1// SPDX-License-Identifier: GPL-2.0
 2#include <perf/threadmap.h>
 3#include <stdlib.h>
 4#include <linux/refcount.h>
 5#include <internal/threadmap.h>
 6#include <string.h>
 7#include <asm/bug.h>
 8#include <stdio.h>
 9
10static void perf_thread_map__reset(struct perf_thread_map *map, int start, int nr)
11{
12	size_t size = (nr - start) * sizeof(map->map[0]);
13
14	memset(&map->map[start], 0, size);
15	map->err_thread = -1;
16}
17
18struct perf_thread_map *perf_thread_map__realloc(struct perf_thread_map *map, int nr)
19{
20	size_t size = sizeof(*map) + sizeof(map->map[0]) * nr;
21	int start = map ? map->nr : 0;
22
23	map = realloc(map, size);
24	/*
25	 * We only realloc to add more items, let's reset new items.
26	 */
27	if (map)
28		perf_thread_map__reset(map, start, nr);
29
30	return map;
31}
32
33#define thread_map__alloc(__nr) perf_thread_map__realloc(NULL, __nr)
34
35void perf_thread_map__set_pid(struct perf_thread_map *map, int thread, pid_t pid)
36{
37	map->map[thread].pid = pid;
38}
39
40char *perf_thread_map__comm(struct perf_thread_map *map, int thread)
41{
42	return map->map[thread].comm;
43}
44
45struct perf_thread_map *perf_thread_map__new_dummy(void)
46{
47	struct perf_thread_map *threads = thread_map__alloc(1);
48
49	if (threads != NULL) {
50		perf_thread_map__set_pid(threads, 0, -1);
51		threads->nr = 1;
52		refcount_set(&threads->refcnt, 1);
53	}
54	return threads;
55}
56
57static void perf_thread_map__delete(struct perf_thread_map *threads)
58{
59	if (threads) {
60		int i;
61
62		WARN_ONCE(refcount_read(&threads->refcnt) != 0,
63			  "thread map refcnt unbalanced\n");
64		for (i = 0; i < threads->nr; i++)
65			free(perf_thread_map__comm(threads, i));
66		free(threads);
67	}
68}
69
70struct perf_thread_map *perf_thread_map__get(struct perf_thread_map *map)
71{
72	if (map)
73		refcount_inc(&map->refcnt);
74	return map;
75}
76
77void perf_thread_map__put(struct perf_thread_map *map)
78{
79	if (map && refcount_dec_and_test(&map->refcnt))
80		perf_thread_map__delete(map);
81}
82
83int perf_thread_map__nr(struct perf_thread_map *threads)
84{
85	return threads ? threads->nr : 1;
86}
87
88pid_t perf_thread_map__pid(struct perf_thread_map *map, int thread)
89{
90	return map->map[thread].pid;
91}