Reducing proc_print_time size

Using smaller operations

wtfstopw.asm

@@ -27,6 +27,8 @@
 ; 	press enter to exit
 ; 	send SIGINT (with kill -2 ir ctrl + c) for a new lap
 [bits 64]
+%use smartalign
+ALIGNMODE k8
 
 %define O_NONBLOCK 0x800
 
@@ -72,10 +74,13 @@ align 8
 
 section .data
 align 8
+
 	TIMESPEC ts_start
 	TIMESPEC ts_cur
 	TIMESPEC ts_sleep
 
+	time_res: dq 2 ; 2 digits bellow seconds can grow to 8
+
 	hours: times 8 db '0' ; 8 digits is the max for hours
 	timestr: db ":00:00." 
 		times 8 db '0' ; 8 digits for nanoseconds
@@ -85,8 +90,6 @@ align 8
 	timestrlen: equ timestrend - timestr
 	%define HOURSLEN 8
 
-	time_res: dq 2 ; 2 digits bellow seconds can grow to 8
-
 	fcntl_flag: dq 0
 	SIGACTION sigaction
 
@@ -197,6 +200,8 @@ setsleep_endloop:
 mov [ts_sleep.tv_nsec], rax
 
 std ; set DF for string operations
+
+align 32
 main_loop:
 	push 2 ; stderr
 	push 0xD ; \r
@@ -225,9 +230,9 @@ flush_stdin:
 	mov rsi, buf
 	mov rdx, 1
 	syscall
-	cmp rax, 0
-	je newline_exit
-	jg flush_stdin
+	test rax, rax
+	jz newline_exit
+	jns flush_stdin
 
 mov rdi, 0 ; EXIT OK
 ;
@@ -241,8 +246,8 @@ exit:
 	mov rsi, 4 ; F_SETFL
 	mov rdx, [fcntl_flag]
 	syscall
-	cmp rax, 0
-	je exit_end
+	test rax, rax
+	jz exit_end
 	pop rdi ; failed to restore
 	push 1 ; exit FAIL
 
@@ -268,42 +273,43 @@ newline_exit:
 	mov rdi, 0 ; exit OK
 	jmp exit
 
-
-
-align 32
 ;
 ; Print current time on FD and add a leading char CHR
 ; push FD & push CHR to set arguments
 ; 	Warning : DF must be set
 ;
+align 32
 proc_print_time:
 	
 	mov rax, 228 ; clock_gettime
-	mov rdi, 0 ; CLOCK_REALTIME
+	xor rdi, rdi ; CLOCK_REALTIME
 	mov rsi, ts_cur
 	syscall ; updating ts_cur time
 
-	mov rax, [ts_cur.tv_nsec]
-	sub rax, [ts_start.tv_nsec]
-	cmp rax, 0 ; Calculating elapsed ns
+	mov rsi, ts_start.tv_sec
+%define RSIPTR(x) rsi + x - ts_start.tv_sec
+
+	mov rax, [RSIPTR(ts_cur.tv_nsec)] ;mov rax, [ts_cur.tv_nsec]
+	sub rax, [RSIPTR(ts_start.tv_nsec)] ;sub rax, [ts_start.tv_nsec]
 	jge print_time_us_cont
 	add rax, 1000000000 ; negativ result
-	sub qword [ts_cur.tv_sec], 1
+	sub qword [RSIPTR(ts_cur.tv_sec)], 1 ;sub qword [ts_cur.tv_sec], 1
 	print_time_us_cont:
 
 	xor rdx, rdx
-	div qword [ts_sleep.tv_nsec] ; Divide result given time_res
+	div qword [RSIPTR(ts_sleep.tv_nsec)] ; Divide result given time_res
 
-	mov r9, [rsp + 8]
-	mov byte [timestr+timestrlen-1], r9b ; set last chr from 1st argument
+	mov rbx, [rsp + 8]
+	mov byte [RSIPTR(timestrend - 1)], bl ; set last chr from 1st argument
 
 	; set the nanosec chars (time_res chars) in timestr
 	mov rbx, 0x2020202020202020
-	mov rcx, [time_res]
-	mov r8, 10
+	mov rcx, [RSIPTR(time_res)]
+	mov rdi, 10
+	xor rdx, rdx
 	procpt_loopns:
-		xor rdx, rdx
-		div r8
+		xor dl, dl
+		div rdi
 		or dl, 0x30 ; '0'
 		shl rbx, 8
 		mov bl, dl
@@ -311,46 +317,46 @@ proc_print_time:
 	mov [timestr+7], rbx
 
 	; filling timestr with seconds & minutes chars
-	mov rax, [ts_cur.tv_sec]
-	sub rax, [ts_start.tv_sec] ; rax now contain elapsed seconds
+	mov rax, [RSIPTR(ts_cur.tv_sec)]
+	sub rax, [RSIPTR(ts_start.tv_sec)] ; rax now contain elapsed seconds
 
-	mov r8, 10
-	mov r9, 6
+	mov rcx, 10
+	mov rdi, 6
 
 	xor rdx, rdx
-	div r8
+	div rcx
 	mov bh, dl
 	xor dl, dl
-	div r9
+	div rdi
 	mov bl, dl
 	or bx, 0x3030
-	mov word [timestr + 4], bx
+	mov word [RSIPTR(timestr + 4)], bx
 
 	xor dl, dl
-	div r8
+	div rcx
 	mov bh, dl
 	xor dl, dl
-	div r9
+	div rdi
 	mov bl, dl
 	or bx, 0x3030
-	mov word [timestr + 1], bx
+	mov word [RSIPTR(timestr + 1)], bx
 
 	; using rbx to stores the hours string
 	xor rbx, rbx
 	mov rcx, HOURSLEN
-	mov r8, 10
+	mov rdi, 10
 	procpt_looph:
 		xor dl, dl
-		div r8
+		div rdi
 		shl rbx, 8
 		mov bl, dl
 		cmp rax, 0
 		loopne procpt_looph
 	procpt_looph_end:
 
-	mov r8, HOURSLEN - 2
+	mov rax, HOURSLEN - 2
 	cmp rcx, HOURSLEN - 2
-	cmovl r8, rcx ; r8 stores hours digit count (at least 2)
+	cmovl rax, rcx ; rsi stores hours digit count (at least 2)
 	cmp rcx, 0
 	jz procpt_looph2_end
 	procpt_looph2:
@@ -358,19 +364,23 @@ proc_print_time:
 		loopne procpt_looph2
 	procpt_looph2_end:
 
-	mov r9, 0x3030303030303030 ; ASCII convertion
-	or rbx, r9
-	mov [hours], rbx
+	mov rdi, 0x3030303030303030 ; ASCII convertion
+	or rbx, rdi
+	mov [RSIPTR(hours)], rbx
 
-	mov rax, 1 ; write
 	mov rdi, [rsp + 16] ; fd as 2nd argument
-	lea rsi, [hours + r8] ; hours start pointer
 	mov rdx, timestrlen + HOURSLEN
-	sub rdx, r8 ; timestr + hours len
+	sub rdx, rax ; timestr + hours len
+	lea rsi, [RSIPTR(hours + rax)] ; hours start pointer
+	mov rax, 1 ; write
 	syscall
+	test rax, rax
+	js fault
 
 	ret 16
 
+
+
 ;
 ;	sig handler for SIGINT displaying lap count and time on stdout
 ;