如果在关机前保存数据？

zhoulifa

有了上面这个例子，我们就可以完成一个关机前保存数据的程序了。

首先，C程序代码如下：

1. 
   
2. #include <sys/types.h>
   
3. #include <time.h>
   
4. #include <signal.h>
   
5. #include <stdio.h>
   
6. #include <stdlib.h>
   
7. #include <unistd.h>
   
8. 
   
9. #define SIGTOCAP SIGTERM
   
10. 
    
11. void sighandler(int);
    
12. char status;
    
13. time_t t;
    
14. 
    
15. /************关于本文档********************************************
    
16. *filename: SaveData.c
    
17. *purpose: 展示如何在关机前保存数据
    
18. *wrote by: zhoulifa(zhoulifa@163.com) 周立发(http://zhoulifa.bokee.com)
    
19. Linux爱好者 Linux知识传播者 SOHO族 开发者 最擅长C语言
    
20. *date time:2006-09-02 22:11:54
    
21. *Note: 任何人可以任意复制代码并运用这些文档，当然包括你的商业用途
    
22. * 但请遵循GPL
    
23. *Hope:希望越来越多的人贡献自己的力量，为科学技术发展出力
    
24. *         科技站在巨人的肩膀上进步更快！感谢有开源前辈的贡献！
    
25. *********************************************************************/
    
26. int main(int argc, char ** argv)
    
27. {
    
28.   FILE * fp = 0;
    
29. 
    
30.   if(fork()) exit(0);
    
31.   if(fork()) exit(0); /* 变成 daemon 精灵程序进行后台服务 */
    
32. 
    
33.   signal(SIGTOCAP, sighandler); /* 捕捉退出信号 */
    
34.   if(!(fp = fopen("/var/run/save-data.pid", "w"))) {
    
35.     fprintf(stdout, "cannot open file '/var/run/save-data.pid'\n");
    
36.     exit(1);
    
37.   }
    
38.   fprintf(fp, "%d\n", getpid()); /* 保存运行程序的 pid */
    
39.   fclose(fp);
    
40. 
    
41.   if(!(fp = fopen("/home/test/running.data", "a+"))) {
    
42.     fprintf(stdout, "cannot open file '/home/test/running.data'\n");
    
43.     exit(1);
    
44.   }
    
45.   status = 0;
    
46.   while(1) {
    
47.     t = time(0);
    
48.     fprintf(fp, "now running at %s\n", ctime(&t)); /* 每隔一定时间写一些数据，表明程序在运行 */
    
49.     if(status) break; /* 如果收到退出信号，并保存完数据， 则退出程序运行 */
    
50.     usleep(1000000);
    
51.   }
    
52.   fclose(fp);
    
53.   remove("/var/run/save-data.pid"); /* 退出前删除 pid 文件 */
    
54.   return 0;
    
55. }
    
56. 
    
57. void sighandler(int sig)
    
58. {
    
59.   FILE * fp = 0;
    
60.   int i = 0;
    
61.   if(sig == SIGTOCAP) { /* 捕捉到退出信号后开始保存数据 */
    
62.     if(!(fp = fopen("/home/test/saving.data", "a+"))) {
    
63.       fprintf(stdout, "cannot open file '/home/test/saving.data'\n");
    
64.       exit(-1);
    
65.     }
    
66.     for(i = 0; i < 10000; i++) {
    
67.       t = time(0);
    
68.       fprintf(fp, "now get ready to exit at %s\n", ctime(&t));
    
69.     }
    
70.     fclose(fp);
    
71.     status= 1; /* 数据保存完毕，设置退出标志 */
    
72.   }
    
73. }
    

复制代码

然后编译此程序：

gcc SaveData.c -o save-data

再把这个程序放到/bin目录下去：

mv save-data /bin

然后编写一个脚本/etc/init.d/save-data，内容如下：

1. 
   
2. #!/bin/sh
   
3. #
   
4. # $Id: save-data.init.d,v 1.0.0.0 2006/09/02 22:11:54
   
5. #
   
6. 
   
7. test -x /bin/save-data || exit 0
   
8. 
   
9. MYPID=/var/run/save-data.pid
   
10. 
    
11. case "$1" in
    
12.         start)
    
13.                 echo -n "Starting save-data daemon: "
    
14.                 echo -n ""
    
15.                 sleep 10 # Just for test , you'd better delete this line for use
    
16.                 start-stop-daemon --start --quiet --pidfile $MYPID --exec /bin/save-data
    
17.                 if [ -f "$MYPID" ] && ps h `cat "$MYPID"` >/dev/null; then
    
18.                         echo " OK."
    
19.                 else
    
20.                         echo " Failed."
    
21.                 fi
    
22.                 ;;
    
23.         stop)
    
24.                 echo -n "Stopping save-data daemon: save-data"
    
25.                 echo -n ""
    
26.                 sleep 10 # Just for test , you'd better delete this line for use
    
27.                 start-stop-daemon --stop --quiet --pidfile $MYPID
    
28.                 echo "."
    
29.                 ;;
    
30.         restart | force-reload)
    
31.                 $0 stop
    
32.                 sleep 2
    
33.                 $0 start
    
34.                 ;;
    
35.         *)
    
36.                 echo "Usage: /etc/init.d/save-data {start|stop|restart}"
    
37.                 exit 1
    
38. esac
    
39. 
    
40. exit 0
    

复制代码

再创建链接：

ln -s /etc/init.d/save-data /etc/rc2.d/S70save-data
ln -s /etc/init.d/save-data /etc/rc0.d/K10save-data

然后重启机器试试，会看到数据被保存到saving.data文件里了

zhoulifa

关于“ daemon 精灵程序 ” ， 我这里只是简单的 fork 了两次，事实上还有其它许多工作要做，比如关闭终端， google 一下很多的，就不赘述了