WebSVN – Kolibri OS – Diff – /programs/network/autodhcp/trunk/autodhcp.asm

+;
 ; with thanks to authors of DHCP client for menuetos: Mike Hibbet
+;
 ; by HidnPlayr & Derpenguin
-TIMEOUT equ 60 ; in seconds
-BUFFER	equ 1024
-__DEBUG__ equ 1
-__DEBUG_LEVEL__ equ 1; 1 = all, 2 = errors
 use32
 	       org    0x0
-	       dd     IM_END		      ; size of image
+	       dd     IM_END		    ; size of image
-	       dd     I_END		      ; memory for app
+	       dd     I_END		    ; memory for app
-	       dd     I_END		      ; esp
+	       dd     I_END		    ; esp
-	       dd     0x0 , 0x0 	      ; I_Param , I_Icon
+	       dd     0x0 , 0x0 	    ; I_Param , I_Icon
-;include 'macros.inc'
-include 'eth.inc'
-include 'debug-fdo.inc'
- ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
-; CONFIGURATION FOR LINK-LOCAL                                                         ;
+; CONFIGURATION
- ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
-				       ;                                               ;
-PROBE_WAIT	    equ 1	       ; second  (initial random delay)                ;
-PROBE_MIN	    equ 1	       ; second  (minimum delay till repeated probe)   ;
-PROBE_MAX	    equ 2	       ; seconds (maximum delay till repeated probe)   ;
-PROBE_NUM	    equ 3	       ;         (number of probe packets)             ;
-				       ;                                               ;
-ANNOUNCE_NUM	    equ 2	       ;         (number of announcement packets)      ;
-ANNOUNCE_INTERVAL   equ 2	       ; seconds (time between announcement packets)   ;
-ANNOUNCE_WAIT	    equ 2	       ; seconds (delay before announcing)             ;
-				       ;                                               ;
-MAX_CONFLICTS	    equ 10	       ;         (max conflicts before rate limiting)  ;
-				       ;                                               ;
-RATE_LIMIT_INTERVAL equ 60	       ; seconds (delay between successive attempts)   ;
+TIMEOUT 	    equ 60		    ; in seconds
-				       ;                                               ;
+BUFFER		    equ 1024		    ; in bytes
+__DEBUG__	    equ 1		    ; enable/disable
+__DEBUG_LEVEL__     equ 1		    ; 1 = all, 2 = errors
-DEFEND_INTERVAL     equ 10	       ; seconds (min. wait between defensive ARPs)    ;
+; CONFIGURATION FOR LINK-LOCAL
-				       ;                                               ;
- ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
+PROBE_WAIT	    equ 1		    ; second  (initial random delay)
+PROBE_MIN	    equ 1		    ; second  (minimum delay till repeated probe)
+PROBE_MAX	    equ 2		    ; seconds (maximum delay till repeated probe)
+PROBE_NUM	    equ 3		    ;         (number of probe packets)
+ANNOUNCE_NUM	    equ 2		    ;         (number of announcement packets)
+ANNOUNCE_INTERVAL   equ 2		    ; seconds (time between announcement packets)
+ANNOUNCE_WAIT	    equ 2		    ; seconds (delay before announcing)
-START:				       ; start of execution
+MAX_CONFLICTS	    equ 10		    ;         (max conflicts before rate limiting)
-    mov     eax,40		       ; Report events
+RATE_LIMIT_INTERVAL equ 60		    ; seconds (delay between successive attempts)
+DEFEND_INTERVAL     equ 10		    ; seconds (min. wait between defensive ARPs)
+include 'macros.inc'
+include 'eth.inc'
+include 'debug-fdo.inc'
+include 'dhcp.inc'
+include 'events.inc'
-    mov     ebx,10000000b	       ; Only Stack
+include 'common.inc'
     int     0x40
-    mov     eax,52		       ; first, enable the stack (packet driver)
+START:					    ; start of execution
-    mov     ebx,2
-    mov     ecx,0x00000383
+    set_event_mask	event_network
-    int     0x40
+    eth.set_network_drv 0x00000383
-    DEBUGF  1,"DHCP: Stack Initialized.\n"
+    DEBUGF  1,"Stack Initialized.\n"
-    eth.status eax		       ; Read the Stack status
+    eth.status eax			    ; Read the Stack status
-    test    eax,eax		       ; if eax is zero, no driver was found
+    test    eax,eax			    ; if eax is zero, no driver was found
     jnz     @f
