#include <string.h>
#include <wsutil/file_util.h>
#include <wsutil/crash_info.h>
-#include <ws_version_info.h>
+#include <version_info.h>
#include <wsutil/inet_addr.h>
+#ifdef _WIN32
+#include <io.h> /* for _setmode */
+#include <fcntl.h> /* for O_BINARY */
+#endif
+
#include <time.h>
#include <glib.h>
#include "writecap/pcapio.h"
#include "text2pcap.h"
+#include "wiretap/wtap.h"
+
#ifdef _WIN32
#include <wsutil/unicode-utils.h>
#endif /* _WIN32 */
/*--- Local date -----------------------------------------------------------------*/
/* This is where we store the packet currently being built */
-#define MAX_PACKET 65535
-static guint8 packet_buf[MAX_PACKET];
+static guint8 packet_buf[WTAP_MAX_PACKET_SIZE_STANDARD];
static guint32 header_length;
static guint32 ip_offset;
static guint32 curr_offset;
-static guint32 max_offset = MAX_PACKET;
+static guint32 max_offset = WTAP_MAX_PACKET_SIZE_STANDARD;
static guint32 packet_start = 0;
-static void start_new_packet(gboolean);
+static int start_new_packet(gboolean);
/* This buffer contains strings present before the packet offset 0 */
#define PACKET_PREAMBLE_MAX_LEN 2048
static guint8* pkt_lnstart;
/* Input file */
-static const char *input_filename;
+static char *input_filename;
static FILE *input_file = NULL;
/* Output file */
-static const char *output_filename;
+static char *output_filename;
static FILE *output_file = NULL;
/* Offset base to parse */
static guint32 offset_base = 16;
-extern FILE *yyin;
+extern FILE *text2pcap_in;
/* ----- State machine -----------------------------------------------------------*/
* Will abort the program if it can't parse the number
* Pass in TRUE if this is an offset, FALSE if not
*/
-static guint32
-parse_num (const char *str, int offset)
+static int
+parse_num(const char *str, int offset, guint32* num)
{
- guint32 num;
char *c;
if (str == NULL) {
fprintf(stderr, "FATAL ERROR: str is NULL\n");
- exit(1);
+ return EXIT_FAILURE;
}
- num = (guint32)strtoul(str, &c, offset ? offset_base : 16);
+ *num = (guint32)strtoul(str, &c, offset ? offset_base : 16);
if (c == str) {
fprintf(stderr, "FATAL ERROR: Bad hex number? [%s]\n", str);
- exit(1);
+ return EXIT_FAILURE;
}
- return num;
+ return EXIT_SUCCESS;
}
/*----------------------------------------------------------------------
* Write this byte into current packet
*/
-static void
-write_byte (const char *str)
+static int
+write_byte(const char *str)
{
guint32 num;
- num = parse_num(str, FALSE);
+ if (parse_num(str, FALSE, &num) != EXIT_SUCCESS)
+ return EXIT_FAILURE;
+
packet_buf[curr_offset] = (guint8) num;
curr_offset++;
if (curr_offset - header_length >= max_offset) /* packet full */
- start_new_packet(TRUE);
+ if (start_new_packet(TRUE) != EXIT_SUCCESS)
+ return EXIT_FAILURE;
+
+ return EXIT_SUCCESS;
}
/*----------------------------------------------------------------------
{
guint32 i;
- if (curr_offset + nbytes < MAX_PACKET) {
+ if (curr_offset + nbytes < WTAP_MAX_PACKET_SIZE_STANDARD) {
for (i = 0; i < nbytes; i++) {
packet_buf[curr_offset] = bytes[i];
curr_offset++;
/*----------------------------------------------------------------------
* Write current packet out
*/
-static void
+static int
write_current_packet (gboolean cont)
{
guint32 length = 0;
if (!success) {
fprintf(stderr, "File write error [%s] : %s\n",
output_filename, g_strerror(err));
- exit(1);
+ return EXIT_FAILURE;
}
if (ts_fmt == NULL) {
/* fake packet counter */
packet_start += curr_offset - header_length;
curr_offset = header_length;
- return;
+ return EXIT_SUCCESS;
}
/*----------------------------------------------------------------------
* Write file header and trailer
*/
-static void
+static int
write_file_header (void)
{
int err;
if (!success) {
fprintf(stderr, "File write error [%s] : %s\n",
output_filename, g_strerror(err));
- exit(1);
+ return EXIT_FAILURE;
}
+
+ return EXIT_SUCCESS;
}
/*----------------------------------------------------------------------
/*----------------------------------------------------------------------
* Start a new packet
*/
-static void
+static int
start_new_packet (gboolean cont)
{
if (debug >= 1)
fprintf(stderr, "Start new packet (cont = %s).\n", cont ? "TRUE" : "FALSE");
/* Write out the current packet, if required */
- write_current_packet(cont);
+ if (write_current_packet(cont) != EXIT_SUCCESS)
+ return EXIT_FAILURE;
num_packets_read++;
/* Ensure we parse the packet preamble as it may contain the time */
parse_preamble();
+
+ return EXIT_SUCCESS;
}
/*----------------------------------------------------------------------
/*----------------------------------------------------------------------
* Parse a single token (called from the scanner)
*/
-void
+int
parse_token (token_t token, char *str)
{
guint32 num;
process_directive(str);
break;
case T_OFFSET:
- num = parse_num(str, TRUE);
+ if (parse_num(str, TRUE, &num) != EXIT_SUCCESS)
+ return EXIT_FAILURE;
if (num == 0) {
/* New packet starts here */
- start_new_packet(FALSE);
+ if (start_new_packet(FALSE) != EXIT_SUCCESS)
+ return EXIT_FAILURE;
state = READ_OFFSET;
pkt_lnstart = packet_buf + num;
}
process_directive(str);
break;
case T_OFFSET:
- num = parse_num(str, TRUE);
+ if (parse_num(str, TRUE, &num) != EXIT_SUCCESS)
+ return EXIT_FAILURE;
if (num == 0) {
/* New packet starts here */
- start_new_packet(FALSE);
+ if (start_new_packet(FALSE) != EXIT_SUCCESS)
+ return EXIT_FAILURE;
packet_start = 0;
state = READ_OFFSET;
} else if ((num - packet_start) != curr_offset - header_length) {
if (debug >= 1)
fprintf(stderr, "Inconsistent offset. Expecting %0X, got %0X. Ignoring rest of packet\n",
curr_offset, num);
- write_current_packet(FALSE);
+ if (write_current_packet(FALSE) != EXIT_SUCCESS)
+ return EXIT_FAILURE;
state = INIT;
}
} else {
/* Record the byte */
state = READ_BYTE;
if (!str) goto fail_null_str;
- write_byte(str);
+ if (write_byte(str) != EXIT_SUCCESS)
+ return EXIT_FAILURE;
break;
case T_TEXT:
case T_DIRECTIVE:
switch (token) {
case T_BYTE:
/* Record the byte */
- write_byte(str);
+ if (write_byte(str) != EXIT_SUCCESS)
+ return EXIT_FAILURE;
break;
case T_TEXT:
case T_DIRECTIVE:
default:
fprintf(stderr, "FATAL ERROR: Bad state (%d)", state);
- exit(1);
+ return EXIT_FAILURE;
}
if (debug >= 2)
fprintf(stderr, ", %s)\n", state_str[state]);
- return;
+ return EXIT_SUCCESS;
fail_null_str:
fprintf(stderr, "FATAL ERROR: got NULL str pointer in state (%d)", state);
- exit(1);
-
+ return EXIT_FAILURE;
}
/*----------------------------------------------------------------------
" -q generate no output at all (automatically disables -d).\n"
" -n use PCAP-NG instead of PCAP as output format.\n"
"",
- MAX_PACKET);
+ WTAP_MAX_PACKET_SIZE_STANDARD);
}
/*----------------------------------------------------------------------
* Parse CLI options
*/
-static void
+static int
parse_options (int argc, char *argv[])
{
GString *comp_info_str;
{"version", no_argument, NULL, 'v'},
{0, 0, 0, 0 }
};
+ struct tm *now_tm;
#ifdef _WIN32
arg_list_utf_16to8(argc, argv);
if (optarg[0] != 'h' && optarg[0] != 'o' && optarg[0] != 'd') {
fprintf(stderr, "Bad argument for '-o': %s\n", optarg);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
switch (optarg[0]) {
case 'o': offset_base = 8; break;
if (sscanf(optarg, "%x", &hdr_ethernet_proto) < 1) {
fprintf(stderr, "Bad argument for '-e': %s\n", optarg);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
break;
hdr_ip_proto > 255) {
fprintf(stderr, "Bad argument for '-i': %s\n", optarg);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
hdr_ethernet = TRUE;
hdr_ethernet_proto = 0x800;
if (p == optarg || (*p != ',' && *p != '\0')) {
fprintf(stderr, "Bad src port for '-%c'\n", c);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
if (*p == '\0') {
fprintf(stderr, "No dest port specified for '-%c'\n", c);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
p++;
optarg = p;
if (p == optarg || (*p != ',' && *p != '\0')) {
fprintf(stderr, "Bad dest port for '-s'\n");
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
if (*p == '\0') {
fprintf(stderr, "No tag specified for '-%c'\n", c);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
p++;
optarg = p;
if (p == optarg || *p != '\0') {
fprintf(stderr, "Bad tag for '-%c'\n", c);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
hdr_ip = TRUE;
if (p == optarg || (*p != ',' && *p != '\0')) {
fprintf(stderr, "Bad src port for '-%c'\n", c);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
if (*p == '\0') {
fprintf(stderr, "No dest port specified for '-%c'\n", c);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
p++;
optarg = p;
if (p == optarg || (*p != ',' && *p != '\0')) {
fprintf(stderr, "Bad dest port for '-s'\n");
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
if (*p == '\0') {
fprintf(stderr, "No ppi specified for '-%c'\n", c);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
p++;
optarg = p;
if (p == optarg || *p != '\0') {
fprintf(stderr, "Bad ppi for '-%c'\n", c);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
hdr_ip = TRUE;
if (p == optarg || (*p != ',' && *p != '\0')) {
fprintf(stderr, "Bad src port for '-u'\n");
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
if (*p == '\0') {
fprintf(stderr, "No dest port specified for '-u'\n");
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
p++;
optarg = p;
if (p == optarg || *p != '\0') {
fprintf(stderr, "Bad dest port for '-u'\n");
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
hdr_ip = TRUE;
hdr_ip_proto = 17;
if (p == optarg || (*p != ',' && *p != '\0')) {
fprintf(stderr, "Bad src port for '-T'\n");
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
if (*p == '\0') {
fprintf(stderr, "No dest port specified for '-u'\n");
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
p++;
optarg = p;
if (p == optarg || *p != '\0') {
fprintf(stderr, "Bad dest port for '-T'\n");
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
hdr_ip = TRUE;
hdr_ip_proto = 6;
if (!p) {
fprintf(stderr, "Bad source param addr for '-%c'\n", c);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
*p = '\0';
if (!ws_inet_pton6(optarg, &hdr_ipv6_src_addr)) {
fprintf(stderr, "Bad src addr -%c '%s'\n", c, p);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
} else {
if (!ws_inet_pton4(optarg, &hdr_ip_src_addr)) {
fprintf(stderr, "Bad src addr -%c '%s'\n", c, p);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
}
if (*p == '\0') {
fprintf(stderr, "No dest addr specified for '-%c'\n", c);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
if (hdr_ipv6 == TRUE) {
if (!ws_inet_pton6(p, &hdr_ipv6_dest_addr)) {
fprintf(stderr, "Bad dest addr for -%c '%s'\n", c, p);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
} else {
if (!ws_inet_pton4(p, &hdr_ip_dest_addr)) {
fprintf(stderr, "Bad dest addr for -%c '%s'\n", c, p);
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
}
break;
case '?':
default:
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
}
}
if (optind >= argc || argc-optind < 2) {
fprintf(stderr, "Must specify input and output filename\n");
print_usage(stderr);
- exit(1);
+ return EXIT_FAILURE;
+ }
+
+ if (max_offset > WTAP_MAX_PACKET_SIZE_STANDARD) {
+ fprintf(stderr, "Maximum packet length cannot be more than %d bytes\n",
+ WTAP_MAX_PACKET_SIZE_STANDARD);
+ return EXIT_FAILURE;
}
- if (strcmp(argv[optind], "-")) {
- input_filename = g_strdup(argv[optind]);
+ if (strcmp(argv[optind], "-") != 0) {
+ input_filename = argv[optind];
input_file = ws_fopen(input_filename, "rb");
if (!input_file) {
fprintf(stderr, "Cannot open file [%s] for reading: %s\n",
input_filename, g_strerror(errno));
- exit(1);
+ return EXIT_FAILURE;
}
} else {
input_filename = "Standard input";
input_file = stdin;
}
- if (strcmp(argv[optind+1], "-")) {
- output_filename = g_strdup(argv[optind+1]);
+ if (strcmp(argv[optind+1], "-") != 0) {
+ /* Write to a file. Open the file, in binary mode. */
+ output_filename = argv[optind+1];
output_file = ws_fopen(output_filename, "wb");
if (!output_file) {
fprintf(stderr, "Cannot open file [%s] for writing: %s\n",
output_filename, g_strerror(errno));
- exit(1);
+ return EXIT_FAILURE;
}
} else {
+ /* Write to the standard output. */
+#ifdef _WIN32
+ /* Put the standard output in binary mode. */
+ if (_setmode(1, O_BINARY) == -1) {
+ /* "Should not happen" */
+ fprintf(stderr, "Cannot put standard output in binary mode: %s\n",
+ g_strerror(errno));
+ return EXIT_FAILURE;
+ }
+#endif
output_filename = "Standard output";
output_file = stdout;
}
/* Some validation */
if (pcap_link_type != 1 && hdr_ethernet) {
fprintf(stderr, "Dummy headers (-e, -i, -u, -s, -S -T) cannot be specified with link type override (-l)\n");
- exit(1);
+ return EXIT_FAILURE;
}
/* Set up our variables */
}
ts_sec = time(0); /* initialize to current time */
- /* We trust the OS to return a time after the Epoch. */
- timecode_default = *localtime(&ts_sec);
+ now_tm = localtime(&ts_sec);
+ if (now_tm == NULL) {
+ /*
+ * This shouldn't happen - on UN*X, this should Just Work, and
+ * on Windows, it won't work if ts_sec is before the Epoch,
+ * but it's long after 1970, so....
+ */
+ fprintf(stderr, "localtime(right now) failed\n");
+ return EXIT_FAILURE;
+ }
+ timecode_default = *now_tm;
timecode_default.tm_isdst = -1; /* Unknown for now, depends on time given to the strptime() function */
/* Display summary of our state */
if (hdr_data_chunk) fprintf(stderr, "Generate dummy DATA chunk header: TSN: %u. SID: %u. SSN: %u. PPID: %u\n",
hdr_data_chunk_tsn, hdr_data_chunk_sid, hdr_data_chunk_ssn, hdr_data_chunk_ppid);
}
+
+ return EXIT_SUCCESS;
}
int
main(int argc, char *argv[])
{
- parse_options(argc, argv);
+ int ret = EXIT_SUCCESS;
+
+ if (parse_options(argc, argv) != EXIT_SUCCESS) {
+ ret = EXIT_FAILURE;
+ goto clean_exit;
+ }
assert(input_file != NULL);
assert(output_file != NULL);
- write_file_header();
+ if (write_file_header() != EXIT_SUCCESS) {
+ ret = EXIT_FAILURE;
+ goto clean_exit;
+ }
header_length = 0;
if (hdr_ethernet) {
}
curr_offset = header_length;
- yyin = input_file;
- yylex();
-
- write_current_packet(FALSE);
- fclose(input_file);
- fclose(output_file);
+ text2pcap_in = input_file;
+ if (text2pcap_lex() == EXIT_SUCCESS) {
+ if (write_current_packet(FALSE) != EXIT_SUCCESS)
+ ret = EXIT_FAILURE;
+ } else {
+ ret = EXIT_FAILURE;
+ }
if (debug)
fprintf(stderr, "\n-------------------------\n");
if (!quiet) {
num_packets_written, (num_packets_written == 1) ? "" : "s",
bytes_written, (bytes_written == 1) ? "" : "s");
}
- return 0;
+clean_exit:
+ text2pcap_lex_destroy();
+ if (input_file) {
+ fclose(input_file);
+ }
+ if (output_file) {
+ fclose(output_file);
+ }
+ return ret;
}
/*
git.samba.org / metze / wireshark / wip.git / blobdiff