main_loop and _start proc enhancement

main_loop alignement and size reduction under 32 bytes
Size reduction of _start code

wtfstopw.asm

@@ -68,10 +68,6 @@ ENDSTRUC
 	%define sigaction.sa_mask sigaction+SIGACTION_STRUC.sa_mask
 %endmacro
 
-section .bss
-align 8
-	buf: resb 1
-
 section .data
 align 8
 
@@ -90,6 +86,8 @@ align 8
 	timestrlen: equ timestrend - timestr
 	%define HOURSLEN 8
 
+	buf: db 0
+
 	fcntl_flag: dq 0
 	SIGACTION sigaction
 
@@ -141,13 +139,19 @@ _start:
 jmp arg_parse
 arg_ok:
 
+mov rbx, ts_start
+%define RBXPTR(x) rbx + x - ts_start
+
 ; set stdin reads non blocking
 xor rdx, rdx
 xor rdi, rdi
 mov rax, 72 ; fcntl
 mov rsi, 3 ; F_GETFL
 syscall
-mov [fcntl_flag], rax
+					mov rdx, O_NONBLOCK
+					not rdx
+					and rax, rdx
+mov [RBXPTR(fcntl_flag)], rax
 mov rdx, rax
 or rdx, O_NONBLOCK
 mov rax, 72 ; fcntl
@@ -158,15 +162,17 @@ jne fault
 
 ; preparing SIGINT catch
 mov rax, proc_lap_handler
-mov qword [sigaction.sa_handler], rax
-mov eax, 0x14000000 ; SA_RESTART | SA_RESTORER
-mov dword [sigaction.sa_flags], eax
-mov rax, sig_restorer
-mov qword [sigaction.sa_restorer], rax
+mov qword [RBXPTR(sigaction.sa_handler)], rax
+mov ecx, 0x14000000 ; SA_RESTART | SA_RESTORER
+mov dword [RBXPTR(sigaction.sa_flags)], ecx
+add rax, sig_restorer - proc_lap_handler
+mov qword [RBXPTR(sigaction.sa_restorer)], rax
 
 mov rax, 13 ; sys_rt_sigaction
 mov rdi, 2 ; SIGINT
-mov rsi, sigaction
+mov rsi, rbx
+add rsi, sigaction - ts_start
+;mov rsi, sigaction
 mov rdx, 0 ; NULL
 mov r10, 8 ; sig_size
 syscall
@@ -175,12 +181,13 @@ jne fault
 
 mov rax, 228 ; clock_gettime
 mov rdi, 0 ; CLOCK_REALTIME
-mov rsi, ts_start
+mov rsi, rbx
 syscall
 
 mov rax, 1 ; write
 mov rdi, 2 ; stderr
-mov rsi, startmsg
+mov rsi, rbx
+add rsi, startmsg - ts_start
 mov rdx, startmsglen
 syscall
 
@@ -188,41 +195,49 @@ syscall
 ; div sleep time by 10 for each digits added bellow seconds
 mov rax, 100000000
 xor rcx, rcx
-mov cl, [time_res]
+mov cl, [RBXPTR(time_res)]
 dec cl
 test cl, cl
 jz setsleep_loopend
 
-mov rbx, 10
+mov rsi, 10
 xor rdx, rdx
 setsleep_loop:
-	div rbx
+	div rsi
 	loop setsleep_loop
 setsleep_loopend:
 
-mov [ts_sleep.tv_nsec], rax
+mov [RBXPTR(ts_sleep.tv_nsec)], rax
+mov r12, proc_print_time
+mov r10, 0x0d02
 
 std ; set DF for string operations
 
-align 32
+align 16
 main_loop:
-	push 2 ; stderr
-	push 0xD ; \r
-	call proc_print_time
 
+	push rbx
+	push r10 ; \r & stderr
+	call r12
+
+	pop rbx
 	; Attempt to read from stdin
 	; if something read, enter has been pressed
 	xor rax, rax
-	xor rdi, rdi
-	mov rsi, buf
+	push rax
+	push rax
+	pop rdi
+	mov rsi, rbx
+	add rsi, buf - ts_start
 	mov rdx, 1
 	syscall
 	test rax, rax
 	jns flush_stdin ; flush & exit
 
 	mov rax, 35 ; nanosleep
-	mov rdi, ts_sleep
-	xor rsi, rsi
+	mov rdi, rbx
+	add rdi, ts_sleep - ts_start
+	pop rsi
 	syscall
 
 jmp main_loop ; main_loop
@@ -281,7 +296,7 @@ newline_exit:
 ; push FD & push CHR to set arguments
 ; 	Warning : DF must be set
 ;
-align 32
+align 16
 proc_print_time:
 	
 	mov rax, 228 ; clock_gettime
@@ -303,7 +318,7 @@ proc_print_time:
 	xor rdx, rdx
 	div qword [RSIPTR(ts_sleep.tv_nsec)] ; Divide result given time_res
 
-	mov rbx, [rsp + 8]
+	mov bl, [rsp + 9]
 	mov byte [RSIPTR(timestrend - 1)], bl ; set last chr from 1st argument
 
 	; set the nanosec chars (time_res chars) in timestr
@@ -373,7 +388,8 @@ proc_print_time:
 	or rbx, rdi
 	mov [RSIPTR(hours)], rbx
 
-	mov rdi, [rsp + 16] ; fd as 2nd argument
+	xor rdi, rdi
+	mov dil, [rsp + 8] ; fd as 2nd argument
 	mov rdx, timestrlen + HOURSLEN
 	sub rdx, rax ; timestr + hours len
 	lea rsi, [RSIPTR(hours + rax)] ; hours start pointer
@@ -382,7 +398,8 @@ proc_print_time:
 	test rax, rax
 	js fault
 
-	ret 16
+	ret 8
+%undef RSIPTR
 
 
 
@@ -438,8 +455,7 @@ proc_lap_handler:
 	js fault
 
 	std ; set DF for string operations
-	push 1 ; stdout
-	push 0xA ; \n
+	push 0x0A01 ; \n & stdout
 	call proc_print_time
 
 	ret