Subversion Repositories Kolibri OS

Compare Revisions

Regard whitespace Rev 7637 → Rev 7638

/programs/network/sntp/sntp.asm
0,0 → 1,905
;
; SNTP client for KolibriOS
;
; (C) 2019 Coldy
; Thank's you for use this code and software based on it!
; I will glad if it's will be helpful.
;
; Distributed under terms of GPL
;
 
format binary as ""
 
use32
org 0x0
 
db 'MENUET01' ; signature
dd 1 ; header version
dd START ; entry point
dd I_END ; initialized size
dd MEM ; required memory
dd STACKTOP ; stack pointer
dd params ; parameters
dd 0 ; path
__DEBUG__ = 1
__DEBUG_LEVEL__ = 2
 
include '../../proc32.inc'
include '../../macros.inc'
include '../../dll.inc'
include '../../struct.inc'
include '../../network.inc'
include '../../debug-fdo.inc'
include 'time.inc'
 
 
START:
; init heap
mcall 68, 11
test eax, eax
; fatal error (not enough memory)
jz exit_now
; load libraries
stdcall dll.Load, @IMPORT
test eax, eax
; fatal error(imports not loaded)
jnz exit_now
; initialize console
; push 1
; call [con_start]
push str_title
push 250
push 80
push 25
push 80
call [con_init]
;test eax, eax
; fatal error(console error)
;jnz exit
 
; setup params
call parse_params
call tz_validate
; is TZ correct?
cmp ebx,0
je @f
mov eax, ebx
mov ebx, str_err11
jmp .tz_error
@@:
; is command line correct?
cmp eax, 10
jne @f
mov ebx, str_err10
jmp .error
@@:
; empty command line (need help)?
cmp eax, 0
je @f
cinvoke con_printf, str_help
jmp exit
 
@@:
; Prepare to do query time
; Convert host name to IP address
invoke inet_addr, params
; Host name by IP address was provided?
cmp eax, -1
jne .resolved
push esp ; reserve stack place
invoke getaddrinfo, params, 0, 0, esp
; Get ptr to result addrinfo struct
pop esi
; Test for error
test eax, eax
jz @f
mov eax, 30
mov ebx, str_err3
;ret ; Error: Name not resolved!
jmp .error
@@:
mov eax, [esi+addrinfo.ai_addr]
mov eax, [eax+sockaddr_in.sin_addr]
push eax ; Store IP to stack
invoke inet_ntoa, eax
; Store string of IP
mov edx, eax
pop eax ; Load IP from stack
jmp @f
.resolved:
clear edx ; mark host is IP format
@@:
mov [sockaddr1.ip], eax
clear ebx
mov bx, [port]
;cinvoke con_printf, str_query, params, eax, ebx
cinvoke con_printf, str_query, params
cmp edx,0
je @f ; Skip IP display
cinvoke con_printf, str_ip, edx
@@:
cinvoke con_printf, str_port, ebx
; free allocated memory
invoke freeaddrinfo, esi
; Now we ready to query server
call sntp_query_time
cmp eax,1
jne @f
mov eax, 59
jmp .warning
@@:
cmp eax,2
jne @f
mov eax, 61
.warning:
cinvoke con_printf, str_warn, eax
jmp .display
@@:
cmp eax, 41
jge .error
.display:
; Tell user results
;mov esi, datetime ; ?
; Display server date and time
xor eax, eax
mov al, [esi + DateTime.sec]
push eax
mov al, [esi + DateTime.min]
push eax
mov al, [esi + DateTime.hour]
push eax
mov ax, [esi + DateTime.year]
push eax
xor eax, eax
mov al, [esi + DateTime.month]
push eax
mov al, [esi + DateTime.day]
push eax
push str_dt
call [con_printf]
add esp, 7*4
; Display timezone
cmp [tz_h],0
jne @f
cmp [tz_m],0
jne @f
jmp .no_bias
@@:
clear eax, ebx
mov ecx, str_tz
mov al, [tz_h]
test al, al
jns @f
mov byte[ecx+1],'-' ; Change sign
sub bl, al
mov al, bl
@@:
cmp [tz_m],0
jne @f
mov word[ecx+4],10 ; \n\0
@@:
mov bl, [tz_m]
cinvoke con_printf, str_tz, eax, ebx
.no_bias:
cmp [sync],0
je exit
cmp [sync], SYNC_S
jne @f
mov eax, str_s
jmp .sync_ok
@@:
cmp [sync], SYNC_ST
jne @f
mov eax, str_st
jmp .sync_ok
@@:
cmp [sync], SYNC_ST
jne .sync_ok
mov eax, str_ss
.sync_ok:
cinvoke con_printf, str_sync, eax
jmp exit
.error:
mov esi, params
.tz_error:
; Note: esi assign by parse_params
 
