init

hc

2024-03-22 a0752693d998599af469473b8dc239ef973a012f

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