mux/src/slave.cpp File Reference

#include "autoconf.h"
#include "config.h"
#include <netdb.h>
#include <netinet/in.h>
#include <sys/wait.h>
#include <sys/file.h>
#include <sys/ioctl.h>
#include <signal.h>
#include "slave.h"
#include <arpa/inet.h>

Include dependency graph for slave.cpp:

Go to the source code of this file.

Defines
#define	CAST_SIGNAL_FUNC
#define	MAX_STRING   1000
#define	INADDR_NONE   ((in_addr_t)-1)
#define	MAX_CHILDREN   20
Functions
char *	format_inet_addr (char *dest, long addr)
char *	stpcpy (char *dest, const char *src)
RETSIGTYPE	child_timeout_signal (int iSig)
int	query (char *ip, char *orig_arg)
RETSIGTYPE	alarm_signal (int iSig)
RETSIGTYPE	child_signal (int iSig)
int	main (int argc, char *argv[])
Variables
pid_t	parent_pid
char *	arg_for_errors
volatile int	nChildrenStarted = 0
volatile int	nChildrenEndedSIGCHLD = 0
volatile int	nChildrenEndedMain = 0

---

Define Documentation

#define CAST_SIGNAL_FUNC

Definition at line 26 of file slave.cpp.

#define INADDR_NONE   ((in_addr_t)-1)

Definition at line 35 of file slave.cpp.

Referenced by query().

#define MAX_CHILDREN   20

Definition at line 223 of file slave.cpp.

Referenced by main().

#define MAX_STRING   1000

Definition at line 31 of file slave.cpp.

Referenced by main(), and query().

---

Function Documentation

RETSIGTYPE alarm_signal

(

int

iSig

)

Definition at line 206 of file slave.cpp.

References CAST_SIGNAL_FUNC, and parent_pid.

Referenced by main().

{
    struct itimerval itime;
    struct timeval interval;

    if (getppid() != parent_pid)
    {
        exit(1);
    }
    signal(SIGALRM, CAST_SIGNAL_FUNC alarm_signal);
    interval.tv_sec = 120;  // 2 minutes.
    interval.tv_usec = 0;
    itime.it_interval = interval;
    itime.it_value = interval;
    setitimer(ITIMER_REAL, &itime, 0);
}

RETSIGTYPE child_signal

(

int

iSig

)

Definition at line 228 of file slave.cpp.

References CAST_SIGNAL_FUNC, nChildrenEndedMain, nChildrenEndedSIGCHLD, and nChildrenStarted.

Referenced by main().

{
    // Collect the children.
    //
    while (waitpid(0, NULL, WNOHANG) > 0)
    {
        int nChildren = nChildrenStarted - nChildrenEndedSIGCHLD
            - nChildrenEndedMain;
        if (0 < nChildren)
        {
            nChildrenEndedSIGCHLD++;
        }
    }

    signal(SIGCHLD, CAST_SIGNAL_FUNC child_signal);
}

RETSIGTYPE child_timeout_signal

(

int

iSig

)

Definition at line 61 of file slave.cpp.

Referenced by main().

{
    exit(1);
}

char* format_inet_addr	(	char *	dest,
		long	addr
	)

Definition at line 38 of file slave.cpp.

Referenced by query().

{
    sprintf(dest, "%ld.%ld.%ld.%ld",
        (addr & 0xFF000000) >> 24,
        (addr & 0x00FF0000) >> 16,
        (addr & 0x0000FF00) >> 8,
        (addr & 0x000000FF));
    return (dest + strlen(dest));
}

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 245 of file slave.cpp.

References alarm_signal(), CAST_SIGNAL_FUNC, child_signal(), child_timeout_signal(), MAX_CHILDREN, MAX_STRING, nChildrenEndedMain, nChildrenEndedSIGCHLD, nChildrenStarted, parent_pid, and query().

{
    char arg[MAX_STRING];
    char *p;
    int len;
    pid_t child;

    parent_pid = getppid();
    if (parent_pid == 1)
    {
        // Our real parent process is gone, and we have been inherited by the
        // init process.
        //
        exit(1);
    }

    alarm_signal(SIGALRM);
    signal(SIGCHLD, CAST_SIGNAL_FUNC child_signal);
    signal(SIGPIPE, SIG_DFL);

    for (;;)
    {
        len = read(0, arg, MAX_STRING - 1);
        if (len == 0)
        {
            break;
        }
        if (len < 0)
        {
            if (errno == EINTR)
            {
                errno = 0;
                continue;
            }
            break;
        }
        arg[len] = '\0';
        p = strchr(arg, '\n');
        if (p)
        {
            *p = '\0';
        }
        child = fork();
        switch (child)
        {
        case -1:
            exit(1);
            break;

        case 0: // child.
            {
                // We don't want to try this for more than 5 minutes.
                //
                struct itimerval itime;
                struct timeval interval;

                interval.tv_sec = 300;  // 5 minutes.
                interval.tv_usec = 0;
                itime.it_interval = interval;
                itime.it_value = interval;
                signal(SIGALRM, CAST_SIGNAL_FUNC child_timeout_signal);
                setitimer(ITIMER_REAL, &itime, 0);
            }
            exit(query(arg, p + 1) != 0);
            break;
        }
        if (child > 0)
        {
            nChildrenStarted++;
        }

        int nChildren = nChildrenStarted - nChildrenEndedSIGCHLD
            - nChildrenEndedMain;

        // Collect the children.
        //
        while (waitpid(0, NULL, (nChildren < MAX_CHILDREN) ? WNOHANG : 0) > 0)
        {
            if (0 < nChildren)
            {
                nChildrenEndedMain++;
            }
        }
    }
    exit(0);
}

