Linux 系统 select I/O 复用

       I/O多路复用技术通过把多个I/O的阻塞复用到同一个select的阻塞上，从而使得系统在单线程的情况下可以同时处理多个客户端请求。与传统的多线程/多进程模型比，I/O多路复用的最大优势是系统开销小，系统不需要创建新的额外进程或者线程，也不需要维护这些进程和线程的运行，降底了系统的维护工作量，节省了系统资源。

请根据实际情况设置IP和端口号

Server端代码：

#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>
#define MAXLINE 50
typedef struct sockaddr SA;
int main(int argc, char **argv)
{
    int listenfd, connfd, maxfd, i, nbyte;
    struct sockaddr_in myaddr;
    char buf[MAXLINE];
    fd_set global_rdfs, current_rdfs;
    if ((listenfd = socket(PF_INET, SOCK_STREAM, 0)) < 0)
    {
        perror("fail to socket");
        exit(-1);
    }
    bzero(&myaddr, sizeof(myaddr));
    myaddr.sin_family = PF_INET;
    myaddr.sin_addr.s_addr = inet_addr("?.?.?.?"); //htonl(INADDR_ANY);
    myaddr.sin_port = htons(?); /* port number */
    if (bind(listenfd, (SA *) &myaddr, sizeof(myaddr)) < 0)
    {
        perror("fail to bind");
        exit(-1);
    }
    listen(listenfd, 5);
    FD_ZERO(&global_rdfs);
    FD_SET(listenfd, &global_rdfs);
    maxfd = listenfd;
    while ( 1 )
    {
        current_rdfs = global_rdfs;
        if (select(maxfd+1, ¤t_rdfs, NULL, NULL, 0) < 0)
        {
            perror("fail to select");
            exit(-1);
        }
        else
        {
            for (i=0; i<=maxfd; i++)
            {
                if (FD_ISSET(i, ¤t_rdfs)) // fd i is ready
                {
                    if (i == listenfd) // new connection is coming
                    {
                        connfd = accept(listenfd, NULL, NULL);
                        FD_SET(connfd, &global_rdfs);
                        maxfd = maxfd > connfd ? maxfd : connfd;
                    }
                    else // client send message 
                    {
                        if ((nbyte = recv(i, buf, sizeof(buf), 0)) <= 0)
                        {
                            close(i);
                            FD_CLR(i, &global_rdfs);
                        }
                        else
                        {
                            send(i, buf, sizeof(buf), 0);
                        }

                    }
                }
            } // end for
        }
    }
    return 0;
}

Client端代码：

#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define N 64
typedef struct sockaddr SA;
int main(int argc, char *argv[]) 
{
    int sockfd;
    char buf[N];
    struct sockaddr_in servaddr;
    if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) < 0)
    {
        perror("fail to socket");
        exit(-1);
    }
    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = PF_INET;
    servaddr.sin_addr.s_addr = inet_addr("?.?.?.?"); //server ip
    servaddr.sin_port = htons(?); /* port number */  
    if (connect(sockfd, (SA *)&servaddr, sizeof(servaddr)) < 0)
    {
        perror("fail to connect");
        exit(-1);
    }
    while ( 1 )
    {
        printf("client > ");
        fgets(buf, N, stdin);
        if (strncmp(buf, "quit", 4) == 0)
        {
            close(sockfd);
            exit(0);
        }
        send(sockfd, buf, N, 0);
        bzero(buf, N);
        recv(sockfd, buf, N, 0);
        printf("recv from server : %s", buf);
    }
    return 0;
}