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 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}