聊天服务器 selectServer.c
程序代码:
/* * selectserver.c -- 多人聊天服务器,可以通过 $ telnet 192.168.1.102 9034 来连接到服务器 */ #include <stdio.h> #include <stdlib.h> #include <string.h> #include <unistd.h> #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> #include <netdb.h> #define PORT "9034" // port we're listening on // get sockaddr, IPv4 or IPv6: void *get_in_addr(struct sockaddr *sa) { if (sa->sa_family == AF_INET) return &(((struct sockaddr_in*)sa)->sin_addr); return &(((struct sockaddr_in6*)sa)->sin6_addr); } int main(void) { fd_set master; // master file descriptor list fd_set read_fds; // temp file descriptor list for select() int fdmax; // maximum file descriptor number int listener; // listening socket descriptor int newfd; // newly accept()ed socket descriptor struct sockaddr_storage remoteaddr; // client address socklen_t addrlen; char buf[256]; // buffer for client data int nbytes; char remoteIP[INET6_ADDRSTRLEN]; int yes=1; // for setsockopt() SO_REUSEADDR, below int i, j, rv; struct addrinfo hints, *ai, *p; FD_ZERO(&master); // clear the master and temp sets FD_ZERO(&read_fds); // get us a socket and bind it memset(&hints, 0, sizeof hints); hints.ai_family = AF_UNSPEC; hints.ai_socktype = SOCK_STREAM; hints.ai_flags = AI_PASSIVE; if ((rv = getaddrinfo(NULL, PORT, &hints, &ai)) != 0) { fprintf(stderr, "selectserver: %s\n", gai_strerror(rv)); exit(1); } for(p = ai; p != NULL; p = p->ai_next) { listener = socket(p->ai_family, p->ai_socktype, p->ai_protocol); if (listener < 0) continue; // lose the pesky "address already in use" error message setsockopt(listener, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)); if (bind(listener, p->ai_addr, p->ai_addrlen) < 0) { close(listener); continue; } break; } // if we got here, it means we didn't get bound if (p == NULL) { fprintf(stderr, "selectserver: failed to bind\n"); exit(2); } freeaddrinfo(ai); // all done with this // listen if (listen(listener, 10) == -1) { perror("listen"); exit(3); } // add the listener to the master set FD_SET(listener, &master); // keep track of the biggest file descriptor fdmax = listener; // so far, it's this one // main loop for(;;) { read_fds = master; // copy it if (select(fdmax+1, &read_fds, NULL, NULL, NULL) == -1) { perror("select"); exit(4); } /* run through the existing connections looking for data to read*/ /*每增加一个新客户,fdmax都会增加1*/ for(i = 0; i <= fdmax; i++) { /*可以在下面添加一个printf() 用来观察i 的变化*/ // printf("%d",i ) if (FD_ISSET(i, &read_fds)) { // we got one!! /* either listen on socket or recvive data from client * 这里的 listener 可能一直为4 * newfd 则是5,6,7的变化 */ if (i == listener) { // handle new connections addrlen = sizeof remoteaddr; /* always use the minimum numbber*/ newfd = accept(listener,(struct sockaddr *)&remoteaddr,&addrlen); if (newfd == -1) perror("accept"); else { FD_SET(newfd, &master); // add to master set if (newfd > fdmax) // keep track of the max fdmax = newfd; printf("selectserver: new connection from %s on socket %d\n",inet_ntop(remoteaddr.ss_family,get_in_addr((struct sockaddr*)&remoteaddr),remoteIP, INET6_ADDRSTRLEN), newfd); } } else { // handle data from a client if ((nbytes = recv(i, buf, sizeof buf, 0)) <= 0) { // got error or connection closed by client if (nbytes == 0) // connection closed printf("selectserver: socket %d hung up\n", i); else perror("recv"); close(i); // bye! FD_CLR(i, &master); // remove from master set } else // 从client 那儿得到数据了 for(j = 0; j <= fdmax; j++) // 发送给所有人 if (FD_ISSET(j, &master)) // 除了listener 和我们自己不发送 if (j != listener && j != i) if (send(j, buf, nbytes, 0) == -1) perror("send"); } } } } return 0; }
[ 本帖最后由 madfrogme 于 2012-8-31 15:22 编辑 ]