• UDP聊天室项目

    代码思路

    服务器

    1. #include 
    2. #include  /* See NOTES */
    3. #include 
    4. #include 
    5. #include 
    6. #include 
    7. #include 
    8. #include 
    9. #include 
    10. #include 
    11. #include 
    12.  
    13. //链表节点结构体：
    14. typedef struct node
    15. {
    16.     struct sockaddr_in addr; //data   memcmp
    17.     struct node *next;
    18. } link_t;
    19.  
    20. //类型
    21. enum type_t
    22. {
    23.     login,
    24.     chat,
    25.     quit,
    26. };
    27.  
    28. //消息对应的结构体(同一个协议)
    29. typedef struct msg_t
    30. {
    31.     int type;       //'L' C  Q    enum un{login,chat,quit};
    32.     char name[32];  //用户名
    33.     char text[128]; //消息正文
    34. } MSG_t;
    35.  
    36. link_t *link_creat();
    37. void client_login(int sockfd, link_t *p, MSG_t msg, struct sockaddr_in caddr);
    38. void client_quit(int sockfd, link_t *p, MSG_t msg, struct sockaddr_in caddr);
    39. void client_chat(int sockfd, link_t *p, MSG_t msg, struct sockaddr_in caddr);
    40.  
    41. int main(int argc, char const *argv[])
    42. {
    43.     // 创建套接字
    44.     int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    45.     if (sockfd < 0)
    46.     {
    47.         perror("socket err");
    48.         return -1;
    49.     }
    50.     printf("sockfd:%d\n", sockfd);
    51.     // 绑定
    52.     struct sockaddr_in saddr, caddr;
    53.     saddr.sin_family = AF_INET;
    54.     saddr.sin_port = htons(atoi(argv[1]));
    55.     saddr.sin_addr.s_addr = INADDR_ANY;
    56.     socklen_t len = sizeof(caddr);
    57.     if (bind(sockfd, (struct sockaddr *)&saddr, sizeof(saddr)) < 0)
    58.     {
    59.         perror("bind err");
    60.         return -1;
    61.     }
    62.     printf("bind ok\n");
    63.     // 创建链表
    64.     link_t *p = link_creat();
    65.     printf("creat ok\n");
    66.     MSG_t msg;
    67.     // 可扩展功能，服务器可以给所有用户发送“公告”
    68.     pid_t pid = fork();
    69.     if (pid < 0)
    70.     {
    71.         perror("fork err");
    72.         return -1;
    73.     }
    74.     else if (pid == 0)
    75.     {
    76.         while (1)
    77.         {
    78.  
    79.             fgets(msg.text, sizeof(msg.text), stdin);
    80.             if (msg.text[strlen(msg.text) - 1] == '\n')
    81.                 msg.text[strlen(msg.text) - 1] = '\0';
    82.             msg.type = chat;
    83.             strcpy(msg.name, "server");
    84.             sendto(sockfd, &msg, sizeof(MSG_t), 0, (struct sockaddr *)&saddr, sizeof(saddr));
    85.         }
    86.     }
    87.     else
    88.     {
    89.         while (1)
    90.         {
    91.             if (recvfrom(sockfd, &msg, sizeof(msg), 0, (struct sockaddr *)&caddr, &len) < 0)
    92.             {
    93.                 perror("recvfrom err");
    94.                 return -1;
    95.             }
    96.             printf("type:%d\n", msg.type);
    97.             switch (msg.type)
    98.             {
    99.             case login: // 上线函数
    100.                 client_login(sockfd, p, msg, caddr);
    101.                 break;
    102.             case chat: // 聊天函数
    103.                 client_chat(sockfd, p, msg, caddr);
    104.                 break;
    105.             case quit: // 下线函数
    106.                 client_quit(sockfd, p, msg, caddr);
    107.                 break;
    108.             default:
    109.                 break;
    110.             }
    111.         }
    112.     }
    113.  
    114.     close(sockfd);
    115.  
    116.     return 0;
    117. }
    118. // 创建链表
    119. link_t *link_creat()
    120. {
    121.     // 给头结点开辟空间
    122.     link_t *p = (link_t *)malloc(sizeof(link_t));
    123.     if (p == NULL)
    124.     {
    125.         perror("malloc err");
    126.         return NULL;
    127.     }
    128.     // 初始化头结点
    129.     p->next = NULL;
    130.     return p;
    131. }
    132.  
    133. // 客户端上线功能：将用户上线消息发送给其他用户，并将用户信息添加到链表中
    134. void client_login(int sockfd, link_t *p, MSG_t msg, struct sockaddr_in caddr)
    135. {
    136.     printf("%s  login....\n", msg.name);
    137.     // 先遍历链表
    138.     // 再插入新的节点
    139.     while (p->next != NULL)
    140.     {
    141.         p = p->next;
    142.         sendto(sockfd, &msg, sizeof(MSG_t), 0, (struct sockaddr *)&(p->addr), sizeof(p->addr));
    143.     }
    144.     link_t *pnew = (link_t *)malloc(sizeof(link_t));
    145.     pnew->addr = caddr;
    146.     pnew->next = NULL;
    147.     p->next = pnew;
    148.  
    149.     return;
    150. }
    151. // 聊天功能：转发消息到在链表中除了自己的所有用户
    152. void client_chat(int sockfd, link_t *p, MSG_t msg, struct sockaddr_in caddr)
    153. {
    154.     printf("%s says %s\n", msg.name, msg.text);
    155.     while (p->next != NULL)
    156.     {
    157.         p = p->next;
    158.         // strcmp:对比前后两个字符串中的内容是否一致
    159.         // memcmp:对比前后两个对应的地址(任意类型)的内容是否一致
    160.         if (memcmp(&(p->addr), &caddr, sizeof(caddr)))
    161.         {
    162.             sendto(sockfd, &msg, sizeof(MSG_t), 0, (struct sockaddr *)&(p->addr), sizeof(p->addr));
    163.         }
    164.     }
    165.  
    166.     return;
    167. }
    168. // 客户端退出功能：将用户下线消息发送给其他用户，将下线用户从链表中删除
    169. void client_quit(int sockfd, link_t *p, MSG_t msg, struct sockaddr_in caddr)
    170. {
    171.     printf("%s quit.......\n", msg.name);
    172.     link_t *pdel = NULL;
    173.     while (p->next != NULL)
    174.     {
    175.         if (memcmp(&(p->next->addr), &caddr, sizeof(caddr)))
    176.         {
    177.             // 不是自己
    178.             p = p->next;
    179.             sendto(sockfd, &msg, sizeof(MSG_t), 0, (struct sockaddr *)&(p->addr), sizeof(p->addr));
    180.         }
    181.         else
    182.         {
    183.             // 是自己,删除节点
    184.             pdel = p->next;
    185.             p->next = pdel->next;
    186.             free(pdel);
    187.             pdel = NULL;
    188.         }
    189.     }
    190.     return;
    191. }

    客户端

    1. #include 
    2. #include  /* See NOTES */
    3. #include 
    4. #include 
    5. #include 
    6. #include 
    7. #include 
    8. #include 
    9. #include 
    10. #include 
    11. #include 
    12.  
    13. //链表节点结构体：
    14. typedef struct node
    15. {
    16.     struct sockaddr_in addr; //data   memcmp
    17.     struct node *next;
    18. } link_t;
    19.  
    20. //类型
    21. enum type_t
    22. {
    23.     login,
    24.     chat,
    25.     quit,
    26. };
    27.  
    28. //消息对应的结构体(同一个协议)
    29. typedef struct msg_t
    30. {
    31.     int type;       //'L' C  Q    enum un{login,chat,quit};
    32.     char name[32];  //用户名
    33.     char text[128]; //消息正文
    34. } MSG_t;
    35.  
    36. int main(int argc, char const *argv[])
    37. {
    38.     int ret;
    39.     // 创建套接字
    40.     int sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    41.     if (sockfd < 0)
    42.     {
    43.         perror("socket err");
    44.         return -1;
    45.     }
    46.     printf("sockfd:%d\n", sockfd);
    47.     // 绑定
    48.     struct sockaddr_in saddr, caddr;
    49.     saddr.sin_family = AF_INET;
    50.     saddr.sin_port = htons(atoi(argv[2]));
    51.     saddr.sin_addr.s_addr = inet_addr(argv[1]);
    52.     socklen_t len = sizeof(caddr);
    53.  
    54.     MSG_t msg;
    55.     // 给服务器发送上线消息
    56.     printf("请输入用户名:");
    57.     fgets(msg.name, sizeof(msg.name), stdin);
    58.     if (msg.name[strlen(msg.name) - 1] == '\n')
    59.         msg.name[strlen(msg.name) - 1] = '\0';
    60.     msg.type = login;
    61.     sendto(sockfd, &msg, sizeof(MSG_t), 0, (struct sockaddr *)&saddr, sizeof(saddr));
    62.  
    63.     pid_t pid = fork();
    64.     if (pid < 0)
    65.     {
    66.         perror("fork err");
    67.         return -1;
    68.     }
    69.     else if (pid == 0) // 循环接受服务器消息
    70.     {
    71.         while (1)
    72.         {
    73.             ret = recvfrom(sockfd, &msg, sizeof(MSG_t), 0, NULL, NULL);
    74.             if (ret < 0)
    75.             {
    76.                 perror("recv from err");
    77.                 return -1;
    78.             }
    79.             if (msg.type == login)
    80.                 printf("%s login.......\n", msg.name);
    81.             else if (msg.type == chat)
    82.                 printf("%s says %s\n", msg.name, msg.text);
    83.             else if (msg.type == quit)
    84.                 printf("%s quit.......\n", msg.name);
    85.         }
    86.     }
    87.     else // 循环发送消息
    88.     {
    89.         while (1)
    90.         {
    91.             fgets(msg.text, sizeof(msg.text), stdin);
    92.             if (msg.text[strlen(msg.text) - 1] == '\n')
    93.                 msg.text[strlen(msg.text) - 1] = '\0';
    94.             if (!strcmp(msg.text, "quit"))
    95.             {
    96.                 msg.type = quit;
    97.                 sendto(sockfd, &msg, sizeof(MSG_t), 0, (struct sockaddr *)&saddr, sizeof(saddr));
    98.                 break;
    99.             }
    100.             else
    101.             {
    102.                 msg.type = chat;
    103.                 sendto(sockfd, &msg, sizeof(MSG_t), 0, (struct sockaddr *)&saddr, sizeof(saddr));
    104.             }
    105.         }
    106.         kill(pid, SIGINT);
    107.         wait(NULL);
    108.     }
    109.     close(sockfd);
    110.     return 0;
    111. }
  • 相关阅读:
    如何开通腾讯云Redis云数据库并创建连接？
    Unity Metaverse（三）、Protobuf & Socket 实现多人在线
    shell语法（一）
    切糕 小白月赛45
    大学生游戏静态HTML网页设计 (HTML+CSS+JS仿英雄联盟网站15页)
    【云原生丨Docker系列11】Docker Machine 操作详解
    自定义Kotlin协程调度器
    RSS订阅
    基于SSM的人才招聘网站的研究与实现
    关于redis的一主三从三哨兵的实现
  • 原文地址：https://blog.csdn.net/qq_64136247/article/details/142262423