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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
#!perl
use 5.14.2;
use warnings;
use Test::More;
use Test::Fatal;
use Zonemaster::Engine::Util;
use Zonemaster::Engine::Nameserver;
use Zonemaster::LDNS;
use Sub::Override;
my $datafile = 't/nameserver-axfr.data';
my %saved_axfr;
my $override = Sub::Override->new();
setup( $datafile );
# This should be a successful AXFR
my $ns = Zonemaster::Engine::Nameserver->new( { name => 'kennedy.faerywicca.se', address => '46.21.106.227' } );
my $counter = 0;
is(
exception {
$ns->axfr( 'cyberpomo.com', sub { $counter += 1; return 1; } );
},
undef,
'No exception'
);
ok( ( $counter > 10 ), 'At least ten records seen' );
# This should be a refused AXFR
$counter = 0;
my $ns2 = Zonemaster::Engine::Nameserver->new( { name => 'ns.nic.se', address => '91.226.36.45' } );
like(
exception {
$ns2->axfr( 'iis.se', sub { $counter += 1; return 1; } );
},
qr/REFUSED/,
'AXFR was refused'
);
is( $counter, 0, 'No records seen' );
finish( $datafile );
done_testing;
###
### Functions to record and replay AXFRs.
###
sub setup {
my ( $datafile ) = @_;
if ( not $ENV{ZONEMASTER_RECORD} ) {
# Replay
die "Stored data file missing" if not -r $datafile;
open my $fh, '<', $datafile or die "Failed to open $datafile for reading: $!\n";
while ( my $line = $fh->getline ) {
my ( $domain, $type, $str ) = split( /\t/, $line, 3 );
if ( $type eq 'RR' ) {
my $rr = eval { Zonemaster::LDNS::RR->new( $str ) };
if ( $rr ) {
push @{ $saved_axfr{$domain} }, $rr;
}
else {
warn "Failed to parse: $str\n";
}
}
elsif ( $type eq 'EXCEPTION' ) {
$saved_axfr{$domain} = $str;
}
}
Zonemaster::Engine::Profile->effective->set( q{no_network}, 1 );
$override->override(
'Zonemaster::Engine::Nameserver::axfr',
sub {
my ( $self, $domain, $callback, $class ) = @_;
if ( exists $saved_axfr{$domain} ) {
if ( ref( $saved_axfr{$domain} ) ) {
while ( my $rr = pop @{ $saved_axfr{$domain} } ) {
$callback->( $rr );
}
}
else {
die $saved_axfr{$domain};
}
}
else {
die "AXFR Request for domain that has not been saved.";
}
}
);
} ## end if ( not $ENV{ZONEMASTER_RECORD...})
else {
# Record
$override->wrap(
'Zonemaster::Engine::Nameserver::axfr',
sub {
my ( $old_axfr, $self, $domain, $callback, $class ) = @_;
my @rrs;
my $new_cb = sub {
push @rrs, $_[0];
$callback->( $_[0] );
};
my $result = eval { $old_axfr->( $self, $domain, $new_cb, $class ) };
if ( $@ ) {
$saved_axfr{$domain} = "$@";
die $@;
}
else {
$saved_axfr{$domain} = \@rrs;
}
}
);
} ## end else [ if ( not $ENV{ZONEMASTER_RECORD...})]
} ## end sub setup
sub finish {
my ( $datafile ) = @_;
if ( $ENV{ZONEMASTER_RECORD} ) {
open my $fh, '>', $datafile or die "Failed to open $datafile for writing: $!\n";
while ( my ( $domain, $aref ) = each %saved_axfr ) {
if ( ref $aref ) {
say $fh $domain . "\tRR\t" . $_->string for @$aref;
}
else {
chomp( $aref );
say $fh $domain . "\tEXCEPTION\t$aref";
}
}
close $fh;
}
}
|
