[mirror_ubuntu-hirsute-kernel.git] / tools / perf / util / cpumap.c

// SPDX-License-Identifier: GPL-2.0
#include <api/fs/fs.h>
#include "../perf.h"
#include "cpumap.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__default_new(void)
{
	struct perf_cpu_map *cpus;
	int nr_cpus;

	nr_cpus = sysconf(_SC_NPROCESSORS_ONLN);
	if (nr_cpus < 0)
		return NULL;

	cpus = malloc(sizeof(*cpus) + nr_cpus * sizeof(int));
	if (cpus != NULL) {
		int i;
		for (i = 0; i < nr_cpus; ++i)
			cpus->map[i] = i;

		cpus->nr = nr_cpus;
		refcount_set(&cpus->refcnt, 1);
	}

	return cpus;
}

static struct perf_cpu_map *cpu_map__trim_new(int nr_cpus, int *tmp_cpus)
{
	size_t payload_size = nr_cpus * sizeof(int);
	struct perf_cpu_map *cpus = malloc(sizeof(*cpus) + payload_size);

	if (cpus != NULL) {
		cpus->nr = nr_cpus;
		memcpy(cpus->map, tmp_cpus, payload_size);
		refcount_set(&cpus->refcnt, 1);
	}

	return cpus;
}

struct perf_cpu_map *cpu_map__read(FILE *file)
{
	struct perf_cpu_map *cpus = NULL;
	int nr_cpus = 0;
	int *tmp_cpus = NULL, *tmp;
	int max_entries = 0;
	int n, cpu, prev;
	char sep;

	sep = 0;
	prev = -1;
	for (;;) {
		n = fscanf(file, "%u%c", &cpu, &sep);
		if (n <= 0)
			break;
		if (prev >= 0) {
			int new_max = nr_cpus + cpu - prev - 1;

			if (new_max >= max_entries) {
				max_entries = new_max + MAX_NR_CPUS / 2;
				tmp = realloc(tmp_cpus, max_entries * sizeof(int));
				if (tmp == NULL)
					goto out_free_tmp;
				tmp_cpus = tmp;
			}

			while (++prev < cpu)
				tmp_cpus[nr_cpus++] = prev;
		}
		if (nr_cpus == max_entries) {
			max_entries += MAX_NR_CPUS;
			tmp = realloc(tmp_cpus, max_entries * sizeof(int));
			if (tmp == NULL)
				goto out_free_tmp;
			tmp_cpus = tmp;
		}

		tmp_cpus[nr_cpus++] = cpu;
		if (n == 2 && sep == '-')
			prev = cpu;
		else
			prev = -1;
		if (n == 1 || sep == '\n')
			break;
	}

	if (nr_cpus > 0)
		cpus = cpu_map__trim_new(nr_cpus, tmp_cpus);
	else
		cpus = cpu_map__default_new();
out_free_tmp:
	free(tmp_cpus);
	return cpus;
}

static struct perf_cpu_map *cpu_map__read_all_cpu_map(void)
{
	struct perf_cpu_map *cpus = NULL;
	FILE *onlnf;

	onlnf = fopen("/sys/devices/system/cpu/online", "r");
	if (!onlnf)
		return cpu_map__default_new();

	cpus = cpu_map__read(onlnf);
	fclose(onlnf);
	return cpus;
}

struct perf_cpu_map *cpu_map__new(const char *cpu_list)
{
	struct perf_cpu_map *cpus = NULL;
	unsigned long start_cpu, end_cpu = 0;
	char *p = NULL;
	int i, nr_cpus = 0;
	int *tmp_cpus = NULL, *tmp;
	int max_entries = 0;

	if (!cpu_list)
		return cpu_map__read_all_cpu_map();

	/*
	 * must handle the case of empty cpumap to cover
	 * TOPOLOGY header for NUMA nodes with no CPU
	 * ( e.g., because of CPU hotplug)
	 */
	if (!isdigit(*cpu_list) && *cpu_list != '\0')
		goto out;

