服务器:
#include<myhead.h>//do-while只是为了不让花括号单独存在,并不循环
#define ERR_MSG(msg) do{\fprintf(stderr,"%d:",__LINE__);\perror(msg);\
}while(0);#define PORT 8888//端口号1024-49151
#define IP "192.168.2.54"//本机IP,终端输入ifconfig可得int main(int argc, const char *argv[])
{//创建流式套接字 socketint sfd = socket(AF_INET,SOCK_STREAM,0);if(sfd<0){//printf("%d:",__LINE__);//perror("socket error");ERR_MSG("socket");return -1;}printf("socket create success sfd=%d\n",sfd);//允许端口快速的被复用int reuse = 1; if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0){ERR_MSG("setsockopt");return -1;}printf("允许端口快速的被复用成功\n");//填充地址信息结构体给bind函数绑定//真是的地址信息结构体根据地址族制定 AF_INET:man 7 ipstruct sockaddr_in sin;sin.sin_family = AF_INET;//必须填AF_INETsin.sin_port = htons(PORT);//端口号的网络字节序sin.sin_addr.s_addr = inet_addr(IP);//本机IP//绑定服务器的地址信息 -->必须绑定 bindif(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0){ERR_MSG("bind");return -1;}//指针类型的强转不会有数据丢失printf("bind success\n");//将套接字设置为被动监听状态 listenif(listen(sfd,128)<0){ERR_MSG("listen");return -1;}printf("listen success\n");//定义文件描述符集合fd_set readfds,tmpfds;//将文件描述符集合清空FD_ZERO(&readfds);//将文件描述符放入集合之中FD_SET(sfd,&readfds);FD_SET(0, &readfds);int maxfd = sfd;int res;char buf[128] = "";struct sockaddr_in cin; //存储客户端的地址信息socklen_t addrlen = sizeof(cin); //真实的地址信息结构体的大小//获取一个已经完成的客户端信息,生成一个新的文件描述符 acceptint newfd;struct sockaddr_in saveCin[1024]; //备份连接成功的客户端的地址信息,用下标来对应地址信息 while(1){tmpfds = readfds;//调用select函数res = select(maxfd+1,&tmpfds, NULL, NULL, NULL);if(res < 0){ERR_MSG("select error");break;}else if(0 == res){printf("time out\n");break;}else{for(int i=0; i<=maxfd; i++){if(0==FD_ISSET(i, &tmpfds)){continue;}else if(-1==FD_ISSET(i, &tmpfds)){ERR_MSG("FD_ISSET error");break;}if(0==i){//终端输入printf("从终端读取成功::");fgets(buf,sizeof(buf),stdin);buf[strlen(buf)-1] = 0;printf("%s\n",buf);}else if(sfd == i){if((newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen))<0){ERR_MSG("accept");return -1;}printf("[%s:%d]客户端链接成功 newfd = %d\n",\inet_ntoa(cin.sin_addr),\ntohs(cin.sin_port),newfd);saveCin[newfd] = cin; //将cin另存到下标为文件描述符的位置//将newfd添加到集合中FD_SET(newfd, &readfds);//更新maxfdmaxfd = maxfd>newfd?maxfd:newfd;}else{if(FD_ISSET(i, &readfds)){//清空字符串bzero(buf,sizeof(buf)); //memset//接收res = recv(i, buf, sizeof(buf) ,0);if(res<0){ERR_MSG("recv");return -1;}else if(0==res){printf("[%s:%d]客户端下线 newfd=%d\n",\inet_ntoa(saveCin[i].sin_addr),\ntohs(saveCin[i].sin_port),i);//将文件描述符从集合中踢出FD_CLR(i, &readfds);//由于踢出的文件描述符可能是最大文件描述符,更新maxfd/* for(; maxfd>=0; maxfd--){if(FD_ISSET(maxfd, &readfds)){break;}} */while(FD_ISSET(maxfd, &readfds)==0 && maxfd-->=0);//关闭文件描述符close(i);continue;}printf("[%s:%d]客户端消息 newfd=%d:%s\n",\inet_ntoa(saveCin[i].sin_addr),\ntohs(saveCin[i].sin_port),i,buf); //发送strcat(buf,">_<");if(send(i,buf,strlen(buf),0)<0){ERR_MSG("send");return -1;}printf("send success\n"); } }}
/*//如果是终端输入就绪if(FD_ISSET(0,&tmpfds)){printf("从终端读取成功::");fgets(buf,sizeof(buf),stdin);buf[strlen(buf)-1] = 0;printf("%s\n",buf);}//如果是客户端连接就绪if(FD_ISSET(sfd,&tmpfds)){if((newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen))<0){ERR_MSG("accept");return -1;}printf("[%s:%d]客户端链接成功 newfd = %d\n",\inet_ntoa(cin.sin_addr),\ntohs(cin.sin_port),newfd);saveCin[newfd] = cin; //将cin另存到下标为文件描述符的位置//将newfd添加到集合中FD_SET(newfd, &readfds);//更新maxfdmaxfd = maxfd>newfd?maxfd:newfd;}for(int i=4; i<=maxfd; i++){if(FD_ISSET(i, &readfds)){//清空字符串bzero(buf,sizeof(buf)); //memset//接收res = recv(i, buf, sizeof(buf) ,0);if(res<0){ERR_MSG("recv");return -1;}else if(0==res){printf("[%s:%d]客户端下线 newfd=%d\n",\inet_ntoa(saveCin[i].sin_addr),\ntohs(saveCin[i].sin_port),i);//将文件描述符从集合中踢出FD_CLR(i, &readfds);//由于踢出的文件描述符可能是最大文件描述符,更新maxfdfor(; maxfd>=0; maxfd--){if(FD_ISSET(maxfd, &readfds)){break;}} while(FD_ISSET(maxfd, &readfds)==0 && maxfd-->=0);//关闭文件描述符close(i);continue;}printf("[%s:%d]客户端消息 newfd=%d:%s\n",\inet_ntoa(saveCin[i].sin_addr),\ntohs(saveCin[i].sin_port),i,buf); //发送strcat(buf,">_<");if(send(i,buf,strlen(buf),0)<0){ERR_MSG("send");return -1;}printf("send success\n"); }}*/}}//关闭文件描述符close(newfd);if(close(sfd)<0){ERR_MSG("close");return -1;}return 0;
}
服务器:
#include<myhead.h>//do-while只是为了不让花括号单独存在,并不循环
#define ERR_MSG(msg) do{\fprintf(stderr,"%d:",__LINE__);\perror(msg);\
}while(0);#define PORT 8888//端口号1024-49151
#define IP "192.168.2.54"//本机IP,终端输入ifconfig可得int main(int argc, const char *argv[])
{//创建流式套接字 socketint cfd = socket(AF_INET,SOCK_STREAM,0);if(cfd<0){//printf("%d:",__LINE__);//perror("socket error");ERR_MSG("socket");return -1;}printf("socket create success sfd=%d\n",cfd);//填充服务器的地址信息结构体给connect函数连接//相连接哪个服务器就输入哪个服务器的地址信息//真是的地址信息结构体根据地址族制定 AF_INET:man 7 ipstruct sockaddr_in sin;sin.sin_family = AF_INET;//必须填AF_INETsin.sin_port = htons(PORT);//服务器绑定的端口号的网络字节序sin.sin_addr.s_addr = inet_addr(IP);//服务器绑定的IPsocklen_t addrlen = sizeof(sin);//连接指定服务器 connectif(connect(cfd,(struct sockaddr*)&sin,addrlen)<0){ERR_MSG("connect");return -1;}char buf[128] = "";ssize_t res = 0;//定义文件描述符集合fd_set readfds,tmpfds;//初始化集合FD_ZERO(&readfds);//把cfd加入集合FD_SET(cfd, &readfds);//把标准输入加入集合FD_SET(0, &readfds);int ret = -1;while(1){tmpfds = readfds;ret = select(cfd+1, &tmpfds, NULL, NULL, NULL);if(ret<0){ERR_MSG("select error");break;}else if(0 == ret){printf("time out\n");break;}else{if(FD_ISSET(0, &tmpfds)){//说明有来自终端的输入,需要发送//清空字符串bzero(buf,sizeof(buf)); //memsetprintf("请输入>> ");fgets(buf,sizeof(buf),stdin);buf[strlen(buf)-1] = 0;//发送strcat(buf,">_<");if(send(cfd,buf,strlen(buf),0)<0){ERR_MSG("send");return -1;}printf("send success\n");}if(FD_ISSET(cfd, &tmpfds)){//套接字文件的接收缓冲区有数据,需要接收//说明有来自终端的输入,需要发送//清空字符串bzero(buf,sizeof(buf)); //memset//接收res = recv(cfd, buf, sizeof(buf) ,0);if(res<0){ERR_MSG("recv");return -1;}else if(0==res){printf("[%s:%d]服务器下线 cfd=%d\n",\inet_ntoa(sin.sin_addr),\ntohs(sin.sin_port),cfd);break;}printf("[%s:%d]服务器消息 cfd=%d:%s\n",\inet_ntoa(sin.sin_addr),\ntohs(sin.sin_port),cfd,buf);}}}/*while(1){//清空字符串bzero(buf,sizeof(buf)); //memsetprintf("请输入>> ");fgets(buf,sizeof(buf),stdin);buf[strlen(buf)-1] = 0;//发送strcat(buf,">_<");if(send(cfd,buf,strlen(buf),0)<0){ERR_MSG("send");return -1;}printf("send success\n");//接收res = recv(cfd, buf, sizeof(buf) ,0);if(res<0){ERR_MSG("recv");return -1;}else if(0==res){printf("[%s:%d]服务器下线 cfd=%d\n",\inet_ntoa(sin.sin_addr),\ntohs(sin.sin_port),cfd);break;}printf("[%s:%d]服务器消息 cfd=%d:%s\n",\inet_ntoa(sin.sin_addr),\ntohs(sin.sin_port),cfd,buf); }
*///关闭文件描述符//close(newfd);if(close(cfd)<0){ERR_MSG("close");return -1;}return 0;
}
效果: