socket实现简易聊天小程序

这是网络编程的小练习，在Linux下用socket，epoll实现服务器端与客户端的通信。 代码如下：

服务器端： #include"func.h" //用进程池实现服务器端向客户端并行传输文件 int main() { int sfd=socket(AF_INET,SOCK_STREAM,0); struct sockaddr_in seraddr,cliaddr;//cliaddr用于accept函数 int ret; int len_client=sizeof(cliaddr); seraddr.sin_family=AF_INET; seraddr.sin_port=htons(2000); seraddr.sin_addr.s_addr=inet_addr("192.168.13.128"); ret=bind(sfd,(struct sockaddr*)&seraddr,sizeof(seraddr));//绑定IP和端口号 if(ret==-1){ perror("bind"); return -1; } listen(sfd,10);//开始监听 //构造进程池 int epfd=epoll_create(1); struct epoll_event event,evs[3]; memset(&event,0,sizeof(event)); event.data.fd=sfd; event.events=EPOLLIN; epoll_ctl(epfd,EPOLL_CTL_ADD,sfd,&event);//注册sfd memset(&event,0,sizeof(event)); event.data.fd=0; event.events=EPOLLIN; epoll_ctl(epfd,EPOLL_CTL_ADD,0,&event);//注册标准输入 memset(&cliaddr,0,sizeof(cliaddr)); int readyNum,i,new_fd; char buf[128]; while(1) { memset(evs,0,sizeof(evs)); readyNum=epoll_wait(epfd,evs,3,-1); for(i=0;i<readyNum;i++)//遍历就绪态的文件描述符，并执行相应的任务 { if(evs[i].data.fd==sfd) { new_fd=accept(sfd,(struct sockaddr*)&cliaddr,&len_client); if(new_fd==-1){ perror("accept"); return -1; } event.data.fd=new_fd; event.events=EPOLLIN; epoll_ctl(epfd,EPOLL_CTL_ADD,new_fd,&event);//注册new_fd printf("connect successfully\n"); } if(evs[i].data.fd==new_fd)//从对端读取 { memset(buf,0,sizeof(buf)); ret=recv(new_fd,buf,sizeof(buf),0); if(ret>0) printf("%s\n",buf); else if(ret==0){ printf("byebye\n"); close(new_fd); event.data.fd=new_fd; event.events=EPOLLIN;// epoll_ctl(epfd,EPOLL_CTL_DEL,new_fd,&event);//移除new_fd } } if(evs[i].data.fd==0)//向对端发送数据 { memset(buf,0,sizeof(buf)); read(0,buf,sizeof(buf)); send(new_fd,buf,strlen(buf)-1,0); } } } close(new_fd); close(sfd); return 0; } 客户端： #include"func.h" //客户端 int main() { int sfd=socket(AF_INET,SOCK_STREAM,0); struct sockaddr_in seraddr; int ret; seraddr.sin_family=AF_INET; seraddr.sin_port=htons(2000); seraddr.sin_addr.s_addr=inet_addr("192.168.13.128"); ret=connect(sfd,(struct sockaddr*)&seraddr,sizeof(seraddr));//客户端请求连接 if(ret==-1){ perror("connect"); return -1; } //构造进程池 int epfd=epoll_create(1); struct epoll_event event,evs[2]; memset(&event,0,sizeof(event)); event.data.fd=sfd; event.events=EPOLLIN; epoll_ctl(epfd,EPOLL_CTL_ADD,sfd,&event);//注册sfd memset(&event,0,sizeof(event)); event.data.fd=0; event.events=EPOLLIN; epoll_ctl(epfd,EPOLL_CTL_ADD,0,&event);//注册标准输入 char buf[128]; int readyNum,i; while(1) { memset(evs,0,sizeof(evs));//每次用epoll_wait前先清空evs readyNum=epoll_wait(epfd,evs,2,-1); for(i=0;i<readyNum;i++) { if(evs[i].data.fd==sfd)//从服务器端读数据 { memset(buf,0,sizeof(buf)); ret=recv(sfd,buf,sizeof(buf),0); if(ret>0) printf("%s\n",buf); else if(ret==0){ printf("byebye\n"); exit(0); } } if(evs[i].data.fd==0)//向对端发送数据 { memset(buf,0,sizeof(buf)); ret=read(0,buf,sizeof(buf)); ret=send(sfd,buf,strlen(buf)-1,0); if(ret==-1){ perror("send"); return -1; } } } } return 0; }