	while (isdigit(*cpu_list)) {
		p = NULL;
		start_cpu = strtoul(cpu_list, &p, 0);
		if (start_cpu >= INT_MAX
		    || (*p != '\0' && *p != ',' && *p != '-'))
			goto invalid;

		if (*p == '-') {
			cpu_list = ++p;
			p = NULL;
			end_cpu = strtoul(cpu_list, &p, 0);

			if (end_cpu >= INT_MAX || (*p != '\0' && *p != ','))
				goto invalid;

			if (end_cpu < start_cpu)
				goto invalid;
		} else {
			end_cpu = start_cpu;
		}

		for (; start_cpu <= end_cpu; start_cpu++) {
			/* check for duplicates */
			for (i = 0; i < nr_cpus; i++)
				if (tmp_cpus[i] == (int)start_cpu)
					goto invalid;

			if (nr_cpus == max_entries) {
				max_entries += MAX_NR_CPUS;
				tmp = realloc(tmp_cpus, max_entries * sizeof(int));
				if (tmp == NULL)
					goto invalid;
				tmp_cpus = tmp;
			}
			tmp_cpus[nr_cpus++] = (int)start_cpu;
		}
		if (*p)
			++p;

		cpu_list = p;
	}

	if (nr_cpus > 0)
		cpus = cpu_map__trim_new(nr_cpus, tmp_cpus);
	else if (*cpu_list != '\0')
		cpus = cpu_map__default_new();
	else
		cpus = perf_cpu_map__dummy_new();
invalid:
	free(tmp_cpus);
out:
	return cpus;
}

static struct perf_cpu_map *cpu_map__from_entries(struct cpu_map_entries *cpus)
{
	struct perf_cpu_map *map;

	map = 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 cpu_map_mask *mask)
{
	struct perf_cpu_map *map;
	int nr, nbits = mask->nr * mask->long_size * BITS_PER_BYTE;

	nr = bitmap_weight(mask->mask, nbits);

	map = 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 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 cpu_map_mask *)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 *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_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__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);
}

/* 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 cpu_map__idx(cpus, cpu) != -1;
}

int cpu_map__idx(struct perf_cpu_map *cpus, int cpu)
{
	int i;

	for (i = 0; i < cpus->nr; ++i) {
		if (cpus->map[i] == cpu)
			return i;
	}

	return -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);

	bitmap = zalloc((last_cpu + 7) / 8);
	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 = cpu_map__new(NULL); /* from /sys/devices/system/cpu/online */

	return online;
}
Commit	Line	Data
b2441318	1	// SPDX-License-Identifier: GPL-2.0
cd0cfad7	2	#include <api/fs/fs.h>
a12b51c4 PM	3	#include "../perf.h"
	4	#include "cpumap.h"
	5	#include <assert.h>
76b31a29	6	#include <dirent.h>
a12b51c4	7	#include <stdio.h>
86ee6e18	8	#include <stdlib.h>
f77b57ad	9	#include <linux/bitmap.h>
f30a79b0	10	#include "asm/bug.h"
a12b51c4	11
3052ba56	12	#include <linux/ctype.h>
7f7c536f	13	#include <linux/zalloc.h>
3d689ed6	14
5ac76283	15	static int max_cpu_num;
92a7e127	16	static int max_present_cpu_num;
5ac76283 ACM	17	static int max_node_num;
	18	static int *cpunode_map;
	19
f854839b	20	static struct perf_cpu_map *cpu_map__default_new(void)
a12b51c4	21	{
f854839b	22	struct perf_cpu_map *cpus;
60d567e2	23	int nr_cpus;
a12b51c4 PM	24
a12b51c4 PM	25	nr_cpus = sysconf(_SC_NPROCESSORS_ONLN);
60d567e2 ACM	26	if (nr_cpus < 0)
	27	return NULL;
	28
	29	cpus = malloc(sizeof(cpus) + nr_cpus sizeof(int));
	30	if (cpus != NULL) {
	31	int i;
	32	for (i = 0; i < nr_cpus; ++i)
	33	cpus->map[i] = i;
a12b51c4	34
60d567e2	35	cpus->nr = nr_cpus;
ec09a42a	36	refcount_set(&cpus->refcnt, 1);
60d567e2	37	}
a12b51c4	38
60d567e2	39	return cpus;
a12b51c4 PM	40	}
a12b51c4 PM	41
f854839b	42	static struct perf_cpu_map cpu_map__trim_new(int nr_cpus, int tmp_cpus)
a12b51c4	43	{
60d567e2	44	size_t payload_size = nr_cpus * sizeof(int);
f854839b	45	struct perf_cpu_map cpus = malloc(sizeof(cpus) + payload_size);
60d567e2 ACM	46
	47	if (cpus != NULL) {
	48	cpus->nr = nr_cpus;
	49	memcpy(cpus->map, tmp_cpus, payload_size);
ec09a42a	50	refcount_set(&cpus->refcnt, 1);
60d567e2 ACM	51	}
	52
	53	return cpus;
	54	}
	55
