123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588 |
- // SPDX-License-Identifier: GPL-2.0
- #include <api/fs/fs.h>
- #include "cpumap.h"
- #include "debug.h"
- #include "event.h"
- #include <assert.h>
- #include <dirent.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <linux/bitmap.h>
- #include "asm/bug.h"
- #include <linux/ctype.h>
- #include <linux/zalloc.h>
- static int max_cpu_num;
- static int max_present_cpu_num;
- static int max_node_num;
- static int *cpunode_map;
- static struct perf_cpu_map *cpu_map__from_entries(struct cpu_map_entries *cpus)
- {
- struct perf_cpu_map *map;
- map = perf_cpu_map__empty_new(cpus->nr);
- if (map) {
- unsigned i;
- for (i = 0; i < cpus->nr; i++) {
- /*
- * Special treatment for -1, which is not real cpu number,
- * and we need to use (int) -1 to initialize map[i],
- * otherwise it would become 65535.
- */
- if (cpus->cpu[i] == (u16) -1)
- map->map[i] = -1;
- else
- map->map[i] = (int) cpus->cpu[i];
- }
- }
- return map;
- }
- static struct perf_cpu_map *cpu_map__from_mask(struct perf_record_record_cpu_map *mask)
- {
- struct perf_cpu_map *map;
- int nr, nbits = mask->nr * mask->long_size * BITS_PER_BYTE;
- nr = bitmap_weight(mask->mask, nbits);
- map = perf_cpu_map__empty_new(nr);
- if (map) {
- int cpu, i = 0;
- for_each_set_bit(cpu, mask->mask, nbits)
- map->map[i++] = cpu;
- }
- return map;
- }
- struct perf_cpu_map *cpu_map__new_data(struct perf_record_cpu_map_data *data)
- {
- if (data->type == PERF_CPU_MAP__CPUS)
- return cpu_map__from_entries((struct cpu_map_entries *)data->data);
- else
- return cpu_map__from_mask((struct perf_record_record_cpu_map *)data->data);
- }
- size_t cpu_map__fprintf(struct perf_cpu_map *map, FILE *fp)
- {
- #define BUFSIZE 1024
- char buf[BUFSIZE];
- cpu_map__snprint(map, buf, sizeof(buf));
- return fprintf(fp, "%s\n", buf);
- #undef BUFSIZE
- }
- struct perf_cpu_map *perf_cpu_map__empty_new(int nr)
- {
- struct perf_cpu_map *cpus = malloc(sizeof(*cpus) + sizeof(int) * nr);
- if (cpus != NULL) {
- int i;
- cpus->nr = nr;
- for (i = 0; i < nr; i++)
- cpus->map[i] = -1;
- refcount_set(&cpus->refcnt, 1);
- }
- return cpus;
- }
- static int cpu__get_topology_int(int cpu, const char *name, int *value)
- {
- char path[PATH_MAX];
- snprintf(path, PATH_MAX,
- "devices/system/cpu/cpu%d/topology/%s", cpu, name);
- return sysfs__read_int(path, value);
- }
- int cpu_map__get_socket_id(int cpu)
- {
- int value, ret = cpu__get_topology_int(cpu, "physical_package_id", &value);
- return ret ?: value;
- }
- int cpu_map__get_socket(struct perf_cpu_map *map, int idx, void *data __maybe_unused)
- {
- int cpu;
- if (idx > map->nr)
- return -1;
- cpu = map->map[idx];
- return cpu_map__get_socket_id(cpu);
- }
- static int cmp_ids(const void *a, const void *b)
- {
- return *(int *)a - *(int *)b;
- }
- int cpu_map__build_map(struct perf_cpu_map *cpus, struct perf_cpu_map **res,
- int (*f)(struct perf_cpu_map *map, int cpu, void *data),
- void *data)
- {
- struct perf_cpu_map *c;
- int nr = cpus->nr;
- int cpu, s1, s2;
- /* allocate as much as possible */
- c = calloc(1, sizeof(*c) + nr * sizeof(int));
- if (!c)
- return -1;
- for (cpu = 0; cpu < nr; cpu++) {
- s1 = f(cpus, cpu, data);
- for (s2 = 0; s2 < c->nr; s2++) {
- if (s1 == c->map[s2])
- break;
- }
- if (s2 == c->nr) {
- c->map[c->nr] = s1;
- c->nr++;
- }
- }
- /* ensure we process id in increasing order */
- qsort(c->map, c->nr, sizeof(int), cmp_ids);
- refcount_set(&c->refcnt, 1);
- *res = c;
- return 0;
- }
- int cpu_map__get_die_id(int cpu)
- {
- int value, ret = cpu__get_topology_int(cpu, "die_id", &value);
- return ret ?: value;
- }
- int cpu_map__get_die(struct perf_cpu_map *map, int idx, void *data)
- {
- int cpu, die_id, s;
- if (idx > map->nr)
- return -1;
- cpu = map->map[idx];
- die_id = cpu_map__get_die_id(cpu);
- /* There is no die_id on legacy system. */
- if (die_id == -1)
- die_id = 0;
- s = cpu_map__get_socket(map, idx, data);
- if (s == -1)
- return -1;
- /*
- * Encode socket in bit range 15:8
- * die_id is relative to socket, and
- * we need a global id. So we combine
- * socket + die id
- */
- if (WARN_ONCE(die_id >> 8, "The die id number is too big.\n"))
- return -1;
- if (WARN_ONCE(s >> 8, "The socket id number is too big.\n"))
- return -1;
- return (s << 8) | (die_id & 0xff);
- }
- int cpu_map__get_core_id(int cpu)
- {
- int value, ret = cpu__get_topology_int(cpu, "core_id", &value);
- return ret ?: value;
- }
- int cpu_map__get_node_id(int cpu)
- {
- return cpu__get_node(cpu);
- }
- int cpu_map__get_core(struct perf_cpu_map *map, int idx, void *data)
- {
- int cpu, s_die;
- if (idx > map->nr)
- return -1;
- cpu = map->map[idx];
- cpu = cpu_map__get_core_id(cpu);
- /* s_die is the combination of socket + die id */
- s_die = cpu_map__get_die(map, idx, data);
- if (s_die == -1)
- return -1;
- /*
- * encode socket in bit range 31:24
- * encode die id in bit range 23:16
- * core_id is relative to socket and die,
- * we need a global id. So we combine
- * socket + die id + core id
- */
- if (WARN_ONCE(cpu >> 16, "The core id number is too big.\n"))
- return -1;
- return (s_die << 16) | (cpu & 0xffff);
- }
- int cpu_map__get_node(struct perf_cpu_map *map, int idx, void *data __maybe_unused)
- {
- if (idx < 0 || idx >= map->nr)
- return -1;
- return cpu_map__get_node_id(map->map[idx]);
- }
- int cpu_map__build_socket_map(struct perf_cpu_map *cpus, struct perf_cpu_map **sockp)
- {
- return cpu_map__build_map(cpus, sockp, cpu_map__get_socket, NULL);
- }
- int cpu_map__build_die_map(struct perf_cpu_map *cpus, struct perf_cpu_map **diep)
- {
- return cpu_map__build_map(cpus, diep, cpu_map__get_die, NULL);
- }
- int cpu_map__build_core_map(struct perf_cpu_map *cpus, struct perf_cpu_map **corep)
- {
- return cpu_map__build_map(cpus, corep, cpu_map__get_core, NULL);
- }
- int cpu_map__build_node_map(struct perf_cpu_map *cpus, struct perf_cpu_map **numap)
- {
- return cpu_map__build_map(cpus, numap, cpu_map__get_node, NULL);
- }
- /* setup simple routines to easily access node numbers given a cpu number */
- static int get_max_num(char *path, int *max)
- {
- size_t num;
- char *buf;
- int err = 0;
- if (filename__read_str(path, &buf, &num))
- return -1;
- buf[num] = '\0';
- /* start on the right, to find highest node num */
- while (--num) {
- if ((buf[num] == ',') || (buf[num] == '-')) {
- num++;
- break;
- }
- }
- if (sscanf(&buf[num], "%d", max) < 1) {
- err = -1;
- goto out;
- }
- /* convert from 0-based to 1-based */
- (*max)++;
- out:
- free(buf);
- return err;
- }
- /* Determine highest possible cpu in the system for sparse allocation */
- static void set_max_cpu_num(void)
- {
- const char *mnt;
- char path[PATH_MAX];
- int ret = -1;
- /* set up default */
- max_cpu_num = 4096;
- max_present_cpu_num = 4096;
- mnt = sysfs__mountpoint();
- if (!mnt)
- goto out;
- /* get the highest possible cpu number for a sparse allocation */
- ret = snprintf(path, PATH_MAX, "%s/devices/system/cpu/possible", mnt);
- if (ret >= PATH_MAX) {
- pr_err("sysfs path crossed PATH_MAX(%d) size\n", PATH_MAX);
- goto out;
- }
- ret = get_max_num(path, &max_cpu_num);
- if (ret)
- goto out;
- /* get the highest present cpu number for a sparse allocation */
- ret = snprintf(path, PATH_MAX, "%s/devices/system/cpu/present", mnt);
- if (ret >= PATH_MAX) {
- pr_err("sysfs path crossed PATH_MAX(%d) size\n", PATH_MAX);
- goto out;
- }
- ret = get_max_num(path, &max_present_cpu_num);
- out:
- if (ret)
- pr_err("Failed to read max cpus, using default of %d\n", max_cpu_num);
- }
- /* Determine highest possible node in the system for sparse allocation */
- static void set_max_node_num(void)
- {
- const char *mnt;
- char path[PATH_MAX];
- int ret = -1;
- /* set up default */
- max_node_num = 8;
- mnt = sysfs__mountpoint();
- if (!mnt)
- goto out;
- /* get the highest possible cpu number for a sparse allocation */
- ret = snprintf(path, PATH_MAX, "%s/devices/system/node/possible", mnt);
- if (ret >= PATH_MAX) {
- pr_err("sysfs path crossed PATH_MAX(%d) size\n", PATH_MAX);
- goto out;
- }
- ret = get_max_num(path, &max_node_num);
- out:
- if (ret)
- pr_err("Failed to read max nodes, using default of %d\n", max_node_num);
- }
- int cpu__max_node(void)
- {
- if (unlikely(!max_node_num))
- set_max_node_num();
- return max_node_num;
- }
- int cpu__max_cpu(void)
- {
- if (unlikely(!max_cpu_num))
- set_max_cpu_num();
- return max_cpu_num;
- }
- int cpu__max_present_cpu(void)
- {
- if (unlikely(!max_present_cpu_num))
- set_max_cpu_num();
- return max_present_cpu_num;
- }
- int cpu__get_node(int cpu)
- {
- if (unlikely(cpunode_map == NULL)) {
- pr_debug("cpu_map not initialized\n");
- return -1;
- }
- return cpunode_map[cpu];
- }
- static int init_cpunode_map(void)
- {
- int i;
- set_max_cpu_num();
- set_max_node_num();
- cpunode_map = calloc(max_cpu_num, sizeof(int));
- if (!cpunode_map) {
- pr_err("%s: calloc failed\n", __func__);
- return -1;
- }
- for (i = 0; i < max_cpu_num; i++)
- cpunode_map[i] = -1;
- return 0;
- }
- int cpu__setup_cpunode_map(void)
- {
- struct dirent *dent1, *dent2;
- DIR *dir1, *dir2;
- unsigned int cpu, mem;
- char buf[PATH_MAX];
- char path[PATH_MAX];
- const char *mnt;
- int n;
- /* initialize globals */
- if (init_cpunode_map())
- return -1;
- mnt = sysfs__mountpoint();
- if (!mnt)
- return 0;
- n = snprintf(path, PATH_MAX, "%s/devices/system/node", mnt);
- if (n >= PATH_MAX) {
- pr_err("sysfs path crossed PATH_MAX(%d) size\n", PATH_MAX);
- return -1;
- }
- dir1 = opendir(path);
- if (!dir1)
- return 0;
- /* walk tree and setup map */
- while ((dent1 = readdir(dir1)) != NULL) {
- if (dent1->d_type != DT_DIR || sscanf(dent1->d_name, "node%u", &mem) < 1)
- continue;
- n = snprintf(buf, PATH_MAX, "%s/%s", path, dent1->d_name);
- if (n >= PATH_MAX) {
- pr_err("sysfs path crossed PATH_MAX(%d) size\n", PATH_MAX);
- continue;
- }
- dir2 = opendir(buf);
- if (!dir2)
- continue;
- while ((dent2 = readdir(dir2)) != NULL) {
- if (dent2->d_type != DT_LNK || sscanf(dent2->d_name, "cpu%u", &cpu) < 1)
- continue;
- cpunode_map[cpu] = mem;
- }
- closedir(dir2);
- }
- closedir(dir1);
- return 0;
- }
- bool cpu_map__has(struct perf_cpu_map *cpus, int cpu)
- {
- return perf_cpu_map__idx(cpus, cpu) != -1;
- }
- int cpu_map__cpu(struct perf_cpu_map *cpus, int idx)
- {
- return cpus->map[idx];
- }
- size_t cpu_map__snprint(struct perf_cpu_map *map, char *buf, size_t size)
- {
- int i, cpu, start = -1;
- bool first = true;
- size_t ret = 0;
- #define COMMA first ? "" : ","
- for (i = 0; i < map->nr + 1; i++) {
- bool last = i == map->nr;
- cpu = last ? INT_MAX : map->map[i];
- if (start == -1) {
- start = i;
- if (last) {
- ret += snprintf(buf + ret, size - ret,
- "%s%d", COMMA,
- map->map[i]);
- }
- } else if (((i - start) != (cpu - map->map[start])) || last) {
- int end = i - 1;
- if (start == end) {
- ret += snprintf(buf + ret, size - ret,
- "%s%d", COMMA,
- map->map[start]);
- } else {
- ret += snprintf(buf + ret, size - ret,
- "%s%d-%d", COMMA,
- map->map[start], map->map[end]);
- }
- first = false;
- start = i;
- }
- }
- #undef COMMA
- pr_debug2("cpumask list: %s\n", buf);
- return ret;
- }
- static char hex_char(unsigned char val)
- {
- if (val < 10)
- return val + '0';
- if (val < 16)
- return val - 10 + 'a';
- return '?';
- }
- size_t cpu_map__snprint_mask(struct perf_cpu_map *map, char *buf, size_t size)
- {
- int i, cpu;
- char *ptr = buf;
- unsigned char *bitmap;
- int last_cpu = cpu_map__cpu(map, map->nr - 1);
- if (buf == NULL)
- return 0;
- bitmap = zalloc(last_cpu / 8 + 1);
- if (bitmap == NULL) {
- buf[0] = '\0';
- return 0;
- }
- for (i = 0; i < map->nr; i++) {
- cpu = cpu_map__cpu(map, i);
- bitmap[cpu / 8] |= 1 << (cpu % 8);
- }
- for (cpu = last_cpu / 4 * 4; cpu >= 0; cpu -= 4) {
- unsigned char bits = bitmap[cpu / 8];
- if (cpu % 8)
- bits >>= 4;
- else
- bits &= 0xf;
- *ptr++ = hex_char(bits);
- if ((cpu % 32) == 0 && cpu > 0)
- *ptr++ = ',';
- }
- *ptr = '\0';
- free(bitmap);
- buf[size - 1] = '\0';
- return ptr - buf;
- }
- const struct perf_cpu_map *cpu_map__online(void) /* thread unsafe */
- {
- static const struct perf_cpu_map *online = NULL;
- if (!online)
- online = perf_cpu_map__new(NULL); /* from /sys/devices/system/cpu/online */
- return online;
- }
|