[转]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;
     }