[1/2] geoip-locations.pl: Add get_geoip_locations().

Message ID 20190110194004.3341-1-stefan.schantl@ipfire.org
State Accepted
Commit 30c59cbb0b370a0fd1470087a94d91b35d4e54a3
Headers
Series [1/2] geoip-locations.pl: Add get_geoip_locations(). |

Commit Message

Stefan Schantl Jan. 11, 2019, 6:40 a.m. UTC
  This function is used to get all available GeoIP locations.

The functions returns them as array, sorted in alphabetical order.

Reference #11959

Signed-off-by: Stefan Schantl <stefan.schantl@ipfire.org>
---
 config/cfgroot/geoip-functions.pl | 33 +++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
  

Patch

diff --git a/config/cfgroot/geoip-functions.pl b/config/cfgroot/geoip-functions.pl
index e8ce8377f..9f86a09ba 100644
--- a/config/cfgroot/geoip-functions.pl
+++ b/config/cfgroot/geoip-functions.pl
@@ -178,4 +178,37 @@  sub get_full_country_name($) {
 	return $name;
 }
 
+# Function to get all available GeoIP locations.
+sub get_geoip_locations() {
+	my @locations;
+
+	# Open the location database.
+	open(LOCATION, "$geoip_database_dir/$location_database") or die "Could not open $geoip_database_dir/$location_database. $!\n";
+
+	# Loop through the file.
+	while(my $line = <LOCATION>) {
+		# Remove newlines.
+		chomp($line);
+
+		# Split the line content.
+		my ($geoname_id, $locale_code, $continent_code, $continent_name, $country_iso_code, $country_name, $is_in_european_union) = split(/\,/, $line);
+
+		# Check if the country_iso_code is upper case.
+		if($country_iso_code =~ /[A-Z]/) {
+			# Add the current ISO code.
+			push(@locations, $country_iso_code);
+		}
+	}
+
+	# Close filehandle.
+	close(LOCATION);
+
+	# Sort locations array in alphabetical order.
+	my @sorted_locations = sort(@locations);
+
+	# Return the array..
+	return @sorted_locations;
+}
+
+
 1;