Как установить тайм-аут в сокете UDP с C / C ++ в WINDOWS?

Question

Я пытался установить timeout в своем классе реализации DatagramSocket в C / C ++ для WINDOWS, но не могу, я пробовал setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, (char *) &timeout, sizeof(struct timeval)); как linux и использовал select функция, но без задержки, что я делаю не так?

Это файл интерфейса:

#ifndef __DatagramSocket_H__
#define __DatagramSocket_H__


#ifdef linux
    #include <sys/socket.h>
    #include <netinet/in.h>
    #include <arpa/inet.h>
#else
    // #include <ws2tcpip.h>
    #include <winsock2.h>
    // #pragma comment(lib,"ws2_32.lib") // Winsock Library, it only works in the microsoft compiler, MinGW ignores it
#endif

#include <iostream>
#include <strings.h>
#include <unistd.h>
#include <sys/time.h>

#include <errno.h>
#include "DatagramPacket.h"

class DatagramSocket {
public:
    DatagramSocket(uint16_t iport, const std::string & addr);
    ~DatagramSocket();

    void unbind();

    int send(DatagramPacket &);
    int receive(DatagramPacket &);

    void setTimeout(long, long);
    int receiveTimeout(DatagramPacket &, time_t, time_t);
private:
    sockaddr_in localAddress, remoteAddress;

    struct timeval timeout;
    bool timeout_set;
    #ifdef linux
        int s;
    #else
        SOCKET s;
        struct fd_set fds;
    #endif
};

#endif

DatagramPacket определяется как:

class DatagramPacket {
public:
    // data, len, ip, port
    DatagramPacket(char* , size_t, const string &, uint16_t );
    DatagramPacket(char* , size_t);
    DatagramPacket();
    ~DatagramPacket();
    string getAddress();
    char *getData();
    size_t getLength();
    uint16_t getPort();

    void setAddress(const string &);
    void setData(char* , size_t);
    void setLength(size_t);
    void setPort(uint16_t);

private:
    char *data;
    uint16_t port;
    string ip;
    size_t length;
};

определение функций-членов из класса DatagramSocket:

DatagramSocket::DatagramSocket(uint16_t iport, const std::string &addr): timeout_set(false) {
    #ifdef _WIN32  // detect windows of 32 and 64 bits
        WSAData wsaData;
        WORD word = MAKEWORD(2, 2);
        if (WSAStartup(word, &wsaData) != 0) {
            std::cerr << "Server: WSAStartup failed with error: " << WSAGetLastError() << std::endl;
            exit(1);
        }
    #endif 

    s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
    // bzero((char *)&localAddress, sizeof(localAddress));
    memset((char *) &localAddress, 0, sizeof(localAddress));
    localAddress.sin_family = AF_INET;
    localAddress.sin_addr.s_addr = inet_addr(addr.c_str());
    localAddress.sin_port = htons(iport);
    bind(s, (struct sockaddr *) &localAddress, sizeof(localAddress));
}

void DatagramSocket::setTimeout(long secs, long u_secs) {
    timeout = { 
        .tv_sec = secs, 
        .tv_usec = u_secs };  
    timeout_set = true;
    #ifdef __linux__
        setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, (char *) &timeout, sizeof(struct timeval));
    #else 
        FD_ZERO(&fds);
        FD_SET(s, &fds);
    #endif
}


int DatagramSocket::receiveTimeout(DatagramPacket &p, time_t secs, time_t u_secs) {
    int len = sizeof(remoteAddress);
    if (!timeout_set) setTimeout(secs, u_secs);  
    #ifdef _WIN32
        int sret = select(0, &fds, NULL, NULL, &timeout);
        if (sret == 0) {
            cout << "Timeout! " << sret << endl;
            return NULL;
        }
    #endif
    int n = recvfrom(s, p.getData(), p.getLength(), 0, (struct sockaddr *) &remoteAddress, &len);
    if (n < 0) {  // deal with errors 
        #ifdef __linux__
            if (errno == EWOULDBLOCK) 
                fprintf(stderr, "Timeout \n");
            else 
                fprintf(stderr, "Error in recvfrom. \n");
        #endif
        return n;
    }
    p.setPort(remoteAddress.sin_port);  
    p.setAddress(string(inet_ntoa(remoteAddress.sin_addr)));
    p.setLength(n);
    return n;
}

