linux socket select
#include <stdio.h>#include <stdlib.h>#include <unistd.h>#include <errno.h>#include <string.h>#include <sys/types.h>#include <sys/socket.h>#include <netinet/in.h>#include <arpa/inet.h>#define MYPORT 1234 // 侦听端口#define BACKLOG 5 // 最大可连接客户端数量#define BUF_SIZE 200int fd_A; // 连接的FD数组int conn_amount; // 当前连接的数量void showclient(){ int i; printf("client amount: %d\n", conn_amount); for (i = 0; i < BACKLOG; i++) { printf("[%d]:%d", i, fd_A); } printf("\n\n");}int main(void){ int sock_fd, new_fd; // 侦听sock_fd, 新连接new_fd struct sockaddr_in server_addr; // server address information struct sockaddr_in client_addr; // connector's address information socklen_t sin_size; int yes = 1; char buf; int ret; int i; //创建侦听Socket if ((sock_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) { perror("Create listening socket error!"); exit(1); } //配置侦听Socket //SO_REUSEADDR BOOL 允许套接口和一个已在使用中的地址捆绑。 if (setsockopt(sock_fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) == -1) { perror("setsockopt error!"); exit(1); } server_addr.sin_family = AF_INET; // host byte order server_addr.sin_port = htons(MYPORT); // short, network byte order server_addr.sin_addr.s_addr = INADDR_ANY; // automatically fill with my IP memset(server_addr.sin_zero, '\0', sizeof(server_addr.sin_zero)); //绑定新创建的Socket到指定的IP和端口 if (bind(sock_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) { perror("bind error!"); exit(1); } //开始侦听，最大连接数为BACKLOG if (listen(sock_fd, BACKLOG) == -1) { perror("listen error!"); exit(1); } printf("listen port %d\n", MYPORT); //监控文件描述符集合 fd_set fdsr; //监控文件描述符集合中最大的文件号 int maxsock; //Select超时返回的时间。 struct timeval tv; conn_amount = 0; sin_size = sizeof(client_addr); maxsock = sock_fd; while (1) { // 初始化文件描述符集合 initialize file descriptor set FD_ZERO(&fdsr); // 把Sock_fd加入到文件描述符集合 FD_SET(sock_fd, &fdsr); // 超时设置30秒 tv.tv_sec = 30; tv.tv_usec = 0; // 把活动的socket的句柄加入到文件描述符集合中 for (i = 0; i < BACKLOG; i++) { if (fd_A != 0) { FD_SET(fd_A, &fdsr); } } //Select 函数原型 //int select（nfds， readfds， writefds， exceptfds， timeout） //nfds: select监视的文件句柄数,视进程中打开的文件数而定，一般设为呢要监视各文件中的 //最大文件号加一 //readfds:select监视的可读文件句柄集合 //writefds:select监视的可写文件句柄集合。 //exceptfds:select监视的异常文件句柄集合。 //timeout:本次select的超时结束时间。 ret = select(maxsock + 1, &fdsr, NULL, NULL, &tv); if (ret < 0) { perror("select error!"); break; } else if (ret == 0) { printf("timeout\n"); continue; } // 轮询各个文件描述符（socket） for (i = 0; i < conn_amount; i++) { //FD_ISSET（int fd， fdset *fdset）：检查fdset联系的文件句柄fd是否可读写， // >0表示可读写。 if (FD_ISSET(fd_A, &fdsr)) { //接收数据 ret = recv(fd_A, buf, sizeof(buf), 0); if (ret <= 0) //接收数据出错 { printf("client[%d] close\n", i); close(fd_A); FD_CLR(fd_A, &fdsr); fd_A = 0; } else// 数据接收成功 { //将接收数据的最后一位补0 if (ret < BUF_SIZE) memset(&buf, '\0', 1); printf("client[%d] send:%s\n", i, buf); } } } // 检查是否有新连接进来，如果有新连接进来，接收连接，生成新socket， //并加入到监控文件描述符集合中。 if (FD_ISSET(sock_fd, &fdsr)) { //接受连接 new_fd = accept(sock_fd, (struct sockaddr *)&client_addr, &sin_size); if (new_fd <= 0) { perror("accept socket error!"); continue; } // 将新的连接加入到监控文件描述符集合 if (conn_amount < BACKLOG) { fd_A = new_fd; printf("new connection client[%d] %s:%d\n", conn_amount,inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port)); if (new_fd > maxsock) maxsock = new_fd; } else { printf("max connections arrive, exit\n"); send(new_fd, "bye", 4, 0); close(new_fd); break; } } showclient(); } // 关闭所有连接 for (i = 0; i < BACKLOG; i++) { if (fd_A != 0) { close(fd_A); } } exit(0);}
页:
[1]