家在寫server的時候，不管server寫的是多麼健壯，還是經常出現core dump等程式異常退出的，但是一般情況下需要在無人為干預情況下，能夠自動重新啟動，保證server程式能夠服務使用者。這時就需要一個監控程式來實現能夠讓程式自動重新啟動，現在筆者在寫portmap就遇到了這個問題，通過網上查詢資料，找到了一個相對靠譜的exec+fork解決方法。

使用指令碼實現自動重啟

首先想到的最簡單的使用shell指令碼，大概思路：

ps -ef | grep “$1″ | grep -v “grep” | wc –l 是獲取 $1 （$1 代表程序的名字）的程序數，指令碼根據程序數來決定下一步的操作。通過一個死迴圈，每隔 1 秒檢查一次系統中的指定程式的程序數，這裡也可使用crontab來實現。

這種方法比較土，還是可以基本解決問題，但是有1s的延遲，筆者在應用中未採用這種方法，有關這個shell指令碼，請參看文章後面的附件程式碼。

exec+fork方式

筆者最終採用的exec+fork方式來實現的，具體思想如下：

1，exec函式把當前程序替換為一個新的程序，新程序由path或file引數指定。可以使用exec函式將程式的執行從一個程式切換到另一個程式；

2，fork函式是建立一個新的程序，在程序表中建立一個新的表項，而建立者（即父程序）按原來的流程繼續執行，子程序執行自己的控制流程；

3，wait 當fork啟動一個子程序時，子程序就有了它自己的生命週期並將獨立執行，我們可以在父程序中呼叫wait函式讓父程序等待子程序的結束；

相信介紹到這裡，讀者已經能夠想到解決方法了：1)首先使用fork系統呼叫，建立子程序，2)在子程序中使用exec函式，執行需要自動重啟的程式，3) 在父程序中執行wait等待子程序的結束，然後重新建立一個新的子程序。

使用方法：

#./portmap 需要監控的程式的路徑
#args portmap 需要的引數
$ ./supervisor ./portmap  args.....

程式碼如下：

/**
 *
 * supervisor 
 *
 * author: liyangguang ([email protected])
 *
 * date: 2011-01-21 21:04:01
 *
 * changes
 * 1, execl to execv
 */
 

 
#include <stdio.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <stdlib.h>
 
int
main(int argc, char **argv)
{
    int ret, i, status;
    char *child_argv[100] = {0};
    pid_t pid;
    if (argc < 2) {
 
        fprintf(stderr, "Usage:%s <exe_path> <args...>n", argv[0]);
        return -1;
    }
    for (i = 1; i < argc; ++i) {
        child_argv[i-1] = (char *)malloc(strlen(argv[i])+1);
        strncpy(child_argv[i-1], argv[i], strlen(argv[i]));
        child_argv[i-1][strlen(argv[i])] = '0';
    }
    while(1){
 
        pid = fork(); 
        if (pid == -1) {
            fprintf(stderr, "fork() error.errno:%d error:%sn", errno, strerror(errno));
            break;
        }
        if (pid == 0) {
            ret = execv(child_argv[0], (char **)child_argv);
            //ret = execl(child_argv[0], "portmap", NULL, 0);
            if (ret < 0) {
                fprintf(stderr, "execv ret:%d errno:%d error:%sn", ret, errno, strerror(errno));
                continue;
            }
            exit(0);
        }
 
        if (pid > 0) {
            pid = wait(&status);
 
            fprintf(stdout, "wait return");
        }
 
    }
 
 
    return 0;
}

shell指令碼方式的程式碼如下：

# 函式: CheckProcess
# 功能: 檢查一個程序是否存在
# 引數: $1 --- 要檢查的程序名稱
# 返回: 如果存在返回0, 否則返回1.
#------------------------------------------------------------------------------
CheckProcess()
{
  # 檢查輸入的引數是否有效
  if [ "$1" = "" ];
  then
    return 1
  fi
 
  #$PROCESS_NUM獲取指定程序名的數目，為1返回0，表示正常，不為1返回1，表示有錯誤，需要重新啟動
  PROCESS_NUM=`ps -ef | grep "$1" | grep -v "grep" | wc -l` 
  if [ $PROCESS_NUM -eq 1 ];
  then
    return 0
  else
    return 1
  fi
}
 
 
# 檢查test例項是否已經存在
while [ 1 ] ; do
 CheckProcess "test"
 CheckQQ_RET=$?
 if [ $CheckQQ_RET -eq 1 ];
 then
 
# 殺死所有test程序，可換任意你需要執行的操作
 
 
  killall -9 test
  exec ./test &  
 fi
 sleep 1
done