sys_macros.m

# Tim Henderson & Steve Johnson
# System Macros for jist
# Meant to be available to every program for jist

#define __save_frame global
    sw      $fp 0($sp)          # save the old frame pointer
    addu    $fp $sp $0          # move the frame pointer to point at top of frame
    subu    $sp $sp 44          # move the stack pointer down 44
    sw      $fp 40($sp)         # save the old stack pointer
    sw      $ra 36($sp)         # save the return address
    sw      $s0 32($sp)         # save registers $s0 - $s7
    sw      $s1 28($sp)
    sw      $s2 24($sp)
    sw      $s3 20($sp)
    sw      $s4 16($sp)
    sw      $s5 12($sp)
    sw      $s6 8($sp)
    sw      $s7 4($sp)
#end

#define __restore_frame global
    subu    $sp $fp 44          # move the stack pointer to the orginal unmodified bottom
    lw      $ra 36($sp)         # load the return address
    lw      $s0 32($sp)         # load registers $s0 - $s7
    lw      $s1 28($sp)
    lw      $s2 24($sp)
    lw      $s3 20($sp)
    lw      $s4 16($sp)
    lw      $s5 12($sp)
    lw      $s6 8($sp)
    lw      $s7 4($sp)
    lw      $fp 44($sp)         # load the old frame pointer
    lw      $sp 40($sp)         # load the old stack pointer
#end

#define ___stshortcut global
    li $a0 %1
    var_store $a0 %2
#end

#define __save_temps global
    li $a0 10
    init_varstore $a0
    ___stshortcut 10 $t0
    ___stshortcut 9  $t1
    ___stshortcut 8  $t2
    ___stshortcut 7  $t3
    ___stshortcut 6  $t4
    ___stshortcut 5  $t5
    ___stshortcut 4  $t6
    ___stshortcut 3  $t7
    ___stshortcut 2  $t8
    ___stshortcut 1  $t9
#end

#define ___rtshortcut global
    li $a0 %1
    var_restore %2 $s0
#end

#define __restore_temps global
    ___rtshortcut 10 $t0
    ___rtshortcut 9  $t1
    ___rtshortcut 8  $t2
    ___rtshortcut 7  $t3
    ___rtshortcut 6  $t4
    ___rtshortcut 5  $t5
    ___rtshortcut 4  $t6
    ___rtshortcut 3  $t7
    ___rtshortcut 2  $t8
    ___rtshortcut 1  $t9
#end

# var_store dst src
#     dst : a number (1-N) indicating which spot you want to store it in, gets trampled
#     src : a register containing the value you want to store
#
#     varstore will store exactly one word of data at a time. It stores it on the stack and is 
#     compatable with call/return
#define var_store global
    @dst = %1
    @src = %2
    sll     @dst @dst 2         # mul the dst by 4
    addu    @dst @dst $sp       # add the stack pointer to the @dst
    sw      @src 0(@dst)
#end

# var_restore dst src
#     dst : a register where you want the value placed
#     src : a number (1-N) indicating which spot you want to restore from, gets trampled
#
#     varstore will store exactly one word of data at a time. It stores it on the stack and is 
#     compatable with call/return
#define var_restore global
    @dst = %1
    @src = %2
    sll     @src @src 2         # mul the dst by 4
    addu    @src @src $sp       # add the stack pointer to the @dst
    lw      @dst 0(@src)
#end

# init_varstore reg
#     reg : contains the size it words you want the store to be, note it gets trampled
#
#     varstore is a place to put variables when you are inside a procedure and run out of $s regs
#     it can only be initiated once per procedure call.
#define init_varstore global
    @size = %1
    
    sll     @size @size 2       # mul the size by 4
    subu    $sp $sp @size
#end

# store_arg arg
#     arg : the register you would like to store on frame
#define store_arg global
    sw      %1 0($sp)
    subu    $sp $sp 4           # move the stack pointer down 4
