WebSVN – Kolibri OS – Diff – /kernel/branches/net/applications/zeroconf/zeroconf.asm

+;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
+;;                                                                 ;;
+;; Copyright (C) KolibriOS team 2010-2013. All rights reserved.    ;;
+;; Distributed under terms of the GNU General Public License       ;;
+;;                                                                 ;;
-; Zero-config
+;;  zeroconfig.asm - Zeroconfig service for KolibriOS              ;;
-; v 1.4
-;
+;;                                                                 ;;
+;;  Written by hidnplayr@kolibrios.org                             ;;
+;;    Some code contributed by Derpenguin                          ;;
+;;                                                                 ;;
-; DHCP code is based on that by Mike Hibbet (DHCP client for menuetos)
+;;  DHCP code is based on that by Mike Hibbet                      ;;
-;
-; Written by HidnPlayr & Derpenguin
+;       (DHCP client for menuetos)                                 ;;
+;;                                                                 ;;
+;;          GNU GENERAL PUBLIC LICENSE                             ;;
+;;             Version 2, June 1991                                ;;
+;;                                                                 ;;
+;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 format binary as ""
                dd     (I_END+0x100)         ; esp
                dd     0x0 , path            ; I_Param , I_Icon
-; CONFIGURATION
 ; CONFIGURATION
 TIMEOUT             equ 60                  ; in seconds
 BUFFER              equ 1024                ; in bytes
 START:                                      ; start of execution
 START:
         mcall   40, 1 shl 7 ; network event
         mcall   40, EVM_STACK   ; network event
         DEBUGF  1,">Zero-config service:\n"
-        mcall   76, API_ETH + 4
+        DEBUGF  1,">Zero-config service loaded\n"
-        cmp     eax, -1
+        mcall   76, API_ETH + 4 ; get MAC of ethernet interface 0
-        je      exit
+        cmp     eax, -1
         je      exit
-        mov     word[MAC], bx
-        mov     dword[MAC+2], eax
+        mov     word[MAC], bx
         mov     dword[MAC+2], eax
-        DEBUGF  1,"->MAC: %x-%x-%x-%x-%x-%x\n",[MAC]:2,[MAC+1]:2,[MAC+2]:2,[MAC+3]:2,[MAC+4]:2,[MAC+5]:2
         DEBUGF  1,"->MAC: %x-%x-%x-%x-%x-%x\n",[MAC]:2,[MAC+1]:2,[MAC+2]:2,[MAC+3]:2,[MAC+4]:2,[MAC+5]:2
         cld
-        mov     edi, path      ; Calculate the length of zero-terminated string
+        mov     edi, path       ; Calculate the length of zero-terminated string
-        xor     al , al
+        xor     al, al
-        mov     ecx, 1024
-        repnz   scas byte[es:edi]
+        mov     ecx, 1024
-        dec     edi
+        repne   scasb
-        mov     esi, filename
-        movsd
+        dec     edi
         movsb
         mov     esi, filename   ; append with .ini
-        DEBUGF  1,"->path to ini: %s\n", path
+        movsd
         movsb
-        mcall  -1
+        mcall   -1
-skip_ini:
+try_dhcp:
-        DEBUGF  1,"->Skip ini\n"
+        DEBUGF  1,"->Trying DHCP\n"
-        mcall 75, 0, AF_INET4, SOCK_DGRAM, 0      ; open socket (parameters: domain, type, reserved)
+        mcall   75, 0, AF_INET4, SOCK_DGRAM, 0          ; open socket (parameters: domain, type, reserved)
-        cmp   eax, -1
+        cmp     eax, -1
-        cmp   eax, -1
+        cmp     eax, -1
-        je    error
+        je      error
         DEBUGF  1,"->Connected to 255.255.255.255 on port 67\n"
-        mov     byte [dhcpMsgType], 0x01        ; DHCP discover
+        mov     [dhcpMsgType], 0x01                     ; DHCP discover
-        mov     dword [dhcpLease], esi          ; esi is still -1 (-1 = forever)
+        mov     [dhcpLease], esi                        ; esi is still -1 (-1 = forever)
-        mcall   26, 9
+        mcall   26, 9                                   ; Get system time
-        imul    eax,100
+        imul    eax, 100
-        mov     [currTime],eax
+        mov     [currTime], eax
             ;;; todo: skip this bullcrap
         mov     edi, eax
         mov     ecx, BUFFER
-        xor     eax, eax
-        cld
+        xor     eax, eax
         rep     stosb
         jne     request_options
         mov     [edx+240+21], byte 0xff         ; "Discover" options
         mov     [dhcpMsgLen], dword 262         ; end of options marker
-        jmp     send_request
+        jmp     send_dhcpmsg
 request_options:
         mov     [edx+240+21], word 0x0436       ; server IP
         mov     eax, [dhcpServerIP]
         mov     [edx+240+23], eax
         mov     [edx+240+27], byte 0xff         ; end of options marker
