WebSVN – Kolibri OS – Diff – /programs/network/icmp/ping.asm

 ;;          GNU GENERAL PUBLIC LICENSE                             ;;
 ;;             Version 2, June 1991                                ;;
 ;;                                                                 ;;
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
+; TODO: more precise timer, ttl, user selectable size/number of packets
 format binary as ""
 BUFFERSIZE      = 1500
-BUFFERSIZE      = 1500
+IDENTIFIER      = 0x1337
         mcall   40, 1 shl 7 ; + 7
 ;        call    [con_cls]
-        mov     [count], 4
         mov     [count], 4
 mainloop:
+        push    str3
-        push    str3
+        call    [con_write_asciiz]
         call    [con_write_asciiz]
+        push    [ip_ptr]
+        call    [con_write_asciiz]
+        push    (icmp_packet.length - ICMP_Packet.Data)
+        push    str3b
+        call    [con_printf]
         push    [ip_ptr]
-        call    [con_write_asciiz]
+mainloop:
         inc     [stats.tx]
         mcall   26,9
         mov     [time_reference], eax
         mcall   send, [socketnum], icmp_packet, icmp_packet.length, 0
         mcall   23, 300 ; 3 seconds time-out
         mcall   26,9
         neg     [time_reference]
+        add     [time_reference], eax
+        mcall   recv, [socketnum], buffer_ptr, BUFFERSIZE, MSG_DONTWAIT
+        cmp     eax, -1
+        je      .no_response
+        sub     eax, ICMP_Packet.Data
+        jb      .no_response            ; FIXME: use other error message?
+        mov     [recvd], eax
+        cmp     word[buffer_ptr + ICMP_Packet.Identifier], IDENTIFIER
+        jne     .no_response            ; FIXME: use other error message?
+; OK, we have a response, update stats and let the user know
+        inc     [stats.rx]
-        add     [time_reference], eax
+        mov     eax, [time_reference]
         add     [stats.time], eax
-        mcall   recv, [socketnum], buffer_ptr, BUFFERSIZE, 0
+        push    str11                   ; TODO: print IP address of packet sender
-        cmp     eax, -1
-        je      .no_response
+        call    [con_write_asciiz]
+; validate the packet
-; validate the packet
+        lea     esi, [buffer_ptr + ICMP_Packet.Data]
+        mov     ecx, [recvd]
+        mov     edi, icmp_packet.data
+        repe    cmpsb
+        jne     .miscomp
         lea     esi, [buffer_ptr + ICMP_Packet.Data]
-        mov     edi, icmp_packet.data
+; All OK, print to the user!
-        mov     ecx, 32/4
+        push    [time_reference]
+        movzx   eax, word[buffer_ptr + ICMP_Packet.SequenceNumber]
-        repe    cmpsd
+        push    eax
-        jne     .miscomp
+        push    [recvd]
-        push    [time_reference]
+        push    str7
-        push    str7
+        call    [con_printf]
         call    [con_printf]
+        jmp     continue
-        jmp     continue
+; Error in packet, print it to user
   .miscomp:
+        sub     edi, icmp_packet.data
-  .miscomp:
+        push    edi
-        sub     edi, icmp_packet.data
+        push    str9
-        push    edi
+        call    [con_printf]
-        push    str9
-        call    [con_printf]
-        jmp     continue
+        jmp     continue
+; Timeout!
+  .no_response:
         push    str8
+        call    [con_write_asciiz]
+; Send more ICMP packets ?
+   continue:
+        dec     [count]
+        jz      done
+        mcall   5, 100  ; wait a second
-  .no_response:
+        inc     [icmp_packet.seq]
-        push    str8
+        jmp     mainloop
         call    [con_write_asciiz]
 ; Done..
-   continue:
+done:
+        xor     edx, edx
+        mov     eax, [stats.time]
-        dec     [count]
+        div     [stats.rx]
-        jz      done
+        push    eax
+        push    [stats.rx]
-        mcall   5, 100  ; wait a second
+        push    [stats.tx]
-        inc     [icmp_packet.id]
+        push    str12
-        jmp     mainloop
+        call    [con_printf]
+        push    str10
+        call    [con_write_asciiz]
         call    [con_getch2]
         push    1
-done:
+        call    [con_exit]
         push    str10
-        call    [con_write_asciiz]
+; Finally.. exit!
-        call    [con_getch2]
+exit:
-        push    1
+        mcall   -1
-        call    [con_exit]
+; DNS error
+fail:
-exit:
+        push    str5
-        mcall   -1
+        call    [con_write_asciiz]
         jmp     done
+; Socket error
-fail:
+fail2:
-        push    str5
+        push    str6
         call    [con_write_asciiz]
         jmp     done
-fail2:
+; data
-        push    str6
+title   db      'ICMP - echo client',0
-        call    [con_write_asciiz]
+str2    db      '> ',0
-        jmp     done
+str3    db      'Pinging to ',0
 str3b   db      ' with %u data bytes',10,0
-; data
+str4    db      10,0
-title   db      'ICMP - echo client',0
+str5    db      'Name resolution failed.',10,0
+str6    db      'Could not open socket',10,0
+str11   db      'Answer: ',0
+str7    db      'bytes=%u seq=%u time=%u0 ms',10,0
+str8    db      'timeout!',10,0
-str2    db      '> ',0
+str9    db      'miscompare at offset %u',10,0
-str3    db      'Ping to ',0
+str10   db      10,'Press any key to exit',0
 str4    db      10,0
 socketnum       dd ?
 icmp_packet:    db 8            ; type
                 db 0            ; code
                 dw 0            ;
- .id            dw 0x0000       ; identifier
+ .id            dw IDENTIFIER   ; identifier
- .seq           dw 0x0001       ; sequence number
+ .seq           dw 0x0000       ; sequence number
- .data          db 'abcdefghijklmnopqrstuvwxyz012345678'
+ .data          db 'abcdefghijklmnopqrstuvwxyz012345'
  .length = $ - icmp_packet

Subversion Repositories Kolibri OS

(root)/programs/network/icmp/ping.asm – Rev 3618 → 4012