]>
Commit | Line | Data |
---|---|---|
1dcd8715 SS |
1 | #!/usr/bin/perl -w |
2 | ############################################################################ | |
3 | # # | |
4 | # This file is part of the IPFire Firewall. # | |
5 | # # | |
6 | # IPFire is free software; you can redistribute it and/or modify # | |
7 | # it under the terms of the GNU General Public License as published by # | |
8 | # the Free Software Foundation; either version 2 of the License, or # | |
9 | # (at your option) any later version. # | |
10 | # # | |
11 | # IPFire is distributed in the hope that it will be useful, # | |
12 | # but WITHOUT ANY WARRANTY; without even the implied warranty of # | |
13 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # | |
14 | # GNU General Public License for more details. # | |
15 | # # | |
16 | # You should have received a copy of the GNU General Public License # | |
17 | # along with IPFire; if not, write to the Free Software # | |
18 | # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA # | |
19 | # # | |
48152fae | 20 | # Copyright (C) 2015 - 2020 IPFire Team <info@ipfire.org>. # |
1dcd8715 SS |
21 | # # |
22 | ############################################################################ | |
23 | ||
48152fae | 24 | package Location::Functions; |
1dcd8715 | 25 | |
8a64d10f | 26 | use Location; |
1dcd8715 | 27 | |
e34dbea7 SS |
28 | # Hash which contains country codes and their names which are special or not |
29 | # part of ISO 3166-1. | |
30 | my %not_iso_3166_location = ( | |
b868abd2 SS |
31 | "A1" => "Anonymous Proxy", |
32 | "A2" => "Satellite Provider", | |
33 | "A3" => "Worldwide Anycast Instance", | |
e34dbea7 SS |
34 | ); |
35 | ||
d443f504 SS |
36 | # Hash which contains possible network flags and their mapped location codes. |
37 | my %network_flags = ( | |
38 | "LOC_NETWORK_FLAG_ANONYMOUS_PROXY" => "A1", | |
39 | "LOC_NETWORK_FLAG_SATELLITE_PROVIDER" => "A2", | |
40 | "LOC_NETWORK_FLAG_ANYCAST" => "A3", | |
41 | ); | |
42 | ||
79b564c8 SS |
43 | # Array which contains special country codes. |
44 | my @special_locations = ( "A1", "A2", "A3" ); | |
45 | ||
8a64d10f | 46 | # Directory where the libloc database and keyfile lives. |
a3afe905 | 47 | our $location_dir = "/var/lib/location/"; |
8a64d10f SS |
48 | |
49 | # Libloc database file. | |
9b2594d8 | 50 | our $database = "$location_dir/database.db"; |
8a64d10f SS |
51 | |
52 | # Libloc keyfile to verify the database. | |
9b2594d8 SS |
53 | our $keyfile = "$location_dir/signing-key.pem"; |
54 | ||
55 | # Directory which contains the exported databases. | |
56 | our $xt_geoip_db_directory = "/usr/share/xt_geoip/"; | |
8a64d10f | 57 | |
b62d7e0c SS |
58 | # Create libloc database handle. |
59 | my $db_handle = &init(); | |
60 | ||
8a64d10f SS |
61 | # |
62 | ## Tiny function to init the location database. | |
63 | # | |
64 | sub init () { | |
65 | # Init and open the database. | |
66 | my $db = &Location::init($database); | |
67 | ||
68 | # Return the database handle. | |
69 | return $db; | |
70 | } | |
00793c27 | 71 | |
8a64d10f SS |
72 | # |
73 | ## Function to verify the integrity of the location database. | |
74 | # | |
75 | sub verify ($) { | |
76 | my ($db_handle) = @_; | |
77 | ||
78 | # Verify the integrity of the database. | |
79 | if(&Location::verify($db_handle, $keyfile)) { | |
80 | # Success, return "1". | |
81 | return 1; | |
00793c27 MT |
82 | } |
83 | ||
8a64d10f SS |
84 | # If we got here, return nothing. |
85 | return; | |
86 | } | |
87 | ||
88 | # | |
89 | ## Function to the the country code of a given address. | |
90 | # | |
91 | sub lookup_country_code($$) { | |
b62d7e0c | 92 | my ($address) = @_; |
8a64d10f SS |
93 | |
94 | # Lookup the given address. | |
95 | my $country_code = &Location::lookup_country_code($db_handle, $address); | |
96 | ||
0ca3baed | 97 | # Return the name of the country |
8a64d10f | 98 | return $country_code; |
00793c27 MT |
99 | } |
100 | ||
1dcd8715 SS |
101 | # Function to get the flag icon for a specified country code. |
102 | sub get_flag_icon($) { | |
103 | my ($input) = @_; | |
104 | ||
105 | # Webserver's root dir. (Required for generating full path) | |
106 | my $webroot = "/srv/web/ipfire/html"; | |
107 | ||
108 | # Directory which contains the flag icons. | |
109 | my $flagdir = "/images/flags"; | |
110 | ||
111 | # File extension of the country flags. | |
112 | my $ext = "png"; | |
113 | ||
114 | # Remove whitespaces. | |
115 | chomp($input); | |
116 | ||
dfbee171 SS |
117 | # Convert given country code to upper case. |
118 | my $ccode = uc($input); | |
1dcd8715 SS |
119 | |
120 | # Generate filename, based on the contry code in lower case | |
121 | # and the defined file extension. | |
122 | my $file = join('.', $ccode,$ext); | |
123 | ||
124 | # Generate path inside webroot to the previously generated file. | |
125 | my $flag_icon = join('/', $flagdir,$file); | |
126 | ||
127 | # Generate absolute path to the icon file. | |
128 | my $absolute_path = join('', $webroot,$flag_icon); | |
129 | ||
130 | # Check if the a icon file exists. | |
131 | if (-e "$absolute_path") { | |
132 | # Return content of flag_icon. | |
133 | return $flag_icon; | |
dfbee171 SS |
134 | } else { |
135 | # If no icon for the specified country exists, try to use | |
136 | # the icon for "unknown". | |
137 | my $ccode = "unknown"; | |
138 | ||
b1ad5b8b MT |
139 | # Redoing all the stuff from above for the "unknown" icon. |
140 | my $file = join('.', $ccode, $ext); | |
141 | my $flag_icon = join('/', $flagdir, $file); | |
142 | my $absolute_path = join('', $webroot, $flag_icon); | |
dfbee171 SS |
143 | |
144 | # Check if the icon is present. | |
145 | if (-e "$absolute_path") { | |
146 | # Return "unknown" icon. | |
147 | return $flag_icon; | |
148 | } | |
1dcd8715 SS |
149 | } |
150 | } | |
151 | ||
152 | # Function to get the county name by a given country code. | |
153 | sub get_full_country_name($) { | |
154 | my ($input) = @_; | |
155 | my $name; | |
156 | ||
157 | # Remove whitespaces. | |
158 | chomp($input); | |
159 | ||
b868abd2 SS |
160 | # Convert input into upper case format. |
161 | my $code = uc($input); | |
8a64d10f | 162 | |
b868abd2 | 163 | # Handle country codes which are special or not part of the list. |
e34dbea7 SS |
164 | if ($not_iso_3166_location{$code}) { |
165 | # Grab location name from hash. | |
166 | $name = $not_iso_3166_location{$code}; | |
167 | } else { | |
b868abd2 SS |
168 | # Init libloc database connection. |
169 | my $db_handle = &init(); | |
170 | ||
171 | # Get the country name by using the location module. | |
172 | $name = &Location::get_country_name($db_handle, $code); | |
1dcd8715 SS |
173 | } |
174 | ||
175 | return $name; | |
176 | } | |
177 | ||
48152fae SS |
178 | # Function to get all available locations. |
179 | sub get_locations() { | |
427190d5 SS |
180 | my ($mode) = @_; |
181 | ||
182 | # Set default mode to add_special_locations. | |
183 | $mode = $mode ? $mode : "add_special_locations"; | |
184 | ||
79b564c8 | 185 | # Get locations which are stored in the location database. |
427190d5 | 186 | my @locations = &Location::database_countries($db_handle); |
51b6f07c | 187 | |
427190d5 SS |
188 | # Check if the special locations should be added. |
189 | if ($mode ne "no_special_locations") { | |
190 | # Merge special locations array and the database locations array. | |
191 | @locations = (@special_locations, @locations); | |
192 | } | |
30c59cbb | 193 | |
30c59cbb SS |
194 | # Sort locations array in alphabetical order. |
195 | my @sorted_locations = sort(@locations); | |
196 | ||
427190d5 | 197 | # Return the array. |
30c59cbb SS |
198 | return @sorted_locations; |
199 | } | |
200 | ||
5bf91fe1 SS |
201 | # Function to get the continent code of a given country code. |
202 | sub get_continent_code($) { | |
203 | my ($country_code) = @_; | |
204 | ||
205 | # Use location module to grab the continent code. | |
206 | my $continent_code = &Location::get_continent_code($db_handle, $country_code); | |
207 | ||
208 | return $continent_code; | |
209 | } | |
210 | ||
33975f57 SS |
211 | # Function to check if a given address has one ore more special flags. |
212 | sub address_has_flags($) { | |
d443f504 SS |
213 | my ($address) = @_; |
214 | ||
33975f57 SS |
215 | # Array to store the flags of the address. |
216 | my @flags; | |
217 | ||
d443f504 SS |
218 | # Loop through the hash of possible network flags. |
219 | foreach my $flag (keys(%network_flags)) { | |
220 | # Check if the address has the current flag. | |
221 | if (&Location::lookup_network_has_flag($db_handle, $address, $flag)) { | |
222 | # The given address has the requested flag. | |
223 | # | |
224 | # Grab the mapped location code for this flag. | |
225 | $mapped_code = $network_flags{$flag}; | |
226 | ||
33975f57 SS |
227 | # Add the mapped code to the array of flags. |
228 | push(@flags, $mapped_code); | |
d443f504 SS |
229 | } |
230 | } | |
33975f57 SS |
231 | |
232 | # Sort the array of flags. | |
233 | @flags = sort(@flags); | |
234 | ||
235 | # Return the array of flags. | |
236 | return @flags; | |
d443f504 SS |
237 | } |
238 | ||
f46fd078 SS |
239 | # Custom END declaration which will be executed when perl |
240 | # ends, to release the database handle to libloc. | |
241 | END { | |
242 | # Check if a database handle exists. | |
243 | if ($db_handle) { | |
244 | # Destroy libloc database handle. | |
245 | &Location::DESTROY($db_handle); | |
246 | } | |
247 | } | |
248 | ||
1dcd8715 | 249 | 1; |