]> www.dxcluster.org Git - spider.git/blob - perl/DXCIDR.pm
remove 5.10.1 isms (//= etc)
[spider.git] / perl / DXCIDR.pm
1 #
2 # IP Address block list / checker
3 #
4 # This is a DXSpider compatible, optional skin over Net::CIDR::Lite
5 # If Net::CIDR::Lite is not present, then a find will always returns 0
6 #
7
8 package DXCIDR;
9
10 use strict;
11 use warnings;
12 use 5.16.1;
13 use DXVars;
14 use DXDebug;
15 use DXUtil;
16 use DXLog;
17 use IO::File;
18 use File::Copy;
19
20 our $active = 0;
21 our $badipfn = "badip";
22 my $ipv4;
23 my $ipv6;
24 my $count4 = 0;
25 my $count6 = 0;
26
27 # load the badip file
28 sub load
29 {
30         if ($active) {
31                 $count4 = _get($ipv4, 4);
32                 $count6 = _get($ipv6, 6);
33         }
34         LogDbg('DXProt', "DXCIDR: loaded $count4 IPV4 addresses and $count6 IPV6 addresses");
35         return $count4 + $count6;
36 }
37
38 sub _fn
39 {
40         return localdata($badipfn) . "$_[0]";
41 }
42
43 sub _get
44 {
45         my $list = shift;
46         my $sort = shift;
47         my $fn = _fn($sort);
48         my $fh = IO::File->new($fn);
49         my $count = 0;
50         
51         if ($fh) {
52                 while (<$fh>) {
53                         chomp;
54                         next if /^\s*\#/;
55                         $list->add($_);
56                         ++$count;
57                 }
58                 $fh->close;
59                 $list->clean if $count;
60         } elsif (-r $fn) {
61                 LogDbg('err', "DXCIDR: $fn not found ($!)");
62         }
63         return $count;
64 }
65
66 sub _put
67 {
68         my $list = shift;
69         my $sort = shift;
70         my $fn = _fn($sort);
71         my $r = rand;
72         my $fh = IO::File->new (">$fn.$r");
73         if ($fh) {
74                 for ($list->list) {
75                         $fh->print("$_\n");
76                 }
77                 move "$fn.$r", $fn;
78         } else {
79                 LogDbg('err', "DXCIDR: cannot write $fn.$r $!");
80         }
81 }
82
83 sub add
84 {
85         for (@_) {
86                 # protect against stupid or malicious
87                 next if /^127\./;
88                 next if /^::1$/;
89                 if (/\./) {
90                         $ipv4->add($_);
91                         ++$count4;
92                         LogDbg('DXProt', "DXCIDR: Added IPV4 $_ address");
93                 } else {
94                         $ipv6->add($_);
95                         ++$count6;
96                         LogDbg('DXProt', "DXCIDR: Added IPV6 $_ address");
97                 }
98         }
99         if ($ipv4 && $count4) {
100                 $ipv4->prep_find;
101                 _put($ipv4, 4);
102         }
103         if ($ipv6 && $count6) {
104                 $ipv6->prep_find;
105                 _put($ipv6, 6);
106         }
107 }
108
109 sub save
110 {
111         return 0 unless $active;
112         my $list = $ipv4->list;
113         _put($list, 4) if $list;
114         $list = $ipv6->list;
115         _put($list, 6) if $list;
116 }
117
118 sub list
119 {
120         my @out;
121         push @out, $ipv4->list;
122         push @out, $ipv6->list;
123         return (1, sort @out);
124 }
125
126 sub find
127 {
128         return 0 unless $active;
129         return 0 unless $_[0];
130         
131         if ($_[0] =~ /\./) {
132                 return $ipv4->find($_[0]) if $count4;
133         }
134         return $ipv6->find($_[0]) if $count6;
135 }
136
137 sub init
138 {
139         eval { require Net::CIDR::Lite };
140         if ($@) {
141                 LogDbg('DXProt', "DXCIDR: load (cpanm) the perl module Net::CIDR::Lite to check for bad IP addresses (or CIDR ranges)");
142                 return;
143         }
144
145         import Net::CIDR::Lite;
146
147         $ipv4 = Net::CIDR::Lite->new;
148         $ipv6 = Net::CIDR::Lite->new;
149
150         load();
151         $active = 1;
152 }
153
154
155
156 1;