src/scripts/xt_geoip_update

   1 #!/bin/bash
   2 ###############################################################################
   3 #                                                                             #
   4 # IPFire.org - A linux based firewall                                         #
   5 # Copyright (C) 2014 IPFire Development Team <info@ipfire.org>                #
   6 #                                                                             #
   7 # This program is free software: you can redistribute it and/or modify        #
   8 # it under the terms of the GNU General Public License as published by        #
   9 # the Free Software Foundation, either version 3 of the License, or           #
  10 # (at your option) any later version.                                         #
  11 #                                                                             #
  12 # This program is distributed in the hope that it will be useful,             #
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of              #
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
  15 # GNU General Public License for more details.                                #
  16 #                                                                             #
  17 # You should have received a copy of the GNU General Public License           #
  18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
  19 #                                                                             #
  20 ###############################################################################
  21
  22 TMP_PATH=$(mktemp -d)
  23 TMP_FILE=$(mktemp -p $TMP_PATH)
  24
  25 SCRIPT_PATH=/usr/local/bin
  26 DEST_PATH=/usr/share/xt_geoip
  27 DB_PATH=/var/lib/GeoIP
  28
  29 DL_URL=http://geolite.maxmind.com/download/geoip/database/
  30 DL_FILE=GeoLite2-Country-CSV.zip
  31
  32 eval $(/usr/local/bin/readhash /var/ipfire/proxy/settings)
  33
  34 function download() {
  35         echo "Downloading latest GeoIP ruleset..."
  36
  37         # Create temporary directory.
  38         mkdir -pv $TMP_PATH
  39
  40         # Proxy settings.
  41         # Check if a proxy should be used.
  42         if [[ $UPSTREAM_PROXY ]]; then
  43                 PROXYSETTINGS="-e http_proxy=http://"
  44
  45                 # Check if authentication against the proxy is configured.
  46                 if [[ $UPSTREAM_USER && $UPSTREAM_PASSWORD ]]; then
  47                         PROXYSETTINGS="$PROXYSETTINGS$UPSTREAM_USER:$UPSTREAM_PASSWORD@"
  48                 fi
  49
  50                 # Add proxy server.
  51                 PROXYSETTINGS="$PROXYSETTINGS$UPSTREAM_PROXY"
  52         fi
  53
  54         # Get the latest GeoIP database from server.
  55         wget $DL_URL/$DL_FILE $PROXYSETTINGS -O $TMP_FILE
  56
  57         # Extract files to database path.
  58         unzip $TMP_FILE -d $TMP_PATH
  59
  60         return 0
  61 }
  62
  63 function install() {
  64         echo "Install CSV database..."
  65
  66         # Check if the database dir exists.
  67         if [ ! -e "$DB_PATH" ]; then
  68                 mkdir -p $DB_PATH &>/dev/null
  69         fi
  70
  71         # Check if the directory for binary databases exists.
  72         if [ ! -e "$DEST_PATH" ]; then
  73                 mkdir -p $DEST_PATH &>/dev/null
  74         fi
  75
  76         # Install CSV databases.
  77         if ! cp -af $TMP_PATH/*/* $DB_PATH &>/dev/null; then
  78                 echo "Could not copy files. Aborting." >&2
  79                 return 1
  80         fi
  81
  82         return 0
  83 }
  84
  85 function build() {
  86         echo "Convert database..."
  87
  88         # Run script to convert the CSV file into several xtables
  89         # compatible binary files.
  90         if ! $SCRIPT_PATH/xt_geoip_build -S $DB_PATH -D $DEST_PATH; then
  91                 echo "Could not convert ruleset. Aborting." >&2
  92                 return 1
  93         fi
  94
  95         return 0
  96 }
  97
  98 function cleanup() {
  99         echo "Cleaning up temporary files..."
 100         if ! rm -rf $TMP_PATH &>/dev/null; then
 101                 echo "Could not remove files. Aborting." >&2
 102                 return 1
 103         fi
 104
 105         return 0
 106 }
 107
 108 function main() {
 109         # Download ruleset.
 110         download || exit $?
 111
 112         if ! install; then
 113                 # Do cleanup.
 114                 cleanup || exit $?
 115                 exit 1
 116         fi
 117
 118         # Remove temporary files.
 119         cleanup || exit $?
 120
 121         # Convert the ruleset.
 122         build || exit $?
 123
 124         return 0
 125 }
 126
 127 # Run the main function.
 128 main