/* SPDX-License-Identifier: GPL-2.0 */
/* Copyright (C) 2020 ARM Limited */

#include "mte_def.h"

.arch   armv8.5-a+memtag

#define ENTRY(name) \
        .globl name ;\
        .p2align 2;\
        .type name, @function ;\
name:

#define ENDPROC(name) \
        .size name, .-name ;

        .text
/*
 * mte_insert_random_tag: Insert random tag and might be same as the source tag if
 *                        the source pointer has it.
 * Input:
 *              x0 - source pointer with a tag/no-tag
 * Return:
 *              x0 - pointer with random tag
 */
ENTRY(mte_insert_random_tag)
        irg     x0, x0, xzr
        ret
ENDPROC(mte_insert_random_tag)

/*
 * mte_insert_new_tag: Insert new tag and different from the source tag if
 *                     source pointer has it.
 * Input:
 *              x0 - source pointer with a tag/no-tag
 * Return:
 *              x0 - pointer with random tag
 */
ENTRY(mte_insert_new_tag)
        gmi     x1, x0, xzr
        irg     x0, x0, x1
        ret
ENDPROC(mte_insert_new_tag)

/*
 * mte_get_tag_address: Get the tag from given address.
 * Input:
 *              x0 - source pointer
 * Return:
 *              x0 - pointer with appended tag
 */
ENTRY(mte_get_tag_address)
        ldg     x0, [x0]
        ret
ENDPROC(mte_get_tag_address)

/*
 * mte_set_tag_address_range: Set the tag range from the given address
 * Input:
 *              x0 - source pointer with tag data
 *              x1 - range
 * Return:
 *              none
 */
ENTRY(mte_set_tag_address_range)
        cbz     x1, 2f
1:
        stg     x0, [x0, #0x0]
        add     x0, x0, #MT_GRANULE_SIZE
        sub     x1, x1, #MT_GRANULE_SIZE
        cbnz    x1, 1b
2:
        ret
ENDPROC(mte_set_tag_address_range)

/*
 * mt_clear_tag_address_range: Clear the tag range from the given address
 * Input:
 *              x0 - source pointer with tag data
 *              x1 - range
 * Return:
 *              none
 */
ENTRY(mte_clear_tag_address_range)
        cbz     x1, 2f
1:
        stzg    x0, [x0, #0x0]
        add     x0, x0, #MT_GRANULE_SIZE
        sub     x1, x1, #MT_GRANULE_SIZE
        cbnz    x1, 1b
2:
        ret
ENDPROC(mte_clear_tag_address_range)

/*
 * mte_enable_pstate_tco: Enable PSTATE.TCO (tag check override) field
 * Input:
 *              none
 * Return:
 *              none
 */
ENTRY(mte_enable_pstate_tco)
        msr     tco, #MT_PSTATE_TCO_EN
        ret
ENDPROC(mte_enable_pstate_tco)

/*
 * mte_disable_pstate_tco: Disable PSTATE.TCO (tag check override) field
 * Input:
 *              none
 * Return:
 *              none
 */
ENTRY(mte_disable_pstate_tco)
        msr     tco, #MT_PSTATE_TCO_DIS
        ret
ENDPROC(mte_disable_pstate_tco)

/*
 * mte_get_pstate_tco: Get PSTATE.TCO (tag check override) field
 * Input:
 *              none
 * Return:
 *              x0
 */
ENTRY(mte_get_pstate_tco)
        mrs     x0, tco
        ubfx    x0, x0, #MT_PSTATE_TCO_SHIFT, #1
        ret
ENDPROC(mte_get_pstate_tco)