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