C语言写监控守护进程

水深无声 2022-05-30 10:40 177阅读 0赞

转自 [http://blog.csdn.net/heiyeshuwu/article/details/1015544][http_blog.csdn.net_heiyeshuwu_article_details_1015544]

来源：[http://bbs.chinaunix.net/archiver/?tid-393658.html][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;
    }

[http_blog.csdn.net_heiyeshuwu_article_details_1015544]: http://blog.csdn.net/heiyeshuwu/article/details/1015544
[http_bbs.chinaunix.net_archiver_tid-393658.html]: http://bbs.chinaunix.net/archiver/?tid-393658.html