c++ - 使用不带 listen() 的 select，可能吗？

Question

我正在构建一个客户端:

1. 应该能够从服务器和标准输入端接收信息
2. 应该能够在不询问的情况下从服务器接收信息，例如当另一个客户端发送消息时。

为此，我尝试使用 select 来监控两种可能的输入。

发生的事情是，当键盘输入被监视时，我向客户端发送了一条消息，我希望收到一条消息，所以没有问题。但是当服务器发送意外消息时，什么也没有发生，我不知道为什么。使用 select() 是正确的方法吗？甚至可以在没有 listen()ing 的情况下使用 select() 吗？

这是我的代码(可编译):

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

#define MAX_CLIENT_NAME 30
#define MAX_TWIT_SIZE 140
#define NUM_OF_ARG 4
#define ERROR -1
#define GREAT_SUCCESS 0
#define OK "OK"
#define EXIT "EXIT"


using std::string;
using std::cerr;
using std::endl;
using std::cout;

string clientName;

int srverfd, numbytes, status, maxSock ;

fd_set inputFdSet;        /* Socket file descriptors we want to wake
                       up for, using select() */

int establishConnection(char * serverAddress,char * port){
    if ((srverfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
        perror("socket");
        return ERROR;
    }
    struct sockaddr_in server;
    server.sin_family = AF_INET;
    inet_aton(serverAddress, &server.sin_addr);
    server.sin_port = htons(atoi(port));
    memset(&(server.sin_zero), '\0', 8);

    if (connect(srverfd,(const struct sockaddr *)&server,sizeof(struct sockaddr)) == -1) {
        perror("connect");
        close(srverfd);
        return ERROR;
    }
    maxSock = srverfd;
    return GREAT_SUCCESS;
}

const char * getUserTweet(){
    string temp;
getline(std::cin,temp);
    return temp.c_str();
}

void sendMessage(string message){
    if ((numbytes = send(srverfd, message.c_str(), message.length(), 0)) == -1) {
        perror("sendMessage");
        close(srverfd);
    }
    cout<<"Message sent: "<< message << endl;
    return;
}

const char * getMessage(){
    char buf[MAX_TWIT_SIZE];
    memset(buf,'\0',MAX_TWIT_SIZE);
    if ((numbytes = recv(srverfd, buf, 140, 0)) == -1) {
        perror("getMessage");
        close(srverfd);
    }
    string temp = buf;
    return temp.c_str();
}

void build_select_list() {
    FD_ZERO(&inputFdSet);
    FD_SET(srverfd,&inputFdSet);
    FD_SET(STDIN_FILENO,&inputFdSet);
    if (STDIN_FILENO > maxSock)
        maxSock = STDIN_FILENO;
    return;
}

void readSocket(fd_set tempfd) {
    const char * tweet, * inMessage;
    if (FD_ISSET(srverfd,&tempfd)) {
        inMessage = getMessage();
        cout << inMessage << endl;
    }

    if (FD_ISSET(STDIN_FILENO,&tempfd)) {
        tweet = getUserTweet();
        sendMessage(tweet);
        inMessage = getMessage();
        if (strcmp(inMessage,OK) != 0) {
            cout << inMessage << endl;
        }
        if (strcmp(inMessage,EXIT) == 0) {
            return;
        }
    }
    return;
}

int main (int argc, char *argv[] ){
    int value;
    bool clientON = false;
    if(establishConnection(argv[2],argv[3])){
        cerr << "usage: failed to make connection" << endl << "exiting..." << endl;
        exit(EXIT_FAILURE);
    }

    cout << "Connected successfully" << endl;
    sendMessage("CONNECT "+clientName); //Connect
    if(strcmp(getMessage(),OK) == 0){
        clientON = true;
    }
    while(clientON){
        build_select_list();
        value = select(maxSock, &inputFdSet, NULL, NULL, NULL);
        if (value < 0) {
            perror("select");
            exit(EXIT_FAILURE);
        }
        if (value == 0) {
            continue;
        }
        else {
            readSocket(inputFdSet);
        }
    }
    sendMessage("DISCONNECT");
    if(strcmp(getMessage(),OK) == 0){
        // do nothing
    }
    close(srverfd);
    return 0;
}

Answer 1

您的 select 调用无效。第一个参数必须是任何集合中最高的文件描述符，加一个。
如您所见， srverfd 上的事件不会“唤醒” select 调用(除非 STDIN_FILENO 以某种方式小于 srverfd ，在这种情况下，stdin 事件不会解锁 select - 但实际上不会发生)。

---

您的代码还有很多其他问题。 (它看起来不像 C++。)

getUserTweet 不可靠(未定义的行为 - temp 在函数返回后立即被销毁，因此您返回的 char* 已消失它的调用者将尝试使用它的时间)。 getMessage 也一样。要解决这个问题，请在任何地方使用 std::string，并且只在调用 C 库函数时提取 char*。

readSocket 不必要地复制 FD 集(可能很昂贵)。

您真的应该摆脱所有这些全局变量 - 构建一个或两个类来封装该状态和网络功能，或类似的东西。