diff options
Diffstat (limited to 'tests/test-progs/asmtest/src/riscv/isa/rv64samt/sysclone_d.S')
| -rw-r--r-- | tests/test-progs/asmtest/src/riscv/isa/rv64samt/sysclone_d.S | 158 |
1 files changed, 158 insertions, 0 deletions
diff --git a/tests/test-progs/asmtest/src/riscv/isa/rv64samt/sysclone_d.S b/tests/test-progs/asmtest/src/riscv/isa/rv64samt/sysclone_d.S new file mode 100644 index 000000000..f178476ce --- /dev/null +++ b/tests/test-progs/asmtest/src/riscv/isa/rv64samt/sysclone_d.S @@ -0,0 +1,158 @@ +/* + * Copyright (c) 2018, Cornell University + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or + * without modification, are permitted provided that the following + * conditions are met: + * + * Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * Redistributions in binary form must reproduce the above + * copyright notice, this list of conditions and the following + * disclaimer in the documentation and/or other materials provided + * with the distribution. + * + * Neither the name of Cornell University nor the names of its + * contributors may be used to endorse or promote products derived + * from this software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND + * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, + * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF + * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR + * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF + * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED + * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN + * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE + * POSSIBILITY OF SUCH DAMAGE. + * + * Authors: Tuan Ta + */ + +//------------------------------------------------------------------------ +// sysclone_d tests basic functionalities of clone system call: +// - create a new thread +// - assign a new per-thread stack frame to the child thread +// - assign a new per-thread TLS to the child thread +// +// In addition to testing clone(), sysclone_d partially checks +// functionalities of futex and exit system calls that are used to +// facilitate thread exit and synchronization. +//------------------------------------------------------------------------ + +#include "riscv_test.h" +#include "test_macros.h" +#include "test_macros_mt_ecall.h" + + RVTEST_RV64U + RVTEST_CODE_BEGIN + +#define MAX_NUM_THREADS 20 + +//------------------------------------------------------------------------ +// Master thread creates new threads, waits for all threads to complete, +// deallocates threads and checks result +//------------------------------------------------------------------------ + li a0, MAX_NUM_THREADS + call _create_threads + + la t6, n_worker_threads + ld a0, (t6) + beqz a0, _fail // exit if there's no worker thread + call _join + + la t6, n_worker_threads + ld a0, (t6) + call _check + + la t6, n_worker_threads + ld a0, (t6) + call _delete_threads + + li a0, SUCCESS + + RVTEST_CODE_END + +//------------------------------------------------------------------------ +// mt_test function executed by child threads +//------------------------------------------------------------------------ +_mt_test: + // get this thread's TID + li a7, SYSCALL_GETTID + ecall + + // store the TID to both stack and TLS of this thread + addi sp, sp, -8 + sd a0, (sp) + sd a0, (tp) + + RVTEST_CODE_END + +//------------------------------------------------------------------------ +// _check: +// The master thread looks into the stack and TLS of each child thread +// and check if the child thread's TID was written in both places. +// +// This function assumes the following structure in the calling thread's +// stack frame +// +// | child_stack_ptr_0 | << fp: frame pointer +// | child_tls_ptr_0 | +// | child_thread_id_0 | +// | saved_child_thread_id_0 | +// | child_stack_ptr_1 | +// | child_tls_ptr_1 | +// | child_thread_id_1 | +// | saved_child_thread_id_1 | +// | ... | << sp: stack pointer +// +// This function takes a number of threads to check in a0 +//------------------------------------------------------------------------ + +_check: + mv t0, a0 // get the number of threads + mv s0, ra // save return register + mv s1, sp // save stack pointer +1: + ld t1, (sp) // get child_thread_saved_id + + addi sp, sp, 8 + ld t2, (sp) // get child_thread_id + bnez t2, _fail // this child_thread_id should have been cleared + + addi sp, sp, 8 + ld t3, (sp) // get child_tls_ptr + ld t3, (t3) // get the first value stored in child's TLS + bne t1, t3, _fail // child_tid should have been saved in the TLS + + addi sp, sp, 8 + ld t4, (sp) // get child_stack_ptr + li t5, MEM_SIZE + add t4, t4, t5 // get the high address of child's stack + ld t4, -8(t4) // get the first value stored in child's stack + bne t1, t4, _fail // child_tid should have been saved in the stack + + addi sp, sp, 8 + + // decrement the number of threads to wait for + addi t0, t0, -1 + bnez t0, 1b + + // finish checking all threads + mv ra, s0 // restore return register + mv sp, s1 // restore stack pointer + ret + +_fail: + li a0, FAILURE + RVTEST_CODE_END + + .data + +MT_DATA |