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

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

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

#include 
#include 
#include  // 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。


你可能感兴趣的:(编程杂记)