int query	(	char *	ip,
		char *	orig_arg
	)

Definition at line 66 of file slave.cpp.

References arg_for_errors, format_inet_addr(), INADDR_NONE, MAX_STRING, and stpcpy().

Referenced by main().

{
    char *comma;
    char *port_pair;
    struct hostent *hp;
    struct sockaddr_in sin;
    int s;
    FILE *f;
    char result[MAX_STRING];
    char buf[MAX_STRING * 2];
    char buf2[MAX_STRING * 2];
    char buf3[MAX_STRING * 4];
    char arg[MAX_STRING];
    size_t len;
    char *p;
    in_addr_t addr;

    addr = inet_addr(ip);
    if (addr == INADDR_NONE)
    {
        return -1;
    }
    const char *pHName = ip;
    hp = gethostbyaddr((char *)&addr, sizeof(addr), AF_INET);
    if (  hp
       && strlen(hp->h_name) < MAX_STRING)
    {
        pHName = hp->h_name;
    }
    p = stpcpy(buf, ip);
    *p++ = ' ';
    p = stpcpy(p, pHName);
    *p++ = '\n';
    *p++ = '\0';

    arg_for_errors = orig_arg;
    strcpy(arg, orig_arg);
    comma = (char *)strrchr(arg, ',');
    if (comma == NULL)
    {
        return -1;
    }
    *comma = 0;
    port_pair = (char *)strrchr(arg, ',');
    if (port_pair == NULL)
    {
        return -1;
    }
    *port_pair++ = 0;
    *comma = ',';

    hp = gethostbyname(arg);
    if (hp == NULL)
    {
        static struct hostent def;
        static struct in_addr defaddr;
        static char *alist[1];
        static char namebuf[MAX_STRING];

        defaddr.s_addr = inet_addr(arg);
        if (defaddr.s_addr == INADDR_NONE)
        {
            return -1;
        }
        strcpy(namebuf, arg);
        def.h_name = namebuf;
        def.h_addr_list = alist;
        def.h_addr = (char *)&defaddr;
        def.h_length = sizeof(struct in_addr);

        def.h_addrtype = AF_INET;
        def.h_aliases = 0;
        hp = &def;
    }
    sin.sin_family = hp->h_addrtype;
    bcopy(hp->h_addr, (char *)&sin.sin_addr, hp->h_length);
    sin.sin_port = htons(113); // ident port
    s = socket(hp->h_addrtype, SOCK_STREAM, 0);
    if (s < 0)
    {
        return -1;
    }
    if (connect(s, (struct sockaddr *)&sin, sizeof(sin)) < 0) {
        if (   errno != ECONNREFUSED
            && errno != ETIMEDOUT
            && errno != ENETUNREACH
            && errno != EHOSTUNREACH)
        {
            close(s);
            return -1;
        }
        buf2[0] = '\0';
    }
    else
    {
        len = strlen(port_pair);
        if ((size_t)write(s, port_pair, len) != len)
        {
            close(s);
            return (-1);
        }
        if (write(s, "\r\n", 2) != 2)
        {
            close(s);
            return (-1);
        }
        f = fdopen(s, "r");
        {
            int c;

            p = result;
            while ((c = fgetc(f)) != EOF)
            {
                if (c == '\n')
                {
                    break;
                }
                if (0x20 <= c && c <= 0x7E)
                {
                    *p++ = c;
                    if (p - result == MAX_STRING - 1)
                    {
                        break;
                    }
                }
            }
            *p = '\0';
        }
        fclose(f);
        p = (char *)format_inet_addr(buf2, ntohl(sin.sin_addr.s_addr));
        *p++ = ' ';
        p = stpcpy(p, result);
        *p++ = '\n';
        *p++ = '\0';
    }
    sprintf(buf3, "%s%s", buf, buf2);
    write(1, buf3, strlen(buf3));
    return 0;
}

char* stpcpy	(	char *	dest,
		const char *	src
	)

Definition at line 51 of file slave.cpp.

Referenced by query().

{
    while ((*dest = *src))
    {
        ++dest;
        ++src;
    }
    return (dest);
}

---

Variable Documentation

char* arg_for_errors

Definition at line 32 of file slave.cpp.

Referenced by query().

volatile int nChildrenEndedMain = 0

Definition at line 226 of file slave.cpp.

Referenced by child_signal(), and main().

volatile int nChildrenEndedSIGCHLD = 0

Definition at line 225 of file slave.cpp.

Referenced by child_signal(), and main().

volatile int nChildrenStarted = 0

Definition at line 224 of file slave.cpp.

Referenced by child_signal(), and main().

pid_t parent_pid

Definition at line 29 of file slave.cpp.

Referenced by alarm_signal(), and main().

---