在C编程中使用select()函数中的'writefds'参数的问题

时间:2015-06-15 13:45:52

标签: c sockets

这是查询与套接字编程有关,服务器正在使用select()函数找出可以读取的套接字。

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

我在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(当其中一个readfd被设置时将返回)并且一旦返回,我将在接受套接字后从套接字读取。这个程序按预期工作,现在

在服务器程序中:

//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)

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

我在服务器和客户端的recv语句中使用UN评论了send语句。如下,

在服务器程序中:

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'设置了准备写入的fds,并且设置那些fds已经准备好在fd_set中写入。

**现在,根据修改后的代码,执行以下更改并保留其余代码,

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上的第一篇文章。因此,我试图清楚地解释,以避免数据不足。如果需要添加/修改任何内容,请发表评论。

1 个答案:

答案 0 :(得分:1)

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

从技术上讲,您不会检查新接受的连接是否已准备就绪。

master集中唯一的套接字是侦听套接字,它永远不可写。

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

另外,在检查就绪套接字时,您不应该真正遍历所有可能的文件描述符。就这样做。

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

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

要知道哪些已连接的客户端已发送数据,请将连接的客户端保留在列表中,并仅循环显示该列表。