freebsd-dev/usr.sbin/nghook/main.c
Matthew Dillon ebdb21f429 Put include <netgraph.h> back in - problem was due to a cvsup mess on
freefall.  There may also be a real problem w/ buildworld but the fix
    would go somewhere else, not here.
1999-10-24 02:58:39 +00:00

221 lines
5.7 KiB
C

/*
* 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$
* $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;
char *path = NULL, *hook = DEFAULT_HOOKNAME;
int csock, dsock;
int asciiFlag = 0;
int ch;
/* Parse flags */
while ((ch = getopt(ac, av, "da")) != EOF) {
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");
(void) 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]");
}