f854839b	56	struct perf_cpu_map cpu_map__read(FILE file)
60d567e2	57	{
f854839b	58	struct perf_cpu_map *cpus = NULL;
a12b51c4	59	int nr_cpus = 0;
60d567e2 ACM	60	int tmp_cpus = NULL, tmp;
60d567e2 ACM	61	int max_entries = 0;
a12b51c4 PM	62	int n, cpu, prev;
	63	char sep;
	64
a12b51c4 PM	65	sep = 0;
	66	prev = -1;
	67	for (;;) {
7ae92e74	68	n = fscanf(file, "%u%c", &cpu, &sep);
a12b51c4 PM	69	if (n <= 0)
	70	break;
	71	if (prev >= 0) {
60d567e2 ACM	72	int new_max = nr_cpus + cpu - prev - 1;
	73
	74	if (new_max >= max_entries) {
	75	max_entries = new_max + MAX_NR_CPUS / 2;
	76	tmp = realloc(tmp_cpus, max_entries * sizeof(int));
	77	if (tmp == NULL)
	78	goto out_free_tmp;
	79	tmp_cpus = tmp;
	80	}
	81
a12b51c4	82	while (++prev < cpu)
60d567e2 ACM	83	tmp_cpus[nr_cpus++] = prev;
	84	}
	85	if (nr_cpus == max_entries) {
	86	max_entries += MAX_NR_CPUS;
	87	tmp = realloc(tmp_cpus, max_entries * sizeof(int));
	88	if (tmp == NULL)
	89	goto out_free_tmp;
	90	tmp_cpus = tmp;
a12b51c4	91	}
60d567e2 ACM	92
60d567e2 ACM	93	tmp_cpus[nr_cpus++] = cpu;
a12b51c4 PM	94	if (n == 2 && sep == '-')
	95	prev = cpu;
	96	else
	97	prev = -1;
	98	if (n == 1 \|\| sep == '\n')
	99	break;
	100	}
a12b51c4	101
60d567e2 ACM	102	if (nr_cpus > 0)
	103	cpus = cpu_map__trim_new(nr_cpus, tmp_cpus);
	104	else
	105	cpus = cpu_map__default_new();
	106	out_free_tmp:
	107	free(tmp_cpus);
7ae92e74 YZ	108	return cpus;
	109	}
	110
f854839b	111	static struct perf_cpu_map *cpu_map__read_all_cpu_map(void)
7ae92e74	112	{
f854839b	113	struct perf_cpu_map *cpus = NULL;
7ae92e74 YZ	114	FILE *onlnf;
	115
	116	onlnf = fopen("/sys/devices/system/cpu/online", "r");
	117	if (!onlnf)
	118	return cpu_map__default_new();
	119
	120	cpus = cpu_map__read(onlnf);
60d567e2 ACM	121	fclose(onlnf);
60d567e2 ACM	122	return cpus;
a12b51c4	123	}
c45c6ea2	124
f854839b	125	struct perf_cpu_map cpu_map__new(const char cpu_list)
c45c6ea2	126	{
f854839b	127	struct perf_cpu_map *cpus = NULL;
c45c6ea2 SE	128	unsigned long start_cpu, end_cpu = 0;
	129	char *p = NULL;
	130	int i, nr_cpus = 0;
60d567e2 ACM	131	int tmp_cpus = NULL, tmp;
60d567e2 ACM	132	int max_entries = 0;
c45c6ea2 SE	133
c45c6ea2 SE	134	if (!cpu_list)
60d567e2	135	return cpu_map__read_all_cpu_map();
c45c6ea2	136
1497e804 SE	137	/*
	138	* must handle the case of empty cpumap to cover
	139	* TOPOLOGY header for NUMA nodes with no CPU
	140	* ( e.g., because of CPU hotplug)
	141	*/
	142	if (!isdigit(cpu_list) && cpu_list != '\0')
60d567e2	143	goto out;
c45c6ea2 SE	144
	145	while (isdigit(*cpu_list)) {
	146	p = NULL;
	147	start_cpu = strtoul(cpu_list, &p, 0);
	148	if (start_cpu >= INT_MAX
	149	\|\| (p != '\0' && p != ',' && *p != '-'))
	150	goto invalid;
	151
	152	if (*p == '-') {
	153	cpu_list = ++p;
	154	p = NULL;
	155	end_cpu = strtoul(cpu_list, &p, 0);
	156
	157	if (end_cpu >= INT_MAX \|\| (p != '\0' && p != ','))
	158	goto invalid;
	159
	160	if (end_cpu < start_cpu)
	161	goto invalid;
	162	} else {
	163	end_cpu = start_cpu;
	164	}
	165
	166	for (; start_cpu <= end_cpu; start_cpu++) {
	167	/* check for duplicates */
	168	for (i = 0; i < nr_cpus; i++)
60d567e2	169	if (tmp_cpus[i] == (int)start_cpu)
c45c6ea2 SE	170	goto invalid;
c45c6ea2 SE	171
60d567e2 ACM	172	if (nr_cpus == max_entries) {
	173	max_entries += MAX_NR_CPUS;
	174	tmp = realloc(tmp_cpus, max_entries * sizeof(int));
	175	if (tmp == NULL)
	176	goto invalid;
	177	tmp_cpus = tmp;
	178	}
	179	tmp_cpus[nr_cpus++] = (int)start_cpu;
c45c6ea2 SE	180	}
	181	if (*p)
	182	++p;
	183
	184	cpu_list = p;
	185	}
c45c6ea2	186
60d567e2 ACM	187	if (nr_cpus > 0)
60d567e2 ACM	188	cpus = cpu_map__trim_new(nr_cpus, tmp_cpus);
1497e804	189	else if (*cpu_list != '\0')
60d567e2	190	cpus = cpu_map__default_new();
1497e804	191	else
397721e0	192	cpus = perf_cpu_map__dummy_new();
c45c6ea2	193	invalid:
60d567e2 ACM	194	free(tmp_cpus);
	195	out:
	196	return cpus;
	197	}
	198
