1*4882a593Smuzhiyun #include <stdlib.h>
2*4882a593Smuzhiyun #include <string.h>
3*4882a593Smuzhiyun #include <stdio.h>
4*4882a593Smuzhiyun #include <signal.h>
5*4882a593Smuzhiyun #include <unistd.h>
6*4882a593Smuzhiyun #include <sys/mman.h>
7*4882a593Smuzhiyun
8*4882a593Smuzhiyun #include "utils.h"
9*4882a593Smuzhiyun
10*4882a593Smuzhiyun extern char __start___ex_table[];
11*4882a593Smuzhiyun extern char __stop___ex_table[];
12*4882a593Smuzhiyun
13*4882a593Smuzhiyun #if defined(__powerpc64__)
14*4882a593Smuzhiyun #define UCONTEXT_NIA(UC) (UC)->uc_mcontext.gp_regs[PT_NIP]
15*4882a593Smuzhiyun #elif defined(__powerpc__)
16*4882a593Smuzhiyun #define UCONTEXT_NIA(UC) (UC)->uc_mcontext.uc_regs->gregs[PT_NIP]
17*4882a593Smuzhiyun #else
18*4882a593Smuzhiyun #error implement UCONTEXT_NIA
19*4882a593Smuzhiyun #endif
20*4882a593Smuzhiyun
segv_handler(int signr,siginfo_t * info,void * ptr)21*4882a593Smuzhiyun static void segv_handler(int signr, siginfo_t *info, void *ptr)
22*4882a593Smuzhiyun {
23*4882a593Smuzhiyun ucontext_t *uc = (ucontext_t *)ptr;
24*4882a593Smuzhiyun unsigned long addr = (unsigned long)info->si_addr;
25*4882a593Smuzhiyun unsigned long *ip = &UCONTEXT_NIA(uc);
26*4882a593Smuzhiyun unsigned long *ex_p = (unsigned long *)__start___ex_table;
27*4882a593Smuzhiyun
28*4882a593Smuzhiyun while (ex_p < (unsigned long *)__stop___ex_table) {
29*4882a593Smuzhiyun unsigned long insn, fixup;
30*4882a593Smuzhiyun
31*4882a593Smuzhiyun insn = *ex_p++;
32*4882a593Smuzhiyun fixup = *ex_p++;
33*4882a593Smuzhiyun
34*4882a593Smuzhiyun if (insn == *ip) {
35*4882a593Smuzhiyun *ip = fixup;
36*4882a593Smuzhiyun return;
37*4882a593Smuzhiyun }
38*4882a593Smuzhiyun }
39*4882a593Smuzhiyun
40*4882a593Smuzhiyun printf("No exception table match for NIA %lx ADDR %lx\n", *ip, addr);
41*4882a593Smuzhiyun abort();
42*4882a593Smuzhiyun }
43*4882a593Smuzhiyun
setup_segv_handler(void)44*4882a593Smuzhiyun static void setup_segv_handler(void)
45*4882a593Smuzhiyun {
46*4882a593Smuzhiyun struct sigaction action;
47*4882a593Smuzhiyun
48*4882a593Smuzhiyun memset(&action, 0, sizeof(action));
49*4882a593Smuzhiyun action.sa_sigaction = segv_handler;
50*4882a593Smuzhiyun action.sa_flags = SA_SIGINFO;
51*4882a593Smuzhiyun sigaction(SIGSEGV, &action, NULL);
52*4882a593Smuzhiyun }
53*4882a593Smuzhiyun
54*4882a593Smuzhiyun unsigned long COPY_LOOP(void *to, const void *from, unsigned long size);
55*4882a593Smuzhiyun unsigned long test_copy_tofrom_user_reference(void *to, const void *from, unsigned long size);
56*4882a593Smuzhiyun
57*4882a593Smuzhiyun static int total_passed;
58*4882a593Smuzhiyun static int total_failed;
59*4882a593Smuzhiyun
do_one_test(char * dstp,char * srcp,unsigned long len)60*4882a593Smuzhiyun static void do_one_test(char *dstp, char *srcp, unsigned long len)
61*4882a593Smuzhiyun {
62*4882a593Smuzhiyun unsigned long got, expected;
63*4882a593Smuzhiyun
64*4882a593Smuzhiyun got = COPY_LOOP(dstp, srcp, len);
65*4882a593Smuzhiyun expected = test_copy_tofrom_user_reference(dstp, srcp, len);
66*4882a593Smuzhiyun
67*4882a593Smuzhiyun if (got != expected) {
68*4882a593Smuzhiyun total_failed++;
69*4882a593Smuzhiyun printf("FAIL from=%p to=%p len=%ld returned %ld, expected %ld\n",
70*4882a593Smuzhiyun srcp, dstp, len, got, expected);
71*4882a593Smuzhiyun //abort();
72*4882a593Smuzhiyun } else
73*4882a593Smuzhiyun total_passed++;
74*4882a593Smuzhiyun }
75*4882a593Smuzhiyun
76*4882a593Smuzhiyun //#define MAX_LEN 512
77*4882a593Smuzhiyun #define MAX_LEN 16
78*4882a593Smuzhiyun
test_copy_exception(void)79*4882a593Smuzhiyun int test_copy_exception(void)
80*4882a593Smuzhiyun {
81*4882a593Smuzhiyun int page_size;
82*4882a593Smuzhiyun static char *p, *q;
83*4882a593Smuzhiyun unsigned long src, dst, len;
84*4882a593Smuzhiyun
85*4882a593Smuzhiyun page_size = getpagesize();
86*4882a593Smuzhiyun p = mmap(NULL, page_size * 2, PROT_READ|PROT_WRITE,
87*4882a593Smuzhiyun MAP_PRIVATE|MAP_ANONYMOUS, -1, 0);
88*4882a593Smuzhiyun
89*4882a593Smuzhiyun if (p == MAP_FAILED) {
90*4882a593Smuzhiyun perror("mmap");
91*4882a593Smuzhiyun exit(1);
92*4882a593Smuzhiyun }
93*4882a593Smuzhiyun
94*4882a593Smuzhiyun memset(p, 0, page_size);
95*4882a593Smuzhiyun
96*4882a593Smuzhiyun setup_segv_handler();
97*4882a593Smuzhiyun
98*4882a593Smuzhiyun if (mprotect(p + page_size, page_size, PROT_NONE)) {
99*4882a593Smuzhiyun perror("mprotect");
100*4882a593Smuzhiyun exit(1);
101*4882a593Smuzhiyun }
102*4882a593Smuzhiyun
103*4882a593Smuzhiyun q = p + page_size - MAX_LEN;
104*4882a593Smuzhiyun
105*4882a593Smuzhiyun for (src = 0; src < MAX_LEN; src++) {
106*4882a593Smuzhiyun for (dst = 0; dst < MAX_LEN; dst++) {
107*4882a593Smuzhiyun for (len = 0; len < MAX_LEN+1; len++) {
108*4882a593Smuzhiyun // printf("from=%p to=%p len=%ld\n", q+dst, q+src, len);
109*4882a593Smuzhiyun do_one_test(q+dst, q+src, len);
110*4882a593Smuzhiyun }
111*4882a593Smuzhiyun }
112*4882a593Smuzhiyun }
113*4882a593Smuzhiyun
114*4882a593Smuzhiyun printf("Totals:\n");
115*4882a593Smuzhiyun printf(" Pass: %d\n", total_passed);
116*4882a593Smuzhiyun printf(" Fail: %d\n", total_failed);
117*4882a593Smuzhiyun
118*4882a593Smuzhiyun return 0;
119*4882a593Smuzhiyun }
120*4882a593Smuzhiyun
main(void)121*4882a593Smuzhiyun int main(void)
122*4882a593Smuzhiyun {
123*4882a593Smuzhiyun return test_harness(test_copy_exception, str(COPY_LOOP));
124*4882a593Smuzhiyun }
125