我正在尝试用 C 语言(在 Linux 上)编写一个程序,该程序循环直到用户按下一个键,但不应该需要按键来继续每个循环。
有没有一种简单的方法可以做到这一点?我想我可以做到这一点,select()
但这似乎需要做很多工作。
或者,有没有办法在程序关闭而不是非阻塞 io 之前捕获ctrl-按键来进行清理?c
我正在尝试用 C 语言(在 Linux 上)编写一个程序,该程序循环直到用户按下一个键,但不应该需要按键来继续每个循环。
有没有一种简单的方法可以做到这一点?我想我可以做到这一点,select()
但这似乎需要做很多工作。
或者,有没有办法在程序关闭而不是非阻塞 io 之前捕获ctrl-按键来进行清理?c
如前所述,您可以使用sigaction
ctrl-cselect
来捕获或捕获任何标准输入。
但是请注意,使用后一种方法,您还需要设置 TTY,使其处于一次字符而不是一次一行模式。后者是默认设置 - 如果您输入一行文本,它不会被发送到正在运行的程序的标准输入,直到您按下回车键。
您需要使用该tcsetattr()
功能关闭 ICANON 模式,并且可能还禁用 ECHO。从内存中,您还必须在程序退出时将终端设置回 ICANON 模式!
为了完整起见,这里有一些我刚刚敲出的代码(nb:没有错误检查!)它设置了一个 Unix TTY 并模拟了 DOS<conio.h>
函数kbhit()
和getch()
:
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/select.h>
#include <termios.h>
struct termios orig_termios;
void reset_terminal_mode()
{
tcsetattr(0, TCSANOW, &orig_termios);
}
void set_conio_terminal_mode()
{
struct termios new_termios;
/* take two copies - one for now, one for later */
tcgetattr(0, &orig_termios);
memcpy(&new_termios, &orig_termios, sizeof(new_termios));
/* register cleanup handler, and set the new terminal mode */
atexit(reset_terminal_mode);
cfmakeraw(&new_termios);
tcsetattr(0, TCSANOW, &new_termios);
}
int kbhit()
{
struct timeval tv = { 0L, 0L };
fd_set fds;
FD_ZERO(&fds);
FD_SET(0, &fds);
return select(1, &fds, NULL, NULL, &tv) > 0;
}
int getch()
{
int r;
unsigned char c;
if ((r = read(0, &c, sizeof(c))) < 0) {
return r;
} else {
return c;
}
}
int main(int argc, char *argv[])
{
set_conio_terminal_mode();
while (!kbhit()) {
/* do some work */
}
(void)getch(); /* consume the character */
}
select()
为了方便起见,它有点太低级了。我建议你使用ncurses
库将终端置于 cbreak 模式和延迟模式,然后调用,如果没有准备好字符getch()
,它将返回:ERR
WINDOW *w = initscr();
cbreak();
nodelay(w, TRUE);
到那时,您可以在getch
没有阻塞的情况下调用。
在 UNIX 系统上,您可以使用call 为表示 Control+C 键序列sigaction
的信号注册信号处理程序。SIGINT
信号处理程序可以设置一个标志,该标志将在循环中进行检查,使其适当地中断。
你可能想要kbhit();
//Example will loop until a key is pressed
#include <conio.h>
#include <iostream>
using namespace std;
int main()
{
while(1)
{
if(kbhit())
{
break;
}
}
}
这可能不适用于所有环境。一种可移植的方法是创建一个监视线程并设置一些标志getch();
另一种获得非阻塞键盘输入的方法是打开设备文件并读取它!
您必须知道您要查找的设备文件,即 /dev/input/event* 之一。你可以运行 cat /proc/bus/input/devices 来找到你想要的设备。
此代码适用于我(以管理员身份运行)。
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <linux/input.h>
int main(int argc, char** argv)
{
int fd, bytes;
struct input_event data;
const char *pDevice = "/dev/input/event2";
// Open Keyboard
fd = open(pDevice, O_RDONLY | O_NONBLOCK);
if(fd == -1)
{
printf("ERROR Opening %s\n", pDevice);
return -1;
}
while(1)
{
// Read Keyboard Data
bytes = read(fd, &data, sizeof(data));
if(bytes > 0)
{
printf("Keypress value=%x, type=%x, code=%x\n", data.value, data.type, data.code);
}
else
{
// Nothing read
sleep(1);
}
}
return 0;
}
curses 库可用于此目的。当然,select()
在一定程度上也可以使用信号处理程序。
如果你只是高兴地抓住了 Control-C,那就大功告成了。如果你真的想要非阻塞 I/O 但又不想使用 curses 库,另一种选择是将 lock、stock 和 bucket 移至AT&Tsfio
库。它是一个很好的基于 C 模式的库,stdio
但更灵活、线程安全且性能更好。(sfio 代表安全、快速的 I/O。)
没有可移植的方法来做到这一点,但 select() 可能是一个好方法。有关更多可能的解决方案,请参阅http://c-faq.com/osdep/readavail.html。
您可以使用 select 执行此操作,如下所示:
int nfds = 0;
fd_set readfds;
FD_ZERO(&readfds);
FD_SET(0, &readfds); /* set the stdin in the set of file descriptors to be selected */
while(1)
{
/* Do what you want */
int count = select(nfds, &readfds, NULL, NULL, NULL);
if (count > 0) {
if (FD_ISSET(0, &readfds)) {
/* If a character was pressed then we get it and exit */
getchar();
break;
}
}
}
没有太多的工作:D
这是为您执行此操作的功能。您需要termios.h
POSIX 系统自带的。
#include <termios.h>
void stdin_set(int cmd)
{
struct termios t;
tcgetattr(1,&t);
switch (cmd) {
case 1:
t.c_lflag &= ~ICANON;
break;
default:
t.c_lflag |= ICANON;
break;
}
tcsetattr(1,0,&t);
}
分解:tcgetattr
获取当前终端信息并将其存储在t
. 如果cmd
为 1,则将本地输入标志t
设置为非阻塞输入。否则将被重置。然后tcsetattr
将标准输入更改为t
.
如果你没有在程序结束时重置标准输入,你的 shell 就会出现问题。
在 C++ 中,我这样做了:
#include <chrono>
#include <thread>
using namespace std::chrono_literals;
void OnEnter()
{
while (true)
{
getchar();
// do something when enter is pressed
}
}
int main()
{
std::thread keyBoardCommands(OnEnter);
while(true)
{
// code for main loop
std::this_thread::sleep_for(16ms);
}
}
此代码将与平台无关。