Commit | Line | Data |
---|---|---|
b2441318 | 1 | // SPDX-License-Identifier: GPL-2.0 |
38af91f0 | 2 | #include <stdlib.h> |
134aa44f JO |
3 | #include <sys/types.h> |
4 | #include <unistd.h> | |
8fbc38aa | 5 | #include <sys/prctl.h> |
134aa44f JO |
6 | #include "tests.h" |
7 | #include "thread_map.h" | |
8 | #include "debug.h" | |
9 | ||
8fbc38aa JO |
10 | #define NAME (const char *) "perf" |
11 | #define NAMEUL (unsigned long) NAME | |
12 | ||
81f17c90 | 13 | int test__thread_map(struct test *test __maybe_unused, int subtest __maybe_unused) |
134aa44f JO |
14 | { |
15 | struct thread_map *map; | |
16 | ||
8fbc38aa JO |
17 | TEST_ASSERT_VAL("failed to set process name", |
18 | !prctl(PR_SET_NAME, NAMEUL, 0, 0, 0)); | |
19 | ||
134aa44f JO |
20 | /* test map on current pid */ |
21 | map = thread_map__new_by_pid(getpid()); | |
22 | TEST_ASSERT_VAL("failed to alloc map", map); | |
23 | ||
24 | thread_map__read_comms(map); | |
25 | ||
26 | TEST_ASSERT_VAL("wrong nr", map->nr == 1); | |
27 | TEST_ASSERT_VAL("wrong pid", | |
28 | thread_map__pid(map, 0) == getpid()); | |
29 | TEST_ASSERT_VAL("wrong comm", | |
30 | thread_map__comm(map, 0) && | |
8fbc38aa | 31 | !strcmp(thread_map__comm(map, 0), NAME)); |
35318d20 | 32 | TEST_ASSERT_VAL("wrong refcnt", |
364fed35 | 33 | refcount_read(&map->refcnt) == 1); |
134aa44f JO |
34 | thread_map__put(map); |
35 | ||
36 | /* test dummy pid */ | |
37 | map = thread_map__new_dummy(); | |
38 | TEST_ASSERT_VAL("failed to alloc map", map); | |
39 | ||
40 | thread_map__read_comms(map); | |
41 | ||
42 | TEST_ASSERT_VAL("wrong nr", map->nr == 1); | |
43 | TEST_ASSERT_VAL("wrong pid", thread_map__pid(map, 0) == -1); | |
44 | TEST_ASSERT_VAL("wrong comm", | |
45 | thread_map__comm(map, 0) && | |
46 | !strcmp(thread_map__comm(map, 0), "dummy")); | |
35318d20 | 47 | TEST_ASSERT_VAL("wrong refcnt", |
364fed35 | 48 | refcount_read(&map->refcnt) == 1); |
134aa44f JO |
49 | thread_map__put(map); |
50 | return 0; | |
51 | } | |
99471c96 JO |
52 | |
53 | static int process_event(struct perf_tool *tool __maybe_unused, | |
54 | union perf_event *event, | |
55 | struct perf_sample *sample __maybe_unused, | |
56 | struct machine *machine __maybe_unused) | |
57 | { | |
58 | struct thread_map_event *map = &event->thread_map; | |
59660942 | 59 | struct thread_map *threads; |
99471c96 JO |
60 | |
61 | TEST_ASSERT_VAL("wrong nr", map->nr == 1); | |
62 | TEST_ASSERT_VAL("wrong pid", map->entries[0].pid == (u64) getpid()); | |
8fbc38aa | 63 | TEST_ASSERT_VAL("wrong comm", !strcmp(map->entries[0].comm, NAME)); |
59660942 JO |
64 | |
65 | threads = thread_map__new_event(&event->thread_map); | |
66 | TEST_ASSERT_VAL("failed to alloc map", threads); | |
67 | ||
68 | TEST_ASSERT_VAL("wrong nr", threads->nr == 1); | |
69 | TEST_ASSERT_VAL("wrong pid", | |
70 | thread_map__pid(threads, 0) == getpid()); | |
71 | TEST_ASSERT_VAL("wrong comm", | |
72 | thread_map__comm(threads, 0) && | |
8fbc38aa | 73 | !strcmp(thread_map__comm(threads, 0), NAME)); |
59660942 | 74 | TEST_ASSERT_VAL("wrong refcnt", |
364fed35 | 75 | refcount_read(&threads->refcnt) == 1); |
59660942 | 76 | thread_map__put(threads); |
99471c96 JO |
77 | return 0; |
78 | } | |
79 | ||
81f17c90 | 80 | int test__thread_map_synthesize(struct test *test __maybe_unused, int subtest __maybe_unused) |
99471c96 JO |
81 | { |
82 | struct thread_map *threads; | |
83 | ||
8fbc38aa JO |
84 | TEST_ASSERT_VAL("failed to set process name", |
85 | !prctl(PR_SET_NAME, NAMEUL, 0, 0, 0)); | |
86 | ||
99471c96 JO |
87 | /* test map on current pid */ |
88 | threads = thread_map__new_by_pid(getpid()); | |
89 | TEST_ASSERT_VAL("failed to alloc map", threads); | |
90 | ||
91 | thread_map__read_comms(threads); | |
92 | ||
93 | TEST_ASSERT_VAL("failed to synthesize map", | |
94 | !perf_event__synthesize_thread_map2(NULL, threads, process_event, NULL)); | |
95 | ||
96 | return 0; | |
97 | } | |
38af91f0 | 98 | |
81f17c90 | 99 | int test__thread_map_remove(struct test *test __maybe_unused, int subtest __maybe_unused) |
38af91f0 JO |
100 | { |
101 | struct thread_map *threads; | |
102 | char *str; | |
103 | int i; | |
104 | ||
105 | TEST_ASSERT_VAL("failed to allocate map string", | |
106 | asprintf(&str, "%d,%d", getpid(), getppid()) >= 0); | |
107 | ||
108 | threads = thread_map__new_str(str, NULL, 0); | |
109 | ||
110 | TEST_ASSERT_VAL("failed to allocate thread_map", | |
111 | threads); | |
112 | ||
bb963e16 | 113 | if (verbose > 0) |
38af91f0 JO |
114 | thread_map__fprintf(threads, stderr); |
115 | ||
116 | TEST_ASSERT_VAL("failed to remove thread", | |
117 | !thread_map__remove(threads, 0)); | |
118 | ||
119 | TEST_ASSERT_VAL("thread_map count != 1", threads->nr == 1); | |
120 | ||
bb963e16 | 121 | if (verbose > 0) |
38af91f0 JO |
122 | thread_map__fprintf(threads, stderr); |
123 | ||
124 | TEST_ASSERT_VAL("failed to remove thread", | |
125 | !thread_map__remove(threads, 0)); | |
126 | ||
127 | TEST_ASSERT_VAL("thread_map count != 0", threads->nr == 0); | |
128 | ||
bb963e16 | 129 | if (verbose > 0) |
38af91f0 JO |
130 | thread_map__fprintf(threads, stderr); |
131 | ||
132 | TEST_ASSERT_VAL("failed to not remove thread", | |
133 | thread_map__remove(threads, 0)); | |
134 | ||
135 | for (i = 0; i < threads->nr; i++) | |
136 | free(threads->map[i].comm); | |
137 | ||
138 | free(threads); | |
139 | return 0; | |
140 | } |