; cmp [notify],0
; jne @f
cinvoke con_printf, str_err, eax, esi, ebx
;jmp exit
;@@:
 
; Do call @NOTIFY
 
 
; Finally... exit!
exit:
push 0
call [con_exit]
 
exit_now:
mcall -1
; End of program
 
;notify:
 
; Time zone check helper
tz_validate:
;jmp .exit
cmp [tz_h],-12
jl .fail
cmp [tz_h],14
jg .fail
cmp [tz_m],0
je .exit
cmp [tz_m],30
jne .tz_45m
cmp [tz_h],-9
je .exit
cmp [tz_h],-3
je .exit
cmp [tz_h],3
jl .fail
je .exit
cmp [tz_h],6
jg @f
jle .exit
@@:
cmp [tz_h],9
je .exit
cmp [tz_h],10
jne .fail
je .exit
.tz_45m:
cmp [tz_m],45
jne .fail
cmp [tz_h],5
je .exit
cmp [tz_h],8
je .exit
cmp [tz_h],12
jne .fail
.exit:
clear ebx
ret
.fail:
mov ebx,11
ret
; Sycronization constants
SYNC_S = 1
SYNC_SS = 2
SYNC_ST = 3
 
parse_params:
mov esi, params
mov ebx, esi
.f00:
lodsb
cmp al, 0
jne .f01
dec esi
cmp esi, ebx
jne @f
;no params
mov eax, -1
ret
.exit:
; mark end of TZ
mov byte [ecx+1], 0
; now esi = start of TZ
mov esi,ebp
 
@@:
mov eax, 0
ret
 
.f01:
cmp al, ' '
jne .f00
; Save end of host position
mov edi, esi
dec edi
;mov byte [esi-1], 0
jmp .param
 
.param_loop:
lodsb
cmp al, 0
je @f
cmp al, ' '
jne .invalid
jmp .param
@@:
mov byte [edi], 0
jmp .exit ;ret
.param:
lodsb
cmp al, '-'
jne .invalid
 
lodsb
; cmp al, 'n'
; jne @f
; mov [notify],1
; jmp .param_loop
;@@:
; cmp al, 'p'
; je .p
cmp al, 't'
jne @f
lodsb
cmp al, 'z'
jne .invalid
je .tz
@@:
cmp al, 's'
je .sync
;.p:
; port setup
; lodsb
; cmp al, ' '
; jne .invalid
; call c2n
; test ebx, ebx
; jz .invalid
; mov [port], bx
; jmp .param_loop
.tz:
; tz setup
lodsb
cmp al, ' '
jne .invalid
; save start of TZ
;push esi
mov ebp, esi
call c2n
;cmp ebx, ebx ; 0 is possible
;jz .invalid
mov [tz_h], bl
cmp al, ':'
je .tz_m
;dec esi
jmp @f;.param_loop
 
.tz_m:
call c2n
;test ebx, ebx ; 0 is possible
;jz .invalid
mov [tz_m], bl
@@:
; save end of TZ
;push esi
mov ecx, esi
jmp .param_loop
 
.sync:
; sync setup
lodsb
cmp al, 's'
jne .st
mov [sync], SYNC_SS
jmp .param_loop
.st:
cmp al, 't'
jne .s
mov [sync], SYNC_ST
jmp .param_loop
.s:
mov [sync], SYNC_S
dec esi
jmp .param_loop
 
