【网络编程学习】+LoIIs+实现一个简单的TCP聊天客户端

发布于 2018-08-01 18:27:48 浏览：1211 订阅该版

* 本帖最后由 jychal@163.com 于 2018-8-1 18:27 编辑 *

一. 开发环境的搭建
详细教程见：[http://ci.rt-thread.com/3067/site/topics/qemu/setup/](http://ci.rt-thread.com/3067/site/topics/qemu/setup/)

二. 此次试验任务
在服务器 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 
TIPS：认真审题，否则。。。。。:'(

三. 实现过程

基于示例tcpclient.c代码进行修改。创建了一个线程用于指令的发送，废话不多说（没表达能力。。），上代码吧
```/*
 * 程序清单：tcp 客户端
 *
 * 这是一个 tcp 客户端的例程
 * 导出 tcpclient 命令到控制终端
 * 命令调用格式：tcpclient URL PORT
 * URL：服务器地址 PORT:：端口号
 * 程序功能：接收并显示从服务端发送过来的信息，接收到开头是 'q' 或 'Q' 的信息退出程序
*/
#include <rtthread.h>
#include <sys/socket.h> /* 使用BSD socket，需要包含socket.h头文件 */
#include "netdb.h"

#define BUFSZ   1024
#define THREAD_PRIORITY      25
#define THREAD_STACK_SIZE    2048
#define THREAD_TIMESLICE     5

static rt_thread_t tid1 = RT_NULL;
char *myName = "LoI";
int sock,flag=1,inputCnt;
char names[200]={0};
uint32_t tick_Count = 0;

enum steps{LOGIN,SAY_Name,LOOK,WHO,SAY_IDs,SAY_Quit,LOGOUT};
enum steps step = LOGIN;

void SendMessage(void* parameter)
{
        int ret;
    char c,str2Send[100]={0};

if(sock == 0)
 {
 rt_kprintf("Please connect... ");
 return;
 }
 while(1)
 {
 rt_thread_delay(RT_TICK_PER_SECOND);
 if(flag)
 {
 flag = 0;
 switch(step)
 {
 case LOGIN:
 sprintf(str2Send,"login %s",myName);
 rt_kprintf("<COMMAND:%s> ",str2Send);
 flag = 1;step++;
 break;
 case SAY_Name:
 sprintf(str2Send,"say My name is %s,I'm doing this job at 20180801",myName);
 rt_kprintf("<COMMAND:%s> ",str2Send);
 break;
 case LOOK:
 sprintf(str2Send,"look",myName);
 rt_kprintf("<COMMAND:%s> ",str2Send);
 break;
 case WHO:
 sprintf(str2Send,"who",myName);
 rt_kprintf("<COMMAND:%s> ",str2Send);
 break;
 case SAY_IDs:
 sprintf(str2Send,"say I can see %s in the room",names);
 rt_kprintf("<COMMAND:%s> ",str2Send);
 break;
 case SAY_Quit:
 sprintf(str2Send,"say I am quiting");
 rt_kprintf("<COMMAND:%s> ",str2Send);
 break;
 case LOGOUT:
 sprintf(str2Send,"logout");
 rt_kprintf("<COMMAND:%s> ",str2Send);
 break;
 }
 
 strncat(str2Send," ",4);
 ret = send(sock, str2Send, strlen(str2Send), 0);
 if (ret < 0)
 {
 /* 接收失败，关闭这个连接 */
 closesocket(sock);
 rt_kprintf(" send error,close the socket. ");

break;
 }
 else if (ret == 0)
 {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
 rt_memset(str2Send,0,sizeof(str2Send));
 // tick_Count = rt_tick_get(); 
 }
 
 }
}

void StartThread()
{
    tid1 = rt_thread_create("SendMessage",SendMessage,0,
                            THREAD_STACK_SIZE, THREAD_PRIORITY,
                            THREAD_TIMESLICE);
    if(tid1 != RT_NULL)rt_thread_startup(tid1);
}

static const char send_data[] = "This is TCP Client from RT-Thread."; /* 发送用到的数据 */
void tcpclient(int argc, char **argv)
{
    int ret,cnt;
    char *recv_data,*token;
    
    struct hostent *host;
    int bytes_received;
    struct sockaddr_in server_addr;
    const char *url;
    int port;

if (argc < 3)
 {
 rt_kprintf("Usage: tcpclient URL PORT ");
 rt_kprintf("Like: tcpclient 192.168.12.44 5000 ");
 return ;
 }

url = argv[1];
    port = strtoul(argv[2], 0, 10);

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

/* 分配用于存放接收数据的缓冲 */
 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(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;
    }

/*打开发送信息线程*/
    StartThread();

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

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

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

if(!flag)
 {
 switch(step)
 {
 case LOGIN:
 recv_data[bytes_received] = '\0';
 rt_kprintf("%s", recv_data);
 flag = 1;step++;
 break;
 case SAY_Name:
 recv_data[bytes_received] = '\0';
 rt_kprintf("%s", recv_data);
 flag = 1;step++;
 break;
 case LOOK:
 recv_data[bytes_received] = '\0';
 rt_kprintf("%s", recv_data);
 if(NULL!=strstr(recv_data,": "))
 {

}
 else
 {
 token = strtok(recv_data," ");
 while(token!=NULL)
 {
 strcat(names,token);
 strcat(names," ");
 token = strtok(NULL," ");
 }
 flag = 1;step++;
 } 
 break;
 case WHO:
 recv_data[bytes_received] = '\0';
 rt_kprintf("%s", recv_data);
 if(NULL!=strstr(recv_data,": "))
 {

}
 else
 {
 // token = strtok(recv_data," ");
 // while(token!=NULL)
 // {
 // strcat(names,token);
 // strcat(names," ");
 // token = strtok(NULL," ");
 // }
 flag = 1;step++;
 }
 break;
 case SAY_IDs:
 recv_data[bytes_received] = '\0';
 rt_kprintf("%s", recv_data);
 flag = 1;step++;
 break;
 case SAY_Quit:
 recv_data[bytes_received] = '\0';
 rt_kprintf("%s", recv_data);
 flag = 1;step++;
 break;
 case LOGOUT:
 recv_data[bytes_received] = '\0';
 rt_kprintf("%s", recv_data);
 flag = 1;step = LOGIN;
 break;
 default:
 step = LOGIN;
 break;
 }

}
 }
 return;
}
MSH_CMD_EXPORT(tcpclient, a tcp client sample);
```由于审题错误，还做了一个可以直接利用服务器进行聊天的版本，按enter键以进入命令输入模式，大家可以运行一下试试，代码如下：```/*
 * 程序清单：tcp 客户端
 *
 * 这是一个 tcp 客户端的例程
 * 导出 tcpclient 命令到控制终端
 * 命令调用格式：tcpclient URL PORT
 * URL：服务器地址 PORT:：端口号
 * 程序功能：接收并显示从服务端发送过来的信息，接收到开头是 'q' 或 'Q' 的信息退出程序
*/
#include <rtthread.h>
#include <sys/socket.h> /* 使用BSD socket，需要包含socket.h头文件 */
#include "netdb.h"

#define BUFSZ   1024
#define THREAD_PRIORITY      25
#define THREAD_STACK_SIZE    2048
#define THREAD_TIMESLICE     5

static rt_thread_t tid1 = RT_NULL;
int sock,flag=0,inputCnt;
char inputStr[200]={0};
uint32_t tick_Count = 0;

void SendMessage(void* parameter)
{
        int ret;
    char c;

if(sock == 0)
 {
 rt_kprintf("Please connect... ");
 return;
 }
 while(1)
 {
 rt_thread_delay(RT_TICK_PER_SECOND);
 while((c=getchar())!=' ') ;
 loop1: 
 flag = 1;
 rt_kprintf("<Command:");
 inputCnt = 0;
 // rt_memset(inputStr,0,sizeof(inputStr));
 while((inputStr[inputCnt]=getchar())!=' ')
 {
 if(inputStr[inputCnt]=='\b')
 {
 rt_kprintf("%c", inputStr[inputCnt]);
 inputStr[inputCnt] = '\0';
 inputCnt--;
 }
 else
 {
 rt_kprintf("%c", inputStr[inputCnt]);
 inputCnt++;
 }
 // inputStr[inputCnt] = c; 
 }
 if(inputCnt<3)goto loop1;
 // inputStr[inputCnt] = '>';
 flag = 0;
 // rt_kprintf("%c", inputStr[inputCnt]);
 rt_kprintf("> ");
 inputStr[inputCnt] = ' ';
 inputCnt++;
 inputStr[inputCnt] = ' ';

ret = send(sock, inputStr, strlen(inputStr), 0);
 if (ret < 0)
 {
 /* 接收失败，关闭这个连接 */
 closesocket(sock);
 rt_kprintf(" send error,close the socket. ");

break;
 }
 else if (ret == 0)
 {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
 rt_memset(inputStr,0,sizeof(inputStr));
 // tick_Count = rt_tick_get(); 
 }
}

void StartThread()
{
    tid1 = rt_thread_create("SendMessage",SendMessage,0,
                            THREAD_STACK_SIZE, THREAD_PRIORITY,
                            THREAD_TIMESLICE);
    if(tid1 != RT_NULL)rt_thread_startup(tid1);
}

static const char send_data[] = "This is TCP Client from RT-Thread."; /* 发送用到的数据 */
void tcpclient(int argc, char **argv)
{
    int ret;
    char *recv_data;
    
    struct hostent *host;
    int bytes_received;
    struct sockaddr_in server_addr;
    const char *url;
    int port;

if (argc < 3)
 {
 rt_kprintf("Usage: tcpclient URL PORT ");
 rt_kprintf("Like: tcpclient 192.168.12.44 5000 ");
 return ;
 }

url = argv[1];
    port = strtoul(argv[2], 0, 10);

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

/* 分配用于存放接收数据的缓冲 */
 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(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;
    }

/*打开发送信息线程*/
    StartThread();

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

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

/* 释放接收缓冲 */
 rt_free(recv_data);
 break;
 }
 
 /*判断是否在Command输入状态，不在即可以直接输出接收到的内容*/
 if(!flag)
 {
 recv_data[bytes_received] = '\0';
 rt_kprintf("%s", recv_data);
 }
 else
 {
 /*在Command输入时，保证接收数据显示时不会紊乱*/
 for(int i=0; i<inputCnt+9; i++)
 {
 rt_kprintf("\b");
 }
 recv_data[bytes_received] = '\0';
 rt_kprintf("%s", recv_data);
 rt_kprintf("<Command:");
 for(int i=0; i<inputCnt; i++)
 {
 rt_kprintf("%c", inputStr*);
 }
 }
 }
 return;
}
MSH_CMD_EXPORT(tcpclient, a tcp client sample);
```

![微信图片_20180801181407.png](https://oss-club.rt-thread.org/uploads/201808/01/181432i7ji6o7d8kjri8o4.png)

下载附件 [tcpclient.c](https://oss-club.rt-thread.org/uploads/201808/01/182300hz4j387z97j7g4gs.attach)

下载附件 [tcpclient聊天版本.c](https://oss-club.rt-thread.org/uploads/201808/01/182300xafmgeedgdodyspa.attach)

![微信截图_20180801182557.png](https://oss-club.rt-thread.org/uploads/201808/01/182712pifhlvwx9vwhxx5q.png)