select实现TCP并发服务器
#include<myhead.h> #define SER_PORT 8888 //服务器端口号 #define SER_IP "192.168.125.192" //服务器ip地址 int main(int argc, const char *argv[]) { //1、为通信创建一个端点 int sfd = socket(AF_INET, SOCK_STREAM, 0); //参数1:说明使用的是ipv4通信域 //参数2:说明使用的是TCP面向连接的通信方式 //参数3:由于参数2中已经指定通信方式,填0即可 if(sfd == -1) { perror("socket error"); return -1; } printf("socket success sfd = %d\n", sfd); //3\、设置地址能够快速重用 int reuse = 1; if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse))==-1) { perror("setsockopt error"); return -1; } //2、绑定ip和端口号 //2.1 准备地址信息结构体 struct sockaddr_in sin; sin.sin_family = AF_INET; //通信域 sin.sin_port = htons(SER_PORT); //端口号 sin.sin_addr.s_addr = inet_addr(SER_IP); //ip地址 //2.2 绑定工作 if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) ==-1) { perror("bind error"); return -1; } printf("bind success\n"); //3、将套接字设置成被动监听状态 if(listen(sfd, 128)==-1) { perror("listen error"); return -1; } printf("listen success\n"); //4.1 定义用于接受客户端信息的容器 struct sockaddr_in cin; socklen_t addrlen = sizeof(cin); //11、准备检测文件描述符的集合 fd_set readfds, tempfds; //22、清空集合中的内容 FD_ZERO(&readfds); //33、将要检测的文件描述符放入集合中 FD_SET(sfd, &readfds); FD_SET(0, &readfds); int newfd = 0; int maxfd = sfd; struct sockaddr_in cin_arr[1024]; while(1) { //执行阻塞函数之前将文件描述符集合备份一份 tempfds = readfds; //使用阻塞函数,检测文件描述符集合中是否有事件产生 int res = select(maxfd+1, &tempfds, NULL, NULL, NULL); if(res == -1) { perror("select error"); return -1; }else if(res == 0) { printf("time out\n"); return -1; } //当程序执行至此,表示检测容器中已经有一个或多个事件产生 //只需要判断哪些文件描述符还在集合中,就一定产生了事件,直接取执行相关逻辑即可 //将所有的文件描述符全部遍历一遍 for(int i=0; i<=maxfd; i++) { //判断当前的i文件描述符是否没有触发事件 if(!FD_ISSET(i, &tempfds)) { continue; } //如果程序执行至此,说明当前的 i //触发了事件,要判断该i是什么文件描述符 //判断sfd是否触发了相关事件 if(i == sfd) { newfd = accept(i, (struct sockaddr*)&cin, &addrlen); if(newfd == -1) { perror("accept error"); return -1; } printf("[%s:%d]:发来连接请求\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port)); //将newfd放入到 readfds容器中,以便于检测是否有客户端发来数据 FD_SET(newfd, &readfds); //更新maxfd if(newfd >= maxfd) { maxfd = newfd; } //将套接字地址信息结构体放入数组中 cin_arr[newfd] = cin; }else if(i == 0)//判断是否是0号文件描述符解除的阻塞 { char buf[128] = ""; fgets(buf, sizeof(buf), stdin); buf[strlen(buf)-1] = 0; printf("键盘输入事件完成:%s\n", buf); //循环将消息发送给所有客户端 for(int k=4; k<=maxfd; k++) { send(k, buf, strlen(buf), 0); } }else { //5、与客户端进行相互通信 char rbuf[128] = ""; //清空容器 bzero(rbuf, sizeof(rbuf)); //从套接字中读取数据 //int res = read(newfd, rbuf, sizeof(rbuf));= int res = recv(i, rbuf, sizeof(rbuf), 0); if(res == 0) { printf("客户端已经下线\n"); //6、关闭套接字 close(i); //将当前的套接字文件描述符从容器中移除 FD_CLR(i, &readfds); //更新maxfd for(int j=maxfd; j>=sfd; j--) { if(FD_ISSET(j, &readfds)) { maxfd = j; break; } } continue; } //将读取的消息展示出来 printf("[%s:%d]:%s\n", inet_ntoa(cin_arr[i].sin_addr),ntohs(cin_arr[i].sin_port), rbuf); //将消息转发给所有客户端 for(int k=4; k<=maxfd; k++) { if(k != i) //不转发给自己 { send(k, rbuf, strlen(rbuf), 0); } } } } } close(sfd); return 0; }
poll实现TCP客户端并发
#include<myhead.h> #define SER_PORT 8888 #define SER_IP "192.168.125.192" #define CLI_PORT 6667 #define CLI_IP "192.168.125.66" int main(int argc, const char *argv[]) { //1、创建用于通信的套接字文件描述符 int cfd = socket(AF_INET, SOCK_STREAM, 0); if(cfd == -1) { perror("socket error"); return -1; } printf("cfd = %d\n", cfd); //2、绑定IP地址和端口号 //2.1 填充客户端地址信息结构体 struct sockaddr_in cin; cin.sin_family = AF_INET; cin.sin_port = htons(CLI_PORT); cin.sin_addr.s_addr = inet_addr(CLI_IP); //2.2 绑定 if( bind(cfd, (struct sockaddr*)&cin, sizeof(cin)) ==-1) { perror("bind error"); return -1; } printf("bind success\n"); //3、连接服务器 //3.1 准备对端地址信息结构体 struct sockaddr_in sin; sin.sin_family = AF_INET; sin.sin_port = htons(SER_PORT); sin.sin_addr.s_addr = inet_addr(SER_IP); //3.2 连接服务器 if(connect(cfd, (struct sockaddr*)&sin, sizeof(sin))==-1) { perror("connect error"); return -1; } printf("connect success\n"); //使用poll完成多个阻塞任务的并发执行 struct pollfd pfd[2]; pfd[0].fd = 0; pfd[0].events = POLLIN; pfd[1].fd = cfd; pfd[1].events = POLLIN; //4、数据收发 char wbuf[128] = ""; char rbuf[128] = ""; while(1) { int res = poll(pfd, 2, -1); if(res == -1) { perror("poll error"); return -1; }else if(res == 0) { printf("time out\n"); return -1; } //当程序执行至此,说明检测容器中有事件产生 //判断是否为0号文件描述符产生的事件 if(pfd[0].revents == POLLIN) { //从终端上获取要发送的数据 fgets(wbuf, sizeof(wbuf), stdin); wbuf[strlen(wbuf)-1] = '\0'; //将数据发送给服务器 send(cfd, wbuf, strlen(wbuf), 0); printf("发送成功\n"); } //判断是否为cfd文件描述符产生事件 if(pfd[1].revents == POLLIN) { //接收服务器发来的消息 bzero(rbuf, sizeof(rbuf)); recv(cfd, rbuf, sizeof(rbuf), 0); printf("服务器发来的消息为:%s\n", rbuf); } } //5、关闭套接字 close(cfd); return 0; }
5.22作业
最新推荐文章于 2024-10-04 13:19:28 发布