基于UDP的网络聊天室

1、如果有用户登录，其他用户可以收到这个人的登录信息

2、如果有人发送信息，其他用户可以收到这个人的群聊信息

3、如果有人下线，其他用户可以收到这个人的下线信息

4、服务器可以发送系统信息 服务器

#include<myhead.h> typedef struct Node { char usrName[20];//用户名 struct sockaddr_in cin; struct Node *next; //指针 }__link; //收发信息结构体 typedef struct msgtyp { int type; char name[20]; char text[128]; }msg; int do_lodin(int sfd,__link* head,msg recv_msg,struct sockaddr_in cin) { printf("%s[%s:%d]登陆成功\n",recv_msg.name,inet_ntoa(cin.sin_addr),ntohs(cin.sin_port)); sprintf(recv_msg.text,"----%s登陆成功----",recv_msg.name); while(head->next!=NULL) { head=head->next; if(sendto(sfd,&recv_msg,sizeof(recv_msg),0,(struct sockaddr*)&(head->cin),sizeof(head->cin))==-1) { perror("sendto"); return -1; } } //添加客户到链表 __link *new=(__link*)malloc(sizeof(__link)); new->cin=cin; new->next=NULL; head->next=new; return 0; } int do_chat(int sfd,__link* head, msg recv_msg, struct sockaddr_in cin) { printf("%s[%s:%d]chat成功\n",recv_msg.name,inet_ntoa(cin.sin_addr),ntohs(cin.sin_port)); char buf[258]=""; sprintf(buf,"%s:%s",recv_msg.name,recv_msg.text); strcpy(recv_msg.text,buf); //给其他客户端发送消息 while(head->next!=NULL) { head=head->next; head->cin=cin; if(memcmp(&cin, &head->cin, sizeof(cin)) != 0) { if(sendto(sfd,&recv_msg,sizeof(recv_msg),0,(struct sockaddr*)&(head->cin),sizeof(head->cin))==-1) { perror("sendto"); return -1; } } } return 0; } int do_quit(int sfd,__link* head, msg recv_msg, struct sockaddr_in cin) { sprintf(recv_msg.text, "-----%s 已下线-----\n", recv_msg.name); while(head->next != NULL) { if(memcmp(&cin, &head->next->cin, sizeof(cin)) == 0) { //删除 __link* temp = head->next; head->next = temp->next; free(temp); } else { head = head->next; if(sendto(sfd, &recv_msg, sizeof(recv_msg),0, (struct sockaddr*)&(head->cin), sizeof(head->cin))==-1) { perror("sendto"); return -1; } } } return 0; } int main(int argc, char *argv[]) { if(argc<3) { printf("请输入ip和端口号\n"); return -1; } int sfd=socket(AF_INET,SOCK_DGRAM,0); if(sfd==-1) { perror("socket"); return -1; } int reuse=-1; if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))==-1) { perror("setsockopt"); return -1; } struct sockaddr_in sin; sin.sin_family=AF_INET; sin.sin_port=htons(atoi(argv[2])); sin.sin_addr.s_addr=inet_addr(argv[1]); if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))==-1) { perror("bind"); return -1; } //printf("bind success\n"); int res=fork(); //父进程接收消息 if(res>0) { __link *head=(__link*)malloc(sizeof(__link)); head->next=NULL; msg recv_msg; int recv_len=0; struct sockaddr_in cin; socklen_t len=sizeof(cin); while(1) { recv_len=recvfrom(sfd,&recv_msg,sizeof(recv_msg),0,(struct sockaddr*)&cin,&len); if(recv_len<0) { perror("recvfrom"); return -1; } int type=ntohl(recv_msg.type); switch(type) { case 1: { do_lodin(sfd,head,recv_msg,cin); } break; case 2: { do_chat(sfd,head,recv_msg,cin); } break; case 3: { do_quit(sfd,head,recv_msg,cin); } break; } } }else if(res==0) { //子进程,发送系统消息 msg sys_msg={htonl(1),"**system**"}; while(1) { bzero(sys_msg.text,1024); fgets(sys_msg.text,1024,stdin); sys_msg.text[strlen(sys_msg.text)-1]=0; if(sendto(sfd,&sys_msg,sizeof(sys_msg),0,(struct sockaddr*)&sin,sizeof(sin))==-1) { perror("sendto"); return -1; } } printf("系统消息发送成功\n"); } close(sfd); return 0; }

客户端代码

#include<myhead.h> #define LOGIN 1 #define CHAT 2 #define QUIT 3 #define ERR_LOG(msg) do{\ printf("%d %s %s \n", __LINE__, __func__, __FILE__);\ perror(msg);\ }while(0) //收发信息的结构体 typedef struct { int type; char name[20]; char text[128] ; }MSG; typedef void (*sighandler_t)(int); int do_chat(int sfd, MSG msg, struct sockaddr_in sin) { while(1) { //从终端获取聊天信息 bzero(msg.text, 128); fgets(msg.text, 128,stdin); msg.text[strlen(msg.text)-1] = 0; if(strncasecmp(msg.text, "QUIT",4) == 0) { msg.type = htonl(QUIT); } else { msg.type = htonl(CHAT); } //发送 if(sendto(sfd, &msg, sizeof(msg), 0, (void*)&sin, sizeof(sin)) < 0) { ERR_LOG("sendto"); return -1; } //如果是退出信息,则客户端要退出 if(msg.type == htonl(QUIT)) { exit(0); //退出子进程; } } } int do_recv(int sfd) { MSG rcv_msg ; while(1) { if(recvfrom(sfd, &rcv_msg, sizeof(rcv_msg), 0, NULL, NULL) < 0) { perror("recvfrom"); return -1; } printf("%s\n",rcv_msg.text); } } void handler(int sig) { //回收子进程资源并退出 while(waitpid(-1,NULL, WNOHANG)>0); exit(0); } int main(int argc, char *argv[]) { if(argc < 3) { printf("请输入 ip 和端口号\n"); return -1; } sighandler_t s = signal(SIGCHLD, handler); if(s == SIG_ERR) { ERR_LOG("signal"); return -1; } int sfd = socket(AF_INET, SOCK_DGRAM, 0); if(sfd < 0) { perror("socket"); return -1; } //2.填充服务器ip和端口号 struct sockaddr_in sin; sin.sin_family=AF_INET; sin.sin_port=htons(atoi(argv[2])); sin.sin_addr.s_addr=inet_addr(argv[1]); socklen_t addrlen = sizeof(sin); //登录协议 MSG msg ; msg.type = htonl(1); printf("请输入姓名>>>"); fgets(msg.name, 20, stdin); msg.name[strlen(msg.name)-1] = 0; if(sendto(sfd, &msg, sizeof(msg), 0, (void*)&sin, sizeof(sin)) < 0) { ERR_LOG("sendto"); return -1; } pid_t pid = fork(); if(pid > 0) { //父进程获取信息 do_recv(sfd); } else if(0 == pid) { //子进程发送信息 do_chat(sfd, msg, sin); } //4.关闭套接字 close(sfd); return 0; }

效果