/* File autogenerated by gengetopt version 2.22.5 generated with the following command: gengetopt -C --no-help --no-version -i zopt.ggo -F zopt The developers of gengetopt consider the fixed text that goes in all gengetopt output files to be in the public domain: we make no copyright claims on it. */ /* If we use autoconf. */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #include #include #include #ifndef FIX_UNUSED #define FIX_UNUSED(X) (void) (X) /* avoid warnings for unused params */ #endif #include #include "zopt.h" const char *gengetopt_args_info_purpose = "A fast Internet-wide scanner."; const char *gengetopt_args_info_usage = "Usage: zmap [OPTIONS]..."; const char *gengetopt_args_info_description = ""; const char *gengetopt_args_info_help[] = { "Basic arguments:", " -p, --target-port=port TCP port number to scan (for SYN scans)", " -o, --output-file=name Output file", " -b, --blacklist-file=path File of subnets to exclude, in CIDR notation, \n e.g. 192.168.0.0/16", " -w, --whitelist-file=path File of subnets to constrain scan to, in CIDR \n notation, e.g. 192.168.0.0/16", "\nScan options:", " -n, --max-targets=n Cap number of targets to probe (as a number or \n a percentage of the address space)", " -N, --max-results=n Cap number of results to return", " -t, --max-runtime=ses Cap length of time for sending packets", " -r, --rate=pps Set send rate in packets/sec", " -B, --bandwidth=bps Set send rate in bits/second (supports suffixes \n G, M and K)", " -c, --cooldown-time=secs How long to continue receiving after sending \n last probe (default=`8')", " -e, --seed=n Seed used to select address permutation", " -T, --sender-threads=n Threads used to send packets (default=`1')", " -P, --probes=n Number of probes to send to each IP \n (default=`1')", " -d, --dryrun Don't actually send packets", "\nNetwork options:", " -s, --source-port=port|range Source port(s) for scan packets", " -S, --source-ip=ip|range Source address(es) for scan packets", " -G, --gateway-mac=addr Specify gateway MAC address", " -i, --interface=name Specify network interface to use", "\nAdvanced options:", " -M, --probe-module=name Select probe module (default=`tcp_synscan')", " -O, --output-module=name Select output module (default=`simple_file')", " --probe-args=args Arguments to pass to probe module", " --output-args=args Arguments to pass to output module", " --list-output-modules List available output modules", " --list-probe-modules List available probe modules", "\nAdditional options:", " -C, --config=filename Read a configuration file, which can specify \n any of these options \n (default=`/etc/zmap/zmap.conf')", " -q, --quiet Do not print status updates", " -g, --summary Print configuration and summary at end of scan", " -v, --verbosity=n Level of log detail (0-5) (default=`3')", " -h, --help Print help and exit", " -V, --version Print version and exit", "\nExamples:\n zmap -p 443 (scans the whole Internet for hosts with port 443 open)\n zmap -N 5 -B 10M -p 80 -o - (find 5 HTTP servers, scanning at 10 Mb/s)", 0 }; typedef enum {ARG_NO , ARG_STRING , ARG_INT } cmdline_parser_arg_type; static void clear_given (struct gengetopt_args_info *args_info); static void clear_args (struct gengetopt_args_info *args_info); static int cmdline_parser_internal (int argc, char **argv, struct gengetopt_args_info *args_info, struct cmdline_parser_params *params, const char *additional_error); struct line_list { char * string_arg; struct line_list * next; }; static struct line_list *cmd_line_list = 0; static struct line_list *cmd_line_list_tmp = 0; static void free_cmd_list(void) { /* free the list of a previous call */ if (cmd_line_list) { while (cmd_line_list) { cmd_line_list_tmp = cmd_line_list; cmd_line_list = cmd_line_list->next; free (cmd_line_list_tmp->string_arg); free (cmd_line_list_tmp); } } } static char * gengetopt_strdup (const char *s); static void clear_given (struct gengetopt_args_info *args_info) { args_info->target_port_given = 0 ; args_info->output_file_given = 0 ; args_info->blacklist_file_given = 0 ; args_info->whitelist_file_given = 0 ; args_info->max_targets_given = 0 ; args_info->max_results_given = 0 ; args_info->max_runtime_given = 0 ; args_info->rate_given = 0 ; args_info->bandwidth_given = 0 ; args_info->cooldown_time_given = 0 ; args_info->seed_given = 0 ; args_info->sender_threads_given = 0 ; args_info->probes_given = 0 ; args_info->dryrun_given = 0 ; args_info->source_port_given = 0 ; args_info->source_ip_given = 0 ; args_info->gateway_mac_given = 0 ; args_info->interface_given = 0 ; args_info->probe_module_given = 0 ; args_info->output_module_given = 0 ; args_info->probe_args_given = 0 ; args_info->output_args_given = 0 ; args_info->list_output_modules_given = 0 ; args_info->list_probe_modules_given = 0 ; args_info->config_given = 0 ; args_info->quiet_given = 0 ; args_info->summary_given = 0 ; args_info->verbosity_given = 0 ; args_info->help_given = 0 ; args_info->version_given = 0 ; } static void clear_args (struct gengetopt_args_info *args_info) { FIX_UNUSED (args_info); args_info->target_port_orig = NULL; args_info->output_file_arg = NULL; args_info->output_file_orig = NULL; args_info->blacklist_file_arg = NULL; args_info->blacklist_file_orig = NULL; args_info->whitelist_file_arg = NULL; args_info->whitelist_file_orig = NULL; args_info->max_targets_arg = NULL; args_info->max_targets_orig = NULL; args_info->max_results_orig = NULL; args_info->max_runtime_orig = NULL; args_info->rate_orig = NULL; args_info->bandwidth_arg = NULL; args_info->bandwidth_orig = NULL; args_info->cooldown_time_arg = 8; args_info->cooldown_time_orig = NULL; args_info->seed_orig = NULL; args_info->sender_threads_arg = 1; args_info->sender_threads_orig = NULL; args_info->probes_arg = 1; args_info->probes_orig = NULL; args_info->source_port_arg = NULL; args_info->source_port_orig = NULL; args_info->source_ip_arg = NULL; args_info->source_ip_orig = NULL; args_info->gateway_mac_arg = NULL; args_info->gateway_mac_orig = NULL; args_info->interface_arg = NULL; args_info->interface_orig = NULL; args_info->probe_module_arg = gengetopt_strdup ("tcp_synscan"); args_info->probe_module_orig = NULL; args_info->output_module_arg = gengetopt_strdup ("simple_file"); args_info->output_module_orig = NULL; args_info->probe_args_arg = NULL; args_info->probe_args_orig = NULL; args_info->output_args_arg = NULL; args_info->output_args_orig = NULL; args_info->config_arg = gengetopt_strdup ("/etc/zmap/zmap.conf"); args_info->config_orig = NULL; args_info->verbosity_arg = 3; args_info->verbosity_orig = NULL; } static void init_args_info(struct gengetopt_args_info *args_info) { args_info->target_port_help = gengetopt_args_info_help[1] ; args_info->output_file_help = gengetopt_args_info_help[2] ; args_info->blacklist_file_help = gengetopt_args_info_help[3] ; args_info->whitelist_file_help = gengetopt_args_info_help[4] ; args_info->max_targets_help = gengetopt_args_info_help[6] ; args_info->max_results_help = gengetopt_args_info_help[7] ; args_info->max_runtime_help = gengetopt_args_info_help[8] ; args_info->rate_help = gengetopt_args_info_help[9] ; args_info->bandwidth_help = gengetopt_args_info_help[10] ; args_info->cooldown_time_help = gengetopt_args_info_help[11] ; args_info->seed_help = gengetopt_args_info_help[12] ; args_info->sender_threads_help = gengetopt_args_info_help[13] ; args_info->probes_help = gengetopt_args_info_help[14] ; args_info->dryrun_help = gengetopt_args_info_help[15] ; args_info->source_port_help = gengetopt_args_info_help[17] ; args_info->source_ip_help = gengetopt_args_info_help[18] ; args_info->gateway_mac_help = gengetopt_args_info_help[19] ; args_info->interface_help = gengetopt_args_info_help[20] ; args_info->probe_module_help = gengetopt_args_info_help[22] ; args_info->output_module_help = gengetopt_args_info_help[23] ; args_info->probe_args_help = gengetopt_args_info_help[24] ; args_info->output_args_help = gengetopt_args_info_help[25] ; args_info->list_output_modules_help = gengetopt_args_info_help[26] ; args_info->list_probe_modules_help = gengetopt_args_info_help[27] ; args_info->config_help = gengetopt_args_info_help[29] ; args_info->quiet_help = gengetopt_args_info_help[30] ; args_info->summary_help = gengetopt_args_info_help[31] ; args_info->verbosity_help = gengetopt_args_info_help[32] ; args_info->help_help = gengetopt_args_info_help[33] ; args_info->version_help = gengetopt_args_info_help[34] ; } void cmdline_parser_print_version (void) { printf ("%s %s\n", (strlen(CMDLINE_PARSER_PACKAGE_NAME) ? CMDLINE_PARSER_PACKAGE_NAME : CMDLINE_PARSER_PACKAGE), CMDLINE_PARSER_VERSION); } static void print_help_common(void) { cmdline_parser_print_version (); if (strlen(gengetopt_args_info_purpose) > 0) printf("\n%s\n", gengetopt_args_info_purpose); if (strlen(gengetopt_args_info_usage) > 0) printf("\n%s\n", gengetopt_args_info_usage); printf("\n"); if (strlen(gengetopt_args_info_description) > 0) printf("%s\n\n", gengetopt_args_info_description); } void cmdline_parser_print_help (void) { int i = 0; print_help_common(); while (gengetopt_args_info_help[i]) printf("%s\n", gengetopt_args_info_help[i++]); } void cmdline_parser_init (struct gengetopt_args_info *args_info) { clear_given (args_info); clear_args (args_info); init_args_info (args_info); } void cmdline_parser_params_init(struct cmdline_parser_params *params) { if (params) { params->override = 0; params->initialize = 1; params->check_required = 1; params->check_ambiguity = 0; params->print_errors = 1; } } struct cmdline_parser_params * cmdline_parser_params_create(void) { struct cmdline_parser_params *params = (struct cmdline_parser_params *)malloc(sizeof(struct cmdline_parser_params)); cmdline_parser_params_init(params); return params; } static void free_string_field (char **s) { if (*s) { free (*s); *s = 0; } } static void cmdline_parser_release (struct gengetopt_args_info *args_info) { free_string_field (&(args_info->target_port_orig)); free_string_field (&(args_info->output_file_arg)); free_string_field (&(args_info->output_file_orig)); free_string_field (&(args_info->blacklist_file_arg)); free_string_field (&(args_info->blacklist_file_orig)); free_string_field (&(args_info->whitelist_file_arg)); free_string_field (&(args_info->whitelist_file_orig)); free_string_field (&(args_info->max_targets_arg)); free_string_field (&(args_info->max_targets_orig)); free_string_field (&(args_info->max_results_orig)); free_string_field (&(args_info->max_runtime_orig)); free_string_field (&(args_info->rate_orig)); free_string_field (&(args_info->bandwidth_arg)); free_string_field (&(args_info->bandwidth_orig)); free_string_field (&(args_info->cooldown_time_orig)); free_string_field (&(args_info->seed_orig)); free_string_field (&(args_info->sender_threads_orig)); free_string_field (&(args_info->probes_orig)); free_string_field (&(args_info->source_port_arg)); free_string_field (&(args_info->source_port_orig)); free_string_field (&(args_info->source_ip_arg)); free_string_field (&(args_info->source_ip_orig)); free_string_field (&(args_info->gateway_mac_arg)); free_string_field (&(args_info->gateway_mac_orig)); free_string_field (&(args_info->interface_arg)); free_string_field (&(args_info->interface_orig)); free_string_field (&(args_info->probe_module_arg)); free_string_field (&(args_info->probe_module_orig)); free_string_field (&(args_info->output_module_arg)); free_string_field (&(args_info->output_module_orig)); free_string_field (&(args_info->probe_args_arg)); free_string_field (&(args_info->probe_args_orig)); free_string_field (&(args_info->output_args_arg)); free_string_field (&(args_info->output_args_orig)); free_string_field (&(args_info->config_arg)); free_string_field (&(args_info->config_orig)); free_string_field (&(args_info->verbosity_orig)); clear_given (args_info); } static void write_into_file(FILE *outfile, const char *opt, const char *arg, const char *values[]) { FIX_UNUSED (values); if (arg) { fprintf(outfile, "%s=\"%s\"\n", opt, arg); } else { fprintf(outfile, "%s\n", opt); } } int cmdline_parser_dump(FILE *outfile, struct gengetopt_args_info *args_info) { int i = 0; if (!outfile) { fprintf (stderr, "%s: cannot dump options to stream\n", CMDLINE_PARSER_PACKAGE); return EXIT_FAILURE; } if (args_info->target_port_given) write_into_file(outfile, "target-port", args_info->target_port_orig, 0); if (args_info->output_file_given) write_into_file(outfile, "output-file", args_info->output_file_orig, 0); if (args_info->blacklist_file_given) write_into_file(outfile, "blacklist-file", args_info->blacklist_file_orig, 0); if (args_info->whitelist_file_given) write_into_file(outfile, "whitelist-file", args_info->whitelist_file_orig, 0); if (args_info->max_targets_given) write_into_file(outfile, "max-targets", args_info->max_targets_orig, 0); if (args_info->max_results_given) write_into_file(outfile, "max-results", args_info->max_results_orig, 0); if (args_info->max_runtime_given) write_into_file(outfile, "max-runtime", args_info->max_runtime_orig, 0); if (args_info->rate_given) write_into_file(outfile, "rate", args_info->rate_orig, 0); if (args_info->bandwidth_given) write_into_file(outfile, "bandwidth", args_info->bandwidth_orig, 0); if (args_info->cooldown_time_given) write_into_file(outfile, "cooldown-time", args_info->cooldown_time_orig, 0); if (args_info->seed_given) write_into_file(outfile, "seed", args_info->seed_orig, 0); if (args_info->sender_threads_given) write_into_file(outfile, "sender-threads", args_info->sender_threads_orig, 0); if (args_info->probes_given) write_into_file(outfile, "probes", args_info->probes_orig, 0); if (args_info->dryrun_given) write_into_file(outfile, "dryrun", 0, 0 ); if (args_info->source_port_given) write_into_file(outfile, "source-port", args_info->source_port_orig, 0); if (args_info->source_ip_given) write_into_file(outfile, "source-ip", args_info->source_ip_orig, 0); if (args_info->gateway_mac_given) write_into_file(outfile, "gateway-mac", args_info->gateway_mac_orig, 0); if (args_info->interface_given) write_into_file(outfile, "interface", args_info->interface_orig, 0); if (args_info->probe_module_given) write_into_file(outfile, "probe-module", args_info->probe_module_orig, 0); if (args_info->output_module_given) write_into_file(outfile, "output-module", args_info->output_module_orig, 0); if (args_info->probe_args_given) write_into_file(outfile, "probe-args", args_info->probe_args_orig, 0); if (args_info->output_args_given) write_into_file(outfile, "output-args", args_info->output_args_orig, 0); if (args_info->list_output_modules_given) write_into_file(outfile, "list-output-modules", 0, 0 ); if (args_info->list_probe_modules_given) write_into_file(outfile, "list-probe-modules", 0, 0 ); if (args_info->config_given) write_into_file(outfile, "config", args_info->config_orig, 0); if (args_info->quiet_given) write_into_file(outfile, "quiet", 0, 0 ); if (args_info->summary_given) write_into_file(outfile, "summary", 0, 0 ); if (args_info->verbosity_given) write_into_file(outfile, "verbosity", args_info->verbosity_orig, 0); if (args_info->help_given) write_into_file(outfile, "help", 0, 0 ); if (args_info->version_given) write_into_file(outfile, "version", 0, 0 ); i = EXIT_SUCCESS; return i; } int cmdline_parser_file_save(const char *filename, struct gengetopt_args_info *args_info) { FILE *outfile; int i = 0; outfile = fopen(filename, "w"); if (!outfile) { fprintf (stderr, "%s: cannot open file for writing: %s\n", CMDLINE_PARSER_PACKAGE, filename); return EXIT_FAILURE; } i = cmdline_parser_dump(outfile, args_info); fclose (outfile); return i; } void cmdline_parser_free (struct gengetopt_args_info *args_info) { cmdline_parser_release (args_info); } /** @brief replacement of strdup, which is not standard */ char * gengetopt_strdup (const char *s) { char *result = 0; if (!s) return result; result = (char*)malloc(strlen(s) + 1); if (result == (char*)0) return (char*)0; strcpy(result, s); return result; } int cmdline_parser (int argc, char **argv, struct gengetopt_args_info *args_info) { return cmdline_parser2 (argc, argv, args_info, 0, 1, 1); } int cmdline_parser_ext (int argc, char **argv, struct gengetopt_args_info *args_info, struct cmdline_parser_params *params) { int result; result = cmdline_parser_internal (argc, argv, args_info, params, 0); if (result == EXIT_FAILURE) { cmdline_parser_free (args_info); exit (EXIT_FAILURE); } return result; } int cmdline_parser2 (int argc, char **argv, struct gengetopt_args_info *args_info, int override, int initialize, int check_required) { int result; struct cmdline_parser_params params; params.override = override; params.initialize = initialize; params.check_required = check_required; params.check_ambiguity = 0; params.print_errors = 1; result = cmdline_parser_internal (argc, argv, args_info, ¶ms, 0); if (result == EXIT_FAILURE) { cmdline_parser_free (args_info); exit (EXIT_FAILURE); } return result; } int cmdline_parser_required (struct gengetopt_args_info *args_info, const char *prog_name) { FIX_UNUSED (args_info); FIX_UNUSED (prog_name); return EXIT_SUCCESS; } static char *package_name = 0; /** * @brief updates an option * @param field the generic pointer to the field to update * @param orig_field the pointer to the orig field * @param field_given the pointer to the number of occurrence of this option * @param prev_given the pointer to the number of occurrence already seen * @param value the argument for this option (if null no arg was specified) * @param possible_values the possible values for this option (if specified) * @param default_value the default value (in case the option only accepts fixed values) * @param arg_type the type of this option * @param check_ambiguity @see cmdline_parser_params.check_ambiguity * @param override @see cmdline_parser_params.override * @param no_free whether to free a possible previous value * @param multiple_option whether this is a multiple option * @param long_opt the corresponding long option * @param short_opt the corresponding short option (or '-' if none) * @param additional_error possible further error specification */ static int update_arg(void *field, char **orig_field, unsigned int *field_given, unsigned int *prev_given, char *value, const char *possible_values[], const char *default_value, cmdline_parser_arg_type arg_type, int check_ambiguity, int override, int no_free, int multiple_option, const char *long_opt, char short_opt, const char *additional_error) { char *stop_char = 0; const char *val = value; int found; char **string_field; FIX_UNUSED (field); stop_char = 0; found = 0; if (!multiple_option && prev_given && (*prev_given || (check_ambiguity && *field_given))) { if (short_opt != '-') fprintf (stderr, "%s: `--%s' (`-%c') option given more than once%s\n", package_name, long_opt, short_opt, (additional_error ? additional_error : "")); else fprintf (stderr, "%s: `--%s' option given more than once%s\n", package_name, long_opt, (additional_error ? additional_error : "")); return 1; /* failure */ } FIX_UNUSED (default_value); if (field_given && *field_given && ! override) return 0; if (prev_given) (*prev_given)++; if (field_given) (*field_given)++; if (possible_values) val = possible_values[found]; switch(arg_type) { case ARG_INT: if (val) *((int *)field) = strtol (val, &stop_char, 0); break; case ARG_STRING: if (val) { string_field = (char **)field; if (!no_free && *string_field) free (*string_field); /* free previous string */ *string_field = gengetopt_strdup (val); } break; default: break; }; /* check numeric conversion */ switch(arg_type) { case ARG_INT: if (val && !(stop_char && *stop_char == '\0')) { fprintf(stderr, "%s: invalid numeric value: %s\n", package_name, val); return 1; /* failure */ } break; default: ; }; /* store the original value */ switch(arg_type) { case ARG_NO: break; default: if (value && orig_field) { if (no_free) { *orig_field = value; } else { if (*orig_field) free (*orig_field); /* free previous string */ *orig_field = gengetopt_strdup (value); } } }; return 0; /* OK */ } int cmdline_parser_internal ( int argc, char **argv, struct gengetopt_args_info *args_info, struct cmdline_parser_params *params, const char *additional_error) { int c; /* Character of the parsed option. */ int error = 0; struct gengetopt_args_info local_args_info; int override; int initialize; int check_required; int check_ambiguity; package_name = argv[0]; override = params->override; initialize = params->initialize; check_required = params->check_required; check_ambiguity = params->check_ambiguity; if (initialize) cmdline_parser_init (args_info); cmdline_parser_init (&local_args_info); optarg = 0; optind = 0; opterr = params->print_errors; optopt = '?'; while (1) { int option_index = 0; static struct option long_options[] = { { "target-port", 1, NULL, 'p' }, { "output-file", 1, NULL, 'o' }, { "blacklist-file", 1, NULL, 'b' }, { "whitelist-file", 1, NULL, 'w' }, { "max-targets", 1, NULL, 'n' }, { "max-results", 1, NULL, 'N' }, { "max-runtime", 1, NULL, 't' }, { "rate", 1, NULL, 'r' }, { "bandwidth", 1, NULL, 'B' }, { "cooldown-time", 1, NULL, 'c' }, { "seed", 1, NULL, 'e' }, { "sender-threads", 1, NULL, 'T' }, { "probes", 1, NULL, 'P' }, { "dryrun", 0, NULL, 'd' }, { "source-port", 1, NULL, 's' }, { "source-ip", 1, NULL, 'S' }, { "gateway-mac", 1, NULL, 'G' }, { "interface", 1, NULL, 'i' }, { "probe-module", 1, NULL, 'M' }, { "output-module", 1, NULL, 'O' }, { "probe-args", 1, NULL, 0 }, { "output-args", 1, NULL, 0 }, { "list-output-modules", 0, NULL, 0 }, { "list-probe-modules", 0, NULL, 0 }, { "config", 1, NULL, 'C' }, { "quiet", 0, NULL, 'q' }, { "summary", 0, NULL, 'g' }, { "verbosity", 1, NULL, 'v' }, { "help", 0, NULL, 'h' }, { "version", 0, NULL, 'V' }, { 0, 0, 0, 0 } }; c = getopt_long (argc, argv, "p:o:b:w:n:N:t:r:B:c:e:T:P:ds:S:G:i:M:O:C:qgv:hV", long_options, &option_index); if (c == -1) break; /* Exit from `while (1)' loop. */ switch (c) { case 'p': /* TCP port number to scan (for SYN scans). */ if (update_arg( (void *)&(args_info->target_port_arg), &(args_info->target_port_orig), &(args_info->target_port_given), &(local_args_info.target_port_given), optarg, 0, 0, ARG_INT, check_ambiguity, override, 0, 0, "target-port", 'p', additional_error)) goto failure; break; case 'o': /* Output file. */ if (update_arg( (void *)&(args_info->output_file_arg), &(args_info->output_file_orig), &(args_info->output_file_given), &(local_args_info.output_file_given), optarg, 0, 0, ARG_STRING, check_ambiguity, override, 0, 0, "output-file", 'o', additional_error)) goto failure; break; case 'b': /* File of subnets to exclude, in CIDR notation, e.g. 192.168.0.0/16. */ if (update_arg( (void *)&(args_info->blacklist_file_arg), &(args_info->blacklist_file_orig), &(args_info->blacklist_file_given), &(local_args_info.blacklist_file_given), optarg, 0, 0, ARG_STRING, check_ambiguity, override, 0, 0, "blacklist-file", 'b', additional_error)) goto failure; break; case 'w': /* File of subnets to constrain scan to, in CIDR notation, e.g. 192.168.0.0/16. */ if (update_arg( (void *)&(args_info->whitelist_file_arg), &(args_info->whitelist_file_orig), &(args_info->whitelist_file_given), &(local_args_info.whitelist_file_given), optarg, 0, 0, ARG_STRING, check_ambiguity, override, 0, 0, "whitelist-file", 'w', additional_error)) goto failure; break; case 'n': /* Cap number of targets to probe (as a number or a percentage of the address space). */ if (update_arg( (void *)&(args_info->max_targets_arg), &(args_info->max_targets_orig), &(args_info->max_targets_given), &(local_args_info.max_targets_given), optarg, 0, 0, ARG_STRING, check_ambiguity, override, 0, 0, "max-targets", 'n', additional_error)) goto failure; break; case 'N': /* Cap number of results to return. */ if (update_arg( (void *)&(args_info->max_results_arg), &(args_info->max_results_orig), &(args_info->max_results_given), &(local_args_info.max_results_given), optarg, 0, 0, ARG_INT, check_ambiguity, override, 0, 0, "max-results", 'N', additional_error)) goto failure; break; case 't': /* Cap length of time for sending packets. */ if (update_arg( (void *)&(args_info->max_runtime_arg), &(args_info->max_runtime_orig), &(args_info->max_runtime_given), &(local_args_info.max_runtime_given), optarg, 0, 0, ARG_INT, check_ambiguity, override, 0, 0, "max-runtime", 't', additional_error)) goto failure; break; case 'r': /* Set send rate in packets/sec. */ if (update_arg( (void *)&(args_info->rate_arg), &(args_info->rate_orig), &(args_info->rate_given), &(local_args_info.rate_given), optarg, 0, 0, ARG_INT, check_ambiguity, override, 0, 0, "rate", 'r', additional_error)) goto failure; break; case 'B': /* Set send rate in bits/second (supports suffixes G, M and K). */ if (update_arg( (void *)&(args_info->bandwidth_arg), &(args_info->bandwidth_orig), &(args_info->bandwidth_given), &(local_args_info.bandwidth_given), optarg, 0, 0, ARG_STRING, check_ambiguity, override, 0, 0, "bandwidth", 'B', additional_error)) goto failure; break; case 'c': /* How long to continue receiving after sending last probe. */ if (update_arg( (void *)&(args_info->cooldown_time_arg), &(args_info->cooldown_time_orig), &(args_info->cooldown_time_given), &(local_args_info.cooldown_time_given), optarg, 0, "8", ARG_INT, check_ambiguity, override, 0, 0, "cooldown-time", 'c', additional_error)) goto failure; break; case 'e': /* Seed used to select address permutation. */ if (update_arg( (void *)&(args_info->seed_arg), &(args_info->seed_orig), &(args_info->seed_given), &(local_args_info.seed_given), optarg, 0, 0, ARG_INT, check_ambiguity, override, 0, 0, "seed", 'e', additional_error)) goto failure; break; case 'T': /* Threads used to send packets. */ if (update_arg( (void *)&(args_info->sender_threads_arg), &(args_info->sender_threads_orig), &(args_info->sender_threads_given), &(local_args_info.sender_threads_given), optarg, 0, "1", ARG_INT, check_ambiguity, override, 0, 0, "sender-threads", 'T', additional_error)) goto failure; break; case 'P': /* Number of probes to send to each IP. */ if (update_arg( (void *)&(args_info->probes_arg), &(args_info->probes_orig), &(args_info->probes_given), &(local_args_info.probes_given), optarg, 0, "1", ARG_INT, check_ambiguity, override, 0, 0, "probes", 'P', additional_error)) goto failure; break; case 'd': /* Don't actually send packets. */ if (update_arg( 0 , 0 , &(args_info->dryrun_given), &(local_args_info.dryrun_given), optarg, 0, 0, ARG_NO, check_ambiguity, override, 0, 0, "dryrun", 'd', additional_error)) goto failure; break; case 's': /* Source port(s) for scan packets. */ if (update_arg( (void *)&(args_info->source_port_arg), &(args_info->source_port_orig), &(args_info->source_port_given), &(local_args_info.source_port_given), optarg, 0, 0, ARG_STRING, check_ambiguity, override, 0, 0, "source-port", 's', additional_error)) goto failure; break; case 'S': /* Source address(es) for scan packets. */ if (update_arg( (void *)&(args_info->source_ip_arg), &(args_info->source_ip_orig), &(args_info->source_ip_given), &(local_args_info.source_ip_given), optarg, 0, 0, ARG_STRING, check_ambiguity, override, 0, 0, "source-ip", 'S', additional_error)) goto failure; break; case 'G': /* Specify gateway MAC address. */ if (update_arg( (void *)&(args_info->gateway_mac_arg), &(args_info->gateway_mac_orig), &(args_info->gateway_mac_given), &(local_args_info.gateway_mac_given), optarg, 0, 0, ARG_STRING, check_ambiguity, override, 0, 0, "gateway-mac", 'G', additional_error)) goto failure; break; case 'i': /* Specify network interface to use. */ if (update_arg( (void *)&(args_info->interface_arg), &(args_info->interface_orig), &(args_info->interface_given), &(local_args_info.interface_given), optarg, 0, 0, ARG_STRING, check_ambiguity, override, 0, 0, "interface", 'i', additional_error)) goto failure; break; case 'M': /* Select probe module. */ if (update_arg( (void *)&(args_info->probe_module_arg), &(args_info->probe_module_orig), &(args_info->probe_module_given), &(local_args_info.probe_module_given), optarg, 0, "tcp_synscan", ARG_STRING, check_ambiguity, override, 0, 0, "probe-module", 'M', additional_error)) goto failure; break; case 'O': /* Select output module. */ if (update_arg( (void *)&(args_info->output_module_arg), &(args_info->output_module_orig), &(args_info->output_module_given), &(local_args_info.output_module_given), optarg, 0, "simple_file", ARG_STRING, check_ambiguity, override, 0, 0, "output-module", 'O', additional_error)) goto failure; break; case 'C': /* Read a configuration file, which can specify any of these options. */ if (update_arg( (void *)&(args_info->config_arg), &(args_info->config_orig), &(args_info->config_given), &(local_args_info.config_given), optarg, 0, "/etc/zmap/zmap.conf", ARG_STRING, check_ambiguity, override, 0, 0, "config", 'C', additional_error)) goto failure; break; case 'q': /* Do not print status updates. */ if (update_arg( 0 , 0 , &(args_info->quiet_given), &(local_args_info.quiet_given), optarg, 0, 0, ARG_NO, check_ambiguity, override, 0, 0, "quiet", 'q', additional_error)) goto failure; break; case 'g': /* Print configuration and summary at end of scan. */ if (update_arg( 0 , 0 , &(args_info->summary_given), &(local_args_info.summary_given), optarg, 0, 0, ARG_NO, check_ambiguity, override, 0, 0, "summary", 'g', additional_error)) goto failure; break; case 'v': /* Level of log detail (0-5). */ if (update_arg( (void *)&(args_info->verbosity_arg), &(args_info->verbosity_orig), &(args_info->verbosity_given), &(local_args_info.verbosity_given), optarg, 0, "3", ARG_INT, check_ambiguity, override, 0, 0, "verbosity", 'v', additional_error)) goto failure; break; case 'h': /* Print help and exit. */ if (update_arg( 0 , 0 , &(args_info->help_given), &(local_args_info.help_given), optarg, 0, 0, ARG_NO, check_ambiguity, override, 0, 0, "help", 'h', additional_error)) goto failure; break; case 'V': /* Print version and exit. */ if (update_arg( 0 , 0 , &(args_info->version_given), &(local_args_info.version_given), optarg, 0, 0, ARG_NO, check_ambiguity, override, 0, 0, "version", 'V', additional_error)) goto failure; break; case 0: /* Long option with no short option */ /* Arguments to pass to probe module. */ if (strcmp (long_options[option_index].name, "probe-args") == 0) { if (update_arg( (void *)&(args_info->probe_args_arg), &(args_info->probe_args_orig), &(args_info->probe_args_given), &(local_args_info.probe_args_given), optarg, 0, 0, ARG_STRING, check_ambiguity, override, 0, 0, "probe-args", '-', additional_error)) goto failure; } /* Arguments to pass to output module. */ else if (strcmp (long_options[option_index].name, "output-args") == 0) { if (update_arg( (void *)&(args_info->output_args_arg), &(args_info->output_args_orig), &(args_info->output_args_given), &(local_args_info.output_args_given), optarg, 0, 0, ARG_STRING, check_ambiguity, override, 0, 0, "output-args", '-', additional_error)) goto failure; } /* List available output modules. */ else if (strcmp (long_options[option_index].name, "list-output-modules") == 0) { if (update_arg( 0 , 0 , &(args_info->list_output_modules_given), &(local_args_info.list_output_modules_given), optarg, 0, 0, ARG_NO, check_ambiguity, override, 0, 0, "list-output-modules", '-', additional_error)) goto failure; } /* List available probe modules. */ else if (strcmp (long_options[option_index].name, "list-probe-modules") == 0) { if (update_arg( 0 , 0 , &(args_info->list_probe_modules_given), &(local_args_info.list_probe_modules_given), optarg, 0, 0, ARG_NO, check_ambiguity, override, 0, 0, "list-probe-modules", '-', additional_error)) goto failure; } break; case '?': /* Invalid option. */ /* `getopt_long' already printed an error message. */ goto failure; default: /* bug: option not considered. */ fprintf (stderr, "%s: option unknown: %c%s\n", CMDLINE_PARSER_PACKAGE, c, (additional_error ? additional_error : "")); abort (); } /* switch */ } /* while */ cmdline_parser_release (&local_args_info); if ( error ) return (EXIT_FAILURE); return 0; failure: cmdline_parser_release (&local_args_info); return (EXIT_FAILURE); } #ifndef CONFIG_FILE_LINE_SIZE #define CONFIG_FILE_LINE_SIZE 2048 #endif #define ADDITIONAL_ERROR " in configuration file " #define CONFIG_FILE_LINE_BUFFER_SIZE (CONFIG_FILE_LINE_SIZE+3) /* 3 is for "--" and "=" */ static int _cmdline_parser_configfile (const char *filename, int *my_argc) { FILE* file; char my_argv[CONFIG_FILE_LINE_BUFFER_SIZE+1]; char linebuf[CONFIG_FILE_LINE_SIZE]; int line_num = 0; int result = 0, equal; char *fopt, *farg; char *str_index; size_t len, next_token; char delimiter; if ((file = fopen(filename, "r")) == 0) { fprintf (stderr, "%s: Error opening configuration file '%s'\n", CMDLINE_PARSER_PACKAGE, filename); return EXIT_FAILURE; } while ((fgets(linebuf, CONFIG_FILE_LINE_SIZE, file)) != 0) { ++line_num; my_argv[0] = '\0'; len = strlen(linebuf); if (len > (CONFIG_FILE_LINE_BUFFER_SIZE-1)) { fprintf (stderr, "%s:%s:%d: Line too long in configuration file\n", CMDLINE_PARSER_PACKAGE, filename, line_num); result = EXIT_FAILURE; break; } /* find first non-whitespace character in the line */ next_token = strspn (linebuf, " \t\r\n"); str_index = linebuf + next_token; if ( str_index[0] == '\0' || str_index[0] == '#') continue; /* empty line or comment line is skipped */ fopt = str_index; /* truncate fopt at the end of the first non-valid character */ next_token = strcspn (fopt, " \t\r\n="); if (fopt[next_token] == '\0') /* the line is over */ { farg = 0; equal = 0; goto noarg; } /* remember if equal sign is present */ equal = (fopt[next_token] == '='); fopt[next_token++] = '\0'; /* advance pointers to the next token after the end of fopt */ next_token += strspn (fopt + next_token, " \t\r\n"); /* check for the presence of equal sign, and if so, skip it */ if ( !equal ) if ((equal = (fopt[next_token] == '='))) { next_token++; next_token += strspn (fopt + next_token, " \t\r\n"); } str_index += next_token; /* find argument */ farg = str_index; if ( farg[0] == '\"' || farg[0] == '\'' ) { /* quoted argument */ str_index = strchr (++farg, str_index[0] ); /* skip opening quote */ if (! str_index) { fprintf (stderr, "%s:%s:%d: unterminated string in configuration file\n", CMDLINE_PARSER_PACKAGE, filename, line_num); result = EXIT_FAILURE; break; } } else { /* read up the remaining part up to a delimiter */ next_token = strcspn (farg, " \t\r\n#\'\""); str_index += next_token; } /* truncate farg at the delimiter and store it for further check */ delimiter = *str_index, *str_index++ = '\0'; /* everything but comment is illegal at the end of line */ if (delimiter != '\0' && delimiter != '#') { str_index += strspn(str_index, " \t\r\n"); if (*str_index != '\0' && *str_index != '#') { fprintf (stderr, "%s:%s:%d: malformed string in configuration file\n", CMDLINE_PARSER_PACKAGE, filename, line_num); result = EXIT_FAILURE; break; } } noarg: if (!strcmp(fopt,"include")) { if (farg && *farg) { result = _cmdline_parser_configfile(farg, my_argc); } else { fprintf(stderr, "%s:%s:%d: include requires a filename argument.\n", CMDLINE_PARSER_PACKAGE, filename, line_num); } continue; } len = strlen(fopt); strcat (my_argv, len > 1 ? "--" : "-"); strcat (my_argv, fopt); if (len > 1 && ((farg && *farg) || equal)) strcat (my_argv, "="); if (farg && *farg) strcat (my_argv, farg); ++(*my_argc); cmd_line_list_tmp = (struct line_list *) malloc (sizeof (struct line_list)); cmd_line_list_tmp->next = cmd_line_list; cmd_line_list = cmd_line_list_tmp; cmd_line_list->string_arg = gengetopt_strdup(my_argv); } /* while */ if (file) fclose(file); return result; } int cmdline_parser_configfile ( const char *filename, struct gengetopt_args_info *args_info, int override, int initialize, int check_required) { struct cmdline_parser_params params; params.override = override; params.initialize = initialize; params.check_required = check_required; params.check_ambiguity = 0; params.print_errors = 1; return cmdline_parser_config_file (filename, args_info, ¶ms); } int cmdline_parser_config_file (const char *filename, struct gengetopt_args_info *args_info, struct cmdline_parser_params *params) { int i, result; int my_argc = 1; char **my_argv_arg; char *additional_error; /* store the program name */ cmd_line_list_tmp = (struct line_list *) malloc (sizeof (struct line_list)); cmd_line_list_tmp->next = cmd_line_list; cmd_line_list = cmd_line_list_tmp; cmd_line_list->string_arg = gengetopt_strdup (CMDLINE_PARSER_PACKAGE); result = _cmdline_parser_configfile(filename, &my_argc); if (result != EXIT_FAILURE) { my_argv_arg = (char **) malloc((my_argc+1) * sizeof(char *)); cmd_line_list_tmp = cmd_line_list; for (i = my_argc - 1; i >= 0; --i) { my_argv_arg[i] = cmd_line_list_tmp->string_arg; cmd_line_list_tmp = cmd_line_list_tmp->next; } my_argv_arg[my_argc] = 0; additional_error = (char *)malloc(strlen(filename) + strlen(ADDITIONAL_ERROR) + 1); strcpy (additional_error, ADDITIONAL_ERROR); strcat (additional_error, filename); result = cmdline_parser_internal (my_argc, my_argv_arg, args_info, params, additional_error); free (additional_error); free (my_argv_arg); } free_cmd_list(); if (result == EXIT_FAILURE) { cmdline_parser_free (args_info); exit (EXIT_FAILURE); } return result; }