/*
 * Copyright (c) 2015, Google Inc. All rights reserved
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files
 * (the "Software"), to deal in the Software without restriction,
 * including without limitation the rights to use, copy, modify, merge,
 * publish, distribute, sublicense, and/or sell copies of the Software,
 * and to permit persons to whom the Software is furnished to do so,
 * subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

#include <asm.h>
#include <arch/asm_macros.h>
#include <err.h>

/**
 * int mmutest_arch_rodata_pnx(void) - Test that rodata section is mapped pnx
 *
 * Returns ERR_FAULT if rodata is not executable.
 * Return 0 if rodata is executable.
 */
.section .rodata
.align 2
 set_fault_handler .Lmmutest_fault
FUNCTION(mmutest_arch_rodata_pnx)
 mov x0, #0

 ret

/**
 * int mmutest_arch_data_pnx(void) - Test that data section is mapped pnx
 *
 * Returns ERR_FAULT if data is not executable.
 * Return 0 if data is executable.
 */
.section .data
.align 2
 set_fault_handler .Lmmutest_fault
FUNCTION(mmutest_arch_data_pnx)
 mov x0, #0

 ret
.section .text

/**
 * int mmutest_arch_rodata_ro(void) - Test that rodata section is mapped read-only
 *
 * Returns ERR_FAULT if rodata is not writable
 * Returns 1 if write to rodata is silently dropped
 * Returns 0 if rodata is writable
 */
FUNCTION(mmutest_arch_rodata_ro)
 adrl x0, .Ltest_rodata_long

 set_fault_handler .Lmmutest_fault
 str wzr, [x0]

 ldr w0, [x0]
 ret

.section .rodata
.Ltest_rodata_long:
 .long 0x1

.section .text

/**
 * int mmutest_arch_store_uint32(uint32_t *ptr, bool user) - Test if ptr is writable
 * @ptr:  Memory location to test
 * @user: Use unprivileged store
 *
 * Returns ERR_FAULT if ptr is not writable
 * Returns ERR_GENERIC if ptr is not readable
 * Returns 2 if write does not fault, but data is lost on readback from memory
 * Returns 1 if write does not fault, but data is lost on readback from cache
 * Returns 0 if ptr is writable
 */
FUNCTION(mmutest_arch_store_uint32)
 cbnz x1, .Lmmutest_arch_store_uint32_user

 set_fault_handler .Lmmutest_setup_fault
 ldr w3, [x0]
 mvn w3, w3

 set_fault_handler .Lmmutest_fault
 str w3, [x0]
 dmb sy
 ldr w2, [x0]

 b .Lmmutest_arch_store_uint32_str_done

.Lmmutest_arch_store_uint32_user:
 set_fault_handler .Lmmutest_setup_fault
 ldtr w3, [x0]
 mvn w3, w3

 set_fault_handler .Lmmutest_fault
 sttr w3, [x0]
 dmb sy
 ldtr w2, [x0]

.Lmmutest_arch_store_uint32_str_done:

 cmp w3, w2
 bne .Lmmutest_arch_store_uint32_cache_read_mismatch

 push x0, x1
 push x3, x30
 mov x1, #4
 bl arch_clean_invalidate_cache_range
 pop x3, x30
 pop x0, x1

 cbnz x1, .Lmmutest_arch_store_uint32_memory_reload_user

 ldr w2, [x0]
 b .Lmmutest_arch_store_uint32_memory_reload_done

.Lmmutest_arch_store_uint32_memory_reload_user:
 ldtr w2, [x0]

.Lmmutest_arch_store_uint32_memory_reload_done:
 cmp w3, w2
 bne .Lmmutest_arch_store_uint32_memory_mismatch

 mov x0, xzr
 ret

.Lmmutest_arch_store_uint32_cache_read_mismatch:
 mov x0, #1
 ret

.Lmmutest_arch_store_uint32_memory_mismatch:
 mov x0, #2
 ret

/**
 * int mmutest_arch_nop(int ret) - Return ret
 *
 * Returns ret if run from executable page.
 * Does not return if run from non-executable page.
 */
FUNCTION(mmutest_arch_nop)
 ret
FUNCTION(mmutest_arch_nop_end)

.Lmmutest_setup_fault:
 mov x0, #ERR_GENERIC
 ret

.Lmmutest_fault:
 mov x0, #ERR_FAULT
 ret