1 // SPDX-License-Identifier: GPL-2.0
2 #include <stdarg.h>
3 #include <stdio.h>
4 #include <perf/threadmap.h>
5 #include <internal/tests.h>
6 #include "tests.h"
7 
8 static int libperf_print(enum libperf_print_level level,
9 			 const char *fmt, va_list ap)
10 {
11 	return vfprintf(stderr, fmt, ap);
12 }
13 
14 static int test_threadmap_array(int nr, pid_t *array)
15 {
16 	struct perf_thread_map *threads;
17 	int i;
18 
19 	threads = perf_thread_map__new_array(nr, array);
20 	__T("Failed to allocate new thread map", threads);
21 
22 	__T("Unexpected number of threads", perf_thread_map__nr(threads) == nr);
23 
24 	for (i = 0; i < nr; i++) {
25 		__T("Unexpected initial value of thread",
26 		    perf_thread_map__pid(threads, i) == (array ? array[i] : -1));
27 	}
28 
29 	for (i = 1; i < nr; i++)
30 		perf_thread_map__set_pid(threads, i, i * 100);
31 
32 	__T("Unexpected value of thread 0",
33 	    perf_thread_map__pid(threads, 0) == (array ? array[0] : -1));
34 
35 	for (i = 1; i < nr; i++) {
36 		__T("Unexpected thread value",
37 		    perf_thread_map__pid(threads, i) == i * 100);
38 	}
39 
40 	perf_thread_map__put(threads);
41 
42 	return 0;
43 }
44 
45 #define THREADS_NR	10
46 int test_threadmap(int argc, char **argv)
47 {
48 	struct perf_thread_map *threads;
49 	pid_t thr_array[THREADS_NR];
50 	int i;
51 
52 	__T_START;
53 
54 	libperf_init(libperf_print);
55 
56 	threads = perf_thread_map__new_dummy();
57 	if (!threads)
58 		return -1;
59 
60 	perf_thread_map__get(threads);
61 	perf_thread_map__put(threads);
62 	perf_thread_map__put(threads);
63 
64 	test_threadmap_array(THREADS_NR, NULL);
65 
66 	for (i = 0; i < THREADS_NR; i++)
67 		thr_array[i] = i + 100;
68 
69 	test_threadmap_array(THREADS_NR, thr_array);
70 
71 	__T_END;
72 	return tests_failed == 0 ? 0 : -1;
73 }
74