1187767Sluigi/* 2204591Sluigi * Copyright (c) 2002-2003,2010 Luigi Rizzo 3187767Sluigi * Copyright (c) 1996 Alex Nash, Paul Traina, Poul-Henning Kamp 4187767Sluigi * Copyright (c) 1994 Ugen J.S.Antsilevich 5187767Sluigi * 6187767Sluigi * Idea and grammar partially left from: 7187767Sluigi * Copyright (c) 1993 Daniel Boulet 8187767Sluigi * 9187767Sluigi * Redistribution and use in source forms, with and without modification, 10187767Sluigi * are permitted provided that this entire comment appears intact. 11187767Sluigi * 12187767Sluigi * Redistribution in binary form may occur without any restrictions. 13187767Sluigi * Obviously, it would be nice if you gave credit where credit is due 14187767Sluigi * but requiring it would be too onerous. 15187767Sluigi * 16187767Sluigi * This software is provided ``AS IS'' without any warranties of any kind. 17187767Sluigi * 18187767Sluigi * Command line interface for IP firewall facility 19187767Sluigi * 20187767Sluigi * $FreeBSD$ 21187767Sluigi */ 22187767Sluigi 23187767Sluigi#include <sys/wait.h> 24187767Sluigi#include <ctype.h> 25187767Sluigi#include <err.h> 26187767Sluigi#include <errno.h> 27187767Sluigi#include <signal.h> 28187767Sluigi#include <stdio.h> 29187767Sluigi#include <stdlib.h> 30187767Sluigi#include <string.h> 31187767Sluigi#include <sysexits.h> 32187767Sluigi#include <unistd.h> 33187767Sluigi 34187767Sluigi#include "ipfw2.h" 35187767Sluigi 36187767Sluigistatic void 37187767Sluigihelp(void) 38187767Sluigi{ 39187767Sluigi fprintf(stderr, 40187767Sluigi"ipfw syntax summary (but please do read the ipfw(8) manpage):\n\n" 41187767Sluigi"\tipfw [-abcdefhnNqStTv] <command>\n\n" 42187767Sluigi"where <command> is one of the following:\n\n" 43187767Sluigi"add [num] [set N] [prob x] RULE-BODY\n" 44187767Sluigi"{pipe|queue} N config PIPE-BODY\n" 45187767Sluigi"[pipe|queue] {zero|delete|show} [N{,N}]\n" 46187767Sluigi"nat N config {ip IPADDR|if IFNAME|log|deny_in|same_ports|unreg_only|reset|\n" 47187767Sluigi" reverse|proxy_only|redirect_addr linkspec|\n" 48187767Sluigi" redirect_port linkspec|redirect_proto linkspec}\n" 49187767Sluigi"set [disable N... enable N...] | move [rule] X to Y | swap X Y | show\n" 50187767Sluigi"set N {show|list|zero|resetlog|delete} [N{,N}] | flush\n" 51187767Sluigi"table N {add ip[/bits] [value] | delete ip[/bits] | flush | list}\n" 52187767Sluigi"table all {flush | list}\n" 53187767Sluigi"\n" 54187767Sluigi"RULE-BODY: check-state [PARAMS] | ACTION [PARAMS] ADDR [OPTION_LIST]\n" 55187767Sluigi"ACTION: check-state | allow | count | deny | unreach{,6} CODE |\n" 56187767Sluigi" skipto N | {divert|tee} PORT | forward ADDR |\n" 57190633Spiso" pipe N | queue N | nat N | setfib FIB | reass\n" 58187767Sluigi"PARAMS: [log [logamount LOGLIMIT]] [altq QUEUE_NAME]\n" 59187767Sluigi"ADDR: [ MAC dst src ether_type ] \n" 60187767Sluigi" [ ip from IPADDR [ PORT ] to IPADDR [ PORTLIST ] ]\n" 61187767Sluigi" [ ipv6|ip6 from IP6ADDR [ PORT ] to IP6ADDR [ PORTLIST ] ]\n" 62187767Sluigi"IPADDR: [not] { any | me | ip/bits{x,y,z} | table(t[,v]) | IPLIST }\n" 63187767Sluigi"IP6ADDR: [not] { any | me | me6 | ip6/bits | IP6LIST }\n" 64187767Sluigi"IP6LIST: { ip6 | ip6/bits }[,IP6LIST]\n" 65187767Sluigi"IPLIST: { ip | ip/bits | ip:mask }[,IPLIST]\n" 66187767Sluigi"OPTION_LIST: OPTION [OPTION_LIST]\n" 67187767Sluigi"OPTION: bridged | diverted | diverted-loopback | diverted-output |\n" 68187767Sluigi" {dst-ip|src-ip} IPADDR | {dst-ip6|src-ip6|dst-ipv6|src-ipv6} IP6ADDR |\n" 69187767Sluigi" {dst-port|src-port} LIST |\n" 70187767Sluigi" estab | frag | {gid|uid} N | icmptypes LIST | in | out | ipid LIST |\n" 71187767Sluigi" iplen LIST | ipoptions SPEC | ipprecedence | ipsec | iptos SPEC |\n" 72187767Sluigi" ipttl LIST | ipversion VER | keep-state | layer2 | limit ... |\n" 73187767Sluigi" icmp6types LIST | ext6hdr LIST | flow-id N[,N] | fib FIB |\n" 74187767Sluigi" mac ... | mac-type LIST | proto LIST | {recv|xmit|via} {IF|IPADDR} |\n" 75187767Sluigi" setup | {tcpack|tcpseq|tcpwin} NN | tcpflags SPEC | tcpoptions SPEC |\n" 76187767Sluigi" tcpdatalen LIST | verrevpath | versrcreach | antispoof\n" 77187767Sluigi); 78187767Sluigi 79187767Sluigi exit(0); 80187767Sluigi} 81187767Sluigi 82187767Sluigi/* 83187767Sluigi * Called with the arguments, including program name because getopt 84187767Sluigi * wants it to be present. 85187767Sluigi * Returns 0 if successful, 1 if empty command, errx() in case of errors. 86204591Sluigi * First thing we do is process parameters creating an argv[] array 87204591Sluigi * which includes the program name and a NULL entry at the end. 88204591Sluigi * If we are called with a single string, we split it on whitespace. 89204591Sluigi * Also, arguments with a trailing ',' are joined to the next one. 90227489Seadler * The pointers (av[]) and data are in a single chunk of memory. 91204591Sluigi * av[0] points to the original program name, all other entries 92204591Sluigi * point into the allocated chunk. 93187767Sluigi */ 94187767Sluigistatic int 95187767Sluigiipfw_main(int oldac, char **oldav) 96187767Sluigi{ 97204591Sluigi int ch, ac; 98187767Sluigi const char *errstr; 99187767Sluigi char **av, **save_av; 100187767Sluigi int do_acct = 0; /* Show packet/byte count */ 101189396Sluigi int try_next = 0; /* set if pipe cmd not found */ 102204591Sluigi int av_size; /* compute the av size */ 103204591Sluigi char *av_p; /* used to build the av list */ 104187767Sluigi 105187767Sluigi#define WHITESP " \t\f\v\n\r" 106187767Sluigi if (oldac < 2) 107187767Sluigi return 1; /* need at least one argument */ 108187767Sluigi 109187767Sluigi if (oldac == 2) { 110187767Sluigi /* 111204591Sluigi * If we are called with one argument, try to split it into 112204591Sluigi * words for subsequent parsing. Spaces after a ',' are 113204591Sluigi * removed by copying the string in-place. 114187767Sluigi */ 115187767Sluigi char *arg = oldav[1]; /* The string is the first arg. */ 116187767Sluigi int l = strlen(arg); 117187767Sluigi int copy = 0; /* 1 if we need to copy, 0 otherwise */ 118187767Sluigi int i, j; 119187767Sluigi 120187767Sluigi for (i = j = 0; i < l; i++) { 121187767Sluigi if (arg[i] == '#') /* comment marker */ 122187767Sluigi break; 123187767Sluigi if (copy) { 124187767Sluigi arg[j++] = arg[i]; 125229403Sed copy = !strchr("," WHITESP, arg[i]); 126187767Sluigi } else { 127229403Sed copy = !strchr(WHITESP, arg[i]); 128187767Sluigi if (copy) 129187767Sluigi arg[j++] = arg[i]; 130187767Sluigi } 131187767Sluigi } 132187767Sluigi if (!copy && j > 0) /* last char was a 'blank', remove it */ 133187767Sluigi j--; 134187767Sluigi l = j; /* the new argument length */ 135187767Sluigi arg[j++] = '\0'; 136187767Sluigi if (l == 0) /* empty string! */ 137187767Sluigi return 1; 138187767Sluigi 139187767Sluigi /* 140187767Sluigi * First, count number of arguments. Because of the previous 141187767Sluigi * processing, this is just the number of blanks plus 1. 142187767Sluigi */ 143187767Sluigi for (i = 0, ac = 1; i < l; i++) 144229403Sed if (strchr(WHITESP, arg[i]) != NULL) 145187767Sluigi ac++; 146187767Sluigi 147187767Sluigi /* 148204591Sluigi * Allocate the argument list structure as a single block 149204591Sluigi * of memory, containing pointers and the argument 150204591Sluigi * strings. We include one entry for the program name 151204591Sluigi * because getopt expects it, and a NULL at the end 152204591Sluigi * to simplify further parsing. 153187767Sluigi */ 154204591Sluigi ac++; /* add 1 for the program name */ 155204591Sluigi av_size = (ac+1) * sizeof(char *) + l + 1; 156204591Sluigi av = safe_calloc(av_size, 1); 157187767Sluigi 158187767Sluigi /* 159204591Sluigi * Init the argument pointer to the end of the array 160204591Sluigi * and copy arguments from arg[] to av[]. For each one, 161187767Sluigi * j is the initial character, i is the one past the end. 162187767Sluigi */ 163204591Sluigi av_p = (char *)&av[ac+1]; 164204591Sluigi for (ac = 1, i = j = 0; i < l; i++) { 165229403Sed if (strchr(WHITESP, arg[i]) != NULL || i == l-1) { 166187767Sluigi if (i == l-1) 167187767Sluigi i++; 168204591Sluigi bcopy(arg+j, av_p, i-j); 169204591Sluigi av[ac] = av_p; 170229778Suqs av_p += i-j; /* the length of the string */ 171204591Sluigi *av_p++ = '\0'; 172187767Sluigi ac++; 173187767Sluigi j = i + 1; 174187767Sluigi } 175204591Sluigi } 176187767Sluigi } else { 177187767Sluigi /* 178187767Sluigi * If an argument ends with ',' join with the next one. 179187767Sluigi */ 180204591Sluigi int first, i, l=0; 181187767Sluigi 182204591Sluigi /* 183204591Sluigi * Allocate the argument list structure as a single block 184204591Sluigi * of memory, containing both pointers and the argument 185204591Sluigi * strings. We include some space for the program name 186204591Sluigi * because getopt expects it. 187204591Sluigi * We add an extra pointer to the end of the array, 188204591Sluigi * to make simpler further parsing. 189204591Sluigi */ 190204591Sluigi for (i=0; i<oldac; i++) 191204591Sluigi l += strlen(oldav[i]); 192204591Sluigi 193204591Sluigi av_size = (oldac+1) * sizeof(char *) + l + oldac; 194204591Sluigi av = safe_calloc(av_size, 1); 195204591Sluigi 196204591Sluigi /* 197204591Sluigi * Init the argument pointer to the end of the array 198204591Sluigi * and copy arguments from arg[] to av[] 199204591Sluigi */ 200204591Sluigi av_p = (char *)&av[oldac+1]; 201187767Sluigi for (first = i = ac = 1, l = 0; i < oldac; i++) { 202187767Sluigi char *arg = oldav[i]; 203187767Sluigi int k = strlen(arg); 204187767Sluigi 205187767Sluigi l += k; 206187767Sluigi if (arg[k-1] != ',' || i == oldac-1) { 207187767Sluigi /* Time to copy. */ 208204591Sluigi av[ac] = av_p; 209187767Sluigi for (l=0; first <= i; first++) { 210204591Sluigi strcat(av_p, oldav[first]); 211204591Sluigi av_p += strlen(oldav[first]); 212187767Sluigi } 213204591Sluigi *av_p++ = '\0'; 214187767Sluigi ac++; 215187767Sluigi l = 0; 216187767Sluigi first = i+1; 217187767Sluigi } 218187767Sluigi } 219187767Sluigi } 220187767Sluigi 221204591Sluigi /* 222204591Sluigi * set the progname pointer to the original string 223204591Sluigi * and terminate the array with null 224204591Sluigi */ 225204591Sluigi av[0] = oldav[0]; 226204591Sluigi av[ac] = NULL; 227204591Sluigi 228187767Sluigi /* Set the force flag for non-interactive processes */ 229187767Sluigi if (!co.do_force) 230187767Sluigi co.do_force = !isatty(STDIN_FILENO); 231187767Sluigi 232204591Sluigi#ifdef EMULATE_SYSCTL /* sysctl emulation */ 233204591Sluigi if ( ac >= 2 && !strcmp(av[1], "sysctl")) { 234204591Sluigi char *s; 235204591Sluigi int i; 236204591Sluigi 237204591Sluigi if (ac != 3) { 238204591Sluigi printf( "sysctl emulation usage:\n" 239204591Sluigi " ipfw sysctl name[=value]\n" 240204591Sluigi " ipfw sysctl -a\n"); 241204591Sluigi return 0; 242204591Sluigi } 243229403Sed s = strchr(av[2], '='); 244204591Sluigi if (s == NULL) { 245204591Sluigi s = !strcmp(av[2], "-a") ? NULL : av[2]; 246204591Sluigi sysctlbyname(s, NULL, NULL, NULL, 0); 247204591Sluigi } else { /* ipfw sysctl x.y.z=value */ 248204591Sluigi /* assume an INT value, will extend later */ 249204591Sluigi if (s[1] == '\0') { 250204591Sluigi printf("ipfw sysctl: missing value\n\n"); 251204591Sluigi return 0; 252204591Sluigi } 253204591Sluigi *s = '\0'; 254204591Sluigi i = strtol(s+1, NULL, 0); 255204591Sluigi sysctlbyname(av[2], NULL, NULL, &i, sizeof(int)); 256204591Sluigi } 257204591Sluigi return 0; 258204591Sluigi } 259204591Sluigi#endif 260204591Sluigi 261187767Sluigi /* Save arguments for final freeing of memory. */ 262187767Sluigi save_av = av; 263187767Sluigi 264187767Sluigi optind = optreset = 1; /* restart getopt() */ 265223661Sae while ((ch = getopt(ac, av, "abcdefhinNp:qs:STtv")) != -1) 266187767Sluigi switch (ch) { 267187767Sluigi case 'a': 268187767Sluigi do_acct = 1; 269187767Sluigi break; 270187767Sluigi 271187767Sluigi case 'b': 272187767Sluigi co.comment_only = 1; 273187767Sluigi co.do_compact = 1; 274187767Sluigi break; 275187767Sluigi 276187767Sluigi case 'c': 277187767Sluigi co.do_compact = 1; 278187767Sluigi break; 279187767Sluigi 280187767Sluigi case 'd': 281187767Sluigi co.do_dynamic = 1; 282187767Sluigi break; 283187767Sluigi 284187767Sluigi case 'e': 285187767Sluigi co.do_expired = 1; 286187767Sluigi break; 287187767Sluigi 288187767Sluigi case 'f': 289187767Sluigi co.do_force = 1; 290187767Sluigi break; 291187767Sluigi 292187767Sluigi case 'h': /* help */ 293204591Sluigi free(save_av); 294187767Sluigi help(); 295187767Sluigi break; /* NOTREACHED */ 296187767Sluigi 297187767Sluigi case 'i': 298187767Sluigi co.do_value_as_ip = 1; 299187767Sluigi break; 300187767Sluigi 301187767Sluigi case 'n': 302187767Sluigi co.test_only = 1; 303187767Sluigi break; 304187767Sluigi 305187767Sluigi case 'N': 306187767Sluigi co.do_resolv = 1; 307187767Sluigi break; 308187767Sluigi 309223661Sae case 'p': 310223661Sae errx(EX_USAGE, "An absolute pathname must be used " 311223661Sae "with -p option."); 312228871Seadler /* NOTREACHED */ 313223661Sae 314187767Sluigi case 'q': 315187767Sluigi co.do_quiet = 1; 316187767Sluigi break; 317187767Sluigi 318187767Sluigi case 's': /* sort */ 319187767Sluigi co.do_sort = atoi(optarg); 320187767Sluigi break; 321187767Sluigi 322187767Sluigi case 'S': 323187767Sluigi co.show_sets = 1; 324187767Sluigi break; 325187767Sluigi 326187767Sluigi case 't': 327187767Sluigi co.do_time = 1; 328187767Sluigi break; 329187767Sluigi 330187767Sluigi case 'T': 331187767Sluigi co.do_time = 2; /* numeric timestamp */ 332187767Sluigi break; 333187767Sluigi 334187767Sluigi case 'v': /* verbose */ 335187767Sluigi co.verbose = 1; 336187767Sluigi break; 337187767Sluigi 338187767Sluigi default: 339204591Sluigi free(save_av); 340187767Sluigi return 1; 341187767Sluigi } 342187767Sluigi 343187767Sluigi ac -= optind; 344187767Sluigi av += optind; 345187767Sluigi NEED1("bad arguments, for usage summary ``ipfw''"); 346187767Sluigi 347187767Sluigi /* 348187767Sluigi * An undocumented behaviour of ipfw1 was to allow rule numbers first, 349187767Sluigi * e.g. "100 add allow ..." instead of "add 100 allow ...". 350187767Sluigi * In case, swap first and second argument to get the normal form. 351187767Sluigi */ 352187767Sluigi if (ac > 1 && isdigit(*av[0])) { 353187767Sluigi char *p = av[0]; 354187767Sluigi 355187767Sluigi av[0] = av[1]; 356187767Sluigi av[1] = p; 357187767Sluigi } 358187767Sluigi 359187767Sluigi /* 360187767Sluigi * Optional: pipe, queue or nat. 361187767Sluigi */ 362187767Sluigi co.do_nat = 0; 363187767Sluigi co.do_pipe = 0; 364222745Sae co.use_set = 0; 365187767Sluigi if (!strncmp(*av, "nat", strlen(*av))) 366220802Sglebius co.do_nat = 1; 367187767Sluigi else if (!strncmp(*av, "pipe", strlen(*av))) 368187767Sluigi co.do_pipe = 1; 369187767Sluigi else if (_substrcmp(*av, "queue") == 0) 370187767Sluigi co.do_pipe = 2; 371204591Sluigi else if (_substrcmp(*av, "flowset") == 0) 372204591Sluigi co.do_pipe = 2; 373204591Sluigi else if (_substrcmp(*av, "sched") == 0) 374204591Sluigi co.do_pipe = 3; 375187767Sluigi else if (!strncmp(*av, "set", strlen(*av))) { 376187767Sluigi if (ac > 1 && isdigit(av[1][0])) { 377187767Sluigi co.use_set = strtonum(av[1], 0, resvd_set_number, 378187767Sluigi &errstr); 379187767Sluigi if (errstr) 380187767Sluigi errx(EX_DATAERR, 381187767Sluigi "invalid set number %s\n", av[1]); 382187767Sluigi ac -= 2; av += 2; co.use_set++; 383187767Sluigi } 384187767Sluigi } 385187767Sluigi 386187767Sluigi if (co.do_pipe || co.do_nat) { 387187767Sluigi ac--; 388187767Sluigi av++; 389187767Sluigi } 390187767Sluigi NEED1("missing command"); 391187767Sluigi 392187767Sluigi /* 393187767Sluigi * For pipes, queues and nats we normally say 'nat|pipe NN config' 394187767Sluigi * but the code is easier to parse as 'nat|pipe config NN' 395187767Sluigi * so we swap the two arguments. 396187767Sluigi */ 397187767Sluigi if ((co.do_pipe || co.do_nat) && ac > 1 && isdigit(*av[0])) { 398187767Sluigi char *p = av[0]; 399187767Sluigi 400187767Sluigi av[0] = av[1]; 401187767Sluigi av[1] = p; 402187767Sluigi } 403187767Sluigi 404187767Sluigi if (co.use_set == 0) { 405187767Sluigi if (_substrcmp(*av, "add") == 0) 406204591Sluigi ipfw_add(av); 407187767Sluigi else if (co.do_nat && _substrcmp(*av, "show") == 0) 408187767Sluigi ipfw_show_nat(ac, av); 409187767Sluigi else if (co.do_pipe && _substrcmp(*av, "config") == 0) 410187767Sluigi ipfw_config_pipe(ac, av); 411187767Sluigi else if (co.do_nat && _substrcmp(*av, "config") == 0) 412187767Sluigi ipfw_config_nat(ac, av); 413187767Sluigi else if (_substrcmp(*av, "set") == 0) 414204591Sluigi ipfw_sets_handler(av); 415187767Sluigi else if (_substrcmp(*av, "table") == 0) 416187767Sluigi ipfw_table_handler(ac, av); 417187767Sluigi else if (_substrcmp(*av, "enable") == 0) 418204591Sluigi ipfw_sysctl_handler(av, 1); 419187767Sluigi else if (_substrcmp(*av, "disable") == 0) 420204591Sluigi ipfw_sysctl_handler(av, 0); 421187767Sluigi else 422187767Sluigi try_next = 1; 423187767Sluigi } 424187767Sluigi 425187767Sluigi if (co.use_set || try_next) { 426187767Sluigi if (_substrcmp(*av, "delete") == 0) 427204591Sluigi ipfw_delete(av); 428187767Sluigi else if (_substrcmp(*av, "flush") == 0) 429187767Sluigi ipfw_flush(co.do_force); 430187767Sluigi else if (_substrcmp(*av, "zero") == 0) 431187767Sluigi ipfw_zero(ac, av, 0 /* IP_FW_ZERO */); 432187767Sluigi else if (_substrcmp(*av, "resetlog") == 0) 433187767Sluigi ipfw_zero(ac, av, 1 /* IP_FW_RESETLOG */); 434187767Sluigi else if (_substrcmp(*av, "print") == 0 || 435220802Sglebius _substrcmp(*av, "list") == 0) 436187767Sluigi ipfw_list(ac, av, do_acct); 437187767Sluigi else if (_substrcmp(*av, "show") == 0) 438187767Sluigi ipfw_list(ac, av, 1 /* show counters */); 439187767Sluigi else 440187767Sluigi errx(EX_USAGE, "bad command `%s'", *av); 441187767Sluigi } 442187767Sluigi 443187767Sluigi /* Free memory allocated in the argument parsing. */ 444204591Sluigi free(save_av); 445187767Sluigi return 0; 446187767Sluigi} 447187767Sluigi 448187767Sluigi 449187767Sluigistatic void 450187767Sluigiipfw_readfile(int ac, char *av[]) 451187767Sluigi{ 452187767Sluigi#define MAX_ARGS 32 453222744Sae char buf[4096]; 454187767Sluigi char *progname = av[0]; /* original program name */ 455187767Sluigi const char *cmd = NULL; /* preprocessor name, if any */ 456187767Sluigi const char *filename = av[ac-1]; /* file to read */ 457187767Sluigi int c, lineno=0; 458187767Sluigi FILE *f = NULL; 459187767Sluigi pid_t preproc = 0; 460187767Sluigi 461187767Sluigi while ((c = getopt(ac, av, "cfNnp:qS")) != -1) { 462187767Sluigi switch(c) { 463187767Sluigi case 'c': 464187767Sluigi co.do_compact = 1; 465187767Sluigi break; 466187767Sluigi 467187767Sluigi case 'f': 468187767Sluigi co.do_force = 1; 469187767Sluigi break; 470187767Sluigi 471187767Sluigi case 'N': 472187767Sluigi co.do_resolv = 1; 473187767Sluigi break; 474187767Sluigi 475187767Sluigi case 'n': 476187767Sluigi co.test_only = 1; 477187767Sluigi break; 478187767Sluigi 479187767Sluigi case 'p': 480187767Sluigi /* 481187767Sluigi * ipfw -p cmd [args] filename 482187767Sluigi * 483187767Sluigi * We are done with getopt(). All arguments 484187767Sluigi * except the filename go to the preprocessor, 485187767Sluigi * so we need to do the following: 486187767Sluigi * - check that a filename is actually present; 487187767Sluigi * - advance av by optind-1 to skip arguments 488187767Sluigi * already processed; 489187767Sluigi * - decrease ac by optind, to remove the args 490187767Sluigi * already processed and the final filename; 491187767Sluigi * - set the last entry in av[] to NULL so 492187767Sluigi * popen() can detect the end of the array; 493187767Sluigi * - set optind=ac to let getopt() terminate. 494187767Sluigi */ 495187767Sluigi if (optind == ac) 496187767Sluigi errx(EX_USAGE, "no filename argument"); 497187767Sluigi cmd = optarg; 498187767Sluigi av[ac-1] = NULL; 499187767Sluigi av += optind - 1; 500187767Sluigi ac -= optind; 501187767Sluigi optind = ac; 502187767Sluigi break; 503187767Sluigi 504187767Sluigi case 'q': 505187767Sluigi co.do_quiet = 1; 506187767Sluigi break; 507187767Sluigi 508187767Sluigi case 'S': 509187767Sluigi co.show_sets = 1; 510187767Sluigi break; 511187767Sluigi 512187767Sluigi default: 513187767Sluigi errx(EX_USAGE, "bad arguments, for usage" 514187767Sluigi " summary ``ipfw''"); 515187767Sluigi } 516187767Sluigi 517187767Sluigi } 518187767Sluigi 519187767Sluigi if (cmd == NULL && ac != optind + 1) 520187767Sluigi errx(EX_USAGE, "extraneous filename arguments %s", av[ac-1]); 521187767Sluigi 522187767Sluigi if ((f = fopen(filename, "r")) == NULL) 523187767Sluigi err(EX_UNAVAILABLE, "fopen: %s", filename); 524187767Sluigi 525187767Sluigi if (cmd != NULL) { /* pipe through preprocessor */ 526187767Sluigi int pipedes[2]; 527187767Sluigi 528187767Sluigi if (pipe(pipedes) == -1) 529187767Sluigi err(EX_OSERR, "cannot create pipe"); 530187767Sluigi 531187767Sluigi preproc = fork(); 532187767Sluigi if (preproc == -1) 533187767Sluigi err(EX_OSERR, "cannot fork"); 534187767Sluigi 535187767Sluigi if (preproc == 0) { 536187767Sluigi /* 537187767Sluigi * Child, will run the preprocessor with the 538187767Sluigi * file on stdin and the pipe on stdout. 539187767Sluigi */ 540187767Sluigi if (dup2(fileno(f), 0) == -1 541187767Sluigi || dup2(pipedes[1], 1) == -1) 542187767Sluigi err(EX_OSERR, "dup2()"); 543187767Sluigi fclose(f); 544187767Sluigi close(pipedes[1]); 545187767Sluigi close(pipedes[0]); 546187767Sluigi execvp(cmd, av); 547187767Sluigi err(EX_OSERR, "execvp(%s) failed", cmd); 548187767Sluigi } else { /* parent, will reopen f as the pipe */ 549187767Sluigi fclose(f); 550187767Sluigi close(pipedes[1]); 551187767Sluigi if ((f = fdopen(pipedes[0], "r")) == NULL) { 552187767Sluigi int savederrno = errno; 553187767Sluigi 554187767Sluigi (void)kill(preproc, SIGTERM); 555187767Sluigi errno = savederrno; 556187767Sluigi err(EX_OSERR, "fdopen()"); 557187767Sluigi } 558187767Sluigi } 559187767Sluigi } 560187767Sluigi 561222744Sae while (fgets(buf, sizeof(buf), f)) { /* read commands */ 562206494Sluigi char linename[20]; 563187767Sluigi char *args[2]; 564187767Sluigi 565187767Sluigi lineno++; 566206494Sluigi snprintf(linename, sizeof(linename), "Line %d", lineno); 567187767Sluigi setprogname(linename); /* XXX */ 568187767Sluigi args[0] = progname; 569187767Sluigi args[1] = buf; 570187767Sluigi ipfw_main(2, args); 571187767Sluigi } 572187767Sluigi fclose(f); 573187767Sluigi if (cmd != NULL) { 574187767Sluigi int status; 575187767Sluigi 576187767Sluigi if (waitpid(preproc, &status, 0) == -1) 577187767Sluigi errx(EX_OSERR, "waitpid()"); 578187767Sluigi if (WIFEXITED(status) && WEXITSTATUS(status) != EX_OK) 579187767Sluigi errx(EX_UNAVAILABLE, 580187767Sluigi "preprocessor exited with status %d", 581187767Sluigi WEXITSTATUS(status)); 582187767Sluigi else if (WIFSIGNALED(status)) 583187767Sluigi errx(EX_UNAVAILABLE, 584187767Sluigi "preprocessor exited with signal %d", 585187767Sluigi WTERMSIG(status)); 586187767Sluigi } 587187767Sluigi} 588187767Sluigi 589187767Sluigiint 590187767Sluigimain(int ac, char *av[]) 591187767Sluigi{ 592204716Sluigi#if defined(_WIN32) && defined(TCC) 593204716Sluigi { 594204716Sluigi WSADATA wsaData; 595204716Sluigi int ret=0; 596204716Sluigi unsigned short wVersionRequested = MAKEWORD(2, 2); 597204716Sluigi ret = WSAStartup(wVersionRequested, &wsaData); 598204716Sluigi if (ret != 0) { 599204716Sluigi /* Tell the user that we could not find a usable */ 600220802Sglebius /* Winsock DLL. */ 601204716Sluigi printf("WSAStartup failed with error: %d\n", ret); 602204716Sluigi return 1; 603204716Sluigi } 604204716Sluigi } 605204716Sluigi#endif 606187767Sluigi /* 607187767Sluigi * If the last argument is an absolute pathname, interpret it 608187767Sluigi * as a file to be preprocessed. 609187767Sluigi */ 610187767Sluigi 611223661Sae if (ac > 1 && av[ac - 1][0] == '/') { 612223661Sae if (access(av[ac - 1], R_OK) == 0) 613223661Sae ipfw_readfile(ac, av); 614223661Sae else 615223661Sae err(EX_USAGE, "pathname: %s", av[ac - 1]); 616223661Sae } else { 617187767Sluigi if (ipfw_main(ac, av)) { 618187767Sluigi errx(EX_USAGE, 619187767Sluigi "usage: ipfw [options]\n" 620187767Sluigi "do \"ipfw -h\" or \"man ipfw\" for details"); 621187767Sluigi } 622187767Sluigi } 623187767Sluigi return EX_OK; 624187767Sluigi} 625