f854839b	199	static struct perf_cpu_map cpu_map__from_entries(struct cpu_map_entries cpus)
f77b57ad	200	{
f854839b	201	struct perf_cpu_map *map;
f77b57ad JO	202
	203	map = cpu_map__empty_new(cpus->nr);
	204	if (map) {
	205	unsigned i;
	206
15d2b995 JO	207	for (i = 0; i < cpus->nr; i++) {
	208	/*
	209	* Special treatment for -1, which is not real cpu number,
	210	* and we need to use (int) -1 to initialize map[i],
	211	* otherwise it would become 65535.
	212	*/
	213	if (cpus->cpu[i] == (u16) -1)
	214	map->map[i] = -1;
	215	else
	216	map->map[i] = (int) cpus->cpu[i];
	217	}
f77b57ad JO	218	}
	219
	220	return map;
	221	}
	222
f854839b	223	static struct perf_cpu_map cpu_map__from_mask(struct cpu_map_mask mask)
f77b57ad	224	{
f854839b	225	struct perf_cpu_map *map;
f77b57ad JO	226	int nr, nbits = mask->nr * mask->long_size * BITS_PER_BYTE;
	227
	228	nr = bitmap_weight(mask->mask, nbits);
	229
	230	map = cpu_map__empty_new(nr);
	231	if (map) {
	232	int cpu, i = 0;
	233
	234	for_each_set_bit(cpu, mask->mask, nbits)
	235	map->map[i++] = cpu;
	236	}
	237	return map;
	238
	239	}
	240
f854839b	241	struct perf_cpu_map cpu_map__new_data(struct cpu_map_data data)
f77b57ad JO	242	{
	243	if (data->type == PERF_CPU_MAP__CPUS)
	244	return cpu_map__from_entries((struct cpu_map_entries *)data->data);
	245	else
	246	return cpu_map__from_mask((struct cpu_map_mask *)data->data);
	247	}
	248
f854839b	249	size_t cpu_map__fprintf(struct perf_cpu_map map, FILE fp)
9ae7d335	250	{
a24020e6 JO	251	#define BUFSIZE 1024
a24020e6 JO	252	char buf[BUFSIZE];
9ae7d335	253
a24020e6 JO	254	cpu_map__snprint(map, buf, sizeof(buf));
	255	return fprintf(fp, "%s\n", buf);
	256	#undef BUFSIZE
9ae7d335 ACM	257	}
9ae7d335 ACM	258
f854839b	259	struct perf_cpu_map *cpu_map__empty_new(int nr)
2322f573	260	{
f854839b	261	struct perf_cpu_map cpus = malloc(sizeof(cpus) + sizeof(int) * nr);
2322f573 JO	262
	263	if (cpus != NULL) {
	264	int i;
	265
	266	cpus->nr = nr;
	267	for (i = 0; i < nr; i++)
	268	cpus->map[i] = -1;
	269
ec09a42a	270	refcount_set(&cpus->refcnt, 1);
2322f573 JO	271	}
	272
	273	return cpus;
	274	}
	275
