byterazor/qpsmtpd
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add headers with GeoIP data

Browse Source
This commit is contained in:
Jared Johnson 2014-11-07 16:30:54 -06:00
parent da3ed5ebbf
commit d22396c298
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
plugins/ident/geoip
View File

@ -130,6 +130,7 @@ sub register {
$self->{_args}{db_dir} ||= '/usr/local/share/GeoIP'; $self->{_args}{db_dir} ||= '/usr/local/share/GeoIP';
$self->load_geoip() or return; $self->load_geoip() or return;
$self->register_hook( data_post => 'add_headers' );
} }
sub load_geoip { sub load_geoip {
@ -197,6 +198,17 @@ sub load_geoip2 {
return; return;
} }
sub add_headers {
my ( $self, $txn ) = @_;
for my $h (qw( Country Country-Name Continent City ASN )) {
my $note = lc "geoip_$h";
$h =~ s/-/_/;
next if ! $self->connection->notes($note);
$txn->header->delete("X-GeoIP-$h");
$txn->header->add( "X-GeoIP-$h", $self->connection->notes($note), 0 );
}
}
sub geoip2_lookup { sub geoip2_lookup {
my $self = shift; my $self = shift;