125763b3cSThomas Gleixner // SPDX-License-Identifier: GPL-2.0-only
25c7fc2d2SAlexei Starovoitov /* Copyright (c) 2013-2015 PLUMgrid, http://plumgrid.com
35c7fc2d2SAlexei Starovoitov */
45c7fc2d2SAlexei Starovoitov #include <stdio.h>
55c7fc2d2SAlexei Starovoitov #include <stdlib.h>
65c7fc2d2SAlexei Starovoitov #include <signal.h>
75c7fc2d2SAlexei Starovoitov #include <unistd.h>
85c7fc2d2SAlexei Starovoitov #include <stdbool.h>
95c7fc2d2SAlexei Starovoitov #include <string.h>
10e00c7b21SDaniel Borkmann
112bf3e2efSJakub Kicinski #include <bpf/bpf.h>
1263841bc0SDaniel T. Lee #include <bpf/libbpf.h>
13e00c7b21SDaniel Borkmann #include "bpf_util.h"
145c7fc2d2SAlexei Starovoitov
155c7fc2d2SAlexei Starovoitov #define SLOTS 100
165c7fc2d2SAlexei Starovoitov
clear_stats(int fd)175c7fc2d2SAlexei Starovoitov static void clear_stats(int fd)
185c7fc2d2SAlexei Starovoitov {
19e00c7b21SDaniel Borkmann unsigned int nr_cpus = bpf_num_possible_cpus();
203059303fSAlexei Starovoitov __u64 values[nr_cpus];
215c7fc2d2SAlexei Starovoitov __u32 key;
225c7fc2d2SAlexei Starovoitov
233059303fSAlexei Starovoitov memset(values, 0, sizeof(values));
245c7fc2d2SAlexei Starovoitov for (key = 0; key < SLOTS; key++)
25d40fc181SJoe Stringer bpf_map_update_elem(fd, &key, values, BPF_ANY);
265c7fc2d2SAlexei Starovoitov }
275c7fc2d2SAlexei Starovoitov
285c7fc2d2SAlexei Starovoitov const char *color[] = {
295c7fc2d2SAlexei Starovoitov "\033[48;5;255m",
305c7fc2d2SAlexei Starovoitov "\033[48;5;252m",
315c7fc2d2SAlexei Starovoitov "\033[48;5;250m",
325c7fc2d2SAlexei Starovoitov "\033[48;5;248m",
335c7fc2d2SAlexei Starovoitov "\033[48;5;246m",
345c7fc2d2SAlexei Starovoitov "\033[48;5;244m",
355c7fc2d2SAlexei Starovoitov "\033[48;5;242m",
365c7fc2d2SAlexei Starovoitov "\033[48;5;240m",
375c7fc2d2SAlexei Starovoitov "\033[48;5;238m",
385c7fc2d2SAlexei Starovoitov "\033[48;5;236m",
395c7fc2d2SAlexei Starovoitov "\033[48;5;234m",
405c7fc2d2SAlexei Starovoitov "\033[48;5;232m",
415c7fc2d2SAlexei Starovoitov };
425c7fc2d2SAlexei Starovoitov const int num_colors = ARRAY_SIZE(color);
435c7fc2d2SAlexei Starovoitov
445c7fc2d2SAlexei Starovoitov const char nocolor[] = "\033[00m";
455c7fc2d2SAlexei Starovoitov
465c7fc2d2SAlexei Starovoitov const char *sym[] = {
475c7fc2d2SAlexei Starovoitov " ",
485c7fc2d2SAlexei Starovoitov " ",
495c7fc2d2SAlexei Starovoitov ".",
505c7fc2d2SAlexei Starovoitov ".",
515c7fc2d2SAlexei Starovoitov "*",
525c7fc2d2SAlexei Starovoitov "*",
535c7fc2d2SAlexei Starovoitov "o",
545c7fc2d2SAlexei Starovoitov "o",
555c7fc2d2SAlexei Starovoitov "O",
565c7fc2d2SAlexei Starovoitov "O",
575c7fc2d2SAlexei Starovoitov "#",
585c7fc2d2SAlexei Starovoitov "#",
595c7fc2d2SAlexei Starovoitov };
605c7fc2d2SAlexei Starovoitov
615c7fc2d2SAlexei Starovoitov bool full_range = false;
625c7fc2d2SAlexei Starovoitov bool text_only = false;
635c7fc2d2SAlexei Starovoitov
print_banner(void)645c7fc2d2SAlexei Starovoitov static void print_banner(void)
655c7fc2d2SAlexei Starovoitov {
665c7fc2d2SAlexei Starovoitov if (full_range)
675c7fc2d2SAlexei Starovoitov printf("|1ns |10ns |100ns |1us |10us |100us"
685c7fc2d2SAlexei Starovoitov " |1ms |10ms |100ms |1s |10s\n");
695c7fc2d2SAlexei Starovoitov else
705c7fc2d2SAlexei Starovoitov printf("|1us |10us |100us |1ms |10ms "
715c7fc2d2SAlexei Starovoitov "|100ms |1s |10s\n");
725c7fc2d2SAlexei Starovoitov }
735c7fc2d2SAlexei Starovoitov
print_hist(int fd)745c7fc2d2SAlexei Starovoitov static void print_hist(int fd)
755c7fc2d2SAlexei Starovoitov {
76e00c7b21SDaniel Borkmann unsigned int nr_cpus = bpf_num_possible_cpus();
775c7fc2d2SAlexei Starovoitov __u64 total_events = 0;
783059303fSAlexei Starovoitov long values[nr_cpus];
793059303fSAlexei Starovoitov __u64 max_cnt = 0;
803059303fSAlexei Starovoitov __u64 cnt[SLOTS];
813059303fSAlexei Starovoitov __u64 value;
823059303fSAlexei Starovoitov __u32 key;
833059303fSAlexei Starovoitov int i;
845c7fc2d2SAlexei Starovoitov
855c7fc2d2SAlexei Starovoitov for (key = 0; key < SLOTS; key++) {
86d40fc181SJoe Stringer bpf_map_lookup_elem(fd, &key, values);
875c7fc2d2SAlexei Starovoitov value = 0;
883059303fSAlexei Starovoitov for (i = 0; i < nr_cpus; i++)
893059303fSAlexei Starovoitov value += values[i];
905c7fc2d2SAlexei Starovoitov cnt[key] = value;
915c7fc2d2SAlexei Starovoitov total_events += value;
925c7fc2d2SAlexei Starovoitov if (value > max_cnt)
935c7fc2d2SAlexei Starovoitov max_cnt = value;
945c7fc2d2SAlexei Starovoitov }
955c7fc2d2SAlexei Starovoitov clear_stats(fd);
965c7fc2d2SAlexei Starovoitov for (key = full_range ? 0 : 29; key < SLOTS; key++) {
975c7fc2d2SAlexei Starovoitov int c = num_colors * cnt[key] / (max_cnt + 1);
985c7fc2d2SAlexei Starovoitov
995c7fc2d2SAlexei Starovoitov if (text_only)
1005c7fc2d2SAlexei Starovoitov printf("%s", sym[c]);
1015c7fc2d2SAlexei Starovoitov else
1025c7fc2d2SAlexei Starovoitov printf("%s %s", color[c], nocolor);
1035c7fc2d2SAlexei Starovoitov }
1045c7fc2d2SAlexei Starovoitov printf(" # %lld\n", total_events);
1055c7fc2d2SAlexei Starovoitov }
1065c7fc2d2SAlexei Starovoitov
main(int ac,char ** argv)1075c7fc2d2SAlexei Starovoitov int main(int ac, char **argv)
1085c7fc2d2SAlexei Starovoitov {
10963841bc0SDaniel T. Lee struct bpf_link *links[2];
11063841bc0SDaniel T. Lee struct bpf_program *prog;
11163841bc0SDaniel T. Lee struct bpf_object *obj;
1125c7fc2d2SAlexei Starovoitov char filename[256];
11363841bc0SDaniel T. Lee int map_fd, i, j = 0;
1145c7fc2d2SAlexei Starovoitov
1155c7fc2d2SAlexei Starovoitov for (i = 1; i < ac; i++) {
1165c7fc2d2SAlexei Starovoitov if (strcmp(argv[i], "-a") == 0) {
1175c7fc2d2SAlexei Starovoitov full_range = true;
1185c7fc2d2SAlexei Starovoitov } else if (strcmp(argv[i], "-t") == 0) {
1195c7fc2d2SAlexei Starovoitov text_only = true;
1205c7fc2d2SAlexei Starovoitov } else if (strcmp(argv[i], "-h") == 0) {
1215c7fc2d2SAlexei Starovoitov printf("Usage:\n"
1225c7fc2d2SAlexei Starovoitov " -a display wider latency range\n"
1235c7fc2d2SAlexei Starovoitov " -t text only\n");
1245c7fc2d2SAlexei Starovoitov return 1;
1255c7fc2d2SAlexei Starovoitov }
1265c7fc2d2SAlexei Starovoitov }
1275c7fc2d2SAlexei Starovoitov
128*4a0ee788SDaniel T. Lee snprintf(filename, sizeof(filename), "%s.bpf.o", argv[0]);
12963841bc0SDaniel T. Lee obj = bpf_object__open_file(filename, NULL);
13063841bc0SDaniel T. Lee if (libbpf_get_error(obj)) {
13163841bc0SDaniel T. Lee fprintf(stderr, "ERROR: opening BPF object file failed\n");
13263841bc0SDaniel T. Lee return 0;
13363841bc0SDaniel T. Lee }
13463841bc0SDaniel T. Lee
13563841bc0SDaniel T. Lee /* load BPF program */
13663841bc0SDaniel T. Lee if (bpf_object__load(obj)) {
13763841bc0SDaniel T. Lee fprintf(stderr, "ERROR: loading BPF object file failed\n");
13863841bc0SDaniel T. Lee goto cleanup;
13963841bc0SDaniel T. Lee }
14063841bc0SDaniel T. Lee
14163841bc0SDaniel T. Lee map_fd = bpf_object__find_map_fd_by_name(obj, "lat_map");
14263841bc0SDaniel T. Lee if (map_fd < 0) {
14363841bc0SDaniel T. Lee fprintf(stderr, "ERROR: finding a map in obj file failed\n");
14463841bc0SDaniel T. Lee goto cleanup;
14563841bc0SDaniel T. Lee }
14663841bc0SDaniel T. Lee
14763841bc0SDaniel T. Lee bpf_object__for_each_program(prog, obj) {
14863841bc0SDaniel T. Lee links[j] = bpf_program__attach(prog);
14963841bc0SDaniel T. Lee if (libbpf_get_error(links[j])) {
15063841bc0SDaniel T. Lee fprintf(stderr, "ERROR: bpf_program__attach failed\n");
15163841bc0SDaniel T. Lee links[j] = NULL;
15263841bc0SDaniel T. Lee goto cleanup;
15363841bc0SDaniel T. Lee }
15463841bc0SDaniel T. Lee j++;
15563841bc0SDaniel T. Lee }
15663841bc0SDaniel T. Lee
1575c7fc2d2SAlexei Starovoitov printf(" heatmap of IO latency\n");
1585c7fc2d2SAlexei Starovoitov if (text_only)
1595c7fc2d2SAlexei Starovoitov printf(" %s", sym[num_colors - 1]);
1605c7fc2d2SAlexei Starovoitov else
1615c7fc2d2SAlexei Starovoitov printf(" %s %s", color[num_colors - 1], nocolor);
1625c7fc2d2SAlexei Starovoitov printf(" - many events with this latency\n");
1635c7fc2d2SAlexei Starovoitov
1645c7fc2d2SAlexei Starovoitov if (text_only)
1655c7fc2d2SAlexei Starovoitov printf(" %s", sym[0]);
1665c7fc2d2SAlexei Starovoitov else
1675c7fc2d2SAlexei Starovoitov printf(" %s %s", color[0], nocolor);
1685c7fc2d2SAlexei Starovoitov printf(" - few events\n");
1695c7fc2d2SAlexei Starovoitov
1705c7fc2d2SAlexei Starovoitov for (i = 0; ; i++) {
1715c7fc2d2SAlexei Starovoitov if (i % 20 == 0)
1725c7fc2d2SAlexei Starovoitov print_banner();
17363841bc0SDaniel T. Lee print_hist(map_fd);
1745c7fc2d2SAlexei Starovoitov sleep(2);
1755c7fc2d2SAlexei Starovoitov }
1765c7fc2d2SAlexei Starovoitov
17763841bc0SDaniel T. Lee cleanup:
17863841bc0SDaniel T. Lee for (j--; j >= 0; j--)
17963841bc0SDaniel T. Lee bpf_link__destroy(links[j]);
18063841bc0SDaniel T. Lee
18163841bc0SDaniel T. Lee bpf_object__close(obj);
1825c7fc2d2SAlexei Starovoitov return 0;
1835c7fc2d2SAlexei Starovoitov }
184