UNIX环境高级编程——单实例的守护进程

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <signal.h>
#include <unistd.h>
#include <sys/types.h>
#include <time.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <syslog.h>
/* 创建守护进程函数 */
int daemonize(){
	int childpid, fd, fdtablesize;
	int error, in, out;
	/* 忽略终端I/O信号，STOP信号 */
	signal(SIGTTOU, SIG_IGN); /* 后台进程写控制终端 */
	signal(SIGTTIN, SIG_IGN); /* 后台进程读控制终端 */
	signal(SIGTSTP, SIG_IGN); /* 终端挂起 */
	signal(SIGHUP, SIG_IGN); /* 进程组长退出时向所有会议成员发出的 */
	/* 父进程退出，子进程成为孤儿进程 */
	if (fork()!=0)
	    exit(1);
	/* 设置新会话的领头进程，并与原来的登录会话和进程组脱离 */
	if (setsid() == -1)
    	exit(1);
	/* 防止会话组长重新申请控制终端,子进程退出，孙进程没有控制终端了 */
	if (fork()!=0)
	    exit(1);
	/* 关闭打开的文件描述符，防止资源浪费和防止引起无法预料的错误 */
	for (fd = 3, fdtablesize = getdtablesize();fd < fdtablesize;fd++){
		close(fd);
	}
	/* 重定向标准输入／标准输出和标准错误输出 */
	error = open("./stderr", O_WRONLY|O_CREAT|O_APPEND, 0600);
	dup2(error, 2);
	close(error);
	in = open("./stdin", O_RDONLY|O_CREAT,0600);
	dup2(in, 0);
	close(in);
	out = open("./stdout", O_WRONLY|O_CREAT|O_APPEND,0600);
	dup2(out, 1);
	close(out);
	/* 可以改变工作目录 */
	/* chdir("/"); */
	/* 重设文件创建掩模 */
	umask(0);
	/* 忽略SIGCHLD信号, 防止僵尸进程 */
	signal(SIGCHLD, SIG_IGN);
	return 0;
}
/* 单实例实现, 成功返回0, 失败程序退出, 打印错误或已运行进程号 */
int already_running(const char *szLockFilePath){
	int fd;
	char szBuf[256] = { 0x00 };

	fd = open(szLockFilePath, O_RDWR|O_CREAT, S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH);
	if (fd < 0){
		memset(szBuf, 0x00, sizeof(szBuf));
		sprintf(szBuf, "open LockFile failed!LockfilePath=[%s]/n", szLockFilePath);
		perror(szBuf);
		exit(1);
	}
	/* 尝试获取文件锁 */
	struct  flock fl;
	memset(&fl, 0x00, sizeof(struct flock));
	if (fcntl(fd, F_GETLK, &fl) < 0){
		perror("fcntl get LockState failed!/n");
		close(fd);
		exit(1);
	}
	if (fl.l_type != F_UNLCK){
		memset(szBuf, 0x00, sizeof(szBuf));
		sprintf(szBuf, "已经有一个运行实例了。进程号为[%ld]/n", (long)fl.l_pid);
		perror(szBuf);
		close(fd);
		exit(1);
	}
	fl.l_whence = SEEK_SET;/* 决定l_start的位置 */
	fl.l_start =  0; /* 设置锁定区域开头位置 */
	fl.l_len = 0; /* 设置锁定区域长度 */
	fl.l_type = F_WRLCK;/* 设置锁定状态为写锁 */
	fl.l_pid = getpid(); /* 设置锁进程号 */
	/* 阻塞式加锁 */
	if (fcntl(fd, F_SETLK, &fl) < 0){
		perror("fcntl set Lockfile failed!/n");
		close(fd);
		exit(1);
	}
	/* 把进程号写入文件 */
	ftruncate(fd, 0);
	memset(szBuf, 0x00, sizeof(szBuf));
	sprintf(szBuf, "%ld", (long)getpid());
	write(fd, szBuf, strlen(szBuf)+1);
	return 0;
}
int main(void){
	time_t now;
	daemonize();
	already_running("./MyDaemon.pid");
	openlog("MyMsgDaemon", LOG_CONS | LOG_PID, 0);
	syslog(LOG_DEBUG, "守护进程测试/n");
	while (1){
		time(&now);
		syslog(LOG_DEBUG, "守护进程测试, 当前时间：[%s]/n", ctime(&now));
		sleep(6);
	}
	return 0;
}