Artifact
d91c9b100dd056b99fcb785245d8654a4ab3946c219fcadf1202746aa3326db9:
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 70 72 clsh..package pr
0020: 65 66 65 72 20 6c 61 74 65 73 74 0a 70 61 63 6b efer latest.pack
0030: 61 67 65 20 72 65 71 75 69 72 65 20 74 6c 73 0a age require tls.
0040: 0a 73 65 74 20 64 69 72 09 09 09 5b 66 69 6c 65 .set dir...[file
0050: 20 6a 6f 69 6e 20 5b 66 69 6c 65 20 64 69 72 6e join [file dirn
0060: 61 6d 65 20 5b 69 6e 66 6f 20 73 63 72 69 70 74 ame [info script
0070: 5d 5d 20 2e 2e 2f 74 65 73 74 73 2f 63 65 72 74 ]] ../tests/cert
0080: 73 5d 0a 73 65 74 20 4f 50 54 53 28 2d 63 61 66 s].set OPTS(-caf
0090: 69 6c 65 29 09 5b 66 69 6c 65 20 6a 6f 69 6e 20 ile).[file join
00a0: 24 64 69 72 20 63 61 2e 70 65 6d 5d 0a 73 65 74 $dir ca.pem].set
00b0: 20 4f 50 54 53 28 2d 63 65 72 74 29 09 09 5b 66 OPTS(-cert)..[f
00c0: 69 6c 65 20 6a 6f 69 6e 20 24 64 69 72 20 63 6c ile join $dir cl
00d0: 69 65 6e 74 2e 70 65 6d 5d 0a 73 65 74 20 4f 50 ient.pem].set OP
00e0: 54 53 28 2d 6b 65 79 29 09 09 5b 66 69 6c 65 20 TS(-key)..[file
00f0: 6a 6f 69 6e 20 24 64 69 72 20 63 6c 69 65 6e 74 join $dir client
0100: 2e 6b 65 79 5d 0a 0a 73 65 74 20 4f 50 54 53 28 .key]..set OPTS(
0110: 2d 68 6f 73 74 29 09 09 6c 6f 63 61 6c 68 6f 73 -host)..localhos
0120: 74 0a 73 65 74 20 4f 50 54 53 28 2d 70 6f 72 74 t.set OPTS(-port
0130: 29 09 09 32 34 36 38 0a 73 65 74 20 4f 50 54 53 )..2468.set OPTS
0140: 28 2d 64 65 62 75 67 29 09 31 0a 73 65 74 20 4f (-debug).1.set O
0150: 50 54 53 28 2d 63 6f 75 6e 74 29 09 38 0a 73 65 PTS(-count).8.se
0160: 74 20 4f 50 54 53 28 2d 70 61 72 61 6c 6c 65 6c t OPTS(-parallel
0170: 29 09 31 0a 73 65 74 20 4f 50 54 53 28 2d 72 65 ).1.set OPTS(-re
0180: 71 75 69 72 65 29 09 30 0a 0a 66 6f 72 65 61 63 quire).0..foreac
0190: 68 20 7b 6b 65 79 20 76 61 6c 7d 20 24 61 72 67 h {key val} $arg
01a0: 76 20 7b 0a 20 20 20 20 69 66 20 7b 21 5b 69 6e v {. if {![in
01b0: 66 6f 20 65 78 69 73 74 73 20 4f 50 54 53 28 24 fo exists OPTS($
01c0: 6b 65 79 29 5d 7d 20 7b 0a 09 70 75 74 73 20 73 key)]} {..puts s
01d0: 74 64 65 72 72 20 22 55 73 61 67 65 3a 20 24 61 tderr "Usage: $a
01e0: 72 67 76 30 20 3f 6f 70 74 69 6f 6e 73 3f 5c 0a rgv0 ?options?\.
01f0: 09 09 5c 6e 5c 74 2d 64 65 62 75 67 20 20 20 20 ..\n\t-debug
0200: 20 62 6f 6f 6c 65 61 6e 20 20 20 44 65 62 75 67 boolean Debug
0210: 67 69 6e 67 20 6f 6e 20 6f 72 20 6f 66 66 20 28 ging on or off (
0220: 24 4f 50 54 53 28 2d 64 65 62 75 67 29 29 5c 0a $OPTS(-debug))\.
0230: 09 09 5c 6e 5c 74 2d 63 61 66 69 6c 65 20 20 20 ..\n\t-cafile
0240: 20 66 69 6c 65 20 20 20 20 20 20 43 65 72 74 2e file Cert.
0250: 20 41 75 74 68 2e 20 46 69 6c 65 20 28 24 4f 50 Auth. File ($OP
0260: 54 53 28 2d 63 61 66 69 6c 65 29 29 5c 0a 09 09 TS(-cafile))\...
0270: 5c 6e 5c 74 2d 63 6c 69 65 6e 74 20 20 20 20 66 \n\t-client f
0280: 69 6c 65 20 20 20 20 20 20 43 6c 69 65 6e 74 20 ile Client
0290: 43 65 72 74 20 28 24 4f 50 54 53 28 2d 63 65 72 Cert ($OPTS(-cer
02a0: 74 29 29 5c 0a 09 09 5c 6e 5c 74 2d 63 6b 65 79 t))\...\n\t-ckey
02b0: 20 20 20 20 20 20 66 69 6c 65 20 20 20 20 20 20 file
02c0: 43 6c 69 65 6e 74 20 4b 65 79 20 28 24 4f 50 54 Client Key ($OPT
02d0: 53 28 2d 6b 65 79 29 29 5c 0a 09 09 5c 6e 5c 74 S(-key))\...\n\t
02e0: 2d 63 6f 75 6e 74 20 20 20 20 20 6e 75 6d 20 20 -count num
02f0: 20 20 20 20 20 4e 6f 20 6f 66 20 73 79 6e 63 2e No of sync.
0300: 20 63 6f 6e 6e 65 63 74 69 6f 6e 73 20 74 6f 20 connections to
0310: 6d 61 6b 65 20 70 65 72 20 63 6c 69 65 6e 74 20 make per client
0320: 28 24 4f 50 54 53 28 2d 63 6f 75 6e 74 29 29 5c ($OPTS(-count))\
0330: 0a 09 09 5c 6e 5c 74 2d 70 61 72 61 6c 6c 65 6c ...\n\t-parallel
0340: 20 20 6e 75 6d 20 20 20 20 20 20 20 4e 6f 20 6f num No o
0350: 66 20 70 61 72 61 6c 6c 65 6c 20 63 6c 69 65 6e f parallel clien
0360: 74 73 20 74 6f 20 72 75 6e 20 28 24 4f 50 54 53 ts to run ($OPTS
0370: 28 2d 70 61 72 61 6c 6c 65 6c 29 29 5c 0a 09 09 (-parallel))\...
0380: 5c 6e 5c 74 2d 68 6f 73 74 20 20 20 20 20 20 68 \n\t-host h
0390: 6f 73 74 6e 61 6d 65 20 20 53 65 72 76 65 72 20 ostname Server
03a0: 68 6f 73 74 6e 61 6d 65 20 28 24 4f 50 54 53 28 hostname ($OPTS(
03b0: 2d 68 6f 73 74 29 29 5c 0a 09 09 5c 6e 5c 74 2d -host))\...\n\t-
03c0: 70 6f 72 74 20 20 20 20 20 20 6e 75 6d 20 20 20 port num
03d0: 20 20 20 20 53 65 72 76 65 72 20 70 6f 72 74 20 Server port
03e0: 28 24 4f 50 54 53 28 2d 70 6f 72 74 29 29 5c 0a ($OPTS(-port))\.
03f0: 09 09 5c 6e 5c 74 2d 72 65 71 75 69 72 65 20 20 ..\n\t-require
0400: 20 62 6f 6f 6c 65 61 6e 20 20 20 52 65 71 75 69 boolean Requi
0410: 72 65 20 43 65 72 74 69 66 69 63 61 74 65 20 28 re Certificate (
0420: 24 4f 50 54 53 28 2d 72 65 71 75 69 72 65 29 29 $OPTS(-require))
0430: 22 0a 09 65 78 69 74 0a 20 20 20 20 7d 0a 20 20 "..exit. }.
0440: 20 20 73 65 74 20 4f 50 54 53 28 24 6b 65 79 29 set OPTS($key)
0450: 20 24 76 61 6c 0a 7d 0a 0a 69 66 20 7b 24 4f 50 $val.}..if {$OP
0460: 54 53 28 2d 70 61 72 61 6c 6c 65 6c 29 20 3e 20 TS(-parallel) >
0470: 31 7d 20 7b 0a 20 20 20 20 23 20 49 66 20 74 68 1} {. # If th
0480: 65 79 20 77 61 6e 74 65 64 20 70 61 72 61 6c 6c ey wanted parall
0490: 65 6c 2c 20 77 65 20 6a 75 73 74 20 73 70 61 77 el, we just spaw
04a0: 6e 20 6f 75 72 73 65 6c 76 65 73 20 73 65 76 65 n ourselves seve
04b0: 72 61 6c 20 74 69 6d 65 73 0a 20 20 20 20 23 20 ral times. #
04c0: 77 69 74 68 20 74 68 65 20 72 69 67 68 74 20 61 with the right a
04d0: 72 67 73 2e 0a 0a 20 20 20 20 73 65 74 20 63 6d rgs... set cm
04e0: 64 09 5b 69 6e 66 6f 20 6e 61 6d 65 6f 66 65 78 d.[info nameofex
04f0: 65 63 75 74 61 62 6c 65 5d 0a 20 20 20 20 73 65 ecutable]. se
0500: 74 20 73 63 72 69 70 74 09 5b 69 6e 66 6f 20 73 t script.[info s
0510: 63 72 69 70 74 5d 0a 20 20 20 20 66 6f 72 20 7b cript]. for {
0520: 73 65 74 20 69 20 30 7d 20 7b 24 69 20 3c 20 24 set i 0} {$i < $
0530: 4f 50 54 53 28 2d 70 61 72 61 6c 6c 65 6c 29 7d OPTS(-parallel)}
0540: 20 7b 69 6e 63 72 20 69 7d 20 7b 0a 09 65 76 61 {incr i} {..eva
0550: 6c 20 5b 6c 69 73 74 20 65 78 65 63 20 24 63 6d l [list exec $cm
0560: 64 20 24 73 63 72 69 70 74 5d 20 5b 61 72 72 61 d $script] [arra
0570: 79 20 67 65 74 20 4f 50 54 53 5d 20 5b 6c 69 73 y get OPTS] [lis
0580: 74 20 2d 70 61 72 61 6c 6c 65 6c 20 30 5d 20 26 t -parallel 0] &
0590: 0a 20 20 20 20 7d 0a 20 20 20 20 65 78 69 74 0a . }. exit.
05a0: 7d 0a 0a 23 20 4c 6f 63 61 6c 20 68 61 6e 64 6c }..# Local handl
05b0: 65 72 20 66 6f 72 20 61 6e 79 20 62 61 63 6b 67 er for any backg
05c0: 72 6f 75 6e 64 20 65 72 72 6f 72 73 2e 0a 70 72 round errors..pr
05d0: 6f 63 20 62 67 65 72 72 6f 72 20 7b 6d 73 67 7d oc bgerror {msg}
05e0: 20 7b 20 70 75 74 73 20 22 42 47 45 52 52 4f 52 { puts "BGERROR
05f0: 3a 20 24 6d 73 67 22 20 7d 0a 0a 23 20 64 65 62 : $msg" }..# deb
0600: 75 67 67 69 6e 67 20 68 65 6c 70 65 72 20 63 6f ugging helper co
0610: 64 65 0a 70 72 6f 63 20 73 68 6f 72 74 73 74 72 de.proc shortstr
0620: 20 7b 73 74 72 7d 20 7b 0a 20 20 20 20 72 65 74 {str} {. ret
0630: 75 72 6e 20 22 5b 73 74 72 69 6e 67 20 72 65 70 urn "[string rep
0640: 6c 61 63 65 20 24 73 74 72 20 31 30 20 65 6e 64 lace $str 10 end
0650: 20 2e 2e 2e 5d 20 5b 73 74 72 69 6e 67 20 6c 65 ...] [string le
0660: 6e 67 74 68 20 24 73 74 72 5d 62 22 0a 7d 0a 70 ngth $str]b".}.p
0670: 72 6f 63 20 64 70 75 74 73 20 7b 6d 73 67 7d 20 roc dputs {msg}
0680: 7b 20 69 66 20 7b 24 3a 3a 4f 50 54 53 28 2d 64 { if {$::OPTS(-d
0690: 65 62 75 67 29 7d 20 7b 20 70 75 74 73 20 73 74 ebug)} { puts st
06a0: 64 65 72 72 20 24 6d 73 67 20 3b 20 66 6c 75 73 derr $msg ; flus
06b0: 68 20 73 74 64 65 72 72 20 7d 20 7d 0a 0a 73 65 h stderr } }..se
06c0: 74 20 4f 50 54 53 28 6f 70 65 6e 70 6f 72 74 73 t OPTS(openports
06d0: 29 09 30 0a 0a 23 20 44 65 66 69 6e 65 20 77 68 ).0..# Define wh
06e0: 61 74 20 77 65 20 77 61 6e 74 20 74 6f 20 66 65 at we want to fe
06f0: 65 64 20 64 6f 77 6e 20 74 68 65 20 70 69 70 65 ed down the pipe
0700: 0a 73 65 74 20 6d 65 67 61 64 61 74 61 20 5b 73 .set megadata [s
0710: 74 72 69 6e 67 20 72 65 70 65 61 74 20 5b 73 74 tring repeat [st
0720: 72 69 6e 67 20 72 65 70 65 61 74 20 41 20 37 36 ring repeat A 76
0730: 5d 5c 6e 20 31 30 30 30 5d 0a 0a 70 72 6f 63 20 ]\n 1000]..proc
0740: 64 72 61 69 6e 20 7b 63 68 61 6e 7d 20 7b 0a 20 drain {chan} {.
0750: 20 20 20 67 6c 6f 62 61 6c 20 4f 50 54 53 0a 20 global OPTS.
0760: 20 20 20 69 66 20 7b 5b 63 61 74 63 68 20 7b 72 if {[catch {r
0770: 65 61 64 20 24 63 68 61 6e 7d 20 64 61 74 61 5d ead $chan} data]
0780: 7d 20 7b 0a 09 23 64 70 75 74 73 20 22 45 4f 46 } {..#dputs "EOF
0790: 20 24 63 68 61 6e 20 28 5b 73 68 6f 72 74 73 74 $chan ([shortst
07a0: 72 20 24 64 61 74 61 5d 29 22 0a 09 69 6e 63 72 r $data])"..incr
07b0: 20 4f 50 54 53 28 6f 70 65 6e 70 6f 72 74 73 29 OPTS(openports)
07c0: 20 2d 31 0a 09 63 61 74 63 68 20 7b 63 6c 6f 73 -1..catch {clos
07d0: 65 20 24 63 68 61 6e 7d 0a 09 72 65 74 75 72 6e e $chan}..return
07e0: 0a 20 20 20 20 7d 0a 20 20 20 20 23 69 66 20 7b . }. #if {
07f0: 24 64 61 74 61 20 6e 65 20 22 22 7d 20 7b 20 64 $data ne ""} { d
0800: 70 75 74 73 20 22 67 6f 74 20 24 63 68 61 6e 20 puts "got $chan
0810: 28 5b 73 68 6f 72 74 73 74 72 20 24 64 61 74 61 ([shortstr $data
0820: 5d 29 22 20 7d 0a 20 20 20 20 69 66 20 7b 5b 73 ])" }. if {[s
0830: 74 72 69 6e 67 20 6d 61 74 63 68 20 2a 43 4c 4f tring match *CLO
0840: 53 45 5c 6e 20 24 64 61 74 61 5d 7d 20 7b 0a 09 SE\n $data]} {..
0850: 64 70 75 74 73 20 22 43 4c 4f 53 45 20 24 63 68 dputs "CLOSE $ch
0860: 61 6e 22 0a 09 69 6e 63 72 20 4f 50 54 53 28 6f an"..incr OPTS(o
0870: 70 65 6e 70 6f 72 74 73 29 20 2d 31 0a 09 63 6c penports) -1..cl
0880: 6f 73 65 20 24 63 68 61 6e 0a 09 72 65 74 75 72 ose $chan..retur
0890: 6e 0a 20 20 20 20 7d 20 65 6c 73 65 69 66 20 7b n. } elseif {
08a0: 5b 65 6f 66 20 24 63 68 61 6e 5d 7d 20 7b 0a 09 [eof $chan]} {..
08b0: 23 20 63 6c 69 65 6e 74 20 67 6f 6e 65 20 6f 72 # client gone or
08c0: 20 66 69 6e 69 73 68 65 64 0a 09 64 70 75 74 73 finished..dputs
08d0: 20 22 45 4f 46 20 24 63 68 61 6e 22 0a 09 69 6e "EOF $chan"..in
08e0: 63 72 20 4f 50 54 53 28 6f 70 65 6e 70 6f 72 74 cr OPTS(openport
08f0: 73 29 20 2d 31 0a 09 63 6c 6f 73 65 20 24 63 68 s) -1..close $ch
0900: 61 6e 0a 09 72 65 74 75 72 6e 0a 20 20 20 20 7d an..return. }
0910: 0a 7d 0a 0a 70 72 6f 63 20 66 65 65 64 20 7b 73 .}..proc feed {s
0920: 6f 63 6b 7d 20 7b 0a 20 20 20 20 64 70 75 74 73 ock} {. dputs
0930: 20 22 66 65 65 64 20 24 73 6f 63 6b 20 28 5b 73 "feed $sock ([s
0940: 68 6f 72 74 73 74 72 20 24 3a 3a 6d 65 67 61 64 hortstr $::megad
0950: 61 74 61 5d 29 22 0a 20 20 20 20 70 75 74 73 20 ata])". puts
0960: 24 73 6f 63 6b 20 24 3a 3a 6d 65 67 61 64 61 74 $sock $::megadat
0970: 61 0a 20 20 20 20 66 6c 75 73 68 20 24 73 6f 63 a. flush $soc
0980: 6b 0a 20 20 20 20 70 75 74 73 20 24 73 6f 63 6b k. puts $sock
0990: 20 43 4c 4f 53 45 0a 20 20 20 20 66 6c 75 73 68 CLOSE. flush
09a0: 20 24 73 6f 63 6b 0a 20 20 20 20 66 69 6c 65 65 $sock. filee
09b0: 76 65 6e 74 20 24 73 6f 63 6b 20 77 72 69 74 61 vent $sock writa
09c0: 62 6c 65 20 7b 7d 0a 7d 0a 0a 70 72 6f 63 20 67 ble {}.}..proc g
09d0: 6f 20 7b 7d 20 7b 0a 20 20 20 20 67 6c 6f 62 61 o {} {. globa
09e0: 6c 20 4f 50 54 53 0a 20 20 20 20 66 6f 72 20 7b l OPTS. for {
09f0: 73 65 74 20 6e 75 6d 20 24 4f 50 54 53 28 2d 63 set num $OPTS(-c
0a00: 6f 75 6e 74 29 7d 20 7b 24 6e 75 6d 20 3e 20 30 ount)} {$num > 0
0a10: 7d 20 7b 69 6e 63 72 20 6e 75 6d 20 2d 31 7d 20 } {incr num -1}
0a20: 7b 0a 09 73 65 74 20 73 6f 63 6b 20 5b 74 6c 73 {..set sock [tls
0a30: 3a 3a 73 6f 63 6b 65 74 20 24 4f 50 54 53 28 2d ::socket $OPTS(-
0a40: 68 6f 73 74 29 20 24 4f 50 54 53 28 2d 70 6f 72 host) $OPTS(-por
0a50: 74 29 5d 0a 09 69 6e 63 72 20 4f 50 54 53 28 6f t)]..incr OPTS(o
0a60: 70 65 6e 70 6f 72 74 73 29 0a 09 66 63 6f 6e 66 penports)..fconf
0a70: 69 67 75 72 65 20 24 73 6f 63 6b 20 2d 62 6c 6f igure $sock -blo
0a80: 63 6b 69 6e 67 20 30 20 2d 62 75 66 66 65 72 73 cking 0 -buffers
0a90: 69 7a 65 20 34 30 39 36 0a 09 66 69 6c 65 65 76 ize 4096..fileev
0aa0: 65 6e 74 20 24 73 6f 63 6b 20 77 72 69 74 61 62 ent $sock writab
0ab0: 6c 65 20 5b 6c 69 73 74 20 66 65 65 64 20 24 73 le [list feed $s
0ac0: 6f 63 6b 20 5d 0a 09 66 69 6c 65 65 76 65 6e 74 ock ]..fileevent
0ad0: 20 24 73 6f 63 6b 20 72 65 61 64 61 62 6c 65 20 $sock readable
0ae0: 5b 6c 69 73 74 20 64 72 61 69 6e 20 24 73 6f 63 [list drain $soc
0af0: 6b 5d 0a 09 64 70 75 74 73 20 22 63 72 65 61 74 k]..dputs "creat
0b00: 65 64 20 24 73 6f 63 6b 22 0a 20 20 20 20 7d 0a ed $sock". }.
0b10: 20 20 20 20 77 68 69 6c 65 20 7b 31 7d 20 7b 0a while {1} {.
0b20: 09 23 20 4d 61 6b 65 20 73 75 72 65 20 74 6f 20 .# Make sure to
0b30: 77 61 69 74 20 75 6e 74 69 6c 20 61 6c 6c 20 6f wait until all o
0b40: 75 72 20 73 6f 63 6b 65 74 73 20 63 6c 6f 73 65 ur sockets close
0b50: 20 64 6f 77 6e 2e 0a 09 76 77 61 69 74 20 4f 50 down...vwait OP
0b60: 54 53 28 6f 70 65 6e 70 6f 72 74 73 29 0a 09 69 TS(openports)..i
0b70: 66 20 7b 24 4f 50 54 53 28 6f 70 65 6e 70 6f 72 f {$OPTS(openpor
0b80: 74 73 29 20 3d 3d 20 30 7d 20 7b 0a 09 20 20 20 ts) == 0} {..
0b90: 20 65 78 69 74 20 30 0a 09 7d 0a 20 20 20 20 7d exit 0..}. }
0ba0: 0a 7d 0a 0a 74 6c 73 3a 3a 69 6e 69 74 20 2d 63 .}..tls::init -c
0bb0: 61 66 69 6c 65 20 24 4f 50 54 53 28 2d 63 61 66 afile $OPTS(-caf
0bc0: 69 6c 65 29 20 2d 63 65 72 74 66 69 6c 65 20 24 ile) -certfile $
0bd0: 4f 50 54 53 28 2d 63 65 72 74 29 20 2d 6b 65 79 OPTS(-cert) -key
0be0: 66 69 6c 65 20 24 4f 50 54 53 28 2d 6b 65 79 29 file $OPTS(-key)
0bf0: 20 2d 72 65 71 75 69 72 65 20 24 4f 50 54 53 28 -require $OPTS(
0c00: 2d 72 65 71 75 69 72 65 29 0a 0a 67 6f 0a -require)..go.