select_socket.c

来自「select socket linux网络编程」· C语言 代码 · 共 67 行

/*select_socket.c*/   
#include <SYS types.h>    
#include <SYS socket.h>    
#include <SYS wait.h>    
#include <STDIO.H>    
#include <STDLIB.H>    
#include <ERRNO.H>    
#include <STRING.H>    
#include <SYS un.h>    
#include <SYS time.h>    
#include <SYS ioctl.h>    
#include <UNISTD.H>    
#include <NETINET in.h>    
#define SERVPORT 3333    
#define BACKLOG 10    
#define MAX_CONNECTED_NO 10    
#define MAXDATASIZE 100    
int main()   
{   
    struct sockaddr_in server_sockaddr,client_sockaddr;   
    int sin_size,recvbytes;   
    fd_set readfd;   
    fd_set writefd;   
    int sockfd,client_fd;   
    char buf[MAXDATASIZE];   
    if((sockfd = socket(AF_INET,SOCK_STREAM,0))==-1){   
        perror("socket");   
        exit(1);   
    }   
    printf("socket success!,sockfd=%d\n",sockfd);   
    server_sockaddr.sin_family=AF_INET;   
    server_sockaddr.sin_port=htons(SERVPORT);   
    server_sockaddr.sin_addr.s_addr=INADDR_ANY;   
    bzero(&(server_sockaddr.sin_zero),8);   
    if(bind(sockfd,(struct sockaddr *)&server_sockaddr,sizeof(struct sockaddr))==-1){   
        perror("bind");   
        exit(1);   
    }   
    printf("bind success!\n");   
    if(listen(sockfd,BACKLOG)==-1){   
        perror("listen");   
        exit(1);   
    }   
    printf("listening....\n");   
/*将调用socket函数的描述符作为文件描述符*/   
    FD_ZERO(&readfd);   
    FD_SET(sockfd,&readfd);   
    while(1){   
    sin_size=sizeof(struct sockaddr_in);   
/*调用select函数*/   
    if(select(MAX_CONNECTED_NO,&readfd,NULL,NULL,(struct timeval *)0)>0){   
        if(FD_ISSET(sockfd,&readfd)>0){   
            if((client_fd=accept(sockfd,(struct sockaddr *)&client_sockaddr,&sin_size))==-1){   
                perror("accept");   
                exit(1);   
            }   
               
            if(read(client_fd,buf,MAXDATASIZE)<0){   
                perror("read");   
                exit(1);   
            }   
            printf("received a connection :%s",buf);   
        }/*if*/   
        close(client_fd);   
        }/*select*/   
    }/*while*/   
}