123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151 |
- /*
- * ngIRCd -- The Next Generation IRC Daemon
- * Copyright (c)2001-2010 Alexander Barton (alex@barton.de)
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- * Please read the file COPYING, README and AUTHORS for more information.
- */
- #include "portab.h"
- /**
- * @file
- * Process management
- */
- #include "imp.h"
- #include <assert.h>
- #include <errno.h>
- #include <signal.h>
- #include <string.h>
- #include <stdlib.h>
- #include <unistd.h>
- #include "log.h"
- #include "io.h"
- #include "conn.h"
- #include "exp.h"
- #include "sighandlers.h"
- #include "proc.h"
- /**
- * Initialize process structure.
- */
- GLOBAL void
- Proc_InitStruct (PROC_STAT *proc)
- {
- assert(proc != NULL);
- proc->pid = 0;
- proc->pipe_fd = -1;
- }
- /**
- * Fork a child process.
- */
- GLOBAL pid_t
- Proc_Fork(PROC_STAT *proc, int *pipefds, void (*cbfunc)(int, short), int timeout)
- {
- pid_t pid;
- unsigned int seed;
- assert(proc != NULL);
- assert(pipefds != NULL);
- assert(cbfunc != NULL);
- if (pipe(pipefds) != 0) {
- Log(LOG_ALERT, "Can't create output pipe for child process: %s!",
- strerror(errno));
- return -1;
- }
- seed = (unsigned int)rand();
- pid = fork();
- switch (pid) {
- case -1:
- /* Error on fork: */
- Log(LOG_CRIT, "Can't fork child process: %s!", strerror(errno));
- close(pipefds[0]);
- close(pipefds[1]);
- return -1;
- case 0:
- /* New child process: */
- srand(seed ^ (unsigned int)time(NULL) ^ getpid());
- Signals_Exit();
- signal(SIGTERM, Proc_GenericSignalHandler);
- signal(SIGALRM, Proc_GenericSignalHandler);
- close(pipefds[0]);
- alarm(timeout);
- Conn_CloseAllSockets();
- return 0;
- }
- /* Old parent process: */
- close(pipefds[1]);
- if (!io_setnonblock(pipefds[0])
- || !io_event_create(pipefds[0], IO_WANTREAD, cbfunc)) {
- Log(LOG_CRIT, "Can't register callback for child process: %s!",
- strerror(errno));
- close(pipefds[0]);
- return -1;
- }
- proc->pid = pid;
- proc->pipe_fd = pipefds[0];
- return pid;
- }
- /**
- * Generic signal handler for forked child processes.
- */
- GLOBAL void
- Proc_GenericSignalHandler(int Signal)
- {
- switch(Signal) {
- case SIGTERM:
- #ifdef DEBUG
- Log_Subprocess(LOG_DEBUG, "Child got TERM signal, exiting.");
- #endif
- exit(1);
- case SIGALRM:
- #ifdef DEBUG
- Log_Subprocess(LOG_DEBUG, "Child got ALARM signal, exiting.");
- #endif
- exit(1);
- }
- }
- /**
- * Read bytes from a pipe of a forked child process.
- * In addition, this function makes sure that the child process is ignored
- * after all data has been read or a fatal error occurred.
- */
- GLOBAL size_t
- Proc_Read(PROC_STAT *proc, void *buffer, size_t buflen)
- {
- ssize_t bytes_read = 0;
- assert(buffer != NULL);
- assert(buflen > 0);
- bytes_read = read(proc->pipe_fd, buffer, buflen);
- if (bytes_read < 0) {
- if (errno == EAGAIN)
- return 0;
- Log(LOG_CRIT, "Can't read from child process %ld: %s",
- proc->pid, strerror(errno));
- bytes_read = 0;
- }
- #if DEBUG
- else if (bytes_read == 0)
- LogDebug("Can't read from child process %ld: EOF", proc->pid);
- #endif
- Proc_InitStruct(proc);
- return (size_t)bytes_read;
- }
- /* -eof- */
|