RT-Thread-【网络编程学习】+ 小燕 + TCP聊天客户端RT-Thread问答社区

【网络编程学习】+ 小燕 + TCP聊天客户端

发布于 2018-08-02 00:15:34 浏览：1314 订阅该版

```/*
 * 程序清单：一个tcp聊天客户端
 *
 在服务器 183.193.243.90 的 11222 端口 上运行了一个基于 TCP 的简单的聊天服务器 ；
 通过编程实现：
 1. 使用 login 命令登录上服务器聊天室，并将服务器返回的信息通过串口打印出来
 2. 使用 say 命令通知服务器你的个人姓名和作业时间 “My name is Yqiu, I'm doing this job at 20180730”，并
 将服务器返回的信息打印出来
 3. 使用 look 命令查看聊天室人员，并将服务器返回的信息打印出来
 4. 使用 who 命令查看我是谁，并将服务器返回的信息打印出来
 5. 使用 say 命令告诉服务器我看到有哪些人在聊天室
 6. 使用 say 命令告诉服务器我准备退出了
 7. 使用 logout 退出聊天室	参考 Log 信息：
 Welcome to TestChat
 <Command: Hello, world> Please log in Use "login "
 <Command: login yqiu>
 <Command: say What are you doing?> yqiu: What are you doing?
 <Command: look> The following are in this room: yqiu
 <Command: who> The following are logged in: yqiu
 <Command: say I can see yqiu in the room> yqiu: I can see yqiu in the room
 <Command: I am quiting> yqiu: I am quiting 
*/
#include <rtthread.h>
#include <sys/socket.h> /* 使用BSD socket，需要包含socket.h头文件 */
#include "netdb.h"
#include <stdio.h>

#define BUFSZ                   1024
#define TCP_SERVER_IP           "183.193.243.90"
#define TCP_SERVER_PORT         11222

typedef enum {
	TCP_CONNECT,
	LOGIN,
	IDENTIFY,
	CHECKMEMBER,
	WHOAMI,
	TELLMEMBER,
	TELLQUIT,
    LOGOUT,
    STEP_MAX
}_CHAT_STEP;

#define CHAT_CMD_LOGIN          "login" // 登录上服务器聊天室
#define CHAT_CMD_SAY            "say"   // 向服务器发送信息
#define CHAT_CMD_LOOK           "look"  // 查看聊天室人员
#define CHAT_CMD_WHO            "who"   // 查看我是谁
#define CHAT_CMD_LOGOUT         "logout"// 退出聊天室

void tcpclientchat(int argc, char **argv)
{
 int ret;
 char *recv_data;
 struct hostent *host;
 int sock, bytes_received;
 struct sockaddr_in server_addr;
 const char *name;
 _CHAT_STEP chat_step = LOGIN;
	char send_data[BUFSZ] = {0};
	char *members = NULL;
	
 if (argc < 2)
 {
 rt_kprintf("Usage: login name ");
 return ;
 }

name = argv[1];

/* 通过函数入口参数url获得host地址（如果是域名，会做域名解析） */
    host = gethostbyname(TCP_SERVER_IP);

/* 分配用于存放接收数据的缓冲 */
 recv_data = rt_malloc(BUFSZ);
 if (recv_data == RT_NULL)
 {
 rt_kprintf("No memory ");
 return;
 }

/* 创建一个socket，类型是SOCKET_STREAM，TCP类型 */
 if ((sock = socket(AF_INET, SOCK_STREAM, 0)) == -1)
 {
 /* 创建socket失败 */
 rt_kprintf("Socket error ");

/* 释放接收缓冲 */
        rt_free(recv_data);
        return;
    }

/* 初始化预连接的服务端地址 */
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(TCP_SERVER_PORT);
    server_addr.sin_addr = *((struct in_addr *)host->h_addr);
    rt_memset(&(server_addr.sin_zero), 0, sizeof(server_addr.sin_zero));

/* 连接到服务端 */
 if (connect(sock, (struct sockaddr *)&server_addr, sizeof(struct sockaddr)) == -1)
 {
 /* 连接失败 */
 rt_kprintf("Connect fail! ");
 closesocket(sock);

/*释放接收缓冲 */
        rt_free(recv_data);
        return;
    }

while (1)
 {
 rt_kprintf("recving ");
 /* 从sock连接中接收最大BUFSZ - 1字节数据 */
 bytes_received = recv(sock, recv_data, BUFSZ - 1, 0);
 if (bytes_received < 0)
 {
 /* 接收失败，关闭这个连接 */
 closesocket(sock);
 rt_kprintf(" received error,close the socket. ");

/* 释放接收缓冲 */
 rt_free(recv_data);
 break;
 }
 else if (bytes_received == 0)
 {
 /* 默认 recv 为阻塞模式，此时收到0认为连接出错，关闭这个连接 */
 closesocket(sock);
 rt_kprintf(" received error,close the socket. ");

/* 释放接收缓冲 */
            rt_free(recv_data);
            break;
        }

/* 有接收到数据，把末端清零 */
 recv_data[bytes_received] = '\0';
		/* 在控制终端显示收到的数据 */
		rt_kprintf(" %s ", recv_data);
		
		switch (chat_step) {
 case TCP_CONNECT:
				chat_step++;
				break;
				
			case LOGIN:
				// 使用 login 命令登录上服务器聊天室
				sprintf(send_data, "%s %s ", CHAT_CMD_LOGIN, name);
				chat_step++;
				break;

case IDENTIFY:
				// 通知服务器你的个人姓名和作业时间 
				sprintf(send_data, "%s my name is %s, i'm doing this job at 20180801 ", CHAT_CMD_SAY, name);
				chat_step++;
				break;

case CHECKMEMBER:
				sprintf(send_data, "%s ", CHAT_CMD_LOOK);
				chat_step++;
				break;

case WHOAMI:
				// 记录哪些人在聊天室
				members = rt_malloc(bytes_received);
				if (NULL == members) {
 rt_kprintf("no memory. ");
					break;
				}
				memcpy(members, recv_data, bytes_received);
				sprintf(send_data, "%s ", CHAT_CMD_WHO);
				chat_step++;
				break;

case TELLMEMBER:
				sprintf(send_data, "%s I can see %s in this room. ", CHAT_CMD_SAY, members);
				if (NULL != members) {
					rt_free(members);
				}
				chat_step++;
				break;

case TELLQUIT:
				sprintf(send_data, "%s I'm quiting. ", CHAT_CMD_SAY);
				chat_step++;
				break;

case LOGOUT:
				sprintf(send_data, "%s ", CHAT_CMD_LOGOUT);
				chat_step++;
				break;

default:
				chat_step = TCP_CONNECT;
				break;
		}

/* 发送数据到sock连接 */
		if (0 == strlen(send_data)) {
 continue;
		}
		rt_kprintf("%s ", send_data);
 ret = send(sock, send_data, strlen(send_data), 0);
 if (ret < 0)
 {
 /* 接收失败，关闭这个连接 */
 closesocket(sock);
 rt_kprintf(" send error,close the socket. ");

rt_free(recv_data);
 break;
 }
 else if (ret == 0)
 {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
		memset(send_data, 0, BUFSZ);
		memset(recv_data, 0, BUFSZ);
		if (chat_step >= LOGOUT) {
			closesocket(sock);
			break;
		}
 }
 return;
}
MSH_CMD_EXPORT(tcpclientchat, a tcp client chat);

```

下载附件 [网络编程学习笔记-tcp聊天客户端.docx](https://oss-club.rt-thread.org/uploads/201808/02/001530me17pj711z1zmy77.attach)