C++ Socket通讯连接Server,并可设置连接超时时间

Socket通讯连接Server,并可设置连接超时时间

非阻塞模式:使用 ioctlsocket 将 socket 设置为非阻塞模式,这样 connect 调用不会阻塞主线程。

连接线程:在一个新线程中执行连接操作,使用 std::atomic 来检查连接是否成功。

超时逻辑:主线程使用一个循环来检查连接是否成功,若超出设定的超时时间则退出。

cpp 复制代码
#include <iostream>
#include <winsock2.h>
#include <ws2tcpip.h>
#include <thread>
#include <atomic>

#pragma comment(lib, "ws2_32.lib")


void ThreadConnect(SOCKET sock, struct sockaddr_in* server, std::atomic<bool>& connected)
{
    if (connect(sock, (struct sockaddr*)server, sizeof(*server)) == SOCKET_ERROR)
    {
        connected = false;
    }
    else
    {
        connected = true;
    }
}

bool ConnectServer()
{
    WSADATA wsaData;
    SOCKET sock;
    struct sockaddr_in server;

    // 初始化 Winsock
    if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
    {
        std::cerr << "Failed to initialize Winsock. Error: " << WSAGetLastError() << std::endl;
        return false;
    }

    // 创建 socket
    sock = socket(AF_INET, SOCK_STREAM, 0);
    if (sock == INVALID_SOCKET)
    {
        std::cerr << "Could not create socket. Error: " << WSAGetLastError() << std::endl;
        WSACleanup();
        return false;
    }

    // 设置 socket 为非阻塞模式
    u_long mode = 1; // 1 表示非阻塞模式
    ioctlsocket(sock, FIONBIO, &mode);

    // 配置服务器地址
    server.sin_family = AF_INET;
    server.sin_port = htons(7999); // 服务器端口
    inet_pton(AF_INET, "127.0.0.1", &server.sin_addr); // 服务器地址

    std::atomic<bool> connected(false);
    std::thread connectThread(ThreadConnect, sock, &server, std::ref(connected));

    // 等待连接或超时
    int connecTimeout = 1000;

    DWORD startTime = GetTickCount();
    while (GetTickCount() - startTime < connecTimeout)
    {
        if (connected)
        {
            break;
        }
        Sleep(100);
    }   
    connectThread.join(); // 关闭连接线程

    std::cout << "Connection time: " << (GetTickCount() - startTime) << " ms" << std::endl;

    if (!connected)
    {
        std::cerr << "Connection timeout." << std::endl;
        closesocket(sock);
        WSACleanup();

        return false;
    }

    mode = 0; // 设置为阻塞模式
    ioctlsocket(sock, FIONBIO, &mode);

    std::cout << "Connected to server!" << std::endl;


    int sendTimeout = 1000; // 发送超时设置
    int recvTimeout = 1000; // 接收超时设置

    if (setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO, (const char*)&sendTimeout, sizeof(sendTimeout)) < 0)
    {
        std::cerr << "Failed to set send timeout. Error: " << WSAGetLastError() << std::endl;
        closesocket(sock);
        WSACleanup();
        return false;
    }

    if (setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, (const char*)&recvTimeout, sizeof(recvTimeout)) < 0)
    {
        std::cerr << "Failed to set receive timeout. Error: " << WSAGetLastError() << std::endl;
        closesocket(sock);
        WSACleanup();
        return false;
    }

    // 这里可以添加发送和接收数据的代码

    // 关闭 socket
    closesocket(sock);
    WSACleanup();

    return true;
}

int main()
{

    ConnectServer();

    std::cin.get();

    return 0;
}
相关推荐
努力的小帅3 小时前
C++_哈希
开发语言·c++·学习·算法·哈希算法·散列表
阿让啊9 小时前
C语言strtol 函数使用方法
c语言·数据结构·c++·单片机·嵌入式硬件
liulilittle10 小时前
OPENPPP2 —— IP标准校验和算法深度剖析:从原理到SSE2优化实现
网络·c++·网络协议·tcp/ip·算法·ip·通信
田里的水稻13 小时前
C++_队列编码实例,从末端添加对象,同时把头部的对象剔除掉,中的队列长度为设置长度NUM_OBJ
java·c++·算法
Jayden_Ruan14 小时前
C++逆向输出一个字符串(三)
开发语言·c++·算法
liulun14 小时前
Skia如何渲染 Lottie 动画
c++·动画
点云SLAM15 小时前
C++ 常见面试题汇总
java·开发语言·c++·算法·面试·内存管理
UnnamedOrange17 小时前
ROS2 配置 linter 的代码格式化工具为 clang-format
c++·cmake
Dobby_0517 小时前
【面试题】C++系列(一)
c++·面经
一拳一个呆瓜17 小时前
【MFC】对话框节点属性:Language(语言)
c++·mfc