5d8cf721	276	static int cpu__get_topology_int(int cpu, const char name, int value)
5ac59a8a	277	{
5ac59a8a	278	char path[PATH_MAX];
5ac59a8a	279
86ee6e18	280	snprintf(path, PATH_MAX,
5d8cf721	281	"devices/system/cpu/cpu%d/topology/%s", cpu, name);
5ac59a8a	282
5d8cf721 ACM	283	return sysfs__read_int(path, value);
5d8cf721 ACM	284	}
193b6bd3	285
5d8cf721 ACM	286	int cpu_map__get_socket_id(int cpu)
	287	{
	288	int value, ret = cpu__get_topology_int(cpu, "physical_package_id", &value);
	289	return ret ?: value;
193b6bd3 KL	290	}
193b6bd3 KL	291
f854839b	292	int cpu_map__get_socket(struct perf_cpu_map map, int idx, void data __maybe_unused)
193b6bd3 KL	293	{
	294	int cpu;
	295
	296	if (idx > map->nr)
	297	return -1;
	298
	299	cpu = map->map[idx];
	300
	301	return cpu_map__get_socket_id(cpu);
5ac59a8a SE	302	}
5ac59a8a SE	303
86ee6e18	304	static int cmp_ids(const void a, const void b)
5ac59a8a	305	{
86ee6e18 SE	306	return (int )a - (int )b;
	307	}
	308
f854839b JO	309	int cpu_map__build_map(struct perf_cpu_map cpus, struct perf_cpu_map *res,
f854839b JO	310	int (f)(struct perf_cpu_map map, int cpu, void *data),
1fe7a300	311	void *data)
86ee6e18	312	{
f854839b	313	struct perf_cpu_map *c;
5ac59a8a SE	314	int nr = cpus->nr;
	315	int cpu, s1, s2;
	316
86ee6e18 SE	317	/* allocate as much as possible */
	318	c = calloc(1, sizeof(c) + nr sizeof(int));
	319	if (!c)
5ac59a8a SE	320	return -1;
	321
	322	for (cpu = 0; cpu < nr; cpu++) {
1fe7a300	323	s1 = f(cpus, cpu, data);
86ee6e18 SE	324	for (s2 = 0; s2 < c->nr; s2++) {
86ee6e18 SE	325	if (s1 == c->map[s2])
5ac59a8a SE	326	break;
5ac59a8a SE	327	}
86ee6e18 SE	328	if (s2 == c->nr) {
	329	c->map[c->nr] = s1;
	330	c->nr++;
5ac59a8a SE	331	}
5ac59a8a SE	332	}
86ee6e18 SE	333	/* ensure we process id in increasing order */
	334	qsort(c->map, c->nr, sizeof(int), cmp_ids);
	335
ec09a42a	336	refcount_set(&c->refcnt, 1);
86ee6e18	337	*res = c;
5ac59a8a SE	338	return 0;
5ac59a8a SE	339	}
86ee6e18	340
b74d8686 KL	341	int cpu_map__get_die_id(int cpu)
	342	{
	343	int value, ret = cpu__get_topology_int(cpu, "die_id", &value);
	344
	345	return ret ?: value;
	346	}
	347
f854839b	348	int cpu_map__get_die(struct perf_cpu_map map, int idx, void data)
db5742b6 KL	349	{
	350	int cpu, die_id, s;
	351
	352	if (idx > map->nr)
	353	return -1;
	354
	355	cpu = map->map[idx];
	356
	357	die_id = cpu_map__get_die_id(cpu);
	358	/* There is no die_id on legacy system. */
	359	if (die_id == -1)
	360	die_id = 0;
	361
	362	s = cpu_map__get_socket(map, idx, data);
	363	if (s == -1)
	364	return -1;
	365
	366	/*
	367	* Encode socket in bit range 15:8
	368	* die_id is relative to socket, and
	369	* we need a global id. So we combine
	370	* socket + die id
	371	*/
	372	if (WARN_ONCE(die_id >> 8, "The die id number is too big.\n"))
	373	return -1;
	374
	375	if (WARN_ONCE(s >> 8, "The socket id number is too big.\n"))
	376	return -1;
	377
	378	return (s << 8) \| (die_id & 0xff);
	379	}
	380
