【7天入门RTOS网络编程】Young day3-5 作业

发布于 2018-10-25 19:24:09 浏览：1343 订阅该版

* 本帖最后由 爱喝柠檬水 于 2018-10-25 21:10 编辑 *

学习了其他同学的方法使用MSH_CMD_EXPORT()函数将命令函数输出到msh列表便于操作
代码部分
```#include <rtthread.h>
#include <sys/socket.h> /* 使用BSD socket，需要包含socket.h头文件 */
#include <netdb.h>
#include <string.h>
#include <stdio.h>

#define BUFSZ  1024
/* 指向线程控制块的指针 */
static rt_thread_t tid1 = RT_NULL;

static int sock = 0;
char *recv_data;
static char send_data[100];
int ret;

/* tcpClient thread entry*/
static void thread_tcpClient_entry(void* parameter) {
 int bytes_received;
 /* 分配用于存放接收数据的缓冲 */
 recv_data = rt_malloc(BUFSZ);
 if (recv_data == RT_NULL) {
 rt_kprintf("No memory ");
 return;
 }
 while(1) {
 /* 从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);
 rt_thread_delete(tid1);
 } 
 else if (bytes_received == 0) {
 /* 打印recv函数返回值为0的警告信息 */
 closesocket(sock);
 rt_kprintf(" received error,close the socket. ");
 /* 释放接收缓冲 */
 rt_free(recv_data);
 rt_thread_delete(tid1);
 }
 /* 有接收到数据，把末端清零 */
 recv_data[bytes_received] = '\0';
 /* 在控制终端显示收到的数据 */
 rt_kprintf(" %s",recv_data);
 }
}

/* 聊天客户端初始化：新建socket，成功建立连接，并发出初始帧信息，创建接收线程 */
static int tcpClient_Start(void) {
        
        struct hostent *host;
        struct sockaddr_in server_addr;
        const char *url;
        int port;
        url = "183.193.243.90";
        port =5000; 
        /* 通过函数入口参数url获得host地址（如果是域名，会做域名解析） */
        host = gethostbyname(url);

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

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

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

/* 初始化预连接的服务端地址 */
        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 0;
 } else {
 rt_kprintf("Connect success! ");
 }
 
 /* 创建线程1 */
 tid1 = rt_thread_create("tcpClient",
 thread_tcpClient_entry, (void*)1, /* 线程入口是thread_entry, 入口参数是1 */
 1024, 20, 5); //THREAD_STACK_SIZE, THREAD_PRIORITY, THREAD_TIMESLICE 
 if (tid1 != RT_NULL){
 rt_kprintf("tcpclient thread init ok,connect tcp server success ");
 rt_thread_startup(tid1);
 }
 else
 return -1;

return 0;
}

void login(int argc, char **argv) {
 if(argv[1]==RT_NULL)
 {
 rt_kprintf("Please log in Use \"login <nick>\" ");
 return;
 }
 sprintf(send_data, "login %s ", argv[1]);
 /* 发送数据到sock连接 */
 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);
 return;
 } else if (ret == 0) {
 //* 打印send函数返回值为0的警告信息 
 rt_kprintf(" Send warning,send function return 0. ");
 rt_free(recv_data);
 return;
 }
}

void say(int argc, char **argv) {

sprintf(send_data, "say%s ", argv[1]);

/* 发送数据到sock连接 */
 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);
 return;
 } else if (ret == 0) {
 //* 打印send函数返回值为0的警告信息 
 rt_kprintf(" Send warning,send function return 0. ");
 }
}

void look(void) {
 sprintf(send_data, "look ");

/* 发送数据到sock连接 */
 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);
 return;
 } else if (ret == 0) {
 //* 打印send函数返回值为0的警告信息 
 rt_kprintf(" Send warning,send function return 0. ");
 }
}

void who(void) {
 sprintf(send_data, "who ");

/* 发送数据到sock连接 */
 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);
 return;
 } else if (ret == 0) {
 //* 打印send函数返回值为0的警告信息 
 rt_kprintf(" Send warning,send function return 0. ");
 }
}

/* logout server 断开连接 */
void logout(void) {
 sprintf(send_data, "logout ");

/* 发送数据到sock连接 */
 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);
 return;
 } else if (ret == 0) {
 //* 打印send函数返回值为0的警告信息 
 rt_kprintf(" Send warning,send function return 0. ");
 }
 rt_thread_delete(tid1);
 closesocket(sock);
}

MSH_CMD_EXPORT(tcpClient_Start, Start Tcp Client );
MSH_CMD_EXPORT(login, a tcp client login);
MSH_CMD_EXPORT(say, a tcp client say);
MSH_CMD_EXPORT(look, a tcp client look);
MSH_CMD_EXPORT(who, a tcp client who);
MSH_CMD_EXPORT(logout, a tcp client logout);
```

![1png.png](/uploads/201810/25/192146o2in8ano92n288h2.png)![2.png](/uploads/201810/25/192150shu9gwk35w59su19.png)

最后发现几个问题，
1、未使用logout 命令退出会让上一个login的用户也在聊天室，不知道用什么方法注销它。
2、say 命令 发送的字符串遇到空格 服务器不会发送回来空格后面的单词。

以上问题还需认真解决！

say命令问题解决：修改say 命令函数如下
```void say(int argc, char **argv) {
 strcpy(send_data,"say ");
 for(int i=1;i<argc;i++)
 {
 strcat(send_data,argv*);
 strcat(send_data," ");
 }
 strcat(send_data," "); 
 //sprintf(send_data, "say %s ", argv[1]); //原使用sprintf()只能写入argv[1]中字符串，使用strcpy()和strcat()即可解决。

/* 发送数据到sock连接 */
 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);
 return;
 } else if (ret == 0) {
 //* 打印send函数返回值为0的警告信息 
 rt_kprintf(" Send warning,send function return 0. ");
 }
}```各位还有更好的方法也请分享