.invalid:
mov eax, 10
ret
 
; Helper to convert char to number
; Input:
; esi - ptr to char of digit
; Output:
; ebx - number
; Use registers (not restore):
; eax, edx
;
c2n:
xor eax, eax
xor ebx, ebx
xor edx, edx
.loop:
lodsb
test al, al
jz .done
cmp al, ' '
je .done
 
cmp al, ':'
;jne .f0
je .done1
 
;.f0:
cmp al, '+'
je .f00
cmp al, '-'
jne .f01
mov dl,1
.f00:
lodsb
.f01:
sub al, '0'
jb .fail
cmp al, 9
ja .fail
lea ebx, [ebx*4+ebx]
lea ebx, [ebx*2+eax]
jmp .loop
.fail:
xor ebx, ebx
.done:
dec esi
.done1:
cmp dl, 1
jne .ret
neg ebx
.ret:
ret
; Sync worker
; Input:
; eax - in_addr (IPv4)
; Output:
; eax - error_code
; ebx - error_string
; esi - ptr to DateTime
; Use registers (not restore):
; eax, edx, ecx, edx, esi,...
;
;sntp_sync_time:
;mov edx, eax
sntp_query_time:
 
; jmp .test
 
; Create socket
mcall socket, AF_INET4, SOCK_DGRAM, IPPROTO_IP
cmp eax, -1
jne @f
mov eax, 41
mov ebx, str_err4
ret ; Connection error (1)
@@:
;mov [socketnum], eax
;mcall connect, [socketnum], sockaddr1, 18
mov ebp, eax ; Store socket
mcall connect, ebp, sockaddr1, 18
cmp eax, -1
jne @f
mov edx, 42
mov edi, str_err4
jmp .error ; Connection error (2)
;DEBUGF 1, "Socket connected.\n"
@@:
; Query system time
;mcall 3
;mov [SystemTime], eax
mcall send, ebp, sntp_packet, SIZEOF_SNTP_PACKET, 0
cmp eax, -1
jne @f
mov edx, 43
mov edi, str_err4
jmp .error ; Connection error (3)
;DEBUGF 1, "send done.\n"
@@:
; Wait 300 msec.
mcall 5, 30
;do_recv:
mcall recv, ebp, sntp_packet, SIZEOF_SNTP_PACKET, MSG_DONTWAIT
cmp eax, -1
jne @f
mov edx, 50
mov edi, str_err5
jmp .error ; no response
@@:
test eax, eax
jnz @f
mov edx, 44
mov edi, str_err4
jmp .error ; ; Connection error (4)
 
;DEBUGF 1, "recv done.\n"
@@:
; Kiss of death?
cmp [sntp_packet.Stratum], 0
jne @f
mov edx, 60
mov edi, str_err6
jmp .error
@@:
; cmp eax, SIZEOF_SNTP_PACKET
; jne do_recv
;push sntp_packet.ReferenceID
;call [con_write_asciiz]
;invoke con_write_asciiz, str_refid
; TODO: calñ roudtrip
mov eax, [sntp_packet.TransmitTime]
bswap eax
; Bias between epoch
sub eax, 0x83AA7E80
;push eax
;cinvoke con_printf, str_t, eax
;cinvoke con_printf, str_tt, [sntp_packet.TransmitTime], [sntp_packet.TransmitTime + 32]
;pop eax
;.test:
mov ebx, datetime
; mov eax, 7fffffffh ; max timestamp
call timestamp2DateTime
; mov esi, datetime
; Calc time zone (hour only)
; mov eax, [SystemTime]
; clear ecx
; mov cl, al
; clear eax
; mov al, [esi + DateTime.hour]
; push ecx
; b2bcd
; pop ecx
; sub cl, al
;sub bh, [esi + DateTime.min]
; correct minutes
clear eax, ebx, ecx
mov al, [tz_m]
test al, al ; tz_m = 0 ?
jz .tz_h
mov bl, [tz_h] ; tz_m < 0 ? This not work with tz_h = 0!
test bl,bl
jns @f
sub [esi + DateTime.min], al
cmp [esi + DateTime.min], 0
jge @f
mov al, 60
mov cl, -1
add [esi + DateTime.min], al;bh
;mov [TimeZone], cx
;@@:
@@:
.tz_h:
; correct hour
cmp [sync], SYNC_SS
; if -ss ignore timezone for hour
je .tz_done
clear eax
mov al, [tz_h]
add al, cl
add [esi + DateTime.hour], al ;3 ; MSK = GMT +3
; Correct day & hour if prev/new day
; hour < 0 ?
mov al,[esi + DateTime.hour]
test al, al
jns @f
add [esi + DateTime.hour],24
dec [esi + DateTime.day]
jmp .tz_done
@@:
; hour >= 24 ?
cmp [esi + DateTime.hour], 24
jl .tz_done
inc [esi + DateTime.day]
mov bl, [esi + DateTime.hour]
sub ebx, 24
mov [esi + DateTime.hour],bl
.tz_done:
 
