select 非阻塞 sock服务端客户端代码实例

sever：

#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：

#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锛宼otalbytes: %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锛乪rrno:%d锛宔rror msg: '%s'\n", errno, strerror(errno));  
                                    else  
                                            printf("other exit锛宼erminal chat\n");  
                                    break;  
                    }  
                            }                 
            }  
    }  
  
    close(sockfd);  
    return 0;  
}