1. 相关概念
在UNIX/Linux下主要有4种I/O 模型:
阻塞I/O:
最常用、最简单、效率最低
非阻塞I/O:
可防止进程阻塞在I/O操作上,需要轮询
I/O 多路复用:
允许同时对多个I/O进行控制
信号驱动I/O:
一种异步通信模型
2. 阻塞IO
阻塞IO几乎是任何一个读写函数都有的性质,比例读函数,
比如要读取管道中的数据,
如果管道中有数据则立即读取,
如果管道中没有数据则需要阻塞等待,
当管道中有数据时,linux内核会立即唤醒当前进程,然后执行读取操作
#include <stdio.h>
#include <unistd.h>
int main(int argc, char const *argv[])
{
char buf[128] = {0};
while(1)
{
fgets(buf, 128, stdin);
sleep(1);
printf("welcome to hqyj\n");
}
return 0;
}
3. 非阻塞IO
如果将一个函数设置为了非阻塞,以读函数读管道数据为例,
如果管道中有数据,读函数正常执行,
如果管道中没有数据,读函数会立即错误返回
如果设置为非阻塞,需要自己写一个循环轮询判断是否有数据,所以很浪费资源
这种模式使用中不普遍
有些函数例如waitpid、recv、recvfrom等可以通过原本的参数就可以设置为非阻塞,
但是大多数函数都没有提供这个标志位,所以需要通过一个函数去实现,
使用fcntl实现非阻塞IO
WNOHANG MSG_DONTWAIT O_NONBLOCK
#include <unistd.h>
#include <fcntl.h>
int fcntl(int fd, int cmd, ... /* arg */ );
功能:
操作一个文件描述符
参数:
fd:
文件描述符
cmd:
具体操作
F_GETFL 获取文件状态标志位
F_SETFL 设置文件状态标志位
O_NONBLOCK 非阻塞
arg:
可变参,是否需要由cmd决定(如果cmd后面括号里面是int类型,则需要这个参数,如果是void类型,则不需要)
返回值:
成功:
F_GETFL 文件状态标志位
F_SETFL 0
失败:
-1
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
int main(int argc, char const *argv[])
{
char buf[128] = {0};
//使用fcntl函数实现非阻塞
//注意:我们在对位或者寄存器操作时,只需要改变对应的位即可,
// 且不能修改其他位
// 所以一般对位或者寄存器的操作,我们要执行三步:读、改、写
//第一步:获取之前的文件状态标志位
int flags;
if((flags = fcntl(0, F_GETFL)) == -1)
{
perror("fcntl error");
exit(1);
}
//第二步:修改为指定的标志位
//将阻塞变成非阻塞,其他位不变
flags = flags | O_NONBLOCK;
//第三步:将新的标志位写回去
if(fcntl(0, F_SETFL, flags) == -1)
{
perror("fcntl error");
exit(1);
}
while(1)
{
//如果设置为非阻塞,缓冲区中没有数据,fgets会错误返回
if(fgets(buf, 128, stdin) == NULL)
{
perror("fgets error");
}
printf("buf = %s\n", buf);
sleep(1);
printf("welcome to hqyj\n");
}
return 0;
}