阅读量: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函数获取连接结果。如果连接成功,可以进行后续操作。
注意,上述代码只是一个简单的示例,实际使用中可能需要根据具体情况进行适当的修改和优化。