diff options
Diffstat (limited to 'tests/test-progs/asmtest/src/riscv/isa/rv64samt/sysfutex3_d.S')
-rw-r--r-- | tests/test-progs/asmtest/src/riscv/isa/rv64samt/sysfutex3_d.S | 172 |
1 files changed, 172 insertions, 0 deletions
diff --git a/tests/test-progs/asmtest/src/riscv/isa/rv64samt/sysfutex3_d.S b/tests/test-progs/asmtest/src/riscv/isa/rv64samt/sysfutex3_d.S new file mode 100644 index 000000000..4616de811 --- /dev/null +++ b/tests/test-progs/asmtest/src/riscv/isa/rv64samt/sysfutex3_d.S @@ -0,0 +1,172 @@ +/* + * 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, Moyang Wang + */ + +//------------------------------------------------------------------------ +// sysfutex3_d tests FUTEX_CMP_REQUEUE functionalities of futex system +// call: +// - make worker threads waiting on futex 1 +// - wake up 1 thread, requeue the rest of the threads to futex 2 +// - wake all threads waiting on futex 1 and futex 2 +//------------------------------------------------------------------------ + +#include "riscv_test.h" +#include "test_macros.h" +#include "test_macros_mt_ecall.h" + + RVTEST_RV64U + RVTEST_CODE_BEGIN + +#define NUM_THREADS 20 + +//------------------------------------------------------------------------ +// Master thread creates new threads, call _master function, waits for all +// threads to complete, deallocates threads and checks result +//------------------------------------------------------------------------ + li a0, NUM_THREADS + call _create_threads + + la t6, n_worker_threads + ld a0, (t6) + beqz a0, _fail // exit if there's no worker thread + + call _master_work + + la t6, n_worker_threads + ld a0, (t6) + 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 + +//------------------------------------------------------------------------ +// master_work function executed by the parent/master thread +//------------------------------------------------------------------------ +_master_work: + mv s0, ra // save return address + li t0, 0 // number of threads that have been waken + la t1, n_worker_threads + ld t1, (t1) + +1: + // futex(futex_X, FUTEX_CMP_REQUEUE, 1, INT_MAX, futex_Y, *futex_X) + la a0, futex_X + li a1, FUTEX_CMP_REQUEUE + li a2, 1 // wake up at most 1 thread + li a3, 1000 // practically is INT_MAX + la a4, futex_Y // move all other waiter to futex_Y + li a5, 0 + li a7, SYSCALL_FUTEX + ecall + + // loop until wake up one thread, all other waiter are requeued to + // futex_Y + beqz a0, 1b + + addi t0, t0, 1 + +2: + // alternating between futex_X and futex_Y + // because there could be new threads added to futex_X's queue + // after our futex_requeue + + // futex(futex_Y, FUTEX_WAKE_PRIVATE, 0) + la a0, futex_Y + li a1, FUTEX_WAKE + li a2, 1 // wake up at most 1 thread + li a7, SYSCALL_FUTEX + ecall + + add t0, t0, a0 // track the number of waken threads so far + + // futex(futex_X, FUTEX_WAKE_PRIVATE, 0) + la a0, futex_X + li a1, FUTEX_WAKE + li a2, 1 // wake up at most 1 thread + li a7, SYSCALL_FUTEX + ecall + + add t0, t0, a0 // track the number of waken threads so far + + // keep waking up until all threads are waken up + blt t0, t1, 2b + + // restore return address and return + mv ra, s0 + ret + +//------------------------------------------------------------------------ +// mt_test function executed by child threads +// +// Wait on futex_X +//------------------------------------------------------------------------ +_mt_test: + // futex(futex_X, FUTEX_WAIT_PRIVATE, 1) + la a0, futex_X + li a1, FUTEX_WAIT + li a2, 0 // expected val of futex_X + li a7, SYSCALL_FUTEX + ecall + + RVTEST_CODE_END + +//------------------------------------------------------------------------ +// _check: +// counter should equals to number of child threads +//------------------------------------------------------------------------ + +_check: + ret + +_fail: + li a0, FAILURE + RVTEST_CODE_END + + .data + +futex_X: .dword 0 +futex_Y: .dword 0 + +MT_DATA |