基于UDP通信的聊天室
项目要求
1、有用户登录，其他用户可以收到这个人的登录信息
2、如果有人发送信息，其他用户可以收到这个人的群聊信息
3、如果有人下线，其他用户可以收到这个人的下线信息
4、服务器发送的消息所有用户可以收到
服务器代码
#include <myhead.h>
#include <poll.h>
#define IP "192.168.60.117"
#define PORT 8888
typedef struct
{char type;char name[100];char text[100];
}msg;// 在文件开头添加函数声明
int find_client_index(msg *e, int count, const char *name) {for(int i = 0; i < count; i++) {if(strcmp(e[i].name, name) == 0) {return i;}}return -1;
}int main(int argc, const char *argv[])
{//创建UDP套接字int oldfd =socket(AF_INET,SOCK_DGRAM,0);if(oldfd==-1){perror("socket");return -1;}//将服务器地址信息填充结构体struct sockaddr_in server={.sin_family=AF_INET,.sin_port=htons(PORT),.sin_addr.s_addr=inet_addr(IP)};//绑定if(bind(oldfd,(struct sockaddr *)&server,sizeof(server))==-1){perror("bind");return -1;}msg e[50];msg e_server;int count =0;//创建存放客户端地址信息的结构体struct sockaddr_in client[50];//存放客户端地址信息长度socklen_t client_len[50];char buff[1024];int sub;struct pollfd fds[2];fds[0].fd=oldfd;fds[1].fd=0;fds[0].events=POLLIN;fds[1].events=POLLIN;//循环接受客户端信息while(1){ int res=poll(fds,2,-1);if(res==-1){perror("poll");return -1;}if(fds[1].revents==POLLIN){fgets(buff,sizeof(buff),stdin);buff[strlen(buff)-1]='\0';strcat(buff,":服务器消息");strcpy(e_server.text,buff);for(int i=0;i<=count;i++){ sendto(oldfd,&e_server,sizeof(e_server),0,(struct sockaddr *)&client[i],client_len[i]); }}if(fds[0].revents==POLLIN){client_len[count]=sizeof(client[count]);recvfrom(oldfd,&e[count],sizeof(e[count]),0,(struct sockaddr *)&client[count],&client_len[count]); if(e[count].type=='L'||e[count].type=='l'){printf("%s客户端上线\n",e[count].name);snprintf(buff,sizeof(buff),"%s",e[count].name);strcat(buff,"已上线");strcpy(e[count].text,buff);for(int i=0; i<=count; i++) { sendto(oldfd,&e[count],sizeof(e[count]),0,(struct sockaddr *)&client[i],client_len[i]); }count++;}else if(e[count].type=='Q'||e[count].type=='q'){ int quit_index = find_client_index(e, count, e[count].name);if(quit_index != -1) {printf("%s客户端退出\n",e[count].name);snprintf(buff,sizeof(buff),"%s",e[count].name);strcat(buff,"已下线");strcpy(e[count].text,buff);for(int i=0; i<count; i++) { sendto(oldfd,&e[count],sizeof(e[count]),0,(struct sockaddr *)&client[i],client_len[i]);}for(int i = quit_index; i < count-1; i++) {memcpy(&client[i], &client[i+1], sizeof(struct sockaddr_in));client_len[i] = client_len[i+1];memcpy(&e[i], &e[i+1], sizeof(msg));}count--;}}else if(e[count].type=='C'||e[count].type=='c'){int sender_index = find_client_index(e, count, e[count].name);if(sender_index != -1) {printf("%s发送的信息：%s\n",e[count].name,e[count].text);for(int i=0; i<count; i++) {if(i != sender_index) {sendto(oldfd,&e[count],sizeof(e[count]),0,(struct sockaddr *)&client[i],client_len[i]);} }}}}}return 0;
}客户端代码
#include <myhead.h>
#include <poll.h>
#define IP "192.168.60.117"
#define PORT 8888
typedef struct
{char type;//消息类型char name[100];//用户名char text[100];//文本
}msg;
int main(int argc, const char *argv[])
{//创建UDP套接字int oldfd =socket(AF_INET,SOCK_DGRAM,0);if(oldfd==-1){perror("socket");return -1;}//将客户端地址信息填充结构体struct sockaddr_in client={.sin_family=AF_INET,.sin_port=htons(PORT),.sin_addr.s_addr=inet_addr(IP)};msg e,e1;struct pollfd fds[2];fds[0].fd=oldfd;fds[1].fd=0;fds[0].events=POLLIN;fds[1].events=POLLIN;printf("请输入用户名：");scanf("%s",e.name);while(getchar()!=10);while(1){printf("请输入消息类型(L(上线),Q(下线),C(进入聊天模式))：");scanf("%c",&e.type);while(getchar()!=10);if(e.type=='C'||e.type=='c'){while(1){int res=poll(fds,2,-1);if(res==-1){perror("poll");}if(res==0){perror("超时");}if(fds[1].revents==POLLIN){scanf("%s",e.text);while(getchar()!=10);if(strcmp(e.text,"quit")==0){break;}sendto(oldfd,&e,sizeof(e),0,(struct sockaddr *)&client,sizeof(client));printf("发送成功\n");}if(fds[0].revents==POLLIN){socklen_t client_len=sizeof(client);recvfrom(oldfd,&e1,sizeof(e1),0,(struct sockaddr *)&client,&client_len);printf("收到%s的信息：%s\n",e1.name,e1.text);}}}else if(e.type=='Q'||e.type=='q'){sendto(oldfd,&e,sizeof(e),0,(struct sockaddr *)&client,sizeof(client));printf("客户端下线\n");return 0;}else if(e.type=='L'||e.type=='l'){sendto(oldfd,&e,sizeof(e),0,(struct sockaddr *)&client,sizeof(client));printf("客户端上线\n"); }else{printf("你输入的消息类型有误，请重新输入\n");}}return 0;
}
