非阻塞socket通讯（select函数的使用）

转自：http://jieastah.blog.sohu.com/84220706.html

我自己写了一个Linux下的利用select的非阻塞的socket通讯的服务器和客户端的程序，供参考。（注：Linux和windows的select有些不同，体现两个方面：一是select函数的第一个参数，在windows下可以忽略，但在linux下必须设为最大文件描述符加1；二是结构fd_set在两个系统里定义不一样。）

server.c程序：
－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－

#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <netinet/in.h>

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/wait.h>
#include <sys/time.h>
#include <sys/types.h>

#define MAXBUF 1024

typedef struct CLIENT {
    int fd;
    struct sockaddr_in addr;    
}CLIENT;

/***************************
**server for multi-client 
**PF_SETSIZE=1024
****************************/
int main(int argc, char** argv)
{
    int i,n,maxi = -1;
    int nready;
    int slisten,sockfd,maxfd=-1,connectfd;
    
    unsigned int myport,lisnum; 

    struct sockaddr_in  my_addr,addr;
    struct timeval tv;
    
    socklen_t len;
    fd_set rset,allset;    
    
    char buf[MAXBUF + 1];
    CLIENT client[FD_SETSIZE];

    if(argv[1]) 
        myport = atoi(argv[1]);
    else
        myport = 1234;

    if(argv[2])
        lisnum = atoi(argv[2]);
    else 
        lisnum = FD_SETSIZE;

    if((slisten = socket(AF_INET,SOCK_STREAM,0)) == -1)
    {
        perror("socket");
        exit(1);
    }

    bzero(&my_addr,sizeof(my_addr));
    my_addr.sin_family = AF_INET;
    my_addr.sin_port = htons(myport);
    my_addr.sin_addr.s_addr = INADDR_ANY;

    if(bind(slisten, (struct sockaddr *)&my_addr, sizeof(my_addr)) == -1) {
            perror("bind");
            exit(1);
        }

    if (listen(slisten, lisnum) == -1) {
            perror("listen");
            exit(1);
        }

    for(i=0;i<FD_SETSIZE;i++)
    {
        client[i].fd = -1;
    }

    FD_ZERO(&allset);           
        FD_SET(slisten, &allset);  
    maxfd = slisten;
    
    printf("Waiting for connections and data...\n");
    while (1)
    {        
        rset = allset;            

        tv.tv_sec = 1;
               tv.tv_usec = 0;
          
               nready = select(maxfd + 1, &rset, NULL, NULL, &tv);

        if(nready == 0)
            continue;
        else if(nready < 0)
        {
            printf("select failed!\n");
            break;
        }    
        else 
        {
            if(FD_ISSET(slisten,&rset)) // new connection
            {            
                len = sizeof(struct sockaddr);
                if((connectfd = accept(slisten,
                    (struct sockaddr*)&addr,&len)) == -1)
                {
                    perror("accept() error\n");
                    continue;
                }
                for(i=0;i<FD_SETSIZE;i++)
                {
                    if(client[i].fd < 0)
                    {
                        client[i].fd = connectfd;
                        client[i].addr = addr;                 
                        printf("Yout got a connection from %s.\n",
                        inet_ntoa(client[i].addr.sin_addr));
                        break;
                    }
                }
                if(i == FD_SETSIZE)            
                    printf("too many connections");            
                FD_SET(connectfd,&allset);
                if(connectfd > maxfd)
                    maxfd = connectfd;
                if(i > maxi)
                    maxi = i;
            }
            else
            {            
                for(i=0;i<=maxi;i++)
                {            
                    if((sockfd = client[i].fd)<0)
                        continue;                
                    if(FD_ISSET(sockfd,&rset))
                    {                        
                        bzero(buf,MAXBUF + 1);
                        if((n = recv(sockfd,buf,MAXBUF,0)) > 0)
                        {
    printf("received data:%s\n from %s\n",buf,inet_ntoa(client[i].addr.sin_addr));    
                        }                
                        else
                        {
                            printf("disconnected by client!\n");
                            close(sockfd);
                            FD_CLR(sockfd,&allset);
                            client[i].fd = -1;
                        }
                    }
                }
            }
        }    
    }
    close(slisten);
}

client.c程序：
－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <sys/socket.h>
#include <resolv.h>
#include <stdlib.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <sys/time.h>
#include <sys/types.h>

#define MAXBUF 1024

int main(int argc, char **argv)
{
        int sockfd, len;
        struct sockaddr_in dest;
        char buf[MAXBUF + 1];
        fd_set rfds;
        struct timeval tv;
    int retval, maxfd = -1;

    if (argc != 3) {
            printf("Usage: %s IP Port",argv[0]);
            exit(0);
        }

    if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
            perror("Socket");
            exit(errno);
        }

    bzero(&dest, sizeof(dest));
        dest.sin_family = AF_INET;
        dest.sin_port = htons(atoi(argv[2]));
        if (inet_aton(argv[1], (struct in_addr *) &dest.sin_addr.s_addr) == 0) {
            perror(argv[1]);
            exit(errno);
        }

    if(connect(sockfd, (struct sockaddr *) &dest, sizeof(dest)) != 0) {
            perror("Connect ");
            exit(errno);
        }

    printf("connect to server...\n");        
    while (1) 
    {

                    FD_ZERO(&rfds);           
                    FD_SET(0, &rfds);
                    maxfd = 0;
            
                    FD_SET(sockfd, &rfds);
                    if (sockfd > maxfd)
                        maxfd = sockfd;
            
                    tv.tv_sec = 1;
                    tv.tv_usec = 0;
          
                    retval = select(maxfd + 1, &rfds, NULL, NULL, &tv);

            if (retval == -1) 
            {
                        printf("select error! %s", strerror(errno));              
                break;
                } else if (retval == 0) {
                         //printf("no msg,no key, and continue to wait……\n"); 
                        continue;
                    } else {
                if (FD_ISSET(0, &rfds)) 
                {                
                                bzero(buf, MAXBUF + 1);
                                fgets(buf, MAXBUF, stdin);                
                                if (!strncasecmp(buf, "quit", 4)) 
                    {
                                    printf("request terminal chat！\n");
                                    break;
                                }
                                len = send(sockfd, buf, strlen(buf) - 1, 0);
                                if (len > 0)
                                    printf("msg:%s send successful，totalbytes: %d！\n", buf, len);
                                else {
                                    printf("msg:'%s  failed!\n", buf);
                                    break;
                                }
                        }
                        else if (FD_ISSET(sockfd, &rfds)) 
                { 
                                bzero(buf, MAXBUF + 1);
                                len = recv(sockfd, buf, MAXBUF, 0);
                                if (len > 0)
                                    printf("recv:'%s, total: %d \n", buf, len);
                                else  
                    {
                        if (len < 0) 
                                            printf("recv failed！errno:%d，error msg: '%s'\n", errno, strerror(errno));
                                    else
                                            printf("other exit，terminal chat\n");
                                    break;
                    }
                            }                
            }
    }

    close(sockfd);
    return 0;
}

编译通过！在处理多个连接请求时没有问题！http://jieastah.blog.sohu.com/84220706.html