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