1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
#!/usr/bin/env perl
use strict;
use warnings;
use File::Basename;
use lib dirname (__FILE__);
use TestUtils;
use TestHTTPD;
use File::Temp;
sub make_wildcard_config($) {
my $proxy_port = shift;
my ($fh, $filename) = File::Temp::tempfile();
# Write out a test config file
print $fh <<END;
# Minimal test configuration
listen 127.0.0.1 $proxy_port {
proto http
}
table {
.* example.com:80
}
END
close ($fh);
return $filename;
}
sub proxy {
my $config = shift;
exec(@_, '../src/sniproxy', '-f', '-c', $config);
}
sub worker($$) {
my $port = shift;
my $requests = shift;
for (my $i = 0; $i < $requests; $i++) {
system('curl',
'-s', '-S',
'-o', '/dev/null',
"http://localhost:$port/");
if ($? == -1) {
die "failed to execute: $!\n";
} elsif ($? & 127) {
printf STDERR "child died with signal %d, %s coredump\n", ($? & 127), ($? & 128) ? 'with' : 'without';
exit 255;
} elsif ($? >> 8) {
exit $? >> 8;
}
}
# Success
exit 0;
}
sub main {
my $proxy_port = $ENV{SNI_PROXY_PORT} || 8080;
my $workers = $ENV{WORKERS} || 3;
my $iterations = $ENV{ITERATIONS} || 3;
my $config = make_wildcard_config($proxy_port);
my $proxy_pid = start_child('server', \&proxy, $config, @ARGV);
# Wait for proxy to load and parse config
wait_for_port(port => $proxy_port);
for (my $i = 0; $i < $workers; $i++) {
start_child('worker', \&worker, $proxy_port, $iterations);
}
# Wait for all our children to finish
wait_for_type('worker');
# Give the proxy a second to flush buffers and close server connections
sleep 1;
# For troubleshooting connections stuck in CLOSE_WAIT state
#kill 10, $proxy_pid;
#system("netstat -ptn | grep $proxy_pid\/sniproxy");
# For troubleshooting 100% CPU usage
#system("top -n 1 -p $proxy_pid -b");
# Orderly shutdown of the server
kill 15, $proxy_pid;
sleep 1;
# Delete our test configuration
unlink($config);
# Kill off any remaining children
reap_children();
}
main();
|
