c - 在 C 编程中的 select() 函数中使用 'writefds' 参数时出现问题

Question

这是与套接字编程相关的查询，服务器正在使用 select() 函数来查找准备读取的套接字。

根据 select() 函数的手册页。简而言之，“select 函数允许程序监视多个网络套接字，这些套接字已准备好读取，哪些 fd 已准备好写入”(暂时保留异常(exception))。以下两个程序，一个用于服务器程序，一个用于客户端程序。

我在 stackoverflow 中搜索了很多文章后提出了这个查询其他网站。我没有找到任何相关的解决方案。

我在 select() 中使用了下面的 readfds 代码并且工作正常。

服务器:使用readfds

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#include <arpa/inet.h>
#include <signal.h>
#include <string.h>

int main(void)
{
fd_set master;
fd_set read_fds;  // file descriptor list used for select()
int fdmax;
int listener;
int newfd;        // newly accepted socket descriptor
struct sockaddr_storage remoteaddr; // client address
socklen_t addrlen;
char remoteIP[INET6_ADDRSTRLEN], buf[20];
int yes=1;        // for setsockopt() SO_REUSEADDR, below
int i, rv, nbytes;
struct addrinfo hints, *ai, *p;

FD_ZERO(&master);
FD_ZERO(&read_fds);

memset(&hints, 0, sizeof hints);
hints.ai_family = AF_INET;
hints.ai_socktype = SOCK_STREAM;
hints.ai_flags = AI_PASSIVE;
if ((rv = getaddrinfo(NULL, "6000", &hints, &ai)) != 0) {
    fprintf(stderr, "selectserver: %s\n", gai_strerror(rv));
    return 1;
}

for(p = ai; p != NULL; p = p->ai_next) {
    listener = socket(p->ai_family, p->ai_socktype, p->ai_protocol);
    if (listener < 0) {
        continue;
    }

    if (setsockopt(listener, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes)) == -1) {
        perror("setsockopt");
        return 1;
    }

    if (bind(listener, p->ai_addr, p->ai_addrlen) < 0)
    {
        close(listener);
        continue;
    }

    break;
}

// if we got here, it means we didn't get bound
if (p == NULL) {
    printf("selectserver: failed to bind\n");
    return 1;
}

freeaddrinfo(ai); // all done with this

if (listen(listener, 5) == -1) {
    perror("listen");
    return 1;
}

FD_SET(listener, &master);

fdmax = listener; // remembering max fd

// main loop
for(;;)
{
    read_fds = master; // copying to read fds it
    if (select(fdmax+1, &read_fds, NULL, NULL, NULL) == -1) {
        perror("select");
        return 1;
    }

    // run through the existing connections looking for data to read
    for(i = fdmax; i >= 0; i--)
    {
        if (FD_ISSET(i, &read_fds))
        { // we got one!!
            if (i == listener)
            {
                // handle new connections
                addrlen = sizeof remoteaddr;
                newfd = accept(listener,
                    (struct sockaddr *)&remoteaddr,
                    &addrlen);

                printf("selectserver: new connection from %s on socket %d\n",
                        inet_ntop(remoteaddr.ss_family,
                        &(((struct sockaddr_in *)&remoteaddr)->sin_addr),
                        remoteIP, INET6_ADDRSTRLEN),
                        listener);

                //if (nbytes = send(newfd, "something", strlen("something"), 0) == -1)
                if ((nbytes = recv(newfd, buf, 20, 0)) <= 0)
                    printf("send failed\n");
                else
                {
                    buf[nbytes] = '\0';
                    printf("%s\n", buf);
                }
            } // END handle data from client
        } // END got new incoming connection
    } // END looping through file descriptors
    close(newfd);
}
close(listener);
return 0;
}

客户:

#include <stdio.h>
#include <unistd.h>
#include <netdb.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <string.h>
#include <arpa/inet.h>