Я включил библиотеки ws2tcpip.h и winsock2.h и скомпилировал с флагом -lwsock32. Я использую MinGW в Windows 10. Я уже мог выполнить эту задачу в linux, но не смог в Windows, я искал ответ в inte rnet, но ни один код не помог мне , Как я уже сказал, установка времени ожидания работает в linux.

Заранее спасибо.

Answer 1

В Windows, SO_RCVTIMEO требуется DWORD с указанием миллисекунд, а не timeval структура, как на других платформах:

void DatagramSocket::setTimeout(long secs, long u_secs) {
    timeout = { 
        .tv_sec = secs, 
        .tv_usec = u_secs };  
    timeout_set = true;

    #ifdef _WIN32
    DWORD dw = (secs * 1000) + ((u_secs + 999) / 1000);
    setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, (char *) &dw, sizeof(dw));
    #else
    setsockopt(s, SOL_SOCKET, SO_RCVTIMEO, (char *) &timeout, sizeof(struct timeval));
    #endif
}

Также обратите внимание, что SO_RCVTIMEO применяется только к блокировке читает, не для неблокирующих чтений. Код, который вы показали, не переводит сокет в неблокирующий режим, поэтому нет смысла использовать select(), если вы используете SO_RCVTIMEO:

int DatagramSocket::receiveTimeout(DatagramPacket &p, time_t secs, time_t u_secs) {
    setTimeout(secs, u_secs);  
    int len = sizeof(remoteAddress);
    int n = recvfrom(s, p.getData(), p.getLength(), 0, (struct sockaddr *) &remoteAddress, &len);
    if (n < 0) {  // deal with errors 
        #ifdef _WIN32
        if (WSAGetLastError() == WSAETIMEDOUT) 
        #else
        if (errno == EAGAIN || errno == EWOULDBLOCK)
        #endif
            std::cout << "Timeout!" << std::endl;
        else
            std::cerr << "Error in recvfrom." << std::endl;
        return -1;
    }
    p.setPort(remoteAddress.sin_port);  
    p.setAddress(string(inet_ntoa(remoteAddress.sin_addr)));
    p.setLength(n);
    return n;
}

Что касается самого select(), он изменяет переданные fd_set (s) на выходе, поэтому вам нужно сбрасывать переменную fds каждый раз, когда вы вызываете select(). Таким образом, если вы собираетесь использовать select(), то вам следует переместить переменные select() в receiveTimeout() локально, и не использовать SO_RCVTIMEO:

int DatagramSocket::receiveTimeout(DatagramPacket &p, time_t secs, time_t u_secs) {
    timeval timeout = { 
        .tv_sec = secs, 
        .tv_usec = u_secs };  

    fd_set fds;
    FD_ZERO(&fds);
    FD_SET(s, &fds);

    int nfds;
    #ifdef _WIN32
    nfds = 0;
    #else
    nfds = s + 1;
    #endif

    int sret = select(nfds, &fds, NULL, NULL, &timeout);
    if (sret <= 0) {
        if (sret == 0) {
            std::cout << "Timeout!" << std::endl;
        else
            std::cerr << "Error in select." << std::endl;
        return -1;
    }

    int len = sizeof(remoteAddress);
    int n = recvfrom(s, p.getData(), p.getLength(), 0, (struct sockaddr *) &remoteAddress, &len);
    if (n < 0) {  // deal with errors 
        std::cerr << "Error in recvfrom." << std::endl;
        return -1;
    }
    p.setPort(remoteAddress.sin_port);  
    p.setAddress(string(inet_ntoa(remoteAddress.sin_addr)));
    p.setLength(n);
    return n;
}