如何在write_fds中正确设置一个标志,在C中选择select()

时间:2016-06-15 14:36:07

标签: c sockets select nonblocking

send()缓冲区已满时,我在给定套接字的write_fds中设置了一个标志,下一次迭代尝试通过检查write_fds是否已设置来发送它。

在一个套接字的write_fds中设置标志可以正常工作。但是,如果我为多个套接字执行FD_SET(i, write_fds),则只有最后一个套接字显示为set,并且只有最后一个套接字才能获取数据。

如果我在write_fds = master;之前select(2),则所有这些都显示为已设置且所有未发送的数据都会发送到所有套接字,但select始终返回。

我确信我一定做错了。

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

// socket setting...
// bind...
// listen...

FD_SET(listener, &master);
fdmax = listener; // so far, it's this one
for(;;){
    read_fds = master; // copy it
    // write_fds = master;
    select(fdmax+1, &read_fds, &write_fds, NULL, NULL);

    for(i = 0; i <= fdmax; i++){
        if(FD_ISSET(i, &read_fds)) {
            if(i == listener){
                addrlen = sizeof remoteaddr;
                newfd = accept(listener, (struct sockaddr * ) & remoteaddr, &addrlen);
                FD_SET(newfd, &master);
                if(newfd > fdmax)
                    fdmax = newfd;
                fcntl(newfd, F_SETFL, fcntl(newfd, F_GETFL, 0) | O_NONBLOCK);
            }else{
                // do some reading and then... send
                sent = send(i, buf, len, 0);
                if ((sent < 0 || (sent == -1 && errno == EWOULDBLOCK))
                    FD_SET(i, &write_fds);
            }  
        }  
        else if(FD_ISSET(i, &write_fds) && i != listener){
            if((sent = send(i, buf, len - sent, 0)) == len - sent)
                FD_CLR(i, &write_fds);
        }  
    }  
}

2 个答案:

答案 0 :(得分:1)

在调用select函数之前,您应该有FD_ZEROFD_SET

E.g。

int res = 0;
for(;;){
     FD_ZERO(&read_fds);
     FD_ZERO(&write_fds);    

     fdmax = 0;
     size_t fd_index;
     for (fd_index=0; fd_index<num_of_fd; fd_index++)
     {
        FD_SET(fd_array[fd_index], &read_fds);  
        FD_SET(fd_array[fd_index], &write_fds);  

        if (fdmax < fd_array[fd_index])
           fdmax = fd_array[fd_index];
     }

    // write_fds = master;
    res = select(fdmax+1, &read_fds, &write_fds, NULL, NULL);

    if( res == 0 )
    {
       // TIMEOUT ...not your case
    }
    else if( res < 0 )
    {
        perror("Select error: ");
    }
    else
    {
        // process your data
    }

你不能写

read_fds = master;

因为fd_set是一个复杂的结构。

修改

由于@jeremyP评论sent < 0应为sent > 0

答案 1 :(得分:0)

在:

FD_SET(i, write_fds);

应该是:

FD_SET(i, &write_fds);

(注意&符号)。