forked from KolibriOS/kolibrios
Trace route program
git-svn-id: svn://kolibrios.org@6923 a494cfbc-eb01-0410-851d-a64ba20cac60
This commit is contained in:
hidnplayr
parent
40a8f693e4
commit
355d9a3988
599
programs/network/tracert/tracert.asm
Normal file
599
programs/network/tracert/tracert.asm
Normal file
@ -0,0 +1,599 @@
|
||||
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
||||
;; ;;
|
||||
;; Copyright (C) KolibriOS team 2010-2017. All rights reserved. ;;
|
||||
;; Distributed under terms of the GNU General Public License ;;
|
||||
;; ;;
|
||||
;; tracert.asm - Trace network route for KolibriOS ;;
|
||||
;; ;;
|
||||
;; Written by hidnplayr@kolibrios.org ;;
|
||||
;; ;;
|
||||
;; GNU GENERAL PUBLIC LICENSE ;;
|
||||
;; Version 2, June 1991 ;;
|
||||
;; ;;
|
||||
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
||||
|
||||
format binary as ""
|
||||
|
||||
BUFFERSIZE = 1500
|
||||
IDENTIFIER = 0x1337
|
||||
|
||||
__DEBUG__ = 1 ; enable/disable
|
||||
__DEBUG_LEVEL__ = 2 ; 1 = all, 2 = errors
|
||||
|
||||
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 '../../debug-fdo.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]
|
||||
; 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
|
||||
|
||||
parse_param:
|
||||
; Check if any additional parameters were given
|
||||
|
||||
DEBUGF 2, "parse parameters\n"
|
||||
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
|
||||
; implement more parameters here
|
||||
.invalid:
|
||||
push str13
|
||||
call [con_write_asciiz]
|
||||
jmp main
|
||||
|
||||
.resolve:
|
||||
DEBUGF 2, "resolve\n"
|
||||
; 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 [icmp_socket], eax
|
||||
|
||||
mcall socket, AF_INET4, SOCK_DGRAM, 0
|
||||
cmp eax, -1
|
||||
jz fail2
|
||||
mov [udp_socket], eax
|
||||
|
||||
mcall connect, [udp_socket], sockaddr1, 18
|
||||
cmp eax, -1
|
||||
je fail2
|
||||
|
||||
mcall 40, EVM_STACK
|
||||
|
||||
push str3
|
||||
call [con_write_asciiz]
|
||||
|
||||
push [ip_ptr]
|
||||
call [con_write_asciiz]
|
||||
|
||||
push str4
|
||||
call [con_write_asciiz]
|
||||
|
||||
mov [ttl], 1
|
||||
|
||||
;; mcall send, [udp_socket], udp_packet, 5, 0 ; dummy send
|
||||
|
||||
mcall recv, [icmp_socket], buffer_ptr, BUFFERSIZE, MSG_DONTWAIT ;; dummy read
|
||||
|
||||
mainloop:
|
||||
call [con_get_flags]
|
||||
test eax, 0x200 ; con window closed?
|
||||
jnz exit_now
|
||||
|
||||
pushd [ttl]
|
||||
pushd str9
|
||||
call [con_printf]
|
||||
add esp, 2*4
|
||||
|
||||
DEBUGF 2, "Setsockopt\n"
|
||||
|
||||
pushd [ttl]
|
||||
pushd 4 ; length of option
|
||||
pushd IP_TTL
|
||||
pushd IPPROTO_IP
|
||||
mcall setsockopt, [udp_socket], esp
|
||||
add esp, 16
|
||||
cmp eax, -1
|
||||
je fail2
|
||||
|
||||
DEBUGF 2, "Sending\n"
|
||||
|
||||
mcall 26, 10 ; Get high precision timer count
|
||||
mov [time_reference], eax
|
||||
mcall send, [udp_socket], udp_packet, 5, 0
|
||||
cmp eax, -1
|
||||
je fail2
|
||||
|
||||
DEBUGF 2, "Packet sent\n", str_ini_int
|
||||
|
||||
.receive:
|
||||
mcall 23, [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_reference], eax
|
||||
|
||||
; Receive reply
|
||||
mcall recv, [icmp_socket], buffer_ptr, BUFFERSIZE, MSG_DONTWAIT
|
||||
cmp eax, -1
|
||||
je .timeout
|
||||
test eax, eax
|
||||
jz fail2
|
||||
|
||||
DEBUGF 2, "Answer after %u\n", eax
|
||||
|
||||
; 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
|
||||
|
||||
DEBUGF 2, "Packet length OK\n", eax
|
||||
|
||||
; make esi point to ICMP packet header
|
||||
add esi, buffer_ptr
|
||||
|
||||
; Verify packet
|
||||
;; movzx eax, [esi + sizeof.ICMP_header + IPv4_header.TimeToLive]
|
||||
;; cmp eax, [ttl]
|
||||
;; jne .receive
|
||||
|
||||
; What kind of response is it?
|
||||
cmp [esi + ICMP_header.Type], ICMP_UNREACH_PORT
|
||||
je .last
|
||||
cmp [esi + ICMP_header.Type], ICMP_TIMXCEED
|
||||
jne .invalid
|
||||
call .print
|
||||
jmp .continue
|
||||
|
||||
.last:
|
||||
call .print
|
||||
jmp main
|
||||
|
||||
.print:
|
||||
DEBUGF 2, "Valid response\n"
|
||||
; we have a response, print a line
|
||||
mov eax, [time_reference]
|
||||
xor edx, edx
|
||||
mov ebx, 10
|
||||
div ebx
|
||||
push edx
|
||||
push eax
|
||||
|
||||
push str1
|
||||
call [con_printf]
|
||||
add esp, 3*4
|
||||
|
||||
mov ebx, [buffer_ptr + IPv4_header.SourceAddress]
|
||||
push ebx
|
||||
call reverse_dns_lookup
|
||||
|
||||
pop eax
|
||||
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 str2
|
||||
call [con_printf]
|
||||
add esp, 5*4
|
||||
|
||||
ret
|
||||
|
||||
|
||||
; Invalid reply
|
||||
.invalid:
|
||||
DEBUGF 2, "Invalid response\n"
|
||||
push str10
|
||||
call [con_write_asciiz]
|
||||
jmp main ;.continue
|
||||
|
||||
; Timeout!
|
||||
.timeout:
|
||||
DEBUGF 2, "Timeout\n", eax
|
||||
push str8
|
||||
call [con_write_asciiz]
|
||||
|
||||
; Send more ICMP packets ?
|
||||
.continue:
|
||||
inc [ttl]
|
||||
|
||||
; wait a second before sending next request
|
||||
mcall 5, 100
|
||||
jmp mainloop
|
||||
|
||||
; 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
|
||||
|
||||
|
||||
; ebx = ip
|
||||
reverse_dns_lookup:
|
||||
|
||||
push ebx
|
||||
mcall socket, AF_INET4, SOCK_DGRAM, 0
|
||||
pop ebx
|
||||
cmp eax, -1
|
||||
je .fail
|
||||
mov [dns_socket], eax
|
||||
|
||||
push ebx
|
||||
mcall connect, [dns_socket], sockaddr2, 18
|
||||
pop ebx
|
||||
cmp eax, -1
|
||||
je .fail
|
||||
|
||||
mov edi, dns_pkt.name
|
||||
rol ebx, 8
|
||||
movzx eax, bl
|
||||
call byte_to_ascii
|
||||
rol ebx, 8
|
||||
movzx eax, bl
|
||||
call byte_to_ascii
|
||||
rol ebx, 8
|
||||
movzx eax, bl
|
||||
call byte_to_ascii
|
||||
rol ebx, 8
|
||||
movzx eax, bl
|
||||
call byte_to_ascii
|
||||
|
||||
mov esi, dns_tr
|
||||
mov ecx, dns_tr.length
|
||||
rep movsb
|
||||
|
||||
sub edi, dns_pkt
|
||||
mov esi, edi
|
||||
|
||||
mcall send, [dns_socket], dns_pkt, , 0
|
||||
cmp eax, -1
|
||||
je .fail
|
||||
|
||||
push esi
|
||||
mcall recv, [dns_socket], buffer_ptr, BUFFERSIZE, 0
|
||||
pop esi
|
||||
|
||||
mcall close, [dns_socket]
|
||||
|
||||
cmp word[buffer_ptr+6], 0 ; answers
|
||||
je .fail
|
||||
|
||||
add esi, buffer_ptr+12
|
||||
mov edi, buffer_ptr
|
||||
xor ecx, ecx
|
||||
lodsb
|
||||
test al, al
|
||||
jz @f
|
||||
movzx ecx, al
|
||||
@@:
|
||||
rep movsb
|
||||
lodsb
|
||||
test al, al
|
||||
jz @f
|
||||
movzx ecx, al
|
||||
mov al, '.'
|
||||
stosb
|
||||
jmp @r
|
||||
@@:
|
||||
stosb
|
||||
|
||||
push buffer_ptr
|
||||
call [con_write_asciiz]
|
||||
|
||||
push str7
|
||||
call [con_write_asciiz]
|
||||
|
||||
ret
|
||||
|
||||
.fail:
|
||||
ret
|
||||
|
||||
|
||||
|
||||
; input: eax - number
|
||||
; edi - ptr
|
||||
byte_to_ascii:
|
||||
|
||||
push ebx ecx edx
|
||||
|
||||
xor edx, edx ; result
|
||||
xor ecx, ecx ; byte count
|
||||
inc ecx
|
||||
mov bl, 10 ; divisor
|
||||
|
||||
div bl
|
||||
mov dl, ah
|
||||
add dl, '0'
|
||||
and ax, 0x00ff
|
||||
jz .ok
|
||||
|
||||
inc ecx
|
||||
shl edx, 8
|
||||
|
||||
div bl
|
||||
mov dl, ah
|
||||
add dl, '0'
|
||||
and ax, 0x00ff
|
||||
jz .ok
|
||||
|
||||
inc ecx
|
||||
shl edx, 8
|
||||
|
||||
mov dl, al
|
||||
add dl, '0'
|
||||
|
||||
.ok:
|
||||
shl edx, 8
|
||||
mov dl, cl
|
||||
mov [edi], edx
|
||||
add edi, ecx
|
||||
inc edi
|
||||
|
||||
pop edx ecx ebx
|
||||
ret
|
||||
|
||||
|
||||
; data
|
||||
title db 'Trace route',0
|
||||
str_welcome db 'Please enter the hostname or IP-address of the host you want to trace,',10
|
||||
db 'or just press enter to exit.',10,10,0
|
||||
str_prompt db 10,'> ',0
|
||||
str3 db 'Tracing route to ',0
|
||||
|
||||
str4 db 10,0
|
||||
str7 db ' ', 0
|
||||
str5 db 'Name resolution failed.',10,0
|
||||
str6 db 'Socket error.',10,0
|
||||
str13 db 'Invalid parameter(s)',10,0
|
||||
|
||||
str9 db '%u ',0
|
||||
str1 db '%u.%u ms ',0
|
||||
str2 db '[%u.%u.%u.%u]',10,0
|
||||
str10 db 'Invalid reply',10,0
|
||||
str8 db 'Timeout!',10,0
|
||||
|
||||
|
||||
sockaddr1:
|
||||
dw AF_INET4
|
||||
.port dw 666
|
||||
.ip dd 0
|
||||
rb 10
|
||||
|
||||
sockaddr2:
|
||||
dw AF_INET4
|
||||
.port dw 53 shl 8 ; DNS port
|
||||
.ip dd 0x08080808 ; Google DNS
|
||||
rb 10
|
||||
|
||||
time_reference dd ?
|
||||
ip_ptr dd ?
|
||||
ttl dd ?
|
||||
timeout dd 500
|
||||
recvd dd ? ; received number of bytes in last packet
|
||||
|
||||
; import
|
||||
align 4
|
||||
@IMPORT:
|
||||
|
||||
library console, 'console.obj', \
|
||||
network, 'network.obj'
|
||||
|
||||
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'
|
||||
|
||||
import network, \
|
||||
getaddrinfo, 'getaddrinfo', \
|
||||
freeaddrinfo, 'freeaddrinfo', \
|
||||
inet_ntoa, 'inet_ntoa'
|
||||
|
||||
include_debug_strings
|
||||
|
||||
icmp_socket dd ?
|
||||
udp_socket dd ?
|
||||
dns_socket dd ?
|
||||
|
||||
udp_packet db 'hello!'
|
||||
|
||||
dns_tr:
|
||||
db 7,'in-addr',4,'arpa',0
|
||||
dw 0x0C00 ; Qtype: PTR
|
||||
dw 0x0100 ; Class: IN
|
||||
|
||||
.length = $ - dns_tr
|
||||
|
||||
dns_pkt:
|
||||
dw 0x9A02 ; Transaction ID
|
||||
dw 0x0001 ; Flags: Recursive desired
|
||||
dw 0x0100 ; Questions
|
||||
dw 0x0000 ; Answers
|
||||
dw 0x0000 ; Authority RR
|
||||
dw 0x0000 ; Additional RR
|
||||
.name rb 512
|
||||
|
||||
I_END:
|
||||
|
||||
params rb 1024
|
||||
buffer_ptr: rb BUFFERSIZE
|
||||
|
||||
IM_END:
|
||||
Loading…
Reference in New Issue
Block a user