third-party-mirrors/sslh
1
0
Fork 0
mirror of https://github.com/yrutschle/sslh.git synced 2025-04-18 17:57:39 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix t_load so it works again!

Browse Source
This commit is contained in:
yrutschle 2021-03-17 18:48:27 +01:00
parent 790d639ad0
commit 0409775a1a
1 changed files with 54 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
t_load
View File

@ -44,8 +44,8 @@ my $stop_client_probability = .001;
# What protocols we test, and on what ports # What protocols we test, and on what ports
# Just comment out protocols you don't want to use. # Just comment out protocols you don't want to use.
my %protocols = ( my %protocols = (
"ssh" => { address => "localhost:9001", client => client("ssh") }, #"ssh" => { address => "localhost:9001", client => client("ssh") },
"ssl" => { address => "localhost:9002", client => client("ssl") }, # "tls" => { address => "localhost:9002", client => client("tls") },
"openvpn" => {address => "localhost:9003", client => client("openvpn") }, "openvpn" => {address => "localhost:9003", client => client("openvpn") },
"tinc" => {address => "localhost:9004", client => client("tinc") }, "tinc" => {address => "localhost:9004", client => client("tinc") },
); );
@ -58,30 +58,57 @@ my %protocols = (
my $sslh_address = "localhost:9000"; my $sslh_address = "localhost:9000";
my $pidfile = "/tmp/sslh_test.pid"; my $pidfile = "/tmp/sslh_test.pid";
sub connect_service {
my ($cnx, $service) = @_;
my ($test_data, $r);
sleep 5 if $service eq "ssh";
if ($service eq "openvpn") {
$test_data = "\x00\x00";
syswrite $cnx, $test_data;
sleep 1;
sysread $cnx, $r, 11; # length "openvpn: \x0\x0" => 11
}
if ($service eq "tinc") {
$test_data = "0 ";
syswrite $cnx, $test_data;
sleep 1;
sysread $cnx, $r, 8; # length "tinc: 0 " => 10
}
my $expected = "$service: $test_data";
($? = 1, die "* $service got [$r] expected [$expected]\n") if ($r ne $expected);
}
sub client { sub client {
my ($service) = @_; my ($service) = @_;
return sub { return sub {
while (1) { my ($client_id) = @_;
my $cnx = new IO::Socket::INET(PeerHost => $sslh_address);
my $test_data = "$service testing " x int(rand($block_rpt)+1) . "\n"; while (1) {
my $r;
my $cnx = new IO::Socket::INET(PeerHost => $sslh_address);
my $cnt = 0;
warn "starting $service\n";
connect_service($cnx, $service);
sleep 5 if $service eq "ssh";
if ($service eq "openvpn") {
syswrite $cnx, "\x00\x0F\x38\n";
my $msg;
sysread $cnx, $msg, 14; # length "openvpn: \x0\xF\x38\n" => 14
}
if ($service eq "tinc") {
syswrite $cnx, "0 \n";
my $msg;
sysread $cnx, $msg, 10; # length "tinc: 0 \n" => 10
}
while (1) { while (1) {
my $test_data = "$service $cnt" x int(rand($block_rpt)+1) . "\n";
print $cnx $test_data; print $cnx $test_data;
my $r = <$cnx>; $r = <$cnx>;
($? = 1, die "$service got [$r]\n") if ($r ne "$service: $test_data"); my $expected= "$test_data";
last if rand(1) < $stop_client_probability; my $r_l = length $r;
my $e_l = length $expected;
warn ("len $r_l / $e_l (". ($r_l - $e_l). ")\t$client_id\t$service\n");
($? = 1, die "$service got [$r] expected [$expected]\n") if ($r ne $expected);
#last if rand(1) < $stop_client_probability;
$cnt++;
} }
} }
exit 0; exit 0;
@ -90,29 +117,32 @@ sub client {
foreach my $p (keys %protocols) { foreach my $p (keys %protocols) {
if (!fork) { if (!fork) {
exec "./echosrv --listen $protocols{$p}->{address} --prefix '$p: '"; my $cmd = "./echosrv --listen $protocols{$p}->{address} --prefix '$p: '";
print "$cmd\n";
exec $cmd;
} }
} }
# Start sslh with the right plumbing # Start sslh with the right plumbing
my $sslh_pid; my $sslh_pid;
if (0) {
if (!($sslh_pid = fork)) { if (!($sslh_pid = fork)) {
my $user = (getpwuid $<)[0]; # Run under current username my $user = (getpwuid $<)[0]; # Run under current username
my $prots = join " ", map "--$_ $protocols{$_}->{address}", keys %protocols; my $prots = join " ", map "--$_ $protocols{$_}->{address}", keys %protocols;
my $cmd = "$sslh_binary -f -t 3 -u $user --listen $sslh_address $prots -P $pidfile"; my $cmd = "$sslh_binary -f -v3 -t 3 -u $user --listen $sslh_address $prots -P $pidfile";
print "$cmd\n"; print "$cmd\n";
exec $cmd; exec $cmd;
exit 0; exit 0;
} }
warn "spawned $sslh_pid\n"; warn "spawned $sslh_pid\n";
sleep 2; # valgrind can be heavy -- wait 5 seconds sleep 2; # valgrind can be heavy -- wait 5 seconds
}
for (1 .. $NUM_CNX) { for my $client_id (1 .. $NUM_CNX) {
foreach my $p (keys %protocols) { foreach my $p (keys %protocols) {
if (!fork) { if (!fork) {
warn "starting $p\n"; &{$protocols{$p}->{client}}($client_id);
&{$protocols{$p}->{client}};
exit; exit;
} }
# Give a little time so we don't overrun the # Give a little time so we don't overrun the