// SPDX-License-Identifier: GPL-2.0-only
// Copyright (C) 2021 ARM Limited.

#include "sme-inst.h"

.arch_extension sve

#define MAGIC     42

#define MAXVL     2048
#define MAXVL_B   (MAXVL / 8)

.pushsection .text
.data
.align 4
scratch:
        .space  MAXVL_B
.popsection

.globl fork_test
fork_test:
        smstart_za

        // For simplicity just set one word in one vector, other tests
        // cover general data corruption issues.
        ldr     x0, =scratch
        mov     x1, #MAGIC
        str     x1, [x0]
        mov     w12, wzr
        _ldr_za 12, 0                   // ZA.H[W12] loaded from [X0]

        // Tail call into the C portion that does the fork & verify
        b       fork_test_c

.globl verify_fork
verify_fork:
        // SVCR should have ZA=1, SM=0
        mrs     x0, S3_3_C4_C2_2
        and     x1, x0, #3
        cmp     x1, #2
        beq     1f
        mov     x0, xzr
        b       100f
1:

        // ZA should still have the value we loaded
        ldr     x0, =scratch
        mov     w12, wzr
        _str_za 12, 0                   // ZA.H[W12] stored to [X0]
        ldr     x1, [x0]
        cmp     x1, #MAGIC
        beq     2f
        mov     x0, xzr
        b       100f

2:
        // All tests passed
        mov     x0, #1
100:
        ret