[英]How to set timeout in UDP socket with C/C++ in WINDOWS?
我一直在尝试在 C/C++ 的 WINDOWS 的DatagramSocket
实现类中设置timeout
,但我不能,我尝试了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 标志进行编译。 我在 Windows 10 中使用 MinGW。我已经可以在 linux 中完成这个任务,但我不能在 Windows 中,我一直在互联网上寻找答案,但没有代码对我有用。 正如我所说,设置超时在 linux 中确实有效。
提前致谢。
在 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
仅适用于阻塞读取,不适用于非阻塞读取。 您显示的代码并未将套接字置于非阻塞模式,因此如果您使用SO_RCVTIMEO
,则使用select()
确实没有意义:
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),因此每次调用select()
都需要重置fds
变量。 因此,如果您打算使用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;
}
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.