【Linux高性能服务器编程】I/O复用的高级应用
文章目录
- 一、基于 select 的非阻塞 connect
- 二、基于 poll 的聊天室程序
-
- 2.1 客户端
- 2.2 服务器
- 三、基于 epoll 实现同时处理 TCP 和 UDP 服务
一、基于 select 的非阻塞 connect
connect系统调用的 man 手册中有如下一段内容：
EINPROGERESS
The socket is nonblocking and the connection cannot be completed immediately. It is possible to select(2) or poll(2) for completion by selecting the socket for writing. After select(2) indicates writability, use getsockopt(2) to read the SO_ERROR option at level SOL_SOCKET to determine whether connect() completed successfully (SO_ERROR is zero) or unsuccessfully (SO_ERROR is one of the usual error codes listed here, explaining the reason for the failure).
这段话描述了 connect 出错时的一种 errno 值：EINPROGRESS。这种错误发生在对非阻塞了socket调用了connect，而连接又没有建立时。根据man文档的解释，在这种情况下我们可以调用select、poll等函数来监听这个连接失败的socket上的可写事件。当select、poll等函数返回后，再利用getsockopt来读取错误码并清除该socket上的错误。如果错误码是0，表示连接成功，否则连接失败。
通过上面描述的非阻塞 connect 方式，我们就能同时发起多个连接并一起等待，下面是非阻塞connect的一种实现方式。
首先是设置文件描述符为非阻塞状态：
// 设置文件描述符为非阻塞
int setnonblocking(int fd)
{
int old_opt = fcntl(fd, F_GETFL);
int new_opt = old_opt | O_NONBLOCK;
fcntl(fd, F_SETFL, new_opt);
// 返回以前的文件描述符状态
return old_opt;
}
先使用fcntl函数获取并保存sockfd描述符的状态，然后再使用fcntl函数将其设置为非阻塞的，然后将原状态返回，便于在建立连接成功后恢复sockfd的状态。
实现unblock_connect函数：
// 非阻塞连接函数，参数是服务器IP、端口以及超时时间。成功返回处于连接状态的socket，失败则返回-1.
int unblock_connect(const std::string &ip, uint16_t port, int time)
{
int ret = 0;
struct sockaddr_in addr;
bzero(&addr, sizeof(addr));
addr.sin_family = AF_INET;
inet_pton(AF_INET, ip.c_str(), &addr.sin_addr);
addr.sin_port = htons(port);
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
// 设置套接字描述符为非阻塞，并保存其原来的状态
int fdopt = setnonblocking(sockfd);
ret = connect(sockfd, (const sockaddr *)&addr, sizeof(addr));
if (ret == 0)
{
// 连接成功，将sockfd恢复原来的状态，然后返回
std::cout << "connect with server immediately!" << std::endl;
fcntl(sockfd, F_SETFL, fdopt);
return sockfd;
}
else if (errno != EINPROGRESS)
{
// 如果连接没有立即建立，那么只有errno为EINPROGRESS时才表示连接还在继续，否则出错返回
std::cout << "unblock connect not support!" << std::endl;
close(sockfd);
return -1;
}
else
{
// 继续连接
// 写文件描述符集
fd_set writefds;
FD_ZERO(&writefds);
// 将sockfd添加到writefds中
FD_SET(sockfd, &writefds);
timeval timeout;
timeout.tv_sec = time;
timeout.tv_usec = 0;
ret = select(sockfd + 1, nullptr, &writefds, nullptr, &timeout);
if (ret <= 0)
{
// select超时或出错，立即返回
std::cout << "connect timeout!" << std::endl;
close(sockfd);
return -1;
}
if (!FD_ISSET(sockfd, &writefds))
{
std::cout << "no events on sockfd found!" << std::endl;
close(sockfd);
return -1;
}
int error = 0;
socklen_t len = sizeof(error);
// 调用 getsockopt来获取并清除sockfd上的错误
if (getsockopt(sockfd, SOL_SOCKET, SO_ERROR, &error, &len) < 0)
{
std::cout << "get socket option failed!" << std::endl;
close(sockfd);
return -1;
}
// 错误码不为0表示连接出错
if (error != 0)
{
std::cout << "conntion failed after select with the error: " << error << std::endl;
close(sockfd);
return -1;
}
// 连接成功，将sockfd恢复原来的状态，然后返回
std::cout << "connect successfully after select with the sockfd: " << sockfd << std::endl;
fcntl(sockfd, F_SETFL, fdopt);
return sockfd;
}
}
在该函数中，首先创建socket套接字，并将其设置为非阻塞的，然后调用connect函数，如果连接成功，恢复sockfd的原状态后返回；如果不能立即连接成功，只有返回的errno是EINPROGRESS才表示连接还在进行，否则直接出错返回。
当返回的errno是EINPROGRESS时，调用select多路复用系统调用对sockfd的写事件进行监听。监听成功后，调用getsockopt函数来获取并清除sockfd上的错误。如果错误码是0，则表示在调用select函数后建立连接成功，恢复sockfd的原状态后返回；否则出错返回。
main函数逻辑：
int main(int argc, char* argv[])
{
if(argc != 3)
{
std::cout << "usage: " << argv[0] << " ip port" << std::endl;
return -1;
}
const std::string ip = argv[1];
uint16_t port = atoi(argv[2]);
int sockfd = unblock_connect(ip, port, 10);
if(sockfd < 0)
{
std::cout << "unblock connect failed!" << std::endl;
return -1;
}
else
{
std::cout << "连接成功，可以向服务器发起请求了..." << std::endl;
}
close(sockfd);
return 0;
}
二、基于 poll 的聊天室程序
这里基于poll系统调用实现一个简单的聊天室程序，以阐述如何使用I/O复用技术来同时处理网络连接给用户输入。该聊天室程序能让所有的用户同时在线群聊，它分为客户端和服务端两个部分。
其中客户端有两个功能：一是从标准输入终端读入用户数据，并将用户数据发送至服务器；二是往标准输出终端打印服务器发送给它的数据。服务器的功能是接收客户端数据，并将接收到的数据发送给每一个登录到该服务器上的客户端（数据发送者除外）。
效果展示如下：
以下是客户端和服务端的代码。
2.1 客户端
客户端程序使用 poll 同时监听用户输入和网络连接，并利用 splice 函数将用户输入的内容直接定向到网络连接上以发送之，从而实现数据的零拷贝，提高了程序的运行效率。客户端代码如下：
#include 2.2 服务器
服务器程序使用poll同时管理监听 socket 和连接 socket，并且使用牺牲空间换取时间的策略来提高服务器的性能。服务器代码如下：
#include 三、基于 epoll 实现同时处理 TCP 和 UDP 服务
以上的两个服务器程序都只能监听一个端口，在实际应用中，有不少程序能够同时监听多个端口，比如超级服务inetd和 android 的调试服务adbd。
从bind系统调用的参数来看，一个socket只能与一个socket地址绑定，即一个socket只能监听一个端口。因此，如果一个程序想要监听多个端口，就必须创建多个socket，并将它们分别绑定到各个端口上。这样一来服务器程序就需要同时管理多个监听socket，I/O复用技术就有了用武之地了。另外，即使在同一个端口，如果服务器要同时处理该端口上的TCP和UDP请求，则也需要创建两个不同的socket：一个是流socket，另一个是数据报socket，如何把它们两个绑定到同一个端口上。
以下的回射服务器就能同时处理一个端口上的TCP和UDP请求：
#include 启动程序，使用netstat -nltup | grep multiport命令就可以看到该程序那个同时处理TCP和UDP请求了：
