175 lines · c
1// SPDX-License-Identifier: GPL-2.0-or-later2/*3 * Ptrace test for GPR/FPR registers4 *5 * Copyright (C) 2015 Anshuman Khandual, IBM Corporation.6 */7#include "ptrace.h"8#include "ptrace-gpr.h"9#include "reg.h"10#include <time.h>11 12/* Tracer and Tracee Shared Data */13int shm_id;14int *cptr, *pptr;15 16extern void gpr_child_loop(int *read_flag, int *write_flag,17 unsigned long *gpr_buf, double *fpr_buf);18 19unsigned long child_gpr_val, parent_gpr_val;20double child_fpr_val, parent_fpr_val;21 22static int child(void)23{24 unsigned long gpr_buf[32];25 double fpr_buf[32];26 int i;27 28 cptr = (int *)shmat(shm_id, NULL, 0);29 memset(gpr_buf, 0, sizeof(gpr_buf));30 memset(fpr_buf, 0, sizeof(fpr_buf));31 32 for (i = 0; i < 32; i++) {33 gpr_buf[i] = child_gpr_val;34 fpr_buf[i] = child_fpr_val;35 }36 37 gpr_child_loop(&cptr[0], &cptr[1], gpr_buf, fpr_buf);38 39 shmdt((void *)cptr);40 41 FAIL_IF(validate_gpr(gpr_buf, parent_gpr_val));42 FAIL_IF(validate_fpr_double(fpr_buf, parent_fpr_val));43 44 return 0;45}46 47int trace_gpr(pid_t child)48{49 __u64 tmp, fpr[32], *peeked_fprs;50 unsigned long gpr[18];51 52 FAIL_IF(start_trace(child));53 54 // Check child GPRs match what we expect using GETREGS55 FAIL_IF(show_gpr(child, gpr));56 FAIL_IF(validate_gpr(gpr, child_gpr_val));57 58 // Check child FPRs match what we expect using GETFPREGS59 FAIL_IF(show_fpr(child, fpr));60 memcpy(&tmp, &child_fpr_val, sizeof(tmp));61 FAIL_IF(validate_fpr(fpr, tmp));62 63 // Check child FPRs match what we expect using PEEKUSR64 peeked_fprs = peek_fprs(child);65 FAIL_IF(!peeked_fprs);66 FAIL_IF(validate_fpr(peeked_fprs, tmp));67 free(peeked_fprs);68 69 // Write child GPRs using SETREGS70 FAIL_IF(write_gpr(child, parent_gpr_val));71 72 // Write child FPRs using SETFPREGS73 memcpy(&tmp, &parent_fpr_val, sizeof(tmp));74 FAIL_IF(write_fpr(child, tmp));75 76 // Check child FPRs match what we just set, using PEEKUSR77 peeked_fprs = peek_fprs(child);78 FAIL_IF(!peeked_fprs);79 FAIL_IF(validate_fpr(peeked_fprs, tmp));80 81 // Write child FPRs using POKEUSR82 FAIL_IF(poke_fprs(child, (unsigned long *)peeked_fprs));83 84 // Child will check its FPRs match before exiting85 FAIL_IF(stop_trace(child));86 87 return TEST_PASS;88}89 90#ifndef __LONG_WIDTH__91#define __LONG_WIDTH__ (sizeof(long) * 8)92#endif93 94static uint64_t rand_reg(void)95{96 uint64_t result;97 long r;98 99 r = random();100 101 // Small values are typical102 result = r & 0xffff;103 if (r & 0x10000)104 return result;105 106 // Pointers tend to have high bits set107 result |= random() << (__LONG_WIDTH__ - 31);108 if (r & 0x100000)109 return result;110 111 // And sometimes we want a full 64-bit value112 result ^= random() << 16;113 114 return result;115}116 117int ptrace_gpr(void)118{119 unsigned long seed;120 int ret, status;121 pid_t pid;122 123 seed = getpid() ^ time(NULL);124 printf("srand(%lu)\n", seed);125 srand(seed);126 127 child_gpr_val = rand_reg();128 child_fpr_val = rand_reg();129 parent_gpr_val = rand_reg();130 parent_fpr_val = rand_reg();131 132 shm_id = shmget(IPC_PRIVATE, sizeof(int) * 2, 0777|IPC_CREAT);133 pid = fork();134 if (pid < 0) {135 perror("fork() failed");136 return TEST_FAIL;137 }138 if (pid == 0)139 exit(child());140 141 if (pid) {142 pptr = (int *)shmat(shm_id, NULL, 0);143 while (!pptr[1])144 asm volatile("" : : : "memory");145 146 ret = trace_gpr(pid);147 if (ret) {148 kill(pid, SIGTERM);149 shmdt((void *)pptr);150 shmctl(shm_id, IPC_RMID, NULL);151 return TEST_FAIL;152 }153 154 pptr[0] = 1;155 shmdt((void *)pptr);156 157 ret = wait(&status);158 shmctl(shm_id, IPC_RMID, NULL);159 if (ret != pid) {160 printf("Child's exit status not captured\n");161 return TEST_FAIL;162 }163 164 return (WIFEXITED(status) && WEXITSTATUS(status)) ? TEST_FAIL :165 TEST_PASS;166 }167 168 return TEST_PASS;169}170 171int main(int argc, char *argv[])172{173 return test_harness(ptrace_gpr, "ptrace_gpr");174}175