量化交易之linux高级篇 - 守护进程的写法

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>

#include <sys/stat.h>
#include <fcntl.h>


// 创建守护进程
int ngx_daemon() {

    int fd;

    switch(fork()) {
    case -1:
        // 创建子进程失败, 这里可以写日志...
        return -1;
    case 0:
        // 子进程, 走到这里直接break
        break;
    default:
        // 父进程, 直接退出
        exit(0);
    }

    // 只有子进程流程才能走到这里
    if (setsid() == -1) { // 脱离终端, 终端关闭将与此子进程无关
        // 记录错误日志...
        return -1;
    }
    umask(0); // 设置为0, 不要让它来限制文件权限, 以免引起混乱

    fd = open("/dev/null", O_RDWR); // 打开黑洞设备 (以读写方式打开)
    if (fd == -1) {
        // 记录错误日志...
        return -1;
    }

    if (dup2(fd, STDIN_FILENO) == -1) { // 先关闭 STDIN_FILENO (这是规矩, 已经打开的描述符, 改动之前先关闭), 类似于指针指向null, 让/dev/null成为标准输入
        // 记录错误日志...
        return -1;
    }

    if (dup2(fd, STDOUT_FILENO) == -1) { // 先关闭 STDOUT_FILENO, 类似于指针指向 null, 让 /dev/null 成为标准输出
        // 记录错误日志...
        return -1;
    }

    if (fd > STDERR_FILENO) { // fd 应该是 3, 这里应该成立
        if (close(fd) == -1) { // 释放资源, 这样这个文件描述符就可以被复用; 不然这个数字(文件描述符)会被一直占用
            // 记录错误日志...
            return -1;
        }
    }

    return 1;
}

int main(int argc, char *argv[]) {

    if (ngx_daemon() != 1) {
        // 创建守护进程失败, 可以做失败后的处理(如 写日志等);
        return 1;
    } else {
        // 创建守护进程成功, 执行守护进程中要做的工作

        for (;;) {
            printf("sleep 1s, process id: %d!\n", getpid()); // 即使打印也没用, 现在标准输出指向黑洞(/dev/null), 打印不出任何结果(不显示任何结果)
            sleep(1);
        }
    }

    return 0;
}