Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit d606ee5c authored by Matteo Croce's avatar Matteo Croce Committed by Alexei Starovoitov
Browse files

samples: bpf: improve xdp1 example 



Store only the total packet count for every protocol, instead of the
whole per-cpu array.
Use bpf_map_get_next_key() to iterate the map, instead of looking up
all the protocols.

Signed-off-by: default avatarMatteo Croce <mcroce@redhat.com>
Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
parent 9ffd05d9

samples/bpf/xdp1_user.c

+8 −10
Original line number Diff line number Diff line
@@ -34,26 +34,24 @@ static void int_exit(int sig) 
static void poll_stats(int map_fd, int interval)

{

	unsigned int nr_cpus = bpf_num_possible_cpus();

	const unsigned int nr_keys = 256;

	__u64 values[nr_cpus], prev[nr_keys][nr_cpus];

	__u32 key;

	__u64 values[nr_cpus], prev[UINT8_MAX] = { 0 };

	int i;



	memset(prev, 0, sizeof(prev));



	while (1) {

		__u32 key = UINT32_MAX;



		sleep(interval);



		for (key = 0; key < nr_keys; key++) {

		while (bpf_map_get_next_key(map_fd, &key, &key) != -1) {

			__u64 sum = 0;



			assert(bpf_map_lookup_elem(map_fd, &key, values) == 0);

			for (i = 0; i < nr_cpus; i++)

				sum += (values[i] - prev[key][i]);

			if (sum)

				sum += values[i];

			if (sum > prev[key])

				printf("proto %u: %10llu pkt/s\n",

				       key, sum / interval);

			memcpy(prev[key], values, sizeof(values));

				       key, (sum - prev[key]) / interval);

			prev[key] = sum;

		}

	}

}