diff options
author | Dario Lombardo <lomato@gmail.com> | 2015-11-10 15:41:52 +0100 |
---|---|---|
committer | Michael Mann <mmann78@netscape.net> | 2015-11-11 13:22:05 +0000 |
commit | f1ff6d62c4fc1db0f5b98efd537f3ec518b0d4b8 (patch) | |
tree | c1bb7ecf4c3aa5ea444752b71cc6508bc145d733 /randpkt-core.c | |
parent | 328fbc001e89d2daaca343c378744e75e0a107fb (diff) |
randpkt: split into a core and an app.
This will allow other apps to use the random packet generation
features.
Change-Id: I7e9af58cbe39da4908242b5fbb292f473e03b4f2
Reviewed-on: https://code.wireshark.org/review/11579
Petri-Dish: Michael Mann <mmann78@netscape.net>
Tested-by: Petri Dish Buildbot <buildbot-no-reply@wireshark.org>
Reviewed-by: Michael Mann <mmann78@netscape.net>
Diffstat (limited to 'randpkt-core.c')
-rw-r--r-- | randpkt-core.c | 800 |
1 files changed, 800 insertions, 0 deletions
diff --git a/randpkt-core.c b/randpkt-core.c new file mode 100644 index 0000000000..a078bd0deb --- /dev/null +++ b/randpkt-core.c @@ -0,0 +1,800 @@ +/* + * randpkt-core.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 "randpkt-core.h" + +#include <config.h> + +#include <time.h> +#include <errno.h> +#include <stdlib.h> +#include <string.h> +#include "wsutil/file_util.h" + +#ifdef _WIN32 +#include <wsutil/unicode-utils.h> +#endif /* _WIN32 */ + +#define array_length(x) (sizeof x / sizeof x[0]) + +/* Types of produceable packets */ +enum { + PKT_ARP, + PKT_BGP, + PKT_BVLC, + PKT_DNS, + PKT_ETHERNET, + PKT_FDDI, + PKT_GIOP, + PKT_ICMP, + PKT_IP, + PKT_LLC, + PKT_M2M, + PKT_MEGACO, + PKT_NBNS, + PKT_NCP2222, + PKT_SCTP, + PKT_SYSLOG, + PKT_TCP, + PKT_TDS, + PKT_TR, + PKT_UDP, + PKT_USB, + PKT_USB_LINUX +}; + +/* Ethernet, indicating ARP */ +guint8 pkt_arp[] = { + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0x00, 0x00, + 0x32, 0x25, 0x0f, 0xff, + 0x08, 0x06 +}; + +/* Ethernet+IP+UDP, indicating DNS */ +guint8 pkt_dns[] = { + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0x01, 0x01, + 0x01, 0x01, 0x01, 0x01, + 0x08, 0x00, + + 0x45, 0x00, 0x00, 0x3c, + 0xc5, 0x9e, 0x40, 0x00, + 0xff, 0x11, 0xd7, 0xe0, + 0xd0, 0x15, 0x02, 0xb8, + 0x0a, 0x01, 0x01, 0x63, + + 0x05, 0xe8, 0x00, 0x35, + 0xff, 0xff, 0x2a, 0xb9, + 0x30 +}; + +/* Ethernet+IP, indicating ICMP */ +guint8 pkt_icmp[] = { + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0x01, 0x01, + 0x01, 0x01, 0x01, 0x01, + 0x08, 0x00, + + 0x45, 0x00, 0x00, 0x54, + 0x8f, 0xb3, 0x40, 0x00, + 0xfd, 0x01, 0x8a, 0x99, + 0xcc, 0xfc, 0x66, 0x0b, + 0xce, 0x41, 0x62, 0x12 +}; + +/* Ethernet, indicating IP */ +guint8 pkt_ip[] = { + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0x01, 0x01, + 0x01, 0x01, 0x01, 0x01, + 0x08, 0x00 +}; + +/* TR, indicating LLC */ +guint8 pkt_llc[] = { + 0x10, 0x40, 0x68, 0x00, + 0x19, 0x69, 0x95, 0x8b, + 0x00, 0x01, 0xfa, 0x68, + 0xc4, 0x67 +}; + +/* Ethernet, indicating WiMAX M2M */ +guint8 pkt_m2m[] = { + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0x00, 0x00, + 0x32, 0x25, 0x0f, 0xff, + 0x08, 0xf0 +}; + +/* Ethernet+IP+UDP, indicating NBNS */ +guint8 pkt_nbns[] = { + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0x01, 0x01, + 0x01, 0x01, 0x01, 0x01, + 0x08, 0x00, + + 0x45, 0x00, 0x00, 0x3c, + 0xc5, 0x9e, 0x40, 0x00, + 0xff, 0x11, 0xd7, 0xe0, + 0xd0, 0x15, 0x02, 0xb8, + 0x0a, 0x01, 0x01, 0x63, + + 0x00, 0x89, 0x00, 0x89, + 0x00, 0x00, 0x2a, 0xb9, + 0x30 +}; + +/* Ethernet+IP+UDP, indicating syslog */ +guint8 pkt_syslog[] = { + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0x01, 0x01, + 0x01, 0x01, 0x01, 0x01, + 0x08, 0x00, + + 0x45, 0x00, 0x00, 0x64, + 0x20, 0x48, 0x00, 0x00, + 0xfc, 0x11, 0xf8, 0x03, + 0xd0, 0x15, 0x02, 0xb8, + 0x0a, 0x01, 0x01, 0x63, + + 0x05, 0xe8, 0x02, 0x02, + 0x00, 0x50, 0x51, 0xe1, + 0x3c +}; + +/* TR+LLC+IP, indicating TCP */ +guint8 pkt_tcp[] = { + 0x10, 0x40, 0x68, 0x00, + 0x19, 0x69, 0x95, 0x8b, + 0x00, 0x01, 0xfa, 0x68, + 0xc4, 0x67, + + 0xaa, 0xaa, 0x03, 0x00, + 0x00, 0x00, 0x08, 0x00, + + 0x45, 0x00, 0x00, 0x28, + 0x0b, 0x0b, 0x40, 0x00, + 0x20, 0x06, 0x85, 0x37, + 0xc0, 0xa8, 0x27, 0x01, + 0xc0, 0xa8, 0x22, 0x3c +}; + +/* Ethernet+IP, indicating UDP */ +guint8 pkt_udp[] = { + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0x01, 0x01, + 0x01, 0x01, 0x01, 0x01, + 0x08, 0x00, + + 0x45, 0x00, 0x00, 0x3c, + 0xc5, 0x9e, 0x40, 0x00, + 0xff, 0x11, 0xd7, 0xe0, + 0xd0, 0x15, 0x02, 0xb8, + 0x0a, 0x01, 0x01, 0x63 +}; + +/* Ethernet+IP+UDP, indicating BVLC */ +guint8 pkt_bvlc[] = { + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0x01, 0x01, + 0x01, 0x01, 0x01, 0x01, + 0x08, 0x00, + + 0x45, 0x00, 0x00, 0x3c, + 0xc5, 0x9e, 0x40, 0x00, + 0xff, 0x11, 0x01, 0xaa, + 0xc1, 0xff, 0x19, 0x1e, + 0xc1, 0xff, 0x19, 0xff, + 0xba, 0xc0, 0xba, 0xc0, + 0x00, 0xff, 0x2d, 0x5e, + 0x81 +}; + +/* TR+LLC+IPX, indicating NCP, with NCP Type == 0x2222 */ +guint8 pkt_ncp2222[] = { + 0x10, 0x40, 0x00, 0x00, + 0xf6, 0x7c, 0x9b, 0x70, + 0x68, 0x00, 0x19, 0x69, + 0x95, 0x8b, 0xe0, 0xe0, + 0x03, 0xff, 0xff, 0x00, + 0x25, 0x02, 0x11, 0x00, + 0x00, 0x74, 0x14, 0x00, + 0x00, 0x00, 0x00, 0x00, + 0x01, 0x04, 0x51, 0x00, + 0x00, 0x00, 0x04, 0x00, + 0x02, 0x16, 0x19, 0x7a, + 0x84, 0x40, 0x01, 0x22, + 0x22 +}; + +/* Ethernet+IP+TCP, indicating GIOP */ +guint8 pkt_giop[] = { + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0x01, 0x01, + 0x01, 0x01, 0x01, 0x01, + 0x08, 0x00, + + 0x45, 0x00, 0x00, 0xa6, + 0x00, 0x2f, 0x40, 0x00, + 0x40, 0x06, 0x3c, 0x21, + 0x7f, 0x00, 0x00, 0x01, + 0x7f, 0x00, 0x00, 0x01, + + 0x30, 0x39, 0x04, 0x05, + 0xac, 0x02, 0x1e, 0x69, + 0xab, 0x74, 0xab, 0x64, + 0x80, 0x18, 0x79, 0x60, + 0xc4, 0xb8, 0x00, 0x00, + 0x01, 0x01, 0x08, 0x0a, + 0x00, 0x00, 0x48, 0xf5, + 0x00, 0x00, 0x48, 0xf5, + + 0x47, 0x49, 0x4f, 0x50, + 0x01, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x30, + 0x00, 0x00, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x01, + 0x01 +}; + +/* Ethernet+IP+TCP, indicating BGP */ +guint8 pkt_bgp[] = { + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0x01, 0x01, + 0x01, 0x01, 0x01, 0x01, + 0x08, 0x00, + + 0x45, 0x00, 0x00, 0xa6, + 0x00, 0x2f, 0x40, 0x00, + 0x40, 0x06, 0x3c, 0x21, + 0x7f, 0x00, 0x00, 0x01, + 0x7f, 0x00, 0x00, 0x01, + + 0x30, 0x39, 0x00, 0xb3, + 0xac, 0x02, 0x1e, 0x69, + 0xab, 0x74, 0xab, 0x64, + 0x80, 0x18, 0x79, 0x60, + 0xc4, 0xb8, 0x00, 0x00, + 0x01, 0x01, 0x08, 0x0a, + 0x00, 0x00, 0x48, 0xf5, + 0x00, 0x00, 0x48, 0xf5, + + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0xff, 0xff, + 0xff, 0xff, 0xff, 0xff, +}; + +/* Ethernet+IP+TCP, indicating TDS NetLib */ +guint8 pkt_tds[] = { + 0x00, 0x50, 0x8b, 0x0d, + 0x7a, 0xed, 0x00, 0x08, + 0xa3, 0x98, 0x39, 0x81, + 0x08, 0x00, + + 0x45, 0x00, 0x03, 0x8d, + 0x90, 0xd4, 0x40, 0x00, + 0x7c, 0x06, 0xc3, 0x1b, + 0xac, 0x14, 0x02, 0x22, + 0x0a, 0xc2, 0xee, 0x82, + + 0x05, 0x99, 0x08, 0xf8, + 0xff, 0x4e, 0x85, 0x46, + 0xa2, 0xb4, 0x42, 0xaa, + 0x50, 0x18, 0x3c, 0x28, + 0x0f, 0xda, 0x00, 0x00, +}; + +/* Ethernet+IP, indicating SCTP */ +guint8 pkt_sctp[] = { + 0x00, 0xa0, 0x80, 0x00, + 0x5e, 0x46, 0x08, 0x00, + 0x03, 0x4a, 0x00, 0x35, + 0x08, 0x00, + + 0x45, 0x00, 0x00, 0x7c, + 0x14, 0x1c, 0x00, 0x00, + 0x3b, 0x84, 0x4a, 0x54, + 0x0a, 0x1c, 0x06, 0x2b, + 0x0a, 0x1c, 0x06, 0x2c, +}; + + +/* Ethernet+IP+SCTP, indicating MEGACO */ +guint8 pkt_megaco[] = { + 0x00, 0xa0, 0x80, 0x00, + 0x5e, 0x46, 0x08, 0x00, + 0x03, 0x4a, 0x00, 0x35, + 0x08, 0x00, + + 0x45, 0x00, 0x00, 0x7c, + 0x14, 0x1c, 0x00, 0x00, + 0x3b, 0x84, 0x4a, 0x54, + 0x0a, 0x1c, 0x06, 0x2b, + 0x0a, 0x1c, 0x06, 0x2c, + + 0x40, 0x00, 0x0b, 0x80, + 0x00, 0x01, 0x6f, 0x0a, + 0x6d, 0xb0, 0x18, 0x82, + 0x00, 0x03, 0x00, 0x5b, + 0x28, 0x02, 0x43, 0x45, + 0x00, 0x00, 0xa0, 0xbd, + 0x00, 0x00, 0x00, 0x07, +}; + +/* This little data table drives the whole program */ +static randpkt_example examples[] = { + { "arp", "Address Resolution Protocol", + PKT_ARP, WTAP_ENCAP_ETHERNET, + pkt_arp, array_length(pkt_arp), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "bgp", "Border Gateway Protocol", + PKT_BGP, WTAP_ENCAP_ETHERNET, + pkt_bgp, array_length(pkt_bgp), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "bvlc", "BACnet Virtual Link Control", + PKT_BVLC, WTAP_ENCAP_ETHERNET, + pkt_bvlc, array_length(pkt_bvlc), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "dns", "Domain Name Service", + PKT_DNS, WTAP_ENCAP_ETHERNET, + pkt_dns, array_length(pkt_dns), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "eth", "Ethernet", + PKT_ETHERNET, WTAP_ENCAP_ETHERNET, + NULL, 0, + NULL, 0, + NULL, NULL, + 1000, + }, + + { "fddi", "Fiber Distributed Data Interface", + PKT_FDDI, WTAP_ENCAP_FDDI, + NULL, 0, + NULL, 0, + NULL, NULL, + 1000, + }, + + { "giop", "General Inter-ORB Protocol", + PKT_GIOP, WTAP_ENCAP_ETHERNET, + pkt_giop, array_length(pkt_giop), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "icmp", "Internet Control Message Protocol", + PKT_ICMP, WTAP_ENCAP_ETHERNET, + pkt_icmp, array_length(pkt_icmp), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "ip", "Internet Protocol", + PKT_IP, WTAP_ENCAP_ETHERNET, + pkt_ip, array_length(pkt_ip), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "llc", "Logical Link Control", + PKT_LLC, WTAP_ENCAP_TOKEN_RING, + pkt_llc, array_length(pkt_llc), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "m2m", "WiMAX M2M Encapsulation Protocol", + PKT_M2M, WTAP_ENCAP_ETHERNET, + pkt_m2m, array_length(pkt_m2m), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "megaco", "MEGACO", + PKT_MEGACO, WTAP_ENCAP_ETHERNET, + pkt_megaco, array_length(pkt_megaco), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "nbns", "NetBIOS-over-TCP Name Service", + PKT_NBNS, WTAP_ENCAP_ETHERNET, + pkt_nbns, array_length(pkt_nbns), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "ncp2222", "NetWare Core Protocol", + PKT_NCP2222, WTAP_ENCAP_TOKEN_RING, + pkt_ncp2222, array_length(pkt_ncp2222), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "sctp", "Stream Control Transmission Protocol", + PKT_SCTP, WTAP_ENCAP_ETHERNET, + pkt_sctp, array_length(pkt_sctp), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "syslog", "Syslog message", + PKT_SYSLOG, WTAP_ENCAP_ETHERNET, + pkt_syslog, array_length(pkt_syslog), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "tds", "TDS NetLib", + PKT_TDS, WTAP_ENCAP_ETHERNET, + pkt_tds, array_length(pkt_tds), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "tcp", "Transmission Control Protocol", + PKT_TCP, WTAP_ENCAP_TOKEN_RING, + pkt_tcp, array_length(pkt_tcp), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "tr", "Token-Ring", + PKT_TR, WTAP_ENCAP_TOKEN_RING, + NULL, 0, + NULL, 0, + NULL, NULL, + 1000, + }, + + { "udp", "User Datagram Protocol", + PKT_UDP, WTAP_ENCAP_ETHERNET, + pkt_udp, array_length(pkt_udp), + NULL, 0, + NULL, NULL, + 1000, + }, + + { "usb", "Universal Serial Bus", + PKT_USB, WTAP_ENCAP_USB, + NULL, 0, + NULL, 0, + NULL, NULL, + 1000, + }, + + { "usb-linux", "Universal Serial Bus with Linux specific header", + PKT_USB_LINUX, WTAP_ENCAP_USB_LINUX, + NULL, 0, + NULL, 0, + NULL, NULL, + 1000, + }, + +}; + +guint randpkt_example_count(void) +{ + return array_length(examples); +} + +/* Find pkt_example record and return pointer to it */ +randpkt_example* randpkt_find_example(int type) +{ + int num_entries = array_length(examples); + int i; + + for (i = 0; i < num_entries; i++) { + if (examples[i].produceable_type == type) { + return &examples[i]; + } + } + + fprintf(stderr, "randpkt: Internal error. Type %d has no entry in examples table.\n", + type); + return NULL; +} + +void randpkt_loop(randpkt_example* example, guint64 produce_count) +{ + guint i; + int j; + int err; + int len_random; + int len_this_pkt; + gchar* err_info; + union wtap_pseudo_header* ps_header; + guint8 buffer[65536]; + struct wtap_pkthdr* pkthdr; + + pkthdr = g_new0(struct wtap_pkthdr, 1); + + pkthdr->rec_type = REC_TYPE_PACKET; + pkthdr->presence_flags = WTAP_HAS_TS; + pkthdr->pkt_encap = example->sample_wtap_encap; + + memset(pkthdr, 0, sizeof(struct wtap_pkthdr)); + memset(buffer, 0, sizeof(buffer)); + + ps_header = &pkthdr->pseudo_header; + + /* Load the sample pseudoheader into our pseudoheader buffer */ + if (example->pseudo_buffer) + memcpy(ps_header, example->pseudo_buffer, example->pseudo_length); + + /* Load the sample into our buffer */ + if (example->sample_buffer) + memcpy(buffer, example->sample_buffer, example->sample_length); + + /* Produce random packets */ + for (i = 0; i < produce_count; i++) { + if (example->produce_max_bytes > 0) { + len_random = (rand() % example->produce_max_bytes + 1); + } + else { + len_random = 0; + } + + len_this_pkt = example->sample_length + len_random; + + pkthdr->caplen = len_this_pkt; + pkthdr->len = len_this_pkt; + pkthdr->ts.secs = i; /* just for variety */ + + for (j = example->pseudo_length; j < (int) sizeof(*ps_header); j++) { + ((guint8*)ps_header)[j] = (rand() % 0x100); + } + + for (j = example->sample_length; j < len_this_pkt; j++) { + /* Add format strings here and there */ + if ((int) (100.0*rand()/(RAND_MAX+1.0)) < 3 && j < (len_random - 3)) { + memcpy(&buffer[j], "%s", 3); + j += 2; + } else { + buffer[j] = (rand() % 0x100); + } + } + + if (!wtap_dump(example->dump, pkthdr, buffer, &err, &err_info)) { + fprintf(stderr, "randpkt: Error writing to %s: %s\n", + example->filename, wtap_strerror(err)); + switch (err) { + + case WTAP_ERR_UNWRITABLE_ENCAP: + /* + * This is a problem with the particular + * frame we're writing and the file type + * and subtype we're writing; note that, + * and report the file type/subtype. + */ + fprintf(stderr, + "Frame has a network type that can't be saved in a \"%s\" file.\n", + wtap_file_type_subtype_short_string(WTAP_FILE_TYPE_SUBTYPE_PCAP)); + break; + + case WTAP_ERR_PACKET_TOO_LARGE: + /* + * This is a problem with the particular + * frame we're writing and the file type + * and subtype we're writing; note that, + * and report the file type/subtype. + */ + fprintf(stderr, + "Frame is too large for a \"%s\" file.\n", + wtap_file_type_subtype_short_string(WTAP_FILE_TYPE_SUBTYPE_PCAP)); + break; + + case WTAP_ERR_UNWRITABLE_REC_TYPE: + /* + * This is a problem with the particular + * record we're writing and the file type + * and subtype we're writing; note that, + * and report the file type/subtype. + */ + fprintf(stderr, + "Record has a record type that can't be saved in a \"%s\" file.\n", + wtap_file_type_subtype_short_string(WTAP_FILE_TYPE_SUBTYPE_PCAP)); + break; + + case WTAP_ERR_UNWRITABLE_REC_DATA: + /* + * This is a problem with the particular + * record we're writing and the file type + * and subtype we're writing; note that, + * and report the file type/subtype. + */ + fprintf(stderr, + "Record has data that can't be saved in a \"%s\" file.\n(%s)\n", + wtap_file_type_subtype_short_string(WTAP_FILE_TYPE_SUBTYPE_PCAP), + err_info != NULL ? err_info : "no information supplied"); + g_free(err_info); + break; + + default: + break; + } + } + } + + g_free(pkthdr); +} + +gboolean randpkt_example_close(randpkt_example* example) +{ + int err; + + if (!wtap_dump_close(example->dump, &err)) { + fprintf(stderr, "Error writing to %s: %s\n", + example->filename, wtap_strerror(err)); + return FALSE; + } + return TRUE; +} + +void randpkt_example_init(randpkt_example* example, char* produce_filename, int produce_max_bytes) +{ + int err; + + if (strcmp(produce_filename, "-") == 0) { + /* Write to the standard output. */ + example->dump = wtap_dump_open_stdout(WTAP_FILE_TYPE_SUBTYPE_PCAP, + example->sample_wtap_encap, produce_max_bytes, FALSE /* compressed */, &err); + example->filename = "the standard output"; + } else { + example->dump = wtap_dump_open(produce_filename, WTAP_FILE_TYPE_SUBTYPE_PCAP, + example->sample_wtap_encap, produce_max_bytes, FALSE /* compressed */, &err); + example->filename = produce_filename; + } + if (!example->dump) { + fprintf(stderr, "randpkt: Error writing to %s\n", example->filename); + exit(2); + } + + /* reduce max_bytes by # of bytes already in sample */ + if (produce_max_bytes <= example->sample_length) { + fprintf(stderr, "randpkt: Sample packet length is %d, which is greater than " + "or equal to\n", example->sample_length); + fprintf(stderr, "your requested max_bytes value of %d\n", produce_max_bytes); + exit(1); + } else { + example->produce_max_bytes -= example->sample_length; + } +} + +/* Seed the random-number generator */ +void +randpkt_seed(void) +{ + unsigned int randomness; + time_t now; +#ifndef _WIN32 + int fd; + ssize_t ret; + +#define RANDOM_DEV "/dev/urandom" + + /* + * Assume it's at least worth trying /dev/urandom on UN*X. + * If it doesn't exist, fall back on time(). + * + * XXX - Use CryptGenRandom on Windows? + */ + fd = ws_open(RANDOM_DEV, O_RDONLY); + if (fd == -1) { + if (errno != ENOENT) { + fprintf(stderr, + "randpkt: Could not open " RANDOM_DEV " for reading: %s\n", + g_strerror(errno)); + exit(2); + } + goto fallback; + } + + ret = ws_read(fd, &randomness, sizeof randomness); + if (ret == -1) { + fprintf(stderr, + "randpkt: Could not read from " RANDOM_DEV ": %s\n", + g_strerror(errno)); + exit(2); + } + if ((size_t)ret != sizeof randomness) { + fprintf(stderr, + "randpkt: Tried to read %lu bytes from " RANDOM_DEV ", got %ld\n", + (unsigned long)sizeof randomness, (long)ret); + exit(2); + } + srand(randomness); + ws_close(fd); + return; + +fallback: +#endif + now = time(NULL); + randomness = (unsigned int) now; + + srand(randomness); +} + +/* Parse command-line option "type" and return enum type */ +int randpkt_parse_type(char *string) +{ + int num_entries = array_length(examples); + int i; + + /* Called with NULL, choose a random packet */ + if (!string) { + return examples[rand() % num_entries].produceable_type; + } + + for (i = 0; i < num_entries; i++) { + if (g_strcmp0(examples[i].abbrev, string) == 0) { + return examples[i].produceable_type; + } + } + + /* Complain */ + fprintf(stderr, "randpkt: Type %s not known.\n", string); + return -1; +} + +void randpkt_example_list(const char*** abbrev_list, const char*** longname_list, unsigned* list_num) +{ + unsigned i; + *list_num = randpkt_example_count(); + *abbrev_list = g_new0(const char*, *list_num); + *longname_list = g_new0(const char*, *list_num); + for (i = 0; i < *list_num; i++) { + (*abbrev_list)[i] = examples[i].abbrev; + (*longname_list)[i] = examples[i].longname; + } +}
\ No newline at end of file |