; {{
; Removed block 1.1
;}}
;clear eax, ebx, ecx
;mov al, '+'
;mov ax, [TimeZone]
;mov bl, al
;mov cl, ah
; {{
; Removed block 1.2
;}}
; FIXED: do sync before display!!!
; It's need to do sync fast ASAP
; Take out any printf from sntp_query_time!
; sync > 0 ?
cmp [sync], 0
je .nosync
;{{
; Removed block 2
; FIXME: Go it from sntp_query_time!
;}}
; Convert time to BCD
clear eax, edx
mov al, [esi + DateTime.sec]
b2bcd
mov ecx, eax
shl ecx, 16
mov al, [esi + DateTime.min]
b2bcd
mov ch, al
mov al, [esi + DateTime.hour]
b2bcd
mov cl, al
; Display BCD time
;cinvoke con_printf, str_t, ecx
; mov ecx, [ebx + DateTime.date]
; push ecx
; push str_d
; call [con_printf]
; add esp, 2*4
 
; Set time
mov eax, 22
mov ebx, 0
int 0x40
; ? error
cmp [sync], SYNC_S
jne .nosync
; Convert date to BCD
mov esi, datetime
clear eax, edx
mov al, [esi + DateTime.day]
b2bcd
mov ecx, eax
shl ecx, 16
mov al, [esi + DateTime.month]
b2bcd
mov ch, al
mov ax, [esi + DateTime.year]
sub ax, 2000
b2bcd
mov cl, al
; Display BCD date
;cinvoke con_printf, str_d, ecx
; Set date
mov eax, 22
mov ebx, 1
; mov ecx, [edx + DateTime.date]
int 0x40
; ? error
.nosync:
; Check LI field
; mov al, byte [sntp_packet]
; bt al,0
; jnc @f
; mov edx,1
;@@:
; bt ax, 1
; jnc @f
; mov edx,2
; jmp .error
@@:
clear edx, edi ; no error
.error:
; Close socket
mcall close, ebp
 
; Ignore error from close,
; but it write result to eax & ebx
; so we need wtire result here
mov eax,edx
mov ebx,edi
ret
;fail:
; cinvoke con_printf, str_err, eax, ebx
; ret
 
