Artifact
38bd23a6f6890008fcfc645b99ae34ac681ec41ad23d9171b3d9a89c2363b9cd
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 0a 66 6f 72 65 61 rallel).1..forea
0160: 63 68 20 7b 6b 65 79 20 76 61 6c 7d 20 24 61 72 ch {key val} $ar
0170: 67 76 20 7b 0a 20 20 20 20 69 66 20 7b 21 5b 69 gv {. if {![i
0180: 6e 66 6f 20 65 78 69 73 74 73 20 4f 50 54 53 28 nfo exists OPTS(
0190: 24 6b 65 79 29 5d 7d 20 7b 0a 09 70 75 74 73 20 $key)]} {..puts
01a0: 73 74 64 65 72 72 20 22 55 73 61 67 65 3a 20 24 stderr "Usage: $
01b0: 61 72 67 76 30 20 3f 6f 70 74 69 6f 6e 73 3f 5c argv0 ?options?\
01c0: 0a 09 09 5c 6e 5c 74 2d 64 65 62 75 67 20 20 20 ...\n\t-debug
01d0: 20 20 62 6f 6f 6c 65 61 6e 20 20 20 44 65 62 75 boolean Debu
01e0: 67 67 69 6e 67 20 6f 6e 20 6f 72 20 6f 66 66 20 gging on or off
01f0: 28 24 4f 50 54 53 28 2d 64 65 62 75 67 29 29 5c ($OPTS(-debug))\
0200: 0a 09 09 5c 6e 5c 74 2d 63 61 66 69 6c 65 20 20 ...\n\t-cafile
0210: 20 20 66 69 6c 65 20 20 20 20 20 20 43 65 72 74 file Cert
0220: 2e 20 41 75 74 68 2e 20 46 69 6c 65 20 28 24 4f . Auth. File ($O
0230: 50 54 53 28 2d 63 61 66 69 6c 65 29 29 5c 0a 09 PTS(-cafile))\..
0240: 09 5c 6e 5c 74 2d 63 6c 69 65 6e 74 20 20 20 20 .\n\t-client
0250: 66 69 6c 65 20 20 20 20 20 20 43 6c 69 65 6e 74 file Client
0260: 20 43 65 72 74 20 28 24 4f 50 54 53 28 2d 63 65 Cert ($OPTS(-ce
0270: 72 74 29 29 5c 0a 09 09 5c 6e 5c 74 2d 63 6b 65 rt))\...\n\t-cke
0280: 79 20 20 20 20 20 20 66 69 6c 65 20 20 20 20 20 y file
0290: 20 43 6c 69 65 6e 74 20 4b 65 79 20 28 24 4f 50 Client Key ($OP
02a0: 54 53 28 2d 6b 65 79 29 29 5c 0a 09 09 5c 6e 5c TS(-key))\...\n\
02b0: 74 2d 63 6f 75 6e 74 20 20 20 20 20 6e 75 6d 20 t-count num
02c0: 20 20 20 20 20 20 4e 6f 20 6f 66 20 73 79 6e 63 No of sync
02d0: 2e 20 63 6f 6e 6e 65 63 74 69 6f 6e 73 20 74 6f . connections to
02e0: 20 6d 61 6b 65 20 70 65 72 20 63 6c 69 65 6e 74 make per client
02f0: 20 28 24 4f 50 54 53 28 2d 63 6f 75 6e 74 29 29 ($OPTS(-count))
0300: 5c 0a 09 09 5c 6e 5c 74 2d 70 61 72 61 6c 6c 65 \...\n\t-paralle
0310: 6c 20 20 6e 75 6d 20 20 20 20 20 20 20 4e 6f 20 l num No
0320: 6f 66 20 70 61 72 61 6c 6c 65 6c 20 63 6c 69 65 of parallel clie
0330: 6e 74 73 20 74 6f 20 72 75 6e 20 28 24 4f 50 54 nts to run ($OPT
0340: 53 28 2d 70 61 72 61 6c 6c 65 6c 29 29 5c 0a 09 S(-parallel))\..
0350: 09 5c 6e 5c 74 2d 68 6f 73 74 20 20 20 20 20 20 .\n\t-host
0360: 68 6f 73 74 6e 61 6d 65 20 20 53 65 72 76 65 72 hostname Server
0370: 20 68 6f 73 74 6e 61 6d 65 20 28 24 4f 50 54 53 hostname ($OPTS
0380: 28 2d 68 6f 73 74 29 29 5c 0a 09 09 5c 6e 5c 74 (-host))\...\n\t
0390: 2d 70 6f 72 74 20 20 20 20 20 20 6e 75 6d 20 20 -port num
03a0: 20 20 20 20 20 53 65 72 76 65 72 20 70 6f 72 74 Server port
03b0: 20 28 24 4f 50 54 53 28 2d 70 6f 72 74 29 29 22 ($OPTS(-port))"
03c0: 0a 09 65 78 69 74 0a 20 20 20 20 7d 0a 20 20 20 ..exit. }.
03d0: 20 73 65 74 20 4f 50 54 53 28 24 6b 65 79 29 20 set OPTS($key)
03e0: 24 76 61 6c 0a 7d 0a 0a 69 66 20 7b 24 4f 50 54 $val.}..if {$OPT
03f0: 53 28 2d 70 61 72 61 6c 6c 65 6c 29 20 3e 20 31 S(-parallel) > 1
0400: 7d 20 7b 0a 20 20 20 20 23 20 49 66 20 74 68 65 } {. # If the
0410: 79 20 77 61 6e 74 65 64 20 70 61 72 61 6c 6c 65 y wanted paralle
0420: 6c 2c 20 77 65 20 6a 75 73 74 20 73 70 61 77 6e l, we just spawn
0430: 20 6f 75 72 73 65 6c 76 65 73 20 73 65 76 65 72 ourselves sever
0440: 61 6c 20 74 69 6d 65 73 0a 20 20 20 20 23 20 77 al times. # w
0450: 69 74 68 20 74 68 65 20 72 69 67 68 74 20 61 72 ith the right ar
0460: 67 73 2e 0a 0a 20 20 20 20 73 65 74 20 63 6d 64 gs... set cmd
0470: 09 5b 69 6e 66 6f 20 6e 61 6d 65 6f 66 65 78 65 .[info nameofexe
0480: 63 75 74 61 62 6c 65 5d 0a 20 20 20 20 73 65 74 cutable]. set
0490: 20 73 63 72 69 70 74 09 5b 69 6e 66 6f 20 73 63 script.[info sc
04a0: 72 69 70 74 5d 0a 20 20 20 20 66 6f 72 20 7b 73 ript]. for {s
04b0: 65 74 20 69 20 30 7d 20 7b 24 69 20 3c 20 24 4f et i 0} {$i < $O
04c0: 50 54 53 28 2d 70 61 72 61 6c 6c 65 6c 29 7d 20 PTS(-parallel)}
04d0: 7b 69 6e 63 72 20 69 7d 20 7b 0a 09 65 76 61 6c {incr i} {..eval
04e0: 20 5b 6c 69 73 74 20 65 78 65 63 20 24 63 6d 64 [list exec $cmd
04f0: 20 24 73 63 72 69 70 74 5d 20 5b 61 72 72 61 79 $script] [array
0500: 20 67 65 74 20 4f 50 54 53 5d 20 5b 6c 69 73 74 get OPTS] [list
0510: 20 2d 70 61 72 61 6c 6c 65 6c 20 30 5d 20 26 0a -parallel 0] &.
0520: 20 20 20 20 7d 0a 20 20 20 20 65 78 69 74 0a 7d }. exit.}
0530: 0a 0a 23 20 4c 6f 63 61 6c 20 68 61 6e 64 6c 65 ..# Local handle
0540: 72 20 66 6f 72 20 61 6e 79 20 62 61 63 6b 67 72 r for any backgr
0550: 6f 75 6e 64 20 65 72 72 6f 72 73 2e 0a 70 72 6f ound errors..pro
0560: 63 20 62 67 65 72 72 6f 72 20 7b 6d 73 67 7d 20 c bgerror {msg}
0570: 7b 20 70 75 74 73 20 22 42 47 45 52 52 4f 52 3a { puts "BGERROR:
0580: 20 24 6d 73 67 22 20 7d 0a 0a 23 20 64 65 62 75 $msg" }..# debu
0590: 67 67 69 6e 67 20 68 65 6c 70 65 72 20 63 6f 64 gging helper cod
05a0: 65 0a 70 72 6f 63 20 73 68 6f 72 74 73 74 72 20 e.proc shortstr
05b0: 7b 73 74 72 7d 20 7b 0a 20 20 20 20 72 65 74 75 {str} {. retu
05c0: 72 6e 20 22 5b 73 74 72 69 6e 67 20 72 65 70 6c rn "[string repl
05d0: 61 63 65 20 24 73 74 72 20 31 30 20 65 6e 64 20 ace $str 10 end
05e0: 2e 2e 2e 5d 20 5b 73 74 72 69 6e 67 20 6c 65 6e ...] [string len
05f0: 67 74 68 20 24 73 74 72 5d 62 22 0a 7d 0a 70 72 gth $str]b".}.pr
0600: 6f 63 20 64 70 75 74 73 20 7b 6d 73 67 7d 20 7b oc dputs {msg} {
0610: 20 69 66 20 7b 24 3a 3a 4f 50 54 53 28 2d 64 65 if {$::OPTS(-de
0620: 62 75 67 29 7d 20 7b 20 70 75 74 73 20 73 74 64 bug)} { puts std
0630: 65 72 72 20 24 6d 73 67 20 3b 20 66 6c 75 73 68 err $msg ; flush
0640: 20 73 74 64 65 72 72 20 7d 20 7d 0a 0a 73 65 74 stderr } }..set
0650: 20 4f 50 54 53 28 6f 70 65 6e 70 6f 72 74 73 29 OPTS(openports)
0660: 09 30 0a 0a 23 20 44 65 66 69 6e 65 20 77 68 61 .0..# Define wha
0670: 74 20 77 65 20 77 61 6e 74 20 74 6f 20 66 65 65 t we want to fee
0680: 64 20 64 6f 77 6e 20 74 68 65 20 70 69 70 65 0a d down the pipe.
0690: 73 65 74 20 6d 65 67 61 64 61 74 61 20 5b 73 74 set megadata [st
06a0: 72 69 6e 67 20 72 65 70 65 61 74 20 5b 73 74 72 ring repeat [str
06b0: 69 6e 67 20 72 65 70 65 61 74 20 41 20 37 36 5d ing repeat A 76]
06c0: 5c 6e 20 31 30 30 30 5d 0a 0a 70 72 6f 63 20 64 \n 1000]..proc d
06d0: 72 61 69 6e 20 7b 63 68 61 6e 7d 20 7b 0a 20 20 rain {chan} {.
06e0: 20 20 67 6c 6f 62 61 6c 20 4f 50 54 53 0a 20 20 global OPTS.
06f0: 20 20 69 66 20 7b 5b 63 61 74 63 68 20 7b 72 65 if {[catch {re
0700: 61 64 20 24 63 68 61 6e 7d 20 64 61 74 61 5d 7d ad $chan} data]}
0710: 20 7b 0a 09 23 64 70 75 74 73 20 22 45 4f 46 20 {..#dputs "EOF
0720: 24 63 68 61 6e 20 28 5b 73 68 6f 72 74 73 74 72 $chan ([shortstr
0730: 20 24 64 61 74 61 5d 29 22 0a 09 69 6e 63 72 20 $data])"..incr
0740: 4f 50 54 53 28 6f 70 65 6e 70 6f 72 74 73 29 20 OPTS(openports)
0750: 2d 31 0a 09 63 61 74 63 68 20 7b 63 6c 6f 73 65 -1..catch {close
0760: 20 24 63 68 61 6e 7d 0a 09 72 65 74 75 72 6e 0a $chan}..return.
0770: 20 20 20 20 7d 0a 20 20 20 20 23 69 66 20 7b 24 }. #if {$
0780: 64 61 74 61 20 21 3d 20 22 22 7d 20 7b 20 64 70 data != ""} { dp
0790: 75 74 73 20 22 67 6f 74 20 24 63 68 61 6e 20 28 uts "got $chan (
07a0: 5b 73 68 6f 72 74 73 74 72 20 24 64 61 74 61 5d [shortstr $data]
07b0: 29 22 20 7d 0a 20 20 20 20 69 66 20 7b 5b 73 74 )" }. if {[st
07c0: 72 69 6e 67 20 6d 61 74 63 68 20 2a 43 4c 4f 53 ring match *CLOS
07d0: 45 5c 6e 20 24 64 61 74 61 5d 7d 20 7b 0a 09 64 E\n $data]} {..d
07e0: 70 75 74 73 20 22 43 4c 4f 53 45 20 24 63 68 61 puts "CLOSE $cha
07f0: 6e 22 0a 09 69 6e 63 72 20 4f 50 54 53 28 6f 70 n"..incr OPTS(op
0800: 65 6e 70 6f 72 74 73 29 20 2d 31 0a 09 63 6c 6f enports) -1..clo
0810: 73 65 20 24 63 68 61 6e 0a 09 72 65 74 75 72 6e se $chan..return
0820: 0a 20 20 20 20 7d 20 65 6c 73 65 69 66 20 7b 5b . } elseif {[
0830: 65 6f 66 20 24 63 68 61 6e 5d 7d 20 7b 0a 09 23 eof $chan]} {..#
0840: 20 63 6c 69 65 6e 74 20 67 6f 6e 65 20 6f 72 20 client gone or
0850: 66 69 6e 69 73 68 65 64 0a 09 64 70 75 74 73 20 finished..dputs
0860: 22 45 4f 46 20 24 63 68 61 6e 22 0a 09 69 6e 63 "EOF $chan"..inc
0870: 72 20 4f 50 54 53 28 6f 70 65 6e 70 6f 72 74 73 r OPTS(openports
0880: 29 20 2d 31 0a 09 63 6c 6f 73 65 20 24 63 68 61 ) -1..close $cha
0890: 6e 0a 09 72 65 74 75 72 6e 0a 20 20 20 20 7d 0a n..return. }.
08a0: 7d 0a 0a 70 72 6f 63 20 66 65 65 64 20 7b 73 6f }..proc feed {so
08b0: 63 6b 7d 20 7b 0a 20 20 20 20 64 70 75 74 73 20 ck} {. dputs
08c0: 22 66 65 65 64 20 24 73 6f 63 6b 20 28 5b 73 68 "feed $sock ([sh
08d0: 6f 72 74 73 74 72 20 24 3a 3a 6d 65 67 61 64 61 ortstr $::megada
08e0: 74 61 5d 29 22 0a 20 20 20 20 70 75 74 73 20 24 ta])". puts $
08f0: 73 6f 63 6b 20 24 3a 3a 6d 65 67 61 64 61 74 61 sock $::megadata
0900: 0a 20 20 20 20 66 6c 75 73 68 20 24 73 6f 63 6b . flush $sock
0910: 0a 20 20 20 20 70 75 74 73 20 24 73 6f 63 6b 20 . puts $sock
0920: 43 4c 4f 53 45 0a 20 20 20 20 66 6c 75 73 68 20 CLOSE. flush
0930: 24 73 6f 63 6b 0a 20 20 20 20 66 69 6c 65 65 76 $sock. fileev
0940: 65 6e 74 20 24 73 6f 63 6b 20 77 72 69 74 61 62 ent $sock writab
0950: 6c 65 20 7b 7d 0a 7d 0a 0a 70 72 6f 63 20 67 6f le {}.}..proc go
0960: 20 7b 7d 20 7b 0a 20 20 20 20 67 6c 6f 62 61 6c {} {. global
0970: 20 4f 50 54 53 0a 20 20 20 20 66 6f 72 20 7b 73 OPTS. for {s
0980: 65 74 20 6e 75 6d 20 24 4f 50 54 53 28 2d 63 6f et num $OPTS(-co
0990: 75 6e 74 29 7d 20 7b 24 6e 75 6d 20 3e 20 30 7d unt)} {$num > 0}
09a0: 20 7b 69 6e 63 72 20 6e 75 6d 20 2d 31 7d 20 7b {incr num -1} {
09b0: 0a 09 73 65 74 20 73 6f 63 6b 20 5b 74 6c 73 3a ..set sock [tls:
09c0: 3a 73 6f 63 6b 65 74 20 24 4f 50 54 53 28 2d 68 :socket $OPTS(-h
09d0: 6f 73 74 29 20 24 4f 50 54 53 28 2d 70 6f 72 74 ost) $OPTS(-port
09e0: 29 5d 0a 09 69 6e 63 72 20 4f 50 54 53 28 6f 70 )]..incr OPTS(op
09f0: 65 6e 70 6f 72 74 73 29 0a 09 66 63 6f 6e 66 69 enports)..fconfi
0a00: 67 75 72 65 20 24 73 6f 63 6b 20 2d 62 6c 6f 63 gure $sock -bloc
0a10: 6b 69 6e 67 20 30 20 2d 62 75 66 66 65 72 73 69 king 0 -buffersi
0a20: 7a 65 20 34 30 39 36 0a 09 66 69 6c 65 65 76 65 ze 4096..fileeve
0a30: 6e 74 20 24 73 6f 63 6b 20 77 72 69 74 61 62 6c nt $sock writabl
0a40: 65 20 5b 6c 69 73 74 20 66 65 65 64 20 24 73 6f e [list feed $so
0a50: 63 6b 20 5d 0a 09 66 69 6c 65 65 76 65 6e 74 20 ck ]..fileevent
0a60: 24 73 6f 63 6b 20 72 65 61 64 61 62 6c 65 20 5b $sock readable [
0a70: 6c 69 73 74 20 64 72 61 69 6e 20 24 73 6f 63 6b list drain $sock
0a80: 5d 0a 09 64 70 75 74 73 20 22 63 72 65 61 74 65 ]..dputs "create
0a90: 64 20 24 73 6f 63 6b 22 0a 20 20 20 20 7d 0a 20 d $sock". }.
0aa0: 20 20 20 77 68 69 6c 65 20 7b 31 7d 20 7b 0a 09 while {1} {..
0ab0: 23 20 4d 61 6b 65 20 73 75 72 65 20 74 6f 20 77 # Make sure to w
0ac0: 61 69 74 20 75 6e 74 69 6c 20 61 6c 6c 20 6f 75 ait until all ou
0ad0: 72 20 73 6f 63 6b 65 74 73 20 63 6c 6f 73 65 20 r sockets close
0ae0: 64 6f 77 6e 2e 0a 09 76 77 61 69 74 20 4f 50 54 down...vwait OPT
0af0: 53 28 6f 70 65 6e 70 6f 72 74 73 29 0a 09 69 66 S(openports)..if
0b00: 20 7b 24 4f 50 54 53 28 6f 70 65 6e 70 6f 72 74 {$OPTS(openport
0b10: 73 29 20 3d 3d 20 30 7d 20 7b 0a 09 20 20 20 20 s) == 0} {..
0b20: 65 78 69 74 20 30 0a 09 7d 0a 20 20 20 20 7d 0a exit 0..}. }.
0b30: 7d 0a 0a 74 6c 73 3a 3a 69 6e 69 74 20 2d 63 61 }..tls::init -ca
0b40: 66 69 6c 65 20 24 4f 50 54 53 28 2d 63 61 66 69 file $OPTS(-cafi
0b50: 6c 65 29 20 2d 63 65 72 74 66 69 6c 65 20 24 4f le) -certfile $O
0b60: 50 54 53 28 2d 63 65 72 74 29 20 2d 6b 65 79 66 PTS(-cert) -keyf
0b70: 69 6c 65 20 24 4f 50 54 53 28 2d 6b 65 79 29 0a ile $OPTS(-key).
0b80: 0a 67 6f 0a .go.