C++: websocket 连接超时处理

belien

已于 2023-04-11 17:35:00 修改

阅读量973

点赞数

分类专栏： C++ websocket 文章标签： c++ websocket 网络

于 2023-03-11 08:36:33 首次发布

本文链接：https://blog.csdn.net/weixin_43718346/article/details/129458647

版权

C++ 同时被 2 个专栏收录

14 篇文章 1 订阅

订阅专栏

websocket

4 篇文章 0 订阅

订阅专栏

头文件

#include <WinSock2.h>
#include <WS2tcpip.h>

1. 初始化

WSAStartup() 规范套接字版本号

WSADATA wsddata; // 套接字详细信息
WSAStartup(0x0202, &wsddata) != 0;

2. 创建套接字

socket() 创建指定面向网络或文件的规范套接字

SOCKET sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

3. 控制 I/O 模式

ioctlsocket() 控制套接字的 I/O 模式，ioctlsocket ；

int WSAAPI ioctlsocket(
  [in]      SOCKET sock, // 标识套接字描述字
  [in]      long   cmd, // 套接字上执行的命令
  [in, out] u_long *iMode // 指向 cmd 参数的指针
);

iMode : 为 0 时则为阻塞模式，非 0 时则为非阻塞模式（TCP 连接时，当网络异常连接失败时，会使得整个线程阻塞）；
cmd :
- FIONBIO，允许或禁止套接字的非阻塞模式；
- FIONREAD，确定套接字自动读入的数据量；
- SIOCATMARK，确定是否所有的带外数据都已被读入（仅适用于 SOCK_STREAM）。

返回值 : 成功返回 0，失败时返回 SOCKET_ERROR 错误代码：

错误代码	描述
WSANOTINITIALISED	未初始化 WSAStartup
WSAENETDOWN	套接字检测到网络子系统失效
WSAEINVAL	cmd 非法命令
WSAEINPROGRESS	阻塞的套接字调用正在运行中
WSAENOTSOCK	描述字不是一个套接字

4. 连接套接字

connect(sock, (struct sockaddr*)&address, sizeof(address)) == false;

5. 自定义连接

fd_set Write, Err;
FD_ZERO(&Write);
FD_ZERO(&Err);
FD_SET(sock, &Write);
FD_SET(sock, &Err);


int select ( 
    int nfds, // 0，仅为与伯克利Socket兼容而提供
    fd_set FAR * readfds, // 指定 socket 数组，存放符合‘可读性’的 socket 成员
    fd_set FAR * writefds, // 检查数组中 socket 成员，存放符合‘可写性’的成员
    fd_set FAR * exceptfds, // 检查所有成员，存放符合‘异常性’的成员 
    const struct timeval FAR * timeout // 指定 select 执行的最长时间
);

timeout :
- NULL，阻塞，直到 sock 位被置位 1 才返回；
- 0，结构内时间设为 0，非阻塞，检查完 sock 才返回；
- 为零，等待时间耗尽。

6. 检查连接

检查指定的 socket 是否存放集合里面。

FD_ISSET(sock, &Write);

7. 实例

#include <iostream>
#include <WinSock2.h>
#include <WS2tcpip.h>
#pragma comment(lib,"ws2_32.lib")

using namespace std;

bool connect(char* host, int port, int timeout)
{
    TIMEVAL Timeout;
    Timeout.tv_sec = timeout;
    Timeout.tv_usec = 0;
    struct sockaddr_in address;

    SOCKET sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

    //address.sin_addr.S_un.S_addr = inet_addr(host);
    inet_pton(AF_INET, host, &address.sin_addr);
    address.sin_port = htons(port);
    address.sin_family = AF_INET;

    //set the socket in non-blocking
    unsigned long iMode = 1;
    int iResult = ioctlsocket(sock, FIONBIO, &iMode);
    if (iResult != NO_ERROR)
    {
        printf("ioctlsocket failed with error: %ld\n", iResult);
    }

    if (connect(sock, (struct sockaddr*)&address, sizeof(address)) == false)
    {
        return false;
    }

    //restart the socket mode
    iMode = 0;
    iResult = ioctlsocket(sock, FIONBIO, &iMode);
    if (iResult != NO_ERROR)
    {
        printf("ioctlsocket failed with error: %ld\n", iResult);
    }

    fd_set Write, Err;
    FD_ZERO(&Write);
    FD_ZERO(&Err);
    FD_SET(sock, &Write);
    FD_SET(sock, &Err);

    //check if the socket is ready
    select(0, NULL, &Write, &Err, &Timeout);
    if (FD_ISSET(sock, &Write)) 
    {
        return true;
    }

    return false;
}

int main()
{
    WSADATA wsaData;
    if (WSAStartup(0x0202, &wsaData) != 0)
    {
        printf("WSAStartup() error!\n");
    }

    char sIP[] = "127.0.0.1";
    if (connect(sIP, 8008, 2000))
        printf("OK\n");
    else
        printf("NO\n");

    WSACleanup();
    system("pause");
    return 0;
}