5-26作业

网络聊天室

服务器：

  1 #include <myhead.h>2 int main(int argc, const char *argv[])3 {4     if(argc!=3)5     {6         printf("请输入IP和端口号\n");7         return -1;8     }9     int sfd = socket(AF_INET, SOCK_DGRAM, 0);10     if(sfd == -1)11     {12         perror("socket error");13         return -1;14     }15 16     int reuse = 1;17     if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse))==-1)18     {19         perror("setsockopt error");20         return -1;21     }22 23     struct sockaddr_in sin;24     sin.sin_family = AF_INET;25     sin.sin_port = htons(atoi(argv[2]));26     sin.sin_addr.s_addr = inet_addr(argv[1]);27 28     //2.2 绑定29     if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) ==-1)30     {31         perror("bind error");32         return -1;33     }34     printf("bind success\n");35 36     struct pollfd fall[2];                                                                                                                                                                                                                                                                                                                                                            37     fall[0].fd=0;38     fall[0].events=POLLIN;39     fall[1].fd=sfd;40     fall[1].events=POLLIN;41 42     //地址组43     struct node144     {45         char type;46         char name[128];47         struct sockaddr_in sin;48     };49     struct node1 node[1024];50     for(int j=0;j<1024;j++)51     {52         node[j].type='0';53     }54     struct sockaddr_in cin;55     socklen_t addrlen = sizeof(cin);56 57     //数据包58     struct rbuf159     {60         char type;61         char name[128];62         char data[1024];63     };64     struct rbuf1 rbuf;65     while(1)66     {67         int res=poll(fall,2,-1);68         if(res==-1)69         {70             perror("poll");71             return -1;72         }73         else if(res==0)74         {75             printf("timeout\n");76             return -1;77         }78 79         if(fall[0].revents==POLLIN)80         {81             bzero(&rbuf,sizeof(rbuf));82             fgets(rbuf.data,sizeof(rbuf.data),stdin);83             rbuf.data[strlen(rbuf.data)-1]='\0';84             char fun[1024]="**system**:";85 86             strcat(fun,rbuf.data);87             strcpy(rbuf.data,fun);88             for(int j=0;j<1024;j++)89             {90                 sendto(sfd,&rbuf,sizeof(rbuf),0,(struct sockaddr*)&node[j].sin,addrlen);91             }92             printf("**system** [%s:%d]chat成功\n",inet_ntoa(sin.sin_addr),ntohs(sin.sin_port));93         }94 95         else if(fall[1].revents==POLLIN)96         {97             bzero(&rbuf,sizeof(rbuf));98             recvfrom(sfd,&rbuf,sizeof(rbuf),0,(struct sockaddr*)&cin,&addrlen);99             if(rbuf.type=='0')
100             {
101                 strcpy(rbuf.data,"请输入姓名>>>");
102                 rbuf.type='1';
103                 sendto(sfd,&rbuf,sizeof(rbuf),0,(struct sockaddr*)&cin,addrlen);
104             }
105 
106             else if(rbuf.type=='1')
107             {
108                 for(int j=0;j<1024;j++)
109                 {
110                     if(strcmp(node[j].name,rbuf.name)==0)
111                     {
112                         node[j].type='0';
113                     }
114                 }
115                 for(int j=0;j<1024;j++)
116                 {
117                     if(node[j].type=='0')
118                     {
119                         node[j].type='1';
120                         strcpy(node[j].name,rbuf.name);
121                         node[j].sin=cin;
122                         break;
123                     }
124                 }
125                 printf("%s [%s:%d]登录成功\n",rbuf.name,inet_ntoa(cin.sin_addr),ntohs(cin.sin_port));
126                 char fun[128]="-----";
127                 strcat(fun,rbuf.data);
128                 strcpy(rbuf.data,fun);
129                 strcat(rbuf.data,"已登录-----");
130 
131                 for(int j=0;j<1024;j++)
132                 {
133                     if(node[j].type=='1')
134                     {
135                         if(strcmp(node[j].name,rbuf.name)!=0)
136                             sendto(sfd,&rbuf,sizeof(rbuf),0,(struct sockaddr*)&node[j].sin,addrlen);
137                     }
138                 }
139             }
140 
141             else if(rbuf.type=='2')
142             {
143                 char run[1024]="";
144                 strcat(run,rbuf.name);
145                 strcat(run,":");
146                 strcat(run,rbuf.data);
147                 strcpy(rbuf.data,run);
148                 strcat(run,":");
149                 for(int j=0;j<1024;j++)
150                 {
151                     if(node[j].type=='1')
152                     {
153                         if(strcmp(node[j].name,rbuf.name)!=0)
154                             sendto(sfd,&rbuf,sizeof(rbuf),0,(struct sockaddr*)&node[j].sin,addrlen);
155                     }
156                 }
157                 printf("%s [%s:%d]chat成功\n",rbuf.name,inet_ntoa(cin.sin_addr),ntohs(cin.sin_port));
158             }
159         }
160     }
161     close(sfd);
162     return 0;
163 }
164 

客户端：

 28 29     //收数据包30     struct rbuf131     {32         char type;33         char name[128];34         char data[1024];35     };36     struct rbuf1 buf;37     buf.type='0';38     strcpy(buf.data,"");39     strcpy(buf.name,"");40 41     //发数据包42     struct rbuf1 rbuf;43     rbuf.type='0';44     strcpy(rbuf.data,"");45     strcpy(rbuf.name,"");46 47     sendto(sfd,&rbuf,sizeof(rbuf),0,(struct sockaddr*)&sin,addrlen);48 49     while(1)50     {51         int fel=poll(fall,2,-1);52         if(fel==-1)53         {54             perror("poll");55             return -1;56         }57         else if(fel==0)58         {59             printf("timeout");60             return -1;                                                                              61         }62 63         if(fall[0].revents==POLLIN)64         {65             if(rbuf.type=='1')66             {67                 rbuf.type='2';68                 bzero(rbuf.data,sizeof(rbuf.data));69                 fgets(rbuf.data,sizeof(rbuf.data),stdin);70                 rbuf.data[strlen(rbuf.data)-1]='\0';71                 sendto(sfd,&rbuf,sizeof(rbuf),0,(struct sockaddr*)&sin,addrlen);72                 rbuf.type='1';73             }74         }75 76         else if(fall[1].revents==POLLIN)77         {78             if(rbuf.type=='0')79             {80                 bzero(&buf,sizeof(buf));81                 bzero(rbuf.data,sizeof(rbuf.data));82                 recvfrom(sfd,&buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);83                 printf("%s",buf.data);84                 fgets(rbuf.data,sizeof(rbuf.data),stdin);85                 rbuf.data[strlen(rbuf.data)-1]='\0';86                 strcpy(rbuf.name,rbuf.data);87                 rbuf.type='1';88                 sendto(sfd,&rbuf,sizeof(rbuf),0,(struct sockaddr*)&sin,addrlen);89 90             }91             else if(rbuf.type=='1')92             {93                 bzero(&buf,sizeof(buf));94                 recvfrom(sfd,&buf,sizeof(rbuf),0,(struct sockaddr*)&sin,&addrlen);95                 printf("%s\n",buf.data);96             }97         }98     }99     close(sfd);
100     return 0;
101 }
~