Subversion Repositories Kolibri OS

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

;;                                                                 ;;

;; Copyright (C) KolibriOS team 2010-2015. All rights reserved.    ;;

;; Distributed under terms of the GNU General Public License       ;;

;;                                                                 ;;

;;  zeroconfig.asm - Zeroconfig service for KolibriOS              ;;

;;                                                                 ;;

;;  Written by hidnplayr@kolibrios.org                             ;;

;;    Some code contributed by Derpenguin                          ;;

;;                                                                 ;;

;;  DHCP code is based on that by Mike Hibbet                      ;;

;;      (DHCP client for menuetos)                                 ;;

;;                                                                 ;;

;;          GNU GENERAL PUBLIC LICENSE                             ;;

;;             Version 2, June 1991                                ;;

;;                                                                 ;;

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

format binary as ""

; CONFIGURATION

TIMEOUT                 = 3             ; in seconds

BUFFER                  = 1024          ; in bytes

DHCP_TRIES              = 3             ; number of times to try contacting DHCP server

__DEBUG__               = 1             ; enable/disable

__DEBUG_LEVEL__         = 2             ; 1 = all, 2 = errors

; CONFIGURATION FOR LINK-LOCAL

PROBE_WAIT              = 1             ; second  (initial random delay)

PROBE_MIN               = 1             ; second  (minimum delay till repeated probe)

PROBE_MAX               = 2             ; seconds (maximum delay till repeated probe)

PROBE_NUM               = 3             ;         (number of probe packets)

ANNOUNCE_NUM            = 2             ;         (number of announcement packets)

ANNOUNCE_INTERVAL       = 2             ; seconds (time between announcement packets)

ANNOUNCE_WAIT           = 2             ; seconds (delay before announcing)

MAX_CONFLICTS           = 10            ;         (max conflicts before rate limiting)

RATE_LIMIT_INTERVAL     = 60            ; seconds (delay between successive attempts)

DEFEND_INTERVAL         = 10            ; seconds (min. wait between defensive ARPs)

use32

        org     0x0

        db      'MENUET01'              ; 8 byte id

        dd      0x01                    ; header version

        dd      START                   ; start of code

        dd      IM_END                  ; size of image

        dd      (I_END+0x100)           ; memory for app

        dd      (I_END+0x100)           ; esp

        dd      0, 0                    ; I_Param, I_Path

include '../../proc32.inc'

include '../../macros.inc'

include '../../debug-fdo.inc'

include '../../network.inc'

include 'dhcp.inc'

include '../../dll.inc'

START:

        mcall   68, 11

        stdcall dll.Load,@IMPORT

        or      eax, eax

        jnz     fail

        DEBUGF  2,"Zero-config service loaded\n"

        mcall   40, EVM_STACK2

wait_for_link_up:

        mov     bh, [device]

        mov     bl, 0           ; Get device type

        mcall   74

        cmp     eax, 1          ; Ethernet

        jne     .wait

        mov     bl, 10          ; Get Link status

        mcall   74

        test    eax, eax

        jnz     .go

  .wait:

        mcall   10

        jmp     wait_for_link_up

  .go:

        mov     ebx, API_ETH + 0

        mov     bh, [device]

        mcall   76              ; get MAC of the ethernet interface

        mov     word[MAC], bx

        mov     dword[MAC+2], eax

        DEBUGF  1,"MAC: %x-%x-%x-%x-%x-%x\n", [MAC+0]:2, [MAC+1]:2, [MAC+2]:2, [MAC+3]:2, [MAC+4]:2, [MAC+5]:2

        invoke  ini.get_str, path, str_ipconfig, str_type, inibuf, 16, 0

        cmp     dword[inibuf], 'stat'

        je      static

        jmp     try_dhcp

wait_for_link_down:

; TODO: detect ARP conflicts

        mcall   40, EVM_STACK2

  .loop:

        mcall   10

        mov     bh, [device]

        mov     bl, 0           ; Get device type

        mcall   74

        cmp     eax, 0          ; No device

        je      .down

        mov     bl, 10          ; Get Link status

        mcall   74

        test    eax, eax

        jnz     .loop

  .down:

        xor     ecx, ecx

        mov     ebx, API_IPv4 + 3

        mov     bh, [device]

        mcall   76              ; ip

        mov     bl, 5

        mcall   76              ; dns

        mov     bl, 7

        mcall   76              ; subnet

        mov     bl, 9

        mcall   76              ; gateway

        jmp     wait_for_link_up

static:

        DEBUGF  1,"Applying Static IP settings\n"

        invoke  ini.get_str, path, str_ipconfig, str_ip, inibuf, 16, 0

        mov     edx, inibuf

        call    ip_str_to_dword

        mov     ecx, edx

        mov     ebx, API_IPv4 + 3       ; set IP

        mov     bh, [device]

        mcall   76

        invoke  ini.get_str, path, str_ipconfig, str_gateway, inibuf, 16, 0

        mov     edx, inibuf

        call    ip_str_to_dword

        mov     ecx, edx

        mov     ebx, API_IPv4 + 9       ; set gateway

        mov     bh, [device]

        mcall   76

        invoke  ini.get_str, path, str_ipconfig, str_dns, inibuf, 16, 0

        mov     edx, inibuf

        call    ip_str_to_dword

        mov     ecx, edx

        mov     ebx, API_IPv4 + 5       ; set DNS

        mov     bh, [device]

        mcall   76

        invoke  ini.get_str, path, str_ipconfig, str_subnet, inibuf, 16, 0

        mov     edx, inibuf

        call    ip_str_to_dword

        mov     ecx, edx

        mov     ebx, API_IPv4 + 7       ; set subnet

        mov     bh, [device]

        mcall   76

        mov     [notify_struct.msg], str_connected

        mcall   70, notify_struct

        jmp     wait_for_link_down

try_dhcp:

        DEBUGF  2,"Trying to contact DHCP server\n"

        mcall   40, EVM_STACK

        mcall   75, 0, AF_INET4, SOCK_DGRAM, 0          ; open socket (parameters: domain, type, reserved)

        cmp     eax, -1

        je      socket_error

        mov     [socketNum], eax

        DEBUGF  1,"Socket %x opened\n", eax

        mcall   75, 2, [socketNum], sockaddr1, 18       ; bind socket to local port 68

        cmp     eax, -1

        je      socket_error

        DEBUGF  1,"Socket Bound to local port 68\n"

        mcall   75, 4, [socketNum], sockaddr2, 18       ; connect to 255.255.255.255 on port 67

        cmp     eax, -1

        je      socket_error

        DEBUGF  1,"Connected to 255.255.255.255 on port 67\n"

        mov     [dhcpMsgType_tx], 0x01                  ; DHCP discover

        mov     [dhcpLease], esi                        ; esi is still -1 (-1 = forever)

        call    random

        mov     [dhcpXID], eax

build_request:                                          ; Creates a DHCP request packet.

        DEBUGF  1,"Building request\n"

        mcall   26, 9                                   ; Get system time

        imul    eax, 100

        mov     [currTime], eax

        mov     [tries], DHCP_TRIES

        stdcall mem.Alloc, BUFFER

        test    eax, eax

        jz      dhcp_fail2

        mov     [dhcpMsg], eax

        ; Fill buffer with zeros

        mov     edi, eax

        mov     ecx, BUFFER

        xor     eax, eax

        rep     stosb

        mov     edx, [dhcpMsg]

        ; Boot protocol legacy

        mov     [edx], byte 0x01                ; Boot request

        mov     [edx+1], byte 0x01              ; Ethernet

        mov     [edx+2], byte 0x06              ; Ethernet h/w len

        mov     eax, [dhcpXID]

        mov     [edx+4], eax                    ; 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

        ; DHCP extension

        mov     [edx+236], dword 0x63538263     ; magic cookie

        mov     [edx+240], word 0x0135          ; option DHCP msg type

        mov     al, [dhcpMsgType_tx]

        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, [dhcp.ip]

        mov     [edx+240+11], eax

        mov     [edx+240+15], word 0x0437       ; option request list

        mov     [edx+240+17], dword 0x0f060301

        cmp     [dhcpMsgType_tx], 0x01          ; Check which msg we are sending

        jne     .options

        mov     [edx+240+21], byte 0xff         ; end of options marker

        mov     [dhcpMsgLen], 262               ; length

        jmp     send_dhcpmsg

  .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

        mov     [dhcpMsgLen], 268               ; length

send_dhcpmsg:

        DEBUGF  1,"Sending DHCP discover/request\n"

        mcall   75, 6, [socketNum], [dhcpMsg], [dhcpMsgLen]             ; write to socket (send broadcast request)

; Wait for data

        mcall   26, 9

        add     eax, TIMEOUT*100

        mov     [timeout], eax

  .wait:

        mcall   23, TIMEOUT

read_data:                                                              ; we have data - this will be the response

        mcall   75, 7, [socketNum], [dhcpMsg], BUFFER, MSG_DONTWAIT     ; read data from socket

        cmp     eax, -1

        jne     .got_data

        mcall   26, 9

        cmp     eax, [timeout]

        jb      send_dhcpmsg.wait

        DEBUGF  2,"No answer from DHCP server\n"

        dec     [tries]

        jnz     send_dhcpmsg                    ; try again

        stdcall mem.Free, [dhcpMsg]

        jmp     dhcp_fail

  .got_data:

        DEBUGF  1,"%d bytes received\n", eax

        mov     [dhcpMsgLen], eax

