fcntl
一个文件描述符,默认都是阻塞IO
函数原型如下:
#include<unistd.h>
#include<fcntl.h>
int fcntl(int fd, int cmd, .../* arg */);
传入的cmd的值不同,后面追加的参数也不相同
fcntl函数有五种功能:
- 复制一个现有的描述符(cmd = F_DUPFD)
- 获得/设置文件描述符标记(cmd=F_GETFD或F_SETFD)
- 获得/设置文件状态标记(cmd=F_GETFL或F_SETFL)
- 获得/设置异步IO所有权(cmd=F_FETOWN或F_SETOWN)
- 获得/设置记录锁(cmd= F_GETLK,F_SETLK或F_SETLKW)
我们此处只用第三种功能,就可以将一个文件描述符设置为非阻塞
实现函数SetNoBlock
基于fcntl我们实现一个SetNoBlock,将文件描述符设置为非阻塞
void SetNoBlock(int fd)
{
int f1 = fcntl(fd, F_GETFL);
if (f1 < 0)
{
perror("fcntl");
return;
}
fcntl(fd, F_SETFL, f1 | O_NONBLOCK);//按位或,将文件描述符设置为非阻塞
}
//使用F_GETFL将当前的文件描述符的属性取出来(这是一个位图).
//然后再使用F_SETFL将文件描述符设置回去. 设置回去的同时, 加上一个O_NONBLOCK参数.
轮询方式读取标准输入
#include<stdio.h>
#include<unistd.h>
#include<fcntl.h>
void SetNoBlock(int fd)
{
int f1 = fcntl(fd, F_GETFL);//将当前文件的描述符属性取出来
if(f1 < 0)
{
perror("fcntl");
return;
}
fcntl(fd, F_SETFL, f1 | O_NONBLOCK);//在使用F_SETFL将文件描述符设置回去,设置回去的同时,加一个O_ONOBLOCK
}
int main()
{
SetNoBlock(0);
while(1)
{
char buff[1024] = {0};
ssize_t read_size = read(0, buff, sizeof(buff)-1);
if(read_size < 0)
{
perror("read");
sleep(1);
continue;
}
printf("input: %s\n", buff);
}
return 0;
}