hc
2024-05-14 bedbef8ad3e75a304af6361af235302bcc61d06b
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
// SPDX-License-Identifier: GPL-2.0-only
/*
 * POWER Data Stream Control Register (DSCR) default test
 *
 * This test modifies the system wide default DSCR through
 * it's sysfs interface and then verifies that all threads
 * see the correct changed DSCR value immediately.
 *
 * Copyright 2012, Anton Blanchard, IBM Corporation.
 * Copyright 2015, Anshuman Khandual, IBM Corporation.
 */
#include "dscr.h"
 
static unsigned long dscr;        /* System DSCR default */
static unsigned long sequence;
static unsigned long result[THREADS];
 
static void *do_test(void *in)
{
   unsigned long thread = (unsigned long)in;
   unsigned long i;
 
   for (i = 0; i < COUNT; i++) {
       unsigned long d, cur_dscr, cur_dscr_usr;
       unsigned long s1, s2;
 
       s1 = READ_ONCE(sequence);
       if (s1 & 1)
           continue;
       rmb();
 
       d = dscr;
       cur_dscr = get_dscr();
       cur_dscr_usr = get_dscr_usr();
 
       rmb();
       s2 = sequence;
 
       if (s1 != s2)
           continue;
 
       if (cur_dscr != d) {
           fprintf(stderr, "thread %ld kernel DSCR should be %ld "
               "but is %ld\n", thread, d, cur_dscr);
           result[thread] = 1;
           pthread_exit(&result[thread]);
       }
 
       if (cur_dscr_usr != d) {
           fprintf(stderr, "thread %ld user DSCR should be %ld "
               "but is %ld\n", thread, d, cur_dscr_usr);
           result[thread] = 1;
           pthread_exit(&result[thread]);
       }
   }
   result[thread] = 0;
   pthread_exit(&result[thread]);
}
 
int dscr_default(void)
{
   pthread_t threads[THREADS];
   unsigned long i, *status[THREADS];
   unsigned long orig_dscr_default;
 
   SKIP_IF(!have_hwcap2(PPC_FEATURE2_DSCR));
 
   orig_dscr_default = get_default_dscr();
 
   /* Initial DSCR default */
   dscr = 1;
   set_default_dscr(dscr);
 
   /* Spawn all testing threads */
   for (i = 0; i < THREADS; i++) {
       if (pthread_create(&threads[i], NULL, do_test, (void *)i)) {
           perror("pthread_create() failed");
           goto fail;
       }
   }
 
   srand(getpid());
 
   /* Keep changing the DSCR default */
   for (i = 0; i < COUNT; i++) {
       double ret = uniform_deviate(rand());
 
       if (ret < 0.0001) {
           sequence++;
           wmb();
 
           dscr++;
           if (dscr > DSCR_MAX)
               dscr = 0;
 
           set_default_dscr(dscr);
 
           wmb();
           sequence++;
       }
   }
 
   /* Individual testing thread exit status */
   for (i = 0; i < THREADS; i++) {
       if (pthread_join(threads[i], (void **)&(status[i]))) {
           perror("pthread_join() failed");
           goto fail;
       }
 
       if (*status[i]) {
           printf("%ldth thread failed to join with %ld status\n",
                               i, *status[i]);
           goto fail;
       }
   }
   set_default_dscr(orig_dscr_default);
   return 0;
fail:
   set_default_dscr(orig_dscr_default);
   return 1;
}
 
int main(int argc, char *argv[])
{
   return test_harness(dscr_default, "dscr_default_test");
}