項目要求：

代碼實現：

服務器端：

#include <myhead.h>//定義協議包
struct proto
{char type;char name[20];char text[128];	
};int main(int argc, const char *argv[])
{//判斷從終端輸入的字符串的個數if(argc != 3){printf("input error\n");printf("usage:./a.out 本機IP 本機端口\n");return -1;}//創建用于通信的套接字int sfd = socket(AF_INET,SOCK_DGRAM,0);if(sfd == -1){perror("socket error");return -1;}//設置端口號快速重用int reuse = 1;if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) == -1){perror("setsockopt error");return -1;}//綁定服務器IP和端口號填充服務器地址信息結構體short port = (short)atoi(argv[2]);struct sockaddr_in sin;sin.sin_family = AF_INET;sin.sin_addr.s_addr = inet_addr(argv[1]);sin.sin_port = htons(port);綁定if(bind(sfd,(struct sockaddr *)&sin,sizeof(sin)) == -1){perror("bind error");return -1;}//定義客戶端地址信息結構體struct sockaddr_in cin;cin.sin_family = AF_INET;socklen_t socklen = sizeof(cin);//定義客戶端地址信息結構體數組，用于存放多個客戶端地址信息struct sockaddr_in savecin[1024];//初始化每個信息結構體內的第一個成員for(int i = 0;i < 1024;i++){savecin[i].sin_family = AF_INET;}定義一個用于檢測文件描述符的集合fd_set readfds, tempfds;                          //在棧區定義清空容器中的內容FD_ZERO(&readfds);將要檢測的文件描述符放入集合中FD_SET(sfd, &readfds);           //將sfd文件描述符放入FD_SET(0, &readfds);             //將0號文件描述符放入//對客戶端的數據進行保存和轉發char buf[256] = "";int res1,res2;int n = 0;//定義協議包結構體變量struct proto pro;struct proto send;while(1){bzero(buf,sizeof(buf));tempfds = readfds;使用select阻塞等待集合中的文件描述符有事件產生res1 = select(sfd+1, &tempfds, NULL, NULL, NULL);if(res1 == -1){perror("select error");return -1;}else if(res1 == 0){printf("time out\n");return -1;}//接收客戶端信息if(FD_ISSET(sfd,&tempfds)){res2 = recvfrom(sfd,&pro,sizeof(pro),0,(struct sockaddr *)&cin,&socklen);if(res2 == -1){perror("recvfrom error");return -1;}//登錄時存儲客戶端到數組中if(pro.type == 'L'){savecin[n] = cin;n++;sprintf(buf,"---%s已上線---",pro.name);printf("%s\n",buf);for(int i = 0;i < n;i++){if(savecin[i].sin_port == cin.sin_port){continue;}	sendto(sfd,&pro,sizeof(pro),0,(struct sockaddr *)&savecin[i],sizeof(savecin[i]));}}if(pro.type == 'C'){//群聊send.type = pro.type;strcpy(send.name,pro.name);strcpy(send.text,pro.text);for(int i = 0;i < n;i++){if(savecin[i].sin_port == cin.sin_port){continue;}	sendto(sfd,&send,sizeof(send),0,(struct sockaddr *)&savecin[i],sizeof(savecin[i]));}}if(pro.type == 'Q'){//下線send.type = pro.type;strcpy(send.name,pro.name);strcpy(send.text,pro.text);for(int i = 0;i < n;i++){bzero(buf,sizeof(buf));sprintf(buf,"---%s已下線---\n",send.name);printf("%s\n",buf);//刪除該用戶if(savecin[i].sin_port == cin.sin_port){int t = i;for(int j = i;j <= n;j++){savecin[t] = savecin[t+1];t++;				}}n--;sendto(sfd,&send,sizeof(send),0,(struct sockaddr *)&savecin[i],sizeof(savecin[i]));}}}if(FD_ISSET(0,&tempfds)){bzero(send.name,sizeof(send.name));bzero(send.text,sizeof(send.text));send.type = 'C';strcpy(send.name,"系統消息");fgets(send.text,sizeof(send.text),stdin);send.text[strlen(send.text)-1] = '\0';for(int i = 0;i <= n;i++){sendto(sfd,&send,sizeof(send),0,(struct sockaddr *)&savecin[i],sizeof(savecin[i]));}	}}//關閉套接字文件描述符close(sfd);return 0;
}

客戶端：

#include <myhead.h>//定義協議包結構體
struct proto
{char type;char name[20];char text[128];
};int main(int argc, const char *argv[])
{//判斷終端輸入的字符串的個數if(argc != 3){printf("input error\n");printf("usage:./a.out 服務器IP 服務器端口號\n");return -1;}//創建用于通信的套接字int cfd = socket(AF_INET,SOCK_DGRAM,0);if(cfd == -1){perror("socket error");return -1;}//填充服務器地址信息結構體short port = (short)atoi(argv[2]);	struct sockaddr_in sin;sin.sin_family = AF_INET;sin.sin_addr.s_addr = inet_addr(argv[1]);sin.sin_port = htons(port);socklen_t socklen = sizeof(sin);//定義協議包結構體變量struct proto pro;//填充登錄協議printf("請輸入姓名>>");//登錄pro.type = 'L'; fgets(pro.name,sizeof(pro.name),stdin);pro.name[strlen(pro.name)-1] = '\0';sendto(cfd,&pro,sizeof(pro),0,(struct sockaddr *)&sin,sizeof(sin));定義一個用于檢測文件描述符的集合fd_set readfds, tempfds;                          //在棧區定義清空容器中的內容FD_ZERO(&readfds);將要檢測的文件描述符放入集合中FD_SET(cfd, &readfds);           //將sfd文件描述符放入FD_SET(0, &readfds);             //將0號文件描述符放入//向服務器發送或從服務器接收消息char rbuf[128] = "";int res = 0;char name1[20] = "";strcpy(name1,pro.name);while(1){將集合內容復制一份tempfds = readfds;使用select阻塞等待集合中的文件描述符有事件產生res = select(cfd+1, &tempfds, NULL, NULL, NULL);if(res == -1){perror("select error");return -1;}else if(res == 0){printf("time out\n");return -1;}//群聊和退出if(FD_ISSET(0,&tempfds)){bzero(pro.text,sizeof(pro.text));//從終端獲取內容fgets(pro.text,sizeof(pro.text),stdin);pro.text[strlen(pro.text)-1] = '\0';if(strcmp(pro.text,"quit") == 0){//退出pro.type = 'Q';sendto(cfd,&pro,sizeof(pro),0,(struct sockaddr *)&sin,sizeof(sin));//關閉套接字文件描述符close(cfd);break;}else{//群聊pro.type = 'C';sendto(cfd,&pro,sizeof(pro),0,(struct sockaddr *)&sin,sizeof(sin));}}//接收來自服務器的消息if(FD_ISSET(cfd,&tempfds)){bzero(rbuf,sizeof(rbuf));bzero(pro.text,sizeof(pro.text));res = recvfrom(cfd,&pro,sizeof(pro),0,NULL,NULL);if(res < 0){perror("recvfrom error");return -1;}if(pro.type == 'L'){printf("---%s已登錄---\n",pro.name);}if(pro.type == 'C'){printf("%s:%s\n",pro.name,pro.text);if(strcmp(pro.name,"系統消息") == 0){strcpy(pro.name,name1);}}if(pro.type == 'Q'){printf("---%s已下線---\n",pro.name);}}}//關閉套接字文件描述符close(cfd);return 0;
}

代碼運行效果圖：