-    DEBUGF  1,"DHCP: No Card detected\n"
-    jmp     close
+    DEBUGF  1,"No Card detected\n"
-   @@:
-    DEBUGF  1,"DHCP: Detected card: %x\n",eax
-   @@:
+    jmp     close
     eth.check_cable eax
-    test    al,al
+   @@:
-    jnz     @f
+    DEBUGF  1,"Detected card: %x\n",eax
-    DEBUGF  1,"DHCP: Ethernet Cable not connected\n"
+   @@:
-    mov     eax,5
-    mov     ebx,500		       ; loop until cable is connected (check every 5 sec)
-    int     0x40
-    jmp     @r
-   @@:
-    DEBUGF  1,"DHCP: Ethernet Cable status: %d\n",al
-    eth.read_mac MAC
+    eth.check_cable eax
-    DEBUGF  1,"DHCP: MAC address: %x-%x-%x-%x-%x-%x\n",[MAC]:2,[MAC+1]:2,[MAC+2]:2,[MAC+3]:2,[MAC+4]:2,[MAC+5]:2
+    test    al,al
     jnz     @f
-;    jmp     apipa   ; comment this out if you want to skip DHCP and continue with link-local
+    DEBUGF  1,"Ethernet Cable not connected\n"
     wait 500				    ; loop until cable is connected (check every 5 sec)
-;***************************************************************************
+    jmp     @r
-;  DHCP rubish starts here
+   @@:
     DEBUGF  1,"Ethernet Cable status: %d\n",al
 ;***************************************************************************
     eth.read_mac MAC
+    DEBUGF  1,"MAC address: %x-%x-%x-%x-%x-%x\n",[MAC]:2,[MAC+1]:2,[MAC+2]:2,[MAC+3]:2,[MAC+4]:2,[MAC+5]:2
-    eth.check_port 68,eax	       ; Check if port 68 is available
     cmp     eax,1
-    je	    @f
+    eth.check_port 68,eax		    ; Check if port 68 is available
-    DEBUGF  1,"DHCP: Port 68 is already in use.\n"
-    jmp     close
-   @@:
-    eth.open_udp 68,67,-1,[socketNum]  ; open socket (local,remote,ip,socket)
+    cmp     eax,1
-    DEBUGF  1,"DHCP: Socket opened: %d\n",eax
-				       ; Setup the first msg we will send
-    mov     byte [dhcpMsgType], 0x01   ; DHCP discover
-    mov     dword [dhcpLease], esi     ; esi is still -1 (-1 = forever)
+    je	    @f
-    mov     eax,26
+    DEBUGF  1,"Port 68 is already in use!\n"
-    mov     ebx,9
+    jmp     close
     int     0x40
     mov     [edx+2], byte 0x06		    ; Ethernet h/w len
     mov     [edx+4], dword 0x11223344	    ; xid
     mov     eax,[currTime]
     mov     [edx+8], eax		    ; secs, our uptime
     mov     [edx+10], byte 0x80 	    ; broadcast flag set
     mov     eax, dword [MAC]		    ; first 4 bytes of MAC
     mov     [edx+28],dword eax
     mov     ax, word [MAC+4]		    ; last 2 bytes of MAC
     mov     [edx+32],word ax
     mov     [edx+236], dword 0x63538263     ; magic number
     mov     [edx+240], word 0x0135	    ; option DHCP msg type
     mov     al, [dhcpMsgType]
     mov     [edx+240+2], al
     mov     [edx+240+3], word 0x0433	    ; option Lease time = infinity
     mov     eax, [dhcpLease]
     mov     [edx+240+5], eax
     mov     [edx+240+9], word 0x0432	    ; option requested IP address
     mov     eax, [dhcpClientIP]
     mov     [edx+240+11], eax
     mov     [edx+240+15], word 0x0437	    ; option request list
     mov     [edx+240+17], dword 0x0f060301
     cmp     [dhcpMsgType], byte 0x01	    ; Check which msg we are sending
-    jne     br001
+    jne     request_options
     mov     [edx+240+21], byte 0xff	    ; "Discover" options
