C++ - 寫一個守護進程A，用於監控進程B，B掛掉就拉起

概述：

1、A可以是守護進程也可以是守護腳本

2、A是守護進程

3、A具有保活其他進程的服務

 

守護進程（已驗證）

//main.c

#include <stdio.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <syslog.h>

//守護應用名
#define APP_NAME "untitled_ATU201console"
//生成日志文件
#define LOG_FILE "/home/root/ljc/folder_daemon/untitled_ATU201console.log"
//匹配成功
#define MATCH_RES "1"
//啟動守護應用的腳本
#define SCRIPT_FILE "start_daemon"



int daemon(int nochdir, int noclose) {
    pid_t pid;

    //讓init進程成為新產生進程的父進程
    pid = fork();
    //如果創建進程失敗
    if (pid < 0) {
        perror("fork");
        return -1;
    }
    //父進程退出運行
    if (pid != 0) {
        exit(0);
    }
    //創建新的會話
    pid = setsid();
    if (pid < -1) {
        perror("set sid");
        return -1;
    }
    //更改當前工作目錄,將工作目錄修改成根目錄
    if (!nochdir) {
        chdir("/");
    }
    //關閉文件描述符，並重定向標准輸入，輸出合錯誤輸出
    //將標准輸入輸出重定向到空設備
    if (!noclose) {
        int fd;
        fd = open("/dev/null", O_RDWR, 0);
        if (fd != -1) {
            dup2(fd, STDIN_FILENO);
            dup2(fd, STDOUT_FILENO);
            dup2(fd, STDERR_FILENO);
            if (fd > 2) {
                close(fd);
            }
        }
    }
    //設置守護進程的文件權限創建掩碼
    umask(0027);

    return 0;
}

//是否有匹配的字符，有則返回1，沒有返回0
//src:源字符串
//dst：目標字符串
//len:源字符串被比較的長度
int match(char *src, char *dst, int len) {
    syslog(LOG_INFO, "--------------------match1");

    int i = 0;
    int j = 0;
    int size_dst = 0;

    //獲得目標字符串的長度
    size_dst = strlen(dst);
    //如果目標字符串的長度大於len，返回失敗
    if (size_dst > len) {
        syslog(LOG_INFO, "--------------------match11");
        return 0;
    }

    syslog(LOG_INFO, "--------------------match2");

    //開始比較
    for (i = 0; i < len; i++) {
        for (j = 0; j < size_dst; j++) {
            if (src[i + j] != dst[j]) {
                break;
            }
        }
        if (j == size_dst) {
            return 1;
        }
    }

    syslog(LOG_INFO, "--------------------match3");

    return 0;
}

int main(int argc, char *argv[]) {
    int fd = 0;
    char buf[100];

    //開啟守護進程
    daemon(0, 0);

    while (1) {
        syslog(LOG_INFO, "--------------------main1");

        //打開日志
        openlog(argv[0], LOG_CONS | LOG_PID, LOG_USER);

        //查看程序是否運行
        //新建輸出文件
        system("touch "LOG_FILE);
        //獲得程序ID
        system("ps -ef | grep "APP_NAME" | grep -v grep | wc -l >> "LOG_FILE);
        //打開輸出文件
        fd = open(LOG_FILE, O_CREAT | O_RDONLY, 0777);
        //清空緩存
        memset(buf, 0, 100);
        //讀取全部
        read(fd, buf, 100);

        syslog(LOG_INFO, "--------------------main2");

        //判斷是否有程序文件運行
//        if (match(buf, MATCH_RES, 1)) {
//            syslog(LOG_INFO, "match success!!!");
//        } else {
//            syslog(LOG_INFO, "match fail!!!");
//            //運行程序
//            system(SCRIPT_FILE);
//        }

        int res = memcmp(buf, MATCH_RES, 1);
        syslog(LOG_INFO, "buf = %s\n", buf);
        syslog(LOG_INFO, "MATCH_RES = %s\n", MATCH_RES);
        syslog(LOG_INFO, "res = %d\n", res);
        if ( res == 0) {
            syslog(LOG_INFO, "match success!!!");
        } else {
            syslog(LOG_INFO, "match fail!!!");
            //運行程序
//            system(SCRIPT_FILE);
            system("/home/root/ljc/folder_daemon/untitled_ATU201console &");
        }

        syslog(LOG_INFO, "--------------------main3");

        //休眠
        sleep(2);
        //刪除輸出文件
        system("rm "LOG_FILE);
        //休眠
        sleep(5);
    }
    //關閉日志
    closelog();

    return 0;
}

 

守護腳本1（待驗證）

//假設要監控的進程名稱為test,TargetProcessName為啟動該進程的腳本，status 有一系列的宏 對應可以檢測。 
while [ 0 ]
do
    Result=$(ps -ef | grep -v grep |grep "test")
    if [ "" == "$Result" ]
    then
RecordTime=$(date)
echo  "Restar test" >>$LogFile
        nohup $TargetProcessName>>$LogFile &
sleep 2
    fi
    Result=
done

 

守護腳本2（待驗證）

#!/bin/bash
# 直接上代碼吧
# watchdog.sh

#定義映射表 k=進程號 v=作業
declare -A mapper
#進程號文件
conf=dog.pid


function debug
{
    echo $@
}

#param pid 進程號 
#使用ps命令查詢該進程是否存在，如果不存在返回"gone"，否則返回"stay"

function watch
{
    local pid=$1
    local index=`ps -ef|awk '{print $2}'|grep -P "^${pid}$"`
    if [ "${index}None" = "None" ]; then
        echo gone
        return
    fi
    echo stay
}

#每過5秒鍾檢查一遍所有的進程，調用上面的watch

function dogit
{
    while [ 1 ]
    do
        sleep 5
        for pid in ${!mapper[@]}
        do
            debug pid:$pid
            local t=`watch $pid`
            debug "test result is $t!!!"
            if [ "$t" = "gone" ]; then
                debug "$c with pid $pid was gone"
                loadscript ${mapper[$pid]}
                unset mapper[$pid]
                sed -i "/$pid/d" $conf
            fi
        done
    done
}

#從作業文件加載需要守護的作業命令

function loadscript
{
    local script=$@
    debug script=$script
    $script > /dev/null &
    local pid=$!
    mapper[$pid]=$script
    echo $pid >> $conf
}

function clean
{
    if [ -f $conf ]; then
        while read line; do
            if [ -n "$line" ]; then
                pid=`ps -ef|awk '{print $2}'|grep $line|grep -v 'grep'`
                if [ -n "${pid}" ]; then
                    debug killing $pid
                    kill $pid
                fi
            fi
        done < $conf
    fi
    echo > $conf
    debug done!
}

function main
{
    clean
    local file=$1
    if [ -f $file ]; then
        while read line
        do
            loadscript $line
        done < $file
        dogit
    else
        echo "Not a file!"
    fi
}

main $1;