// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright 2015, Michael Neuling, IBM Corp.
 * Original: Michael Neuling 19/7/2013
 * Edited: Rashmica Gupta 01/12/2015
 *
 * Do some transactions, see if the tar is corrupted.
 * If the transaction is aborted, the TAR should be rolled back to the
 * checkpointed value before the transaction began. The value written to
 * TAR in suspended mode should only remain in TAR if the transaction
 * completes.
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>

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

int     num_loops       = 10000;

int test_tar(void)
{
        int i;

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

        for (i = 0; i < num_loops; i++)
        {
                uint64_t result = 0;
                asm __volatile__(
                        "li     7, 1;"
                        "mtspr  %[tar], 7;"     /* tar = 1 */
                        "tbegin.;"
                        "beq    3f;"
                        "li     4, 0x7000;"     /* Loop lots, to use time */
                        "2:;"                   /* Start loop */
                        "li     7, 2;"
                        "mtspr  %[tar], 7;"     /* tar = 2 */
                        "tsuspend.;"
                        "li     7, 3;"
                        "mtspr  %[tar], 7;"     /* tar = 3 */
                        "tresume.;"
                        "subi   4, 4, 1;"
                        "cmpdi  4, 0;"
                        "bne    2b;"
                        "tend.;"

                        /* Transaction sucess! TAR should be 3 */
                        "mfspr  7, %[tar];"
                        "ori    %[res], 7, 4;"  // res = 3|4 = 7
                        "b      4f;"

                        /* Abort handler. TAR should be rolled back to 1 */
                        "3:;"
                        "mfspr  7, %[tar];"
                        "ori    %[res], 7, 8;"  // res = 1|8 = 9
                        "4:;"

                        : [res]"=r"(result)
                        : [tar]"i"(SPRN_TAR)
                           : "memory", "r0", "r4", "r7");

                /* If result is anything else other than 7 or 9, the tar
                 * value must have been corrupted. */
                if ((result != 7) && (result != 9))
                        return 1;
        }
        return 0;
}

int main(int argc, char *argv[])
{
        /* A low number of iterations (eg 100) can cause a false pass */
        if (argc > 1) {
                if (strcmp(argv[1], "-h") == 0) {
                        printf("Syntax:\n\t%s [<num loops>]\n",
                               argv[0]);
                        return 1;
                } else {
                        num_loops = atoi(argv[1]);
                }
        }

        printf("Starting, %d loops\n", num_loops);

        return test_harness(test_tar, "tm_tar");
}