linux 客户端 Socket 非阻塞connect编程

avatar
作者
筋斗云
阅读量:3

在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函数获取连接结果。如果连接成功,可以进行后续操作。

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

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!