udpserver_sample.c

/* * Copyright (c) 2006-2018, RT-Thread Development Team * * SPDX-License-Identifier: Apache-2.0 * * Change Logs: * Date Author Notes * *//** 程序清单：udp 服务端 * * 这是一个 udp 服务端的例程 * 导出 udpserv 命令到控制终端 * 命令调用格式：udpserv * 无参数 * 程序功能：作为一个服务端，接收并显示客户端发来的数据 ，接收到 exit 退出程序*/#include <rtthread.h>#include <sys/socket.h> /* 使用BSD socket，需要包含socket.h头文件 */#include <netdb.h>#include <string.h>#include <finsh.h>#define BUFSZ 1024static void udpserv(int argc, char **argv){ int sock; int bytes_read; char *recv_data; socklen_t addr_len; struct sockaddr_in server_addr, client_addr; /* 分配接收用的数据缓冲 */ recv_data = rt_malloc(BUFSZ); if (recv_data == RT_NULL) { /* 分配内存失败，返回 */ rt_kprintf("No memory\n"); return; } /* 创建一个socket，类型是SOCK_DGRAM，UDP类型 */ if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == -1) { rt_kprintf("Socket error\n"); /* 释放接收用的数据缓冲 */ rt_free(recv_data); return; } /* 初始化服务端地址 */ server_addr.sin_family = AF_INET; server_addr.sin_port = htons(5000); 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("Bind error\n"); /* 释放接收用的数据缓冲 */ rt_free(recv_data); return; } addr_len = sizeof(struct sockaddr); rt_kprintf("UDPServer Waiting for client on port 5000...\n"); while (1) { /* 从sock中收取最大BUFSZ - 1字节数据 */ bytes_read = recvfrom(sock, recv_data, BUFSZ - 1, 0, (struct sockaddr *)&client_addr, &addr_len); /* UDP不同于TCP，它基本不会出现收取的数据失败的情况，除非设置了超时等待 */ recv_data[bytes_read] = '\0'; /* 把末端清零 */ /* 输出接收的数据 */ rt_kprintf("\n(%s , %d) said : ", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port)); rt_kprintf("%s", recv_data); /* 如果接收数据是exit，退出 */ if (strcmp(recv_data, "exit") == 0) { closesocket(sock); /* 释放接收用的数据缓冲 */ rt_free(recv_data); break; } } return;}MSH_CMD_EXPORT(udpserv, a udp server sample);