From 067835d5b0160e10a778252cf0dc6b2e4d1a74cb Mon Sep 17 00:00:00 2001 From: Dirk Koopman Date: Fri, 21 Jan 2022 22:46:21 +0000 Subject: [PATCH] add missing DXCIDR --- perl/DXCIDR.pm | 156 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 156 insertions(+) create mode 100644 perl/DXCIDR.pm diff --git a/perl/DXCIDR.pm b/perl/DXCIDR.pm new file mode 100644 index 00000000..495768ed --- /dev/null +++ b/perl/DXCIDR.pm @@ -0,0 +1,156 @@ +# +# IP Address block list / checker +# +# This is a DXSpider compatible, optional skin over Net::CIDR::Lite +# If Net::CIDR::Lite is not present, then a find will always returns 0 +# + +package DXCIDR; + +use strict; +use warnings; +use 5.16.1; +use DXVars; +use DXDebug; +use DXUtil; +use DXLog; +use IO::File; +use File::Copy; + +our $active = 0; +our $badipfn = "badip"; +my $ipv4; +my $ipv6; +my $count4 = 0; +my $count6 = 0; + +# load the badip file +sub load +{ + if ($active) { + $count4 = _get($ipv4, 4); + $count6 = _get($ipv6, 6); + } + LogDbg('DXProt', "DXCIDR: loaded $count4 IPV4 addresses and $count6 IPV6 addresses"); + return $count4 + $count6; +} + +sub _fn +{ + return localdata($badipfn) . "$_[0]"; +} + +sub _get +{ + my $list = shift; + my $sort = shift; + my $fn = _fn($sort); + my $fh = IO::File->new($fn); + my $count = 0; + + if ($fh) { + while (<$fh>) { + chomp; + next if /^\s*\#/; + $list->add($_); + ++$count; + } + $fh->close; + $list->clean if $count; + } elsif (-r $fn) { + LogDbg('err', "DXCIDR: $fn not found ($!)"); + } + return $count; +} + +sub _put +{ + my $list = shift; + my $sort = shift; + my $fn = _fn($sort); + my $r = rand; + my $fh = IO::File->new (">$fn.$r"); + if ($fh) { + for ($list->list) { + $fh->print("$_\n"); + } + move "$fn.$r", $fn; + } else { + LogDbg('err', "DXCIDR: cannot write $fn.$r $!"); + } +} + +sub add +{ + for (@_) { + # protect against stupid or malicious + next if /^127\./; + next if /^::1$/; + if (/\./) { + $ipv4->add($_); + ++$count4; + LogDbg('DXProt', "DXCIDR: Added IPV4 $_ address"); + } else { + $ipv6->add($_); + ++$count6; + LogDbg('DXProt', "DXCIDR: Added IPV6 $_ address"); + } + } + if ($ipv4 && $count4) { + $ipv4->prep_find; + _put($ipv4, 4); + } + if ($ipv6 && $count6) { + $ipv6->prep_find; + _put($ipv6, 6); + } +} + +sub save +{ + return 0 unless $active; + my $list = $ipv4->list; + _put($list, 4) if $list; + $list = $ipv6->list; + _put($list, 6) if $list; +} + +sub list +{ + my @out; + push @out, $ipv4->list; + push @out, $ipv6->list; + return (1, sort @out); +} + +sub find +{ + return 0 unless $active; + return 0 unless $_[0]; + + if ($_[0] =~ /\./) { + return $ipv4->find($_[0]) if $count4; + } + return $ipv6->find($_[0]) if $count6; +} + +sub init +{ + eval { require Net::CIDR::Lite }; + if ($@) { + LogDbg('DXProt', "DXCIDR: load (cpanm) the perl module Net::CIDR::Lite to check for bad IP addresses (or CIDR ranges)"); + return; + } + + import Net::CIDR::Lite; + + $ipv4 = Net::CIDR::Lite->new; + $ipv6 = Net::CIDR::Lite->new; + + load(); + $active = 1; +} + + + +1; -- 2.34.1