193b6bd3	381	int cpu_map__get_core_id(int cpu)
12c08a9f	382	{
5d8cf721 ACM	383	int value, ret = cpu__get_topology_int(cpu, "core_id", &value);
5d8cf721 ACM	384	return ret ?: value;
193b6bd3 KL	385	}
193b6bd3 KL	386
f854839b	387	int cpu_map__get_core(struct perf_cpu_map map, int idx, void data)
193b6bd3	388	{
db5742b6	389	int cpu, s_die;
193b6bd3 KL	390
193b6bd3 KL	391	if (idx > map->nr)
12c08a9f SE	392	return -1;
12c08a9f SE	393
193b6bd3 KL	394	cpu = map->map[idx];
	395
	396	cpu = cpu_map__get_core_id(cpu);
	397
db5742b6 KL	398	/* s_die is the combination of socket + die id */
	399	s_die = cpu_map__get_die(map, idx, data);
	400	if (s_die == -1)
12c08a9f SE	401	return -1;
	402
	403	/*
db5742b6 KL	404	* encode socket in bit range 31:24
	405	* encode die id in bit range 23:16
	406	* core_id is relative to socket and die,
12c08a9f	407	* we need a global id. So we combine
db5742b6	408	* socket + die id + core id
12c08a9f	409	*/
db5742b6 KL	410	if (WARN_ONCE(cpu >> 16, "The core id number is too big.\n"))
	411	return -1;
	412
	413	return (s_die << 16) \| (cpu & 0xffff);
12c08a9f SE	414	}
12c08a9f SE	415
f854839b	416	int cpu_map__build_socket_map(struct perf_cpu_map cpus, struct perf_cpu_map *sockp)
86ee6e18	417	{
1fe7a300	418	return cpu_map__build_map(cpus, sockp, cpu_map__get_socket, NULL);
86ee6e18	419	}
12c08a9f	420
f854839b	421	int cpu_map__build_die_map(struct perf_cpu_map cpus, struct perf_cpu_map *diep)
db5742b6 KL	422	{
	423	return cpu_map__build_map(cpus, diep, cpu_map__get_die, NULL);
	424	}
	425
f854839b	426	int cpu_map__build_core_map(struct perf_cpu_map cpus, struct perf_cpu_map *corep)
12c08a9f	427	{
1fe7a300	428	return cpu_map__build_map(cpus, corep, cpu_map__get_core, NULL);
12c08a9f	429	}
7780c25b DZ	430
	431	/* setup simple routines to easily access node numbers given a cpu number */
	432	static int get_max_num(char path, int max)
	433	{
	434	size_t num;
	435	char *buf;
	436	int err = 0;
	437
	438	if (filename__read_str(path, &buf, &num))
	439	return -1;
	440
	441	buf[num] = '\0';
	442
	443	/* start on the right, to find highest node num */
	444	while (--num) {
	445	if ((buf[num] == ',') \|\| (buf[num] == '-')) {
	446	num++;
	447	break;
	448	}
	449	}
	450	if (sscanf(&buf[num], "%d", max) < 1) {
	451	err = -1;
	452	goto out;
	453	}
	454
	455	/* convert from 0-based to 1-based */
	456	(*max)++;
	457
	458	out:
	459	free(buf);
	460	return err;
	461	}
	462
	463	/* Determine highest possible cpu in the system for sparse allocation */
	464	static void set_max_cpu_num(void)
	465	{
	466	const char *mnt;
	467	char path[PATH_MAX];
	468	int ret = -1;
	469
	470	/* set up default */
	471	max_cpu_num = 4096;
92a7e127	472	max_present_cpu_num = 4096;
7780c25b DZ	473
	474	mnt = sysfs__mountpoint();
	475	if (!mnt)
	476	goto out;
	477
	478	/* get the highest possible cpu number for a sparse allocation */
f5b1f4e4	479	ret = snprintf(path, PATH_MAX, "%s/devices/system/cpu/possible", mnt);
7780c25b DZ	480	if (ret == PATH_MAX) {
	481	pr_err("sysfs path crossed PATH_MAX(%d) size\n", PATH_MAX);
	482	goto out;
	483	}
	484
	485	ret = get_max_num(path, &max_cpu_num);
92a7e127 JS	486	if (ret)
	487	goto out;
	488
	489	/* get the highest present cpu number for a sparse allocation */
	490	ret = snprintf(path, PATH_MAX, "%s/devices/system/cpu/present", mnt);
	491	if (ret == PATH_MAX) {
	492	pr_err("sysfs path crossed PATH_MAX(%d) size\n", PATH_MAX);
	493	goto out;
	494	}
	495
	496	ret = get_max_num(path, &max_present_cpu_num);
7780c25b DZ	497
	498	out:
	499	if (ret)
	500	pr_err("Failed to read max cpus, using default of %d\n", max_cpu_num);
	501	}
	502
	503	/* Determine highest possible node in the system for sparse allocation */
	504	static void set_max_node_num(void)
	505	{
	506	const char *mnt;
	507	char path[PATH_MAX];
	508	int ret = -1;
	509
	510	/* set up default */
	511	max_node_num = 8;
	512
	513	mnt = sysfs__mountpoint();
	514	if (!mnt)
	515	goto out;
	516
	517	/* get the highest possible cpu number for a sparse allocation */
	518	ret = snprintf(path, PATH_MAX, "%s/devices/system/node/possible", mnt);
	519	if (ret == PATH_MAX) {
	520	pr_err("sysfs path crossed PATH_MAX(%d) size\n", PATH_MAX);
	521	goto out;
	522	}
	523
	524	ret = get_max_num(path, &max_node_num);
	525
	526	out:
	527	if (ret)
	528	pr_err("Failed to read max nodes, using default of %d\n", max_node_num);
	529	}
	530
5ac76283 ACM	531	int cpu__max_node(void)
	532	{
	533	if (unlikely(!max_node_num))
	534	set_max_node_num();
	535
	536	return max_node_num;
	537	}
	538
	539	int cpu__max_cpu(void)
	540	{
	541	if (unlikely(!max_cpu_num))
	542	set_max_cpu_num();
	543
	544	return max_cpu_num;
	545	}
	546
92a7e127 JS	547	int cpu__max_present_cpu(void)
	548	{
	549	if (unlikely(!max_present_cpu_num))
	550	set_max_cpu_num();
	551
	552	return max_present_cpu_num;
	553	}
	554
	555
5ac76283 ACM	556	int cpu__get_node(int cpu)
	557	{
	558	if (unlikely(cpunode_map == NULL)) {
	559	pr_debug("cpu_map not initialized\n");
	560	return -1;
	561	}
	562
	563	return cpunode_map[cpu];
	564	}
	565
7780c25b DZ	566	static int init_cpunode_map(void)
	567	{
	568	int i;
	569
	570	set_max_cpu_num();
	571	set_max_node_num();
	572
	573	cpunode_map = calloc(max_cpu_num, sizeof(int));
	574	if (!cpunode_map) {
	575	pr_err("%s: calloc failed\n", __func__);
	576	return -1;
	577	}
	578
	579	for (i = 0; i < max_cpu_num; i++)
	580	cpunode_map[i] = -1;
	581
	582	return 0;
	583	}
	584
	585	int cpu__setup_cpunode_map(void)
	586	{
	587	struct dirent dent1, dent2;
	588	DIR dir1, dir2;
	589	unsigned int cpu, mem;
	590	char buf[PATH_MAX];
	591	char path[PATH_MAX];
	592	const char *mnt;
	593	int n;
	594
	595	/* initialize globals */
	596	if (init_cpunode_map())
	597	return -1;
	598
	599	mnt = sysfs__mountpoint();
	600	if (!mnt)
	601	return 0;
	602
	603	n = snprintf(path, PATH_MAX, "%s/devices/system/node", mnt);
	604	if (n == PATH_MAX) {
	605	pr_err("sysfs path crossed PATH_MAX(%d) size\n", PATH_MAX);
	606	return -1;
	607	}
	608
	609	dir1 = opendir(path);
	610	if (!dir1)
	611	return 0;
	612
	613	/* walk tree and setup map */
	614	while ((dent1 = readdir(dir1)) != NULL) {
	615	if (dent1->d_type != DT_DIR \|\| sscanf(dent1->d_name, "node%u", &mem) < 1)
	616	continue;
	617
	618	n = snprintf(buf, PATH_MAX, "%s/%s", path, dent1->d_name);
	619	if (n == PATH_MAX) {
	620	pr_err("sysfs path crossed PATH_MAX(%d) size\n", PATH_MAX);
	621	continue;
	622	}
	623
	624	dir2 = opendir(buf);
	625	if (!dir2)
	626	continue;
	627	while ((dent2 = readdir(dir2)) != NULL) {
	628	if (dent2->d_type != DT_LNK \|\| sscanf(dent2->d_name, "cpu%u", &cpu) < 1)
	629	continue;
630	cpunode_map[cpu] = mem;
631	}
632	closedir(dir2);
633	}
634	closedir(dir1);
635	return 0;
636	}
e632aa69	637
f854839b	638	bool cpu_map__has(struct perf_cpu_map *cpus, int cpu)
9a6c582d MR	639	{
	640	return cpu_map__idx(cpus, cpu) != -1;
	641	}
	642
f854839b	643	int cpu_map__idx(struct perf_cpu_map *cpus, int cpu)
e632aa69 JO	644	{
	645	int i;
	646
	647	for (i = 0; i < cpus->nr; ++i) {
	648	if (cpus->map[i] == cpu)
9a6c582d	649	return i;
e632aa69 JO	650	}
e632aa69 JO	651
9a6c582d MR	652	return -1;
	653	}
	654
f854839b	655	int cpu_map__cpu(struct perf_cpu_map *cpus, int idx)
9a6c582d MR	656	{
9a6c582d MR	657	return cpus->map[idx];
e632aa69	658	}
a24020e6	659
f854839b	660	size_t cpu_map__snprint(struct perf_cpu_map map, char buf, size_t size)
a24020e6 JO	661	{
	662	int i, cpu, start = -1;
	663	bool first = true;
	664	size_t ret = 0;
	665
	666	#define COMMA first ? "" : ","
	667
	668	for (i = 0; i < map->nr + 1; i++) {
	669	bool last = i == map->nr;
	670
	671	cpu = last ? INT_MAX : map->map[i];
	672
	673	if (start == -1) {
	674	start = i;
	675	if (last) {
	676	ret += snprintf(buf + ret, size - ret,
	677	"%s%d", COMMA,
	678	map->map[i]);
	679	}
	680	} else if (((i - start) != (cpu - map->map[start])) \|\| last) {
	681	int end = i - 1;
	682
	683	if (start == end) {
	684	ret += snprintf(buf + ret, size - ret,
	685	"%s%d", COMMA,
	686	map->map[start]);
	687	} else {
	688	ret += snprintf(buf + ret, size - ret,
	689	"%s%d-%d", COMMA,
	690	map->map[start], map->map[end]);
	691	}
	692	first = false;
	693	start = i;
	694	}
	695	}
	696
	697	#undef COMMA
	698
deb83da1	699	pr_debug2("cpumask list: %s\n", buf);
a24020e6 JO	700	return ret;
a24020e6 JO	701	}
4400ac8a NK	702
	703	static char hex_char(unsigned char val)
	704	{
	705	if (val < 10)
	706	return val + '0';
	707	if (val < 16)
	708	return val - 10 + 'a';
	709	return '?';
	710	}
	711
f854839b	712	size_t cpu_map__snprint_mask(struct perf_cpu_map map, char buf, size_t size)
4400ac8a NK	713	{
	714	int i, cpu;
	715	char *ptr = buf;
	716	unsigned char *bitmap;
	717	int last_cpu = cpu_map__cpu(map, map->nr - 1);
	718
	719	bitmap = zalloc((last_cpu + 7) / 8);
	720	if (bitmap == NULL) {
	721	buf[0] = '\0';
	722	return 0;
	723	}
	724
	725	for (i = 0; i < map->nr; i++) {
	726	cpu = cpu_map__cpu(map, i);
	727	bitmap[cpu / 8] \|= 1 << (cpu % 8);
	728	}
	729
	730	for (cpu = last_cpu / 4 * 4; cpu >= 0; cpu -= 4) {
	731	unsigned char bits = bitmap[cpu / 8];
	732
	733	if (cpu % 8)
	734	bits >>= 4;
	735	else
	736	bits &= 0xf;
	737
	738	*ptr++ = hex_char(bits);
	739	if ((cpu % 32) == 0 && cpu > 0)
	740	*ptr++ = ',';
	741	}
	742	*ptr = '\0';
	743	free(bitmap);
	744
	745	buf[size - 1] = '\0';
	746	return ptr - buf;
	747	}
f13de660	748
f854839b	749	const struct perf_cpu_map cpu_map__online(void) / thread unsafe */
f13de660	750	{
f854839b	751	static const struct perf_cpu_map *online = NULL;
f13de660 AB	752
	753	if (!online)
	754	online = cpu_map__new(NULL); /* from /sys/devices/system/cpu/online */
	755
	756	return online;
	757	}