tests/tcg/aarch64/pauth-1.c


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

#include <assert.h>
#include <sys/prctl.h>
#include <stdio.h>

asm(".arch armv8.4-a");

#ifndef PR_PAC_RESET_KEYS
#define PR_PAC_RESET_KEYS  54
#define PR_PAC_APDAKEY     (1 << 2)
#endif

#define TESTS 1000

int main()
{
    int x, i, count = 0;
    void *p0 = &x, *p1, *p2;
    float perc;

    for (i = 0; i < TESTS; i++) {
        asm volatile("pacdza %0" : "=r"(p1) : "0"(p0));
        prctl(PR_PAC_RESET_KEYS, PR_PAC_APDAKEY, 0, 0, 0);
        asm volatile("pacdza %0" : "=r"(p2) : "0"(p0));

        if (p1 != p0) {
            count++;
        }
        if (p1 != p2) {
            count++;
        }
    }

    perc = (float) count / (float) (TESTS * 2);
    printf("Ptr Check: %0.2f%%", perc * 100.0);
    assert(perc > 0.95);
    return 0;
}