123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404 |
- // SPDX-License-Identifier: GPL-2.0
- #include "cpumap.h"
- #include "debug.h"
- #include "env.h"
- #include "util/header.h"
- #include <linux/ctype.h>
- #include <linux/zalloc.h>
- #include "bpf-event.h"
- #include "cgroup.h"
- #include <errno.h>
- #include <sys/utsname.h>
- #include <bpf/libbpf.h>
- #include <stdlib.h>
- #include <string.h>
- struct perf_env perf_env;
- void perf_env__insert_bpf_prog_info(struct perf_env *env,
- struct bpf_prog_info_node *info_node)
- {
- __u32 prog_id = info_node->info_linear->info.id;
- struct bpf_prog_info_node *node;
- struct rb_node *parent = NULL;
- struct rb_node **p;
- down_write(&env->bpf_progs.lock);
- p = &env->bpf_progs.infos.rb_node;
- while (*p != NULL) {
- parent = *p;
- node = rb_entry(parent, struct bpf_prog_info_node, rb_node);
- if (prog_id < node->info_linear->info.id) {
- p = &(*p)->rb_left;
- } else if (prog_id > node->info_linear->info.id) {
- p = &(*p)->rb_right;
- } else {
- pr_debug("duplicated bpf prog info %u\n", prog_id);
- goto out;
- }
- }
- rb_link_node(&info_node->rb_node, parent, p);
- rb_insert_color(&info_node->rb_node, &env->bpf_progs.infos);
- env->bpf_progs.infos_cnt++;
- out:
- up_write(&env->bpf_progs.lock);
- }
- struct bpf_prog_info_node *perf_env__find_bpf_prog_info(struct perf_env *env,
- __u32 prog_id)
- {
- struct bpf_prog_info_node *node = NULL;
- struct rb_node *n;
- down_read(&env->bpf_progs.lock);
- n = env->bpf_progs.infos.rb_node;
- while (n) {
- node = rb_entry(n, struct bpf_prog_info_node, rb_node);
- if (prog_id < node->info_linear->info.id)
- n = n->rb_left;
- else if (prog_id > node->info_linear->info.id)
- n = n->rb_right;
- else
- goto out;
- }
- node = NULL;
- out:
- up_read(&env->bpf_progs.lock);
- return node;
- }
- bool perf_env__insert_btf(struct perf_env *env, struct btf_node *btf_node)
- {
- struct rb_node *parent = NULL;
- __u32 btf_id = btf_node->id;
- struct btf_node *node;
- struct rb_node **p;
- bool ret = true;
- down_write(&env->bpf_progs.lock);
- p = &env->bpf_progs.btfs.rb_node;
- while (*p != NULL) {
- parent = *p;
- node = rb_entry(parent, struct btf_node, rb_node);
- if (btf_id < node->id) {
- p = &(*p)->rb_left;
- } else if (btf_id > node->id) {
- p = &(*p)->rb_right;
- } else {
- pr_debug("duplicated btf %u\n", btf_id);
- ret = false;
- goto out;
- }
- }
- rb_link_node(&btf_node->rb_node, parent, p);
- rb_insert_color(&btf_node->rb_node, &env->bpf_progs.btfs);
- env->bpf_progs.btfs_cnt++;
- out:
- up_write(&env->bpf_progs.lock);
- return ret;
- }
- struct btf_node *perf_env__find_btf(struct perf_env *env, __u32 btf_id)
- {
- struct btf_node *node = NULL;
- struct rb_node *n;
- down_read(&env->bpf_progs.lock);
- n = env->bpf_progs.btfs.rb_node;
- while (n) {
- node = rb_entry(n, struct btf_node, rb_node);
- if (btf_id < node->id)
- n = n->rb_left;
- else if (btf_id > node->id)
- n = n->rb_right;
- else
- goto out;
- }
- node = NULL;
- out:
- up_read(&env->bpf_progs.lock);
- return node;
- }
- /* purge data in bpf_progs.infos tree */
- static void perf_env__purge_bpf(struct perf_env *env)
- {
- struct rb_root *root;
- struct rb_node *next;
- down_write(&env->bpf_progs.lock);
- root = &env->bpf_progs.infos;
- next = rb_first(root);
- while (next) {
- struct bpf_prog_info_node *node;
- node = rb_entry(next, struct bpf_prog_info_node, rb_node);
- next = rb_next(&node->rb_node);
- rb_erase(&node->rb_node, root);
- free(node->info_linear);
- free(node);
- }
- env->bpf_progs.infos_cnt = 0;
- root = &env->bpf_progs.btfs;
- next = rb_first(root);
- while (next) {
- struct btf_node *node;
- node = rb_entry(next, struct btf_node, rb_node);
- next = rb_next(&node->rb_node);
- rb_erase(&node->rb_node, root);
- free(node);
- }
- env->bpf_progs.btfs_cnt = 0;
- up_write(&env->bpf_progs.lock);
- }
- void perf_env__exit(struct perf_env *env)
- {
- int i;
- perf_env__purge_bpf(env);
- perf_env__purge_cgroups(env);
- zfree(&env->hostname);
- zfree(&env->os_release);
- zfree(&env->version);
- zfree(&env->arch);
- zfree(&env->cpu_desc);
- zfree(&env->cpuid);
- zfree(&env->cmdline);
- zfree(&env->cmdline_argv);
- zfree(&env->sibling_dies);
- zfree(&env->sibling_cores);
- zfree(&env->sibling_threads);
- zfree(&env->pmu_mappings);
- zfree(&env->cpu);
- zfree(&env->cpu_pmu_caps);
- zfree(&env->numa_map);
- for (i = 0; i < env->nr_numa_nodes; i++)
- perf_cpu_map__put(env->numa_nodes[i].map);
- zfree(&env->numa_nodes);
- for (i = 0; i < env->caches_cnt; i++)
- cpu_cache_level__free(&env->caches[i]);
- zfree(&env->caches);
- for (i = 0; i < env->nr_memory_nodes; i++)
- zfree(&env->memory_nodes[i].set);
- zfree(&env->memory_nodes);
- }
- void perf_env__init(struct perf_env *env)
- {
- env->bpf_progs.infos = RB_ROOT;
- env->bpf_progs.btfs = RB_ROOT;
- init_rwsem(&env->bpf_progs.lock);
- }
- int perf_env__set_cmdline(struct perf_env *env, int argc, const char *argv[])
- {
- int i;
- /* do not include NULL termination */
- env->cmdline_argv = calloc(argc, sizeof(char *));
- if (env->cmdline_argv == NULL)
- goto out_enomem;
- /*
- * Must copy argv contents because it gets moved around during option
- * parsing:
- */
- for (i = 0; i < argc ; i++) {
- env->cmdline_argv[i] = argv[i];
- if (env->cmdline_argv[i] == NULL)
- goto out_free;
- }
- env->nr_cmdline = argc;
- return 0;
- out_free:
- zfree(&env->cmdline_argv);
- out_enomem:
- return -ENOMEM;
- }
- int perf_env__read_cpu_topology_map(struct perf_env *env)
- {
- int cpu, nr_cpus;
- if (env->cpu != NULL)
- return 0;
- if (env->nr_cpus_avail == 0)
- env->nr_cpus_avail = cpu__max_present_cpu();
- nr_cpus = env->nr_cpus_avail;
- if (nr_cpus == -1)
- return -EINVAL;
- env->cpu = calloc(nr_cpus, sizeof(env->cpu[0]));
- if (env->cpu == NULL)
- return -ENOMEM;
- for (cpu = 0; cpu < nr_cpus; ++cpu) {
- env->cpu[cpu].core_id = cpu_map__get_core_id(cpu);
- env->cpu[cpu].socket_id = cpu_map__get_socket_id(cpu);
- env->cpu[cpu].die_id = cpu_map__get_die_id(cpu);
- }
- env->nr_cpus_avail = nr_cpus;
- return 0;
- }
- int perf_env__read_cpuid(struct perf_env *env)
- {
- char cpuid[128];
- int err = get_cpuid(cpuid, sizeof(cpuid));
- if (err)
- return err;
- free(env->cpuid);
- env->cpuid = strdup(cpuid);
- if (env->cpuid == NULL)
- return ENOMEM;
- return 0;
- }
- static int perf_env__read_arch(struct perf_env *env)
- {
- struct utsname uts;
- if (env->arch)
- return 0;
- if (!uname(&uts))
- env->arch = strdup(uts.machine);
- return env->arch ? 0 : -ENOMEM;
- }
- static int perf_env__read_nr_cpus_avail(struct perf_env *env)
- {
- if (env->nr_cpus_avail == 0)
- env->nr_cpus_avail = cpu__max_present_cpu();
- return env->nr_cpus_avail ? 0 : -ENOENT;
- }
- const char *perf_env__raw_arch(struct perf_env *env)
- {
- return env && !perf_env__read_arch(env) ? env->arch : "unknown";
- }
- int perf_env__nr_cpus_avail(struct perf_env *env)
- {
- return env && !perf_env__read_nr_cpus_avail(env) ? env->nr_cpus_avail : 0;
- }
- void cpu_cache_level__free(struct cpu_cache_level *cache)
- {
- zfree(&cache->type);
- zfree(&cache->map);
- zfree(&cache->size);
- }
- /*
- * Return architecture name in a normalized form.
- * The conversion logic comes from the Makefile.
- */
- static const char *normalize_arch(char *arch)
- {
- if (!strcmp(arch, "x86_64"))
- return "x86";
- if (arch[0] == 'i' && arch[2] == '8' && arch[3] == '6')
- return "x86";
- if (!strcmp(arch, "sun4u") || !strncmp(arch, "sparc", 5))
- return "sparc";
- if (!strcmp(arch, "aarch64") || !strcmp(arch, "arm64"))
- return "arm64";
- if (!strncmp(arch, "arm", 3) || !strcmp(arch, "sa110"))
- return "arm";
- if (!strncmp(arch, "s390", 4))
- return "s390";
- if (!strncmp(arch, "parisc", 6))
- return "parisc";
- if (!strncmp(arch, "powerpc", 7) || !strncmp(arch, "ppc", 3))
- return "powerpc";
- if (!strncmp(arch, "mips", 4))
- return "mips";
- if (!strncmp(arch, "sh", 2) && isdigit(arch[2]))
- return "sh";
- return arch;
- }
- const char *perf_env__arch(struct perf_env *env)
- {
- char *arch_name;
- if (!env || !env->arch) { /* Assume local operation */
- static struct utsname uts = { .machine[0] = '\0', };
- if (uts.machine[0] == '\0' && uname(&uts) < 0)
- return NULL;
- arch_name = uts.machine;
- } else
- arch_name = env->arch;
- return normalize_arch(arch_name);
- }
- int perf_env__numa_node(struct perf_env *env, int cpu)
- {
- if (!env->nr_numa_map) {
- struct numa_node *nn;
- int i, nr = 0;
- for (i = 0; i < env->nr_numa_nodes; i++) {
- nn = &env->numa_nodes[i];
- nr = max(nr, perf_cpu_map__max(nn->map));
- }
- nr++;
- /*
- * We initialize the numa_map array to prepare
- * it for missing cpus, which return node -1
- */
- env->numa_map = malloc(nr * sizeof(int));
- if (!env->numa_map)
- return -1;
- for (i = 0; i < nr; i++)
- env->numa_map[i] = -1;
- env->nr_numa_map = nr;
- for (i = 0; i < env->nr_numa_nodes; i++) {
- int tmp, j;
- nn = &env->numa_nodes[i];
- perf_cpu_map__for_each_cpu(j, tmp, nn->map)
- env->numa_map[j] = i;
- }
- }
- return cpu >= 0 && cpu < env->nr_numa_map ? env->numa_map[cpu] : -1;
- }
|