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 idx, pid_t pid)
 36{
 37	map->map[idx].pid = pid;
 38}
 39
 40char *perf_thread_map__comm(struct perf_thread_map *map, int idx)
 41{
 42	return map->map[idx].comm;
 43}
 44
 45struct perf_thread_map *perf_thread_map__new_array(int nr_threads, pid_t *array)
 46{
 47	struct perf_thread_map *threads = thread_map__alloc(nr_threads);
 48	int i;
 49
 50	if (!threads)
 51		return NULL;
 52
 53	for (i = 0; i < nr_threads; i++)
 54		perf_thread_map__set_pid(threads, i, array ? array[i] : -1);
 55
 56	threads->nr = nr_threads;
 57	refcount_set(&threads->refcnt, 1);
 58
 
 
 
 
 
 59	return threads;
 60}
 61
 62struct perf_thread_map *perf_thread_map__new_dummy(void)
 63{
 64	return perf_thread_map__new_array(1, NULL);
 65}
 66
 67static void perf_thread_map__delete(struct perf_thread_map *threads)
 68{
 69	if (threads) {
 70		int i;
 71
 72		WARN_ONCE(refcount_read(&threads->refcnt) != 0,
 73			  "thread map refcnt unbalanced\n");
 74		for (i = 0; i < threads->nr; i++)
 75			free(perf_thread_map__comm(threads, i));
 76		free(threads);
 77	}
 78}
 79
 80struct perf_thread_map *perf_thread_map__get(struct perf_thread_map *map)
 81{
 82	if (map)
 83		refcount_inc(&map->refcnt);
 84	return map;
 85}
 86
 87void perf_thread_map__put(struct perf_thread_map *map)
 88{
 89	if (map && refcount_dec_and_test(&map->refcnt))
 90		perf_thread_map__delete(map);
 91}
 92
 93int perf_thread_map__nr(struct perf_thread_map *threads)
 94{
 95	return threads ? threads->nr : 1;
 96}
 97
 98pid_t perf_thread_map__pid(struct perf_thread_map *map, int idx)
 99{
100	return map->map[idx].pid;
101}

 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}