; depending on which msg we sent, handle the response

; accordingly.

; If the response is to a dhcp discover, then:

;  1) If response is DHCP OFFER then

;  1.1) record server IP, lease time & IP address.

;  1.2) send a request packet

; If the response is to a dhcp request, then:

;  1) If the response is DHCP ACK then

;  1.1) extract the DNS & subnet fields. Set them in the stack

        cmp     [dhcpMsgType_tx], 0x01          ; did we send a discover?

        je      discover_sent

        cmp     [dhcpMsgType_tx], 0x03          ; did we send a request?

        je      request_sent

        ; we should never reach here ;)

        stdcall mem.Free, [dhcpMsg]

        jmp     fail

discover_sent:

        call    parse_response

        cmp     [dhcpMsgType_rx], 0x02          ; Was the response an offer?

        jne     read_data

        DEBUGF  1, "Got offer, making request\n"

        mov     [dhcpMsgType_tx], 0x03          ; make it a request

        jmp     build_request

request_sent:

        call    parse_response

        cmp     [dhcpMsgType_rx], 0x05          ; Was the response an ACK? It should be

        jne     read_data                       ; NO - read next packets

        DEBUGF  2, "IP assigned by DHCP server successfully\n"

        mov     [notify_struct.msg], str_connected

        mcall   70, notify_struct

        mcall   close, [socketNum]

        mov     ebx, API_IPv4 + 3

        mov     bh, [device]

        mcall   76, , [dhcp.ip]                 ; ip

        mov     bl, 5

        mcall   76, , [dhcp.dns]                ; dns

        mov     bl, 7

        mcall   76, , [dhcp.subnet]             ; subnet

        mov     bl, 9

        mcall   76, , [dhcp.gateway]            ; gateway

        jmp     wait_for_link_down

;***************************************************************************

;   Function

;      parseResponse

;

;   Description

;      extracts the fields ( client IP address and options ) from

;      a DHCP response

;      The values go into

;       dhcpMsgType,dhcpLease,dhcpClientIP,dhcpServerIP,

;       dhcpDNSIP, dhcpSubnet

;      The message is stored in dhcpMsg

;

;***************************************************************************

parse_response:

        DEBUGF  1,"Parsing response\n"

        mov     edx, [dhcpMsg]

        mov     [dhcpMsgType_rx], 0

; Verify if session ID matches

        mov     eax, [dhcpXID]

        cmp     dword[edx+4], eax

        jne     .done

        push    dword [edx+16]

        pop     [dhcp.ip]

        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

        mov     al, 240                         ; Point to first option

        movzx   ecx, al

  .next_option:

        add     edx, ecx

        mov     al, [edx]                       ; get message identifier

        cmp     al, 0xff                        ; End of options?

        je      .done

        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

        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

  .msgtype:

        mov     al, [edx]

        mov     [dhcpMsgType_rx], al

        DEBUGF  1,"DHCP Msg type: %u\n", al

        jmp     .next_option                    ; Get next option

  .server:

        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

        jmp     .next_option

  .lease:

        pusha

        mov     eax,[edx]

        bswap   eax

        mov     [dhcpLease],eax

        DEBUGF  1,"Lease: %d\n",eax

        popa

        jmp     .next_option

  .subnet:

        push    dword [edx]

        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

  .router:

        push    dword [edx]

        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

  .dns:

        push    dword [edx]

        pop     [dhcp.dns]

        DEBUGF  1,"DNS: %u.%u.%u.%u\n",[edx]:1,[edx+1]:1,[edx+2]:1,[edx+3]:1

        jmp     .next_option

  .done:

        stdcall mem.Free, [dhcpMsg]

        ret

dhcp_fail:

        mcall   close, [socketNum]

dhcp_fail2:

        DEBUGF  1,"DHCP failed\n"

link_local:

        call    random

        mov     cx, ax

        shl     ecx, 16

        mov     cx, 0xfea9                              ; IP 169.254.0.0 link local net, see RFC3927

        mov     ebx, API_IPv4 + 3

        mov     bh, [device]

        mcall   76, , ecx                               ; mask is 255.255.0.0

        DEBUGF  2,"Link Local IP assigned: 169.254.%u.%u\n", [generator+0]:1, [generator+1]:1

        mov     bl, 7

        mcall   76, , 0xffff

        mov     bl, 9

        mcall   76, , 0x0

        mov     bl, 5

        mcall   76, , 0x0

        mcall   5, PROBE_WAIT*100

        xor     esi, esi

   probe_loop:

        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

        jae     @f                                      ; all ok

        add     al, (PROBE_MAX-PROBE_MIN)*100           ; al is too small

   @@:

        cmp     al, PROBE_MAX*100

        jbe     @f

        sub     al, (PROBE_MAX-PROBE_MIN)*100

   @@:

        movzx   ebx,al

        DEBUGF  1,"Waiting %u0ms\n",ebx

        mcall   5

        DEBUGF  1,"Sending Probe\n"

        mov     ebx, API_ARP + 6

        mov     bh, [device]

        mcall   76

        inc     esi

        cmp     esi, PROBE_NUM

        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

        mcall   5, ANNOUNCE_WAIT*100

        xor   esi, esi

   announce_loop:

        DEBUGF  1,"Sending Announce\n"

        mov     ebx, API_ARP + 6

        mov     bh, [device]

        mcall   76

        inc     esi

        cmp     esi,ANNOUNCE_NUM

        je      @f

        DEBUGF  1,"Waiting %us\n", ANNOUNCE_INTERVAL

        mcall   5, ANNOUNCE_INTERVAL*100

        jmp     announce_loop

   @@:

        jmp     wait_for_link_down

