您的位置:首页 > 运维架构 > Linux

Linux下C/C++程序处理Ctrl+C的例子

2014-11-17 01:07 302 查看
遇到一个场景:程序要读入文件,而用户在使用文件时是用/dev/urandom重定向作为输入的。(好吧这个用户就是测试我们作业的助教……)而/dev/urandom显然是没有EOF的,只有用Ctrl+C的方法结束输入。但是Ctrl+C也会同时结束程序,造成不用刻预测的结果。

因此,程序必须设法获知Ctrl+C的发生并进行处理。在Linux下,按下Ctrl+C后,程序会收到一个信号SIGINT;这时操作系统默认会结束程序。如果程序不想被默认处理,就要设置对SIGINT信号的处理函数(Handler)。下面是一个简单的例子:

#include <stdio.h>
#include <stdlib.h>
#include <signal.h> // signal functions
volatile sig_atomic_t flag = 0;
static void my_handler(int sig){ // can be called asynchronously
flag = 1; // set flag
}

int main(){
// Register signals
signal(SIGINT, my_handler);
//      ^          ^
//  which-signal   |-- which user defined function registered
while(1)
if(flag){ // my action when signal set it 1
printf("\n Signal caught!\n");
printf("\n default action it not termination!\n");
flag = 0;
}
return 0;
}
(来源:http://stackoverflow.com/questions/17766550/ctrl-c-interrupt-event-handling-in-linux)

如上面程序所示,一般是在一个循环中检测一个标志变量flag,而这个flag只在信号处理函数中改变。在这里,my_handler()函数会在Ctrl+C按下后被调用,并设置flag。注意,由于我们自己设置了Ctrl+C的处理函数,系统就不会再按默认行为结束程序了。因此flag的变化会在循环中被检测到,并作出响应。

需要注意的是,网上很多例子中出于演示的方便,在消息处理函数中使用printf()。这样是不安全的:设想程序在正常调用printf的过程中收到了Ctrl+C,那么信号处理函数就会再调用一次printf,这有有造成错误的可能。有一些函数经过专门的设计可以安全地在信号处理函数中使用。

具体的解释可以参考:Matthew N, Stones R. Beginning Linux Programming[M]. John Wiley & Sons, 2011. 中的 Chapter 11: Processes and Signals。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: