From a9d7f2034af6a28287ef85682a2fa1b6c8e0d05b Mon Sep 17 00:00:00 2001
From: jim-p <jimp@pfsense.org>
Date: Fri, 3 Apr 2015 13:25:55 -0400
Subject: Add some fixes/additions for better handling of GeoIP data in ntopng.
 Fixes #4567 Also add attribution text for GeoIP data and a link back to
 maxmind for CC BY-SA 3.0.

---
 config/ntopng/ntopng.xml | 42 +++++++++++++++++++++++++++++++++++++++---
 pkg_config.10.xml        |  2 +-
 pkg_config.8.xml         |  2 +-
 pkg_config.8.xml.amd64   |  2 +-
 4 files changed, 42 insertions(+), 6 deletions(-)

diff --git a/config/ntopng/ntopng.xml b/config/ntopng/ntopng.xml
index f287768b..9382912a 100644
--- a/config/ntopng/ntopng.xml
+++ b/config/ntopng/ntopng.xml
@@ -39,7 +39,7 @@
 	]]>
 	</copyright>
 	<name>ntopng</name>
-	<version>1.2 v0.4</version>
+	<version>1.2 v0.5</version>
 	<title>Diagnostics: ntopng Settings</title>
 	<savetext>Change</savetext>
 	<aftersaveredirect>pkg_edit.php?xml=ntopng.xml</aftersaveredirect>
@@ -96,7 +96,12 @@
 		<field>
 			<fielddescr>DNS Mode</fielddescr>
 			<fieldname>dns_mode</fieldname>
-			<description>Configures how name resolution is handled</description>
+			<description>
+				Configures how name resolution is handled&lt;br/&gt;&lt;br/&gt;
+				Additionally, GeoIP Data can provide location information about IP addresses.&lt;br/&gt;
+				This product includes GeoLite data created by MaxMind, available from &lt;a href=&quot;http://www.maxmind.com&quot;&gt;http://www.maxmind.com&lt;/a&gt;.&lt;br/&gt;
+				&lt;input type=&quot;submit&quot; name=&quot;Submit&quot; value=&quot;Update GeoIP Data&quot;&gt;
+			</description>
 			<type>select</type>
 			<default_value>0</default_value>
 			<options>
@@ -134,6 +139,9 @@
 	<custom_php_global_functions>
 	<![CDATA[
 	function sync_package_ntopng() {
+		if ($_POST['Submit'] == "Update GeoIP Data") {
+			return;
+		}
 		conf_mount_rw();
 		config_lock();
 		global $config;
@@ -222,6 +230,7 @@
 			// Wait for ntopng to shut down cleanly.
 			sleep(20);
 		}
+		ntopng_fixup_geoip();
 		start_service("ntopng");
 		sleep(2);
 
@@ -233,6 +242,30 @@
 		conf_mount_ro();
 		config_unlock();
 	}
+	function ntopng_update_geoip() {
+		mwexec("/usr/pbi/ntopng-" . php_uname("m") . "/bin/ntopng-geoipupdate.sh");
+		ntopng_fixup_geoip();
+		restart_service("ntopng");
+	}
+	function ntopng_fixup_geoip() {
+		$target_dir = "/usr/pbi/ntopng-" . php_uname("m") . "/local/share/ntopng/httpdocs/geoip";
+		$source_dir = "/usr/pbi/ntopng-" . php_uname("m") . "/share/ntopng";
+
+		foreach(glob("{$source_dir}/Geo*.dat*") as $geofile) {
+			/* Decompress if needed. */
+			if (substr($geofile, -3, 3) == ".gz") {
+				mwexec("/usr/bin/gzip -d " . escapeshellarg($geofile));
+			}
+		}
+
+		/* Use a separate glob since the filenames could have changed since the last run */
+		foreach(glob("{$source_dir}/Geo*.dat*") as $geofile) {
+			$target_file = $target_dir . '/' . basename($geofile);
+			if (!file_exists($target_file)) {
+				symlink($geofile, $target_file);
+			}
+		}
+	}
 	]]>
 	</custom_php_global_functions>
 	<custom_add_php_command>
@@ -243,7 +276,7 @@
 	</custom_php_resync_config_command>
 	<custom_php_install_command>
 	<![CDATA[
-		mwexec_bg("/usr/pbi/ntopng-" . php_uname("m") . "/bin/geoipupdate.sh");
+		ntopng_update_geoip();
 		sync_package_ntopng();
 	]]>
 	</custom_php_install_command>
@@ -257,6 +290,9 @@
 			$input_errors[] = "You must provide (and confirm) ntopng's password.";
 		if ($_POST['password'] != $_POST['passwordagain'])
 			$input_errors[] = "The provided passwords did not match.";
+		if ($_POST['Submit'] == "Update GeoIP Data") {
+			ntopng_update_geoip();
+		}
 	}
 	]]>
 	</custom_php_validation_command>
diff --git a/pkg_config.10.xml b/pkg_config.10.xml
index a6c25cb9..e4f10f76 100644
--- a/pkg_config.10.xml
+++ b/pkg_config.10.xml
@@ -263,7 +263,7 @@
 			<ports_before>databases/redis databases/gdbm net/GeoIP x11-fonts/font-util x11-fonts/webfonts graphics/graphviz</ports_before>
 			<port>net/ntopng</port>
 		</build_pbi>
-		<version>1.2.1 v0.4</version>
+		<version>1.2.1 v0.5</version>
 		<status>ALPHA</status>
 		<required_version>2.2</required_version>
 		<config_file>https://packages.pfsense.org/packages/config/ntopng/ntopng.xml</config_file>
diff --git a/pkg_config.8.xml b/pkg_config.8.xml
index 19d56a4d..6eefef46 100644
--- a/pkg_config.8.xml
+++ b/pkg_config.8.xml
@@ -389,7 +389,7 @@
 			<ports_before>databases/redis databases/gdbm net/GeoIP x11-fonts/font-util x11-fonts/webfonts graphics/graphviz</ports_before>
 			<port>net/ntopng</port>
 		</build_pbi>
-		<version>1.1 v0.4</version>
+		<version>1.1 v0.5</version>
 		<status>ALPHA</status>
 		<required_version>2.1.4</required_version>
 		<config_file>https://packages.pfsense.org/packages/config/ntopng/ntopng.xml</config_file>
diff --git a/pkg_config.8.xml.amd64 b/pkg_config.8.xml.amd64
index d0887c45..582a602e 100644
--- a/pkg_config.8.xml.amd64
+++ b/pkg_config.8.xml.amd64
@@ -376,7 +376,7 @@
 			<ports_before>databases/redis databases/gdbm net/GeoIP x11-fonts/font-util x11-fonts/webfonts graphics/graphviz</ports_before>
 			<port>net/ntopng</port>
 		</build_pbi>
-		<version>1.1 v0.4</version>
+		<version>1.1 v0.5</version>
 		<status>ALPHA</status>
 		<required_version>2.1.4</required_version>
 		<config_file>https://packages.pfsense.org/packages/config/ntopng/ntopng.xml</config_file>
-- 
cgit v1.2.3