; Copyright (c) 2023 Marvin Borner
%include "config.asm"

global main
extern printf
extern array
extern size

bits 64

;;; main

main:
	push    rbp
	mov     ecx, size

loop:
	push    rcx
	call    thread
	pop     rcx
	dec     ecx
	jnz     loop

end:
	call    thread_wait_all
	pop     rbp
	mov     rax, 0x3c
	mov     rdi, 0
	syscall ; exit

%if USE_SYSCALL

; uses nanosleep syscall
sleep:
	mov     rax, TIMEOUT
	mov     rcx, rsi
	mul     rcx
	push    qword rax ; ns
	push    qword 0 ; s
	mov     rax, 0x23
	mov     rdi, rsp
	xor     rsi, rsi
	syscall
	add     rsp, 16
	ret

%else

; uses busy loop
sleep:
	mov     rax, TIMEOUT
	mov     rcx, rsi
	mul     rcx
.loop:
	dec     rax
	jnz     .loop
	ret

%endif

print:
	push    rsi
	call    sleep
	pop     rsi
	mov     rdi, format
	xor     rax, rax
	call    printf
	ret

thread:
	lea     rax, print
	mov     esi, [array + (ecx - 1) * 4]
	call    thread_run
	ret

;;; utils

thread_run:
	push    rax
	mov     rax, 0x39
	syscall ; fork
	cmp     rax, 0
	mov     r10, rax
	pop     rax
	je      child
	mov     rax, r10
	ret

thread_wait_all:
	sub     rsp, 4
	mov     rdi, -1
	mov     rax, 0x3d
	mov     rsi, rsp
	xor     rdx, rdx
	xor     r10, r10
	syscall ; wait4
	add     rsp, 4
	cmp     rax, -10 ; idk why
	jne     thread_wait_all
	ret

child:
	call    rax

	mov     rax, 0x3c
	mov     rdi, 0
	syscall ; exit

;;; data

section .data

format: db "%d", 10, 0