如何在socket编程中使用select和FD_SET?

时间:2012-02-06 18:01:30

标签: linux sockets select

我是socket编程的新手,我无法理解select()FD_SET()的工作原理。

我修改了Beej教程中的一个例子,试图找出它。我想在for循环中做的是每次迭代我等待4秒。如果有可用的读取,我会打印“按下一个键”,如果超时,则会打印“超时”。然后我会清除该设置并重复该过程9次。但似乎一旦设置了文件描述符0,即使在调用FD_ZERO()和/或FD_CLR()之后也永远不会取消设置。换句话说,在循环的第一次迭代中按下一个键之后,将为剩余的迭代设置文件描述符,并且不再进行等待。所以我必须有一些我想念的东西,但我不知道是什么。

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

#define SERVERPORT 4950

int main(int argc, char *argv[]) {
    struct sockaddr_in their_addr; // connector's address information
    struct hostent *he;
    int numbytes;
    int broadcast = 1;

    if ((he=gethostbyname(argv[1])) == NULL) {  // get the host info
        perror("gethostbyname");
        exit(1);
    }

    // this call is what allows broadcast packets to be sent:
    if (setsockopt(sockfd, SOL_SOCKET, SO_BROADCAST, &broadcast,
        sizeof broadcast) == -1) {
        perror("setsockopt (SO_BROADCAST)");
        exit(1);
    }
    their_addr.sin_family = AF_INET;     // host byte order
    their_addr.sin_port = htons(SERVERPORT); // short, network byte order
    their_addr.sin_addr = *((struct in_addr *)he->h_addr);
    memset(their_addr.sin_zero, '\0', sizeof their_addr.sin_zero);

    struct timeval tv;
    fd_set broadcastfds;

    int i;
    for(i=0; i < 10; i++) {
        tv.tv_sec = 4;
        tv.tv_usec = 500000;

        FD_ZERO(&broadcastfds);
        FD_CLR(0, &broadcastfds);
        FD_SET(0, &broadcastfds);
        if(select(0+1, &broadcastfds, NULL, NULL, &tv) == -1) perror("select");

        if (FD_ISSET(0, &broadcastfds)) printf("A key was pressed!\n");
        else printf("Timed out.\n");
        fflush(stdout); 
    }   
    close(sockfd);
    return 0;
}

2 个答案:

答案 0 :(得分:5)

您正在使用FD_SET。当文件描述符0(标准输入)准备好读取时,您要求select()通知您。它做到了这一点。问题是您没有读取标准输入以消耗可用的输入。因此,当您循环并再次调用select()时,标准输入仍然可以读取,并立即返回。

使用select()(或poll(),这通常是更好的选择)的正确方法是:

  • 将所涉及的所有文件描述符设置为非阻塞模式。对于大多数用例,您需要这样做,因为您希望在select()(或poll())内进行所有阻止,而不是在为单个文件描述符提供服务时。
  • 设置select()poll()的参数,以注册您感兴趣的文件描述符。
  • 致电select()poll()
  • 通过消耗输入和写入输出来对它给出的通知做出反应。
  • 循环回到第2步。

P.S。:你的UDP套接字sockfd与什么有什么关系?你打开它但它没有被用于任何东西。

答案 1 :(得分:4)

问题是你永远不会从文件描述符中读取数据。

select()报告状态,而非事件。

所以在第一次select()返回后,总有数据可供阅读,所以select()会立即报告。

PS。无论你从哪里获得代码,它看起来大约15岁。 poll()通常比select()更方便,getaddrinfo()比gethostbyname()更方便。 [而且他们的工作也更好]。