【操作体系】Linux之网络编程(UDP)（头歌作业）

第1关：UDP套接字创建与端口绑定

1. #include <sys/types.h>
2. #include <sys/socket.h>
3. #include <arpa/inet.h>
4. #include <stdio.h>
5. #include <errno.h>
6. #include <string.h>
8. /************************
9. * port: 需要绑定的端口号
10. * 返回值: 调用成功返回0，否则返回-1
11. *************************/
12. int UDPSocket(unsigned short port)
13. {
14.         int ret = -1;
15.         /********** BEGIN **********/
16.         int sockfd;
17.     struct sockaddr_in server_addr;
18.     // 创建UDP套接字
19.     sockfd = socket(AF_INET, SOCK_DGRAM, 0);
20.     if (sockfd == -1)
21.     {
22.         printf("创建UDP套接字失败: %s\n", strerror(errno));
23.         return -1;
24.     }
26.     // 初始化服务器地址结构体
27.     memset(&server_addr, 0, sizeof(server_addr));
28.     server_addr.sin_family = AF_INET;
29.     server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
30.     server_addr.sin_port = htons(port);
32.     // 绑定端口
33.     ret = bind(sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr));
34.     if (ret == -1)
35.     {
36.         printf("绑定端口失败: %s\n", strerror(errno));
37.         close(sockfd);
38.         return -1;
39.     }
41.     ret = 0;
42.        
43.         /********** END **********/
44.        
45.         return ret;
46. }

复制代码

第2关：UDP数据传送

1. #include <sys/types.h>
2. #include <sys/socket.h>
3. #include <arpa/inet.h>
4. #include <stdio.h>
5. #include <errno.h>
6. #include <string.h>
7. #include <unistd.h>
9. #define PORT 8888
11. int main()
12. {
13.         int sockfd;
14.         sockfd = socket(AF_INET, SOCK_DGRAM, 0);
15.         if(sockfd == -1)
16.         {
17.                 return -1;
18.         }
19.         struct sockaddr_in addr;
20.         bzero(&addr, sizeof(addr));    //清空
21.         addr.sin_family = AF_INET;
22.         addr.sin_port = htons(PORT);
23.         addr.sin_addr.s_addr = htonl(INADDR_ANY);
24.         //与PORT端口进行绑定
25.         if(bind(sockfd, (struct sockaddr *)&addr, sizeof(addr)) == -1)
26.         {
27.                 return -1;
28.         }
29.        
30.         char data[100];
31.         //接收客户端传来的数据，并打印出来(提示：换行打印)
32.         //同时将接收到的数据原样发送给客户端
33.         //当接收到"exit"字符串时，退出当前程序，不打印出"exit"字符串
34.         //提示：在每次接收字符串前要将存放字符串的变量清空
35.         /********** BEGIN **********/
36.         struct sockaddr_in clientAddr;
37.     socklen_t clientAddrLen = sizeof(clientAddr);
38.     while (1)
39.     {
40.         // 每次接收前清空存放字符串的变量
41.         memset(data, 0, sizeof(data));
43.         ssize_t recvLen = recvfrom(sockfd, data, sizeof(data), 0, (struct sockaddr *)&clientAddr, &clientAddrLen);
44.         if (recvLen == -1)
45.         {
46.             printf("接收数据失败: %s\n", strerror(errno));
47.             continue;
48.         }
49.         else if (recvLen > 0)
50.         {
51.             // 去掉字符串末尾可能的换行符等空白字符
52.             while (data[recvLen - 1] == '\n' || data[recvLen - 1] == '\r' || data[recvLen - 1] == '\t' || data[recvLen - 1] == ' ')
53.             {
54.                 data[recvLen - 1] = '\0';
55.                 recvLen--;
56.             }
58.             if (strcmp(data, "exit") == 0)
59.             {
60.                 break;
61.             }
63.             // 换行打印接收到的数据
64.             printf("%s\n", data);
66.             // 将接收到的数据原样发送给客户端
67.             ssize_t sendLen = sendto(sockfd, data, recvLen, 0, (struct sockaddr *)&clientAddr, clientAddrLen);
68.             if (sendLen == -1)
69.             {
70.                 printf("发送数据失败: %s\n", strerror(errno));
71.             }
72.         }
73.     }
74.        
75.         /********** END **********/
77.         close(sockfd);
78.        
79.         return 0;
80. }

复制代码

第3关：项目实战

1. #include <stdio.h>
2. #include <sys/types.h>
3. #include <sys/socket.h>
4. #include <netinet/in.h>
5. #include <arpa/inet.h>
6. #include <string.h>
7. #include <errno.h>
8. #include <sys/types.h>
9. #include <sys/stat.h>
10. #include <fcntl.h>
11. #include <unistd.h>
13. //定义数据块大小
14. char data[16];
15. //定义服务器端口和服务器地址
16. #define PORT 8889
17. #define SERVER_IP "127.0.0.1"
19. int main(int argc, char *argv[])
20. {
21.         int sockfd;
22.         sockfd = socket(AF_INET, SOCK_DGRAM, 0);
23.         if(sockfd == -1)
24.         {
25.                 return -1;
26.         }
27.         struct sockaddr_in servAddr;
28.         int servAddrLen = sizeof(servAddr);
29.         bzero(&servAddr, sizeof(servAddr));
30.         servAddr.sin_family = AF_INET;
31.         servAddr.sin_port = htons(PORT);
32.         servAddr.sin_addr.s_addr = inet_addr(SERVER_IP);
33.        
34.         //由用户传入的要下载文件的名称
35.         char *downLoadFileName = argv[1];
36.         printf("%s\n", argv[1]);
37.         //先在本地创建要下载的文件
38.         int fd = fd = open(downLoadFileName, O_CREAT | O_WRONLY | O_TRUNC, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
39.         //向服务器发送要上传文件的名称
40.         sendto(sockfd, downLoadFileName, strlen(downLoadFileName), 0, (struct sockaddr *)&servAddr, servAddrLen);
41.        
42.     /********** BEGIN **********/
43.         char ack[16];
44.     int recvLen;
46.     // 循环接收服务器发送的数据块
47.     while (1)
48.     {
49.         // 接收服务器发送的数据块
50.         recvLen = recvfrom(sockfd, data, sizeof(data), 0, (struct sockaddr *)&servAddr, &servAddrLen);
51.         if (recvLen == -1)
52.         {
53.             printf("接收数据失败: %s\n", strerror(errno));
54.             continue;
55.         }
57.         // 根据数据块类型判断是否下载完成
58.         if (data[0] == 'e')
59.         {
60.             // 下载完成，关闭文件并退出程序
61.             close(fd);
62.             break;
63.         }
64.         else if (data[0] == 'c')
65.         {
66.             // 数据块为文件内容，将内容写入本地文件
67.             write(fd, &(data[1]), recvLen - 1);
68.         }
70.         // 向服务器发送接收确认请求
71.         sendto(sockfd, "OK", strlen("OK"), 0, (struct sockaddr *)&servAddr, servAddrLen);
73.         // 清空数据块数据部分（除了类型标识字节）
74.         memset(&(data[1]), 0, sizeof(data) - 1);
75.     }
76.        
77.         /********** END **********/
78.        
79.         close(sockfd);
80.        
81.         return 0;
82. }

复制代码