-    ;  1.2) send a request packet
+;  1.2) send a request packet
-    ; If the response is to a dhcp request, then:
+; If the response is to a dhcp request, then:
-    ;  1) If the response is DHCP ACK then
+;  1) If the response is DHCP ACK then
-    ;  1.1) extract the DNS & subnet fields. Set them in the stack
+;  1.1) extract the DNS & subnet fields. Set them in the stack
-    cmp     [dhcpMsgType], byte 0x01        ; did we send a discover?
+        cmp     [dhcpMsgType], 0x01             ; did we send a discover?
+        je      discover
-    je      discover
-    cmp     [dhcpMsgType], byte 0x03        ; did we send a request?
+        cmp     [dhcpMsgType], 0x03             ; did we send a request?
-    je      request
+        je      request
         call    dhcp_end                        ; we should never reach here ;)
-        jmp     exit                           ; really unknown, what we did
+        jmp     exit
+discover:
+        call    parse_response
-discover:
+        cmp     [dhcpMsgType], 0x02             ; Was the response an offer?
+        je      send_request
-        call    parseResponse
+        call    dhcp_end
+        jmp     link_local
         cmp     [dhcpMsgType], byte 0x02        ; Was the response an offer?
-        jne     apipa                           ; NO - so we do zeroconf
+send_request:
+        mov     [dhcpMsgType], 0x03             ; make it a request
-        mov     [dhcpMsgType], byte 0x03        ; DHCP request
+        jmp     build_request
-        jmp     buildRequest
 request:
-request:
+        call    parse_response
-        call    parseResponse
+        call    dhcp_end
-        cmp     [dhcpMsgType], byte 0x05        ; Was the response an ACK? It should be
+        cmp     [dhcpMsgType], 0x05             ; Was the response an ACK? It should be
+        jne     link_local                      ; NO - so we do link-local
+        mcall   76, API_IPv4 + 3, [dhcp.ip]             ; ip
+        mcall   76, API_IPv4 + 5, [dhcp.dns]            ; dns
+        mcall   76, API_IPv4 + 7, [dhcp.subnet]         ; subnet
+        mcall   76, API_IPv4 + 9, [dhcp.gateway]        ; gateway
         jne     apipa                           ; NO - so we do zeroconf
         jmp     exit
         mcall 76, API_IPv4 + 3, [dhcp.ip]       ; ip
-        mcall 76, API_IPv4 + 5, [dhcp.dns]      ; dns
+dhcp_end:
-        mcall 76, API_IPv4 + 7, [dhcp.subnet]   ; subnet
+        mcall   close, [socketNum]
 ;       dhcpMsgType,dhcpLease,dhcpClientIP,dhcpServerIP,
 ;       dhcpDNSIP, dhcpSubnet
 ;      The message is stored in dhcpMsg
+;
 ;***************************************************************************
-parseResponse:
+parse_response:
-    DEBUGF  1,"Data received, parsing response\n"
+        DEBUGF  1,"Data received, parsing response\n"
-    mov     edx, [dhcpMsg]
+        mov     edx, [dhcpMsg]
-    push    dword [edx+16]
+        push    dword [edx+16]
-    pop     [dhcp.ip]
+        pop     [dhcp.ip]
+        DEBUGF  1,"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
+; TODO: check if there really are options
-    add     edx, 240                        ; Point to first option
+        mov     al, 240                         ; Point to first option
-    xor     ecx, ecx
+        movzx   ecx, al
-next_option:
-    add     edx, ecx
-pr001:
-    mov     al, [edx]
-    cmp     al, 0xff                        ; End of options?
-    je      pr_exit
+  .next_option:
+        add     edx, ecx
-    cmp     al, dhcp_msg_type               ; Msg type is a single byte option
+        mov     al, [edx]                       ; get message identifier
-    jne     @f
-    mov     al, [edx+2]
-    mov     [dhcpMsgType], al
+        cmp     al, 0xff                        ; End of options?
+        je      .done
     DEBUGF  1,"DHCP Msg type: %u\n", al
+        cmp     al, 0
+        je      .pad
+; TODO: check if we still are inside the buffer
+        inc     edx
+        movzx   ecx, byte [edx]                 ; get data length
+        inc     edx                             ; point to data
+        cmp     al, dhcp_msg_type               ; Msg type is a single byte option
+        je      .msgtype
+        cmp     al, dhcp_dhcp_server_id
+        je      .server
+        cmp     al, dhcp_address_time
+        je      .lease
+        cmp     al, dhcp_subnet_mask
+        je      .subnet
+        cmp     al, dhcp_router
+        je      .router
-    add     edx, 3
+        cmp     al, dhcp_domain_server
+        je      .dns
+        DEBUGF  1,"Unsupported DHCP option: %u\n", al
+        jmp     .next_option
+  .pad:
+        xor     ecx, ecx
+        inc     ecx
+        jmp     .next_option
     jmp     pr001                           ; Get next option
   .msgtype:
