分别在linux和windows上设置socket为阻塞模式

最新推荐文章于 2024-09-12 07:47:14 发布

原创最新推荐文章于 2024-09-12 07:47:14 发布 · 1.4k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#linux #运维 #服务器 #c++ #windows

C++基础实战同时被 2 个专栏收录

41 篇文章

订阅专栏

C++后台开发

13 篇文章

订阅专栏

博客介绍了在Linux和Windows系统中，将socket设置为非阻塞模式的方法。在Linux系统中可使用特定函数设置，还提供了直接将返回的socket设为非阻塞的函数；在Windows系统中也有相应函数用于设置。此外，还封装了两个函数方便调用。

在 Linux 和 Windows 系统中，都可以将 socket 设置为非阻塞模式。

Linux平台

在 Linux 系统中，可以使用 fcntl 函数来设置 socket 为非阻塞模式。例如：

int flags = fcntl(socket_fd, F_GETFL, 0);
fcntl(socket_fd, F_SETFL, flags | O_NONBLOCK);

此外，Linux 还提供了一个 accept4 函数，可以直接将返回的 socket 设置为非阻塞模式：

int client_fd = accept4(server_fd, (struct sockaddr *)&client_addr, &client_addr_len, SOCK_NONBLOCK);

设置阻塞模式，默认阻塞模式不用单独设置

int flags = fcntl(socket_fd, F_GETFL, 0);
fcntl(socket_fd, F_SETFL, flags | ~O_NONBLOCK);

Windows平台

在 Windows 系统中，可以使用 ioctlsocket 函数来设置 socket 为非阻塞模式。例如：

u_long mode = 1;
ioctlsocket(socket_fd, FIONBIO, &mode);

设置阻塞模式

u_long mode = 0;
ioctlsocket(socket_fd, FIONBIO, &mode);

封窗两个函数方便调用

设置非阻塞模式
void SetNonBlocking(SOCKET sockfd)
设置阻塞模式
void SetBlocking(SOCKET sockfd)

#ifdef __GNUC__
#define SOCKET int
#elif defined(_WIN32)
#endif


void SetNonBlocking(SOCKET sockfd){
#ifdef _WIN32
    unsigned long flag = 1;
    if(ioctlsocket(sockfd, FIONBIO, &flag) == SOCKET_ERROR){
#else
    int cflags = fcntl(sockfd, F_GETFL, 0); 
    if(fcntl(sockfd, F_SETFL, cflags | O_NONBLOCK) == -1){
#endif
        std::cerr << "ioctlsocket or fcntl set non blocking error" << std::endl;
        exit(-1);
    }
}

void SetBlocking(SOCKET fd){
#ifdef _WIN32
    unsigned long flag = 0;
    if(ioctlsocket(sockfd, FIONBIO, &flag) == SOCKET_ERROR){
#else
    int cflags = fcntl(sockfd, F_GETFL, 0); 
    if(fcntl(sockfd, F_SETFL, cflags & ~O_NONBLOCK) == -1){
#endif
        std::cerr << "ioctlsocket or fcntl set non blocking error" << std::endl;
        exit(-1);
    }
}