-    mov     [dhcpMsgLen], dword 262	    ; end of options marker
-    jmp     ctr000
+    mov     [dhcpMsgLen], dword 262	    ; end of options marker
     jmp     send_request
 br001:					    ; "Request" options
 request_options:
     mov     [edx+240+21], word 0x0436	    ; server IP
     mov     eax, [dhcpServerIP]
-    mov     [edx+240+23], eax
     mov     [edx+240+23], eax
     mov     [edx+240+27], byte 0xff	    ; end of options marker
     mov     [edx+240+27], byte 0xff	    ; end of options marker
     int     0x40
     eth.poll [socketNum]
     test    eax,eax
-    jnz     ctr002
+    jnz     read_data
-    DEBUGF  2,"DHCP: Timeout!\n"
+    DEBUGF  2,"Timeout!\n"
     eth.close_udp [socketNum]
     jmp    apipa			    ; no server found, lets try zeroconf
-    call     parseResponse
+    call    parseResponse
     cmp     [dhcpMsgType], byte 0x05	    ; Was the response an ACK? It should be
-    jne     apipa			    ; NO - so we do zeroconf
-close:
-    DEBUGF  1,"DHCP: Exiting\n"
+    jne     apipa			    ; NO - so we do zeroconf
-    mov     eax,-1			    ; at last, exit
     int     0x40
     jmp     close
 ;***************************************************************************
 ;       dhcpDNSIP, dhcpSubnet
 ;      The message is stored in dhcpMsg
+;
 ;***************************************************************************
 parseResponse:
-    DEBUGF  1,"DHCP: Data received, parsing response\n"
+    DEBUGF  1,"Data received, parsing response\n"
     mov     edx, dhcpMsg
     pusha
     eth.set_IP [edx+16]
     mov     eax,[edx]
     mov     [dhcpClientIP],eax
-    DEBUGF  1,"DHCP: Client: %u.%u.%u.%u\n",[edx+16]:1,[edx+17]:1,[edx+18]:1,[edx+19]:1
+    DEBUGF  1,"Client: %u.%u.%u.%u\n",[edx+16]:1,[edx+17]:1,[edx+18]:1,[edx+19]:1
     popa
+    add     edx, 240			    ; Point to first option
+    xor     ecx, ecx
-    add     edx, 240	    ; Point to first option
+next_option:
     add     edx, ecx
 pr001:
     mov     al, [edx]
-    cmp     al, 0xff	    ; End of options?
+    cmp     al, 0xff			    ; End of options?
     je	    pr_exit
-    cmp     al, 53	    ; Msg type is a single byte option
+    cmp     al, dhcp_msg_type		    ; Msg type is a single byte option
-    jne     pr002
+    jne     @f
     mov     al, [edx+2]
     mov     [dhcpMsgType], al
     add     edx, 3
-    jmp     pr001	    ; Get next option
+    jmp     pr001			    ; Get next option
-pr002:
+@@:
     inc     edx
     movzx   ecx, byte [edx]
-    inc     edx 	    ; point to data
+    inc     edx 			    ; point to data
-    cmp     al, 54	    ; server id
+    cmp     al, dhcp_dhcp_server_id	    ; server ip
-    jne     pr0021
+    jne     @f
-    mov     eax, [edx]	    ; All options are 4 bytes, so get it
+    mov     eax, [edx]
-    mov     [dhcpServerIP], eax
-    DEBUGF  1,"DHCP: Server: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
+    mov     [dhcpServerIP], eax
-    jmp     pr003
+    DEBUGF  1,"Server: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
     jmp     next_option
-pr0021:
-    cmp     al, 51	    ; lease
-    jne     pr0022
-    pusha
-    DEBUGF  1,"DHCP: lease: "
-    mov eax,[edx]
-    bswap eax
+@@:
-    mov  [dhcpLease],eax
+    cmp     al, dhcp_address_time
-    cmp  dword[edx],-1	    ; i really don't know, how to test it
+    jne     @f
-    jne  no_lease_forever
-    DEBUGF  1,"forever\n"
+    pusha
-    jmp  @f
+    mov     eax,[edx]
-   no_lease_forever:
+    bswap   eax
-    DEBUGF  1,"%d\n",eax
+    mov     [dhcpLease],eax
-   @@:
+    DEBUGF  1,"lease: %d\n",eax
     popa
