Subversion Repositories Kolibri OS

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

;;                                                                 ;;

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

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

;;                                                                 ;;

;;  ping.asm - ICMP echo client for KolibriOS                      ;;

;;                                                                 ;;

;;  Written by hidnplayr@kolibrios.org                             ;;

;;                                                                 ;;

;;          GNU GENERAL PUBLIC LICENSE                             ;;

;;             Version 2, June 1991                                ;;

;;                                                                 ;;

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

format binary as ""

BUFFERSIZE      = 65536

use32

        org     0x0

        db      'MENUET01'      ; signature

        dd      1               ; header version

        dd      START           ; entry point

        dd      I_END           ; initialized size

        dd      IM_END+0x1000   ; required memory

        dd      IM_END+0x1000   ; stack pointer

        dd      params          ; parameters

        dd      0               ; path

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

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

purge mov,add,sub

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

include '../../struct.inc'

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

include '../icmp.inc'

include '../ip.inc'

START:

; init heap

        mcall   68, 11

        test    eax, eax

        jz      exit

; load libraries

        stdcall dll.Load, @IMPORT

        test    eax, eax

        jnz     exit

; initialize console

        push    1

        call    [con_start]

        push    title

        push    250

        push    80

        push    25

        push    80

        call    [con_init]

; Init identifier with our PID number

        mcall   9, thread_info, -1

        mov     eax, [thread_info.PID]

        mov     [icmp_packet.id], ax

; expand payload to 65504 bytes

        mov     edi, icmp_packet.data+32

        mov     ecx, 65504/32-1

  .expand_payload:

        mov     esi, icmp_packet.data

        movsd

        movsd

        movsd

        movsd

        movsd

        movsd

        movsd

        movsd

        dec     ecx

        jnz     .expand_payload

; main loop

        cmp     byte[params], 0

        jne     parse_param

        push    str_welcome

        call    [con_write_asciiz]

main:

; write prompt

        push    str_prompt

        call    [con_write_asciiz]

; read string

        mov     esi, params

        push    1024

        push    esi

        call    [con_gets]

; check for exit

        test    eax, eax

        jz      exit

        cmp     byte [esi], 10

        jz      exit

; delete terminating '\n'

        push    esi

@@:

        lodsb

        test    al, al

        jnz     @b

        mov     [esi-2], al

        pop     esi

; reset stats

        mov     [stats.tx], 0

        mov     [stats.rx], 0

        mov     [stats.time], 0

parse_param:

; parameters defaults

        mov     [count], 4

        mov     [size], 32

        mov     [ttl], 128

        mov     [timeout], 300

; Check if any additional parameters were given

        mov     esi, params

        mov     ecx, 1024

  .addrloop:

        lodsb

        test    al, al

        jz      .resolve

        cmp     al, ' '

        jne     .addrloop

        mov     byte[esi-1], 0

        jmp     .param

  .param_loop:

        lodsb

        test    al, al

        jz      .resolve

        cmp     al, ' '

        jne     .invalid

  .param:

        lodsb

        cmp     al, '-'

        jne     .invalid

        lodsb

        cmp     al, 't'

        jne     @f

        mov     [count], -1     ; infinite

        jmp     .param_loop

  @@:

        cmp     al, 'n'

        jne     @f

        call    ascii_to_dec

        test    ebx, ebx

        jz      .invalid

        mov     [count], ebx

        jmp     .param_loop

  @@:

        cmp     al, 'l'

        jne     @f

        call    ascii_to_dec

        test    ebx, ebx

        jz      .invalid

        cmp     ebx, 65500

        ja      .invalid

        mov     [size], ebx

        jmp     .param_loop

  @@:

        cmp     al, 'i'

        jne     @f

        call    ascii_to_dec

        test    ebx, ebx

        jz      .invalid

        cmp     ebx, 255

        ja      .invalid

        mov     [ttl], ebx

        jmp     .param_loop

  @@:

        cmp     al, 'w'

        jne     @f

        call    ascii_to_dec

        test    ebx, ebx

        jz      .invalid

        mov     [timeout], ebx

        jmp     .param_loop

  @@:

        ; implement more parameters here

  .invalid:

        push    str13

        call    [con_write_asciiz]

        jmp     main

  .resolve:

; resolve name

        push    esp     ; reserve stack place

        push    esp     ; fourth parameter

        push    0       ; third parameter

        push    0       ; second parameter

        push    params  ; first parameter

        call    [getaddrinfo]

        pop     esi

; test for error

        test    eax, eax

        jnz     fail

; convert IP address to decimal notation

        mov     eax, [esi+addrinfo.ai_addr]

        mov     eax, [eax+sockaddr_in.sin_addr]

        mov     [sockaddr1.ip], eax

        push    eax

        call    [inet_ntoa]

; write result

        mov     [ip_ptr], eax

        push    eax

; free allocated memory

        push    esi

        call    [freeaddrinfo]

        push    str4

        call    [con_write_asciiz]

        mcall   socket, AF_INET4, SOCK_RAW, IPPROTO_ICMP

        cmp     eax, -1

        jz      fail2

        mov     [socketnum], eax

        mcall   connect, [socketnum], sockaddr1, 18

        cmp     eax, -1

        je      fail2

        pushd   [ttl]

        pushd   4                               ; length of option

        pushd   IP_TTL

        pushd   IPPROTO_IP

        mcall   setsockopt, [socketnum], esp

        add     esp, 16

        cmp     eax, -1

        je      fail2

        mcall   40, EVM_STACK

        push    str3

        call    [con_write_asciiz]

        push    [ip_ptr]

        call    [con_write_asciiz]

        push    [size]

        push    str3b

        call    [con_printf]

        add     esp, 2*4

mainloop:

        call    [con_get_flags]

        test    eax, 0x200                      ; con window closed?

        jnz     exit_now

        inc     [stats.tx]

        mcall   26, 10                          ; Get high precision timer count

        mov     [time_reference], eax

        mov     esi, [size]

        add     esi, sizeof.ICMP_header

        xor     edi, edi

        mcall   send, [socketnum], icmp_packet

        cmp     eax, -1

        je      fail2

        mov     [time_exceeded], 0

  .receiveloop:

        mov     ebx, [timeout]

        sub     ebx, [time_exceeded]

        jb      .no_response

        mcall   23                              ; Wait for network event with timeout

        mcall   26, 10                          ; Get high precision timer count

        sub     eax, [time_reference]

        jz      @f

        xor     edx, edx

        mov     ebx, 100000

        div     ebx

        cmp     edx, 50000

        jb      @f

        inc     eax

  @@:

        mov     [time_exceeded], eax           ; Exceeded time in 1/100 s

; Receive reply

        mcall   recv, [socketnum], buffer_ptr, BUFFERSIZE, MSG_DONTWAIT

        cmp     eax, -1

        je      .no_response

        test    eax, eax

        jz      fail2

; IP header length

        movzx   esi, byte[buffer_ptr]

        and     esi, 0xf

        shl     esi, 2

; Check packet length

        sub     eax, esi

        sub     eax, sizeof.ICMP_header

        jb      .invalid

        mov     [recvd], eax

; make esi point to ICMP packet header

        add     esi, buffer_ptr

; Check identifier

        mov     ax, [icmp_packet.id]

        cmp     [esi + ICMP_header.Identifier], ax

        jne     .receiveloop

; we have a response, print the sender IP

        push    esi

        mov     eax, [buffer_ptr + IPv4_header.SourceAddress]

        rol     eax, 16

        movzx   ebx, ah

        push    ebx

        movzx   ebx, al

        push    ebx

        shr     eax, 16

        movzx   ebx, ah

        push    ebx

        movzx   ebx, al

        push    ebx

        push    str11

        call    [con_printf]

        add     esp, 5*4

        pop     esi

; What kind of response is it?

        cmp     [esi + ICMP_header.Type], ICMP_ECHOREPLY

        je      .echo_reply

        cmp     [esi + ICMP_header.Type], ICMP_TIMXCEED

        je      .ttl_exceeded

        jmp     .invalid

  .echo_reply:

; Validate the payload

        add     esi, sizeof.ICMP_header

        mov     ecx, [size]

        mov     edi, icmp_packet.data

        repe    cmpsb

        jne     .miscomp

; update stats

        inc     [stats.rx]

        mov     eax, [time_exceeded]

        add     [stats.time], eax

; Print time exceeded

        movzx   eax, [buffer_ptr + IPv4_header.TimeToLive]

        push    eax

        mov     eax, [time_exceeded]

        xor     edx, edx

        mov     ebx, 10

        div     ebx

        push    edx

        push    eax

        push    [recvd]

        push    str7

        call    [con_printf]

        add     esp, 5*4

        jmp     .continue

  .ttl_exceeded:

        push    str14

        call    [con_write_asciiz]

        jmp     .continue

