![Guy Harris](/assets/img/avatar_default.png)
Add a "report a warning message" routine to the "report_err" code in libwsutil, and rename files and routines appropriately, as they don't only handle errors any more. Have a routine read_enabled_and_disabled_protos() that reads all the files that enable or disable protocols or heuristic dissectors, enables and disables them based on the contents of those files, and reports errors itself (as warnings) using the new "report a warning message" routine. Fix that error reporting to report separately on the disabled protocols, enabled protocols, and heuristic dissectors files. Have a routine to set up the enabled and disabled protocols and heuristic dissectors from the command-line arguments, so it's done the same way in all programs. If we try to enable or disable an unknown heuristic dissector via a command-line argument, report an error. Update a bunch of comments. Update the name of disabled_protos_cleanup(), as it cleans up information for disabled *and* enabled protocols and for heuristic dissectors. Support the command-line flags to enable and disable protocols and heuristic dissectors in tfshark. Change-Id: I9b8bd29947cccdf6dc34a0540b5509ef941391df Reviewed-on: https://code.wireshark.org/review/20966 Reviewed-by: Guy Harris <guy@alum.mit.edu>
289 lines
6.7 KiB
C
289 lines
6.7 KiB
C
/*
|
|
* randpkt.c
|
|
* ---------
|
|
* Creates random packet traces. Useful for debugging sniffers by testing
|
|
* assumptions about the veracity of the data found in the packet.
|
|
*
|
|
* Copyright (C) 1999 by Gilbert Ramirez <gram@alumni.rice.edu>
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
#include <config.h>
|
|
|
|
#include <glib.h>
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <wsutil/clopts_common.h>
|
|
#include <wsutil/cmdarg_err.h>
|
|
#include <wsutil/unicode-utils.h>
|
|
#include <wsutil/file_util.h>
|
|
#include <wsutil/filesystem.h>
|
|
#include <wsutil/privileges.h>
|
|
|
|
#ifdef HAVE_PLUGINS
|
|
#include <wsutil/plugins.h>
|
|
#endif
|
|
|
|
#include <wsutil/report_message.h>
|
|
#ifdef HAVE_GETOPT_H
|
|
#include <getopt.h>
|
|
#endif
|
|
|
|
#ifndef HAVE_GETOPT_LONG
|
|
#include "wsutil/wsgetopt.h"
|
|
#endif
|
|
|
|
#include "randpkt_core/randpkt_core.h"
|
|
|
|
#define INVALID_OPTION 1
|
|
#define INVALID_TYPE 2
|
|
#define CLOSE_ERROR 2
|
|
|
|
/*
|
|
* General errors and warnings are reported with an console message
|
|
* in randpkt.
|
|
*/
|
|
static void
|
|
failure_warning_message(const char *msg_format, va_list ap)
|
|
{
|
|
fprintf(stderr, "randpkt: ");
|
|
vfprintf(stderr, msg_format, ap);
|
|
fprintf(stderr, "\n");
|
|
}
|
|
|
|
/*
|
|
* Report additional information for an error in command-line arguments.
|
|
*/
|
|
static void
|
|
failure_message_cont(const char *msg_format, va_list ap)
|
|
{
|
|
vfprintf(stderr, msg_format, ap);
|
|
fprintf(stderr, "\n");
|
|
}
|
|
|
|
/* Print usage statement and exit program */
|
|
static void
|
|
usage(gboolean is_error)
|
|
{
|
|
FILE *output;
|
|
char** abbrev_list;
|
|
char** longname_list;
|
|
unsigned i = 0;
|
|
|
|
if (!is_error) {
|
|
output = stdout;
|
|
}
|
|
else {
|
|
output = stderr;
|
|
}
|
|
|
|
fprintf(output, "Usage: randpkt [-b maxbytes] [-c count] [-t type] [-r] filename\n");
|
|
fprintf(output, "Default max bytes (per packet) is 5000\n");
|
|
fprintf(output, "Default count is 1000.\n");
|
|
fprintf(output, "-r: random packet type selection\n");
|
|
fprintf(output, "\n");
|
|
fprintf(output, "Types:\n");
|
|
|
|
/* Get the examples list */
|
|
randpkt_example_list(&abbrev_list, &longname_list);
|
|
while (abbrev_list[i] && longname_list[i]) {
|
|
fprintf(output, "\t%-16s%s\n", abbrev_list[i], longname_list[i]);
|
|
i++;
|
|
}
|
|
|
|
g_strfreev(abbrev_list);
|
|
g_strfreev(longname_list);
|
|
|
|
fprintf(output, "\nIf type is not specified, a random packet will be chosen\n\n");
|
|
}
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
char *init_progfile_dir_error;
|
|
int opt;
|
|
int produce_type = -1;
|
|
char *produce_filename = NULL;
|
|
int produce_max_bytes = 5000;
|
|
int produce_count = 1000;
|
|
randpkt_example *example;
|
|
guint8* type = NULL;
|
|
int allrandom = FALSE;
|
|
wtap_dumper *savedump;
|
|
int ret = EXIT_SUCCESS;
|
|
static const struct option long_options[] = {
|
|
{"help", no_argument, NULL, 'h'},
|
|
{0, 0, 0, 0 }
|
|
};
|
|
|
|
/*
|
|
* Get credential information for later use.
|
|
*/
|
|
init_process_policies();
|
|
|
|
/*
|
|
* Attempt to get the pathname of the directory containing the
|
|
* executable file.
|
|
*/
|
|
init_progfile_dir_error = init_progfile_dir(argv[0], main);
|
|
if (init_progfile_dir_error != NULL) {
|
|
fprintf(stderr,
|
|
"capinfos: Can't get pathname of directory containing the capinfos program: %s.\n",
|
|
init_progfile_dir_error);
|
|
g_free(init_progfile_dir_error);
|
|
}
|
|
|
|
wtap_init();
|
|
|
|
cmdarg_err_init(failure_warning_message, failure_message_cont);
|
|
|
|
#ifdef _WIN32
|
|
arg_list_utf_16to8(argc, argv);
|
|
create_app_running_mutex();
|
|
#endif /* _WIN32 */
|
|
|
|
#ifdef HAVE_PLUGINS
|
|
/* Register wiretap plugins */
|
|
init_report_message(failure_warning_message, failure_warning_message,
|
|
NULL, NULL, NULL);
|
|
|
|
/* Scan for plugins. This does *not* call their registration routines;
|
|
that's done later.
|
|
|
|
Don't report failures to load plugins because most
|
|
(non-wiretap) plugins *should* fail to load (because
|
|
we're not linked against libwireshark and dissector
|
|
plugins need libwireshark). */
|
|
scan_plugins(DONT_REPORT_LOAD_FAILURE);
|
|
|
|
/* Register all libwiretap plugin modules. */
|
|
register_all_wiretap_modules();
|
|
#endif
|
|
|
|
while ((opt = getopt_long(argc, argv, "b:c:ht:r", long_options, NULL)) != -1) {
|
|
switch (opt) {
|
|
case 'b': /* max bytes */
|
|
produce_max_bytes = get_positive_int(optarg, "max bytes");
|
|
if (produce_max_bytes > 65536) {
|
|
cmdarg_err("max bytes is > 65536");
|
|
ret = INVALID_OPTION;
|
|
goto clean_exit;
|
|
}
|
|
break;
|
|
|
|
case 'c': /* count */
|
|
produce_count = get_positive_int(optarg, "count");
|
|
break;
|
|
|
|
case 't': /* type of packet to produce */
|
|
type = g_strdup(optarg);
|
|
break;
|
|
|
|
case 'h':
|
|
usage(FALSE);
|
|
goto clean_exit;
|
|
break;
|
|
|
|
case 'r':
|
|
allrandom = TRUE;
|
|
break;
|
|
|
|
default:
|
|
usage(TRUE);
|
|
ret = INVALID_OPTION;
|
|
goto clean_exit;
|
|
break;
|
|
}
|
|
}
|
|
|
|
/* any more command line parameters? */
|
|
if (argc > optind) {
|
|
produce_filename = argv[optind];
|
|
}
|
|
else {
|
|
usage(TRUE);
|
|
ret = INVALID_OPTION;
|
|
goto clean_exit;
|
|
}
|
|
|
|
if (!allrandom) {
|
|
produce_type = randpkt_parse_type(type);
|
|
g_free(type);
|
|
|
|
example = randpkt_find_example(produce_type);
|
|
if (!example) {
|
|
ret = INVALID_OPTION;
|
|
goto clean_exit;
|
|
}
|
|
|
|
ret = randpkt_example_init(example, produce_filename, produce_max_bytes);
|
|
if (ret != EXIT_SUCCESS)
|
|
goto clean_exit;
|
|
randpkt_loop(example, produce_count);
|
|
} else {
|
|
if (type) {
|
|
fprintf(stderr, "Can't set type in random mode\n");
|
|
ret = INVALID_TYPE;
|
|
goto clean_exit;
|
|
}
|
|
|
|
produce_type = randpkt_parse_type(NULL);
|
|
example = randpkt_find_example(produce_type);
|
|
if (!example) {
|
|
ret = INVALID_OPTION;
|
|
goto clean_exit;
|
|
}
|
|
ret = randpkt_example_init(example, produce_filename, produce_max_bytes);
|
|
if (ret != EXIT_SUCCESS)
|
|
goto clean_exit;
|
|
|
|
while (produce_count-- > 0) {
|
|
randpkt_loop(example, 1);
|
|
produce_type = randpkt_parse_type(NULL);
|
|
|
|
savedump = example->dump;
|
|
|
|
example = randpkt_find_example(produce_type);
|
|
if (!example) {
|
|
ret = INVALID_OPTION;
|
|
goto clean_exit;
|
|
}
|
|
example->dump = savedump;
|
|
}
|
|
}
|
|
if (!randpkt_example_close(example)) {
|
|
ret = CLOSE_ERROR;
|
|
}
|
|
|
|
clean_exit:
|
|
wtap_cleanup();
|
|
return ret;
|
|
}
|
|
|
|
/*
|
|
* Editor modelines - http://www.wireshark.org/tools/modelines.html
|
|
*
|
|
* Local variables:
|
|
* c-basic-offset: 8
|
|
* tab-width: 8
|
|
* indent-tabs-mode: t
|
|
* End:
|
|
*
|
|
* vi: set shiftwidth=8 tabstop=8 noexpandtab:
|
|
* :indentSize=8:tabSize=8:noTabs=false:
|
|
*/
|