非阻塞的SOCKET通讯

jidongah

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

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define MAXBUF 1024
typedef struct CLIENT {
    int fd;
    struct sockaddr_in addr;   
}CLIENT;
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 maxfd)
                    maxfd = connectfd;
                if(i > maxi)
                    maxi = i;
            }
            else
            {            
                for(i=0;i 0)
                        {
    printf("received data:%s\n from %s\n",buf,inet_ntoa(client.addr.sin_addr));   
                        }               
                        else
                        {
                            printf("disconnected by client!\n");
                            close(sockfd);
                            FD_CLR(sockfd,&allset);
                            client.fd = -1;
                        }
                    }
                }
            }
        }   
    }
    close(slisten);
}
client.c程序：
－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－－
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#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


本文来自ChinaUnix博客，如果查看原文请点：http://blog.chinaunix.net/u2/66624/showart_542115.html