linux

linux 客户端 Socket 非阻塞connect编程

小亿
79
2023-12-20 19:40:43
栏目: 编程语言

在Linux中,可以使用非阻塞IO(Non-blocking IO)来实现非阻塞connect的编程。

以下是一个简单的示例代码,演示如何在Linux中使用非阻塞connect进行Socket编程:

#include <iostream>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <unistd.h>

int main() {
    // 创建Socket
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd == -1) {
        std::cerr << "Failed to create socket." << std::endl;
        return 1;
    }

    // 设置Socket为非阻塞
    int flags = fcntl(sockfd, F_GETFL, 0);
    if (flags == -1) {
        std::cerr << "Failed to get socket flags." << std::endl;
        return 1;
    }
    if (fcntl(sockfd, F_SETFL, flags | O_NONBLOCK) == -1) {
        std::cerr << "Failed to set socket to non-blocking mode." << std::endl;
        return 1;
    }

    // 设置连接目标地址
    struct sockaddr_in server_addr{};
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(12345);
    server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");

    // 尝试非阻塞连接
    int ret = connect(sockfd, (struct sockaddr*)&server_addr, sizeof(server_addr));
    if (ret == -1) {
        if (errno == EINPROGRESS) {
            // 连接正在进行中,等待连接完成
            fd_set write_fds;
            FD_ZERO(&write_fds);
            FD_SET(sockfd, &write_fds);

            struct timeval timeout{};
            timeout.tv_sec = 5;  // 设置连接超时时间为5秒

            ret = select(sockfd + 1, nullptr, &write_fds, nullptr, &timeout);
            if (ret == -1) {
                std::cerr << "Failed to select." << std::endl;
                return 1;
            } else if (ret == 0) {
                // 连接超时
                std::cerr << "Connect timeout." << std::endl;
                return 1;
            } else {
                // 连接完成,检查连接是否成功
                int error = 0;
                socklen_t len = sizeof(error);
                if (getsockopt(sockfd, SOL_SOCKET, SO_ERROR, &error, &len) == -1) {
                    std::cerr << "Failed to get socket option." << std::endl;
                    return 1;
                }
                if (error != 0) {
                    std::cerr << "Failed to connect: " << strerror(error) << std::endl;
                    return 1;
                }
            }
        } else {
            std::cerr << "Failed to connect: " << strerror(errno) << std::endl;
            return 1;
        }
    }

    std::cout << "Connected successfully." << std::endl;

    // 连接成功后,可以进行后续操作

    close(sockfd);

    return 0;
}

在上述代码中,首先创建了一个Socket,并将其设置为非阻塞模式。然后,通过connect函数进行非阻塞连接。如果connect返回-1,并且errno为EINPROGRESS,说明连接正在进行中,此时可以通过select函数等待连接完成。如果select返回0,表示连接超时。如果select返回大于0,说明连接已完成,可以通过getsockopt函数获取连接结果。如果连接成功,可以进行后续操作。

注意,上述代码只是一个简单的示例,实际使用中可能需要根据具体情况进行适当的修改和优化。

0
看了该问题的人还看了