网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。
一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!
1.6 使用sendmsg()函数发送数据
函数sendmsg()可用于向多个缓冲区发送数据,函数原型如下所示。函数sendmsg()向套接字描述符s中按照结构msg的设定写入数据,其中操作方式由flags指定。
#include <sys/uio.h>
ssize\_t sendmsg(int s, const struct msghdr\* msg, int flags);
2. select()函数和pselect()函数
2.1 select()函数
select()函数是Linux中的一个系统调用,用于监视文件描述符的变化情况,包括读、写、异常等。它可以等待多个文件描述符中的任何一个变为“准备好”的状态,从而完成 I/O 操作。select()函数的语法如下:
#include <sys/select.h>
int select(int nfds, fd_set \*readfds, fd_set \*writefds, fd_set \*exceptfds, struct timeval \*timeout);
其中,nfds 是需要监视的文件描述符数量,readfds、writefds 和 exceptfds 分别是需要监视的读、写和异常事件的文件描述符集合。timeout 是超时时间,如果在超时时间内没有任何事件发生,则 select() 函数返回 0。
select()函数返回值:
负值:select错误
正值:表示某些文件可读或可写
通常会和如下几个函数一起使用:
void FD\_ZERO(fd_set \*set);//清空一个文件描述符的集合
void FD\_SET(int fd, fd_set \*set);//将一个文件描述符添加到一个指定的文件描述符集合中
void FD\_CLR(int fd, fd_set \*set);//将一个指定的文件描述符从集合中清除;
int FD\_ISSET(int fd, fd_set \*set);//检查集合中指定的文件描述符是否可以读写
举个例子,下面的代码使用 select() 函数实现了一个简单的 TCP 服务器:
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>
int main(int argc, char \*argv[]) {
int listen_fd = socket(AF_INET, SOCK_STREAM, 0);
if (listen_fd == -1) {
perror("socket");
exit(EXIT_FAILURE);
}
struct sockaddr\_in addr;
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_addr.s_addr = htonl(INADDR_ANY);
addr.sin_port = htons(8080);
if (bind(listen_fd, (struct sockaddr \*)&addr, sizeof(addr)) == -1) {
perror("bind");
exit(EXIT_FAILURE);
}
if (listen(listen_fd, 10) == -1) {
perror("listen");
exit(EXIT_FAILURE);
}
fd_set read_fds;
FD\_ZERO(&read_fds);
FD\_SET(listen_fd, &read_fds);
while (1) {
fd_set tmp_fds = read_fds;
if (select(listen_fd + 1, &tmp_fds, NULL, NULL, NULL) == -1) {
perror("select");
exit(EXIT_FAILURE);
}
for (int i = 0; i <= listen_fd; i++) {
if (FD\_ISSET(i, &tmp_fds)) {
if (i == listen_fd) {
int conn_fd = accept(listen_fd, NULL, NULL);
if (conn_fd == -1) {
perror("accept");
exit(EXIT_FAILURE);
}
printf("accept connection %d\n", conn_fd);
FD\_SET(conn_fd, &read_fds);
} else {
char buf[1024];
ssize\_t n = recv(i, buf, sizeof(buf), 0);
if (n <= 0) {
printf("close connection %d\n", i);
close(i);
FD\_CLR(i, &read_fds);
} else {
buf[n] = '\0';
printf("recv from connection %d: %s", i, buf);
}
}
}
}
}
return 0;
}
这个服务器使用 select() 函数监听了一个 TCP 端口,并在有新连接到来时接受连接。当有数据到来时,它会打印出数据内容。这个服务器可以同时处理多个连接,并且不会阻塞在任何一个连接上。
2.2 pselect()函数
pselect() 函数是 select() 函数的一个变种,它可以在等待期间阻塞指定的信号。pselect() 函数的语法如下:
#include <sys/select.h>
int pselect(int nfds, fd_set \*readfds, fd_set \*writefds, fd_set \*exceptfds, const struct timespec \*timeout, const sigset\_t \*sigmask);
其中,sigmask 是一个指向信号集的指针,用于阻塞指定的信号。timeout 是超时时间,如果在超时时间内没有任何事件发生,则 pselect() 函数返回 0。
pselect() 函数与 select() 函数的区别在于,它可以在等待期间阻塞指定的信号。这个特性可以用于实现更加复杂的 I/O 操作,例如同时等待多个文件描述符和多个信号。
举个例子,下面的代码使用 pselect() 函数实现了一个简单的 TCP 服务器:
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>
int main(int argc, char \*argv[]) {
int listen_fd = socket(AF_INET, SOCK_STREAM, 0);
if (listen_fd == -1) {
perror("socket");
exit(EXIT_FAILURE);
}
struct sockaddr\_in addr;
memset(&addr, 0, sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_addr.s_addr = htonl(INADDR_ANY);
addr.sin_port = htons(8080);
if (bind(listen_fd, (struct sockaddr \*)&addr, sizeof(addr)) == -1) {
perror("bind");
exit(EXIT_FAILURE);
}
if (listen(listen_fd, 10) == -1) {
perror("listen");
exit(EXIT_FAILURE);
}
fd_set read_fds;
FD\_ZERO(&read_fds);
FD\_SET(listen_fd, &read_fds);
sigset\_t sigmask;
sigemptyset(&sigmask);
sigaddset(&sigmask, SIGINT);
while (1) {
fd_set tmp_fds = read_fds;
struct timespec timeout = { .tv_sec = 1 };
if (pselect(listen_fd + 1, &tmp_fds, NULL, NULL, &timeout, &sigmask) == -1) {
perror("pselect");
exit(EXIT_FAILURE);
}
for (int i = 0; i <= listen_fd; i++) {
if (FD\_ISSET(i, &tmp_fds)) {
if (i == listen_fd) {
int conn_fd = accept(listen_fd, NULL, NULL);
if (conn_fd == -1) {
perror("accept");
exit(EXIT_FAILURE);
}
printf("accept connection %d\n", conn_fd);
FD\_SET(conn_fd, &read_fds);
} else {
char buf[1024];
ssize\_t n = recv(i, buf, sizeof(buf), 0);
if (n <= 0) {
printf("close connection %d\n", i);
close(i);
FD\_CLR(i, &read_fds);
} else {
buf[n] = '\0';
printf("recv from connection %d: %s", i, buf);
}
}
}
}
}
return 0;
}
这个服务器使用 pselect() 函数监听了一个 TCP 端口,并在有新连接到来时接受连接。当有数据到来时,它会打印出数据内容。这个服务器可以同时处理多个连接,并且不会阻塞在任何一个连接上。
pselect() 函数和 select() 函数基本上是一致的,但是有三个区别:
- select 函数用的 timeout 参数,是一个 timeval 的结构体(包含秒和微秒),而 pselect 用的是一个 timespec 结构体(包含秒和纳秒)。
- select 函数可能会为了指示还剩多长时间而更新 timeout 参数,而 pselect 不会改变 timeout 参数。
- select 函数没有 sigmask 参数,当 pselect 的 sigmask 参数为 null 时,两者行为时一致的。
因此,当需要阻塞指定的信号时,可以使用 pselect() 函数。
以下是一个需要阻塞信号的 pselect() 函数的例子:
#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/select.h>
#include <sys/time.h>
#include <unistd.h>
static void sig\_handler(int signo)
{
printf("Caught signal %d\n", signo);
}
int main(void)
{
fd_set rfds;
struct timeval tv;
int retval;
/\* Watch stdin (fd 0) to see when it has input. \*/
FD\_ZERO(&rfds);
FD\_SET(0, &rfds);
/\* Wait up to five seconds. \*/
tv.tv_sec = 5;
tv.tv_usec = 0;
/\* Set up the signal handler. \*/
struct sigaction sa;
sa.sa_handler = sig_handler;
sigemptyset(&sa.sa_mask);
sa.sa_flags = 0;
if (sigaction(SIGINT, &sa, NULL) == -1) {
perror("sigaction");
exit(EXIT_FAILURE);
}
/\* Block SIGINT and wait for input. \*/
sigset\_t mask;
sigemptyset(&mask);
sigaddset(&mask, SIGINT);
if (sigprocmask(SIG_BLOCK, &mask, NULL) == -1) {
perror("sigprocmask");
exit(EXIT_FAILURE);
}
retval = pselect(1, &rfds, NULL, NULL, &tv, &mask);
if (retval == -1)
perror("pselect()");
else if (retval)
printf("Data is available now.\n");
else
printf("No data within five seconds.\n");
exit(EXIT_SUCCESS);
}
3. poll()函数和ppoll()函数
3.1 poll()函数
poll()函数是Linux系统上的一个用于执行I/O多路复用的函数,与select()函数类似。poll()函数的机制与select()类似,管理多个描述符也是进行轮询,根据描述符的状态进行处理,但是poll()没有最大文件描述符数量的限制(但是数量过大后性能也是会下降)。poll()函数需要一个pollfd结构类型的数组,用于存放需要检测其状态的Socket描述符。每当有一个Socket描述符状态发生变化时,poll()函数就会在这个数组中找到对应的元素,然后修改这个元素中revents成员的值,表明这个Socket描述符上发生了什么事件。
poll()函数的声明如下:
#include <poll.h>
int poll(struct pollfd fds[], nfds\_t nfds, int timeout);
其中,fds是一个struct pollfd结构类型的数组,用于存放需要检测其状态的Socket描述符;nfds表示fds数组中元素的数量;timeout表示poll()函数等待的时间(单位为毫秒),如果timeout为0,则表示立即返回;如果timeout为-1,则表示无限等待。
以下是一个简单的poll()函数的例子,用于监测按键按下的事件,如果按下了就将键值打印出来;如果超过5S,还没有按键按下,就打印出超时信息:
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <poll.h>
int main(void)
{
struct pollfd fds;
int ret;
char c;
fds.fd = STDIN_FILENO;
fds.events = POLLIN;
ret = poll(&fds, 1, 5000);
if (ret == -1) {
perror("poll");
return 1;
}
if (!ret) {
printf("timeout\n");
return 0;
}
if (fds.revents & POLLIN) {
scanf("%c", &c);
printf("input = %c\n", c);
}
return 0;
}
POLLIN是poll()函数的一个参数,用于指定需要监测的事件类型。POLLIN表示需要监测的是输入事件,即文件描述符是否可读。其他常见的事件类型还有POLLOUT(文件描述符是否可写)和POLLERR(文件描述符是否出错)等。
3.2 ppoll()函数
ppoll()函数是Linux系统下的一个系统调用,用于监测多个文件描述符的状态,类似于poll()函数。与poll()函数不同的是,ppoll()函数可以指定一个sigmask参数,用于指定需要屏蔽的信号集合。ppoll()函数的函数原型如下:
#include <poll.h>
int ppoll(struct pollfd \*fds, nfds\_t nfds,
const struct timespec \*timeout_ts,
const sigset\_t \*sigmask);
其中,fds参数是一个指向pollfd结构体数组的指针,用于指定需要监测的文件描述符;nfds参数表示fds数组中元素的个数;timeout_ts参数表示超时时间;sigmask参数表示需要屏蔽的信号集合。ppoll()函数返回值与poll()函数相同,表示就绪文件描述符的个数。
其区别同函数select()和pselect()的区别相同,主要有两点:
- 超时时间timeout,采用了纳秒级的变量。
- 可以在ppoll()函数的处理过程中挂接临时的信号掩码。
4. 非阻塞编程
非阻塞方式的操作与阻塞方式的操作最大的不同点是函数的调用立刻返回,不管数据是否成功读取或者成功写入。使用fcntl()将套接字文件描述符按照如下的代码进行设置后,可以进行非阻塞的编程;
fcntl(s, F_SETFL, O_NONBLOCK);
其中的s是套接字文件描述符,使用F_SETFL命令将套接字s设置为非阻塞方式后,再进行读写操作就可以马上返回了.
网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。
一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!
ntl(s, F_SETFL, O_NONBLOCK);
其中的s是套接字文件描述符,使用F\_SETFL命令将套接字s设置为非阻塞方式后,再进行读写操作就可以马上返回了.
[外链图片转存中...(img-C0Kz7LaM-1715671597446)]
[外链图片转存中...(img-CEqC3soc-1715671597446)]
**网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。**
**[需要这份系统化的资料的朋友,可以添加戳这里获取](https://bbs.csdn.net/topics/618668825)**
**一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!**