UNIX環境高級編程學習之第十三章守護進程 - 單實例的守護進程

UNIX環境高級編程學習之第十三章守護進程 - 單實例的守護進程

/* Data: 20100808 * User: lixiujie * Desc: 單實例的守護進程 * gcc daemon.c -o MyDaemon */ #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("/tmp"); */ /* 重設文件創建掩模 */ 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; }