87 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			Factor
		
	
	
			
		
		
	
	
			87 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			Factor
		
	
	
! Copyright (C) 2007 Doug Coleman.
 | 
						|
! See http://factorcode.org/license.txt for BSD license.
 | 
						|
USING: alien alien.c-types alien.syntax byte-arrays io
 | 
						|
io.sockets kernel structs math math.parser
 | 
						|
prettyprint sequences ;
 | 
						|
IN: io.sockets.headers
 | 
						|
 | 
						|
C-STRUCT: etherneth
 | 
						|
    { { "char" 6 } "dmac" }
 | 
						|
    { { "char" 6 } "smac" }
 | 
						|
    { "ushort" "type" } ;
 | 
						|
 | 
						|
: >mac-address ( byte-array -- string )
 | 
						|
    6 memory>byte-array
 | 
						|
    [ >hex 2 48 pad-left ] { } map-as ":" join ;
 | 
						|
 | 
						|
: etherneth. ( etherneth -- )
 | 
						|
    [ etherneth-dmac "Dest   MAC: " write >mac-address . ] keep
 | 
						|
    [ etherneth-smac "Source MAC: " write >mac-address . ] keep
 | 
						|
    [ etherneth-type "Type      : " write .h ] keep
 | 
						|
    drop ;
 | 
						|
 | 
						|
C-STRUCT: iph
 | 
						|
    { "uchar" "hl|v" } ! hl is 4 bits, v is 4 bits
 | 
						|
    { "uchar" "tos" }
 | 
						|
    { "short" "len" }
 | 
						|
    { "short" "id" }
 | 
						|
    { "short" "off" }
 | 
						|
    { "uchar" "ttl" }
 | 
						|
    { "uchar" "p" }
 | 
						|
    { "ushort" "ip_sum" }
 | 
						|
    { "uint" "ip_src" }
 | 
						|
    { "uint" "ip_dst" } ;
 | 
						|
 | 
						|
: iph-hl ( iph -- n )
 | 
						|
    iph-hl|v -4 shift ;
 | 
						|
 | 
						|
: iph-v ( iph -- n )
 | 
						|
    iph-hl|v 0x0f bitand ;
 | 
						|
 | 
						|
: set-iph-hl ( n iph -- )
 | 
						|
    [ iph-hl|v 0x0f bitand >r 4 shift r> bitor ] keep
 | 
						|
    set-iph-hl|v ;
 | 
						|
 | 
						|
: set-iph-v ( n iph -- )
 | 
						|
    [ iph-hl|v 0xf0 bitand bitor ] keep
 | 
						|
    set-iph-hl|v ;
 | 
						|
 | 
						|
C-STRUCT: icmph
 | 
						|
    { "uchar" "type" }
 | 
						|
    { "uchar" "code" }
 | 
						|
    { "short" "chksum" }
 | 
						|
    { "ushort" "id" }
 | 
						|
    { "ushort" "seq" } ;
 | 
						|
 | 
						|
C-STRUCT: udph
 | 
						|
    { "ushort" "sport" }
 | 
						|
    { "ushort" "dport" }
 | 
						|
    { "ushort" "len" }
 | 
						|
    { "ushort" "check" } ;
 | 
						|
 | 
						|
C-STRUCT: tcph
 | 
						|
    { "ushort" "sport" }
 | 
						|
    { "ushort" "dport" }
 | 
						|
    { "uint" "seq" }
 | 
						|
    { "uint" "ack" }
 | 
						|
    { "uchar" "x2|off" }
 | 
						|
    { "uchar" "flags" }
 | 
						|
    { "ushort" "win" }
 | 
						|
    { "ushort" "sum" }
 | 
						|
    { "ushort" "urp" } ;
 | 
						|
 | 
						|
: tcph-x2 ( iph -- n )
 | 
						|
    tcph-x2|off -4 shift ;
 | 
						|
 | 
						|
: tcph-off ( iph -- n )
 | 
						|
    tcph-x2|off 0x0f bitand ;
 | 
						|
 | 
						|
: set-tcph-x2 ( n iph -- )
 | 
						|
    [ tcph-x2|off 0x0f bitand >r 4 shift r> bitor ] keep
 | 
						|
    set-tcph-x2|off ;
 | 
						|
 | 
						|
: set-tcph-off ( n iph -- )
 | 
						|
    [ tcph-x2|off 0x0f bitand bitor ] keep
 | 
						|
    set-tcph-x2|off ;
 | 
						|
 |