;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
; RDOS operating system
; Copyright (C) 1988-2000, Leif Ekblad
;
; This program is free software; you can redistribute it and/or modify
; it under the terms of the GNU General Public License as published by
; the Free Software Foundation; either version 2 of the License, or
; (at your option) any later version. The only exception to this rule
; is for commercial usage in embedded systems. For information on
; usage in commercial embedded systems, contact embedded@rdos.net
;
; This program is distributed in the hope that it will be useful,
; but WITHOUT ANY WARRANTY; without even the implied warranty of
; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
; GNU General Public License for more details.
;
; You should have received a copy of the GNU General Public License
; along with this program; if not, write to the Free Software
; Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
;
; The author of this program may be contacted at leif@rdos.net
;
; NTP.ASM
; NTP protocol
;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

INCLUDE protseg.def
INCLUDE ..\driver.def
INCLUDE ..\user.def
INCLUDE ..\os.def
INCLUDE ..\user.inc
INCLUDE ..\os.inc
INCLUDE exec.def
INCLUDE system.inc
INCLUDE ntp.inc

    extrn WriteIpEnv:near

data    SEGMENT byte public 'DATA'

ntp_req     ntp_header <>

data    ENDS

code    SEGMENT byte public 'CODE'

.386p
    
    assume cs:code

        
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;
;       Name:           GetTimestamp
;
;       Purpose:        Get current time in time-stamp format
;
;       Returns:        EDX:EAX     Time-stamp time
;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

GetTimestamp    Proc near
    push ecx
;
    GetSystemTime
    sub edx,1                   ; UTC + 1 time-zone
    sub edx,0FE2440h        ; rel to 1/1 1900 (no leap in 1900)
    mov ecx,edx
    mov edx,3600
    mul edx
    push eax    
    push edx
    mov eax,3600
    mul ecx
    pop ecx
    add eax,ecx
    mov edx,eax
    pop eax
;
    pop ecx 
    ret
GetTimestamp    Endp

        
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;
;       Name:           SyncTime
;
;       Purpose:        Sync time with NTP
;
;       Parameters:         IP address of NTP-server
;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

sync_time_name DB 'Synchronize Time',0

sync_time       Proc far
    push ds
    push es
    push ax
    push bx
    push cx
    push edx
    push si
    push edi
;
    mov ax,SEG data
    mov ds,ax
    mov es,ax
    mov cx,SIZE ntp_header
    mov di,OFFSET ntp_req
    xor al,al
    rep stosb
    mov di,OFFSET ntp_req
    mov ds:ntp_req.ntp_flag,1Bh
    mov ds:ntp_req.ntp_strat,3
    mov ds:ntp_req.ntp_poll,100
    mov ds:ntp_req.ntp_precision,-20
;
    push edx
    call GetTimestamp
    xchg al,ah
    ror eax,16
    xchg al,ah
    mov ds:ntp_req.ntp_orginate_timestamp+4,eax
    xchg dl,dh
    ror edx,16
    xchg dl,dh
    mov ds:ntp_req.ntp_orginate_timestamp,edx           ; t1
    pop edx
;
    mov edi,OFFSET ntp_req
    mov ecx,SIZE ntp_header
    mov bx,123
    mov eax,2000
    QueryUdp
    jc sync_time_done
;
    call GetTimestamp                                   ; t4
    mov ebx,es:[di].ntp_transmit_timestamp+4        ; t3
    xchg bl,bh
    ror ebx,16
    xchg bl,bh
    mov ecx,es:[di].ntp_transmit_timestamp
    xchg cl,ch
    ror ecx,16
    xchg cl,ch
    sub eax,ebx                                         ; t4 - t3
    sbb edx,ecx
    push eax
    push edx    
;
    mov ebx,ds:ntp_req.ntp_orginate_timestamp+4         ; t1
    xchg bl,bh
    ror ebx,16
    xchg bl,bh
    mov ecx,ds:ntp_req.ntp_orginate_timestamp
    xchg cl,ch
    ror ecx,16
    xchg cl,ch      
    mov eax,es:[di].ntp_receive_timestamp+4         ; t2
    xchg al,ah
    ror eax,16
    xchg al,ah
    mov edx,es:[di].ntp_receive_timestamp
    xchg dl,dh
    ror edx,16
    xchg dl,dh
    sub eax,ebx
    sbb edx,ecx                                         ; t2 - t1
;
    pop ecx
    pop ebx                                         ; t4 - t3
    sub eax,ebx
    sub edx,ecx                                         ; t2 - t1 - (t4 - t3)
    mov ebx,eax
    mov ecx,edx
    test ecx,80000000h
    pushf
    jz diff_pos
diff_neg:
    not ebx
    not ecx
diff_pos:
    mov eax,ecx
    xor edx,edx
    mov edi,3600*2
    div edi
    push eax
    mov eax,ebx
    div edi
    pop edx
    popf
    jz sync_diff_found
    not eax
    not edx
sync_diff_found:
    FreeMem
    UpdateTime
    UpdateRtc
    clc
sync_time_done:
    pop edi
    pop si
    pop edx
    pop cx
    pop bx
    pop ax
    pop es
    pop ds
    retf32
sync_time       Endp

        
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;
;       Name:           define_ntp_server
;
;       Purpose:        Define a NTP server from DHCP
;
;       Parameters:     ECX          Size of msg
;                       ES:EDI       NTP server
;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

ntp_name    DB 'NTP', 0

define_ntp_server       Proc far
    push ds
    push es
    push ax
    push edx
    push di
;
    mov edx,es:[di]
;
    mov ax,cs
    mov es,ax
    mov di,OFFSET ntp_name
    call WriteIpEnv
;
    pop di
    pop edx
    pop ax
    pop es
    pop ds
    retf32
define_ntp_server       Endp


;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;
;
;           NAME:           init_task_ntp
;
;           DESCRIPTION:    Init NTP driver, tasking part
;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

    public init_task_ntp

init_task_ntp    PROC near
    mov ax,cs
    mov ds,ax
    mov es,ax
;
    mov esi,OFFSET sync_time
    mov edi,OFFSET sync_time_name
    xor dx,dx
    mov ax,sync_time_nr
    RegisterBimodalUserGate
;
    mov al,42
    mov edi,OFFSET define_ntp_server
    AddDhcpOption
;
    ret
init_task_ntp    ENDP

code    ENDS

    END