=head1 NAME Net::SIP::Dropper::ByIPPort - drops SIP messages based on senders IP and port =head1 SYNOPSIS use Net::SIP::Dropper::ByIPPort; my $drop_by_ipport = Net::SIP::Dropper::ByIPPort->new( database => '/path/to/database.drop', methods => [ 'REGISTER', '...', '' ], attempts => 10, interval => 60, ); my $dropper = Net::SIP::Dropper->new( cb => $drop_by_ipport ); my $chain = Net::SIP::ReceiveChain->new([ $dropper, ... ]); =head1 DESCRIPTION With C<Net::SIP::Dropper::ByIPPort> one can drop packets, if too much packets are received from the same IP and port within a specific interval. This is to stop bad behaving clients. =cut use strict; use warnings; package Net::SIP::Dropper::ByIPPort; use Net::SIP::Debug; use Net::SIP::Util 'invoke_callback'; use fields qw(interval attempts methods dbcb data); =head1 CONSTRUCTOR =over 4 =item new ( ARGS ) ARGS is a hash with the following keys: =over 8 =item database Optional file name of database or callback for storing/retrieving the data. If it is a callback it will be called with C<< $callback->(\%data) >> to retrieve the data (C<%data> will be updated) and C<< $callback->(\%data,true) >> to save the data. No return value will be expected from the callback. %data contains the number of attempts from a specific IP, port at a specific time in the following format: C<< $data{ip}{port}{time} = count >> =item attempts After how many attempts within the specific interval the packet will be dropped. Argument is required. =item interval The interval for attempts. Argument is required. =item methods Optional argument to restrict dropping to specific methods. Is array reference of method names, if one of the names is empty also responses will be considered. If not given all packets will be checked. =back =back =cut sub new { my ($class,%args) = @_; my $interval = delete $args{interval} or croak('interval should be defined'); my $attempts = delete $args{attempts} or croak('attempts should be defined'); my $methods = delete $args{methods}; # optional my %ips_ports; my $dbcb; if ( my $db = delete $args{database} ) { if ( ! ref $db ) { # file name require Storable; if ( ! -e $db ) { # initialize DB Storable::store(\%ips_ports, $db) or croak("cannot create $db: $!"); } $dbcb = [ sub { my ($file,$data,$save) = @_; if ( $save ) { Storable::store($data,$file); } else { %$data = %{ Storable::retrieve($file) } } }, $db ]; } else { $dbcb = $db } # load contents of database invoke_callback($dbcb,\%ips_ports); DEBUG_DUMP(100, \%ips_ports); } # initialize object my Net::SIP::Dropper::ByIPPort $self = fields::new($class); $self->{data} = \%ips_ports; $self->{interval} = $interval; $self->{attempts} = $attempts; $self->{methods} = $methods; $self->{dbcb} = $dbcb; return $self } =head1 METHODS =over 4 =item run ( PACKET, LEG, FROM ) This method is called as a callback from the L<Net::SIP::Dropper> object. It returns true if the packet should be dropped, e.g. if there are too much packets from the same ip,port within the given interval. =cut sub run { my Net::SIP::Dropper::ByIPPort $self = shift; my ($packet,$leg,$from) = @_; # expire current contents $self->expire; # check if the packet type/method fits if (my $m = $self->{methods}) { if ($packet->is_response) { return if ! grep { !$_ } @$m } else { my $met = $packet->method; return if ! grep { $_ eq $met } @$m } }; # enter ip,port into db my ($ip,$port) = ($from->{addr},$from->{port}); $self->{data}{$ip}{$port}{ time() }++; $self->savedb(); # count attempts in interval # because everything outside of interval is expired we can # just look at all entries for ip,port my $count = 0; for (values %{$self->{data}{$ip}{$port}} ) { $count += $_; } # by using port = 0 one can block the whole IP for (values %{$self->{data}{$ip}{0} || {}} ) { $count += $_; } # drop if too much attempts if ( $count >= $self->{attempts} ) { DEBUG(1,"message dropped because $ip:$port was in database with $count attempts"); return 1; } return; } =item expire This method is called from within C<run> but can also be called by hand. It will expire all entries which are outside of the interval. =cut sub expire { my Net::SIP::Dropper::ByIPPort $self = shift; my $interval = $self->{interval}; my $data = $self->{data}; my $maxtime = time() - $interval; my $changed; for my $ip ( keys %$data ) { my $ipp = $data->{$ip}; for my $port (keys %$ipp) { my $ippt = $ipp->{$port}; for my $time (keys %$ippt) { if ($time<=$maxtime) { delete $ippt->{$time}; $changed = 1; } } delete $ipp->{$port} if ! %$ippt; } delete $data->{$ip} if ! %$ipp; } $self->savedb if $changed; } =item savedb This method is called from C<expire> and C<run> for saving to the database after changes, but can be called by hand to, useful if you made manual changes using the C<data> method. =cut sub savedb { my Net::SIP::Dropper::ByIPPort $self = shift; my $dbcb = $self->{dbcb} or return; invoke_callback($dbcb,$self->{data},'save') } =item data This method gives access to the internal hash which stores the attempts. An attempt from a specific IP and port and a specific time (as int, like time() gives) will be added to C<< $self->data->{ip}{port}{time} >>. By manually manipulating the hash one can restrict a specific IP,port forever (just set time to a large value and add a high number of attempts) or even restrict access for the whole IP (all ports) until time by using a port number of 0. After changes to the data it is advised to call C<savedb>. =cut sub data { my Net::SIP::Dropper::ByIPPort $self = shift; return $self->{data} } =pod =back =cut 1;