File Coverage

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

Criterion	Covered	Total	%
statement	18	35	51.4
branch	0	8	0.0
condition	0	3	0.0
subroutine	6	9	66.6
pod	2	3	66.6
total	26	58	44.8

use strict;
use warnings;

package Net::IMP::HTTP::Example::AddXFooHeader;
use base 'Net::IMP::HTTP::Connection';
use fields qw(pos1);
use Net::IMP;
use Net::IMP::HTTP;

sub RTYPES { ( IMP_PASS, IMP_PREPASS, IMP_REPLACE ) }
sub new_analyzer {
    my ($factory,%args) = @_;
    my $analyzer = $factory->SUPER::new_analyzer(%args);

    # we are not interested in request data, only response
    # but for http we need to see requests to pair with responses
    $analyzer->run_callback([IMP_PREPASS,0,IMP_MAXOFFSET]);
    return $analyzer;
}

# data supports IMP_DATA_HTTP and IMP_DATA_HTTPRQ interface
sub data {
    my ($self,$dir,$data,$offset,$type) = @_;

    # request are handled by the pass maxoffset in new_analyzer
    if ( $dir == 0 ) {
	# if we speak httprq we can make the prepass to a pass
	$self->run_callback([IMP_PASS,0,IMP_MAXOFFSET])
	    if $type == IMP_DATA_HTTPRQ_HEADER;
	return;
    }

    $self->{pos1} += length($data);
    if ( $type == IMP_DATA_HTTP_HEADER
	or $type == IMP_DATA_HTTPRQ_HEADER ) {
	$data =~s{\n}{\nX-Foo: bar\r\n};
	my @rv = [
	    IMP_REPLACE,
	    1,
	    $self->{pos1},
	    $data
	];
	if ( $type == IMP_DATA_HTTPRQ_HEADER ) {
	    # for httprq interface we can pass until end of request (maxoffset)
	    push @rv, [ IMP_PASS,1,IMP_MAXOFFSET ];
	}
	$self->run_callback(@rv);

    } else {
	# for http we might get more requests
	# these might be chunking etc so we probably don't know their length
	# therefore we can pass only data we get until we get the next
	# response header
	$self->run_callback([ IMP_PASS,1,$self->{pos1} ]);
    }
}


1;
__END__