[转]C语言写监控守护进程

来源：http://bbs.chinaunix.net/archiver/?tid-393658.html

UNIX Programming FAQ 中文版 v0.1.0（转）

一个使用以上函数的范例程序:

#include <sys/types.h>;
#include <sys/socket.h>;
#include <netinet/in.h>;
#include <stdio.h>;
#include <stdlib.h>;
#include <syslog.h>;
#include <errno.h>;

int daemon(int,int);
int fork2(void);
void closeall(int);

#define TCP_PORT 8888

void errexit(const char *str)
{
syslog(LOG_INFO, "%s failed: %d (%m)", str, errno);
exit(1);
}

void errreport(const char *str)
{
syslog(LOG_INFO, "%s failed: %d (%m)", str, errno);
}

/* 实际的子进程在此. */

void run_child(int sock)
{
FILE *in = fdopen(sock,"r");
FILE *out = fdopen(sock,"w");
int ch;

setvbuf(in, NULL, _IOFBF, 1024);
setvbuf(out, NULL, _IOLBF, 1024);

while ((ch = fgetc(in)) != EOF)
fputc(toupper(ch), out);

fclose(out);
}

/* 这是守护程序的主要工作 -- 侦听连接并生成子进程 */

void process()
{
struct sockaddr_in addr;
int addrlen = sizeof(addr);
int sock = socket(AF_INET, SOCK_STREAM, 0);
int flag = 1;
int rc = setsockopt(sock, SOL_SOCKET, SO_REUSEADDR,
&flag, sizeof(flag));

if (rc < 0)
errexit("setsockopt");

addr.sin_family = AF_INET;
addr.sin_port = htons(TCP_PORT);
addr.sin_addr.s_addr = INADDR_ANY;

rc = bind(sock, (struct sockaddr *) &addr, addrlen);
if (rc < 0)
errexit("bind");

rc = listen(sock, 5);
if (rc < 0)
errexit("listen");

for (;;)
{
rc = accept(sock, (struct sockaddr *) &addr, &addrlen);

if (rc >;= 0)
switch (fork2())
{
case 0:close(sock); run_child(rc); _exit(0);
case -1: errreport("fork2"); close(rc); break;
default: close(rc);
}
}
}

int main()
{
if (daemon(0,0) < 0)
{
perror("daemon");
exit(2);
}

openlog("test", LOG_PID, LOG_DAEMON);

process();

return 0;
}

#include <unistd.h>;
#include <stdlib.h>;
#include <fcntl.h>;
#include <signal.h>;
#include <sys/types.h>;
#include <sys/wait.h>;
#include <errno.h>;

/* closeall() -- 关闭所有>;=给定值的文件描述符 */

void closeall(int fd)
{
int fdlimit = sysconf(_SC_OPEN_MAX);

while (fd < fdlimit)
close(fd++);
}

/* daemon() - 将进程从用户端脱离并消失进入后台，若失败返回-1，
* 但是在那种情况下你只能退出，因为我们可能已经生成了子进程。
* 这是基于BSD的版本,所以调用方需负责类似umask等等其它的工作。
*/

/* 相信在所有Posix系统上都能工作 */

int daemon(int nochdir, int noclose)
{
switch (fork())
{
case 0:break;
case -1: return -1;
default: _exit(0); /* 原进程退出 */
}

if (setsid() < 0) /* 不应该失败 */
return -1;

/* 如果你希望将来获得一个控制tty,则排除(dyke)以下的switch语句 */
/* -- 正常情况不建议用于守护程序 */

switch (fork())
{
case 0:break;
case -1: return -1;
default: _exit(0);
}

if (!nochdir)
chdir("/");

if (!noclose)
{
closeall(0);
open("/dev/null",O_RDWR);
dup(0); dup(0);
}

return 0;
}

/* fork2() -- 类似fork函数，但子进程立刻变成孤儿进程
* (当它退出时不产生僵死进程)
* 返回1给父进程，不是任何有意义的进程号.
* 父进程不能使用wait函数等待子进程结束 (它们是无关的).
*/

/* 这个版本假设你没有捕获和忽略SIGCHLD信号. */
/* 如果你有设定，则不管怎样应使用fork函数 */

int fork2()
{
pid_t pid;
int rc;
int status;

if (!(pid = fork()))
{
switch (fork())
{
case 0:return 0;
case -1: _exit(errno); /* 假设错误码都小于256 */
default: _exit(0);
}
}

if (pid < 0 || waitpid(pid,&status,0) < 0)
return -1;

if (WIFEXITED(status))
if (WEXITSTATUS(status) == 0)
return 1;
else
errno = WEXITSTATUS(status);
else
errno = EINTR;/* 唉，类似这个 :-) */

return -1;
}