RT-Thread-【网络编程学习】+ Yuzr + 第一周QEMU的学习总结及实战RT-Thread问答社区

【网络编程学习】+ Yuzr + 第一周QEMU的学习总结及实战

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

* 本帖最后由 Yuzr 于 2018-8-2 22:27 编辑 *

本周学习完QEMU上对RT-THREAD的操作，由于时间有限，只能晚上调调代码。本次上传了前半部分学习总结以及实战的效果，即实现聊天室客户端。时间能力有限，客户端代码写得比较粗糙，以下代码实现客户端基本的通信，结果和原先的要求有一点点不同，不过使用SOCKET的方法应该相似，可以手动指令发送，也可以自动发送。
```/**
  ******************************************************************************
  * @file    tcpclient.c
  * @brief   TCP client source File.
  *
  * @version V0.01
  * @data    day/mon/year
  * @author  Yuzr
  * @note
  *
  *******************************************************************************
  */

/* Includes -------------------------------------------------------------------*/ 
#include <rtthread.h>
#include <sys/socket.h> /* 使用BSD socket，需要包含socket.h头文件 */
#include "netdb.h"
#include "stdio.h"

/* Private types --------------------------------------------------------------*/
typedef enum {
    IDEL,
    LOGIN,
    SAY,
    WHO,
    LOOK,
    LOGOUT
}Staus_TypeDef;

/* Private constants ----------------------------------------------------------*/
#define USE_FISH_TCP 0

#define BUF_SIZE   512

/* instruct data */
static const char login_data[] = "login Yuzr ";
static const char say_data[] = "say I am Yuzr ";
static const char look_data[] = "look ";
static const char who_data[] = "who ";
static const char logout_data[] = "logout ";
/* socket */
int socket_ret;
/* client status */
Staus_TypeDef TcpClientStatus = IDEL;

void tcpclient(int argc, char **argv)
{
    int ret;
    char *Rx_Data1;
    struct hostent *host;
    int bytes_count;
    struct sockaddr_in server_addr;
    const char *url;
    int port;

/* 参数错误 */
 if (argc < 2) {
 rt_kprintf("parameter error ");
 return ;
 }
 
 url = argv[1];
 port = strtoul(argv[2], 0, 10);
 
 /* 通过函数入口参数url获得host地址（如果是域名，会做域名解析） */
 host = gethostbyname(url);

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

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

/* 释放接收缓冲 */
        rt_free(Rx_Data1);
        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(socket_ret, (struct sockaddr *)&server_addr, sizeof(struct sockaddr)) == -1) {
 /* 连接失败 */
 rt_kprintf("Connect fail! ");
 closesocket(socket_ret);

/*释放接收缓冲 */
 rt_free(Rx_Data1);
 return;
 }
 
 bytes_count = recv(socket_ret, Rx_Data1, BUF_SIZE - 1, 0);
 rt_kprintf("%s ",Rx_Data1);
 
 TcpClientStatus = LOGIN;
 
 while (1) {
 #if (USE_FISH_TCP == 0)
 {
 switch (TcpClientStatus){
 case LOGIN:{
 ret = send(socket_ret, login_data, sizeof(login_data)-1, 0);
 if (ret < 0) {
 /* 接收失败，关闭这个连接 */
 closesocket(socket_ret);
 rt_kprintf(" 2 error,close the socket. ");
 return;
 }
 else if (ret == 0) {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
 TcpClientStatus = SAY;
 break;
 }
 
 case SAY:{
 memset(Rx_Data1, 0, BUF_SIZE);
 ret = send(socket_ret, say_data, sizeof(say_data)-1, 0);
 if (ret < 0) {
 /* 接收失败，关闭这个连接 */
 closesocket(socket_ret);
 rt_kprintf(" 3 error,close the socket. ");
 return;
 }
 else if (ret == 0) {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }

ret = recv(socket_ret, Rx_Data1, BUF_SIZE, 0);
 if (ret < 0) {
 /* 接收失败，关闭这个连接 */
 closesocket(socket_ret);
 rt_kprintf(" 4 error,close the socket. ");
 return;
 }
 else if (ret == 0) {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
 rt_kprintf("<command say I am yuzr>%s ",Rx_Data1);
 TcpClientStatus = LOOK;
 break;
 }
 
 case LOOK:{
 memset(Rx_Data1, 0, BUF_SIZE);
 ret = send(socket_ret, look_data, sizeof(look_data)-1, 0);
 if (ret < 0) {
 /* 接收失败，关闭这个连接 */
 closesocket(socket_ret);
 rt_kprintf(" 5 error,close the socket. ");
 return;
 }
 else if (ret == 0) {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
 
 ret = recv(socket_ret, Rx_Data1, BUF_SIZE, 0);
 if (ret < 0) {
 /* 接收失败，关闭这个连接 */
 closesocket(socket_ret);
 rt_kprintf(" 6 error,close the socket. ");
 return;
 }
 else if (ret == 0) {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
 
 rt_kprintf("<command look>%s ",Rx_Data1);
 TcpClientStatus = WHO;
 break;
 }
 
 case WHO:{
 memset(Rx_Data1, 0, BUF_SIZE);
 ret = send(socket_ret, who_data, sizeof(look_data)-1, 0);
 if (ret < 0) {
 /* 接收失败，关闭这个连接 */
 closesocket(socket_ret);
 rt_kprintf(" 8 error,close the socket. ");
 return;
 }
 else if (ret == 0) {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
 
 ret = recv(socket_ret, Rx_Data1, BUF_SIZE, 0);
 if (ret < 0) {
 /* 接收失败，关闭这个连接 */
 closesocket(socket_ret);
 rt_kprintf(" 9 error,close the socket. ");
 return;
 }
 else if (ret == 0) {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
 
 rt_kprintf("<command who>%s ",Rx_Data1);
 TcpClientStatus = LOGOUT;
 break;
 }
 
 case LOGOUT:{
 ret = send(socket_ret, look_data, sizeof(look_data)-1, 0);
 if (ret < 0) {
 /* 接收失败，关闭这个连接 */
 closesocket(socket_ret);
 rt_kprintf(" 11 error,close the socket. ");
 return;
 }
 else if (ret == 0) {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
 TcpClientStatus = IDEL;
 break;
 }
 case IDEL:{
 closesocket(socket_ret);
 rt_kprintf(" close the socket. ");
 return;
 }
 
 }
 }
 #endif
 
 }
 
}

#if (USE_FISH_TCP == 1)
/* 登陆 */
void login(int argc, char **argv)
{
        char send_data[512] = "login ";
        char recei_data[512];
    int ret, dat_len;

if (argc < 2)
 {
 rt_kprintf("Usage: longin ID ");
 rt_kprintf("Like: login skawu ");
 return ;
 }
 
 strcat(&send_data[6], argv[1]);
 dat_len = rt_strlen(send_data);

send_data[dat_len] = ' ';
 send_data[dat_len + 1] = ' ';
 send_data[dat_len + 2] = '\0';
 rt_kprintf("<Command:%s>", send_data);
 ret = send(sock, send_data, dat_len + 2, 0);
 if (ret < 0)
 {
 /* 接收失败，关闭这个连接 */
 closesocket(sock);
 rt_kprintf(" login error,close the socket. ");
 return;
 }
 else if (ret == 0)
 {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
}
/* 发言 */
void say(int argc, char **argv)
{
 char send_data[1024] = "say ";
 char recei_data[512];
 int ret, dat_len = 4, i;

if (argc < 2)
 {
 rt_kprintf("Usage: say content ");
 rt_kprintf("Like: say I am yuzr ");
 return ;
 }

if(rt_strlen(argv[1]) > 1000){
 rt_kprintf("The content is to long, and it should be less than 1000 byte ");
 return;
 }

for(i = 1; i < argc; i++)
 {
 strcat(&send_data[dat_len], argv*);
 dat_len += rt_strlen(argv*);
 send_data[dat_len] = ' ';
 dat_len++; 
 }
 dat_len = rt_strlen(send_data);

send_data[dat_len] = ' ';
 send_data[dat_len + 1] = ' ';
 send_data[dat_len + 2] = '\0';
 rt_kprintf("<Command:%s>", send_data);
 ret = send(sock, send_data, dat_len + 2, 0);
 if (ret < 0)
 {
 /* 接收失败，关闭这个连接 */
 closesocket(sock);
 rt_kprintf(" login error,close the socket. ");
 return;
 }
 else if (ret == 0)
 {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
 
 ret = recv(sock, recei_data, 511, 0);
 recei_data[ret-1]='\0';
 recei_data[ret-2]='\0';
 rt_kprintf("%s ",recei_data);
}

/*  查找 */
void look(int argc, char **argv)
{
        char send_data[1024] = "look";
                char recei_data[512];
        int ret, dat_len;

dat_len = rt_strlen(send_data);
 send_data[dat_len] = ' ';
 send_data[dat_len + 1] = ' ';
 send_data[dat_len + 2] = '\0';
 rt_kprintf("<Command:%s>", send_data);
 ret = send(sock, send_data, dat_len + 2, 0);
 if (ret < 0)
 {
 /* 接收失败，关闭这个连接 */
 closesocket(sock);
 rt_kprintf(" login error,close the socket. ");
 return;
 }
 else if (ret == 0)
 {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
 
 ret = recv(sock, recei_data, 511, 0);
 recei_data[ret-1]='\0';
 recei_data[ret-2]='\0';
 rt_kprintf("%s ",recei_data);
 
}

/* 自己ID */
void who(int argc, char **argv)
{
 char send_data[1024] = "who";
 char recei_data[512];
 int ret, dat_len;
 
 dat_len = rt_strlen(send_data);
 
 send_data[dat_len] = ' ';
 send_data[dat_len + 1] = ' ';
 send_data[dat_len + 2] = '\0';
 rt_kprintf("<Command:%s>", send_data);
 
 ret = send(sock, send_data, dat_len + 2, 0);
 if (ret < 0)
 {
 /* 接收失败，关闭这个连接 */
 closesocket(sock);
 rt_kprintf(" login error,close the socket. ");
 return;
 }
 else if (ret == 0)
 {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
 
 ret = recv(sock, recei_data, 511, 0);
 recei_data[ret-1]='\0';
 recei_data[ret-2]='\0';
 rt_kprintf("%s ",recei_data);
}

/* 退出 */
void logout(int argc, char **argv)
{
 char send_data[1024] = "logout";
 char recei_data[512];
 int ret, dat_len;
 
 dat_len = rt_strlen(send_data);
 
 send_data[dat_len] = ' ';
 send_data[dat_len + 1] = ' ';
 send_data[dat_len + 2] = '\0';
 rt_kprintf("<Command:%s>", send_data);
 
 ret = send(sock, send_data, dat_len + 2, 0);
 if (ret < 0)
 {
 /* 接收失败，关闭这个连接 */
 closesocket(sock);
 rt_kprintf(" login error,close the socket. ");
 return;
 }
 else if (ret == 0)
 {
 /* 打印send函数返回值为0的警告信息 */
 rt_kprintf(" Send warning,send function return 0. ");
 }
}
#endif

#ifdef RT_USING_FINSH
#include <finsh.h>
/* 输出tcpclient函数到finsh shell中 */
MSH_CMD_EXPORT(tcpclient, startup tcp client);
#if (USE_FISH_TCP == 1)
MSH_CMD_EXPORT(login, a tcp client cmmand login);
MSH_CMD_EXPORT(say, a tcp client cmmand say);
MSH_CMD_EXPORT(look, a tcp client cmmand look);
MSH_CMD_EXPORT(who, a tcp client cmmand who);
MSH_CMD_EXPORT(logout, a tcp client cmmand logout);
#endif
#endif
```

下载附件 [rt-thread学习-网络编程基础总结.pdf](https://oss-club.rt-thread.org/uploads/201808/02/004751fm103mimew8qommt.attach)