介绍

  在linux下每打开一个终端,系统自动的就打开了三个文件,它们的文件描述符分别为0,1,2,功能分别是“标准输入”、“标准输出”和“标准错误输出”,同时对应了三个文件流指针,分别是stdin,stdout和stderr。三个文件描述符定义了对应的宏,分别为STDIN_FILENO,STDOUT_FILENO和STDERR_FILENO下表为他们的对应关系:

  在程序中通过read从STDIN_FILENO这个文件描述符中读取数据,实际上就是读取终端上键盘输入的数据,当然也可以通过fread从stdin这个文件流指针中读取数据。通过write往STDOUT_FILENO这个文件描述符中写数据,自然就是往终端输出数据了。标准错误输出也是往终端输出,它与标准输出的区别就是标准输出是有缓冲的,而标准错误输出是无缓冲的。

  在程序中可以通过各种办法从终端读取键盘输入的数据,有时候会希望调用的函数不阻塞,这里有三种方法可以实现。

1、通过ioctl清除非阻塞标志

  函数原型:

    int ioctl(int d, int request, ...);

  参数:

    d:文件描述符

    request:功能码。根据填写的功能码选择第三个参数。

  返回:

    成功返回0,失败返回-1。但是也有一部分功能返回的是一个数值。

  ioctl函数是一个万能函数,这里不详细解释。

  传入的第二个参数为FIONBIO表示“设置/清除非阻塞标志”,那么第三个参数要传入一个int类型的指针,指针指向的值为1表示设置非阻塞标志,那么对应的文件描述符为非阻塞,指针指向的值为0表示清除非阻塞标志,那么对应的文件描述符为阻塞,简单的说就是“0就阻塞,1就非阻塞

  因此调用如下代码即可实现非阻塞:

int attr = 1;
ioctl(STDIN_FILENO, FIONBIO, &attr);   /* 清除非阻塞标志 */

  通过该方法设置完成之后,在没有数据的情况下调用read函数将返回-1。

  完整测试代码如下:

 1 #include <stdio.h>
 2 #include <netinet/in.h>
 3 #include <unistd.h>
 4 #include <fcntl.h>
 5 #include <termios.h>
 6 #include <string.h>
 7 #include <sys/ioctl.h>
 8
 9 int main(int argc, const char *argv[])
10 {
11     char buf[128] = { 0 };
12     int len = 0;
13     int total = 0;
14     int attr = 1;
15
16     ioctl(STDIN_FILENO, FIONBIO, &attr);   /* 清除非阻塞标志 */
17
18     while (1) {
19         len = read(STDIN_FILENO, &buf[total], sizeof(buf) - total);
20         //printf("len = %d\n", len);  /* 如果不相信read变成了非阻塞可以去掉本行注释 */
21         if (len > 0) {
22             total += len;
23             if (buf[total - 1] == '\n') {
24                 printf("total = %d\n", total);
25                 printf("buf = %s\n", buf);
26                 total = 0;
27                 memset(buf, 0, sizeof(buf));
28             }
29         }
30     }
31
32     return 0;
33 }

2、通过fcntl设置非阻塞标志

  函数原型:

    int fcntl(int fd, int cmd, ... /* arg */ );

  参数:

    fd:文件描述符。

    cmd:功能码。根据填写的功能码选择第三个参数。

  返回:

    根据不同的功能决定。

  fcntl函数与ioctl函数看上去有很多相似性,实际上这两个函数确实有很多功能是重叠的。

  如果要通过fcntl让文件不阻塞,那么需要知道两个功能码:

  F_GETFL:取得fd的文件状态标志,如同下面的描述一样(arg被忽略)                    

       F_SETFL:设置给arg描述符状态标志,可以更改的几个标志是:O_APPEND, O_NONBLOCK,O_SYNC和O_ASYNC。

  很明显非阻塞标志是O_NONBLOCK,调用下列代码就可以实现非阻塞:

int attr = 0;

attr = fcntl(STDIN_FILENO, F_GETFL);
attr |= O_NONBLOCK;
fcntl(STDIN_FILENO, F_SETFL, attr);

  通过该方法设置完成之后,在没有数据的情况下调用read函数将返回-1。

  完整测试代码如下:

 1 #include <stdio.h>
 2 #include <netinet/in.h>
 3 #include <unistd.h>
 4 #include <fcntl.h>
 5 #include <termios.h>
 6 #include <string.h>
 7 #include <sys/ioctl.h>
 8
 9 int main(int argc, const char *argv[])
