From c4aafe05d919dedca66869b0cf78866db2da6956 Mon Sep 17 00:00:00 2001
From: Ian Jackson <ijackson@chiark.greenend.org.uk>
Date: Sat, 12 Oct 2019 20:40:55 +0100
Subject: [PATCH] test: Proxy udp packets

We must change the config to specify localhost addrs explicitly,
because we don't implement any special logic for IN[6]ADDR_ANY.

Signed-off-by: Ian Jackson <ijackson@chiark.greenend.org.uk>
---
 test/invoke | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/test/invoke b/test/invoke
index 6678bb6..22554cf 100755
--- a/test/invoke
+++ b/test/invoke
@@ -3,6 +3,7 @@
 package require Tclx
 
 load chiark_tcl_hbytes-1.so
+load chiark_tcl_dgram-1.so
 
 set netlink(inside) {
     local-address "172.18.232.9";
@@ -66,6 +67,7 @@ exec cat
 	append cfg "$delim
 	    udp {
                 port $port;
+                address \"::1\", \"127.0.0.1\";
 		buffer sysbuffer(4096);
 	    }
 	"
@@ -157,7 +159,42 @@ proc sendpkt {} {
 
 file mkdir test/tmp
 set tmp test/tmp
+set socktmp $tmp
+regsub {^(?!/)} $socktmp {./} socktmp ;# dgram-socket wants ./ or /
 
+proc udp-proxy {} {
+    global socktmp udpsock
+    set u $socktmp/udp
+    file delete $u
+    regsub {^(?!/)} $u {./} u
+    set udpsock [dgram-socket create $u]
+    dgram-socket on-receive $udpsock udp-relay
+}
+
+proc udp-relay {data src sock args} {
+    global udpsock socktmp
+    set headerlen [expr {52+1}]
+    set orgsrc $src
+
+    set dst [hbytes range $data 0 $headerlen]
+    regsub {(?:00)*$} $dst {} dst
+    set dst [hbytes h2raw $dst]
+
+    hbytes overwrite data 0 [hbytes zeroes $headerlen]
+    regsub {.*/} $src {} src
+    set srch [hbytes raw2h $src]
+    hbytes append srch 00
+    if {[catch {
+	if {[regexp {[^.,:0-9a-f]} $dst c]} { error "bad dst" }
+	if {[hbytes length $srch] > $headerlen} { error "src addr too long" }
+	hbytes overwrite data 0 $srch
+	dgram-socket transmit $udpsock $data $socktmp/$dst
+    } emsg]} {
+	puts stderr "$orgsrc -> $dst: $emsg"
+    }
+}
+
+udp-proxy
 spawn-secnet inside
 spawn-secnet outside
 
-- 
2.30.2