| #!/usr/bin/perl |
| |
| # (C) Andrey Zelenkov |
| # (C) Nginx, Inc. |
| |
| # Tests for stream geoip module. |
| |
| ############################################################################### |
| |
| use warnings; |
| use strict; |
| |
| use Test::More; |
| |
| use Socket qw/ $CRLF /; |
| |
| BEGIN { use FindBin; chdir($FindBin::Bin); } |
| |
| use lib 'lib'; |
| use Test::Nginx; |
| use Test::Nginx::Stream qw/ stream /; |
| |
| ############################################################################### |
| |
| select STDERR; $| = 1; |
| select STDOUT; $| = 1; |
| |
| my $t = Test::Nginx->new()->has(qw/stream stream_geoip stream_return/) |
| ->has('stream_realip'); |
| |
| $t->write_file_expand('nginx.conf', <<'EOF'); |
| |
| %%TEST_GLOBALS%% |
| |
| daemon off; |
| |
| events { |
| } |
| |
| stream { |
| set_real_ip_from 127.0.0.1/32; |
| |
| geoip_country %%TESTDIR%%/country.dat; |
| geoip_city %%TESTDIR%%/city.dat; |
| geoip_org %%TESTDIR%%/org.dat; |
| |
| server { |
| listen 127.0.0.1:8080 proxy_protocol; |
| return "country_code:$geoip_country_code |
| country_code3:$geoip_country_code3 |
| country_name:$geoip_country_name |
| |
| area_code:$geoip_area_code |
| city_continent_code:$geoip_city_continent_code |
| city_country_code:$geoip_city_country_code |
| city_country_code3:$geoip_city_country_code3 |
| city_country_name:$geoip_city_country_name |
| dma_code:$geoip_dma_code |
| latitude:$geoip_latitude |
| longitude:$geoip_longitude |
| region:$geoip_region |
| region_name:$geoip_region_name |
| city:$geoip_city |
| postal_code:$geoip_postal_code |
| |
| org:$geoip_org"; |
| } |
| } |
| |
| EOF |
| |
| my $d = $t->testdir(); |
| |
| # country database: |
| # |
| # "10.0.0.1","10.0.0.1","RU","Russian Federation" |
| # "2001:db8::","2001:db8::","US","United States" |
| |
| my $data = ''; |
| |
| for my $i (0 .. 156) { |
| # skip to offset 32 if 1st bit set in ipv6 address wins |
| $data .= pack_node($i + 1) . pack_node(32), next if $i == 2; |
| # otherwise default to RU |
| $data .= pack_node(0xffffb9) . pack_node(0xffff00), next if $i == 31; |
| # continue checking bits set in ipv6 address |
| $data .= pack_node(0xffff00) . pack_node($i + 1), next |
| if grep $_ == $i, (44, 49, 50, 52, 53, 55, 56, 57); |
| # last bit set in ipv6 address |
| $data .= pack_node(0xffffe1) . pack_node(0xffff00), next if $i == 156; |
| $data .= pack_node($i + 1) . pack_node(0xffff00); |
| } |
| |
| $data .= chr(0x00) x 3; |
| $data .= chr(0xFF) x 3; |
| $data .= chr(12); |
| |
| $t->write_file('country.dat', $data); |
| |
| # city database: |
| # |
| # "167772161","167772161","RU","48","Moscow","119034","55.7543",37.6202",, |
| |
| $data = ''; |
| |
| for my $i (0 .. 31) { |
| $data .= pack_node(32) . pack_node($i + 1), next if $i == 4 or $i == 6; |
| $data .= pack_node(32) . pack_node($i + 2), next if $i == 31; |
| $data .= pack_node($i + 1) . pack_node(32); |
| } |
| |
| $data .= chr(42); |
| $data .= chr(185); |
| $data .= pack('Z*', 48); |
| $data .= pack('Z*', 'Moscow'); |
| $data .= pack('Z*', 119034); |
| $data .= pack_node(int((55.7543 + 180) * 10000)); |
| $data .= pack_node(int((37.6202 + 180) * 10000)); |
| $data .= chr(0) x 3; |
| $data .= chr(0xFF) x 3; |
| $data .= chr(2); |
| $data .= pack_node(32); |
| |
| $t->write_file('city.dat', $data); |
| |
| # organization database: |
| # |
| # "167772161","167772161","Nginx" |
| |
| $data = ''; |
| |
| for my $i (0 .. 31) { |
| $data .= pack_org(32) . pack_org($i + 1), next if $i == 4 or $i == 6; |
| $data .= pack_org(32) . pack_org($i + 2), next if $i == 31; |
| $data .= pack_org($i + 1) . pack_org(32); |
| } |
| |
| $data .= chr(42); |
| $data .= pack('Z*', 'Nginx'); |
| $data .= chr(0xFF) x 3; |
| $data .= chr(5); |
| $data .= pack_node(32); |
| |
| $t->write_file('org.dat', $data); |
| $t->try_run('no inet6 support')->plan(20); |
| |
| ############################################################################### |
| |
| my %data = stream_pp('10.0.0.1') =~ /(\w+):(.*)/g; |
| is($data{country_code}, 'RU', 'geoip country code'); |
| is($data{country_code3}, 'RUS', 'geoip country code 3'); |
| is($data{country_name}, 'Russian Federation', 'geoip country name'); |
| |
| is($data{area_code}, 0, 'geoip area code'); |
| is($data{city_continent_code}, 'EU', 'geoip city continent code'); |
| is($data{city_country_code}, 'RU', 'geoip city country code'); |
| is($data{city_country_code3}, 'RUS', 'geoip city country code 3'); |
| is($data{city_country_name}, 'Russian Federation', 'geoip city country name'); |
| is($data{dma_code}, 0, 'geoip dma code'); |
| is($data{latitude}, 55.7543, 'geoip latitude'); |
| is($data{longitude}, 37.6202, 'geoip longitude'); |
| is($data{region}, 48, 'geoip region'); |
| is($data{region_name}, 'Moscow City', 'geoip region name'); |
| is($data{city}, 'Moscow', 'geoip city'); |
| is($data{postal_code}, 119034, 'geoip postal code'); |
| |
| is($data{org}, 'Nginx', 'geoip org'); |
| |
| like(stream_pp('::ffff:10.0.0.1'), qr/org:Nginx/, 'geoip ipv6 ipv4-mapped'); |
| |
| %data = stream_pp('2001:db8::') =~ /(\w+):(.*)/g; |
| is($data{country_code}, 'US', 'geoip ipv6 country code'); |
| is($data{country_code3}, 'USA', 'geoip ipv6 country code 3'); |
| is($data{country_name}, 'United States', 'geoip ipv6 country name'); |
| |
| ############################################################################### |
| |
| sub stream_pp { |
| my ($ip) = @_; |
| my $type = ($ip =~ ':' ? 'TCP6' : 'TCP4'); |
| return stream('127.0.0.1:' . port(8080)) |
| ->io("PROXY $type $ip 127.0.0.1 8080 8080${CRLF}"); |
| } |
| |
| sub pack_node { |
| substr pack('V', shift), 0, 3; |
| } |
| |
| sub pack_org { |
| pack('V', shift); |
| } |
| |
| ############################################################################### |