-@@:
+        mov     al, [edx]
-    inc     edx
+        mov     [dhcpMsgType], al
-    movzx   ecx, byte [edx]
-    inc     edx                             ; point to data
-    cmp     al, dhcp_dhcp_server_id         ; server ip
-    jne     @f
+        DEBUGF  1,"DHCP Msg type: %u\n", al
-    mov     eax, [edx]
+        jmp     .next_option                    ; Get next option
     mov     [dhcpServerIP], eax
-    DEBUGF  1,"Server: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
+  .server:
-    jmp     next_option
+        mov     eax, [edx]
         mov     [dhcpServerIP], eax
+        DEBUGF  1,"Server: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
-@@:
-    cmp     al, dhcp_address_time
-    jne     @f
-    pusha
+        jmp     .next_option
-    mov     eax,[edx]
     bswap   eax
-    mov     [dhcpLease],eax
+  .lease:
-    DEBUGF  1,"lease: %d\n",eax
+        pusha
-    popa
+        mov     eax,[edx]
-    jmp     next_option
-@@:
+        bswap   eax
-    cmp     al, dhcp_subnet_mask
+        mov     [dhcpLease],eax
-    jne     @f
+        DEBUGF  1,"lease: %d\n",eax
         popa
-    push    dword [edx]
+        jmp     .next_option
-    pop     [dhcp.subnet]
-    DEBUGF  1,"Subnet: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
-    jmp     next_option
-@@:
-    cmp     al, dhcp_router
+  .subnet:
-    jne     @f
+        push    dword [edx]
         pop     [dhcp.subnet]
-    push    dword [edx]
+        DEBUGF  1,"Subnet: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
-    pop     [dhcp.gateway]
-    DEBUGF  1,"Gateway: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
+        jmp     .next_option
     jmp     next_option
-@@:
+  .router:
-    cmp     al, dhcp_domain_server
+        push    dword [edx]
-    jne     next_option
+        pop     [dhcp.gateway]
+        DEBUGF  1,"Gateway: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
         jmp     .next_option
     push    dword [edx]
-    pop     [dhcp.dns]
+  .dns:
-    DEBUGF  1,"DNS: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1
+        push    dword [edx]
-    jmp     next_option
+        pop     [dhcp.dns]
-    xor esi,esi
+        xor     esi, esi
    probe_loop:
-    call  random                            ; create a pseudo random number in eax (seeded by MAC)
+        call    random                                  ; create a pseudo random number in eax (seeded by MAC)
-    cmp   al,PROBE_MIN*100                  ; check if al is bigger then PROBE_MIN
+        cmp     al, PROBE_MIN*100                       ; check if al is bigger then PROBE_MIN
-    jge   @f                                ; all ok
+        jae     @f                                      ; all ok
-    add   al,(PROBE_MAX-PROBE_MIN)*100      ; al is too small
+        add     al, (PROBE_MAX-PROBE_MIN)*100           ; al is too small
    @@:
-    cmp   al,PROBE_MAX*100
+        cmp     al, PROBE_MAX*100
-    jle   @f
+        jbe     @f
-    sub   al,(PROBE_MAX-PROBE_MIN)*100
+        sub     al, (PROBE_MAX-PROBE_MIN)*100
    @@:
-    movzx ebx,al
+        movzx   ebx,al
-    DEBUGF  1,"Waiting %u0ms\n",ebx
+        DEBUGF  1,"Waiting %u0ms\n",ebx
-    mcall 5
+        mcall   5
-    DEBUGF  1,"Sending Probe\n"
+        DEBUGF  1,"Sending Probe\n"
-;    eth.ARP_PROBE MAC
+        mcall   76, API_ARP + 6
-    inc   esi
+        inc     esi
-    cmp   esi,PROBE_NUM
+        cmp     esi, PROBE_NUM
-    jl    probe_loop
+        jb      probe_loop
 ; now we wait further ANNOUNCE_WAIT seconds and send ANNOUNCE_NUM ARP announces. If any other host has assingned
 ; IP within this time, we should create another adress, that have to be done later
-    DEBUGF  1,"Waiting %us\n",ANNOUNCE_WAIT
+        DEBUGF  1,"Waiting %us\n", ANNOUNCE_WAIT
-    mcall 5, ANNOUNCE_WAIT*100
+        mcall   5, ANNOUNCE_WAIT*100
-    xor   esi,esi
+        xor   esi, esi
    announce_loop:
-    DEBUGF  1,"Sending Announce\n"
+        DEBUGF  1,"Sending Announce\n"
-;    eth.ARP_ANNOUNCE MAC
+        mcall   76, API_ARP + 6
-    inc   esi
+        inc     esi
+        cmp     esi,ANNOUNCE_NUM
-    cmp   esi,ANNOUNCE_NUM
+        je      @f
     je    @f
 socketNum       dd  ?
 MAC             dp  ?
 currTime        dd  ?
-renewTime       dd  ?
+currTime        dd  ?
 generator       dd  ?

Subversion Repositories Kolibri OS

(root)/kernel/branches/net/applications/zeroconf/zeroconf.asm – Rev 2995 → 3200