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
105
106
107
108
109
110
|
use Test::More;
use strict;
use warnings;
use Net::Proxy;
use t::Util;
my @lines = (
"thwapp qunckkk aiieee flrbbbbb clunk zlonk\n",
"tutu bidon toto test3 pipo titi\n",
"pique ratatam am gram stram colegram\n",
"Signs_Point_to_Yes Reply_Hazy_Try_Again Cannot_Predict_Now\n",
);
my $tests = 2 * ( @lines + 1 );
plan tests => $tests;
init_rand(@ARGV);
my @free = find_free_ports(3);
SKIP: {
skip 'Not enough available ports', $tests if @free < 3;
my ( $proxy_port, $ssl_port, $ssh_port ) = @free;
my $pid = fork;
SKIP: {
skip "fork failed", $tests if !defined $pid;
if ( $pid == 0 ) {
# the child process runs the proxy
my $proxy = Net::Proxy->new(
{ in => {
type => 'dual',
port => $proxy_port,
timeout => 0.5,
server_first => {
type => 'tcp',
port => $ssh_port,
},
client_first => {
type => 'tcp',
port => $ssl_port,
}
},
out => { type => 'dummy' },
}
);
$proxy->register();
Net::Proxy->set_verbosity( $ENV{NET_PROXY_VERBOSITY} || 0 );
Net::Proxy->mainloop(2);
exit;
}
else {
# wait for the proxy to set up
sleep 1;
# the parent process does the testing
my $ssh_listener = listen_on_port($ssh_port)
or skip "Couldn't start the ssh server: $!", $tests;
my $ssl_listener = listen_on_port($ssl_port)
or skip "Couldn't start the ssl server: $!", $tests;
my ( $client, $server );
# try 'ssh'
$client = connect_to_port($proxy_port)
or skip_fail "Couldn't start the client: $!", $tests;
sleep 1; # wait for the timeout
$server = $ssh_listener->accept()
or skip_fail "Proxy didn't connect: $!", $tests;
# transmit data
for my $line (@lines) {
print $server $line; # real server speaks first
is( <$client>, $line, "SSH line received" );
( $client, $server ) = random_swap( $client, $server );
}
# close connections
$server->close();
is_closed( $client, 'peer' );
$client->close();
# try ssl
$client = connect_to_port($proxy_port)
or skip_fail "Couldn't start the client: $!", $tests;
print $client $lines[0]; # real client speaks first
$server = $ssl_listener->accept()
or skip_fail "Proxy didn't connect: $!", $tests;
is( <$server>, $lines[0], "SSL line received" );
# transmit the rest of the data
shift @lines;
for my $line ( @lines ) {
( $client, $server ) = random_swap( $client, $server );
print $client $line; # real client speaks first
is( <$server>, $line, "SSL line received" );
}
# close connections
$client->close();
is_closed( $server, 'peer' );
$server->close();
}
}
}
|
