ptrace-vsx.c

  1// SPDX-License-Identifier: GPL-2.0-or-later
  2/*
  3 * Ptrace test for VMX/VSX registers
  4 *
  5 * Copyright (C) 2015 Anshuman Khandual, IBM Corporation.
  6 */
  7#include "ptrace.h"
  8#include "ptrace-vsx.h"
  9
 10/* Tracer and Tracee Shared Data */
 11int shm_id;
 12int *cptr, *pptr;
 13
 14unsigned long fp_load[VEC_MAX];
 15unsigned long fp_load_new[VEC_MAX];
 16unsigned long fp_store[VEC_MAX];
 17
 18void vsx(void)
 19{
 20	int ret;
 21
 22	cptr = (int *)shmat(shm_id, NULL, 0);
 23	loadvsx(fp_load, 0);
 24	cptr[1] = 1;
 25
 26	while (!cptr[0])
 27		asm volatile("" : : : "memory");
 28	shmdt((void *) cptr);
 29
 30	storevsx(fp_store, 0);
 31	ret = compare_vsx_vmx(fp_store, fp_load_new);
 32	if (ret)
 33		exit(1);
 34	exit(0);
 35}
 36
 37int trace_vsx(pid_t child)
 38{
 39	unsigned long vsx[VSX_MAX];
 40	unsigned long vmx[VMX_MAX + 2][2];
 41
 42	FAIL_IF(start_trace(child));
 43	FAIL_IF(show_vsx(child, vsx));
 44	FAIL_IF(validate_vsx(vsx, fp_load));
 45	FAIL_IF(show_vmx(child, vmx));
 46	FAIL_IF(validate_vmx(vmx, fp_load));
 47
 48	memset(vsx, 0, sizeof(vsx));
 49	memset(vmx, 0, sizeof(vmx));
 50	load_vsx_vmx(fp_load_new, vsx, vmx);
 51
 52	FAIL_IF(write_vsx(child, vsx));
 53	FAIL_IF(write_vmx(child, vmx));
 54	FAIL_IF(stop_trace(child));
 55
 56	return TEST_PASS;
 57}
 58
 59int ptrace_vsx(void)
 60{
 61	pid_t pid;
 62	int ret, status, i;
 63
 64	SKIP_IF_MSG(!have_hwcap(PPC_FEATURE_HAS_VSX), "Don't have VSX");
 65
 66	shm_id = shmget(IPC_PRIVATE, sizeof(int) * 2, 0777|IPC_CREAT);
 67
 68	for (i = 0; i < VEC_MAX; i++)
 69		fp_load[i] = i + rand();
 70
 71	for (i = 0; i < VEC_MAX; i++)
 72		fp_load_new[i] = i + 2 * rand();
 73
 74	pid = fork();
 75	if (pid < 0) {
 76		perror("fork() failed");
 77		return TEST_FAIL;
 78	}
 79
 80	if (pid == 0)
 81		vsx();
 82
 83	if (pid) {
 84		pptr = (int *)shmat(shm_id, NULL, 0);
 85		while (!pptr[1])
 86			asm volatile("" : : : "memory");
 87
 88		ret = trace_vsx(pid);
 89		if (ret) {
 90			kill(pid, SIGTERM);
 91			shmdt((void *)pptr);
 92			shmctl(shm_id, IPC_RMID, NULL);
 93			return TEST_FAIL;
 94		}
 95
 96		pptr[0] = 1;
 97		shmdt((void *)pptr);
 98
 99		ret = wait(&status);
100		shmctl(shm_id, IPC_RMID, NULL);
101		if (ret != pid) {
102			printf("Child's exit status not captured\n");
103			return TEST_FAIL;
104		}
105
106		return (WIFEXITED(status) && WEXITSTATUS(status)) ? TEST_FAIL :
107			TEST_PASS;
108	}
109	return TEST_PASS;
110}
111
112int main(int argc, char *argv[])
113{
114	return test_harness(ptrace_vsx, "ptrace_vsx");
115}