Linux下非阻塞地检测键盘输入的方法 (整理)

prc

下面的代码实现了以下特性：
1. getchar获得输入字符不需要按回车
2. 非阻塞得检测键盘输入
3. 输入的字符不回显

1. 
   
2. #include <stdio.h>
   
3. #include <string.h>
   
4. #include <sys/time.h>
   
5. #include <sys/types.h>
   
6. #include <unistd.h>
   
7. #include <termios.h>
   
8. #include <unistd.h>
   
9. 
   
10. static struct termios ori_attr, cur_attr;
    
11. 
    
12. static __inline
    
13. int tty_reset(void)
    
14. {
    
15.         if (tcsetattr(STDIN_FILENO, TCSANOW, &ori_attr) != 0)
    
16.                 return -1;
    
17. 
    
18.         return 0;
    
19. }
    
20. 
    
21. 
    
22. static __inline
    
23. int tty_set(void)
    
24. {
    
25.        
    
26.         if ( tcgetattr(STDIN_FILENO, &ori_attr) )
    
27.                 return -1;
    
28.        
    
29.         memcpy(&cur_attr, &ori_attr, sizeof(cur_attr) );
    
30.         cur_attr.c_lflag &= ~ICANON;
    
31. //        cur_attr.c_lflag |= ECHO;
    
32.         cur_attr.c_lflag &= ~ECHO;
    
33.         cur_attr.c_cc[VMIN] = 1;
    
34.         cur_attr.c_cc[VTIME] = 0;
    
35. 
    
36.         if (tcsetattr(STDIN_FILENO, TCSANOW, &cur_attr) != 0)
    
37.                 return -1;
    
38. 
    
39.         return 0;
    
40. }
    
41. 
    
42. static __inline
    
43. int kbhit(void)
    
44. {
    
45.                   
    
46.         fd_set rfds;
    
47.         struct timeval tv;
    
48.         int retval;
    
49. 
    
50.         /* Watch stdin (fd 0) to see when it has input. */
    
51.         FD_ZERO(&rfds);
    
52.         FD_SET(0, &rfds);
    
53.         /* Wait up to five seconds. */
    
54.         tv.tv_sec  = 0;
    
55.         tv.tv_usec = 0;
    
56. 
    
57.         retval = select(1, &rfds, NULL, NULL, &tv);
    
58.         /* Don't rely on the value of tv now! */
    
59. 
    
60.         if (retval == -1) {
    
61.                 perror("select()");
    
62.                 return 0;
    
63.         } else if (retval)
    
64.                 return 1;
    
65.         /* FD_ISSET(0, &rfds) will be true. */
    
66.         else
    
67.                 return 0;
    
68.         return 0;
    
69. }
    
70. 
    
71. 
    
72. int main()
    
73. {
    
74.         int tty_set_flag;
    
75.         tty_set_flag = tty_set();
    
76.         while(1) {
    
77. 
    
78.                 if( kbhit() ) {
    
79.                         const int key = getchar();
    
80.                         printf("%c pressed\n", key);
    
81.                         if(key == 'q')
    
82.                                 break;
    
83.                 } else {
    
84.                         fprintf(stderr, "<no key detected>\n");
    
85.                 }
    
86.         }
    
87. 
    
88.         if(tty_set_flag == 0)
    
89.                 tty_reset();
    
90.         return 0;
    
91. }
    
92. 
    
93. 
    
94. 
    
95. 
    

复制代码