; Error in packet, print it to user

  .miscomp:

        sub     edi, icmp_packet.data+1

        push    edi

        push    str9

        call    [con_printf]

        add     esp, 2*4

        jmp     .continue

; Invalid reply

  .invalid:

        push    str10

        call    [con_write_asciiz]

        jmp     .continue

; Timeout!

  .no_response:

        push    str8

        call    [con_write_asciiz]

; Send more ICMP packets ?

  .continue:

        inc     [icmp_packet.seq]

        cmp     [count], -1

        je      .forever

        dec     [count]

        jz      .stats

  .forever:

; wait a second before sending next request

        mcall   5, 100

        jmp     mainloop

; Print statistics

  .stats:

        cmp     [stats.rx], 0

        jne     @f

        xor     eax, eax

        xor     edx, edx

        jmp     .zero

  @@:

        xor     edx, edx

        mov     eax, [stats.time]

        div     [stats.rx]

        xor     edx, edx

        mov     ebx, 10

        div     ebx

  .zero:

        push    edx

        push    eax

        push    [stats.rx]

        push    [stats.tx]

        push    str12

        call    [con_printf]

        add     esp, 5*4

        jmp     main

; DNS error

fail:

        push    str5

        call    [con_write_asciiz]

        jmp     main

; Socket error

fail2:

        push    str6

        call    [con_write_asciiz]

        jmp     main

; Finally.. exit!

exit:

        push    1

        call    [con_exit]

exit_now:

        mcall   -1

ascii_to_dec:

        lodsb

        cmp     al, ' '

        jne     .fail

        xor     eax, eax

        xor     ebx, ebx

  .loop:

        lodsb

        test    al, al

        jz      .done

        cmp     al, ' '

        je      .done

        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

        ret

; data

title   db      'ICMP echo (ping) client',0

str_welcome db  'Please enter the hostname or IP-address of the host you want to ping,',10

            db  'or just press enter to exit.',10,10

            db  'Options:',10

            db  ' -t            Send packets till users abort.',10

            db  ' -n number     Number of requests to send.',10

            db  ' -i TTL        Time to live.',10

            db  ' -l size       Size of echo request.',10

            db  ' -w time-out   Time-out in hundredths of a second.',10,0

str_prompt  db  10,'> ',0

str3    db      'Pinging to ',0

str3b   db      ' with %u data bytes',10,0

str4    db      10,0

str5    db      'Name resolution failed.',10,0

str6    db      'Socket error.',10,0

str13   db      'Invalid parameter(s)',10,0

str11   db      'Answer from %u.%u.%u.%u: ',0

str7    db      'bytes=%u time=%u.%u ms TTL=%u',10,0

str8    db      'Timeout',10,0

str9    db      'miscompare at offset %u.',10,0

str10   db      'invalid reply.',10,0

str14   db      'TTL expired.',10,0

str12   db      10,'Statistics:',10,'%u packets sent, %u packets received',10,'average response time=%u.%u ms',10,0

sockaddr1:

        dw AF_INET4

.port   dw 0

.ip     dd 0

        rb 10

time_reference  dd ?    ; start time of sent packet

time_exceeded   dd ?    ; time exceeded between send and receive

ip_ptr          dd ?

count           dd ?

size            dd ?

ttl             dd ?

timeout         dd ?

recvd           dd ?    ; received number of bytes in last packet

stats:

        .tx     dd ?

        .rx     dd ?

        .time   dd ?

; import

align 4

@IMPORT:

library network, 'network.obj', console, 'console.obj'

import  network,        \

        getaddrinfo,    'getaddrinfo',  \

        freeaddrinfo,   'freeaddrinfo', \

        inet_ntoa,      'inet_ntoa'

import  console,        \

        con_start,      'START',        \

        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'

socketnum       dd ?

icmp_packet     db ICMP_ECHO    ; type

                db 0            ; code

                dw 0            ; checksum

 .id            dw 0            ; identifier

 .seq           dw 0x0000       ; sequence number

 .data          db 'abcdefghijklmnopqrstuvwxyz012345'

I_END:

                rb 65504-32

thread_info process_information

params          rb 1024

buffer_ptr:     rb BUFFERSIZE

IM_END: