/*
 * 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
 */

//------------------------------------------------------------------------
// This test_macros includes necessary functions and macros to create
// and exit threads. They're used in multi-threaded assembly tests.
// This assumes the target system can concurrently support 4 different
// threads (i.e., 1 master thread and 3 child threads)
//------------------------------------------------------------------------

#ifndef __TEST_MACROS_MT_H
#define __TEST_MACROS_MT_H

#define SYSCALL_MMAP    222
#define SYSCALL_MUNMAP  215
#define SYSCALL_CLONE   220

#define STACK_SIZE      (4096 * 1024)

#define PROT_READ	      0x1
#define PROT_WRITE	    0x2
#define MMAP_PROT_FLAGS (PROT_READ | PROT_WRITE)

#define MAP_PRIVATE	    0x02
#define MAP_ANONYMOUS	  0x20
#define MAP_STACK	      0x20000
#define MMAP_MAP_FLAGS  (MAP_PRIVATE | MAP_ANONYMOUS | MAP_STACK)

#define CLONE_VM	      0x00000100
#define CLONE_FS	      0x00000200
#define CLONE_FILES	    0x00000400
#define CLONE_SIGHAND	  0x00000800
#define CLONE_PARENT	  0x00008000
#define CLONE_THREAD	  0x00010000
#define CLONE_IO		    0x80000000
#define CLONE_FLAGS     (CLONE_VM | CLONE_FS | CLONE_FILES | CLONE_SIGHAND\
                         | CLONE_PARENT | CLONE_THREAD | CLONE_IO)

#define NUM_THREADS     3

#define FAILURE         1
#define SUCCESS         0

#define HARTID          0xF14

//------------------------------------------------------------------------
// create NUM_THREADS child threads
//------------------------------------------------------------------------
_create_threads:
  li      t0, NUM_THREADS
  mv      s0, ra                  // save return register
1:
  jal     ra, _alloc_stack
  addi    sp, sp, -8
  sd      a0, (sp)                // save pointer to the new stack
  jal     ra, _clone_thread       // clone a new thread
  addi    t0, t0, -1
  bnez    t0, 1b
  mv      ra, s0                  // restore return register
  ret

_alloc_stack:
  li      a0, 0
  li      a1, STACK_SIZE
  li      a2, MMAP_PROT_FLAGS
  li      a3, MMAP_MAP_FLAGS
  li      a4, -1
  li      a5, 0
  li      a7, SYSCALL_MMAP
  ecall
  ret

_clone_thread:
  li      a1, STACK_SIZE
  add     a1, a1, a0
  li      a0, CLONE_FLAGS
  li      a7, SYSCALL_CLONE
  ecall
  beqz    a0, _mt_test
  ret

//------------------------------------------------------------------------
// wait for all child threads to exit
//------------------------------------------------------------------------
_join:
  la      t0, barrier
  li      t1, NUM_THREADS
1:
  ld      t2, (t0)
  bne     t1, t2, 1b
  ret

//------------------------------------------------------------------------
// deallocate NUM_THREADS child threads
//------------------------------------------------------------------------
_delete_threads:
  li      t0, NUM_THREADS
  mv      s0, ra                  // save return register
1:
  ld      a0, (sp)                // pop the new stack's pointer
  addi    sp, sp, 8
  jal     ra, _dealloc_stack
  addi    t0, t0, -1
  bnez    t0, 1b
  mv      ra, s0                  // restore return register
  ret

_dealloc_stack:
  li      a1, STACK_SIZE
  li      a7, SYSCALL_MUNMAP
  ecall
  ret

#define MT_DATA                                                           \
  shared_var:   .dword    0;                                              \
  barrier:      .dword    0;                                              \
  array:        .dword    0x00000000deadbeef,                             \
                          0xdeadbeefdeadbeef,                             \
                          0x12343eeaaf423451;                             \

#endif