aboutsummaryrefslogtreecommitdiff
path: root/tools
diff options
context:
space:
mode:
authorAllan Jude <allanjude@FreeBSD.org>2021-05-17 23:04:08 +0000
committerAllan Jude <allanjude@FreeBSD.org>2021-05-22 01:55:29 +0000
commit20d684ecc9d7d9128774f2e3c287058868f48bb0 (patch)
tree78a04bbb78d6459195fc5e799b53d719a91b3215 /tools
parent73e77cf90ba35809adefcc6cd129da5ef15c8e9d (diff)
downloadsrc-20d684ecc9d7d9128774f2e3c287058868f48bb0.tar.gz
src-20d684ecc9d7d9128774f2e3c287058868f48bb0.zip
pkt-gen: Allow limiting received packets
Makes pkg-gen quit after having received N packets, the same way it already supports doing for sent packets. Reviewed by: vmaffione Sponsored by: Klara Inc. MFC after: 4 weeks Differential Revision: https://reviews.freebsd.org/D30266
Diffstat (limited to 'tools')
-rw-r--r--tools/tools/netmap/pkt-gen.c7
1 files changed, 4 insertions, 3 deletions
diff --git a/tools/tools/netmap/pkt-gen.c b/tools/tools/netmap/pkt-gen.c
index c958af3b9781..b24915e150a5 100644
--- a/tools/tools/netmap/pkt-gen.c
+++ b/tools/tools/netmap/pkt-gen.c
@@ -1816,6 +1816,7 @@ receiver_body(void *data)
struct netmap_ring *rxring;
int i;
struct my_ctrs cur;
+ uint64_t n = targ->g->npackets / targ->g->nthreads;
memset(&cur, 0, sizeof(cur));
@@ -1843,7 +1844,7 @@ receiver_body(void *data)
/* main loop, exit after 1s silence */
clock_gettime(CLOCK_REALTIME_PRECISE, &targ->tic);
if (targ->g->dev_type == DEV_TAP) {
- while (!targ->cancel) {
+ while (!targ->cancel && (n == 0 || targ->ctr.pkts < n)) {) {
char buf[MAX_BODYSIZE];
/* XXX should we poll ? */
i = read(targ->g->main_fd, buf, sizeof(buf));
@@ -1855,7 +1856,7 @@ receiver_body(void *data)
}
#ifndef NO_PCAP
} else if (targ->g->dev_type == DEV_PCAP) {
- while (!targ->cancel) {
+ while (!targ->cancel && (n == 0 || targ->ctr.pkts < n)) {
/* XXX should we poll ? */
pcap_dispatch(targ->g->p, targ->g->burst, receive_pcap,
(u_char *)&targ->ctr);
@@ -1866,7 +1867,7 @@ receiver_body(void *data)
int dump = targ->g->options & OPT_DUMP;
nifp = targ->nmd->nifp;
- while (!targ->cancel) {
+ while (!targ->cancel && (n == 0 || targ->ctr.pkts < n)) {
/* Once we started to receive packets, wait at most 1 seconds
before quitting. */
#ifdef BUSYWAIT