X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ian/git?a=blobdiff_plain;f=stest%2Fcommon.tcl;h=328cb016a6b8b15b3759e5c44c16cd46c3f9547b;hb=98a99d6c348383fafa223cad17410f8918d28e3c;hp=21ed960138f1b9a99cb98e388d6eba2cc2d20c63;hpb=e4b227ddf4c1ba709a5a364f826ca046f712b417;p=secnet.git diff --git a/stest/common.tcl b/stest/common.tcl index 21ed960..328cb01 100644 --- a/stest/common.tcl +++ b/stest/common.tcl @@ -25,22 +25,23 @@ set extra(inside) { } set extra(outside) {} -proc mkconf {which} { +proc mkconf {location site} { global tmp global builddir global netlink global ports global extra global netlinkfh - set pipefp $tmp/$which.netlink + set pipefp $tmp/$site.netlink foreach tr {t r} { file delete $pipefp.$tr exec mkfifo -m600 $pipefp.$tr - set netlinkfh($which.$tr) [set fh [open $pipefp.$tr r+]] + set netlinkfh($site.$tr) [set fh [open $pipefp.$tr r+]] fconfigure $fh -blocking 0 -buffering none -translation binary } - fileevent $netlinkfh($which.r) readable [list netlink-readable $which] - set fakeuf $tmp/$which.fake-userv + fileevent $netlinkfh($site.r) readable \ + [list netlink-readable $location $site] + set fakeuf $tmp/$site.fake-userv set fakeuh [open $fakeuf w 0755] puts $fakeuh "#!/bin/sh set -e @@ -53,18 +54,19 @@ exec cat " close $fakeuh set cfg " + hash sha1; netlink userv-ipif { name \"netlink\"; userv-path \"$fakeuf\"; - $netlink($which) + $netlink($site) mtu 1400; buffer sysbuffer(2048); - interface \"secnet-test-[string range $which 0 0]\"; + interface \"secnet-test-[string range $site 0 0]\"; }; comm " set delim {} - foreach port $ports($which) { + foreach port $ports($site) { append cfg "$delim udp { port $port; @@ -75,15 +77,17 @@ exec cat set delim , } append cfg "; - local-name \"test-example/$which/$which\"; - local-key rsa-private(\"$builddir/test-example/$which.key\"); + local-name \"test-example/$location/$site\"; + local-key rsa-private(\"$builddir/test-example/$site.key\"); " - append cfg $extra($which) - append cfg { + append cfg $extra($site) + append cfg " log logfile { - filename "/dev/tty"; - class "info","notice","warning","error","security","fatal"; + prefix \"$site\"; + class \"debug\",\"info\",\"notice\",\"warning\",\"error\",\"security\",\"fatal\"; }; + " + append cfg { system { }; resolver adns { @@ -93,42 +97,108 @@ exec cat transform eax-serpent { }, serpent256-cbc { }; } + set pubkeys $tmp/$site.pubkeys + file delete -force $pubkeys + exec cp -rl $builddir/test-example/pubkeys $pubkeys + set f [open $builddir/test-example/sites.conf r] + while {[gets $f l] >= 0} { + regsub {\"[^\"]*test-example/pubkeys/} $l "\"$pubkeys/" l + append cfg $l "\n" + } set sites [read $f] close $f append cfg $sites append cfg { - sites map(site,vpn/test-example/all-sites); + sites map(site,all-sites); } + return $cfg } -proc spawn-secnet {which} { +proc spawn-secnet {location site} { global tmp global builddir global netlinkfh - upvar #0 pids($which) pid - set cf $tmp/$which.conf + global env + global pidmap + global readbuf + upvar #0 pids($site) pid + set readbuf($site) {} + set cf $tmp/$site.conf set ch [open $cf w] - puts $ch [mkconf $which] + puts $ch [mkconf $location $site] close $ch set argl [list $builddir/secnet -dvnc $cf] - set pid [fork] - if {!$pid} { - execl [lindex $argl 0] [lrange $argl 1 end] + set divertk SECNET_STEST_DIVERT_$site + puts -nonewline "spawn" + foreach k [array names env] { + switch -glob $k { + SECNET_STEST_DIVERT_* - + SECNET_TEST_BUILDDIR { } + *SECNET* - + *PRELOAD* { puts -nonewline " $k=$env($k)" } + } + } + puts " $argl" + if {[info exists env($divertk)]} { + switch -glob $env($divertk) { + i { + puts -nonewline "run ^ command, hit return " + flush stdout + gets stdin + set argl {} + } + 0 - "" { + } + * { + set argl [split $env($divertk)] + } + } + } + if {[llength $argl]} { + set pid [fork] + set pidmap($pid) "secnet $location/$site" + if {!$pid} { + execl [lindex $argl 0] [lrange $argl 1 end] + } } - puts -nonewline $netlinkfh($which.t) [hbytes h2raw c0] + puts -nonewline $netlinkfh($site.t) [hbytes h2raw c0] } -proc netlink-readable {which} { +proc netlink-readable {location site} { global ok - upvar #0 netlinkfh($which.r) fh - read $fh; # empty the buffer - switch -exact $which { + upvar #0 readbuf($site) buf + upvar #0 netlinkfh($site.r) fh + while 1 { + set x [read $fh] + set h [hbytes raw2h $x] + if {![hbytes length $h]} return + append buf $h + #puts "READABLE $site buf=$buf" + while {[regexp {^((?:..)*?)c0(.*)$} $buf dummy now buf]} { + #puts "READABLE $site now=$now (buf=$buf)" + regsub -all {^((?:..)*?)dbdc} $now {\1c0} now + regsub -all {^((?:..)*?)dbdd} $now {\1db} now + puts "netlink-got-packet $location $site $now" + netlink-got-packet $location $site $now + } + } +} + +proc netlink-got-packet {location site data} { + if {![hbytes length $data]} return + switch -exact $site { inside { - puts OK - set ok 1; # what a bodge - return + switch -glob $data { + 45000054ed9d4000fe0166d9ac12e802ac12e80900* { + puts "OK $data" + finish 0 + } + * { + error "unexpected $site $data" + } + } } outside { error "inside rx'd!" @@ -148,7 +218,7 @@ $message ---------------------------------------- " } - exit 1 + finish 1 } proc sendpkt {} { @@ -167,19 +237,38 @@ proc sendpkt {} { set socktmp $tmp/s exec mkdir -p -m700 $socktmp -regsub {^(?!/)} $socktmp {./} socktmp ;# dgram-socket wants ./ or / +regsub {^(?!/|\./)} $socktmp {./} socktmp ;# dgram-socket wants ./ or / -proc prefix_preload {lib} { - global env - set l {} - catch { set l [split $env(PRELOAD) :] } - set l [concat [list $lib] $l] - set env(LD_PRELOAD) [join $l :] -} +proc prefix_preload {lib} { prefix_some_path LD_PRELOAD $lib } set env(UDP_PRELOAD_DIR) $socktmp prefix_preload $builddir/stest/udp-preload.so +proc finish {estatus} { + puts stderr "FINISHING $estatus" + signal default SIGCHLD + global pidmap + foreach pid [array names pidmap] { + kill KILL $pid + } + exit $estatus +} + +proc reap {} { + global pidmap + #puts stderr REAPING + foreach pid [array names pidmap] { + set got [wait -nohang $pid] + if {![llength $got]} continue + set info $pidmap($pid) + unset pidmap($pid) + puts stderr "reaped $info: $got" + finish 1 + } +} + +signal -restart trap SIGCHLD { after idle reap } + proc udp-proxy {} { global socktmp udpsock set u $socktmp/udp @@ -214,8 +303,8 @@ proc udp-relay {data src sock args} { proc test-kex {} { udp-proxy - spawn-secnet inside - spawn-secnet outside + spawn-secnet in inside + spawn-secnet out outside after 500 sendpkt after 1000 sendpkt