changeset 1062:f1cc64e4e67d

Tests: basic tests for stream geoip module.
author Andrey Zelenkov <zelenkov@nginx.com>
date Fri, 14 Oct 2016 15:20:28 +0300
parents 83357211563e
children d62747c321a6
files stream_geoip.t
diffstat 1 files changed, 192 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
new file mode 100644
--- /dev/null
+++ b/stream_geoip.t
@@ -0,0 +1,192 @@
+#!/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 ipv6/)
+	->has('stream_realip')->plan(20);
+
+$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->run();
+
+###############################################################################
+
+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()->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);
+}
+
+###############################################################################