poll服务器

#include 
   
   
    
    
#include 
    
    
     
     
#include 
     
     
      
      
#include 
      
      
       
       
#include 
       
       
         #include 
        
          #include 
         
           #include 
          
            #define array_size 1024 int startup(const char* _ip, int _port) { int sock = socket(AF_INET, SOCK_STREAM, 0); if(sock < 0) { perror("create sock is fail"); return 2; } struct sockaddr_in peer; peer.sin_family = AF_INET; peer.sin_port = htons(_port); peer.sin_addr.s_addr = inet_addr(_ip); int opt = 1; if(setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)) == -1) { perror("setsockopt"); return 3; } if(bind(sock, (struct sockaddr*)&peer, sizeof(peer)) < 0) { perror("bind"); return 4; } if(listen(sock, 10) < 0) { perror("listen"); return 5; } return sock; } static void Usage(const char* proc) { printf("%s[ip][port]\n", proc); } int main(int argc, char *argv[]) { if(argc != 3) { Usage(argv[0]); return 1; } int listensock = startup(argv[1], atoi(argv[2])); struct pollfd array_fds[1024]; array_fds[0].fd = listensock; array_fds[0].events = POLLIN; int i = 1; for(; i < array_size; i++) { array_fds[i].fd = -1; } int _timeout = 1000; while(1) { switch(poll(array_fds, 1024, _timeout)) { case 0: printf("timeout......"); break; case -1: perror("poll"); break; default: { int j = 0; for(; j < array_size; j++) { if(j == 0 && array_fds[0].revents & POLLIN) { struct sockaddr_in client; socklen_t len = sizeof(client); int new_fd = accept(array_fds[j].fd, (struct sockaddr*)&client, &len); if(new_fd < 0) { perror("new_fd: accept"); continue; }else { printf("connect is success! ip: %s port: %d\n", inet_ntoa(client.sin_addr), ntohs(client.sin_port)); int k = 1; for(; k < array_size; k++) { if(array_fds[k].fd < 0) { array_fds[k].fd = new_fd; array_fds[k].events = POLLIN; break; } } if(k == array_size) { printf("连接超出最大限度"); return; } } } else if(j != 0 && array_fds[j].revents & POLLIN) { char buf[1024]; memset(buf, 0, sizeof(buf)); ssize_t s = read(array_fds[j].fd, buf, sizeof(buf)-1); if(s > 0) { buf[s-1] = '\0'; printf("client say# %s\n", buf); array_fds[j].events = POLLOUT; }else if(s == 0) { printf("client is quit!"); close(array_fds[j].fd); array_fds[j].fd = -1; }else { perror("read"); close(array_fds[j].fd); array_fds[j].fd = -1; } } else if(j != 0 && array_fds[j].revents & POLLOUT) { const char* msg = "HTTP/1.1 200 OK \r\n\r\n 
           
hello poll
"; write(array_fds[j].fd, msg, strlen(msg)); close(array_fds[j].fd); array_fds[j].fd = -1; } } } break; } } return 0; }