void timer_init(void)
{
#if defined(ENABLE_RDTSC)
rdtsc_calibrate();
#endif
time(&start_time);
}
/**
* \brief Does all the prep work before calling tcpreplay_replay()
*
* Technically this validates our config options, preloads the tcpprep
* cache file, loads the packet cache and anything else which might
* cause a delay for starting to send packets with tcpreplay_replay()
*/
int
tcpreplay_prepare(tcpreplay_t *ctx)
{
char *intname, ebuf[SENDPACKET_ERRBUF_SIZE];
int int1dlt, int2dlt, i;
assert(ctx);
/*
* First, process the validations, basically the same we do in
* tcpreplay_post_args() and AutoOpts
*/
if (ctx->options->intf1_name == NULL) {
tcpreplay_seterr(ctx, "%s", "You must specify at least one network interface");
return -1;
}
if (ctx->options->source_cnt == 0) {
tcpreplay_seterr(ctx, "%s", "You must specify at least one source pcap");
return -1;
}
if (ctx->options->dualfile) {
if (! ctx->options->source_cnt >= 2) {
tcpreplay_seterr(ctx, "%s", "Dual file mode requires 2 or more pcap files");
return -1;
}
if (ctx->options->source_cnt % 2 != 0) {
tcpreplay_seterr(ctx, "%s", "Dual file mode requires an even number of pcap files");
return -1;
}
}
if (ctx->options->dualfile && ctx->options->cachedata != NULL) {
tcpreplay_seterr(ctx, "%s", "Can't use dual file mode and tcpprep cache file together");
return -1;
}
if ((ctx->options->dualfile || ctx->options->cachedata != NULL) &&
ctx->options->intf2_name == NULL) {
tcpreplay_seterr(ctx, "%s", "dual file mode and tcpprep cache files require two interfaces");
}
#ifndef HAVE_SELECT
if (ctx->options->accurate == accurate_select) {
tcpreplay_seterr(ctx, "%s", "tcpreplay_api not compiled with select support");
return -1;
}
#endif
#ifndef HAVE_RDTSC
if (ctx->options->accurate == accurate_rdtsc) {
tcpreplay_seterr(ctx, "%s", "tcpreplay_api not compiled with rdtsc support");
return -1;
}
#else
if (ctx->options->rdtsc_clicks > 0)
rdtsc_calibrate(ctx->options->rdtsc_clicks);
#endif
#ifndef HAVE_IOPERM
if (ctx->options->accurate == accurate_ioport) {
tcpreplay_seterr(ctx, "%s", "tcpreplay_api not compiled with IO Port 0x80 support");
return -1;
}
#else
if (ctx->options->accurate == accurate_ioport) {
ioport_sleep_init();
}
#endif
#ifndef HAVE_ABSOLUTE_TIME
if (ctx->options->accurate == accurate_abs_time) {
tcpreplay_seterr(ctx, "%s", "tcpreplay_api only supports absolute time on Apple OS X");
return -1;
}
#endif
if ((intname = get_interface(ctx->intlist, ctx->options->intf1_name)) == NULL) {
tcpreplay_seterr(ctx, "Invalid interface name/alias: %s", OPT_ARG(INTF1));
return -1;
}
/* open interfaces for writing */
if ((ctx->intf1 = sendpacket_open(ctx->options->intf1_name, ebuf, TCPR_DIR_C2S)) == NULL) {
tcpreplay_seterr(ctx, "Can't open %s: %s", ctx->options->intf1_name, ebuf);
return -1;
}
int1dlt = sendpacket_get_dlt(ctx->intf1);
if (ctx->options->intf2_name != NULL) {
if ((intname = get_interface(ctx->intlist, ctx->options->intf2_name)) == NULL) {
tcpreplay_seterr(ctx, "Invalid interface name/alias: %s", OPT_ARG(INTF2));
return -1;
}
/* open interfaces for writing */
if ((ctx->intf2 = sendpacket_open(ctx->options->intf2_name, ebuf, TCPR_DIR_C2S)) == NULL) {
tcpreplay_seterr(ctx, "Can't open %s: %s", ctx->options->intf2_name, ebuf);
return -1;
}
int2dlt = sendpacket_get_dlt(ctx->intf2);
if (int2dlt != int1dlt) {
tcpreplay_seterr(ctx, "DLT type missmatch for %s (%s) and %s (%s)",
ctx->options->intf1_name, pcap_datalink_val_to_name(int1dlt),
ctx->options->intf2_name, pcap_datalink_val_to_name(int2dlt));
return -1;
}
}
/*
* Setup up the file cache, if required
*/
if (ctx->options->enable_file_cache && ctx->options->file_cache == NULL) {
/* Initialise each of the file cache structures */
for (i = 0; i < ctx->options->source_cnt; i++) {
ctx->options->file_cache[i].index = i;
ctx->options->file_cache[i].cached = FALSE;
ctx->options->file_cache[i].packet_cache = NULL;
}
}
return 0;
}
/**
* \brief Parses the GNU AutoOpts options for tcpreplay
*
* If you're using AutoOpts with tcpreplay_api, then just call this after
* optionProcess() and it will parse all the options for you. As always,
* returns 0 on success, and -1 on error & -2 on warning.
*/
int
tcpreplay_post_args(tcpreplay_t *ctx, int argc)
{
char *temp, *intname;
char ebuf[SENDPACKET_ERRBUF_SIZE];
int int1dlt, int2dlt;
tcpreplay_opt_t *options;
int warn = 0;
#ifdef USE_AUTOOPTS
options = ctx->options;
#ifdef DEBUG
if (HAVE_OPT(DBUG))
debug = OPT_VALUE_DBUG;
#else
if (HAVE_OPT(DBUG)) {
warn ++;
tcpreplay_setwarn(ctx, "%s", "not configured with --enable-debug. Debugging disabled.");
}
#endif
options->loop = OPT_VALUE_LOOP;
options->sleep_accel = OPT_VALUE_SLEEP_ACCEL;
if (HAVE_OPT(LIMIT))
options->limit_send = OPT_VALUE_LIMIT;
if (HAVE_OPT(TOPSPEED)) {
options->speed.mode = speed_topspeed;
options->speed.speed = 0.0;
} else if (HAVE_OPT(PPS)) {
options->speed.mode = speed_packetrate;
options->speed.speed = (float)OPT_VALUE_PPS;
options->speed.pps_multi = OPT_VALUE_PPS_MULTI;
} else if (HAVE_OPT(ONEATATIME)) {
options->speed.mode = speed_oneatatime;
options->speed.speed = 0.0;
} else if (HAVE_OPT(MBPS)) {
options->speed.mode = speed_mbpsrate;
options->speed.speed = atof(OPT_ARG(MBPS));
} else if (HAVE_OPT(MULTIPLIER)) {
options->speed.mode = speed_multiplier;
options->speed.speed = atof(OPT_ARG(MULTIPLIER));
}
#ifdef ENABLE_VERBOSE
if (HAVE_OPT(VERBOSE))
options->verbose = 1;
if (HAVE_OPT(DECODE))
options->tcpdump->args = safe_strdup(OPT_ARG(DECODE));
#endif
if (HAVE_OPT(STATS))
options->stats = OPT_VALUE_STATS;
/*
* Check if the file cache should be enabled - if we're looping more than
* once and the command line option has been spec'd
*/
if (HAVE_OPT(ENABLE_FILE_CACHE) && (options->loop != 1)) {
options->enable_file_cache = true;
}
/*
* If we're preloading the pcap before the first run, then
* we're forcing the file cache to be true
*/
if (HAVE_OPT(PRELOAD_PCAP)) {
options->preload_pcap = true;
options->enable_file_cache = true;
}
/* Dual file mode */
if (HAVE_OPT(DUALFILE)) {
options->dualfile = true;
if (argc < 2) {
tcpreplay_seterr(ctx, "%s", "--dualfile mode requires at least two pcap files");
return -1;
}
if (argc % 2 != 0) {
tcpreplay_seterr(ctx, "%s", "--dualfile mode requires an even number of pcap files");
return -1;
}
}
if (HAVE_OPT(TIMER)) {
if (strcmp(OPT_ARG(TIMER), "select") == 0) {
#ifdef HAVE_SELECT
options->accurate = accurate_select;
#else
tcpreplay_seterr(ctx, "%s", "tcpreplay_api not compiled with select support");
return -1;
#endif
} else if (strcmp(OPT_ARG(TIMER), "rdtsc") == 0) {
#ifdef HAVE_RDTSC
options->accurate = accurate_rdtsc;
#else
tcpreplay_seterr(ctx, "%s", "tcpreplay_api not compiled with rdtsc support");
return -1;
#endif
} else if (strcmp(OPT_ARG(TIMER), "ioport") == 0) {
#if defined HAVE_IOPERM && defined(__i386__)
options->accurate = accurate_ioport;
ioport_sleep_init();
#else
tcpreplay_seterr(ctx, "%s", "tcpreplay_api not compiled with IO Port 0x80 support");
return -1;
#endif
} else if (strcmp(OPT_ARG(TIMER), "gtod") == 0) {
options->accurate = accurate_gtod;
} else if (strcmp(OPT_ARG(TIMER), "nano") == 0) {
options->accurate = accurate_nanosleep;
} else if (strcmp(OPT_ARG(TIMER), "abstime") == 0) {
#ifdef HAVE_ABSOLUTE_TIME
options->accurate = accurate_abs_time;
if (!MPLibraryIsLoaded()) {
tcpreplay_seterr(ctx, "%s", "The MP library did not load.\n");
return -1;
}
#else
tcpreplay_seterr(ctx, "%s", "tcpreplay_api only supports absolute time on Apple OS X");
return -1;
#endif
} else {
tcpreplay_seterr(ctx, "Unsupported timer mode: %s", OPT_ARG(TIMER));
return -1;
}
}
#ifdef HAVE_RDTSC
if (HAVE_OPT(RDTSC_CLICKS)) {
rdtsc_calibrate(OPT_VALUE_RDTSC_CLICKS);
}
#endif
if (HAVE_OPT(PKTLEN)) {
options->use_pkthdr_len = true;
warn ++;
tcpreplay_setwarn(ctx, "%s", "--pktlen may cause problems. Use with caution.");
}
if ((intname = get_interface(ctx->intlist, OPT_ARG(INTF1))) == NULL) {
tcpreplay_seterr(ctx, "Invalid interface name/alias: %s", OPT_ARG(INTF1));
return -1;
}
options->intf1_name = safe_strdup(intname);
/* open interfaces for writing */
if ((ctx->intf1 = sendpacket_open(options->intf1_name, ebuf, TCPR_DIR_C2S)) == NULL) {
tcpreplay_seterr(ctx, "Can't open %s: %s", options->intf1_name, ebuf);
return -1;
}
int1dlt = sendpacket_get_dlt(ctx->intf1);
if (HAVE_OPT(INTF2)) {
if ((intname = get_interface(ctx->intlist, OPT_ARG(INTF2))) == NULL) {
tcpreplay_seterr(ctx, "Invalid interface name/alias: %s", OPT_ARG(INTF2));
return -1;
}
options->intf2_name = safe_strdup(intname);
/* open interface for writing */
if ((ctx->intf2 = sendpacket_open(options->intf2_name, ebuf, TCPR_DIR_S2C)) == NULL) {
tcpreplay_seterr(ctx, "Can't open %s: %s", options->intf2_name, ebuf);
}
int2dlt = sendpacket_get_dlt(ctx->intf2);
if (int2dlt != int1dlt) {
tcpreplay_seterr(ctx, "DLT type missmatch for %s (%s) and %s (%s)",
options->intf1_name, pcap_datalink_val_to_name(int1dlt),
options->intf2_name, pcap_datalink_val_to_name(int2dlt));
return -1;
}
}
if (HAVE_OPT(CACHEFILE)) {
temp = safe_strdup(OPT_ARG(CACHEFILE));
options->cache_packets = read_cache(&options->cachedata, temp,
&options->comment);
safe_free(temp);
}
/* return -2 on warnings */
if (warn > 0)
return -2;
return 0;
#else
tcpreplay_seterr(ctx, "autopts support not compiled in. tcpreplay_post_args() not supported");
return -1;
#endif /* USE_AUTOOPTS */
}
