nexmon – Rev 1

Subversion Repositories:
Rev:
/*
 * 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/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_err.h>
#ifdef HAVE_GETOPT_H
#include <getopt.h>
#endif

#ifndef HAVE_GETOPT_LONG
#include "wsutil/wsgetopt.h"
#endif

#include "randpkt_core/randpkt_core.h"

#ifdef HAVE_PLUGINS
/*
 *  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).
 */
static void
failure_message(const char *msg_format _U_, va_list ap _U_)
{
  return;
}
#endif

/* 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");

        exit(is_error ? 1 : 0);
}
int
main(int argc, char **argv)
{
        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;
        static const struct option long_options[] = {
                {"help", no_argument, NULL, 'h'},
                {0, 0, 0, 0 }
        };

#ifdef HAVE_PLUGINS
        char  *init_progfile_dir_error;
#endif

  /*
   * Get credential information for later use.
   */
  init_process_policies();
  init_open_routines();

#ifdef _WIN32
        arg_list_utf_16to8(argc, argv);
        create_app_running_mutex();
#endif /* _WIN32 */

#ifdef HAVE_PLUGINS
        /* Register wiretap plugins */
        if ((init_progfile_dir_error = init_progfile_dir(argv[0], main))) {
                g_warning("randpkt: init_progfile_dir(): %s", init_progfile_dir_error);
                g_free(init_progfile_dir_error);
        } else {
                /* Register all the plugin types we have. */
                wtap_register_plugin_types(); /* Types known to libwiretap */

                init_report_err(failure_message,NULL,NULL,NULL);

                /* Scan for plugins.  This does *not* call their registration routines;
                   that's done later. */
                scan_plugins();

                /* 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 = atoi(optarg);
                                if (produce_max_bytes > 65536) {
                                        fprintf(stderr, "randpkt: Max bytes is 65536\n");
                                        return 1;
                                }
                                break;

                        case 'c':       /* count */
                                produce_count = atoi(optarg);
                                break;

                        case 't':       /* type of packet to produce */
                                type = g_strdup(optarg);
                                break;

                        case 'h':
                                usage(FALSE);
                                break;

                        case 'r':
                                allrandom = TRUE;
                                break;

                        default:
                                usage(TRUE);
                                break;
                }
        }

        /* any more command line parameters? */
        if (argc > optind) {
                produce_filename = argv[optind];
        }
        else {
                usage(TRUE);
        }

        if (!allrandom) {
                produce_type = randpkt_parse_type(type);
                g_free(type);

                example = randpkt_find_example(produce_type);
                if (!example)
                        return 1;

                randpkt_example_init(example, produce_filename, produce_max_bytes);
                randpkt_loop(example, produce_count);
        } else {
                if (type) {
                        fprintf(stderr, "Can't set type in random mode\n");
                        return 2;
                }

                produce_type = randpkt_parse_type(NULL);
                example = randpkt_find_example(produce_type);
                if (!example)
                        return 1;
                randpkt_example_init(example, produce_filename, produce_max_bytes);

                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)
                                return 1;
                        example->dump = savedump;
                }
        }
        if (!randpkt_example_close(example))
                return 2;
        return 0;

}

/*
 * 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:
 */