1*4882a593Smuzhiyun // SPDX-License-Identifier: LGPL-2.1
2*4882a593Smuzhiyun /*
3*4882a593Smuzhiyun * Basic test coverage for critical regions and rseq_current_cpu().
4*4882a593Smuzhiyun */
5*4882a593Smuzhiyun
6*4882a593Smuzhiyun #define _GNU_SOURCE
7*4882a593Smuzhiyun #include <assert.h>
8*4882a593Smuzhiyun #include <sched.h>
9*4882a593Smuzhiyun #include <signal.h>
10*4882a593Smuzhiyun #include <stdio.h>
11*4882a593Smuzhiyun #include <string.h>
12*4882a593Smuzhiyun #include <sys/time.h>
13*4882a593Smuzhiyun
14*4882a593Smuzhiyun #include "rseq.h"
15*4882a593Smuzhiyun
test_cpu_pointer(void)16*4882a593Smuzhiyun void test_cpu_pointer(void)
17*4882a593Smuzhiyun {
18*4882a593Smuzhiyun cpu_set_t affinity, test_affinity;
19*4882a593Smuzhiyun int i;
20*4882a593Smuzhiyun
21*4882a593Smuzhiyun sched_getaffinity(0, sizeof(affinity), &affinity);
22*4882a593Smuzhiyun CPU_ZERO(&test_affinity);
23*4882a593Smuzhiyun for (i = 0; i < CPU_SETSIZE; i++) {
24*4882a593Smuzhiyun if (CPU_ISSET(i, &affinity)) {
25*4882a593Smuzhiyun CPU_SET(i, &test_affinity);
26*4882a593Smuzhiyun sched_setaffinity(0, sizeof(test_affinity),
27*4882a593Smuzhiyun &test_affinity);
28*4882a593Smuzhiyun assert(sched_getcpu() == i);
29*4882a593Smuzhiyun assert(rseq_current_cpu() == i);
30*4882a593Smuzhiyun assert(rseq_current_cpu_raw() == i);
31*4882a593Smuzhiyun assert(rseq_cpu_start() == i);
32*4882a593Smuzhiyun CPU_CLR(i, &test_affinity);
33*4882a593Smuzhiyun }
34*4882a593Smuzhiyun }
35*4882a593Smuzhiyun sched_setaffinity(0, sizeof(affinity), &affinity);
36*4882a593Smuzhiyun }
37*4882a593Smuzhiyun
main(int argc,char ** argv)38*4882a593Smuzhiyun int main(int argc, char **argv)
39*4882a593Smuzhiyun {
40*4882a593Smuzhiyun if (rseq_register_current_thread()) {
41*4882a593Smuzhiyun fprintf(stderr, "Error: rseq_register_current_thread(...) failed(%d): %s\n",
42*4882a593Smuzhiyun errno, strerror(errno));
43*4882a593Smuzhiyun goto init_thread_error;
44*4882a593Smuzhiyun }
45*4882a593Smuzhiyun printf("testing current cpu\n");
46*4882a593Smuzhiyun test_cpu_pointer();
47*4882a593Smuzhiyun if (rseq_unregister_current_thread()) {
48*4882a593Smuzhiyun fprintf(stderr, "Error: rseq_unregister_current_thread(...) failed(%d): %s\n",
49*4882a593Smuzhiyun errno, strerror(errno));
50*4882a593Smuzhiyun goto init_thread_error;
51*4882a593Smuzhiyun }
52*4882a593Smuzhiyun return 0;
53*4882a593Smuzhiyun
54*4882a593Smuzhiyun init_thread_error:
55*4882a593Smuzhiyun return -1;
56*4882a593Smuzhiyun }
57