usr.sbin/nghook/main.c

   1
   2 /*
   3  * main.c
   4  *
   5  * Copyright (c) 1996-1999 Whistle Communications, Inc.
   6  * All rights reserved.
   7  *
   8  * Subject to the following obligations and disclaimer of warranty, use and
   9  * redistribution of this software, in source or object code forms, with or
  10  * without modifications are expressly permitted by Whistle Communications;
  11  * provided, however, that:
  12  * 1. Any and all reproductions of the source or object code must include the
  13  *    copyright notice above and the following disclaimer of warranties; and
  14  * 2. No rights are granted, in any manner or form, to use Whistle
  15  *    Communications, Inc. trademarks, including the mark "WHISTLE
  16  *    COMMUNICATIONS" on advertising, endorsements, or otherwise except as
  17  *    such appears in the above copyright notice or in the software.
  18  *
  19  * THIS SOFTWARE IS BEING PROVIDED BY WHISTLE COMMUNICATIONS "AS IS", AND
  20  * TO THE MAXIMUM EXTENT PERMITTED BY LAW, WHISTLE COMMUNICATIONS MAKES NO
  21  * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING THIS SOFTWARE,
  22  * INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED WARRANTIES OF
  23  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT.
  24  * WHISTLE COMMUNICATIONS DOES NOT WARRANT, GUARANTEE, OR MAKE ANY
  25  * REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS OF THE USE OF THIS
  26  * SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY, RELIABILITY OR OTHERWISE.
  27  * IN NO EVENT SHALL WHISTLE COMMUNICATIONS BE LIABLE FOR ANY DAMAGES
  28  * RESULTING FROM OR ARISING OUT OF ANY USE OF THIS SOFTWARE, INCLUDING
  29  * WITHOUT LIMITATION, ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
  30  * PUNITIVE, OR CONSEQUENTIAL DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR
  31  * SERVICES, LOSS OF USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY
  32  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  33  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  34  * THIS SOFTWARE, EVEN IF WHISTLE COMMUNICATIONS IS ADVISED OF THE POSSIBILITY
  35  * OF SUCH DAMAGE.
  36  *
  37  * $FreeBSD: src/usr.sbin/nghook/main.c,v 1.4 1999/11/30 02:09:36 archie Exp $
  38  * $DragonFly: src/usr.sbin/nghook/main.c,v 1.5 2008/07/10 18:29:52 swildner Exp $
  39  * $Whistle: main.c,v 1.9 1999/01/20 00:26:26 archie Exp $
  40  */
  41
  42 #include <stdio.h>
  43 #include <stdlib.h>
  44 #include <string.h>
  45 #include <ctype.h>
  46 #include <unistd.h>
  47 #include <sysexits.h>
  48 #include <errno.h>
  49 #include <err.h>
  50
  51 #include <sys/types.h>
  52 #include <sys/socket.h>
  53 #include <sys/select.h>
  54
  55 #include <netgraph.h>
  56
  57 #define DEFAULT_HOOKNAME        "debug"
  58 #define NG_SOCK_HOOK_NAME       "hook"
  59
  60 #define BUF_SIZE                (64 * 1024)
  61
  62 static void     WriteAscii(u_char * buf, int len);
  63 static void     Usage(void);
  64
  65 /*
  66  * main()
  67  */
  68 int
  69 main(int ac, char *av[])
  70 {
  71         struct ngm_connect ngc;
  72         const char   *path = NULL, *hook = DEFAULT_HOOKNAME;
  73         int     csock, dsock;
  74         int     asciiFlag = 0;
  75         int     ch;
  76
  77         /* Parse flags */
  78         while ((ch = getopt(ac, av, "da")) != -1) {
  79                 switch (ch) {
  80                 case 'd':
  81                         NgSetDebug(NgSetDebug(-1) + 1);
  82                         break;
  83                 case 'a':
  84                         asciiFlag = 1;
  85                         break;
  86                 case '?':
  87                 default:
  88                         Usage();
  89                 }
  90         }
  91         ac -= optind;
  92         av += optind;
  93
  94         /* Get params */
  95         switch (ac) {
  96         case 2:
  97                 hook = av[1];
  98                 /* FALLTHROUGH */
  99         case 1:
 100                 path = av[0];
 101                 break;
 102         default:
 103                 Usage();
 104         }
 105
 106         /* Get sockets */
 107         if (NgMkSockNode(NULL, &csock, &dsock) < 0)
 108                 errx(EX_OSERR, "can't get sockets");
 109
 110         /* Connect socket node to specified node */
 111         snprintf(ngc.path, sizeof(ngc.path), "%s", path);
 112         snprintf(ngc.ourhook, sizeof(ngc.ourhook), NG_SOCK_HOOK_NAME);
 113         snprintf(ngc.peerhook, sizeof(ngc.peerhook), "%s", hook);
 114
 115         if (NgSendMsg(csock, ".",
 116             NGM_GENERIC_COOKIE, NGM_CONNECT, &ngc, sizeof(ngc)) < 0)
 117                 errx(EX_OSERR, "can't connect to node");
 118
 119         /* Relay data */
 120         while (1) {
 121                 fd_set  rfds;
 122
 123                 /* Setup bits */
 124                 FD_ZERO(&rfds);
 125                 FD_SET(0, &rfds);
 126                 FD_SET(dsock, &rfds);
 127
 128                 /* Wait for something to happen */
 129                 if (select(FD_SETSIZE, &rfds, NULL, NULL, NULL) < 0)
 130                         err(EX_OSERR, "select");
 131
 132                 /* Check data from socket */
 133                 if (FD_ISSET(dsock, &rfds)) {
 134                         char    buf[BUF_SIZE];
 135                         int     rl, wl;
 136
 137                         /* Read packet from socket */
 138                         if ((rl = NgRecvData(dsock,
 139                             buf, sizeof(buf), NULL)) < 0)
 140                                 err(EX_OSERR, "read(hook)");
 141                         if (rl == 0)
 142                                 errx(EX_OSERR, "read EOF from hook?!");
 143
 144                         /* Write packet to stdout */
 145                         if (asciiFlag)
 146                                 WriteAscii((u_char *) buf, rl);
 147                         else if ((wl = write(1, buf, rl)) != rl) {
 148                                 if (wl < 0) {
 149                                         err(EX_OSERR, "write(stdout)");
 150                                 } else {
 151                                         errx(EX_OSERR,
 152                                             "stdout: read %d, wrote %d",
 153                                             rl, wl);
 154                                 }
 155                         }
 156                 }
 157
 158                 /* Check data from stdin */
 159                 if (FD_ISSET(0, &rfds)) {
 160                         char    buf[BUF_SIZE];
 161                         int     rl;
 162
 163                         /* Read packet from stdin */
 164                         if ((rl = read(0, buf, sizeof(buf))) < 0)
 165                                 err(EX_OSERR, "read(stdin)");
 166                         if (rl == 0)
 167                                 errx(EX_OSERR, "EOF(stdin)");
 168
 169                         /* Write packet to socket */
 170                         if (NgSendData(dsock, NG_SOCK_HOOK_NAME, buf, rl) < 0)
 171                                 err(EX_OSERR, "write(hook)");
 172                 }
 173         }
 174 }
 175
 176 /*
 177  * Dump data in hex and ASCII form
 178  */
 179 static void
 180 WriteAscii(u_char *buf, int len)
 181 {
 182         char    ch, sbuf[100];
 183         int     k, count;
 184
 185         for (count = 0; count < len; count += 16) {
 186                 snprintf(sbuf, sizeof(sbuf), "%04x:  ", count);
 187                 for (k = 0; k < 16; k++)
 188                         if (count + k < len)
 189                                 snprintf(sbuf + strlen(sbuf),
 190                                     sizeof(sbuf) - strlen(sbuf),
 191                                     "%02x ", buf[count + k]);
 192                         else
 193                                 snprintf(sbuf + strlen(sbuf),
 194                                     sizeof(sbuf) - strlen(sbuf), "   ");
 195                 snprintf(sbuf + strlen(sbuf), sizeof(sbuf) - strlen(sbuf), " ");
 196                 for (k = 0; k < 16; k++)
 197                         if (count + k < len) {
 198                                 ch = isprint(buf[count + k]) ?
 199                                     buf[count + k] : '.';
 200                                 snprintf(sbuf + strlen(sbuf),
 201                                     sizeof(sbuf) - strlen(sbuf), "%c", ch);
 202                         } else
 203                                 snprintf(sbuf + strlen(sbuf),
 204                                     sizeof(sbuf) - strlen(sbuf), " ");
 205                 snprintf(sbuf + strlen(sbuf),
 206                     sizeof(sbuf) - strlen(sbuf), "\n");
 207                 write(1, sbuf, strlen(sbuf));
 208         }
 209         ch = '\n';
 210         write(1, &ch, 1);
 211 }
 212
 213 /*
 214  * Display usage and exit
 215  */
 216 static void
 217 Usage(void)
 218 {
 219         errx(EX_USAGE, "usage: nghook [-da] path [hookname]");
 220 }
 221