int main(int argc, char *argv[])
{
int clientSocket;
struct sockaddr_in serverAddr;
socklen_t addr_size;
char buf[20];
int nbytes;
int rv;

struct addrinfo hints, *ai, *p;

memset(&hints, 0, sizeof hints);
hints.ai_family = AF_INET;
hints.ai_socktype = SOCK_STREAM;
//hints.ai_flags = AI_PASSIVE;

if ((rv = getaddrinfo(NULL, "6000", &hints, &ai)) != 0) {
  fprintf(stderr, "selectserver: %s\n", gai_strerror(rv));
  return 1;
}

for(p = ai; p != NULL; p = p->ai_next) {
    clientSocket = socket(p->ai_family, p->ai_socktype, p->ai_protocol);
  if (clientSocket < 0) {
      printf("error");
      continue;
  }
  break;
}

addr_size = sizeof serverAddr;
if (connect(clientSocket, (struct sockaddr *) p->ai_addr, p->ai_addrlen) == -1)
{
    printf("Connect failed\n");
    return 1;
}

printf("connect success\n");
//if ((nbytes = recv(clientSocket, buf, 20, 0)) <= 0)
if ((nbytes = send(clientSocket, "something", strlen("something"), 0)) == -1)
{
    printf("failed, Number of bytes: %d\n", nbytes);
    return 1;
}

return 0;
}

我在需要的地方添加了评论。

在服务器中，select()函数用于readfds(当其中一个readfds被设置时它将返回)，一旦返回，我将在接受套接字后从套接字读取。该程序现在正在按预期运行

在服务器程序中:

//if (nbytes = send(newfd, "something", strlen("something"), 0) == -1) <br>
if ((nbytes = recv(newfd, buf, 20, 0)) <= 0)

在客户端程序中:

//if ((nbytes = recv(clientSocket, buf, 20, 0)) <= 0)<br>
if ((nbytes = send(clientSocket, "something", strlen("something"), 0)) == -1)

------------------------------------------------------------ ----------------

我已取消注释服务器中的发送语句和客户端中的接收语句。就像下面这样，

在服务器程序中:

if (nbytes = send(newfd, "something", strlen("something"), 0) == -1)<br>
//if ((nbytes = recv(newfd, buf, 20, 0)) <= 0)

在客户端程序中:

if ((nbytes = recv(clientSocket, buf, 20, 0)) <= 0)<br>
//if ((nbytes = send(clientSocket, "something", strlen("something"), 0)) == -1)

------------------------------------------------------------ ----------------

现在程序被修改为服务器正在发送数据而客户端正在接收(请记住我在 select() 函数中使用“readfds”)。修改后的程序完全可以正常工作。由此我们可以说，select()函数中的参数'readfds'正在设置准备写入的fd，并且也在fd_set中设置这些fd准备写入。

**现在，在修改后的代码中，进行以下更改并保持其余代码不变，

if (select(fdmax+1, &read_fds, NULL, NULL, NULL) == -1) {<br>

替换为下面的行，

if (select(fdmax+1, NULL, &read_fds, NULL, NULL) == -1) {<br>

上述更改将使 select() 函数在找到准备写入的套接字时返回(从等待状态)。 (根据功能)。但是，“select()”根本不返回并等待套接字向其写入数据。并且在这里被阻止..这不符合预期。**

主要要求是，我有一台服务器在一台机器上运行，并等待在同一台机器上运行的两种类型的客户端套接字，这些套接字已准备好读取和写入。为了实现这一点，我通过指定 readfds(select() 中的第二个参数)和 writefds(select() 中的第三个参数)修改了 select 语句。哪个不起作用(即 writefds 未设置，问题已在前面解释过)。主要问题是，我无法区分套接字(准备好读写 fds)，因为 select() 函数仅在 readfds 中捕获读写 sock。

这是我在 stackoverflow 上的第一篇文章。所以，我尽量解释清楚，避免数据不足。如果需要添加/修改任何内容，请评论。

Answer 1

当您选择writefds参数时，该集合不包含任何准备写入的套接字。它不包含任何可供写入的套接字，因为您从未在 master 集中添加新的已接受套接字。

从技术上讲，您不会检查新接受的连接是否已准备好读取。

master集中唯一的套接字是监听套接字，并且它永远不可写。

解决方案:在接受套接字后将套接字添加到 master 集中。

<小时/>

顺便说一句，在检查就绪套接字时，您不应该真正循环遍历所有可能的文件描述符。只需执行例如

if (select(...) < 0)
{
    // Handle error
}

if (FD_ISSET(listener, &read_fds))
{
    // Handle new connection
}

要了解哪些已连接的客户端已发送数据，请将已连接的客户端保留在列表中并仅循环遍历该列表。