-    jmp     pr003
+    jmp     next_option
-pr0022:
+@@:
-    cmp     al, 1	    ; subnet mask
+    cmp     al, dhcp_subnet_mask
-    jne     pr0023
+    jne     @f
+    pusha
+    eth.set_SUBNET [edx]
-    pusha
+    DEBUGF  1,"Subnet: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
-    eth.set_SUBNET [edx]
+    popa
-    DEBUGF  1,"DHCP: Subnet: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
-    popa
+    jmp     next_option
-    jmp     pr003
+@@:
     cmp     al, dhcp_router
-pr0023:
-    cmp     al, 3	    ; gateway ip
-    jne     pr0024
+    jne     @f
     pusha
-    eth.set_GATEWAY [edx]
+    pusha
-    DEBUGF  1,"DHCP: Gateway: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
+    eth.set_GATEWAY [edx]
-    popa
+    DEBUGF  1,"Gateway: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
     popa
-pr0024:
+    jmp     next_option
     cmp     al, 6	    ; dns ip
     jne     pr003
+@@:
-    pusha
+    cmp     al, dhcp_domain_server
-    eth.set_DNS [edx]
+    jne     next_option
     DEBUGF  1,"DHCP: DNS: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
-    popa
+    pusha
-pr003:
+    eth.set_DNS [edx]
-    add     edx, ecx
+    DEBUGF  1,"DNS: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
-    jmp     pr001
+    popa
     jle   @f
     sub   al,(PROBE_MAX-PROBE_MIN)*100
    @@:
     movzx ebx,al
-    DEBUGF  1,"ZeroConf: Waiting %u0ms\n",ebx
+    DEBUGF  1,"Waiting %u0ms\n",ebx
-    mov   eax,5
-    int   0x40
+    wait  ebx
-    DEBUGF  1,"ZeroConf: Sending Probe\n"
+    DEBUGF  1,"Sending Probe\n"
-;    eth.ARP_PROBE MAC2
+;    eth.ARP_PROBE MAC
     inc   esi
     cmp   esi,PROBE_NUM
     jl	  probe_loop
-; now we wait further ANNOUNCE_WAIT seconds and send ANNOUNCE_NUM ARP announces. If any other host has assingnd
-; IP within this time, we should create another adress, that have to be done later
-    DEBUGF  1,"ZeroConf: Waiting %us\n",ANNOUNCE_WAIT
+; now we wait further ANNOUNCE_WAIT seconds and send ANNOUNCE_NUM ARP announces. If any other host has assingned
-    mov   eax,5
+; IP within this time, we should create another adress, that have to be done later
-    mov   ebx,ANNOUNCE_WAIT*100
-    int   0x40
+    DEBUGF  1,"Waiting %us\n",ANNOUNCE_WAIT
     wait ANNOUNCE_WAIT*100
     xor   esi,esi
    announce_loop:
-    DEBUGF  1,"ZeroConf: Sending Announce\n"
-;    eth.ARP_ANNOUNCE MAC2
+    DEBUGF  1,"Sending Announce\n"
-    inc   esi
-    cmp   esi,ANNOUNCE_NUM
+;    eth.ARP_ANNOUNCE MAC
     je	  @f
+    inc   esi
+    cmp   esi,ANNOUNCE_NUM
+    je	  @f
+    DEBUGF  1,"Waiting %us\n",ANNOUNCE_INTERVAL
+    wait  ANNOUNCE_INTERVAL*100
-    DEBUGF  1,"ZeroConf: Waiting %us\n",ANNOUNCE_INTERVAL
+    jmp   announce_loop
+   @@:
-    mov   eax,5
+    ; we should, instead of closing, detect ARP conflicts and detect if cable keeps connected ;)
     mov   ebx,ANNOUNCE_INTERVAL*100
-    int   0x40
+close:
     DEBUGF  1,"Exiting\n"
-    jmp   announce_loop
+    exit				    ; at last, exit
    @@:
     jmp   close   ; we should, instead of closing, detect ARP conflicts and detect if cable keeps connected ;)
 random:
-random:
-    mov   eax,[generator]
-    add   eax,-43ab45b5h
+    mov   eax,[generator]
-    ror   eax,1
+    add   eax,-43ab45b5h
-    bswap eax
+    ror   eax,1
-    xor   eax,dword[MAC]
+    bswap eax
-    ror   eax,1
+    xor   eax,dword[MAC]

Subversion Repositories Kolibri OS

(root)/programs/network/autodhcp/trunk/autodhcp.asm – Rev 337 → 339