#end

# load_arg arg_num destination
#     arg_num : the number of the argument you want it must an immediate value
#     destination : register you want your argument in
#     temp_reg : this macro requires a temporary register
#     args number works like this
#     store_arg $t3  --> arg 3
#     store_arg $s3  --> arg 2
#     store_arg $v1  --> arg 1
#     call my_procedure
#define load_arg global
    li      %2 %1
    mul     %2 %2 4
    addu    %2 $fp %2
    lw      %2 0(%2)
#end

# same as load_arg, but specify arg_num in a register instead
#define load_arg_by_reg global
    add     %2 %1 $zero
    mul     %2 %2 4
    addu    %2 $fp %2
    lw      %2 0(%2)
#end

# call label
#     label : label you are jumping to
#define call global
    __save_frame
    jal     %1
    __restore_frame
#end

# call_reg reg
#     label : label you are jumping to
#define call_reg global
    __save_frame
    la      $ra ret
    jr      %1
ret:
    __restore_frame
#end

# __min_save
#     most minimal stack frame save
#define __min_save
    sw      $fp 0($sp)          # save the old frame pointer
    addu    $fp $sp $0          # move the frame pointer to point at top of frame
    subu    $sp $sp 12          # move the stack pointer down 12
    sw      $fp 8($sp)         # save the old stack pointer
    sw      $ra 4($sp)         # save the return address
#end

# __min_restore
#     most minimal stack frame restore
#define __min_restore
    subu    $sp $fp 12           # move the stack pointer to the orginal unmodified bottom
    lw      $ra 4($sp)         # save the return address
    lw      $fp 12($sp)         # load the old frame pointer
    lw      $sp 8($sp)         # load the old stack pointer
#end


# exec label
#     label : label you are jumping to
# like call but more minimal stack save
#define exec global
    __min_save
    jal     %1
    __min_restore
#end

#define return global
    jr      $ra
#end

# __save_args
#     this macro saves $a0, $a1, $a2, $a3, $v0, $v1 on the stack
#define __save_args global
    sw      $fp 0($sp)          # save the old frame pointer
    addu    $fp $sp $0          # move the frame pointer to point at top of frame
    subu    $sp $sp 32          # move the stack pointer down 32
    sw      $fp 28($sp)         # save the old stack pointer
    sw      $a0 24($sp)         # save $a0
    sw      $a1 20($sp)         # save $a1
    sw      $a2 16($sp)         # save $a2
    sw      $a3 12($sp)         # save $a3
    sw      $v0 8($sp)          # save $v0
    sw      $v1 4($sp)          # save $v1
#end

# __restore_args
#     this macro restores $a0, $a1, $a2, $a3, $v0, $v1 on the stack
#define __restore_args global
    subu    $sp $fp 32          # move the stack pointer to the orginal unmodified bottom
    lw      $a0 24($sp)         # load $a0
    lw      $a1 20($sp)         # load $a1
    lw      $a2 16($sp)         # load $a2
    lw      $a3 12($sp)         # load $a3
    lw      $v0 8($sp)          # load $v0
    lw      $v1 4($sp)          # load $v1
    lw      $fp 32($sp)         # load the old frame pointer
    lw      $sp 28($sp)         # load the old stack pointer
#end

# quickstore reg
#     reg : the register you want to store
#     stores one register on the static heap (use with quickrestore)
#define quickstore global
    sw      %1 0($gp)           # save %1 at the current $gp location
    addu    $gp $gp 4           # increment $gp by 4
#end

# quickrestore dst
#     dst : the destination you want the restored value placed
#     restores one register from the static heap (use with quickstore)
#define quickrestore global
    subu    $gp $gp 4           # decrement $gp by 4
    lw      %1 0($gp)           # save %1 at the current $gp location
#end

