一个简单的程序timeout控制器

key

很简单，不过比较实用，用来妨止程序失去控制。以分钟为单位
涉及问题：
1. linux下c++/c编程
2. 多进程控制
3. 简单信号处理

1. #include <iostream>
   
2. 
   
3. #include <stdlib.h>
   
4. #include <unistd.h>
   
5. #include <stdint.h>
   
6. 
   
7. #include <sys/types.h>
   
8. #include <sys/time.h>
   
9. #include <time.h>
   
10. 
    
11. #include <string.h>
    
12. #include <errno.h>
    
13. #include <signal.h>
    
14. 
    
15. using std::cout;
    
16. using std::cerr;
    
17. using std::endl;
    
18. 
    
19. static void usage(const char * prog)
    
20. {
    
21.   cerr << "Usage : " << prog << " <minutes> <prog> <arg1> <arg2...>" << endl;
    
22. }
    
23. 
    
24. static void child_return(int sig)
    
25. {
    
26.   cerr << "Child return, parent exits." << endl;
    
27.   exit(0);
    
28. }
    
29. 
    
30. int main(int argc, char ** argv)
    
31. {
    
32.   const char * prog = argv[0];
    
33. 
    
34.   if(argc<3)
    
35.   {
    
36.     usage(prog);
    
37.     exit(-1);
    
38.   }
    
39. 
    
40.   const char * min_str = argv[1];
    
41.   int timeout_min = atoi(min_str);
    
42. 
    
43.   if(min_str <= 0)
    
44.   {
    
45.     cerr << "Error : minutes should be positive." << endl << endl;
    
46.     usage(prog);
    
47.     exit(-2);
    
48.   }
    
49. 
    
50.   time_t begin_time = time(NULL);
    
51.   pid_t child_pid = -1;
    
52. 
    
53.   child_pid = fork();
    
54. 
    
55.   if(child_pid == 0) //child
    
56.   {
    
57.     char * child_args[argc - 2 + 1];
    
58.     int k;
    
59. 
    
60.     bzero(child_args, sizeof(child_args));
    
61. 
    
62.     for(k=0; k < argc - 2 + 1; ++k)
    
63.     {
    
64.       child_args[k] = argv[k+2];
    
65.       cout << "DEBUG " << child_args[k] << endl;
    
66.     }
    
67. 
    
68.     execvp(child_args[0], child_args);
    
69. 
    
70.     exit(-1);
    
71.   }
    
72.   else if(child_pid < 0) //fork fails
    
73.   {
    
74.     cerr << "fork() error: " << strerror(errno) << endl;
    
75.     exit(-3);
    
76.   }
    
77. 
    
78.   //parent
    
79.   
    
80.   signal(SIGCHLD, child_return);
    
81. 
    
82.   while(1)
    
83.   {
    
84.     time_t cur = time(NULL);
    
85.     if(cur - begin_time < (timeout_min * 60))
    
86.     {
    
87.       cout << "DEBUG p " << timeout_min * 60 - cur + begin_time << endl;
    
88.       sleep(timeout_min * 60 - cur + begin_time);
    
89.     }
    
90.     else
    
91.       break;
    
92.   }
    
93. 
    
94.   kill(child_pid, SIGKILL);
    
95. 
    
96.   cerr << "Kill child process." << endl;
    
97. 
    
98.   return 0;
    
99. }

复制代码

ubuntuhk

哈哈，就是单片机的看门狗机制。