/*
 * main.c
 *
 * Copyright (c) 1996-1999 Whistle Communications, Inc.
 * All rights reserved.
 * 
 * Subject to the following obligations and disclaimer of warranty, use and
 * redistribution of this software, in source or object code forms, with or
 * without modifications are expressly permitted by Whistle Communications;
 * provided, however, that:
 * 1. Any and all reproductions of the source or object code must include the
 *    copyright notice above and the following disclaimer of warranties; and
 * 2. No rights are granted, in any manner or form, to use Whistle
 *    Communications, Inc. trademarks, including the mark "WHISTLE
 *    COMMUNICATIONS" on advertising, endorsements, or otherwise except as
 *    such appears in the above copyright notice or in the software.
 * 
 * THIS SOFTWARE IS BEING PROVIDED BY WHISTLE COMMUNICATIONS "AS IS", AND
 * TO THE MAXIMUM EXTENT PERMITTED BY LAW, WHISTLE COMMUNICATIONS MAKES NO
 * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING THIS SOFTWARE,
 * INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT.
 * WHISTLE COMMUNICATIONS DOES NOT WARRANT, GUARANTEE, OR MAKE ANY
 * REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS OF THE USE OF THIS
 * SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY, RELIABILITY OR OTHERWISE.
 * IN NO EVENT SHALL WHISTLE COMMUNICATIONS BE LIABLE FOR ANY DAMAGES
 * RESULTING FROM OR ARISING OUT OF ANY USE OF THIS SOFTWARE, INCLUDING
 * WITHOUT LIMITATION, ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
 * PUNITIVE, OR CONSEQUENTIAL DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES, LOSS OF USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF WHISTLE COMMUNICATIONS IS ADVISED OF THE POSSIBILITY
 * OF SUCH DAMAGE.
 *
 * $FreeBSD: src/usr.sbin/nghook/main.c,v 1.4 1999/11/30 02:09:36 archie Exp $
 * $DragonFly: src/usr.sbin/nghook/main.c,v 1.5 2008/07/10 18:29:52 swildner Exp $
 * $Whistle: main.c,v 1.9 1999/01/20 00:26:26 archie Exp $
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <unistd.h>
#include <sysexits.h>
#include <errno.h>
#include <err.h>

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/select.h>

#include <netgraph.h>

#define DEFAULT_HOOKNAME        "debug"
#define NG_SOCK_HOOK_NAME       "hook"

#define BUF_SIZE                (64 * 1024)

static void     WriteAscii(u_char * buf, int len);
static void     Usage(void);

/*
 * main()
 */
int
main(int ac, char *av[])
{
        struct ngm_connect ngc;
        const char   *path = NULL, *hook = DEFAULT_HOOKNAME;
        int     csock, dsock;
        int     asciiFlag = 0;
        int     ch;

        /* Parse flags */
        while ((ch = getopt(ac, av, "da")) != -1) {
                switch (ch) {
                case 'd':
                        NgSetDebug(NgSetDebug(-1) + 1);
                        break;
                case 'a':
                        asciiFlag = 1;
                        break;
                case '?':
                default:
                        Usage();
                }
        }
        ac -= optind;
        av += optind;

        /* Get params */
        switch (ac) {
        case 2:
                hook = av[1];
                /* FALLTHROUGH */
        case 1:
                path = av[0];
                break;
        default:
                Usage();
        }

        /* Get sockets */
        if (NgMkSockNode(NULL, &csock, &dsock) < 0)
                errx(EX_OSERR, "can't get sockets");

        /* Connect socket node to specified node */
        snprintf(ngc.path, sizeof(ngc.path), "%s", path);
        snprintf(ngc.ourhook, sizeof(ngc.ourhook), NG_SOCK_HOOK_NAME);
        snprintf(ngc.peerhook, sizeof(ngc.peerhook), "%s", hook);

        if (NgSendMsg(csock, ".",
            NGM_GENERIC_COOKIE, NGM_CONNECT, &ngc, sizeof(ngc)) < 0)
                errx(EX_OSERR, "can't connect to node");

        /* Relay data */
        while (1) {
                fd_set  rfds;

                /* Setup bits */
                FD_ZERO(&rfds);
                FD_SET(0, &rfds);
                FD_SET(dsock, &rfds);

                /* Wait for something to happen */
                if (select(FD_SETSIZE, &rfds, NULL, NULL, NULL) < 0)
                        err(EX_OSERR, "select");

                /* Check data from socket */
                if (FD_ISSET(dsock, &rfds)) {
                        char    buf[BUF_SIZE];
                        int     rl, wl;

                        /* Read packet from socket */
                        if ((rl = NgRecvData(dsock,
                            buf, sizeof(buf), NULL)) < 0)
                                err(EX_OSERR, "read(hook)");
                        if (rl == 0)
                                errx(EX_OSERR, "read EOF from hook?!");

                        /* Write packet to stdout */
                        if (asciiFlag)
                                WriteAscii((u_char *) buf, rl);
                        else if ((wl = write(1, buf, rl)) != rl) {
                                if (wl < 0) {
                                        err(EX_OSERR, "write(stdout)");
                                } else {
                                        errx(EX_OSERR,
                                            "stdout: read %d, wrote %d",
                                            rl, wl);
                                }
                        }
                }

                /* Check data from stdin */
                if (FD_ISSET(0, &rfds)) {
                        char    buf[BUF_SIZE];
                        int     rl;

                        /* Read packet from stdin */
                        if ((rl = read(0, buf, sizeof(buf))) < 0)
                                err(EX_OSERR, "read(stdin)");
                        if (rl == 0)
                                errx(EX_OSERR, "EOF(stdin)");

                        /* Write packet to socket */
                        if (NgSendData(dsock, NG_SOCK_HOOK_NAME, buf, rl) < 0)
                                err(EX_OSERR, "write(hook)");
                }
        }
}

/*
 * Dump data in hex and ASCII form
 */
static void
WriteAscii(u_char *buf, int len)
{
        char    ch, sbuf[100];
        int     k, count;

        for (count = 0; count < len; count += 16) {
                snprintf(sbuf, sizeof(sbuf), "%04x:  ", count);
                for (k = 0; k < 16; k++)
                        if (count + k < len)
                                snprintf(sbuf + strlen(sbuf),
                                    sizeof(sbuf) - strlen(sbuf),
                                    "%02x ", buf[count + k]);
                        else
                                snprintf(sbuf + strlen(sbuf),
                                    sizeof(sbuf) - strlen(sbuf), "   ");
                snprintf(sbuf + strlen(sbuf), sizeof(sbuf) - strlen(sbuf), " ");
                for (k = 0; k < 16; k++)
                        if (count + k < len) {
                                ch = isprint(buf[count + k]) ?
                                    buf[count + k] : '.';
                                snprintf(sbuf + strlen(sbuf),
                                    sizeof(sbuf) - strlen(sbuf), "%c", ch);
                        } else
                                snprintf(sbuf + strlen(sbuf),
                                    sizeof(sbuf) - strlen(sbuf), " ");
                snprintf(sbuf + strlen(sbuf),
                    sizeof(sbuf) - strlen(sbuf), "\n");
                write(1, sbuf, strlen(sbuf));
        }
        ch = '\n';
        write(1, &ch, 1);
}

/*
 * Display usage and exit
 */
static void
Usage(void)
{
        errx(EX_USAGE, "usage: nghook [-da] path [hookname]");
}