// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright 2015, Michael Neuling, IBM Corp.
 *
 * Original: Michael Neuling 4/12/2013
 * Edited: Rashmica Gupta 4/12/2015
 *
 * See if the altivec state is leaked out of an aborted transaction due to
 * kernel vmx copy loops.
 *
 * When the transaction aborts, VSR values should rollback to the values
 * they held before the transaction commenced. Using VSRs while transaction
 * is suspended should not affect the checkpointed values.
 *
 * (1) write A to a VSR
 * (2) start transaction
 * (3) suspend transaction
 * (4) change the VSR to B
 * (5) trigger kernel vmx copy loop
 * (6) abort transaction
 * (7) check that the VSR value is A
 */

#include <inttypes.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/mman.h>
#include <string.h>
#include <assert.h>

#include "tm.h"
#include "utils.h"

int test_vmxcopy()
{
        long double vecin = 1.3;
        long double vecout;
        unsigned long pgsize = getpagesize();
        int i;
        int fd;
        int size = pgsize*16;
        char tmpfile[] = "/tmp/page_faultXXXXXX";
        char buf[pgsize];
        char *a;
        uint64_t aborted = 0;

        SKIP_IF(!have_htm());
        SKIP_IF(htm_is_synthetic());
        SKIP_IF(!is_ppc64le());

        fd = mkstemp(tmpfile);
        assert(fd >= 0);

        memset(buf, 0, pgsize);
        for (i = 0; i < size; i += pgsize)
                assert(write(fd, buf, pgsize) == pgsize);

        unlink(tmpfile);

        a = mmap(NULL, size, PROT_READ|PROT_WRITE, MAP_PRIVATE, fd, 0);
        assert(a != MAP_FAILED);

        asm __volatile__(
                "lxvd2x 40,0,%[vecinptr];"      /* set 40 to initial value*/
                "tbegin.;"
                "beq    3f;"
                "tsuspend.;"
                "xxlxor 40,40,40;"              /* set 40 to 0 */
                "std    5, 0(%[map]);"          /* cause kernel vmx copy page */
                "tabort. 0;"
                "tresume.;"
                "tend.;"
                "li     %[res], 0;"
                "b      5f;"

                /* Abort handler */
                "3:;"
                "li     %[res], 1;"

                "5:;"
                "stxvd2x 40,0,%[vecoutptr];"
                : [res]"=&r"(aborted)
                : [vecinptr]"r"(&vecin),
                  [vecoutptr]"r"(&vecout),
                  [map]"r"(a)
                : "memory", "r0", "r3", "r4", "r5", "r6", "r7");

        if (aborted && (vecin != vecout)){
                printf("FAILED: vector state leaked on abort %f != %f\n",
                       (double)vecin, (double)vecout);
                return 1;
        }

        munmap(a, size);

        close(fd);

        return 0;
}

int main(void)
{
        return test_harness(test_vmxcopy, "tm_vmxcopy");
}