# sbrk_imm amt dst
#     amt : the amount of memory you want, must be an immediate value
#define sbrk_imm global
    __save_args
    li      $v0 9               # system call code for sbrk
    addi    $a0 $0 %1           # amount
    syscall                     # make the call
    quickstore $v0
    __restore_args
    quickrestore %2
#end

# sbrk_addr addr dst
#     addr : location in memory whos value equals the amount of memory you 
#            want to allocate
#define sbrk_addr global
    __save_args
    li      $v0 9               # system call code for sbrk
    lw      $a0 %1              # amount
    syscall                     # make the call
    quickstore $v0
    __restore_args
    quickrestore %2
#end

# sbrk reg dst
#     reg : stored in the reg should be the amount of memory you want to allocate
#define sbrk global
    quickstore %1
    __save_args
    quickrestore %1
    li      $v0 9               # system call code for sbrk
    add     $a0 %1 $0           # amount
    syscall                     # make the call
    quickstore $v0
    __restore_args
    quickrestore %2
#end

#define exit global
{
    __save_args
    
    println exit_msg
    
    la      $a0 KMSG
    li      $a1 2
    sw      $a1 0($a0)  #KMSG = 2
    
    la      $a0 wait_return
    subu    $a0 $a0 4
    mtc0    $a0 $14
    li      $a0 0
    mtc0    $a0 $13
    la      $a0 exception_handler
    jr      $a0
wait_return:
    __restore_args
.data
exit_msg: .asciiz "------Process exit------"
.text
}
#end

#define kill_jist global
    li $v0 10
    syscall
#end

#define wait global
    __save_frame
    __save_args
    # __save_temps
    
    la      $a0 KMSG
    li      $a1 1
    sw      $a1 0($a0)  #KMSG = 1
    
    la      $a0 wait_return
    subu    $a0 $a0 4
    mtc0    $a0 $14
    li      $a0 0
    mtc0    $a0 $13
    la      $a0 exception_handler
    jr      $a0
wait_return:
    # __restore_temps
    __restore_args
    __restore_frame
#end

#define disable_interrupts global
    __save_args
    mfc0    $a0 $12             # load the status register
    lui     $a1 0xffff
    ori     $a1 0xfffc
    and     $a0 $a0 $a1         # enable the interrupts
    mtc0    $a0 $12             # push the changes back to the co-proc
    nop
    nop
    __restore_args
#end

#define enable_interrupts global
    __save_args
    mfc0    $a0 $12             # load the status register
    lui     $t1 0xffff
    ori     $t1 0xfffd
    and     $a0 $a0 $t1         # enable the interrupts
    ori     $a0 $a0 0x1         # set exception level to 0 this re-enables interrupts
    mtc0    $a0 $12             # push the changes back to the co-proc
    nop
    nop
    __restore_args
#end

#define enable_clock_interrupt global
    __save_args
    mfc0    $a0 $9              # get the current clock value
    add     $a0 $a0 1           # add 1
    mtc0    $a0 $11             # push to compare
    __restore_args
#end


#define disable_clock_interrupt global
    mtc0    $0 $11             # push to compare
#end

# sem_wait loc mem_id hcb_addr
#define sem_wait global
    @loc = %1
    @mem_id = %2
    @hcb_addr = %3
    @sem = $a0
    @err = $a1
loop:
    disable_interrupts
    get     @loc @mem_id @hcb_addr @sem @err
    enable_interrupts
    bne     @err $0 getfail
    lw      $a0 0($a1)
    beq     $a0 $0 lock
    wait
    j       loop
lock:
    disable_interrupts
    addi    $a0 $a0 1
    sw      $a0 0($a1)
    enable_interrupts
    j       end

    getfail:
        println getfail_msg
        j       end
.data
    getfail_msg: .asciiz "get failed in sem_wait"
.text
end:
#end

