C语言写监控守护进程

转自 http://blog.csdn.net/heiyeshuwu/article/details/1015544

来源：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;
}