10 {
11     char buf[128] = { 0 };
12     int len = 0;
13     int total = 0;
14     int attr = 0;
15
16     /* 设置 O_NONBLOCK属性 */
17     attr = fcntl(STDIN_FILENO, F_GETFL);
18     attr |= O_NONBLOCK;
19     fcntl(STDIN_FILENO, F_SETFL, attr);
20
21     while (1) {
22         len = read(STDIN_FILENO, &buf[total], sizeof(buf) - total);
23         //printf("len = %d\n", len);  /* 如果不相信read变成了非阻塞可以去掉本行注释 */
24         if (len > 0) {
25             total += len;
26             if (buf[total - 1] == '\n') {
27                 printf("total = %d\n", total);
28                 printf("buf = %s\n", buf);
29                 total = 0;
30                 memset(buf, 0, sizeof(buf));
31             }
32         }
33     }
34
35     return 0;
36 }

3、通过设置termios实现

  termios是一个结构体,详细介绍可以参见这篇博客:https://www.cnblogs.com/dartagnan/archive/2013/04/25/3042417.html

  需要用到的函数有tcgetattr和tcsetattr,这两个函数的用法在这篇博客中有介绍:https://www.cnblogs.com/Suzkfly/p/11055532.html

  操作流程就是先用tcgetattr获取文件属性,修改属性之后再用tcsetattr设置进去,关键代码如下:

    tcgetattr(STDIN_FILENO, &attr);
    attr.c_cc[VTIME] = 0;
    attr.c_cc[VMIN] = 0;
    attr.c_lflag &= ~ICANON;        /* 禁用规范输入模式 */
    tcsetattr(STDIN_FILENO, TCSANOW, &attr);

  经过测试,attr.c_lflag &= ~ICANON;这一句一定要写,原因我也不知道,但是写了这句话会带来某些问题,在最后进行分析。

  将attr.c_cc[VTIME]和attr.c_cc[VMIN]的值都设置为0。attr.c_cc[VTIME]和attr.c_cc[VMIN]共同决定了read函数的返回时机。表示读阻塞时间,单位是1/10秒。

  如果经过了c_cc[VTIME]这么长时间,缓冲区内有数据,但是还没达到c_cc[VMIN]个数据,read也会返回。而如果当缓冲区内有了c_cc[VMIN]个数据时,无论等待时间是否到了c_cc[VTIME],read都会返回,但返回值可能比c_cc[VMIN]还大,根据实际数据量而定。如果将c_cc[VMIN]的值设置为0,那么当经过c_cc[VTIME]时间后read也会返回,返回值为0。如果将c_cc[VTIME]和c_cc[VMIN]都设置为0,那么程序运行的效果与设置O_NONBLOCK类似,不同的是如果设置了O_NONBLOCK,那么在没有数据时read返回-1,而如果没有设置O_NONBLOCK,那么在没有数据时read返回的是0。

  完整测试代码如下:

 1 #include <stdio.h>
 2 #include <netinet/in.h>
 3 #include <unistd.h>
 4 #include <fcntl.h>
 5 #include <termios.h>
 6 #include <string.h>
 7 #include <sys/ioctl.h>
 8
 9 int main(int argc, const char *argv[])
10 {
11     int port = 0;
12     char buf[128] = { 0 };
13     int len = 0;
14     int total = 0;
15     struct termios attr;
16
17     tcgetattr(STDIN_FILENO, &attr);
18     attr.c_cc[VTIME] = 0;
19     attr.c_cc[VMIN] = 0;
20     attr.c_lflag &= ~ICANON;        /* 禁用规范输入模式 */
21     tcsetattr(STDIN_FILENO, TCSANOW, &attr);
22
23     while (1) {
24         len = read(STDIN_FILENO, &buf[total], sizeof(buf) - total);
25         //printf("len = %d\n", len);
26         if (len > 0) {
27             total += len;
28             if (buf[total - 1] == '\n') {
29                 printf("total = %d\n", total);
30                 printf("buf = %s\n", buf);
31                 total = 0;
32                 memset(buf, 0, sizeof(buf));
33             }
34         }
35     }
36
37     return 0;
38 }

4、分析总结

  特别要注意的一点是,在测试的时候,如果你不知道你的操作会产生什么结果,在下次测试之前一定要重新开一个终端,因为如果终端没有结束的话,就表示文件没有关闭,即使程序退出了,之前的设置也是会保存的。

  通过这三种方法都实现了设置属性为非阻塞,其实第一和第二种方法是一样的,第三种方法由于禁用了规范输入模式,会带来某些问题。所谓规范输入模式就是对某些特殊的键会产生特殊的效果,比如退格键应该回删一个字符,但通过第三种方法会将退格键也当成一个字符读取进去了。比如,在终端输入“abc”再输入一个退格,然后按回车,通过第一和第二种方法会接收到“ab\n”三个字符,而通过第三种方法则会收到“abc”+“退格”+“\n”5个字符。

  第一和第二种方法虽然read不会阻塞,但是并不是说只要通过键盘输入了数据,它就能读到数据,即使通过键盘输入了一些数据,read函数返回的还是-1,直到按下回车键为止。

01-27 02:29