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$ 38 */ 39 40 #include "ngctl.h" 41 42 #define PROMPT "+ " 43 #define MAX_ARGS 512 44 #define WHITESPACE " \t\r\n\v\f" 45 #define NG_SOCKET_KLD "ng_socket.ko" 46 47 /* Internal functions */ 48 static int ReadFile(FILE *fp); 49 static int DoParseCommand(char *line); 50 static int DoCommand(int ac, char **av); 51 static int DoInteractive(void); 52 static const struct ngcmd *FindCommand(const char *string); 53 static void Usage(const char *msg); 54 static int ReadCmd(int ac, char **av); 55 static int HelpCmd(int ac, char **av); 56 static int QuitCmd(int ac, char **av); 57 58 /* List of commands */ 59 static const struct ngcmd *const cmds[] = { 60 &connect_cmd, 61 &debug_cmd, 62 &help_cmd, 63 &list_cmd, 64 &mkpeer_cmd, 65 &name_cmd, 66 &read_cmd, 67 &rmhook_cmd, 68 &show_cmd, 69 &shutdown_cmd, 70 &status_cmd, 71 &types_cmd, 72 &quit_cmd, 73 NULL 74 }; 75 76 /* Commands defined in this file */ 77 const struct ngcmd read_cmd = { 78 ReadCmd, 79 "read <filename>", 80 "Read and execute commands from a file", 81 NULL 82 }; 83 const struct ngcmd help_cmd = { 84 HelpCmd, 85 "help [command]", 86 "Show command summary or get more help on a specific command", 87 NULL 88 }; 89 const struct ngcmd quit_cmd = { 90 QuitCmd, 91 "quit", 92 "Exit program", 93 NULL 94 }; 95 96 /* Our control and data sockets */ 97 int csock, dsock; 98 99 /* 100 * main() 101 */ 102 int 103 main(int ac, char *av[]) 104 { 105 char name[NG_NODELEN + 1]; 106 int interactive = isatty(0) && isatty(1); 107 FILE *fp = NULL; 108 int ch, rtn = 0; 109 110 /* Set default node name */ 111 snprintf(name, sizeof(name), "ngctl%d", getpid()); 112 113 /* Parse command line */ 114 while ((ch = getopt(ac, av, "df:n:")) != EOF) { 115 switch (ch) { 116 case 'd': 117 NgSetDebug(NgSetDebug(-1) + 1); 118 break; 119 case 'f': 120 if (strcmp(optarg, "-") == 0) 121 fp = stdin; 122 else if ((fp = fopen(optarg, "r")) == NULL) 123 err(EX_NOINPUT, "%s", optarg); 124 break; 125 case 'n': 126 snprintf(name, sizeof(name), "%s", optarg); 127 break; 128 case '?': 129 default: 130 Usage((char *)NULL); 131 break; 132 } 133 } 134 ac -= optind; 135 av += optind; 136 137 /* Create a new socket node */ 138 if (NgMkSockNode(name, &csock, &dsock) < 0) { 139 if (errno == EPROTONOSUPPORT) { 140 if (kldload(NG_SOCKET_KLD) < 0) 141 err(EX_OSERR, "can't load %s", NG_SOCKET_KLD); 142 if (NgMkSockNode(name, &csock, &dsock) >= 0) 143 goto gotNode; 144 } 145 err(EX_OSERR, "can't create node"); 146 } 147 gotNode: 148 149 /* Do commands as requested */ 150 if (ac == 0) { 151 if (fp != NULL) { 152 rtn = ReadFile(fp); 153 } else if (interactive) { 154 rtn = DoInteractive(); 155 } else 156 Usage("no command specified"); 157 } else { 158 rtn = DoCommand(ac, av); 159 } 160 161 /* Convert command return code into system exit code */ 162 switch (rtn) { 163 case CMDRTN_OK: 164 case CMDRTN_QUIT: 165 rtn = 0; 166 break; 167 case CMDRTN_USAGE: 168 rtn = EX_USAGE; 169 break; 170 case CMDRTN_ERROR: 171 rtn = EX_OSERR; 172 break; 173 } 174 return(rtn); 175 } 176 177 /* 178 * Process commands from a file 179 */ 180 static int 181 ReadFile(FILE *fp) 182 { 183 char line[LINE_MAX]; 184 int num, rtn; 185 186 for (num = 1; fgets(line, sizeof(line), fp) != NULL; num++) { 187 if (*line == '#') 188 continue; 189 if ((rtn = DoParseCommand(line)) != 0) { 190 warnx("line %d: error in file", num); 191 return(rtn); 192 } 193 } 194 return(CMDRTN_OK); 195 } 196 197 /* 198 * Interactive mode 199 */ 200 static int 201 DoInteractive(void) 202 { 203 char buf[LINE_MAX]; 204 205 /* Read commands from stdin */ 206 (*help_cmd.func)(0, NULL); 207 do { 208 printf("%s", PROMPT); 209 if (fgets(buf, sizeof(buf), stdin) == NULL) 210 break; 211 fflush(stdout); 212 } while (DoParseCommand(buf) != CMDRTN_QUIT); 213 return(CMDRTN_QUIT); 214 } 215 216 /* 217 * Parse a command line and execute the command 218 */ 219 static int 220 DoParseCommand(char *line) 221 { 222 char *av[MAX_ARGS]; 223 int ac; 224 225 /* Parse line */ 226 for (ac = 0, av[0] = strtok(line, WHITESPACE); 227 ac < MAX_ARGS - 1 && av[ac]; 228 av[++ac] = strtok(NULL, WHITESPACE)); 229 230 /* Do command */ 231 return(DoCommand(ac, av)); 232 } 233 234 /* 235 * Execute the command 236 */ 237 static int 238 DoCommand(int ac, char **av) 239 { 240 const struct ngcmd *cmd; 241 int rtn; 242 243 if (ac == 0 || *av[0] == 0) 244 return(CMDRTN_OK); 245 if ((cmd = FindCommand(av[0])) == NULL) 246 return(CMDRTN_ERROR); 247 if ((rtn = (*cmd->func)(ac, av)) == CMDRTN_USAGE) 248 warnx("usage: %s", cmd->cmd); 249 return(rtn); 250 } 251 252 /* 253 * Find a command 254 */ 255 static const struct ngcmd * 256 FindCommand(const char *string) 257 { 258 const struct ngcmd *cmd; 259 int k, len, found; 260 261 if (strcmp(string, "?") == 0) 262 string = "help"; 263 for (k = 0, found = -1; cmds[k]; k++) { 264 cmd = cmds[k]; 265 len = strcspn(cmd->cmd, WHITESPACE); 266 if (len > strlen(string)) 267 len = strlen(string); 268 if (!strncasecmp(string, cmd->cmd, len)) { 269 if (found != -1) { 270 warnx("\"%s\": ambiguous command", string); 271 return(NULL); 272 } 273 found = k; 274 } 275 } 276 if (found == -1) { 277 warnx("\"%s\": unknown command", string); 278 return(NULL); 279 } 280 return(cmds[found]); 281 } 282 283 /* 284 * ReadCmd() 285 */ 286 static int 287 ReadCmd(int ac, char **av) 288 { 289 FILE *fp; 290 int rtn; 291 292 /* Open file */ 293 switch (ac) { 294 case 2: 295 if ((fp = fopen(av[1], "r")) == NULL) 296 warn("%s", av[1]); 297 return(CMDRTN_ERROR); 298 default: 299 return(CMDRTN_USAGE); 300 } 301 302 /* Process it */ 303 rtn = ReadFile(fp); 304 fclose(fp); 305 return(rtn); 306 } 307 308 /* 309 * HelpCmd() 310 */ 311 static int 312 HelpCmd(int ac, char **av) 313 { 314 const struct ngcmd *cmd; 315 int k; 316 317 switch (ac) { 318 case 0: 319 case 1: 320 /* Show all commands */ 321 printf("Available commands:\n"); 322 for (k = 0; cmds[k] != NULL; k++) { 323 char *s, buf[100]; 324 325 cmd = cmds[k]; 326 snprintf(buf, sizeof(buf), "%s", cmd->cmd); 327 for (s = buf; *s != '\0' && !isspace(*s); s++); 328 *s = '\0'; 329 printf(" %-10s %s\n", buf, cmd->desc); 330 } 331 return(CMDRTN_OK); 332 default: 333 /* Show help on a specific command */ 334 if ((cmd = FindCommand(av[1])) != NULL) { 335 printf("Usage: %s\n", cmd->cmd); 336 printf("Summary: %s\n", cmd->desc); 337 if (cmd->help != NULL) { 338 const char *s; 339 char buf[65]; 340 int tot, len, done; 341 342 printf("Description:\n"); 343 for (s = cmd->help; *s != '\0'; s += len) { 344 while (isspace(*s)) 345 s++; 346 tot = snprintf(buf, 347 sizeof(buf), "%s", s); 348 len = strlen(buf); 349 done = len == tot; 350 if (!done) { 351 while (len > 0 352 && !isspace(buf[len-1])) 353 buf[--len] = '\0'; 354 } 355 printf(" %s\n", buf); 356 } 357 } 358 } 359 } 360 return(CMDRTN_OK); 361 } 362 363 /* 364 * QuitCmd() 365 */ 366 static int 367 QuitCmd(int ac, char **av) 368 { 369 return(CMDRTN_QUIT); 370 } 371 372 /* 373 * Usage() 374 */ 375 static void 376 Usage(const char *msg) 377 { 378 if (msg) 379 warnx("%s", msg); 380 errx(EX_USAGE, "usage: ngctl [-d] [-f file] [-n name] [command ...]"); 381 } 382 383