; data
str_title db 'SNTP client',0
str_help db 'sntp host [-tz [-[+]]hh[:ss]] [-s]|[-st]|[-ss]',10
;str_help db 'sntp host [-c][[-p] [-tz [-[+]]hh[:ss]] [-s]|[-st]|[-ss]]',10
;'-ñ Load config from Time.ini',10
;'-p Set port, default is 123',10
db 'host Name or IP address of NTP/SNTP server',10
db 'Options:',10
db '-tz Set time zone, default is GMT',10
;'-qm Try query time from master SNTP server (if stratum > 1), defautl is disabled',10
db 10
db 'Syncronization, default is disabled',10
db '-s System date and time',10
db '-st System time (hours, mitutes and seconds) only',10
db '-ss Save current hour (syncronize minutes and seconds only)',10
db 10
;-dr Display SNTP server reserence ID (if stratum = 0)
;-dt Display accurate time from SNTP request, default is enabled
;-da Display all SNTP server information
db 'Examples:',10
db 'sntp pool.ntp.org -tz 1 -s',10
db 'sntp 88.147.254.227 -tz 1 -ss',10,0
;str_badoption db 'Warning: unknown option %s, ignored',10, 0
;str_query db 'Query - %s [%s] :%i',10,0
str_query db 'Query: %s',0
str_ip db ' [%s]',0
str_port db ' :%i',10,0
;str_tt db 'Transmit time - 0x%X.%X',10,0
;str_t db 'Timestamp - %i',10,0
str_dt db 'Date & time: %i.%02i.%02i %i:%02i:%02i GMT',0 ; ' UTC %%i:%02i'
str_tz db ' +%i:%02i',10,0 ;Time zone: GMT +%i:%02i
;str_d db 'BCD date - 0x%08X',10,0
;str_t db 'BCD time - 0x%08X',10,0
str_err db 'Error: #%i, %s => %s',0
str_err10 db 'Bad command line, type ',39,'sntp',39,' for help.',10,0
str_err11 db 'Incorrect time zone! Visit https://www.timeanddate.com/time/map for details.',10,0
;str_err2 db 'Can',39,'t run @NOTIFY!',10,0
str_err3 db 'Host name not resolved!',0 ; @notify => ntp.example.com \nError 30: Host name not resolved!
str_err4 db 'Connection failed!',0 ; @notify => ntp.example.com:100 \nError 4x: Connection failed!
str_err5 db 'Host not responce!',0 ; @notify => ntp.example.com:100 \nError 50: Host not responce!
str_err6 db 'Received Kiss-o',39,'-Death (KoD) packet, repeat later or try another server!',0
 
;str_err7 db 'Server clock not syncronizing!',0 Not needed, see RFC 4330 about LI field
str_err7 db 'Clock syncronizing failed!',10,0
;str_err62 db 'Date syncronizing failed',0
str_warn db 'Warning: Last minute will have %i seconds!',10,0
str_s db 'Date & time',0
str_st db 'Time',0
str_ss db 'Mitutes & seconds',0
str_sync db '%s syncronized',10,0
datetime DateTime ?
;SystemTime dd ?
;TimeZone dw 180 ; GMT + 3
;Flags db 0
 
port dw 123
tz_h db 0
tz_m db 0
sync db 0
;notify db 0
 
sockaddr1:
dw AF_INET4
.port dw 0x7b00 ; 123 in network order (big endian)
.ip dd 0
rb 10
 
SIZEOF_SNTP_PACKET = 48
sntp_packet db 0x23 ; Li = 0 Vn = 4 Mode = 3 (client) FIX: Why 0x0b?
.Stratum db 0
.Pool db 0
.Precision db 0
.RootDelay dd 0
.RootDispersion dd 0
.ReferenceID dd 0
.ReferenceTime dq 0
.OriginateTime dq 0
.ReceiveTime dq 0
.TransmitTime dd 0
 
; import
align 4
@IMPORT:
 
library network, 'network.obj', console, 'console.obj'
;library network, 'network.obj', console, 'console.obj'
import network, \
inet_addr, 'inet_addr', \
getaddrinfo, 'getaddrinfo', \
freeaddrinfo, 'freeaddrinfo', \
inet_ntoa, 'inet_ntoa'
 
; con_start, 'START', \
import console, \
con_init, 'con_init', \
con_write_asciiz, 'con_write_asciiz', \
con_printf, 'con_printf', \
con_exit, 'con_exit', \
con_gets, 'con_gets',\
con_cls, 'con_cls',\
con_getch2, 'con_getch2',\
con_set_cursor_pos, 'con_set_cursor_pos',\
con_get_flags, 'con_get_flags'
;, \
;con_set_flags, 'con_set_flags'
 
;socketnum dd ?
 
I_END:
rb 4096
align 16
;buffer_ptr: rb BUFFERSIZE
STACKTOP:
 
MEM:
params rb 1024
;buffer_ptr: rb BUFFERSIZE
IM_END: