网络编程0x05 select函数

2023-12-09 21:52
文章标签 函数 编程 网络 select 0x05

本文主要是介绍网络编程0x05 select函数,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

文章目录

    • 事件分类
      • 读事件就绪
      • 写事件就绪
      • 异常事件就绪
    • 接口API
    • select示例
    • nc安装和使用
      • nc安装
      • nc常用命令
    • select函数使用注释事项

select函数的作用是检测一组socket中某个或者某几个是否有"事件"就绪。

事件分类

读事件就绪

  • socket内核中,接收缓冲区中的子节数大于等于低水位标记SO_RCVLOWAT, 此时调用recv或者read函数可以无阻塞的读该描述符,并且返回值大于0
  • TCP对端关闭连接,此时调用recv或read函数对socket进行读,返回值为0
  • 监听fd上有新连接请求
  • socket上有未处理的错误

写事件就绪

  • socket内核中,发送缓冲区中的可用字节数(发送缓冲区的空闲位置)大于等于低水位标记SO_SNDLOWAT, 此时可以无阻塞的写,并且返回值大于0
  • socket的写操作被关闭(调用了close或者shutdown函数),对于一个写操作被关闭的socket进行写操作,会触发SIGPIPE信号
  • socket有非阻塞的connect连接成功或者失败之后

异常事件就绪

  • socket上收到带外数据

接口API

/* According to POSIX.1-2001 */
#include <sys/select.h>/* According to earlier standards */
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>int select(int nfds, fd_set *readfds, fd_set *writefds,fd_set *exceptfds, struct timeval *timeout);void FD_CLR(int fd, fd_set *set);
int  FD_ISSET(int fd, fd_set *set);
void FD_SET(int fd, fd_set *set);
void FD_ZERO(fd_set *set);

select参数说明:

  • nfds: 所有需要使用select函数监听fd中最大fd值加1
  • readfds: 需要监听可读事件的fd集合
  • writefds: 需要监听可写事件的fd集合
  • exceptfds: 需要监听异常事件的fd集合
  • timeout: 超时时间,即在这个参数设定的时间内检测这些fd事件,超过这个时间后select函数将立即返回。

相关结构体:

timeout超时时间的结构体为struct timeval

struct timeval {long    tv_sec;         /* seconds */long    tv_usec;        /* microseconds */
};

select函数超时总时间是timeout->tv_sec和timeout->tv_usec之后,前者单位是秒,后者单位是微秒

readfds/writefds/exceptfds的类型都是fd_set。其结构如下:

//  sys/select.h /* The fd_set member is required to be an array of longs.  */
typedef long int __fd_mask;/* Some versions of <linux/posix_types.h> define this macros.  */
#undef	__NFDBITS
/* It's easier to assume 8-bit bytes than to get CHAR_BIT.  */
#define __NFDBITS	(8 * (int) sizeof (__fd_mask))    // 64
#define	__FD_ELT(d)	((d) / __NFDBITS)
#define	__FD_MASK(d)	((__fd_mask) 1 << ((d) % __NFDBITS))/* fd_set for select and pselect.  */
typedef struct{/* XPG4.2 requires this member name.  Otherwise avoid the namefrom the global namespace.  */
#ifdef __USE_XOPEN__fd_mask fds_bits[__FD_SETSIZE / __NFDBITS];
# define __FDS_BITS(set) ((set)->fds_bits)
#else__fd_mask __fds_bits[__FD_SETSIZE / __NFDBITS];  // 1024/64 = 16
# define __FDS_BITS(set) ((set)->__fds_bits)
#endif} fd_set;/* Maximum number of file descriptors in `fd_set'.  */
#define	FD_SETSIZE		__FD_SETSIZE    // 1024/* Access macros for `fd_set'.  */
#define	FD_SET(fd, fdsetp)	__FD_SET (fd, fdsetp)
#define	FD_CLR(fd, fdsetp)	__FD_CLR (fd, fdsetp)
#define	FD_ISSET(fd, fdsetp)	__FD_ISSET (fd, fdsetp)
#define	FD_ZERO(fdsetp)		__FD_ZERO (fdsetp)extern int select (int __nfds, fd_set *__restrict __readfds,fd_set *__restrict __writefds,fd_set *__restrict __exceptfds,struct timeval *__restrict __timeout);/* We don't use `memset' because this would require a prototype andthe array isn't too big.  */
#define __FD_ZERO(s) \do {									      \unsigned int __i;							      \fd_set *__arr = (s);						      \for (__i = 0; __i < sizeof (fd_set) / sizeof (__fd_mask); ++__i)	      \__FDS_BITS (__arr)[__i] = 0;					      \} while (0)
#define __FD_SET(d, s) \((void) (__FDS_BITS (s)[__FD_ELT(d)] |= __FD_MASK(d)))
#define __FD_CLR(d, s) \((void) (__FDS_BITS (s)[__FD_ELT(d)] &= ~__FD_MASK(d)))
#define __FD_ISSET(d, s) \((__FDS_BITS (s)[__FD_ELT (d)] & __FD_MASK (d)) != 0)
  • FD_SET(fd, fdsetp)宏 将一个fd添加到fdsetp这个集合中
  • FD_CLR(fd, fdsetp)宏 从fdsetp集合中删除一个fd
  • FD_ISSET(fd, fdsetp)宏 检测对应位置上是否置1
  • FD_ZERO(fdsetp)宏 将fdsetp中所有fd都清除掉

select示例

/*** @brief select示例*/ 
#include <sys/select.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>#include <iostream>
#include <vector>
#include <mutex>std::mutex g_mutex;#ifndef LOG_MSG
#define LOG_MSG(x)  do {                                            \std::lock_guard<std::mutex> locker(g_mutex);\std::cout << x << std::endl;                \} while(0)
#endif //LOG_MSGconstexpr int32_t kInvalidFd = -1;int main(int argc, char *argv[])
{if (argc < 3){LOG_MSG("Usage: ");LOG_MSG("          " << argv[0] << " $port $ip");return -1;}// 创建监听套接字int listenfd = socket(AF_INET, SOCK_STREAM, 0);if (listenfd < 0){LOG_MSG("create socket fail");return -1;}// init server addrsockaddr_in bindaddr;bindaddr.sin_family = AF_INET;bindaddr.sin_addr.s_addr = inet_addr(argv[2]);      // htonl(INADDR_ANY);bindaddr.sin_port= htons(::atoi(argv[1]));if(bind(listenfd,(struct sockaddr*) &bindaddr, sizeof(bindaddr)) == -1){LOG_MSG("bind error");close(listenfd);return -1;}// start listenif(listen(listenfd, SOMAXCONN) == -1){LOG_MSG("listen error");close(listenfd);return -1;}std::vector<int> clientfds;char recvbuf[256];int maxfd = listenfd;do{fd_set readset;FD_ZERO(&readset);  FD_SET(listenfd, &readset);for (auto& item : clientfds){if (item != kInvalidFd){FD_SET(item, &readset);}}timeval tm;tm.tv_sec = 1;tm.tv_usec = 0;const auto ret = select(maxfd + 1, &readset, nullptr, nullptr, &tm);if (ret < 0){if (errno != EINTR)break;}else if (ret == 0){// timeoutcontinue;}else{if (FD_ISSET(listenfd, &readset)){// 检查监听fd上是否有读事件struct sockaddr_in clientaddr;socklen_t client_len = sizeof(clientaddr);int clientfd = accept(listenfd, (struct sockaddr*)&clientaddr, &client_len);if (clientfd < 0){// 连接出错,退出break;}LOG_MSG("accept a client connection, fd " << clientfd << " : " << inet_ntoa(clientaddr.sin_addr));clientfds.emplace_back(clientfd);if (clientfd > maxfd)maxfd = clientfd;}else{for (auto& clifd : clientfds){if (clifd != kInvalidFd && FD_ISSET(clifd, &readset)){memset(recvbuf, 0, sizeof(recvbuf));int length = recv(clifd, recvbuf, 256, 0);if (length <= 0 && errno != EINTR){LOG_MSG("recv data error, clienfd: " << clifd);close(clifd);clifd = kInvalidFd;continue;}LOG_MSG("clientfd: " << clifd << ", recv data: " << recvbuf);}}}}} while (true);for (const auto& clifd : clientfds){if (clifd != kInvalidFd){close(clifd);}}close(listenfd);return 0;
}

程序执行后输出结果如下:

# 启动服务端
➜  build ./test_select 8899 127.0.0.1
accept a client connection, fd 4 : 127.0.0.1
clientfd: 4, recv data: 33312222clientfd: 4, recv data: 455533322332clientfd: 4, recv data: 4444recv data error, clienfd: 4
# 启动客户端
➜  ~ nc -v 127.0.0.1 8899
Ncat: Version 7.50 ( https://nmap.org/ncat )
Ncat: Connected to 127.0.0.1:8899.
33312222
455533322332
4444

由于nc发送的数据是按换行符来区分的,每个数据包默认的换行符以\n结束,所以服务端收到数据后,显示出来的数据每一行下面都一个空白行。

nc安装和使用

nc安装

在CentOS上,可以使用以下命令安装nc:

yum update  
yum install nc

nc常用命令

  1. 打开一个终端并启动nc监听器:nc -lnvp <端口号>
# 监听端口号为1234的端口
nc -lnvp 1234
  1. 在另一个终端中,输入以下命令以连接到nc监听器:nc <IP地址> <端口号>
# 连接到IP地址为192.168.1.100的计算机上的端口号为1234
nc 192.168.1.100 1234

select函数使用注释事项

  1. select函数调用前后会修改readfds/writefds/exceptfds这3个集合中的内容,所以下次调用select复用这个变量,需要再次调用FD_ZERO将集合清理,然后调用FD_SET将需要检测的事件fd再次添加进去。
  2. select函数也会修改timeval结构体的值,如果我们需要复用这个变量,需要给timeval变量重新设置值。
  3. select函数的timeval结构体的tv_sec和tv_usec这个两个值设置为0, 即检测事件设置超时时间为0. 其行为是select会检测相关集合中的fd, 如果没有就绪的事件,则立即返回。

这篇关于网络编程0x05 select函数的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/475207

相关文章

MySQL 中的 CAST 函数详解及常见用法

《MySQL中的CAST函数详解及常见用法》CAST函数是MySQL中用于数据类型转换的重要函数,它允许你将一个值从一种数据类型转换为另一种数据类型,本文给大家介绍MySQL中的CAST... 目录mysql 中的 CAST 函数详解一、基本语法二、支持的数据类型三、常见用法示例1. 字符串转数字2. 数字

Python内置函数之classmethod函数使用详解

《Python内置函数之classmethod函数使用详解》:本文主要介绍Python内置函数之classmethod函数使用方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地... 目录1. 类方法定义与基本语法2. 类方法 vs 实例方法 vs 静态方法3. 核心特性与用法(1编程客

Python函数作用域示例详解

《Python函数作用域示例详解》本文介绍了Python中的LEGB作用域规则,详细解析了变量查找的四个层级,通过具体代码示例,展示了各层级的变量访问规则和特性,对python函数作用域相关知识感兴趣... 目录一、LEGB 规则二、作用域实例2.1 局部作用域(Local)2.2 闭包作用域(Enclos

Linux中压缩、网络传输与系统监控工具的使用完整指南

《Linux中压缩、网络传输与系统监控工具的使用完整指南》在Linux系统管理中,压缩与传输工具是数据备份和远程协作的桥梁,而系统监控工具则是保障服务器稳定运行的眼睛,下面小编就来和大家详细介绍一下它... 目录引言一、压缩与解压:数据存储与传输的优化核心1. zip/unzip:通用压缩格式的便捷操作2.

MySQL count()聚合函数详解

《MySQLcount()聚合函数详解》MySQL中的COUNT()函数,它是SQL中最常用的聚合函数之一,用于计算表中符合特定条件的行数,本文给大家介绍MySQLcount()聚合函数,感兴趣的朋... 目录核心功能语法形式重要特性与行为如何选择使用哪种形式?总结深入剖析一下 mysql 中的 COUNT

Go语言数据库编程GORM 的基本使用详解

《Go语言数据库编程GORM的基本使用详解》GORM是Go语言流行的ORM框架,封装database/sql,支持自动迁移、关联、事务等,提供CRUD、条件查询、钩子函数、日志等功能,简化数据库操作... 目录一、安装与初始化1. 安装 GORM 及数据库驱动2. 建立数据库连接二、定义模型结构体三、自动迁

MySQL 中 ROW_NUMBER() 函数最佳实践

《MySQL中ROW_NUMBER()函数最佳实践》MySQL中ROW_NUMBER()函数,作为窗口函数为每行分配唯一连续序号,区别于RANK()和DENSE_RANK(),特别适合分页、去重... 目录mysql 中 ROW_NUMBER() 函数详解一、基础语法二、核心特点三、典型应用场景1. 数据分

MySQL数据库的内嵌函数和联合查询实例代码

《MySQL数据库的内嵌函数和联合查询实例代码》联合查询是一种将多个查询结果组合在一起的方法,通常使用UNION、UNIONALL、INTERSECT和EXCEPT关键字,下面:本文主要介绍MyS... 目录一.数据库的内嵌函数1.1聚合函数COUNT([DISTINCT] expr)SUM([DISTIN

Python get()函数用法案例详解

《Pythonget()函数用法案例详解》在Python中,get()是字典(dict)类型的内置方法,用于安全地获取字典中指定键对应的值,它的核心作用是避免因访问不存在的键而引发KeyError错... 目录简介基本语法一、用法二、案例:安全访问未知键三、案例:配置参数默认值简介python是一种高级编

python 常见数学公式函数使用详解(最新推荐)

《python常见数学公式函数使用详解(最新推荐)》文章介绍了Python的数学计算工具,涵盖内置函数、math/cmath标准库及numpy/scipy/sympy第三方库,支持从基础算术到复杂数... 目录python 数学公式与函数大全1. 基本数学运算1.1 算术运算1.2 分数与小数2. 数学函数