Linux Audio

Check our new training course

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