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
|
#!/usr/bin/env perl
package Net::Server::Test;
use strict;
use File::Temp qw(tempdir);
use File::Spec::Functions qw(catfile);
use English qw($UID $GID);
use FindBin qw($Bin);
use lib $Bin;
use NetServerTest qw(prepare_test ok use_ok note skip);
my $env = prepare_test({n_tests => 5, start_port => 20800, n_ports => 1}); # runs three of its own tests
if ($^O eq 'MSWin32') {
SKIP: { skip("UNIX Sockets will not work on Win32", 2) };
exit;
}
use_ok('Net::Server');
@Net::Server::Test::ISA = qw(Net::Server);
sub accept {
$env->{'signal_ready_to_test'}->();
return shift->SUPER::accept(@_);
}
my $socket_dir = tempdir(CLEANUP => 1);
my $socket_file = catfile($socket_dir, 'socket'); # must do before fork
sub allow_deny_hook {
my ($server, $client) = @_;
### check the properties of the client socket
if($client->NS_proto eq 'UNIX') {
return $client->NS_port eq $socket_file;
} else {
return $client->NS_port == $env->{'ports'}->[0];
}
}
sub process_request {
my ($self, $client) = @_;
print $client "NS_port: ".$client->NS_port."\n";
return $self->SUPER::process_request($client);
}
my $ok = eval {
local $SIG{'ALRM'} = sub { die "Timeout\n" };
alarm $env->{'timeout'};
my $ppid = $$;
my $pid = fork;
die "Trouble forking: $!" if ! defined $pid;
### parent does the client
if ($pid) {
$env->{'block_until_ready_to_test'}->();
### connect to child under unix
my $remote = IO::Socket::UNIX->new(Peer => $socket_file);
die "No socket returned [$!]" if ! defined $remote;
my $line = <$remote>;
note "# unix port - $line";
$line = <$remote>;
die "Didn't get the type of line we were expecting: ($line)" if $line !~ /Net::Server/;
print $remote "quite\n";
### connect to child under tcp
$remote = NetServerTest::client_connect(
PeerAddr => $env->{'hostname'},
PeerPort => $env->{'ports'}->[0],
Proto => 'tcp') || die "Couldn't open to sock: $!";
$line = <$remote>;
note "# tcp port - $line";
$line = <$remote>;
die "Didn't get the type of line we were expecting: ($line)" if $line !~ /Net::Server/;
print $remote "exit\n";
return 1;
### child does the server
} else {
eval {
close STDERR;
Net::Server::Test->run(
port => "$env->{'ports'}->[0]/tcp",
port => "$socket_file|unix",
user => $UID, # user accepts id as well
group => $GID, # group accepts id as well
host => $env->{'hostname'},
ipv => $env->{'ipv'},
background => 0,
setsid => 0,
);
} || do {
note("Trouble running server: $@");
kill(9, $ppid) && ok(0, "Failed during run of server");
};
exit;
}
alarm(0);
};
alarm(0);
ok($ok, "Got the correct output from the server") || note("Error: $@");
|
