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

我需要这个服务器能够侦听和建立新的客户端连接，同时写入现有的连接。例如，异步非阻塞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，22000）；在循环内检查sockfd或newsockfd是否有任何活动，在这种情况下，我使用适当的读或写。如果有人能给我一些指导，我将非常感激。

最佳答案

内核将填充在revents数组的struct pollfd字段中发生的事件。
从手册页：
字段revents是一个输出参数，由内核用实际发生的事件填充。revents中返回的位可以包括events中指定的任何位，或者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，因为它需要更多的簿记才能可靠地完成。见下一段。
不幸的是，当使用“cc>”时，没有辅助每个连接数据存储状态的空间。根据您的平台不同，有多种选择，例如，ePoll for Linux、kQueue for*BSD和少量Windows选项。如果要对上下文数据使用poll，则必须使用可以使用文件描述符或数组索引进行搜索的数据结构。