Linux环境下使用fcntl实现C语言非阻塞输入

Linux环境下使用fcntl实现C语言非阻塞输入

在使用C语言编程时,常常会用到scanf()getchar()等函数来获取键盘的输入,当键盘没有输入时程序会发生阻塞并一直保持在阻塞状态,此时如果想要继续执行下一步的程序,就需要将IO状态置为非阻塞,以确保无用户输入的情况下程序的正常执行。

fcntl是Linux中的函数,用来改变已打开文件的性质,其定义为int fcntl(int fd, int cmd);,其中参数fd表示欲设置的文件描述符,参数cmd代表将要执行的操作,如果该操作需要键入参数,fcntl能够接收第三个参数int arg。返回值与命令相关,如果出错,所有命令都返回-1,如果成功则返回某个其他值。

用法举例:

#include <stdio.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>


int main()
{
    
    int flag1, flag2;
    if(flag1=(fcntl(STDIN_FILENO, F_GETFL, 0)) < 0)
    {
        perror("fcntl");
        return -1;
    }
    flag2 = flag1 | O_NONBLOCK;
    fcntl(STDIN_FILENO, F_SETFL, flag2);


    int a, ret;
    while(1)
    {
        sleep(2);
        int ret = scanf("%d", &a);
        
        if(ret >= 0)
        {
            printf("%d\n", a);
        }
        else
        {
            printf("--------------\n");
        }
    }
    return 0;
}

运行以上程序得到的输出效果为:

ubuntu@ubuntu:~/Desktop$ ./a.out 
--------------
--------------
2  # 这个2为获取键盘输入
2
--------------
--------------
--------------
^C

IO设置为非阻塞状态后,可以通过修改程序使其恢复到阻塞状态:

#include <stdio.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>


int main()
{
    
    int flag1, flag2;
    if(flag1=(fcntl(STDIN_FILENO, F_GETFL, 0)) < 0)
    {
        perror("fcntl");
        return -1;
    }
    flag2 = flag1 | O_NONBLOCK;
    fcntl(STDIN_FILENO, F_SETFL, flag2);


    int a, ret;
    while(1)
    {
        sleep(2);
        int ret = scanf("%d", &a);
        
        if(ret >= 0)
        {
            printf("%d\n", a);
        }
        else
        {
            printf("--------------\n");
            fcntl(STDIN_FILENO, F_SETFL, flag1);
        }
    }
    return 0;
}

此时的效果为:

ubuntu@ubuntu:~/Desktop$ ./a.out 
--------------
  # 此时光标一直停留,直至获取到键盘输入
©️2020 CSDN 皮肤主题: 数字20 设计师:CSDN官方博客 返回首页