coydog/zmap-freebsd
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
ce5be6e30f
zmap-freebsd/src/zopt.c
Zakir Durumeric 490054d239 inital public release
2013-08-16 11:12:47 -04:00

1340 lines
44 KiB
C
Raw Blame History

/*
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 <stdio.h>
#include <stdlib.h>
#include <string.h>
#ifndef FIX_UNUSED
#define FIX_UNUSED(X) (void) (X) /* avoid warnings for unused params */
#endif
#include <getopt.h>
#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, &params, 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, &params);
}
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;
}
Reference in New Issue View Git Blame Copy Permalink