Linux上用select()实现异步的Echo服务器

Linux下用select()实现异步的Echo服务器

本例子使用异步socket（select方法）实现了ECHO服务器程序。

搞了一个晚上，终于弄好了，出现的问题主要如下：

• 这是最重要的问题！当读取完数据后，需要将数据重新FD_SET进去，特别是serverFd，注意这个testFd意义非常重大，相当于参数传递中的复制行参，需要好好体会。
• 当read(rd)后，返回为0表示客户端的socket已经关闭，此时除了要FD_CLR，还要关闭fd!!否则fd资源没有被释放，很快就会达到select监听的上限！
  #相关代码, [四号程序员] http://www.coder4.com/*?* main.cc?*?* ?Created on: 2009-11-30?* ? ? ?Author: liheyuan?* ? ?Describe: 非阻塞模式服务器(Echo服务器)?*?* ? Last Date: 2009-11-30?* ? CopyRight: 2009 @ ICT LiHeyuan?*/?#include <iostream>using namespace std;?#include <sys/types.h>#include <sys/socket.h>#include <netinet/in.h>#include <arpa/inet.h>#include <unistd.h>#include <sys/ioctl.h>#include <fcntl.h>#include <errno.h>#include <sys/wait.h>#include <signal.h>?#define SERVER_PORT 18000?#define SERVER_QUEUE 10#define FD_SET_SIZE 10#define MAX_BUF 16?int main() {??? ?//设置服务器Addr，在18000，任意IP监听?? ?int serverFd;?? ?serverFd = socket(AF_INET, SOCK_STREAM, 0);??? ?/* 设置 serverFd 为非阻塞方式 */?? ?int opt = SO_REUSEADDR;?? ?setsockopt(serverFd, SOL_SOCKET, opt, &opt, sizeof(opt));??? ?struct sockaddr_in serverAddr;?? ?socklen_t serverAddrLen = sizeof(sockaddr_in);?? ?serverAddr.sin_addr.s_addr = htonl(INADDR_ANY);?? ?serverAddr.sin_port = htons(SERVER_PORT);??? ?//绑定?? ?if (bind(serverFd, (sockaddr*) &serverAddr, serverAddrLen)) {?? ? ? ?cout << "Binding on " << SERVER_PORT << " fail." << endl;?? ? ? ?return -1;?? ?}??? ?//创建等待队列?? ?listen(serverFd, SERVER_QUEUE);??? ?//设定fd_set?? ?fd_set readfds, testfds;?? ?FD_ZERO(&readfds);?? ?FD_SET(serverFd,&readfds);??? ?//非阻塞模式等待客户连接?? ?struct sockaddr_in clientAddr;?? ?int clientFd;?? ?socklen_t clientAddrLen;?? ?int len;?? ?char buf[MAX_BUF];?? ?while (1) {?? ? ? ?FD_SET(serverFd,&readfds);?? ? ? ?testfds = readfds;??? ? ? ?//选择readfds中可用的fd?? ? ? ?if (select(FD_SET_SIZE, &testfds, (fd_set *) NULL, (fd_set *) NULL,?? ? ? ? ? ? ? ?(struct timeval *) NULL) > 0) {?? ? ? ? ? ?if (FD_ISSET(serverFd,&testfds)) {?? ? ? ? ? ? ? ?//如果服务器fd可用，则为accept?? ? ? ? ? ? ? ?clientAddrLen = sizeof(sockaddr_in);?? ? ? ? ? ? ? ?clientFd = accept(serverFd, (struct sockaddr*) &clientAddr,?? ? ? ? ? ? ? ? ? ? ? ?&clientAddrLen);?? ? ? ? ? ? ? ?if (clientFd == -1) {?? ? ? ? ? ? ? ? ? ?cout << "accept() error" << endl;?? ? ? ? ? ? ? ? ? ?return -1;?? ? ? ? ? ? ? ?} else {?? ? ? ? ? ? ? ? ? ?FD_SET(clientFd,&readfds);?? ? ? ? ? ? ? ? ? ?cout << inet_ntoa(clientAddr.sin_addr) << " connect"?? ? ? ? ? ? ? ? ? ? ? ? ? ?<< " at fd" << clientFd << endl;?? ? ? ? ? ? ? ?}?? ? ? ? ? ?}?? ? ? ? ? ?//依次查询?? ? ? ? ? ?for (int fd = 0; fd < FD_SET_SIZE; fd++) {?? ? ? ? ? ? ? ?if (FD_ISSET(fd,&testfds)) {?? ? ? ? ? ? ? ? ? ?//如果是Client活动，进行Echo?? ? ? ? ? ? ? ? ? ?len = read(fd, buf, MAX_BUF);?? ? ? ? ? ? ? ? ? ?if (len > 0) {?? ? ? ? ? ? ? ? ? ? ? ?write(fd, buf, len);?? ? ? ? ? ? ? ? ? ?} else {?? ? ? ? ? ? ? ? ? ? ? ?//结束?? ? ? ? ? ? ? ? ? ? ? ?cout << "Client end at" << clientFd << endl;?? ? ? ? ? ? ? ? ? ? ? ?FD_CLR(fd,&readfds);?? ? ? ? ? ? ? ? ? ?}?? ? ? ? ? ? ? ?}?? ? ? ? ? ?}?? ? ? ?}?? ?}??? ?close(serverFd);}
  转自：http://www.coder4.com/archives/130

  ?