Artifact
2ba4c91b9e7d6c3c533bb1abc6f07ae500f53f85f915bacf6fe04e8de1fc6521:
0000: 23 21 2f 75 73 72 2f 62 69 6e 2f 65 6e 76 20 74 #!/usr/bin/env t
0010: 63 6c 73 68 0a 0a 70 61 63 6b 61 67 65 20 72 65 clsh..package re
0020: 71 75 69 72 65 20 74 6c 73 0a 0a 73 65 74 20 64 quire tls..set d
0030: 69 72 09 09 09 5b 66 69 6c 65 20 6a 6f 69 6e 20 ir...[file join
0040: 5b 66 69 6c 65 20 64 69 72 6e 61 6d 65 20 5b 69 [file dirname [i
0050: 6e 66 6f 20 73 63 72 69 70 74 5d 5d 20 2e 2e 2f nfo script]] ../
0060: 74 65 73 74 73 2f 63 65 72 74 73 5d 0a 73 65 74 tests/certs].set
0070: 20 4f 50 54 53 28 2d 63 61 66 69 6c 65 29 09 5b OPTS(-cafile).[
0080: 66 69 6c 65 20 6a 6f 69 6e 20 24 64 69 72 20 63 file join $dir c
0090: 61 2e 70 65 6d 5d 0a 73 65 74 20 4f 50 54 53 28 a.pem].set OPTS(
00a0: 2d 63 65 72 74 29 09 09 5b 66 69 6c 65 20 6a 6f -cert)..[file jo
00b0: 69 6e 20 24 64 69 72 20 63 6c 69 65 6e 74 2e 70 in $dir client.p
00c0: 65 6d 5d 0a 73 65 74 20 4f 50 54 53 28 2d 6b 65 em].set OPTS(-ke
00d0: 79 29 09 09 5b 66 69 6c 65 20 6a 6f 69 6e 20 24 y)..[file join $
00e0: 64 69 72 20 63 6c 69 65 6e 74 2e 6b 65 79 5d 0a dir client.key].
00f0: 0a 73 65 74 20 4f 50 54 53 28 2d 68 6f 73 74 29 .set OPTS(-host)
0100: 09 09 6c 6f 72 61 78 0a 73 65 74 20 4f 50 54 53 ..lorax.set OPTS
0110: 28 2d 70 6f 72 74 29 09 09 32 34 36 38 0a 73 65 (-port)..2468.se
0120: 74 20 4f 50 54 53 28 2d 64 65 62 75 67 29 09 31 t OPTS(-debug).1
0130: 0a 73 65 74 20 4f 50 54 53 28 2d 63 6f 75 6e 74 .set OPTS(-count
0140: 29 09 38 0a 73 65 74 20 4f 50 54 53 28 2d 70 61 ).8.set OPTS(-pa
0150: 72 61 6c 6c 65 6c 29 09 31 0a 73 65 74 20 4f 50 rallel).1.set OP
0160: 54 53 28 2d 72 65 71 75 69 72 65 29 09 30 0a 0a TS(-require).0..
0170: 66 6f 72 65 61 63 68 20 7b 6b 65 79 20 76 61 6c foreach {key val
0180: 7d 20 24 61 72 67 76 20 7b 0a 20 20 20 20 69 66 } $argv {. if
0190: 20 7b 21 5b 69 6e 66 6f 20 65 78 69 73 74 73 20 {![info exists
01a0: 4f 50 54 53 28 24 6b 65 79 29 5d 7d 20 7b 0a 09 OPTS($key)]} {..
01b0: 70 75 74 73 20 73 74 64 65 72 72 20 22 55 73 61 puts stderr "Usa
01c0: 67 65 3a 20 24 61 72 67 76 30 20 3f 6f 70 74 69 ge: $argv0 ?opti
01d0: 6f 6e 73 3f 5c 0a 09 09 5c 6e 5c 74 2d 64 65 62 ons?\...\n\t-deb
01e0: 75 67 20 20 20 20 20 62 6f 6f 6c 65 61 6e 20 20 ug boolean
01f0: 20 44 65 62 75 67 67 69 6e 67 20 6f 6e 20 6f 72 Debugging on or
0200: 20 6f 66 66 20 28 24 4f 50 54 53 28 2d 64 65 62 off ($OPTS(-deb
0210: 75 67 29 29 5c 0a 09 09 5c 6e 5c 74 2d 63 61 66 ug))\...\n\t-caf
0220: 69 6c 65 20 20 20 20 66 69 6c 65 20 20 20 20 20 ile file
0230: 20 43 65 72 74 2e 20 41 75 74 68 2e 20 46 69 6c Cert. Auth. Fil
0240: 65 20 28 24 4f 50 54 53 28 2d 63 61 66 69 6c 65 e ($OPTS(-cafile
0250: 29 29 5c 0a 09 09 5c 6e 5c 74 2d 63 6c 69 65 6e ))\...\n\t-clien
0260: 74 20 20 20 20 66 69 6c 65 20 20 20 20 20 20 43 t file C
0270: 6c 69 65 6e 74 20 43 65 72 74 20 28 24 4f 50 54 lient Cert ($OPT
0280: 53 28 2d 63 65 72 74 29 29 5c 0a 09 09 5c 6e 5c S(-cert))\...\n\
0290: 74 2d 63 6b 65 79 20 20 20 20 20 20 66 69 6c 65 t-ckey file
02a0: 20 20 20 20 20 20 43 6c 69 65 6e 74 20 4b 65 79 Client Key
02b0: 20 28 24 4f 50 54 53 28 2d 6b 65 79 29 29 5c 0a ($OPTS(-key))\.
02c0: 09 09 5c 6e 5c 74 2d 63 6f 75 6e 74 20 20 20 20 ..\n\t-count
02d0: 20 6e 75 6d 20 20 20 20 20 20 20 4e 6f 20 6f 66 num No of
02e0: 20 73 79 6e 63 2e 20 63 6f 6e 6e 65 63 74 69 6f sync. connectio
02f0: 6e 73 20 74 6f 20 6d 61 6b 65 20 70 65 72 20 63 ns to make per c
0300: 6c 69 65 6e 74 20 28 24 4f 50 54 53 28 2d 63 6f lient ($OPTS(-co
0310: 75 6e 74 29 29 5c 0a 09 09 5c 6e 5c 74 2d 70 61 unt))\...\n\t-pa
0320: 72 61 6c 6c 65 6c 20 20 6e 75 6d 20 20 20 20 20 rallel num
0330: 20 20 4e 6f 20 6f 66 20 70 61 72 61 6c 6c 65 6c No of parallel
0340: 20 63 6c 69 65 6e 74 73 20 74 6f 20 72 75 6e 20 clients to run
0350: 28 24 4f 50 54 53 28 2d 70 61 72 61 6c 6c 65 6c ($OPTS(-parallel
0360: 29 29 5c 0a 09 09 5c 6e 5c 74 2d 68 6f 73 74 20 ))\...\n\t-host
0370: 20 20 20 20 20 68 6f 73 74 6e 61 6d 65 20 20 53 hostname S
0380: 65 72 76 65 72 20 68 6f 73 74 6e 61 6d 65 20 28 erver hostname (
0390: 24 4f 50 54 53 28 2d 68 6f 73 74 29 29 5c 0a 09 $OPTS(-host))\..
03a0: 09 5c 6e 5c 74 2d 70 6f 72 74 20 20 20 20 20 20 .\n\t-port
03b0: 6e 75 6d 20 20 20 20 20 20 20 53 65 72 76 65 72 num Server
03c0: 20 70 6f 72 74 20 28 24 4f 50 54 53 28 2d 70 6f port ($OPTS(-po
03d0: 72 74 29 29 5c 0a 09 09 5c 6e 5c 74 2d 72 65 71 rt))\...\n\t-req
03e0: 75 69 72 65 20 20 20 62 6f 6f 6c 65 61 6e 20 20 uire boolean
03f0: 20 52 65 71 75 69 72 65 20 43 65 72 74 69 66 69 Require Certifi
0400: 63 61 74 65 20 28 24 4f 50 54 53 28 2d 72 65 71 cate ($OPTS(-req
0410: 75 69 72 65 29 29 22 0a 09 65 78 69 74 0a 20 20 uire))"..exit.
0420: 20 20 7d 0a 20 20 20 20 73 65 74 20 4f 50 54 53 }. set OPTS
0430: 28 24 6b 65 79 29 20 24 76 61 6c 0a 7d 0a 0a 69 ($key) $val.}..i
0440: 66 20 7b 24 4f 50 54 53 28 2d 70 61 72 61 6c 6c f {$OPTS(-parall
0450: 65 6c 29 20 3e 20 31 7d 20 7b 0a 20 20 20 20 23 el) > 1} {. #
0460: 20 49 66 20 74 68 65 79 20 77 61 6e 74 65 64 20 If they wanted
0470: 70 61 72 61 6c 6c 65 6c 2c 20 77 65 20 6a 75 73 parallel, we jus
0480: 74 20 73 70 61 77 6e 20 6f 75 72 73 65 6c 76 65 t spawn ourselve
0490: 73 20 73 65 76 65 72 61 6c 20 74 69 6d 65 73 0a s several times.
04a0: 20 20 20 20 23 20 77 69 74 68 20 74 68 65 20 72 # with the r
04b0: 69 67 68 74 20 61 72 67 73 2e 0a 0a 20 20 20 20 ight args...
04c0: 73 65 74 20 63 6d 64 09 5b 69 6e 66 6f 20 6e 61 set cmd.[info na
04d0: 6d 65 6f 66 65 78 65 63 75 74 61 62 6c 65 5d 0a meofexecutable].
04e0: 20 20 20 20 73 65 74 20 73 63 72 69 70 74 09 5b set script.[
04f0: 69 6e 66 6f 20 73 63 72 69 70 74 5d 0a 20 20 20 info script].
0500: 20 66 6f 72 20 7b 73 65 74 20 69 20 30 7d 20 7b for {set i 0} {
0510: 24 69 20 3c 20 24 4f 50 54 53 28 2d 70 61 72 61 $i < $OPTS(-para
0520: 6c 6c 65 6c 29 7d 20 7b 69 6e 63 72 20 69 7d 20 llel)} {incr i}
0530: 7b 0a 09 65 76 61 6c 20 5b 6c 69 73 74 20 65 78 {..eval [list ex
0540: 65 63 20 24 63 6d 64 20 24 73 63 72 69 70 74 5d ec $cmd $script]
0550: 20 5b 61 72 72 61 79 20 67 65 74 20 4f 50 54 53 [array get OPTS
0560: 5d 20 5b 6c 69 73 74 20 2d 70 61 72 61 6c 6c 65 ] [list -paralle
0570: 6c 20 30 5d 20 26 0a 20 20 20 20 7d 0a 20 20 20 l 0] &. }.
0580: 20 65 78 69 74 0a 7d 0a 0a 23 20 4c 6f 63 61 6c exit.}..# Local
0590: 20 68 61 6e 64 6c 65 72 20 66 6f 72 20 61 6e 79 handler for any
05a0: 20 62 61 63 6b 67 72 6f 75 6e 64 20 65 72 72 6f background erro
05b0: 72 73 2e 0a 70 72 6f 63 20 62 67 65 72 72 6f 72 rs..proc bgerror
05c0: 20 7b 6d 73 67 7d 20 7b 20 70 75 74 73 20 22 42 {msg} { puts "B
05d0: 47 45 52 52 4f 52 3a 20 24 6d 73 67 22 20 7d 0a GERROR: $msg" }.
05e0: 0a 23 20 64 65 62 75 67 67 69 6e 67 20 68 65 6c .# debugging hel
05f0: 70 65 72 20 63 6f 64 65 0a 70 72 6f 63 20 73 68 per code.proc sh
0600: 6f 72 74 73 74 72 20 7b 73 74 72 7d 20 7b 0a 20 ortstr {str} {.
0610: 20 20 20 72 65 74 75 72 6e 20 22 5b 73 74 72 69 return "[stri
0620: 6e 67 20 72 65 70 6c 61 63 65 20 24 73 74 72 20 ng replace $str
0630: 31 30 20 65 6e 64 20 2e 2e 2e 5d 20 5b 73 74 72 10 end ...] [str
0640: 69 6e 67 20 6c 65 6e 67 74 68 20 24 73 74 72 5d ing length $str]
0650: 62 22 0a 7d 0a 70 72 6f 63 20 64 70 75 74 73 20 b".}.proc dputs
0660: 7b 6d 73 67 7d 20 7b 20 69 66 20 7b 24 3a 3a 4f {msg} { if {$::O
0670: 50 54 53 28 2d 64 65 62 75 67 29 7d 20 7b 20 70 PTS(-debug)} { p
0680: 75 74 73 20 73 74 64 65 72 72 20 24 6d 73 67 20 uts stderr $msg
0690: 3b 20 66 6c 75 73 68 20 73 74 64 65 72 72 20 7d ; flush stderr }
06a0: 20 7d 0a 0a 73 65 74 20 4f 50 54 53 28 6f 70 65 }..set OPTS(ope
06b0: 6e 70 6f 72 74 73 29 09 30 0a 0a 23 20 44 65 66 nports).0..# Def
06c0: 69 6e 65 20 77 68 61 74 20 77 65 20 77 61 6e 74 ine what we want
06d0: 20 74 6f 20 66 65 65 64 20 64 6f 77 6e 20 74 68 to feed down th
06e0: 65 20 70 69 70 65 0a 73 65 74 20 6d 65 67 61 64 e pipe.set megad
06f0: 61 74 61 20 5b 73 74 72 69 6e 67 20 72 65 70 65 ata [string repe
0700: 61 74 20 5b 73 74 72 69 6e 67 20 72 65 70 65 61 at [string repea
0710: 74 20 41 20 37 36 5d 5c 6e 20 31 30 30 30 5d 0a t A 76]\n 1000].
0720: 0a 70 72 6f 63 20 64 72 61 69 6e 20 7b 63 68 61 .proc drain {cha
0730: 6e 7d 20 7b 0a 20 20 20 20 67 6c 6f 62 61 6c 20 n} {. global
0740: 4f 50 54 53 0a 20 20 20 20 69 66 20 7b 5b 63 61 OPTS. if {[ca
0750: 74 63 68 20 7b 72 65 61 64 20 24 63 68 61 6e 7d tch {read $chan}
0760: 20 64 61 74 61 5d 7d 20 7b 0a 09 23 64 70 75 74 data]} {..#dput
0770: 73 20 22 45 4f 46 20 24 63 68 61 6e 20 28 5b 73 s "EOF $chan ([s
0780: 68 6f 72 74 73 74 72 20 24 64 61 74 61 5d 29 22 hortstr $data])"
0790: 0a 09 69 6e 63 72 20 4f 50 54 53 28 6f 70 65 6e ..incr OPTS(open
07a0: 70 6f 72 74 73 29 20 2d 31 0a 09 63 61 74 63 68 ports) -1..catch
07b0: 20 7b 63 6c 6f 73 65 20 24 63 68 61 6e 7d 0a 09 {close $chan}..
07c0: 72 65 74 75 72 6e 0a 20 20 20 20 7d 0a 20 20 20 return. }.
07d0: 20 23 69 66 20 7b 24 64 61 74 61 20 6e 65 20 22 #if {$data ne "
07e0: 22 7d 20 7b 20 64 70 75 74 73 20 22 67 6f 74 20 "} { dputs "got
07f0: 24 63 68 61 6e 20 28 5b 73 68 6f 72 74 73 74 72 $chan ([shortstr
0800: 20 24 64 61 74 61 5d 29 22 20 7d 0a 20 20 20 20 $data])" }.
0810: 69 66 20 7b 5b 73 74 72 69 6e 67 20 6d 61 74 63 if {[string matc
0820: 68 20 2a 43 4c 4f 53 45 5c 6e 20 24 64 61 74 61 h *CLOSE\n $data
0830: 5d 7d 20 7b 0a 09 64 70 75 74 73 20 22 43 4c 4f ]} {..dputs "CLO
0840: 53 45 20 24 63 68 61 6e 22 0a 09 69 6e 63 72 20 SE $chan"..incr
0850: 4f 50 54 53 28 6f 70 65 6e 70 6f 72 74 73 29 20 OPTS(openports)
0860: 2d 31 0a 09 63 6c 6f 73 65 20 24 63 68 61 6e 0a -1..close $chan.
0870: 09 72 65 74 75 72 6e 0a 20 20 20 20 7d 20 65 6c .return. } el
0880: 73 65 69 66 20 7b 5b 65 6f 66 20 24 63 68 61 6e seif {[eof $chan
0890: 5d 7d 20 7b 0a 09 23 20 63 6c 69 65 6e 74 20 67 ]} {..# client g
08a0: 6f 6e 65 20 6f 72 20 66 69 6e 69 73 68 65 64 0a one or finished.
08b0: 09 64 70 75 74 73 20 22 45 4f 46 20 24 63 68 61 .dputs "EOF $cha
08c0: 6e 22 0a 09 69 6e 63 72 20 4f 50 54 53 28 6f 70 n"..incr OPTS(op
08d0: 65 6e 70 6f 72 74 73 29 20 2d 31 0a 09 63 6c 6f enports) -1..clo
08e0: 73 65 20 24 63 68 61 6e 0a 09 72 65 74 75 72 6e se $chan..return
08f0: 0a 20 20 20 20 7d 0a 7d 0a 0a 70 72 6f 63 20 66 . }.}..proc f
0900: 65 65 64 20 7b 73 6f 63 6b 7d 20 7b 0a 20 20 20 eed {sock} {.
0910: 20 64 70 75 74 73 20 22 66 65 65 64 20 24 73 6f dputs "feed $so
0920: 63 6b 20 28 5b 73 68 6f 72 74 73 74 72 20 24 3a ck ([shortstr $:
0930: 3a 6d 65 67 61 64 61 74 61 5d 29 22 0a 20 20 20 :megadata])".
0940: 20 70 75 74 73 20 24 73 6f 63 6b 20 24 3a 3a 6d puts $sock $::m
0950: 65 67 61 64 61 74 61 0a 20 20 20 20 66 6c 75 73 egadata. flus
0960: 68 20 24 73 6f 63 6b 0a 20 20 20 20 70 75 74 73 h $sock. puts
0970: 20 24 73 6f 63 6b 20 43 4c 4f 53 45 0a 20 20 20 $sock CLOSE.
0980: 20 66 6c 75 73 68 20 24 73 6f 63 6b 0a 20 20 20 flush $sock.
0990: 20 66 69 6c 65 65 76 65 6e 74 20 24 73 6f 63 6b fileevent $sock
09a0: 20 77 72 69 74 61 62 6c 65 20 7b 7d 0a 7d 0a 0a writable {}.}..
09b0: 70 72 6f 63 20 67 6f 20 7b 7d 20 7b 0a 20 20 20 proc go {} {.
09c0: 20 67 6c 6f 62 61 6c 20 4f 50 54 53 0a 20 20 20 global OPTS.
09d0: 20 66 6f 72 20 7b 73 65 74 20 6e 75 6d 20 24 4f for {set num $O
09e0: 50 54 53 28 2d 63 6f 75 6e 74 29 7d 20 7b 24 6e PTS(-count)} {$n
09f0: 75 6d 20 3e 20 30 7d 20 7b 69 6e 63 72 20 6e 75 um > 0} {incr nu
0a00: 6d 20 2d 31 7d 20 7b 0a 09 73 65 74 20 73 6f 63 m -1} {..set soc
0a10: 6b 20 5b 74 6c 73 3a 3a 73 6f 63 6b 65 74 20 24 k [tls::socket $
0a20: 4f 50 54 53 28 2d 68 6f 73 74 29 20 24 4f 50 54 OPTS(-host) $OPT
0a30: 53 28 2d 70 6f 72 74 29 5d 0a 09 69 6e 63 72 20 S(-port)]..incr
0a40: 4f 50 54 53 28 6f 70 65 6e 70 6f 72 74 73 29 0a OPTS(openports).
0a50: 09 66 63 6f 6e 66 69 67 75 72 65 20 24 73 6f 63 .fconfigure $soc
0a60: 6b 20 2d 62 6c 6f 63 6b 69 6e 67 20 30 20 2d 62 k -blocking 0 -b
0a70: 75 66 66 65 72 73 69 7a 65 20 34 30 39 36 0a 09 uffersize 4096..
0a80: 66 69 6c 65 65 76 65 6e 74 20 24 73 6f 63 6b 20 fileevent $sock
0a90: 77 72 69 74 61 62 6c 65 20 5b 6c 69 73 74 20 66 writable [list f
0aa0: 65 65 64 20 24 73 6f 63 6b 20 5d 0a 09 66 69 6c eed $sock ]..fil
0ab0: 65 65 76 65 6e 74 20 24 73 6f 63 6b 20 72 65 61 eevent $sock rea
0ac0: 64 61 62 6c 65 20 5b 6c 69 73 74 20 64 72 61 69 dable [list drai
0ad0: 6e 20 24 73 6f 63 6b 5d 0a 09 64 70 75 74 73 20 n $sock]..dputs
0ae0: 22 63 72 65 61 74 65 64 20 24 73 6f 63 6b 22 0a "created $sock".
0af0: 20 20 20 20 7d 0a 20 20 20 20 77 68 69 6c 65 20 }. while
0b00: 7b 31 7d 20 7b 0a 09 23 20 4d 61 6b 65 20 73 75 {1} {..# Make su
0b10: 72 65 20 74 6f 20 77 61 69 74 20 75 6e 74 69 6c re to wait until
0b20: 20 61 6c 6c 20 6f 75 72 20 73 6f 63 6b 65 74 73 all our sockets
0b30: 20 63 6c 6f 73 65 20 64 6f 77 6e 2e 0a 09 76 77 close down...vw
0b40: 61 69 74 20 4f 50 54 53 28 6f 70 65 6e 70 6f 72 ait OPTS(openpor
0b50: 74 73 29 0a 09 69 66 20 7b 24 4f 50 54 53 28 6f ts)..if {$OPTS(o
0b60: 70 65 6e 70 6f 72 74 73 29 20 3d 3d 20 30 7d 20 penports) == 0}
0b70: 7b 0a 09 20 20 20 20 65 78 69 74 20 30 0a 09 7d {.. exit 0..}
0b80: 0a 20 20 20 20 7d 0a 7d 0a 0a 74 6c 73 3a 3a 69 . }.}..tls::i
0b90: 6e 69 74 20 2d 63 61 66 69 6c 65 20 24 4f 50 54 nit -cafile $OPT
0ba0: 53 28 2d 63 61 66 69 6c 65 29 20 2d 63 65 72 74 S(-cafile) -cert
0bb0: 66 69 6c 65 20 24 4f 50 54 53 28 2d 63 65 72 74 file $OPTS(-cert
0bc0: 29 20 2d 6b 65 79 66 69 6c 65 20 24 4f 50 54 53 ) -keyfile $OPTS
0bd0: 28 2d 6b 65 79 29 20 2d 72 65 71 75 69 72 65 20 (-key) -require
0be0: 24 4f 50 54 53 28 2d 72 65 71 75 69 72 65 29 0a $OPTS(-require).
0bf0: 0a 67 6f 0a .go.