File Coverage

blib/lib/Net/IMP/HTTP/Example/BlockContentType.pm

Criterion	Covered	Total	%
statement	15	62	24.1
branch	0	20	0.0
condition	0	8	0.0
subroutine	5	14	35.7
pod	8	9	88.8
total	28	113	24.7

use strict;
use warnings;

package Net::IMP::HTTP::Example::BlockContentType;
use base 'Net::IMP::HTTP::Request';
use Net::IMP;  # import IMP_ constants
use Net::IMP::Debug;

sub RTYPES { ( IMP_PASS, IMP_DENY ) }
sub new_analyzer {
    my ($factory,%args) = @_;
    my $self = $factory->SUPER::new_analyzer(%args);
    # request data do not matter
    $self->run_callback([ IMP_PASS,0,IMP_MAXOFFSET ]);
    if ( ! $self->{factory_args}{whiterx}
	&& ! $self->{factory_args}{blackrx} ) {
	# nothing to analyze
	$self->run_callback([ IMP_PASS,1,IMP_MAXOFFSET ]);
    }
    return $self;
}

sub validate_cfg {
    my ($class,%cfg) = @_;
    my @err;
    for my $k (qw(whiterx blackrx)) {
	my $rx = delete $cfg{$k} or next;
	ref($rx) and next;
	push @err,"$k is no valid regexp: $@" if ! eval { qr/$rx/ };
    }
    return (@err,$class->SUPER::validate_cfg(%cfg));
}

sub str2cfg {
    my ($class,$str) = @_;
    my %cfg = $class->SUPER::str2cfg($str);
    for my $k (qw(whiterx blackrx)) {
	next if ! $cfg{$k} or ref $cfg{$k};
        $cfg{$k} = eval { qr/$cfg{$k}/ }
	    or die "invalid rx in $k: $@";
    }
    return %cfg;
}

sub request_hdr {}
sub request_body {}
sub response_body {}
sub any_data {}

sub response_hdr {
    my ($self,$hdr) = @_;
    # we only want selected image/ content types and not too big
    my $ct = $hdr =~m{\nContent-type:[ \t]*([^\s;]+)}i && lc($1)
	|| 'unknown/unknown';

    my $reason;
    if ( my $white = $self->{factory_args}{whiterx} ) {
	if ( $ct =~ $white ) {
	    debug("allowed $ct because of white list");
	    goto pass;
	} else {
	    debug("denied $ct because not in white list");
	    $reason = "denied $ct because not in white list";
	    goto deny;
	}
    }
    if ( my $black = $self->{factory_args}{blackrx} ) {
	if ( $ct =~ $black ) {
	    debug("denied $ct because in black list");
	    $reason = "denied $ct because in black list";
	    goto deny;
	} else {
	    debug("allow $ct because not in black list");
	    goto pass;
	}
    }

    pass:
    $self->run_callback([ IMP_PASS,1,IMP_MAXOFFSET ]);
    return;

    deny:
    $self->run_callback([ IMP_DENY,1,$reason ]);
    return;
}


1;
__END__