diff --git a/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/SConscript b/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/SConscript index 7bb0b8524..0c762e3fe 100644 --- a/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/SConscript +++ b/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/SConscript @@ -10,6 +10,11 @@ CPPPATH = [cwd] if GetDepend('BSP_USING_LCD'): src += ['lcd_test.c'] +## 设置 tcp_client.c 和 tcp_server.c 的依赖宏 +if GetDepend('PKG_USING_WIZNET'): + src += ['tcp_client.c'] + src += ['tcp_server.c'] + group = DefineGroup('Applications', src, depend = [''], CPPPATH = CPPPATH) Return('group') diff --git a/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/tcp_client.c b/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/tcp_client.c new file mode 100644 index 000000000..f5548502d --- /dev/null +++ b/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/tcp_client.c @@ -0,0 +1,144 @@ +/* + * 程序清单:tcp 客户端 + * + * 这是一个 tcp 客户端的例程 + * 导出 tcpclient 命令到控制终端 + * 命令调用格式:tcpclient URL PORT + * URL:服务器地址 PORT::端口号 + * 程序功能:接收并显示从服务端发送过来的信息,接收到开头是 'q' 或 'Q' 的信息退出程序 + * Created by Ybeichen on 2022/7/28. +*/ + +#include "tcp_client.h" +#include +#include /* 使用BSD socket,需要包含socket.h头文件 */ +#include +#include + +#define BUFSZ 1024 + +static const char send_data[] = "This is TCP Client from RT-Thread."; /* 发送用到的数据 */ +static void tcp_client(int argc, char **argv) +{ + int ret; + char *recv_data; + struct hostent *host; + int sock, bytes_received; + struct sockaddr_in server_addr; + const char *url; + int port; + + if (argc < 3) + { + rt_kprintf("Usage: tcp_client URL PORT\n"); + rt_kprintf("Like: tcp_client 192.168.12.44 5000\n"); + 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\n"); + return; + } + + /* 创建一个socket,类型是SOCKET_STREAM,TCP类型 */ + if ((sock = socket(AF_WIZ, SOCK_STREAM, 0)) == -1) + { + /* 创建socket失败 */ + rt_kprintf("Socket error\n"); + + /* 释放已分配的接收缓冲 */ + rt_free(recv_data); + return; + } + + /* 初始化预连接的服务端地址 */ + server_addr.sin_family = AF_WIZ; + 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!\n"); + closesocket(sock); + + /*释放接收缓冲 */ + rt_free(recv_data); + return; + } + + while (1) + { + /* 从sock连接中接收最大BUFSZ - 1字节数据 */ + bytes_received = recv(sock, recv_data, BUFSZ - 1, 0); + if (bytes_received < 0) + { + /* 接收失败,关闭这个连接 */ + closesocket(sock); + rt_kprintf("\nreceived error,close the socket.\r\n"); + + /* 释放接收缓冲 */ + rt_free(recv_data); + break; + } + else if (bytes_received == 0) + { + /* 默认 recv 为阻塞模式,此时收到0认为连接出错,关闭这个连接 */ + closesocket(sock); + rt_kprintf("\nreceived error,close the socket.\r\n"); + + /* 释放接收缓冲 */ + rt_free(recv_data); + break; + } + + /* 有接收到数据,把末端清零 */ + recv_data[bytes_received] = '\0'; + + if (strncmp(recv_data, "q", 1) == 0 || strncmp(recv_data, "Q", 1) == 0) + { + /* 如果是首字母是q或Q,关闭这个连接 */ + closesocket(sock); + rt_kprintf("\n got a 'q' or 'Q',close the socket.\r\n"); + + /* 释放接收缓冲 */ + rt_free(recv_data); + break; + } + else + { + /* 在控制终端显示收到的数据 */ + rt_kprintf("\nReceived data = %s ", recv_data); + } + + /* 发送数据到sock连接 */ + ret = send(sock, send_data, strlen(send_data), 0); + if (ret < 0) + { + /* 接收失败,关闭这个连接 */ + closesocket(sock); + rt_kprintf("\nsend error,close the socket.\r\n"); + + rt_free(recv_data); + break; + } + else if (ret == 0) + { + /* 打印send函数返回值为0的警告信息 */ + rt_kprintf("\n Send warning,send function return 0.\r\n"); + } + } + return; +} +MSH_CMD_EXPORT(tcp_client, a tcp client sample); \ No newline at end of file diff --git a/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/tcp_client.h b/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/tcp_client.h new file mode 100644 index 000000000..6190811cc --- /dev/null +++ b/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/tcp_client.h @@ -0,0 +1,10 @@ +// +// Created by Y北辰 on 2022/7/28. +// + +#ifndef RT_THREAD_FUSION_XIUOS_TCP_CLIENT_H +#define RT_THREAD_FUSION_XIUOS_TCP_CLIENT_H + +#endif //RT_THREAD_FUSION_XIUOS_TCP_CLIENT_H + +static void tcpclient(int argc, char **argv); \ No newline at end of file diff --git a/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/tcp_server.c b/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/tcp_server.c new file mode 100644 index 000000000..287881827 --- /dev/null +++ b/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/tcp_server.c @@ -0,0 +1,174 @@ +/* +* 程序清单:tcp 服务端 + * + * 这是一个 tcp 服务端的例程 + * 导出 tcp_server 命令到控制终端 + * 命令调用格式:tcp_server + * 无参数 + * 程序功能:作为一个服务端,接收并显示客户端发来的数据 ,接收到 exit 退出程序 + * Created by Ybeichen on 2022/7/28. +*/ + +#include "tcp_server.h" +#include +#include /* 使用BSD socket,需要包含socket.h头文件 */ +#include "netdb.h" +#include +#include + +#define BUFSZ (2048) +static int port = 5000; +static int is_running = 0; /* 停止标志 */ +static const char send_data[] = "This is TCP Server from RT-Thread."; /* 发送用到的数据 */ + +static void tcp_server(void *argr) +{ + char *recv_data; /* 用于接收的指针,后面会做一次动态分配以请求可用内存 */ + socklen_t sin_size; + int sock, connected, bytes_received; + struct sockaddr_in server_addr, client_addr; + int ret; + + recv_data = rt_malloc(BUFSZ + 1); /* 分配接收用的数据缓冲 */ + if (recv_data == RT_NULL) + { + rt_kprintf("No memory\n"); + return; + } + + /* 创建一个socket,类型是SOCKET_STREAM,TCP类型 */ + if ((sock = socket(AF_WIZ, SOCK_STREAM, IPPROTO_TCP)) == -1) + { + /* 创建socket失败 */ + rt_kprintf("Socket error\n"); + + /* 释放已分配的接收缓冲 */ + rt_free(recv_data); + return; + } + + /* 初始化本地服务端地址 */ + server_addr.sin_family = AF_WIZ; + server_addr.sin_port = htons(port); /* 服务端工作的端口 */ + server_addr.sin_addr.s_addr = INADDR_ANY; + rt_memset(&(server_addr.sin_zero), 0, sizeof(server_addr.sin_zero)); + + /* 绑定socket到服务端地址 */ + if (bind(sock, (struct sockaddr *)&server_addr, sizeof(struct sockaddr)) == -1) + { + /* 绑定失败 */ + rt_kprintf("Unable to bind\n"); + + /* 释放已分配的接收缓冲 */ + rt_free(recv_data); + return; + } + + /* 在socket上进行监听 */ + if (listen(sock, 5) == -1) + { + rt_kprintf("Listen error\n"); + + /* release recv buffer */ + rt_free(recv_data); + return; + } + + rt_kprintf("\nTCPServer Waiting for client on port %d...\n", port); + + is_running=1; + while (is_running) + { + + /* 接受一个客户端连接socket的请求,这个函数调用是阻塞式的 */ + connected = accept(sock, (struct sockaddr *)&client_addr, (socklen_t *)sizeof(struct sockaddr_in)); + /* 返回的是连接成功的socket */ + if (connected < 0) + { + rt_kprintf("accept connection failed! errno = %d\n", errno); + /* release recv buffer */ + rt_free(recv_data); + continue; + } + + /* 接受返回的client_addr指向了客户端的地址信息 */ + rt_kprintf("I got a connection from (%s , %d)\n", + inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port)); + + /* 客户端连接的处理 */ + while (is_running) + { + /* 发送数据到connected socket */ + ret = send(connected, send_data, strlen(send_data), 0); + if (ret < 0) + { + /* 发送失败,关闭这个连接 */ + rt_kprintf("\nsend error,close the socket.\r\n"); + closesocket(connected); + /* 释放接收缓冲 */ + rt_free(recv_data); + break; + } + else if (ret == 0) + { + /* 打印send函数返回值为0的警告信息 */ + rt_kprintf("\n Send warning,send function return 0.\r\n"); + + } + + /* 从connected socket中接收数据,接收buffer是1024大小,但并不一定能够收到1024大小的数据 */ + bytes_received = recv(connected, recv_data, BUFSZ, 0); + if (bytes_received < 0) + { + /* 接收失败,关闭这个connected socket */ + rt_kprintf("\nReceived error, close the connect.\r\n"); + closesocket(connected); + /* 释放接收缓冲 */ + rt_free(recv_data); + break; + } + else if (bytes_received == 0) + { + /* 打印recv函数返回值为0的警告信息 */ + rt_kprintf("\nReceived warning,recv function return 0.\r\n"); + closesocket(connected); + /* 释放接收缓冲 */ + rt_free(recv_data); + break; + } + + /* 有接收到数据,把末端清零 */ + recv_data[bytes_received] = '\0'; + if (strncmp(recv_data, "q", 1) == 0 || strncmp(recv_data, "Q", 1) == 0) + { + /* 如果是首字母是q或Q,关闭这个连接 */ + rt_kprintf("\nGot a 'q' or 'Q', close the connect.\r\n"); + closesocket(connected); + /* 释放接收缓冲 */ + rt_free(recv_data); + break; + } + else if (strcmp(recv_data, "exit") == 0) + { + /* 如果接收的是exit,则关闭整个服务端 */ + closesocket(connected); + is_running=0; + break; + } + else + { + /* 在控制终端显示收到的数据 */ + rt_kprintf("RECEIVED DATA = %s \n", recv_data); + } + } + } + + /* 退出服务 */ + closesocket(sock); + + /* 释放接收缓冲 */ + rt_free(recv_data); + + return; +} +MSH_CMD_EXPORT(tcp_server, a tcp server sample); \ No newline at end of file diff --git a/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/tcp_server.h b/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/tcp_server.h new file mode 100644 index 000000000..2a79578b4 --- /dev/null +++ b/Ubiquitous/RT-Thread_Fusion_XiUOS/aiit_board/xidatong-riscv64/applications/tcp_server.h @@ -0,0 +1,10 @@ +// +// Created by Y北辰 on 2022/7/28. +// + +#ifndef RT_THREAD_FUSION_XIUOS_TCP_SERVER_H +#define RT_THREAD_FUSION_XIUOS_TCP_SERVER_H + +#endif //RT_THREAD_FUSION_XIUOS_TCP_SERVER_H + +static void tcp_server(void *argr); \ No newline at end of file