c-如何将一个简单的客户端-服务器TCP程序转换为非阻塞程序

Question

嗨，我在Beej的指南中读到了使用select()的非阻塞调用，但我仍然不知道如何将我的简单客户端-服务器代码更改为非阻塞代码。有人能告诉我我需要对服务器代码和客户端代码进行哪些更改吗？

这是服务器代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h> 
#include <sys/socket.h>
#include <netinet/in.h>
void error(const char *msg)
{
    perror(msg);
    exit(1);
}
int main(int argc, char *argv[])
{
     int sockfd, 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 providedn");
         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);
     newsockfd = accept(sockfd, 
                 (struct sockaddr *) &cli_addr, 
                 &clilen);
     if (newsockfd < 0) 
          error("ERROR on accept");
     bzero(buffer,256);
     n = read(newsockfd,buffer,255);
     if (n < 0) error("ERROR reading from socket");
     printf("Here is the message: %sn",buffer);
     n = write(newsockfd,"I got your message",18);
     if (n < 0) error("ERROR writing to socket");
     close(newsockfd);
     close(sockfd);
     return 0; 
}

这是客户端代码：

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h> 
void error(const char *msg)
{
    perror(msg);
    exit(0);
}
int main(int argc, char *argv[])
{
    int sockfd, portno, n;
    struct sockaddr_in serv_addr;
    struct hostent *server;
    char buffer[256];
    if (argc < 3) {
       fprintf(stderr,"usage %s hostname portn", argv[0]);
       exit(0);
    }
    portno = atoi(argv[2]);
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0) 
        error("ERROR opening socket");
    server = gethostbyname(argv[1]);
    if (server == NULL) {
        fprintf(stderr,"ERROR, no such hostn");
        exit(0);
    }
    bzero((char *) &serv_addr, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    bcopy((char *)server->h_addr, 
         (char *)&serv_addr.sin_addr.s_addr,
         server->h_length);
    serv_addr.sin_port = htons(portno);
    if (connect(sockfd,(struct sockaddr *) &serv_addr,sizeof(serv_addr)) < 0) 
        error("ERROR connecting");
    printf("Please enter the message: ");
    bzero(buffer,256);
    fgets(buffer,255,stdin);
    n = write(sockfd,buffer,strlen(buffer));
    if (n < 0) 
         error("ERROR writing to socket");
    bzero(buffer,256);
    n = read(sockfd,buffer,255);
    if (n < 0) 
         error("ERROR reading from socket");
    printf("%sn",buffer);
    close(sockfd);
    return 0;
}

Answer 1

select(2)使事情不阻塞的方法是等待I/O成为可能而不阻塞（例如，当新数据可用时）。当使用select()时，通常不必将受监控的描述符置于非阻塞模式（例如SOCK_NONBLOCK），因此从某种意义上讲，select()特别是关于避免必须使用非阻塞I/O。

CCD_ 6用于同时等待多个描述符上的事件。这里的事件可以使read(2)从描述符或（取决于如何使用select()）write(2)到描述符而不阻塞。

例如，您可以使用select()同时等待新的客户端连接和来自服务器中已连接客户端的数据（假设您将其扩展为处理多个客户端）。为此，您将使用select()来监视sockfd和从accept(2)返回的任何描述符。如果没有select()，您将不得不使用某种形式的非阻塞I/O（或单独的线程），以避免在新客户端连接之前被卡在例如accept()中，这将使您在此期间无法看到来自其他客户端的数据。这将比睡在一个地方更难实现，而且效率也更低。

select()本身不进行任何I/O操作。它只会在I/O成为可能时通知您，而不会阻塞。您向它传递一组描述符，它"告诉"您何时可以对其中任何一个进行I/O。（同时告诉你可以在哪些描述符上。）

除了等待套接字上的数据外，您还可以使用select()同时等待例如stdin上的用户输入。有许多不同类型的描述符可以在上select()。