12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466 |
- /* $Id$ */
- /*
- * Copyright (c) 2001-2010 Aaron Turner <aturner at synfin dot net>
- * Copyright (c) 2013-2024 Fred Klassen <tcpreplay at appneta dot com> - AppNeta
- *
- * The Tcpreplay Suite of tools 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 3 of the
- * License, or with the authors permission any later version.
- *
- * The Tcpreplay Suite 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 the Tcpreplay Suite. If not, see <http://www.gnu.org/licenses/>.
- */
- #include "tcpreplay_api.h"
- #include "defines.h"
- #include "config.h"
- #include "common.h"
- #include "replay.h"
- #include "send_packets.h"
- #include "sleep.h"
- #include <ctype.h>
- #include <errno.h>
- #include <fcntl.h>
- #include <stdarg.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <sys/types.h>
- #include <time.h>
- #include <unistd.h>
- #ifdef TCPREPLAY_EDIT
- #include "tcpreplay_edit_opts.h"
- #else
- #include "tcpreplay_opts.h"
- #endif
- /**
- * \brief Returns a string describing the last error.
- *
- * Value when the last call does not result in an error is undefined
- * (may be NULL, may be garbage)
- */
- char *
- tcpreplay_geterr(tcpreplay_t *ctx)
- {
- assert(ctx);
- return(ctx->errstr);
- }
- /**
- * \brief Returns a string describing the last warning.
- *
- * Value when the last call does not result in an warning is undefined
- * (may be NULL, may be garbage)
- */
- char *
- tcpreplay_getwarn(tcpreplay_t *ctx)
- {
- assert(ctx);
- return(ctx->warnstr);
- }
- /**
- * \brief Initialize a new tcpreplay context
- *
- * Allocates memory and stuff like that. Always returns a buffer or completely
- * fails by calling exit() on malloc failure.
- */
- tcpreplay_t *
- tcpreplay_init()
- {
- tcpreplay_t *ctx;
- /* allocations will reset everything to zeros */
- ctx = safe_malloc(sizeof(tcpreplay_t));
- ctx->options = safe_malloc(sizeof(tcpreplay_opt_t));
- /* replay packets only once */
- ctx->options->loop = 1;
- /* Default mode is to replay pcap once in real-time */
- ctx->options->speed.mode = speed_multiplier;
- ctx->options->speed.multiplier = 1.0;
- /* Set the default timing method */
- ctx->options->accurate = accurate_gtod;
- /* set the default MTU size */
- ctx->options->mtu = DEFAULT_MTU;
- /* disable periodic statistics */
- ctx->options->stats = -1;
- /* disable limit send */
- ctx->options->limit_send = -1;
- /* default unique-loops */
- ctx->options->unique_loops = 1.0;
- #ifdef ENABLE_VERBOSE
- /* clear out tcpdump struct */
- ctx->options->tcpdump = (tcpdump_t *)safe_malloc(sizeof(tcpdump_t));
- #endif
- if (fcntl(STDERR_FILENO, F_SETFL, O_NONBLOCK) < 0)
- tcpreplay_setwarn(ctx, "Unable to set STDERR to non-blocking: %s", strerror(errno));
- #ifdef ENABLE_PCAP_FINDALLDEVS
- ctx->intlist = get_interface_list();
- #else
- ctx->intlist = NULL;
- #endif
- /* set up flows - on by default*/
- ctx->options->flow_stats = 1;
- ctx->flow_hash_table = flow_hash_table_init(DEFAULT_FLOW_HASH_BUCKET_SIZE);
- ctx->sp_type = SP_TYPE_NONE;
- ctx->intf1dlt = -1;
- ctx->intf2dlt = -1;
- ctx->abort = false;
- ctx->first_time = true;
- return ctx;
- }
- /**
- * \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;
- tcpreplay_opt_t *options;
- int warn = 0;
- float n;
- int ret = 0;
- options = ctx->options;
- dbg(2, "tcpreplay_post_args: parsing command arguments");
- ebuf = safe_malloc(SENDPACKET_ERRBUF_SIZE);
- #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->loopdelay_ms = OPT_VALUE_LOOPDELAY_MS;
- options->loopdelay_ns = OPT_VALUE_LOOPDELAY_NS;
- if (HAVE_OPT(LIMIT))
- options->limit_send = OPT_VALUE_LIMIT;
- if (HAVE_OPT(DURATION))
- options->limit_time = OPT_VALUE_DURATION;
- if (HAVE_OPT(TOPSPEED)) {
- options->speed.mode = speed_topspeed;
- options->speed.speed = 0;
- } else if (HAVE_OPT(PPS)) {
- n = atof(OPT_ARG(PPS));
- if (!n) {
- tcpreplay_seterr(ctx, "invalid pps value '%s'", OPT_ARG(PPS));
- ret = -1;
- goto out;
- }
- options->speed.speed = (COUNTER)(n * 60.0 * 60.0); /* convert to packets per hour */
- options->speed.mode = speed_packetrate;
- options->speed.pps_multi = OPT_VALUE_PPS_MULTI;
- } else if (HAVE_OPT(ONEATATIME)) {
- options->speed.mode = speed_oneatatime;
- options->speed.speed = 0;
- } else if (HAVE_OPT(MBPS)) {
- n = atof(OPT_ARG(MBPS));
- if (n) {
- options->speed.mode = speed_mbpsrate;
- options->speed.speed = (COUNTER)(n * 1000000.0); /* convert to bps */
- } else {
- options->speed.mode = speed_topspeed;
- options->speed.speed = 0;
- }
- } else if (HAVE_OPT(MULTIPLIER)) {
- options->speed.mode = speed_multiplier;
- options->speed.multiplier = atof(OPT_ARG(MULTIPLIER));
- }
- if (HAVE_OPT(MAXSLEEP)) {
- options->maxsleep.tv_sec = OPT_VALUE_MAXSLEEP / 1000;
- options->maxsleep.tv_nsec = (OPT_VALUE_MAXSLEEP % 1000) * 1000 * 1000;
- }
- if (HAVE_OPT(SUPPRESS_WARNINGS))
- print_warnings = 0;
- #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;
- /*
- * preloading the pcap before the first run
- */
- if (HAVE_OPT(PRELOAD_PCAP)) {
- options->preload_pcap = true;
- }
- #ifdef TCPREPLAY_EDIT
- if (HAVE_OPT(PRELOAD_PCAP) && OPT_VALUE_LOOP > 1) {
- tcpreplay_seterr(ctx,
- "%s",
- "tcpreplay_edit --loop (-l) and --preload_pcap (-K) options are mutually exclusive");
- ret = -1;
- goto out;
- }
- #endif
- /* 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");
- ret = -1;
- goto out;
- }
- if (argc % 2 != 0) {
- tcpreplay_seterr(ctx, "%s", "--dualfile mode requires an even number of pcap files");
- ret = -1;
- goto out;
- }
- }
- #ifdef HAVE_NETMAP
- options->netmap_delay = OPT_VALUE_NM_DELAY;
- #endif
- if (HAVE_OPT(NETMAP)) {
- #ifdef HAVE_NETMAP
- options->netmap = 1;
- ctx->sp_type = SP_TYPE_NETMAP;
- #else
- err(-1, "--netmap feature was not compiled in. See INSTALL.");
- #endif
- }
- if (HAVE_OPT(XDP)) {
- #ifdef HAVE_LIBXDP
- options->xdp = 1;
- ctx->sp_type = SP_TYPE_LIBXDP;
- #else
- err(-1, "--xdp feature was not compiled in. See INSTALL.");
- #endif
- }
- if (HAVE_OPT(UNIQUE_IP))
- options->unique_ip = 1;
- if (HAVE_OPT(UNIQUE_IP_LOOPS)) {
- options->unique_loops = atof(OPT_ARG(UNIQUE_IP_LOOPS));
- if (options->unique_loops < 1.0) {
- tcpreplay_seterr(ctx, "%s", "--unique-ip-loops requires loop count >= 1.0");
- ret = -1;
- goto out;
- }
- }
- /* flow statistics */
- if (HAVE_OPT(NO_FLOW_STATS))
- options->flow_stats = 0;
- if (HAVE_OPT(FLOW_EXPIRY)) {
- options->flow_expiry = OPT_VALUE_FLOW_EXPIRY;
- }
- 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");
- ret = -1;
- goto out;
- #endif
- } else if (strcmp(OPT_ARG(TIMER), "ioport") == 0) {
- #if defined HAVE_IOPORT_SLEEP__
- options.accurate = ACCURATE_IOPORT;
- ioport_sleep_init();
- #else
- err(-1, "tcpreplay not compiled with IO Port 0x80 support");
- #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) {
- tcpreplay_seterr(ctx, "%s", "abstime is deprecated");
- ret = -1;
- goto out;
- } else {
- tcpreplay_seterr(ctx, "Unsupported timer mode: %s", OPT_ARG(TIMER));
- ret = -1;
- goto out;
- }
- }
- #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.");
- }
- switch (WHICH_IDX_INTF1) {
- case INDEX_OPT_WRITE:
- options->intf1_name = safe_strdup(OPT_ARG(INTF1));
- ctx->sp_type = SP_TYPE_LIBPCAP_DUMP;
- /* open interfaces for writing */
- if ((ctx->intf1 = sendpacket_open(options->intf1_name, ebuf, TCPR_DIR_C2S, ctx->sp_type, ctx)) == NULL) {
- tcpreplay_seterr(ctx, "Can't open %s: %s", options->intf1_name, ebuf);
- ret = -1;
- goto out;
- }
- break;
- case INDEX_OPT_INTF1:
- if ((intname = get_interface(ctx->intlist, OPT_ARG(INTF1))) == NULL) {
- if (!strncmp(OPT_ARG(INTF1), "netmap:", 7) || !strncmp(OPT_ARG(INTF1), "vale", 4)) {
- tcpreplay_seterr(
- ctx,
- "Unable to connect to netmap interface %s. Ensure netmap module is installed (see INSTALL).",
- OPT_ARG(INTF1));
- } else {
- tcpreplay_seterr(ctx, "Invalid interface name/alias: %s", OPT_ARG(INTF1));
- }
- ret = -1;
- goto out;
- }
- if (!strncmp(intname, "netmap:", 7) || !strncmp(intname, "vale:", 5)) {
- #ifdef HAVE_NETMAP
- options->netmap = 1;
- ctx->sp_type = SP_TYPE_NETMAP;
- #else
- tcpreplay_seterr(ctx, "%s", "tcpreplay_api not compiled with netmap support");
- ret = -1;
- goto out;
- #endif
- }
- options->intf1_name = safe_strdup(intname);
- /* open interfaces for writing */
- if ((ctx->intf1 = sendpacket_open(options->intf1_name, ebuf, TCPR_DIR_C2S, ctx->sp_type, ctx)) == NULL) {
- tcpreplay_seterr(ctx, "Can't open %s: %s", options->intf1_name, ebuf);
- ret = -1;
- goto out;
- }
- #ifdef HAVE_LIBXDP
- ctx->intf1->batch_size = OPT_VALUE_XDP_BATCH_SIZE;
- #endif
- #if defined HAVE_NETMAP
- ctx->intf1->netmap_delay = ctx->options->netmap_delay;
- #endif
- ctx->intf1dlt = sendpacket_get_dlt(ctx->intf1);
- if (HAVE_OPT(INTF2)) {
- if (!HAVE_OPT(CACHEFILE) && !HAVE_OPT(DUALFILE)) {
- tcpreplay_seterr(ctx, "--intf2=%s requires either --cachefile or --dualfile", OPT_ARG(INTF2));
- ret = -1;
- goto out;
- }
- if ((intname = get_interface(ctx->intlist, OPT_ARG(INTF2))) == NULL) {
- tcpreplay_seterr(ctx, "Invalid interface name/alias: %s", OPT_ARG(INTF2));
- ret = -1;
- goto out;
- }
- options->intf2_name = safe_strdup(intname);
- /* open interface for writing */
- if ((ctx->intf2 = sendpacket_open(options->intf2_name, ebuf, TCPR_DIR_S2C, ctx->sp_type, ctx)) == NULL) {
- tcpreplay_seterr(ctx, "Can't open %s: %s", options->intf2_name, ebuf);
- }
- #if defined HAVE_NETMAP
- ctx->intf2->netmap_delay = ctx->options->netmap_delay;
- #endif
- ctx->intf2dlt = sendpacket_get_dlt(ctx->intf2);
- if (ctx->intf2dlt != ctx->intf1dlt) {
- tcpreplay_seterr(ctx,
- "DLT type mismatch for %s (%s) and %s (%s)",
- options->intf1_name,
- pcap_datalink_val_to_name(ctx->intf1dlt),
- options->intf2_name,
- pcap_datalink_val_to_name(ctx->intf2dlt));
- ret = -1;
- goto out;
- }
- }
- break;
- default:
- assert(false); // shouldn't happen!
- }
- 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)
- ret = -2;
- out:
- safe_free(ebuf);
- return ret;
- }
- /**
- * Closes & free's all memory related to a tcpreplay context
- */
- void
- tcpreplay_close(tcpreplay_t *ctx)
- {
- tcpreplay_opt_t *options;
- interface_list_t *intlist, *intlistnext;
- packet_cache_t *packet_cache, *next;
- assert(ctx);
- assert(ctx->options);
- options = ctx->options;
- #ifdef HAVE_LIBXDP
- if (ctx->intf1->handle_type == SP_TYPE_LIBXDP) {
- free_umem_and_xsk(ctx->intf1);
- if (ctx->intf2) {
- free_umem_and_xsk(ctx->intf2);
- }
- }
- #endif
- safe_free(options->intf1_name);
- safe_free(options->intf2_name);
- sendpacket_close(ctx->intf1);
- if (ctx->intf2 != NULL)
- sendpacket_close(ctx->intf2);
- safe_free(options->cachedata);
- safe_free(options->comment);
- #ifdef ENABLE_VERBOSE
- safe_free(options->tcpdump_args);
- tcpdump_close(options->tcpdump);
- #endif
- /* free the flow hash table */
- flow_hash_table_release(ctx->flow_hash_table);
- /* free the file cache */
- packet_cache = options->file_cache->packet_cache;
- while (packet_cache != NULL) {
- next = packet_cache->next;
- safe_free(packet_cache->pktdata);
- safe_free(packet_cache);
- packet_cache = next;
- }
- /* free our interface list */
- if (ctx->intlist != NULL) {
- intlist = ctx->intlist;
- while (intlist != NULL) {
- intlistnext = intlist->next;
- safe_free(intlist);
- intlist = intlistnext;
- }
- }
- /* free --include / --exclude list */
- free_list(options->list);
- }
- /**
- * \brief Specifies an interface to use for sending.
- *
- * You may call this up to two (2) times with different interfaces
- * when using a tcpprep cache file or dualfile mode. Note, both interfaces
- * must use the same DLT type
- */
- int
- tcpreplay_set_interface(tcpreplay_t *ctx, tcpreplay_intf intf, char *value)
- {
- static int int1dlt = -1, int2dlt = -1;
- char *intname;
- char *ebuf;
- int ret = 0;
- assert(ctx);
- assert(value);
- ebuf = safe_malloc(SENDPACKET_ERRBUF_SIZE);
- if (intf == intf1) {
- if ((intname = get_interface(ctx->intlist, value)) == NULL) {
- if (!strncmp(OPT_ARG(INTF1), "netmap:", 7) || !strncmp(OPT_ARG(INTF1), "vale", 4))
- tcpreplay_seterr(ctx, "Unable to connect to netmap interface %s. Ensure netmap module is installed (see INSTALL).",
- value);
- else
- tcpreplay_seterr(ctx, "Invalid interface name/alias: %s", value);
- ret = -1;
- goto out;
- }
- ctx->options->intf1_name = safe_strdup(intname);
- /* open interfaces for writing */
- if ((ctx->intf1 = sendpacket_open(ctx->options->intf1_name, ebuf, TCPR_DIR_C2S, ctx->sp_type, ctx)) == NULL) {
- tcpreplay_seterr(ctx, "Can't open %s: %s", ctx->options->intf1_name, ebuf);
- ret = -1;
- goto out;
- }
- int1dlt = sendpacket_get_dlt(ctx->intf1);
- } else if (intf == intf2) {
- if ((intname = get_interface(ctx->intlist, value)) == NULL) {
- tcpreplay_seterr(ctx, "Invalid interface name/alias: %s", ctx->options->intf2_name);
- ret = -1;
- goto out;
- }
- ctx->options->intf2_name = safe_strdup(intname);
- /* open interface for writing */
- if ((ctx->intf2 = sendpacket_open(ctx->options->intf2_name, ebuf, TCPR_DIR_S2C, ctx->sp_type, ctx)) == NULL) {
- tcpreplay_seterr(ctx, "Can't open %s: %s", ctx->options->intf2_name, ebuf);
- ret = -1;
- goto out;
- }
- int2dlt = sendpacket_get_dlt(ctx->intf2);
- }
- /*
- * If both interfaces are selected, then make sure both interfaces use
- * the same DLT type
- */
- if (int1dlt != -1 && int2dlt != -1) {
- if (int1dlt != int2dlt) {
- tcpreplay_seterr(ctx, "DLT type mismatch 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));
- ret = -1;
- goto out;
- }
- }
- out:
- safe_free(ebuf);
- return ret;
- }
- /**
- * Set the replay speed mode.
- */
- int
- tcpreplay_set_speed_mode(tcpreplay_t *ctx, tcpreplay_speed_mode value)
- {
- assert(ctx);
- ctx->options->speed.mode = value;
- return 0;
- }
- /**
- * Set the approprate speed value. Value is interpreted based on
- * how tcpreplay_set_speed_mode() value
- */
- int
- tcpreplay_set_speed_speed(tcpreplay_t *ctx, COUNTER value)
- {
- assert(ctx);
- ctx->options->speed.speed = value;
- return 0;
- }
- /**
- * Sending under packets/sec requires an integer value, not float.
- * you must first call tcpreplay_set_speed_mode(ctx, speed_packetrate)
- */
- int
- tcpreplay_set_speed_pps_multi(tcpreplay_t *ctx, int value)
- {
- assert(ctx);
- ctx->options->speed.pps_multi = value;
- return 0;
- }
- /**
- * How many times should we loop through all the pcap files?
- */
- int
- tcpreplay_set_loop(tcpreplay_t *ctx, u_int32_t value)
- {
- assert(ctx);
- ctx->options->loop = value;
- return 0;
- }
- /**
- * Set the unique IP address flag
- */
- int
- tcpreplay_set_unique_ip(tcpreplay_t *ctx, bool value)
- {
- assert(ctx);
- ctx->options->unique_ip = value;
- return 0;
- }
- int
- tcpreplay_set_unique_ip_loops(tcpreplay_t *ctx, int value)
- {
- assert(ctx);
- ctx->options->unique_loops = value;
- return 0;
- }
- /**
- * Set netmap mode
- */
- int
- tcpreplay_set_netmap(_U_ tcpreplay_t *ctx, _U_ bool value)
- {
- assert(ctx);
- #ifdef HAVE_NETMAP
- ctx->options->netmap = value;
- return 0;
- #else
- warn("netmap not compiled in");
- return -1;
- #endif
- }
- /**
- * Tell tcpreplay to ignore the snaplen (default) and use the "actual"
- * packet len instead
- */
- int
- tcpreplay_set_use_pkthdr_len(tcpreplay_t *ctx, bool value)
- {
- assert(ctx);
- ctx->options->use_pkthdr_len = value;
- return 0;
- }
- /**
- * Override the outbound MTU
- */
- int
- tcpreplay_set_mtu(tcpreplay_t *ctx, int value)
- {
- assert(ctx);
- ctx->options->mtu = value;
- return 0;
- }
- /**
- * Sets the accurate timing mode
- */
- int
- tcpreplay_set_accurate(tcpreplay_t *ctx, tcpreplay_accurate value)
- {
- assert(ctx);
- ctx->options->accurate = value;
- return 0;
- }
- /**
- * Sets the number of seconds between printing stats
- */
- int
- tcpreplay_set_stats(tcpreplay_t *ctx, int value)
- {
- assert(ctx);
- ctx->options->stats = value;
- return 0;
- }
- /**
- * \brief Enable or disable dual file mode
- *
- * In dual file mode, we read two files at the same time and use
- * one file for each interface.
- */
- int
- tcpreplay_set_dualfile(tcpreplay_t *ctx, bool value)
- {
- assert(ctx);
- ctx->options->dualfile = value;
- return 0;
- }
- /**
- * \brief Enable or disable preloading the file cache
- *
- * Note: This is a global option and forces all pcaps
- * to be preloaded for this context. If you turn this
- * on, then it forces set_file_cache(true)
- */
- int
- tcpreplay_set_preload_pcap(tcpreplay_t *ctx, bool value)
- {
- assert(ctx);
- ctx->options->preload_pcap = value;
- return 0;
- }
- /**
- * \brief Add a pcap file to be sent via tcpreplay
- *
- * One or more pcap files can be added. Each file will be replayed
- * in order
- */
- int
- tcpreplay_add_pcapfile(tcpreplay_t *ctx, char *pcap_file)
- {
- assert(ctx);
- assert(pcap_file);
- if (ctx->options->source_cnt < MAX_FILES) {
- ctx->options->sources[ctx->options->source_cnt].filename = safe_strdup(pcap_file);
- ctx->options->sources[ctx->options->source_cnt].type = source_filename;
- /*
- * prepare the cache info data struct. This doesn't actually enable
- * file caching for this pcap (that is controlled globally via
- * tcpreplay_set_file_cache())
- */
- ctx->options->file_cache[ctx->options->source_cnt].index = ctx->options->source_cnt;
- ctx->options->file_cache[ctx->options->source_cnt].cached = false;
- ctx->options->file_cache[ctx->options->source_cnt].packet_cache = NULL;
- ctx->options->source_cnt += 1;
- } else {
- tcpreplay_seterr(ctx, "Unable to add more then %u files", MAX_FILES);
- return -1;
- }
- return 0;
- }
- /**
- * Limit the total number of packets to send
- */
- int
- tcpreplay_set_limit_send(tcpreplay_t *ctx, COUNTER value)
- {
- assert(ctx);
- ctx->options->limit_send = value;
- return 0;
- }
- /**
- * \brief Specify the tcpprep cache file to use for replaying with two NICs
- *
- * Note: this only works if you have a single pcap file
- * returns -1 on error
- */
- int
- tcpreplay_set_tcpprep_cache(tcpreplay_t *ctx, char *file)
- {
- assert(ctx);
- char *tcpprep_file;
- if (ctx->options->source_cnt > 1) {
- tcpreplay_seterr(ctx, "%s", "Unable to use tcpprep cache file with a single pcap file");
- return -1;
- }
- tcpprep_file = safe_strdup(file);
- ctx->options->cache_packets = read_cache(&ctx->options->cachedata,
- tcpprep_file, &ctx->options->comment);
- free(tcpprep_file);
- return 0;
- }
- /*
- * Verbose mode requires fork() and tcpdump binary, hence won't work
- * under Win32 without Cygwin
- */
- /**
- * Enable verbose mode
- */
- int
- tcpreplay_set_verbose(tcpreplay_t *ctx, bool value _U_)
- {
- assert(ctx);
- #ifdef ENABLE_VERBOSE
- ctx->options->verbose = value;
- return 0;
- #else
- tcpreplay_seterr(ctx, "%s", "verbose mode not supported");
- return -1;
- #endif
- }
- /**
- * \brief Set the arguments to be passed to tcpdump
- *
- * Specify the additional argument to be passed to tcpdump when enabling
- * verbose mode. See TCPDUMP_ARGS in tcpdump.h for the default options
- */
- int
- tcpreplay_set_tcpdump_args(tcpreplay_t *ctx, char *value _U_)
- {
- assert(ctx);
- #ifdef ENABLE_VERBOSE
- assert(value);
- ctx->options->tcpdump_args = safe_strdup(value);
- return 0;
- #else
- tcpreplay_seterr(ctx, "%s", "verbose mode not supported");
- return -1;
- #endif
- }
- /**
- * \brief Set the path to the tcpdump binary
- *
- * In order to support the verbose feature, tcpreplay needs to know where
- * tcpdump lives
- */
- int
- tcpreplay_set_tcpdump(tcpreplay_t *ctx, tcpdump_t *value _U_)
- {
- assert(ctx);
- #ifdef ENABLE_VERBOSE
- assert(value);
- ctx->options->verbose = true;
- ctx->options->tcpdump = value;
- return 0;
- #else
- tcpreplay_seterr(ctx, "%s", "verbose mode not supported");
- return -1;
- #endif
- }
- /**
- * \brief Set the callback function for handing manual iteration
- *
- * Obviously for this to work, you need to first set speed_mode = speed_oneatatime
- * returns 0 on success, < 0 on error
- */
- int
- tcpreplay_set_manual_callback(tcpreplay_t *ctx, tcpreplay_manual_callback callback)
- {
- assert(ctx);
- assert(callback);
- if (ctx->options->speed.mode != speed_oneatatime) {
- tcpreplay_seterr(ctx, "%s",
- "Unable to set manual callback because speed mode is not 'speed_oneatatime'");
- return -1;
- }
- ctx->options->speed.manual_callback = callback;
- return 0;
- }
- /**
- * \brief return the number of packets sent so far
- */
- COUNTER
- tcpreplay_get_pkts_sent(tcpreplay_t *ctx)
- {
- assert(ctx);
- ctx->static_stats.pkts_sent = ctx->stats.pkts_sent;
- return ctx->static_stats.pkts_sent;
- }
- /**
- * \brief return the number of bytes sent so far
- */
- COUNTER
- tcpreplay_get_bytes_sent(tcpreplay_t *ctx)
- {
- assert(ctx);
- ctx->static_stats.bytes_sent = ctx->stats.bytes_sent;
- return ctx->static_stats.bytes_sent;
- }
- /**
- * \brief return the number of failed attempts to send a packet
- */
- COUNTER
- tcpreplay_get_failed(tcpreplay_t *ctx)
- {
- assert(ctx);
- ctx->static_stats.failed = ctx->stats.failed;
- return ctx->static_stats.failed;
- }
- /**
- * \brief returns a pointer to the timespec structure of when replay first started
- */
- const struct timespec *
- tcpreplay_get_start_time(tcpreplay_t *ctx)
- {
- assert(ctx);
- TIMESPEC_SET(&ctx->static_stats.start_time, &ctx->stats.start_time);
- return &ctx->static_stats.start_time;
- }
- /**
- * \brief returns a pointer to the timespec structure of when replay finished
- */
- const struct timespec *
- tcpreplay_get_end_time(tcpreplay_t *ctx)
- {
- assert(ctx);
- TIMESPEC_SET(&ctx->static_stats.end_time, &ctx->stats.end_time);
- return &ctx->static_stats.end_time;
- }
- /**
- * \brief Internal function to set the tcpreplay error string
- *
- * Used to set the error string when there is an error, result is retrieved
- * using tcpedit_geterr(). You shouldn't ever actually call this, but use
- * tcpreplay_seterr() which is a macro wrapping this instead.
- */
- void
- __tcpreplay_seterr(tcpreplay_t *ctx, const char *func,
- const int line, const char *file, const char *fmt, ...)
- {
- va_list ap;
- char errormsg[TCPREPLAY_ERRSTR_LEN - 32];
- assert(ctx);
- assert(file);
- assert(func);
- assert(line);
- va_start(ap, fmt);
- if (fmt != NULL)
- (void)vsnprintf(errormsg, sizeof(errormsg), fmt, ap);
- va_end(ap);
- #ifdef DEBUG
- snprintf(ctx->errstr, sizeof(ctx->errstr), "From %s:%s() line %d:\n%s",
- file, func, line, errormsg);
- #else
- snprintf(ctx->errstr, sizeof(ctx->errstr), "%s", errormsg);
- #endif
- }
- /**
- * \brief Internal function to set the tcpedit warning string
- *
- * Used to set the warning string when there is an non-fatal issue, result is retrieved
- * using tcpedit_getwarn().
- */
- void
- tcpreplay_setwarn(tcpreplay_t *ctx, const char *fmt, ...)
- {
- va_list ap;
- assert(ctx);
- va_start(ap, fmt);
- if (fmt != NULL)
- (void)vsnprintf(ctx->warnstr, sizeof(ctx->warnstr), fmt, ap);
- va_end(ap);
- }
- /**
- * \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;
- int int1dlt, int2dlt, i;
- int ret = 0;
- assert(ctx);
- ebuf = safe_malloc(SENDPACKET_ERRBUF_SIZE);
- /*
- * 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");
- ret = -1;
- goto out;
- }
- if (ctx->options->source_cnt == 0) {
- tcpreplay_seterr(ctx, "%s", "You must specify at least one source pcap");
- ret = -1;
- goto out;
- }
- if (ctx->options->dualfile) {
- if (!(ctx->options->source_cnt >= 2)) {
- tcpreplay_seterr(ctx, "%s", "Dual file mode requires 2 or more pcap files");
- ret = -1;
- goto out;
- }
- if (ctx->options->source_cnt % 2 != 0) {
- tcpreplay_seterr(ctx, "%s", "Dual file mode requires an even number of pcap files");
- ret = -1;
- goto out;
- }
- }
- if (ctx->options->dualfile && ctx->options->cachedata != NULL) {
- tcpreplay_seterr(ctx, "%s", "Can't use dual file mode and tcpprep cache file together");
- ret = -1;
- goto out;
- }
- 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");
- ret = -1;
- goto out;
- }
- #endif
- if ((intname = get_interface(ctx->intlist, ctx->options->intf1_name)) == NULL) {
- if (!strncmp(OPT_ARG(INTF1), "netmap:", 7) || !strncmp(OPT_ARG(INTF1), "vale", 4))
- tcpreplay_seterr(ctx, "Unable to connect to netmap interface %s. Ensure netmap module is installed (see INSTALL).",
- OPT_ARG(INTF1));
- else
- tcpreplay_seterr(ctx, "Invalid interface name/alias: %s", OPT_ARG(INTF1));
- ret = -1;
- goto out;
- }
- /* open interfaces for writing */
- if ((ctx->intf1 = sendpacket_open(ctx->options->intf1_name, ebuf, TCPR_DIR_C2S, ctx->sp_type, ctx)) == NULL) {
- tcpreplay_seterr(ctx, "Can't open %s: %s", ctx->options->intf1_name, ebuf);
- ret = -1;
- goto out;
- }
- 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));
- ret = -1;
- goto out;
- }
- /* open interfaces for writing */
- if ((ctx->intf2 = sendpacket_open(ctx->options->intf2_name, ebuf, TCPR_DIR_C2S, ctx->sp_type, ctx)) == NULL) {
- tcpreplay_seterr(ctx, "Can't open %s: %s", ctx->options->intf2_name, ebuf);
- ret = -1;
- goto out;
- }
- int2dlt = sendpacket_get_dlt(ctx->intf2);
- if (int2dlt != int1dlt) {
- tcpreplay_seterr(ctx, "DLT type mismatch 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));
- ret = -1;
- goto out;
- }
- }
- /*
- * Setup up the file cache, if required
- */
- if (ctx->options->preload_pcap) {
- /* 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;
- }
- }
- out:
- safe_free(ebuf);
- return ret;
- }
- static bool apply_loop_delay(tcpreplay_t *ctx);
- /**
- * \brief sends the traffic out the interfaces
- *
- * Designed to be called in a separate thread if you need to. Blocks until
- * the replay is complete or you call tcpreplay_abort() in another thread.
- */
- int
- tcpreplay_replay(tcpreplay_t *ctx)
- {
- int rcode;
- COUNTER loop, total_loops;
- assert(ctx);
- if (!ctx->options->source_cnt) {
- tcpreplay_seterr(ctx, "invalid source count: %d", ctx->options->source_cnt);
- return -1;
- }
- if (ctx->options->dualfile && ctx->options->source_cnt < 2) {
- tcpreplay_seterr(ctx, "invalid dualfile source count: %d", ctx->options->source_cnt);
- return -1;
- }
- init_timestamp(&ctx->stats.start_time);
- init_timestamp(&ctx->stats.time_delta);
- init_timestamp(&ctx->stats.end_time);
- init_timestamp(&ctx->stats.pkt_ts_delta);
- init_timestamp(&ctx->stats.last_print);
- ctx->running = true;
- total_loops = ctx->options->loop;
- loop = 0;
- /* main loop, when not looping forever (or until abort) */
- if (ctx->options->loop > 0) {
- while (ctx->options->loop-- && !ctx->abort) { /* limited loop */
- ++loop;
- if (ctx->options->stats == 0) {
- if (!ctx->unique_iteration || loop == ctx->unique_iteration)
- printf("Loop " COUNTER_SPEC " of " COUNTER_SPEC "...\n",
- loop, total_loops);
- else
- printf("Loop " COUNTER_SPEC " of " COUNTER_SPEC " (" COUNTER_SPEC " unique)...\n",
- loop, total_loops,
- ctx->unique_iteration);
- }
- if ((rcode = tcpr_replay_index(ctx)) < 0)
- return rcode;
- if (ctx->options->loop > 0) {
- if (apply_loop_delay(ctx))
- get_current_time(&ctx->stats.end_time);
- if (ctx->options->stats == 0) {
- packet_stats(&ctx->stats);
- }
- }
- #ifdef HAVE_LIBXDP
- sendpacket_t *sp = ctx->intf1;
- if (sp->handle_type == SP_TYPE_LIBXDP) {
- sp->xsk_info->tx.cached_prod = 0;
- sp->xsk_info->tx.cached_cons = sp->tx_size;
- }
- #endif
- }
- } else {
- while (!ctx->abort) { /* loop forever unless user aborts */
- ++loop;
- if (ctx->options->stats == 0) {
- if (!ctx->unique_iteration || loop == ctx->unique_iteration)
- printf("Loop " COUNTER_SPEC "...\n", loop);
- else
- printf("Loop " COUNTER_SPEC " (" COUNTER_SPEC " unique)...\n", loop,
- ctx->unique_iteration);
- }
- if ((rcode = tcpr_replay_index(ctx)) < 0)
- return rcode;
- if (apply_loop_delay(ctx))
- get_current_time(&ctx->stats.end_time);
- if (ctx->options->stats == 0 && !ctx->abort)
- packet_stats(&ctx->stats);
- }
- }
- ctx->running = false;
- if (ctx->options->stats >= 0) {
- char buf[64];
- if (format_date_time(&ctx->stats.end_time, buf, sizeof(buf)) > 0)
- printf("Test complete: %s\n", buf);
- }
- return 0;
- }
- /**
- * \brief Abort the tcpreplay_replay execution.
- *
- * This might take a little while since tcpreplay_replay() only checks this
- * once per packet (sleeping between packets can cause delays), however,
- * this function returns once the signal has been sent and does not block
- */
- int
- tcpreplay_abort(tcpreplay_t *ctx)
- {
- assert(ctx);
- ctx->abort = true;
- printf("sendpacket_abort\n");
- if (ctx->intf1 != NULL)
- sendpacket_abort(ctx->intf1);
- if (ctx->intf2 != NULL)
- sendpacket_abort(ctx->intf2);
- return 0;
- }
- /**
- * \brief Temporarily suspend tcpreplay_replay()
- *
- * This might take a little while since tcpreplay_replay() only checks this
- * once per packet (sleeping between packets can cause delays), however,
- * this function returns once the signal has been sent and does not block
- *
- * Note that suspending a running context can create odd timing
- */
- int
- tcpreplay_suspend(tcpreplay_t *ctx)
- {
- assert(ctx);
- ctx->suspend = true;
- return 0;
- }
- /**
- * \brief Restart tcpreplay_replay() after suspend
- *
- * Causes the worker thread to restart sending packets
- */
- int
- tcpreplay_restart(tcpreplay_t *ctx)
- {
- assert(ctx);
- ctx->suspend = false;
- return 0;
- }
- /**
- * \brief Tells you if the given tcpreplay context is currently suspended
- *
- * Suspended == running, but not sending packets
- */
- bool
- tcpreplay_is_suspended(tcpreplay_t *ctx)
- {
- assert(ctx);
- return ctx->suspend;
- }
- /**
- * \brief Tells you if the tcpreplay context is running (not yet finished)
- *
- * Returns true even if it is suspended
- */
- bool
- tcpreplay_is_running(tcpreplay_t *ctx)
- {
- assert(ctx);
- return ctx->running;
- }
- /**
- * \brief returns the current statistics during or after a replay
- *
- * For performance reasons, I don't bother to put a mutex around this and you
- * don't need to either. Just realize that your values may be off by one until
- * tcreplay_replay() returns.
- */
- const tcpreplay_stats_t *
- tcpreplay_get_stats(tcpreplay_t *ctx)
- {
- const tcpreplay_stats_t *ptr;
- assert(ctx);
- /* copy stats over so they don't change while caller is using the buffer */
- memcpy(&ctx->static_stats, &ctx->stats, sizeof(tcpreplay_stats_t));
- ptr = &ctx->static_stats;
- return ptr;
- }
- /**
- * \brief returns the current number of sources/files to be sent
- */
- int
- tcpreplay_get_source_count(tcpreplay_t *ctx)
- {
- assert(ctx);
- return ctx->options->source_cnt;
- }
- /**
- * \brief Returns the current source id being replayed
- */
- int
- tcpreplay_get_current_source(tcpreplay_t *ctx)
- {
- assert(ctx);
- return ctx->current_source;
- }
- /* vim: set tabstop=8 expandtab shiftwidth=4 softtabstop=4: */
- /**
- * \brief Sets printing of flow statistics
- */
- int tcpreplay_set_flow_stats(tcpreplay_t *ctx, bool value)
- {
- assert(ctx);
- ctx->options->flow_stats = value;
- return 0;
- }
- /**
- * \brief Sets the flow expiry in seconds
- */
- int tcpreplay_set_flow_expiry(tcpreplay_t *ctx, int value)
- {
- assert(ctx);
- ctx->options->flow_expiry = value;
- return 0;
- }
- /**
- * \brief Get whether to printof flow statistics
- */
- bool tcpreplay_get_flow_stats(tcpreplay_t *ctx)
- {
- assert(ctx);
- return ctx->options->flow_stats;
- }
- /**
- * \brief Gets the flow expiry in seconds
- */
- int tcpreplay_get_flow_expiry(tcpreplay_t *ctx)
- {
- assert(ctx);
- return ctx->options->flow_expiry;
- }
- bool
- apply_loop_delay(tcpreplay_t *ctx) {
- if (!ctx->abort && ctx->options->loopdelay_ms > 0) {
- usleep(ctx->options->loopdelay_ms * 1000);
- return true;
- }
- if (!ctx->abort && ctx->options->loopdelay_ns > 0) {
- struct timespec nap;
- nap.tv_sec = 0;
- nap.tv_nsec = ctx->options->loopdelay_ns;
- nanosleep_sleep(NULL, &nap, &ctx->stats.end_time, NULL);
- return true;
- }
- return false;
- }
- #ifdef HAVE_LIBXDP
- void
- delete_xsk_socket(struct xsk_socket *xsk)
- {
- size_t desc_sz = sizeof(struct xdp_desc);
- struct xdp_mmap_offsets off;
- socklen_t optlen;
- int err;
- if (!xsk) {
- return;
- }
- optlen = sizeof(off);
- err = getsockopt(xsk->fd, SOL_XDP, XDP_MMAP_OFFSETS, &off, &optlen);
- if (!err) {
- if (xsk->rx) {
- munmap(xsk->rx->ring - off.rx.desc, off.rx.desc + xsk->config.rx_size * desc_sz);
- }
- if (xsk->tx) {
- munmap(xsk->tx->ring - off.tx.desc, off.tx.desc + xsk->config.tx_size * desc_sz);
- }
- }
- close(xsk->fd);
- }
- void
- free_umem_and_xsk(sendpacket_t *sp)
- {
- xsk_umem__delete(sp->xsk_info->umem->umem);
- delete_xsk_socket(sp->xsk_info->xsk);
- }
- #endif /*HAVE_LIBXDP*/
|