xref: /freebsd/usr.sbin/ngctl/main.c (revision f8307e1233657707bc582110f07373c96d91943b)
14cf49a43SJulian Elischer 
24cf49a43SJulian Elischer /*
34cf49a43SJulian Elischer  * main.c
44cf49a43SJulian Elischer  *
54cf49a43SJulian Elischer  * Copyright (c) 1996-1999 Whistle Communications, Inc.
64cf49a43SJulian Elischer  * All rights reserved.
74cf49a43SJulian Elischer  *
84cf49a43SJulian Elischer  * Subject to the following obligations and disclaimer of warranty, use and
94cf49a43SJulian Elischer  * redistribution of this software, in source or object code forms, with or
104cf49a43SJulian Elischer  * without modifications are expressly permitted by Whistle Communications;
114cf49a43SJulian Elischer  * provided, however, that:
124cf49a43SJulian Elischer  * 1. Any and all reproductions of the source or object code must include the
134cf49a43SJulian Elischer  *    copyright notice above and the following disclaimer of warranties; and
144cf49a43SJulian Elischer  * 2. No rights are granted, in any manner or form, to use Whistle
154cf49a43SJulian Elischer  *    Communications, Inc. trademarks, including the mark "WHISTLE
164cf49a43SJulian Elischer  *    COMMUNICATIONS" on advertising, endorsements, or otherwise except as
174cf49a43SJulian Elischer  *    such appears in the above copyright notice or in the software.
184cf49a43SJulian Elischer  *
194cf49a43SJulian Elischer  * THIS SOFTWARE IS BEING PROVIDED BY WHISTLE COMMUNICATIONS "AS IS", AND
204cf49a43SJulian Elischer  * TO THE MAXIMUM EXTENT PERMITTED BY LAW, WHISTLE COMMUNICATIONS MAKES NO
214cf49a43SJulian Elischer  * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING THIS SOFTWARE,
224cf49a43SJulian Elischer  * INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED WARRANTIES OF
234cf49a43SJulian Elischer  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT.
244cf49a43SJulian Elischer  * WHISTLE COMMUNICATIONS DOES NOT WARRANT, GUARANTEE, OR MAKE ANY
254cf49a43SJulian Elischer  * REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS OF THE USE OF THIS
264cf49a43SJulian Elischer  * SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY, RELIABILITY OR OTHERWISE.
274cf49a43SJulian Elischer  * IN NO EVENT SHALL WHISTLE COMMUNICATIONS BE LIABLE FOR ANY DAMAGES
284cf49a43SJulian Elischer  * RESULTING FROM OR ARISING OUT OF ANY USE OF THIS SOFTWARE, INCLUDING
294cf49a43SJulian Elischer  * WITHOUT LIMITATION, ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
304cf49a43SJulian Elischer  * PUNITIVE, OR CONSEQUENTIAL DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR
314cf49a43SJulian Elischer  * SERVICES, LOSS OF USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY
324cf49a43SJulian Elischer  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
334cf49a43SJulian Elischer  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
344cf49a43SJulian Elischer  * THIS SOFTWARE, EVEN IF WHISTLE COMMUNICATIONS IS ADVISED OF THE POSSIBILITY
354cf49a43SJulian Elischer  * OF SUCH DAMAGE.
364cf49a43SJulian Elischer  *
374cf49a43SJulian Elischer  * $FreeBSD$
38f8307e12SArchie Cobbs  * $Whistle: main.c,v 1.12 1999/11/29 19:17:46 archie Exp $
394cf49a43SJulian Elischer  */
404cf49a43SJulian Elischer 
414cf49a43SJulian Elischer #include "ngctl.h"
424cf49a43SJulian Elischer 
434cf49a43SJulian Elischer #define PROMPT			"+ "
444cf49a43SJulian Elischer #define MAX_ARGS		512
454cf49a43SJulian Elischer #define WHITESPACE		" \t\r\n\v\f"
460d6985e2SArchie Cobbs #define NG_SOCKET_KLD		"ng_socket.ko"
47f8307e12SArchie Cobbs #define DUMP_BYTES_PER_LINE	16
484cf49a43SJulian Elischer 
494cf49a43SJulian Elischer /* Internal functions */
504cf49a43SJulian Elischer static int	ReadFile(FILE *fp);
514cf49a43SJulian Elischer static int	DoParseCommand(char *line);
524cf49a43SJulian Elischer static int	DoCommand(int ac, char **av);
534cf49a43SJulian Elischer static int	DoInteractive(void);
544cf49a43SJulian Elischer static const	struct ngcmd *FindCommand(const char *string);
55f8307e12SArchie Cobbs static int	MatchCommand(const struct ngcmd *cmd, const char *s);
56f8307e12SArchie Cobbs static void	DumpAscii(const u_char *buf, int len);
574cf49a43SJulian Elischer static void	Usage(const char *msg);
584cf49a43SJulian Elischer static int	ReadCmd(int ac, char **av);
594cf49a43SJulian Elischer static int	HelpCmd(int ac, char **av);
604cf49a43SJulian Elischer static int	QuitCmd(int ac, char **av);
614cf49a43SJulian Elischer 
624cf49a43SJulian Elischer /* List of commands */
634cf49a43SJulian Elischer static const struct ngcmd *const cmds[] = {
644cf49a43SJulian Elischer 	&connect_cmd,
654cf49a43SJulian Elischer 	&debug_cmd,
664cf49a43SJulian Elischer 	&help_cmd,
674cf49a43SJulian Elischer 	&list_cmd,
684cf49a43SJulian Elischer 	&mkpeer_cmd,
69f8307e12SArchie Cobbs 	&msg_cmd,
704cf49a43SJulian Elischer 	&name_cmd,
714cf49a43SJulian Elischer 	&read_cmd,
724cf49a43SJulian Elischer 	&rmhook_cmd,
734cf49a43SJulian Elischer 	&show_cmd,
744cf49a43SJulian Elischer 	&shutdown_cmd,
754cf49a43SJulian Elischer 	&status_cmd,
764cf49a43SJulian Elischer 	&types_cmd,
774cf49a43SJulian Elischer 	&quit_cmd,
784cf49a43SJulian Elischer 	NULL
794cf49a43SJulian Elischer };
804cf49a43SJulian Elischer 
814cf49a43SJulian Elischer /* Commands defined in this file */
824cf49a43SJulian Elischer const struct ngcmd read_cmd = {
834cf49a43SJulian Elischer 	ReadCmd,
844cf49a43SJulian Elischer 	"read <filename>",
854cf49a43SJulian Elischer 	"Read and execute commands from a file",
86f8307e12SArchie Cobbs 	NULL,
87f8307e12SArchie Cobbs 	{ "source", "." }
884cf49a43SJulian Elischer };
894cf49a43SJulian Elischer const struct ngcmd help_cmd = {
904cf49a43SJulian Elischer 	HelpCmd,
914cf49a43SJulian Elischer 	"help [command]",
924cf49a43SJulian Elischer 	"Show command summary or get more help on a specific command",
93f8307e12SArchie Cobbs 	NULL,
94f8307e12SArchie Cobbs 	{ "?" }
954cf49a43SJulian Elischer };
964cf49a43SJulian Elischer const struct ngcmd quit_cmd = {
974cf49a43SJulian Elischer 	QuitCmd,
984cf49a43SJulian Elischer 	"quit",
994cf49a43SJulian Elischer 	"Exit program",
100f8307e12SArchie Cobbs 	NULL,
101f8307e12SArchie Cobbs 	{ "exit" }
1024cf49a43SJulian Elischer };
1034cf49a43SJulian Elischer 
1044cf49a43SJulian Elischer /* Our control and data sockets */
1054cf49a43SJulian Elischer int	csock, dsock;
1064cf49a43SJulian Elischer 
1074cf49a43SJulian Elischer /*
1084cf49a43SJulian Elischer  * main()
1094cf49a43SJulian Elischer  */
1104cf49a43SJulian Elischer int
1114cf49a43SJulian Elischer main(int ac, char *av[])
1124cf49a43SJulian Elischer {
1134cf49a43SJulian Elischer 	char	name[NG_NODELEN + 1];
1144cf49a43SJulian Elischer 	int	interactive = isatty(0) && isatty(1);
1154cf49a43SJulian Elischer 	FILE	*fp = NULL;
1164cf49a43SJulian Elischer 	int	ch, rtn = 0;
1174cf49a43SJulian Elischer 
1184cf49a43SJulian Elischer 	/* Set default node name */
1194cf49a43SJulian Elischer 	snprintf(name, sizeof(name), "ngctl%d", getpid());
1204cf49a43SJulian Elischer 
1214cf49a43SJulian Elischer 	/* Parse command line */
1224cf49a43SJulian Elischer 	while ((ch = getopt(ac, av, "df:n:")) != EOF) {
1234cf49a43SJulian Elischer 		switch (ch) {
1244cf49a43SJulian Elischer 		case 'd':
1254cf49a43SJulian Elischer 			NgSetDebug(NgSetDebug(-1) + 1);
1264cf49a43SJulian Elischer 			break;
1274cf49a43SJulian Elischer 		case 'f':
1284cf49a43SJulian Elischer 			if (strcmp(optarg, "-") == 0)
1294cf49a43SJulian Elischer 				fp = stdin;
1304cf49a43SJulian Elischer 			else if ((fp = fopen(optarg, "r")) == NULL)
1314cf49a43SJulian Elischer 				err(EX_NOINPUT, "%s", optarg);
1324cf49a43SJulian Elischer 			break;
1334cf49a43SJulian Elischer 		case 'n':
1344cf49a43SJulian Elischer 			snprintf(name, sizeof(name), "%s", optarg);
1354cf49a43SJulian Elischer 			break;
1364cf49a43SJulian Elischer 		case '?':
1374cf49a43SJulian Elischer 		default:
1384cf49a43SJulian Elischer 			Usage((char *)NULL);
1394cf49a43SJulian Elischer 			break;
1404cf49a43SJulian Elischer 		}
1414cf49a43SJulian Elischer 	}
1424cf49a43SJulian Elischer 	ac -= optind;
1434cf49a43SJulian Elischer 	av += optind;
1444cf49a43SJulian Elischer 
1454cf49a43SJulian Elischer 	/* Create a new socket node */
1460d6985e2SArchie Cobbs 	if (NgMkSockNode(name, &csock, &dsock) < 0) {
1470d6985e2SArchie Cobbs 		if (errno == EPROTONOSUPPORT) {
1480d6985e2SArchie Cobbs 			if (kldload(NG_SOCKET_KLD) < 0)
1490d6985e2SArchie Cobbs 				err(EX_OSERR, "can't load %s", NG_SOCKET_KLD);
1500d6985e2SArchie Cobbs 			if (NgMkSockNode(name, &csock, &dsock) >= 0)
1510d6985e2SArchie Cobbs 				goto gotNode;
1520d6985e2SArchie Cobbs 		}
1534cf49a43SJulian Elischer 		err(EX_OSERR, "can't create node");
1540d6985e2SArchie Cobbs 	}
1550d6985e2SArchie Cobbs gotNode:
1564cf49a43SJulian Elischer 
1574cf49a43SJulian Elischer 	/* Do commands as requested */
1584cf49a43SJulian Elischer 	if (ac == 0) {
1594cf49a43SJulian Elischer 		if (fp != NULL) {
1604cf49a43SJulian Elischer 			rtn = ReadFile(fp);
1614cf49a43SJulian Elischer 		} else if (interactive) {
1624cf49a43SJulian Elischer 			rtn = DoInteractive();
1634cf49a43SJulian Elischer 		} else
1644cf49a43SJulian Elischer 			Usage("no command specified");
1654cf49a43SJulian Elischer 	} else {
1664cf49a43SJulian Elischer 		rtn = DoCommand(ac, av);
1674cf49a43SJulian Elischer 	}
1684cf49a43SJulian Elischer 
1694cf49a43SJulian Elischer 	/* Convert command return code into system exit code */
1704cf49a43SJulian Elischer 	switch (rtn) {
1714cf49a43SJulian Elischer 	case CMDRTN_OK:
1724cf49a43SJulian Elischer 	case CMDRTN_QUIT:
1734cf49a43SJulian Elischer 		rtn = 0;
1744cf49a43SJulian Elischer 		break;
1754cf49a43SJulian Elischer 	case CMDRTN_USAGE:
1764cf49a43SJulian Elischer 		rtn = EX_USAGE;
1774cf49a43SJulian Elischer 		break;
1784cf49a43SJulian Elischer 	case CMDRTN_ERROR:
1794cf49a43SJulian Elischer 		rtn = EX_OSERR;
1804cf49a43SJulian Elischer 		break;
1814cf49a43SJulian Elischer 	}
1824cf49a43SJulian Elischer 	return(rtn);
1834cf49a43SJulian Elischer }
1844cf49a43SJulian Elischer 
1854cf49a43SJulian Elischer /*
1864cf49a43SJulian Elischer  * Process commands from a file
1874cf49a43SJulian Elischer  */
1884cf49a43SJulian Elischer static int
1894cf49a43SJulian Elischer ReadFile(FILE *fp)
1904cf49a43SJulian Elischer {
1914cf49a43SJulian Elischer 	char line[LINE_MAX];
1924cf49a43SJulian Elischer 	int num, rtn;
1934cf49a43SJulian Elischer 
1944cf49a43SJulian Elischer 	for (num = 1; fgets(line, sizeof(line), fp) != NULL; num++) {
1954cf49a43SJulian Elischer 		if (*line == '#')
1964cf49a43SJulian Elischer 			continue;
1974cf49a43SJulian Elischer 		if ((rtn = DoParseCommand(line)) != 0) {
1984cf49a43SJulian Elischer 			warnx("line %d: error in file", num);
1994cf49a43SJulian Elischer 			return(rtn);
2004cf49a43SJulian Elischer 		}
2014cf49a43SJulian Elischer 	}
2024cf49a43SJulian Elischer 	return(CMDRTN_OK);
2034cf49a43SJulian Elischer }
2044cf49a43SJulian Elischer 
2054cf49a43SJulian Elischer /*
2064cf49a43SJulian Elischer  * Interactive mode
2074cf49a43SJulian Elischer  */
2084cf49a43SJulian Elischer static int
2094cf49a43SJulian Elischer DoInteractive(void)
2104cf49a43SJulian Elischer {
211f8307e12SArchie Cobbs 	const int maxfd = MAX(csock, dsock) + 1;
212f8307e12SArchie Cobbs 
213f8307e12SArchie Cobbs 	(*help_cmd.func)(0, NULL);
214f8307e12SArchie Cobbs 	while (1) {
215f8307e12SArchie Cobbs 		struct timeval tv;
216f8307e12SArchie Cobbs 		fd_set rfds;
217f8307e12SArchie Cobbs 
218f8307e12SArchie Cobbs 		/* See if any data or control messages are arriving */
219f8307e12SArchie Cobbs 		FD_ZERO(&rfds);
220f8307e12SArchie Cobbs 		FD_SET(csock, &rfds);
221f8307e12SArchie Cobbs 		FD_SET(dsock, &rfds);
222f8307e12SArchie Cobbs 		memset(&tv, 0, sizeof(tv));
223f8307e12SArchie Cobbs 		if (select(maxfd, &rfds, NULL, NULL, &tv) <= 0) {
224f8307e12SArchie Cobbs 
225f8307e12SArchie Cobbs 			/* Issue prompt and wait for anything to happen */
226f8307e12SArchie Cobbs 			printf("%s", PROMPT);
227f8307e12SArchie Cobbs 			fflush(stdout);
228f8307e12SArchie Cobbs 			FD_ZERO(&rfds);
229f8307e12SArchie Cobbs 			FD_SET(0, &rfds);
230f8307e12SArchie Cobbs 			FD_SET(csock, &rfds);
231f8307e12SArchie Cobbs 			FD_SET(dsock, &rfds);
232f8307e12SArchie Cobbs 			if (select(maxfd, &rfds, NULL, NULL, NULL) < 0)
233f8307e12SArchie Cobbs 				err(EX_OSERR, "select");
234f8307e12SArchie Cobbs 
235f8307e12SArchie Cobbs 			/* If not user input, print a newline first */
236f8307e12SArchie Cobbs 			if (!FD_ISSET(0, &rfds))
237f8307e12SArchie Cobbs 				printf("\n");
238f8307e12SArchie Cobbs 		}
239f8307e12SArchie Cobbs 
240f8307e12SArchie Cobbs 		/* Display any incoming control message */
241f8307e12SArchie Cobbs 		while (FD_ISSET(csock, &rfds)) {
242f8307e12SArchie Cobbs 			u_char buf[2 * sizeof(struct ng_mesg) + 8192];
243f8307e12SArchie Cobbs 			struct ng_mesg *const m = (struct ng_mesg *)buf;
244f8307e12SArchie Cobbs 			struct ng_mesg *const ascii = (struct ng_mesg *)m->data;
245f8307e12SArchie Cobbs 			char path[NG_PATHLEN+1];
246f8307e12SArchie Cobbs 
247f8307e12SArchie Cobbs 			/* Get incoming message (in binary form) */
248f8307e12SArchie Cobbs 			if (NgRecvMsg(csock, m, sizeof(buf), path) < 0) {
249f8307e12SArchie Cobbs 				warn("recv incoming message");
250f8307e12SArchie Cobbs 				break;
251f8307e12SArchie Cobbs 			}
252f8307e12SArchie Cobbs 
253f8307e12SArchie Cobbs 			/* Ask originating node to convert to ASCII */
254f8307e12SArchie Cobbs 			if (NgSendMsg(csock, path, NGM_GENERIC_COOKIE,
255f8307e12SArchie Cobbs 			      NGM_BINARY2ASCII, m,
256f8307e12SArchie Cobbs 			      sizeof(*m) + m->header.arglen) < 0
257f8307e12SArchie Cobbs 			    || NgRecvMsg(csock, m, sizeof(buf), NULL) < 0) {
258f8307e12SArchie Cobbs 				printf("Rec'd %s %d from \"%s\":\n",
259f8307e12SArchie Cobbs 				    (m->header.flags & NGF_RESP) != 0 ?
260f8307e12SArchie Cobbs 				      "response" : "command",
261f8307e12SArchie Cobbs 				    m->header.cmd, path);
262f8307e12SArchie Cobbs 				if (m->header.arglen == 0)
263f8307e12SArchie Cobbs 					printf("No arguments\n");
264f8307e12SArchie Cobbs 				else
265f8307e12SArchie Cobbs 					DumpAscii(m->data, m->header.arglen);
266f8307e12SArchie Cobbs 				break;
267f8307e12SArchie Cobbs 			}
268f8307e12SArchie Cobbs 
269f8307e12SArchie Cobbs 			/* Display message in ASCII form */
270f8307e12SArchie Cobbs 			printf("Rec'd %s \"%s\" (%d) from \"%s\":\n",
271f8307e12SArchie Cobbs 			    (ascii->header.flags & NGF_RESP) != 0 ?
272f8307e12SArchie Cobbs 			      "response" : "command",
273f8307e12SArchie Cobbs 			    ascii->header.cmdstr, ascii->header.cmd, path);
274f8307e12SArchie Cobbs 			if (*ascii->data != '\0')
275f8307e12SArchie Cobbs 				printf("Args:\t%s\n", ascii->data);
276f8307e12SArchie Cobbs 			else
277f8307e12SArchie Cobbs 				printf("No arguments\n");
278f8307e12SArchie Cobbs 			break;
279f8307e12SArchie Cobbs 		}
280f8307e12SArchie Cobbs 
281f8307e12SArchie Cobbs 		/* Display any incoming data packet */
282f8307e12SArchie Cobbs 		while (FD_ISSET(dsock, &rfds)) {
283f8307e12SArchie Cobbs 			u_char buf[8192];
284f8307e12SArchie Cobbs 			char hook[NG_HOOKLEN + 1];
285f8307e12SArchie Cobbs 			int rl;
286f8307e12SArchie Cobbs 
287f8307e12SArchie Cobbs 			/* Read packet from socket */
288f8307e12SArchie Cobbs 			if ((rl = NgRecvData(dsock,
289f8307e12SArchie Cobbs 			    buf, sizeof(buf), hook)) < 0)
290f8307e12SArchie Cobbs 				err(EX_OSERR, "read(hook)");
291f8307e12SArchie Cobbs 			if (rl == 0)
292f8307e12SArchie Cobbs 				errx(EX_OSERR, "EOF from hook \"%s\"?", hook);
293f8307e12SArchie Cobbs 
294f8307e12SArchie Cobbs 			/* Write packet to stdout */
295f8307e12SArchie Cobbs 			printf("Rec'd data packet on hook \"%s\":\n", hook);
296f8307e12SArchie Cobbs 			DumpAscii(buf, rl);
297f8307e12SArchie Cobbs 			break;
298f8307e12SArchie Cobbs 		}
299f8307e12SArchie Cobbs 
300f8307e12SArchie Cobbs 		/* Get any user input */
301f8307e12SArchie Cobbs 		if (FD_ISSET(0, &rfds)) {
3024cf49a43SJulian Elischer 			char buf[LINE_MAX];
3034cf49a43SJulian Elischer 
304f8307e12SArchie Cobbs 			if (fgets(buf, sizeof(buf), stdin) == NULL) {
305f8307e12SArchie Cobbs 				printf("\n");
3064cf49a43SJulian Elischer 				break;
307f8307e12SArchie Cobbs 			}
308f8307e12SArchie Cobbs 			if (DoParseCommand(buf) == CMDRTN_QUIT)
309f8307e12SArchie Cobbs 				break;
310f8307e12SArchie Cobbs 		}
311f8307e12SArchie Cobbs 	}
3124cf49a43SJulian Elischer 	return(CMDRTN_QUIT);
3134cf49a43SJulian Elischer }
3144cf49a43SJulian Elischer 
3154cf49a43SJulian Elischer /*
3164cf49a43SJulian Elischer  * Parse a command line and execute the command
3174cf49a43SJulian Elischer  */
3184cf49a43SJulian Elischer static int
3194cf49a43SJulian Elischer DoParseCommand(char *line)
3204cf49a43SJulian Elischer {
3214cf49a43SJulian Elischer 	char *av[MAX_ARGS];
3224cf49a43SJulian Elischer 	int ac;
3234cf49a43SJulian Elischer 
3244cf49a43SJulian Elischer 	/* Parse line */
3254cf49a43SJulian Elischer 	for (ac = 0, av[0] = strtok(line, WHITESPACE);
3264cf49a43SJulian Elischer 	    ac < MAX_ARGS - 1 && av[ac];
3274cf49a43SJulian Elischer 	    av[++ac] = strtok(NULL, WHITESPACE));
3284cf49a43SJulian Elischer 
3294cf49a43SJulian Elischer 	/* Do command */
3304cf49a43SJulian Elischer 	return(DoCommand(ac, av));
3314cf49a43SJulian Elischer }
3324cf49a43SJulian Elischer 
3334cf49a43SJulian Elischer /*
3344cf49a43SJulian Elischer  * Execute the command
3354cf49a43SJulian Elischer  */
3364cf49a43SJulian Elischer static int
3374cf49a43SJulian Elischer DoCommand(int ac, char **av)
3384cf49a43SJulian Elischer {
3394cf49a43SJulian Elischer 	const struct ngcmd *cmd;
3404cf49a43SJulian Elischer 	int rtn;
3414cf49a43SJulian Elischer 
3424cf49a43SJulian Elischer 	if (ac == 0 || *av[0] == 0)
3434cf49a43SJulian Elischer 		return(CMDRTN_OK);
3444cf49a43SJulian Elischer 	if ((cmd = FindCommand(av[0])) == NULL)
3454cf49a43SJulian Elischer 		return(CMDRTN_ERROR);
3464cf49a43SJulian Elischer 	if ((rtn = (*cmd->func)(ac, av)) == CMDRTN_USAGE)
3474cf49a43SJulian Elischer 		warnx("usage: %s", cmd->cmd);
3484cf49a43SJulian Elischer 	return(rtn);
3494cf49a43SJulian Elischer }
3504cf49a43SJulian Elischer 
3514cf49a43SJulian Elischer /*
3524cf49a43SJulian Elischer  * Find a command
3534cf49a43SJulian Elischer  */
3544cf49a43SJulian Elischer static const struct ngcmd *
3554cf49a43SJulian Elischer FindCommand(const char *string)
3564cf49a43SJulian Elischer {
357f8307e12SArchie Cobbs 	int k, found = -1;
3584cf49a43SJulian Elischer 
359f8307e12SArchie Cobbs 	for (k = 0; cmds[k] != NULL; k++) {
360f8307e12SArchie Cobbs 		if (MatchCommand(cmds[k], string)) {
3614cf49a43SJulian Elischer 			if (found != -1) {
3624cf49a43SJulian Elischer 				warnx("\"%s\": ambiguous command", string);
3634cf49a43SJulian Elischer 				return(NULL);
3644cf49a43SJulian Elischer 			}
3654cf49a43SJulian Elischer 			found = k;
3664cf49a43SJulian Elischer 		}
3674cf49a43SJulian Elischer 	}
3684cf49a43SJulian Elischer 	if (found == -1) {
3694cf49a43SJulian Elischer 		warnx("\"%s\": unknown command", string);
3704cf49a43SJulian Elischer 		return(NULL);
3714cf49a43SJulian Elischer 	}
3724cf49a43SJulian Elischer 	return(cmds[found]);
3734cf49a43SJulian Elischer }
3744cf49a43SJulian Elischer 
3754cf49a43SJulian Elischer /*
376f8307e12SArchie Cobbs  * See if string matches a prefix of "cmd" (or an alias) case insensitively
377f8307e12SArchie Cobbs  */
378f8307e12SArchie Cobbs static int
379f8307e12SArchie Cobbs MatchCommand(const struct ngcmd *cmd, const char *s)
380f8307e12SArchie Cobbs {
381f8307e12SArchie Cobbs 	int a;
382f8307e12SArchie Cobbs 
383f8307e12SArchie Cobbs 	/* Try to match command, ignoring the usage stuff */
384f8307e12SArchie Cobbs 	if (strlen(s) <= strcspn(cmd->cmd, WHITESPACE)) {
385f8307e12SArchie Cobbs 		if (strncasecmp(s, cmd->cmd, strlen(s)) == 0)
386f8307e12SArchie Cobbs 			return (1);
387f8307e12SArchie Cobbs 	}
388f8307e12SArchie Cobbs 
389f8307e12SArchie Cobbs 	/* Try to match aliases */
390f8307e12SArchie Cobbs 	for (a = 0; a < MAX_CMD_ALIAS && cmd->aliases[a] != NULL; a++) {
391f8307e12SArchie Cobbs 		if (strlen(cmd->aliases[a]) >= strlen(s)) {
392f8307e12SArchie Cobbs 			if (strncasecmp(s, cmd->aliases[a], strlen(s)) == 0)
393f8307e12SArchie Cobbs 				return (1);
394f8307e12SArchie Cobbs 		}
395f8307e12SArchie Cobbs 	}
396f8307e12SArchie Cobbs 
397f8307e12SArchie Cobbs 	/* No match */
398f8307e12SArchie Cobbs 	return (0);
399f8307e12SArchie Cobbs }
400f8307e12SArchie Cobbs 
401f8307e12SArchie Cobbs /*
4024cf49a43SJulian Elischer  * ReadCmd()
4034cf49a43SJulian Elischer  */
4044cf49a43SJulian Elischer static int
4054cf49a43SJulian Elischer ReadCmd(int ac, char **av)
4064cf49a43SJulian Elischer {
4074cf49a43SJulian Elischer 	FILE *fp;
4084cf49a43SJulian Elischer 	int rtn;
4094cf49a43SJulian Elischer 
4104cf49a43SJulian Elischer 	/* Open file */
4114cf49a43SJulian Elischer 	switch (ac) {
4124cf49a43SJulian Elischer 	case 2:
4134cf49a43SJulian Elischer 		if ((fp = fopen(av[1], "r")) == NULL)
4144cf49a43SJulian Elischer 			warn("%s", av[1]);
4154cf49a43SJulian Elischer 		return(CMDRTN_ERROR);
4164cf49a43SJulian Elischer 	default:
4174cf49a43SJulian Elischer 		return(CMDRTN_USAGE);
4184cf49a43SJulian Elischer 	}
4194cf49a43SJulian Elischer 
4204cf49a43SJulian Elischer 	/* Process it */
4214cf49a43SJulian Elischer 	rtn = ReadFile(fp);
4224cf49a43SJulian Elischer 	fclose(fp);
4234cf49a43SJulian Elischer 	return(rtn);
4244cf49a43SJulian Elischer }
4254cf49a43SJulian Elischer 
4264cf49a43SJulian Elischer /*
4274cf49a43SJulian Elischer  * HelpCmd()
4284cf49a43SJulian Elischer  */
4294cf49a43SJulian Elischer static int
4304cf49a43SJulian Elischer HelpCmd(int ac, char **av)
4314cf49a43SJulian Elischer {
4324cf49a43SJulian Elischer 	const struct ngcmd *cmd;
4334cf49a43SJulian Elischer 	int k;
4344cf49a43SJulian Elischer 
4354cf49a43SJulian Elischer 	switch (ac) {
4364cf49a43SJulian Elischer 	case 0:
4374cf49a43SJulian Elischer 	case 1:
4384cf49a43SJulian Elischer 		/* Show all commands */
4394cf49a43SJulian Elischer 		printf("Available commands:\n");
4404cf49a43SJulian Elischer 		for (k = 0; cmds[k] != NULL; k++) {
4414cf49a43SJulian Elischer 			char *s, buf[100];
4424cf49a43SJulian Elischer 
4434cf49a43SJulian Elischer 			cmd = cmds[k];
4444cf49a43SJulian Elischer 			snprintf(buf, sizeof(buf), "%s", cmd->cmd);
4454cf49a43SJulian Elischer 			for (s = buf; *s != '\0' && !isspace(*s); s++);
4464cf49a43SJulian Elischer 			*s = '\0';
4474cf49a43SJulian Elischer 			printf("  %-10s %s\n", buf, cmd->desc);
4484cf49a43SJulian Elischer 		}
4494cf49a43SJulian Elischer 		return(CMDRTN_OK);
4504cf49a43SJulian Elischer 	default:
4514cf49a43SJulian Elischer 		/* Show help on a specific command */
4524cf49a43SJulian Elischer 		if ((cmd = FindCommand(av[1])) != NULL) {
4534cf49a43SJulian Elischer 			printf("Usage:    %s\n", cmd->cmd);
454f8307e12SArchie Cobbs 			if (cmd->aliases[0] != NULL) {
455f8307e12SArchie Cobbs 				int a = 0;
456f8307e12SArchie Cobbs 
457f8307e12SArchie Cobbs 				printf("Aliases:  ");
458f8307e12SArchie Cobbs 				while (1) {
459f8307e12SArchie Cobbs 					printf("%s", cmd->aliases[a++]);
460f8307e12SArchie Cobbs 					if (a == MAX_CMD_ALIAS
461f8307e12SArchie Cobbs 					    || cmd->aliases[a] == NULL) {
462f8307e12SArchie Cobbs 						printf("\n");
463f8307e12SArchie Cobbs 						break;
464f8307e12SArchie Cobbs 					}
465f8307e12SArchie Cobbs 					printf(", ");
466f8307e12SArchie Cobbs 				}
467f8307e12SArchie Cobbs 			}
4684cf49a43SJulian Elischer 			printf("Summary:  %s\n", cmd->desc);
4694cf49a43SJulian Elischer 			if (cmd->help != NULL) {
4704cf49a43SJulian Elischer 				const char *s;
4714cf49a43SJulian Elischer 				char buf[65];
4724cf49a43SJulian Elischer 				int tot, len, done;
4734cf49a43SJulian Elischer 
4744cf49a43SJulian Elischer 				printf("Description:\n");
4754cf49a43SJulian Elischer 				for (s = cmd->help; *s != '\0'; s += len) {
4764cf49a43SJulian Elischer 					while (isspace(*s))
4774cf49a43SJulian Elischer 						s++;
4784cf49a43SJulian Elischer 					tot = snprintf(buf,
4794cf49a43SJulian Elischer 					    sizeof(buf), "%s", s);
4804cf49a43SJulian Elischer 					len = strlen(buf);
4814cf49a43SJulian Elischer 					done = len == tot;
4824cf49a43SJulian Elischer 					if (!done) {
4834cf49a43SJulian Elischer 						while (len > 0
4844cf49a43SJulian Elischer 						    && !isspace(buf[len-1]))
4854cf49a43SJulian Elischer 							buf[--len] = '\0';
4864cf49a43SJulian Elischer 					}
4874cf49a43SJulian Elischer 					printf("  %s\n", buf);
4884cf49a43SJulian Elischer 				}
4894cf49a43SJulian Elischer 			}
4904cf49a43SJulian Elischer 		}
4914cf49a43SJulian Elischer 	}
4924cf49a43SJulian Elischer 	return(CMDRTN_OK);
4934cf49a43SJulian Elischer }
4944cf49a43SJulian Elischer 
4954cf49a43SJulian Elischer /*
4964cf49a43SJulian Elischer  * QuitCmd()
4974cf49a43SJulian Elischer  */
4984cf49a43SJulian Elischer static int
4994cf49a43SJulian Elischer QuitCmd(int ac, char **av)
5004cf49a43SJulian Elischer {
5014cf49a43SJulian Elischer 	return(CMDRTN_QUIT);
5024cf49a43SJulian Elischer }
5034cf49a43SJulian Elischer 
5044cf49a43SJulian Elischer /*
505f8307e12SArchie Cobbs  * Dump data in hex and ASCII form
506f8307e12SArchie Cobbs  */
507f8307e12SArchie Cobbs static void
508f8307e12SArchie Cobbs DumpAscii(const u_char *buf, int len)
509f8307e12SArchie Cobbs {
510f8307e12SArchie Cobbs 	char ch, sbuf[100];
511f8307e12SArchie Cobbs 	int k, count;
512f8307e12SArchie Cobbs 
513f8307e12SArchie Cobbs 	for (count = 0; count < len; count += DUMP_BYTES_PER_LINE) {
514f8307e12SArchie Cobbs 		snprintf(sbuf, sizeof(sbuf), "%04x:  ", count);
515f8307e12SArchie Cobbs 		for (k = 0; k < DUMP_BYTES_PER_LINE; k++) {
516f8307e12SArchie Cobbs 			if (count + k < len) {
517f8307e12SArchie Cobbs 				snprintf(sbuf + strlen(sbuf),
518f8307e12SArchie Cobbs 				    sizeof(sbuf) - strlen(sbuf),
519f8307e12SArchie Cobbs 				    "%02x ", buf[count + k]);
520f8307e12SArchie Cobbs 			} else {
521f8307e12SArchie Cobbs 				snprintf(sbuf + strlen(sbuf),
522f8307e12SArchie Cobbs 				    sizeof(sbuf) - strlen(sbuf), "   ");
523f8307e12SArchie Cobbs 			}
524f8307e12SArchie Cobbs 		}
525f8307e12SArchie Cobbs 		snprintf(sbuf + strlen(sbuf), sizeof(sbuf) - strlen(sbuf), " ");
526f8307e12SArchie Cobbs 		for (k = 0; k < DUMP_BYTES_PER_LINE; k++) {
527f8307e12SArchie Cobbs 			if (count + k < len) {
528f8307e12SArchie Cobbs 				ch = isprint(buf[count + k]) ?
529f8307e12SArchie Cobbs 				    buf[count + k] : '.';
530f8307e12SArchie Cobbs 				snprintf(sbuf + strlen(sbuf),
531f8307e12SArchie Cobbs 				    sizeof(sbuf) - strlen(sbuf), "%c", ch);
532f8307e12SArchie Cobbs 			} else {
533f8307e12SArchie Cobbs 				snprintf(sbuf + strlen(sbuf),
534f8307e12SArchie Cobbs 				    sizeof(sbuf) - strlen(sbuf), " ");
535f8307e12SArchie Cobbs 			}
536f8307e12SArchie Cobbs 		}
537f8307e12SArchie Cobbs 		printf("%s\n", sbuf);
538f8307e12SArchie Cobbs 	}
539f8307e12SArchie Cobbs }
540f8307e12SArchie Cobbs 
541f8307e12SArchie Cobbs /*
5424cf49a43SJulian Elischer  * Usage()
5434cf49a43SJulian Elischer  */
5444cf49a43SJulian Elischer static void
5454cf49a43SJulian Elischer Usage(const char *msg)
5464cf49a43SJulian Elischer {
5474cf49a43SJulian Elischer 	if (msg)
5484cf49a43SJulian Elischer 		warnx("%s", msg);
5494cf49a43SJulian Elischer 	errx(EX_USAGE, "usage: ngctl [-d] [-f file] [-n name] [command ...]");
5504cf49a43SJulian Elischer }
5514cf49a43SJulian Elischer 
552