UDP的聊天室

 定义用户链表

#include <myhead.h>
#define SER_PORT 8888           //服务器端口号
#define SER_IP "192.168.125.78" //服务器ip地址

//定义一个结构体类型，用于向线程体函数传递参数
struct Info
{
  int newfd;
  struct sockaddr_in cin;
};

struct User
{
  char userName[20];
  struct sockaddr_in cin;
  struct User *next;
} User;

 定义链表添加

// 向链表中添加新用户的函数
void addUserToList(struct User **head, const char *newUserName, struct sockaddr_in newCin)
{
  // 1. 分配内存给新节点
  struct User *newNode = (struct User *)malloc(sizeof(struct User));
  if (newNode == NULL)
  {
    perror("Failed to allocate memory for new user node");
    return; // 内存分配失败，退出函数
  }

  // 2. 填充新节点的数据
  strncpy(newNode->userName, newUserName, sizeof(newNode->userName) - 1); // 注意安全复制字符串
  newNode->userName[sizeof(newNode->userName) - 1] = '\0';                // 确保字符串末尾是'\0'
  newNode->cin = newCin;                                                  // 直接复制sockaddr_in结构
  newNode->next = NULL;                                                   // 新节点的next指针初始化为空

  // 3. 链接到链表末尾
  if (*head == NULL)
  { // 如果链表为空，新节点成为头节点
    *head = newNode;
  }
  else
  {
    struct User *current = *head;
    while (current->next != NULL)
    { // 寻找链表的末尾
      current = current->next;
    }
    current->next = newNode; // 将新节点添加到当前末尾节点之后
  }
}
int compareUsername(struct User *user, char *rbuf)
{
  // 查找rbuf中的\0以确定用户名的结束位置
  char *receivedUserName = strtok(rbuf, "\0"); // 使用\0作为分隔符

  if (receivedUserName == NULL)
  {
    perror("Failed to extract username from received data");
    return -1; // 失败，无法找到用户名
  }

  // 现在比较receivedUserName与user->userName
  if (strcmp(receivedUserName, user->userName) == 0)
  {
    return 1; // 用户名相同
  }
  else
  {
    return 0; // 用户名不同
  }
}

监听客户端

//定义线程体函数
void *deal_cli_msg(void *arg)
{
  //解析传递进来的数据
  int newfd = ((struct Info *)arg)->newfd;
  struct sockaddr_in cin = ((struct Info *)arg)->cin;

  //5、与客户端进行相互通信
  char rbuf[128] = ""; //读取消息内容的容器

  while (1)
  {
    //清空容器
    bzero(rbuf, sizeof(rbuf));

    //从套接字中读取数据
    int res = recvfrom(newfd, rbuf, sizeof(rbuf), 0);
    if (res == 0)
    {
      printf("客户端已经下线\n");
      break;
    }
    //判断一下用户信息链表中是否有该用户，如果有该用户，就发送消息，如果没有就存储用户信息，并且广播用户信息
    if (compareUsername(User, rbuf))
    {
      //TODO:转发消息
      printf("Username matches!\n");
    }
    else
    {
      //TODO:存储用户信息，广播用户登录
      addUserToList(&userList, rbuf, cin); // 添加新用户到链表
      sendto(newfd, rbuf, strlen(rbuf), 0);
      printf("Username does not match.\n");
    }

    //将读取的消息展示出来
    printf("[%s:%d]:%s\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), rbuf);
  }

  //6、关闭套接字
  close(newfd);

  //退出线程
  pthread_exit(NULL);
}

服务端主线程

int main(int argc, const char *argv[])
{
  //1、为通信创建一个端点
  int sfd = socket(AF_INET, SOCK_DGRAM, 0);
  //参数1：说明使用的是ipv4通信域
  //参数2：说明使用的是TCP面向连接的通信方式
  //参数3：由于参数2中已经指定通信方式，填0即可
  if (sfd == -1)
  {
    perror("socket error");
    return -1;
  }
  printf("socket success sfd = %d\n", sfd); //3

  //设置地址能够快速重用
  int reuse = 1;
  if (setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) == -1)
  {
    perror("setsockopt error");
    return -1;
  }

  //2、绑定ip和端口号
  //2.1 准备地址信息结构体
  struct sockaddr_in sin;
  sin.sin_family = AF_INET;                //通信域
  sin.sin_port = htons(SER_PORT);          //端口号
  sin.sin_addr.s_addr = inet_addr(SER_IP); //ip地址

  //2.2 绑定工作
  if (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) == -1)
  {
    perror("bind error");
    return -1;
  }
  printf("bind success\n");

  //3、数据收发

  //准备接受对端的地址信息
  struct sockaddr_in cin;
  socklen_t addrlen = sizeof(cin);

  while (1)
  {
    //定义要传递数据的结构体变量
    struct Info buf = {newfd, cin};

    //创建分支线程，用于通信
    pthread_t tid = -1;
    if (pthread_create(&tid, NULL, deal_cli_msg, &buf) == -1)
    {
      printf("pthread_create error\n");
      return -1;
    }

    //将线程设置成分离态
    pthread_detach(tid);
  }

  close(sfd);

  return 0;
}