项目要求:
代码实现:
服务器端:
#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;
}
代码运行效果图: