Orange Pi5 kernel

^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   1) // SPDX-License-Identifier: GPL-2.0
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   2) #include <dirent.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   3) #include <errno.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   4) #include <limits.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   5) #include <stdbool.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   6) #include <stdlib.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   7) #include <stdio.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   8) #include <sys/types.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300   9) #include <sys/stat.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  10) #include <unistd.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  11) #include "string2.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  12) #include "strlist.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  13) #include <string.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  14) #include <api/fs/fs.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  15) #include <linux/string.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  16) #include <linux/zalloc.h>
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  17) #include "asm/bug.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  18) #include "thread_map.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  19) #include "debug.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  20) #include "event.h"
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  21) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  22) /* Skip "." and ".." directories */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  23) static int filter(const struct dirent *dir)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  24) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  25) 	if (dir->d_name[0] == '.')
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  26) 		return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  27) 	else
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  28) 		return 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  29) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  30) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  31) #define thread_map__alloc(__nr) perf_thread_map__realloc(NULL, __nr)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  32) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  33) struct perf_thread_map *thread_map__new_by_pid(pid_t pid)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  34) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  35) 	struct perf_thread_map *threads;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  36) 	char name[256];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  37) 	int items;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  38) 	struct dirent **namelist = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  39) 	int i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  40) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  41) 	sprintf(name, "/proc/%d/task", pid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  42) 	items = scandir(name, &namelist, filter, NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  43) 	if (items <= 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  44) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  45) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  46) 	threads = thread_map__alloc(items);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  47) 	if (threads != NULL) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  48) 		for (i = 0; i < items; i++)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  49) 			perf_thread_map__set_pid(threads, i, atoi(namelist[i]->d_name));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  50) 		threads->nr = items;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  51) 		refcount_set(&threads->refcnt, 1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  52) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  53) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  54) 	for (i=0; i<items; i++)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  55) 		zfree(&namelist[i]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  56) 	free(namelist);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  57) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  58) 	return threads;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  59) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  60) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  61) struct perf_thread_map *thread_map__new_by_tid(pid_t tid)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  62) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  63) 	struct perf_thread_map *threads = thread_map__alloc(1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  64) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  65) 	if (threads != NULL) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  66) 		perf_thread_map__set_pid(threads, 0, tid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  67) 		threads->nr = 1;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  68) 		refcount_set(&threads->refcnt, 1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  69) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  70) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  71) 	return threads;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  72) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  73) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  74) static struct perf_thread_map *__thread_map__new_all_cpus(uid_t uid)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  75) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  76) 	DIR *proc;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  77) 	int max_threads = 32, items, i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  78) 	char path[NAME_MAX + 1 + 6];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  79) 	struct dirent *dirent, **namelist = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  80) 	struct perf_thread_map *threads = thread_map__alloc(max_threads);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  81) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  82) 	if (threads == NULL)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  83) 		goto out;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  84) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  85) 	proc = opendir("/proc");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  86) 	if (proc == NULL)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  87) 		goto out_free_threads;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  88) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  89) 	threads->nr = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  90) 	refcount_set(&threads->refcnt, 1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  91) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  92) 	while ((dirent = readdir(proc)) != NULL) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  93) 		char *end;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  94) 		bool grow = false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  95) 		pid_t pid = strtol(dirent->d_name, &end, 10);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  96) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  97) 		if (*end) /* only interested in proper numerical dirents */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  98) 			continue;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300  99) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 100) 		snprintf(path, sizeof(path), "/proc/%s", dirent->d_name);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 101) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 102) 		if (uid != UINT_MAX) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 103) 			struct stat st;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 104) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 105) 			if (stat(path, &st) != 0 || st.st_uid != uid)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 106) 				continue;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 107) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 108) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 109) 		snprintf(path, sizeof(path), "/proc/%d/task", pid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 110) 		items = scandir(path, &namelist, filter, NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 111) 		if (items <= 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 112) 			goto out_free_closedir;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 113) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 114) 		while (threads->nr + items >= max_threads) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 115) 			max_threads *= 2;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 116) 			grow = true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 117) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 118) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 119) 		if (grow) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 120) 			struct perf_thread_map *tmp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 121) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 122) 			tmp = perf_thread_map__realloc(threads, max_threads);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 123) 			if (tmp == NULL)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 124) 				goto out_free_namelist;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 125) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 126) 			threads = tmp;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 127) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 128) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 129) 		for (i = 0; i < items; i++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 130) 			perf_thread_map__set_pid(threads, threads->nr + i,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 131) 						    atoi(namelist[i]->d_name));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 132) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 133) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 134) 		for (i = 0; i < items; i++)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 135) 			zfree(&namelist[i]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 136) 		free(namelist);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 137) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 138) 		threads->nr += items;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 139) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 140) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 141) out_closedir:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 142) 	closedir(proc);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 143) out:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 144) 	return threads;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 145) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 146) out_free_threads:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 147) 	free(threads);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 148) 	return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 149) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 150) out_free_namelist:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 151) 	for (i = 0; i < items; i++)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 152) 		zfree(&namelist[i]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 153) 	free(namelist);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 154) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 155) out_free_closedir:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 156) 	zfree(&threads);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 157) 	goto out_closedir;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 158) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 159) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 160) struct perf_thread_map *thread_map__new_all_cpus(void)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 161) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 162) 	return __thread_map__new_all_cpus(UINT_MAX);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 163) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 164) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 165) struct perf_thread_map *thread_map__new_by_uid(uid_t uid)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 166) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 167) 	return __thread_map__new_all_cpus(uid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 168) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 169) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 170) struct perf_thread_map *thread_map__new(pid_t pid, pid_t tid, uid_t uid)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 171) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 172) 	if (pid != -1)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 173) 		return thread_map__new_by_pid(pid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 174) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 175) 	if (tid == -1 && uid != UINT_MAX)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 176) 		return thread_map__new_by_uid(uid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 177) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 178) 	return thread_map__new_by_tid(tid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 179) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 180) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 181) static struct perf_thread_map *thread_map__new_by_pid_str(const char *pid_str)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 182) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 183) 	struct perf_thread_map *threads = NULL, *nt;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 184) 	char name[256];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 185) 	int items, total_tasks = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 186) 	struct dirent **namelist = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 187) 	int i, j = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 188) 	pid_t pid, prev_pid = INT_MAX;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 189) 	char *end_ptr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 190) 	struct str_node *pos;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 191) 	struct strlist_config slist_config = { .dont_dupstr = true, };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 192) 	struct strlist *slist = strlist__new(pid_str, &slist_config);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 193) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 194) 	if (!slist)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 195) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 196) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 197) 	strlist__for_each_entry(pos, slist) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 198) 		pid = strtol(pos->s, &end_ptr, 10);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 199) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 200) 		if (pid == INT_MIN || pid == INT_MAX ||
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 201) 		    (*end_ptr != '\0' && *end_ptr != ','))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 202) 			goto out_free_threads;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 203) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 204) 		if (pid == prev_pid)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 205) 			continue;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 206) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 207) 		sprintf(name, "/proc/%d/task", pid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 208) 		items = scandir(name, &namelist, filter, NULL);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 209) 		if (items <= 0)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 210) 			goto out_free_threads;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 211) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 212) 		total_tasks += items;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 213) 		nt = perf_thread_map__realloc(threads, total_tasks);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 214) 		if (nt == NULL)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 215) 			goto out_free_namelist;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 216) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 217) 		threads = nt;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 218) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 219) 		for (i = 0; i < items; i++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 220) 			perf_thread_map__set_pid(threads, j++, atoi(namelist[i]->d_name));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 221) 			zfree(&namelist[i]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 222) 		}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 223) 		threads->nr = total_tasks;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 224) 		free(namelist);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 225) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 226) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 227) out:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 228) 	strlist__delete(slist);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 229) 	if (threads)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 230) 		refcount_set(&threads->refcnt, 1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 231) 	return threads;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 232) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 233) out_free_namelist:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 234) 	for (i = 0; i < items; i++)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 235) 		zfree(&namelist[i]);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 236) 	free(namelist);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 237) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 238) out_free_threads:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 239) 	zfree(&threads);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 240) 	goto out;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 241) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 242) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 243) struct perf_thread_map *thread_map__new_by_tid_str(const char *tid_str)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 244) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 245) 	struct perf_thread_map *threads = NULL, *nt;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 246) 	int ntasks = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 247) 	pid_t tid, prev_tid = INT_MAX;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 248) 	char *end_ptr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 249) 	struct str_node *pos;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 250) 	struct strlist_config slist_config = { .dont_dupstr = true, };
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 251) 	struct strlist *slist;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 252) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 253) 	/* perf-stat expects threads to be generated even if tid not given */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 254) 	if (!tid_str)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 255) 		return perf_thread_map__new_dummy();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 256) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 257) 	slist = strlist__new(tid_str, &slist_config);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 258) 	if (!slist)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 259) 		return NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 260) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 261) 	strlist__for_each_entry(pos, slist) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 262) 		tid = strtol(pos->s, &end_ptr, 10);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 263) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 264) 		if (tid == INT_MIN || tid == INT_MAX ||
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 265) 		    (*end_ptr != '\0' && *end_ptr != ','))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 266) 			goto out_free_threads;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 267) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 268) 		if (tid == prev_tid)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 269) 			continue;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 270) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 271) 		ntasks++;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 272) 		nt = perf_thread_map__realloc(threads, ntasks);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 273) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 274) 		if (nt == NULL)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 275) 			goto out_free_threads;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 276) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 277) 		threads = nt;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 278) 		perf_thread_map__set_pid(threads, ntasks - 1, tid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 279) 		threads->nr = ntasks;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 280) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 281) out:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 282) 	if (threads)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 283) 		refcount_set(&threads->refcnt, 1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 284) 	return threads;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 285) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 286) out_free_threads:
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 287) 	zfree(&threads);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 288) 	strlist__delete(slist);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 289) 	goto out;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 290) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 291) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 292) struct perf_thread_map *thread_map__new_str(const char *pid, const char *tid,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 293) 				       uid_t uid, bool all_threads)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 294) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 295) 	if (pid)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 296) 		return thread_map__new_by_pid_str(pid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 297) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 298) 	if (!tid && uid != UINT_MAX)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 299) 		return thread_map__new_by_uid(uid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 300) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 301) 	if (all_threads)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 302) 		return thread_map__new_all_cpus();
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 303) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 304) 	return thread_map__new_by_tid_str(tid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 305) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 306) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 307) size_t thread_map__fprintf(struct perf_thread_map *threads, FILE *fp)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 308) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 309) 	int i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 310) 	size_t printed = fprintf(fp, "%d thread%s: ",
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 311) 				 threads->nr, threads->nr > 1 ? "s" : "");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 312) 	for (i = 0; i < threads->nr; ++i)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 313) 		printed += fprintf(fp, "%s%d", i ? ", " : "", perf_thread_map__pid(threads, i));
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 314) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 315) 	return printed + fprintf(fp, "\n");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 316) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 317) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 318) static int get_comm(char **comm, pid_t pid)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 319) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 320) 	char *path;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 321) 	size_t size;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 322) 	int err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 323) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 324) 	if (asprintf(&path, "%s/%d/comm", procfs__mountpoint(), pid) == -1)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 325) 		return -ENOMEM;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 326) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 327) 	err = filename__read_str(path, comm, &size);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 328) 	if (!err) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 329) 		/*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 330) 		 * We're reading 16 bytes, while filename__read_str
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 331) 		 * allocates data per BUFSIZ bytes, so we can safely
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 332) 		 * mark the end of the string.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 333) 		 */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 334) 		(*comm)[size] = 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 335) 		strim(*comm);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 336) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 337) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 338) 	free(path);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 339) 	return err;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 340) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 341) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 342) static void comm_init(struct perf_thread_map *map, int i)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 343) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 344) 	pid_t pid = perf_thread_map__pid(map, i);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 345) 	char *comm = NULL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 346) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 347) 	/* dummy pid comm initialization */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 348) 	if (pid == -1) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 349) 		map->map[i].comm = strdup("dummy");
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 350) 		return;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 351) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 352) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 353) 	/*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 354) 	 * The comm name is like extra bonus ;-),
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 355) 	 * so just warn if we fail for any reason.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 356) 	 */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 357) 	if (get_comm(&comm, pid))
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 358) 		pr_warning("Couldn't resolve comm name for pid %d\n", pid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 359) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 360) 	map->map[i].comm = comm;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 361) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 362) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 363) void thread_map__read_comms(struct perf_thread_map *threads)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 364) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 365) 	int i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 366) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 367) 	for (i = 0; i < threads->nr; ++i)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 368) 		comm_init(threads, i);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 369) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 370) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 371) static void thread_map__copy_event(struct perf_thread_map *threads,
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 372) 				   struct perf_record_thread_map *event)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 373) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 374) 	unsigned i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 375) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 376) 	threads->nr = (int) event->nr;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 377) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 378) 	for (i = 0; i < event->nr; i++) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 379) 		perf_thread_map__set_pid(threads, i, (pid_t) event->entries[i].pid);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 380) 		threads->map[i].comm = strndup(event->entries[i].comm, 16);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 381) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 382) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 383) 	refcount_set(&threads->refcnt, 1);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 384) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 385) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 386) struct perf_thread_map *thread_map__new_event(struct perf_record_thread_map *event)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 387) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 388) 	struct perf_thread_map *threads;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 389) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 390) 	threads = thread_map__alloc(event->nr);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 391) 	if (threads)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 392) 		thread_map__copy_event(threads, event);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 393) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 394) 	return threads;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 395) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 396) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 397) bool thread_map__has(struct perf_thread_map *threads, pid_t pid)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 398) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 399) 	int i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 400) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 401) 	for (i = 0; i < threads->nr; ++i) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 402) 		if (threads->map[i].pid == pid)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 403) 			return true;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 404) 	}
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 405) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 406) 	return false;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 407) }
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 408) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 409) int thread_map__remove(struct perf_thread_map *threads, int idx)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 410) {
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 411) 	int i;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 412) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 413) 	if (threads->nr < 1)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 414) 		return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 415) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 416) 	if (idx >= threads->nr)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 417) 		return -EINVAL;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 418) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 419) 	/*
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 420) 	 * Free the 'idx' item and shift the rest up.
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 421) 	 */
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 422) 	zfree(&threads->map[idx].comm);
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 423) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 424) 	for (i = idx; i < threads->nr - 1; i++)
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 425) 		threads->map[i] = threads->map[i + 1];
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 426) 
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 427) 	threads->nr--;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 428) 	return 0;
^8f3ce5b39 (kx 2023-10-28 12:00:06 +0300 429) }