socket_error:

        DEBUGF  2,"Socket error\n"

fail:

        DEBUGF  2,"Zeroconf failed!\n"

        mcall   -1

random:  ; Pseudo random actually

        mov     eax, [generator]

        add     eax, -43ab45b5h

        ror     eax, 1

        bswap   eax

        xor     eax, dword[MAC]

        ror     eax, 1

        xor     eax, dword[MAC+2]

        mov     [generator], eax

        ret

ip_str_to_dword:

    push    edx

    ; This code validates if the query is an IP containing 4 numbers and 3 dots

    xor     al, al            ; make al (dot count) zero

   @@:

    cmp     byte[edx],'0'     ; check if this byte is a number, if not jump to no_IP

    jl      no_IP             ;

    cmp     byte[edx],'9'     ;

    jg      no_IP             ;

    inc     edx               ; the byte was a number, so lets check the next byte

    cmp     byte[edx],0       ; is this byte zero? (have we reached end of query?)

    jz      @f                ; jump to next @@ then

    cmp     byte[edx],':'

    jz      @f

    cmp     byte[edx],'.'     ; is this byte a dot?

    jne     @r                ; if not, jump to previous @@

    inc     al                ; the byte was a dot so increment al(dot count)

    inc     edx               ; next byte

    jmp     @r                ; lets check for numbers again (jump to previous @@)

   @@:                        ; we reach this when end of query reached

    cmp     al,3              ; check if there where 3 dots

    jnz     no_IP             ; if not, jump to no_IP

    ; The following code will convert this IP into a dword and output it in eax

    ; If there is also a port number specified, this will be returned in ebx, otherwise ebx is -1

    pop     esi               ; edx (query address) was pushed onto stack and is now popped in esi

    xor     edx, edx          ; result

    xor     eax, eax          ; current character

    xor     ebx, ebx          ; current byte

  .outer_loop:

    shl     edx, 8

    add     edx, ebx

    xor     ebx, ebx

  .inner_loop:

    lodsb

    test    eax, eax

    jz      .finish

    cmp     al, '.'

    jz      .outer_loop

    sub     eax, '0'

    imul    ebx, 10

    add     ebx, eax

    jmp     .inner_loop

  .finish:

    shl     edx, 8

    add     edx, ebx

    bswap   edx               ; we want little endian order

    ret

no_IP:

    pop     edx

    xor     edx, edx

    ret

; DATA AREA

align 16

@IMPORT:

library \

        libini,'libini.obj'

import  libini, \

        ini.get_str,'ini_get_str'

include_debug_strings

str_ip          db 'ip', 0

str_subnet      db 'subnet', 0

str_gateway     db 'gateway', 0

str_dns         db 'dns', 0

str_ipconfig    db 'ipconfig', 0

str_type        db 'type', 0

sockaddr1:

        dw AF_INET4

        dw 68 shl 8     ; local port

        dd 0            ; local IP

        rb 10

sockaddr2:

        dw AF_INET4

        dw 67 shl 8     ; destination port

        dd -1           ; destination IP

        rb 10

notify_struct:

        dd 7            ; run application

        dd 0

 .msg   dd 0

        dd 0

        dd 0

        db '/sys/@notify', 0

str_connected   db '"You are now connected to the network." -N', 0

path            db '/sys/settings/network.ini',0

IM_END:

device          db 1

inibuf          rb 16

tries           db ?

dhcpMsgType_tx  db ?    ; sent

dhcpMsgType_rx  db ?    ; received

dhcpXID         dd ?

dhcpLease       dd ?

dhcpServerIP    dd ?

dhcp:

.ip             dd ?

.subnet         dd ?

.dns            dd ?

.gateway        dd ?

dhcpMsgLen      dd ?

socketNum       dd ?

MAC             dp ?

currTime        dd ?

generator       dd ?

dhcpMsg         dd ?

timeout         dd ?

I_END: