在TCP服务器上实现poll()读/写

时间:2013-10-24 05:49:41

标签: c sockets client-server polling

我需要这台服务器能够监听并与客户建立新的连接,同时写入现有的连接..即。异步非阻塞i / o。我被告知使用poll()但是在花了大量时间只是试图掌握套接字编程之后,我仍然不确定如何实现poll()函数。

int sockfd; 

int main(int argc, char *argv[])
{
 int newsockfd, portno; 
 socklen_t clilen;  
 char buffer[256];       
 struct sockaddr_in serv_addr, cli_addr;  
 int n;             

 if (argc < 2) {         
     fprintf(stderr,"ERROR, no port provided\n"); 
     exit(1);
 }

 sockfd = socket(AF_INET, SOCK_STREAM, 0);      
 if (sockfd < 0)                    
    error("ERROR opening socket");          

 bzero((char *) &serv_addr, sizeof(serv_addr));     

 portno = atoi(argv[1]);            
 serv_addr.sin_family = AF_INET;        
 serv_addr.sin_addr.s_addr = INADDR_ANY;    
 serv_addr.sin_port = htons(portno);    

 if (bind(sockfd, (struct sockaddr *) &serv_addr, 
          sizeof(serv_addr)) < 0) 
          error("ERROR on binding");        
 listen(sockfd,5);                  

 clilen = sizeof(cli_addr);     

 while(1){          
     newsockfd = accept(sockfd, 
                 (struct sockaddr *) &cli_addr, 
                 &clilen);              
     if (newsockfd < 0)                 
          error("ERROR on accept");

     // READ READ READ READ READ READ READ READ READ READ READ READ READ READ READ READ 
     bzero(buffer,256);
     n = read(newsockfd,buffer,255);    


     if (n < 0) error("ERROR reading from socket");
     printf("Here is the message: %s\n",buffer);

     // WRITE WRITE WRITE WRITE WRITE WRITE WRITE WRITE WRITE WRITE WRITE WRITE WRITE 
     n = write(newsockfd,"I got your message",18); 
     if (n < 0) error("ERROR writing to socket");
     close(newsockfd);  
 }

 return 0; 

}

我的理解是我需要建立这样的东西:

//  Set up array of file descriptors for polling
struct pollfd ufds[2];
ufds[0].fd = sockfd;    
ufds[0].events = POLLIN; 

ufds[1].fd = newsockfd;
ufds[1].events = POLLOUT;

并使用poll(ufds,2,2000);在循环内部检查sockfd或newsockfd是否有任何活动,在这种情况下我使用适当的读或写..如果有人可以给我一些指导我会非常感激。

2 个答案:

答案 0 :(得分:3)

内核将填写revents数组的struct pollfd字段中发生的事件。

从手册页:

  

字段revents是一个输出参数,由内核填充实际发生的事件。 revents中返回的位可以包括事件中指定的任何位,或POLLERR,POLLHUP或POLLNVAL中的值之一。 (这三个位在事件字段中没有意义,只要相应的条件为真,就会在revents字段中设置。)

如果您希望接受连接的事件通知,则需要提前预留空间或为每个连接调整struct pollfd数组的大小。

您需要一些方法来区分侦听套接字。您可以将它存储在数组的索引0中。

int i, n;

n = poll(ufds, num_fds_in_array, timeout_value);

/* errors or timeout? */
if (n < 1)
    ;

for (i = 0; i < num_fds_in_array; i++) {
    /* were there any events for this socket? */
    if (!ufds[i].revents)
        continue;

    /* is it our listening socket? */
    if (!i) {
         if (ufds[0].revents & POLLIN)
             /* call accept() and add the new socket to ufds */
         else
             /* error */

         continue;
    }

    /* is there incoming data on the socket? */
    if (ufds[i].revents & POLLIN)
        /* call recv() on the socket and decide what to do from there */
}

POLLOUT标志用于表示套接字上的发送数据何时不会阻塞调用者。

对于非阻塞I / O,我会使用更强大的API,因为它需要更多的簿记来可靠地执行。见下一段。

不幸的是,使用poll时,没有空间可以使用辅助的每个连接数据来存储状态。根据您的平台,有其他可用选项,例如: G。适用于Linux的epoll,适用于* BSD的kqueue以及适用于Windows的少数选项。如果要将poll与上下文数据一起使用,则必须使用可以使用文件描述符或数组索引搜索的数据结构。

答案 1 :(得分:0)

你为什么不使用libevent?它完全异步且无阻塞。 http://libevent.org/