1// SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB
2
3/**
4 * ibumad BPF sample user side
5 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of version 2 of the GNU General Public
8 * License as published by the Free Software Foundation.
9 *
10 * Copyright(c) 2018 Ira Weiny, Intel Corporation
11 */
12
13#include <linux/bpf.h>
14#include <signal.h>
15#include <stdio.h>
16#include <stdlib.h>
17#include <string.h>
18#include <unistd.h>
19#include <sys/types.h>
20#include <limits.h>
21
22#include <sys/resource.h>
23#include <getopt.h>
24#include <net/if.h>
25
26#include "bpf_load.h"
27#include "bpf_util.h"
28#include <bpf/libbpf.h>
29
30static void dump_counts(int fd)
31{
32	__u32 key;
33	__u64 value;
34
35	for (key = 0; key < 256; key++) {
36		if (bpf_map_lookup_elem(fd, &key, &value)) {
37			printf("failed to read key %u\n", key);
38			continue;
39		}
40		if (value)
41			printf("0x%02x : %llu\n", key, value);
42	}
43}
44
45static void dump_all_counts(void)
46{
47	printf("Read 'Class : count'\n");
48	dump_counts(map_fd[0]);
49	printf("Write 'Class : count'\n");
50	dump_counts(map_fd[1]);
51}
52
53static void dump_exit(int sig)
54{
55	dump_all_counts();
56	exit(0);
57}
58
59static const struct option long_options[] = {
60	{"help",      no_argument,       NULL, 'h'},
61	{"delay",     required_argument, NULL, 'd'},
62};
63
64static void usage(char *cmd)
65{
66	printf("eBPF test program to count packets from various IP addresses\n"
67		"Usage: %s <options>\n"
68		"       --help,   -h  this menu\n"
69		"       --delay,  -d  <delay>  wait <delay> sec between prints [1 - 1000000]\n"
70		, cmd
71		);
72}
73
74int main(int argc, char **argv)
75{
76	unsigned long delay = 5;
77	int longindex = 0;
78	int opt;
79	char bpf_file[256];
80
81	/* Create the eBPF kernel code path name.
82	 * This follows the pattern of all of the other bpf samples
83	 */
84	snprintf(bpf_file, sizeof(bpf_file), "%s_kern.o", argv[0]);
85
86	/* Do one final dump when exiting */
87	signal(SIGINT, dump_exit);
88	signal(SIGTERM, dump_exit);
89
90	while ((opt = getopt_long(argc, argv, "hd:rSw",
91				  long_options, &longindex)) != -1) {
92		switch (opt) {
93		case 'd':
94			delay = strtoul(optarg, NULL, 0);
95			if (delay == ULONG_MAX || delay < 0 ||
96			    delay > 1000000) {
97				fprintf(stderr, "ERROR: invalid delay : %s\n",
98					optarg);
99				usage(argv[0]);
100				return 1;
101			}
102			break;
103		default:
104		case 'h':
105			usage(argv[0]);
106			return 1;
107		}
108	}
109
110	if (load_bpf_file(bpf_file)) {
111		fprintf(stderr, "ERROR: failed to load eBPF from file : %s\n",
112			bpf_file);
113		return 1;
114	}
115
116	while (1) {
117		sleep(delay);
118		dump_all_counts();
119	}
120
121	return 0;
122}
123