# sem_signal loc mem_id hcb_addr
#define sem_signal global
    @loc = %1
    @mem_id = %2
    @hcb_addr = %3
    @sem = $a0
    @err = $a1
    
    disable_interrupts
    get     @loc @mem_id @hcb_addr @sem @err
    bne     @err $0 getfail
    beq     @sem $0 alread_zero
    sub     @sem @sem 1
    put     @loc @mem_id @hcb_addr @sem @err
    bne     @err $0 putfail
alread_zero:
    enable_interrupts
    j       end

    getfail:
        println getfail_msg
        j       end
    putfail:
        println putfail_msg
        j       end
.data
    getfail_msg: .asciiz "get failed in sem_signal"
    putfail_msg: .asciiz "put failed in sem_signal"
.text
end:
#end



# blocksize mem_id hcb_addr dst err
#     mem_id : the memory id for the block you are accessing
#     hcb_addr : the address of the hcb
#     dst : the register you want the result placed
#     err : error code 0 is success
#define blocksize global
    @mem_id = %1
    @hcb_addr = %2
    @dst = %3
    @err = %4
    
    addu    $a1 @mem_id $0
    addu    $a2 @hcb_addr $0
    call    blocksize
    addu    @err $v0 $0
    addu    @dst $v1 $0
#end

# get loc mem_id hcb_addr dst err
#     loc : the word you want to get should be from 0-(n-1) where n is len of block, reg
#     mem_id : the memory id for the block you are accessing
#     hcb_addr : the address of the hcb
#     dst : the register you want the result placed
#     err : error code 0 is success
#define get global
    @word = %1
    @mem_id = %2
    @hcb_addr = %3
    @dst = %4
    @err = %5
    
    addu    $a0 @word $0
    addu    $a1 @mem_id $0
    addu    $a2 @hcb_addr $0
    call    __getword
    addu    @err $v0 $0
    addu    @dst $v1 $0
#end

#define geti global
    @mem_id = %2
    @hcb_addr = %3
    @dst = %4
    @err = %5
    
    addu    $a0 $0 %1
    addu    $a1 @mem_id $0
    addu    $a2 @hcb_addr $0
    call    __getword
    addu    @err $v0 $0
    addu    @dst $v1 $0
#end

# put loc mem_id hcb_addr val err
#     loc : the word you want to put should be from 0-(n-1) where n is len of block
#     mem_id : the memory id for the block you are accessing
#     hcb_addr : the address of the hcb
#     val : the value (in a reg) that you want put into the word
#     err : error code 0 is success
#define put global
    @word = %1
    @mem_id = %2
    @hcb_addr = %3
    @value = %4
    @err = %5
    
    addu    $a0 @value $0
    addu    $a1 @word $0
    addu    $a2 @mem_id $0
    addu    $a3 @hcb_addr $0
    call    __putword
    addu    @err $v0 $0
#end

#define puti global
    @mem_id = %2
    @hcb_addr = %3
    @value = %4
    @err = %5
    
    addu    $a0 @value $0
    addu    $a1 $0 %1
    addu    $a2 @mem_id $0
    addu    $a3 @hcb_addr $0
    call    __putword
    addu    @err $v0 $0
#end

# alloci size hcb_addr dest
#define alloci
    @size = %1
    @hcb_addr = %2
    @dest = %3
    
    addu    $a0 $0 @size
    addu    $a1 @hcb_addr $zero
    call    alloc
    addu    @dest $v0 $zero
    addu    @hcb_addr $v1 $zero
#end

# khcb_writeback hcb_addr
#define khcb_writeback
    @khcb_addr = $t0
    @hcb_addr = %1
    la  $t0  KHCB_ADDR
    sw  %1   0($t0)
#end

# khcb_getaddr hcb_addr
#define khcb_getaddr
    @khcb_addr = $t0
    @hcb_addr = %1
    la  $t0  KHCB_ADDR
    lw  %1   0($t0)
#end

#launch_program pnum
#define launch_program global
    la $a0 user_program_locations
    lw $a0 0($a0)
    call load_process
#end