Useful CLI tools (bash) for Arch Linux administration
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

76 lines
2.5 KiB

#!/bin/bash
# SSH timezone - Retrieve timezone information for SSH users
# Copyright (C) 2018 Pekka Helenius
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
##############################################################################
# This script is meant to be used as a part of provided 'bash.custom' file
##############################################################################
REQ_PKGS=(geoip2-database mmdblookup systemd openssh bind-tools)
for pkg in ${REQ_PKGS[*]}; do
if [[ ! $(pacman -Q | grep $pkg) ]]; then
echo -e "\nMissing package $pkg\n"
kill -INT $$
fi
done
GEOIP2_DATABASE="/usr/share/GeoIP/GeoLite2-City.mmdb"
if [[ ! -f $GEOIP2_DATABASE ]]; then
# GeoIP2 database file couldn't be found
kill -INT $$
fi
# If SSH_CONNECTION variable is defined
if [[ $(export -p | grep -o SSH_CONNECTION) ]]; then
# Get IP address of the user
RUSER_IP=$(echo $SSH_CONNECTION | awk '{print $1}')
# If SSH IP is local, try to get public IP.
if [[ $RUSER_IP =~ ^192\.168\. ]] || [[ $RUSER_IP =~ ^10\. ]] || [[ $USER_IP =~ ^172\.{16..31}\. ]]; then
RUSER_WAN_IP=$(dig +short +time=8 +tries=1 myip.opendns.com @resolver1.opendns.com 2>/dev/null)
# If OpenDNS hostname can't be resolved, use local IP as a fallback
if [[ $(echo $RUSER_WAN_IP | wc -w) -eq 0 ]]; then
RUSER_WAN_IP=$RUSER_IP
fi
else
# WAN IP is used IP
RUSER_WAN_IP=$RUSER_IP
fi
else
# Couldn't find IP address from SSH_CONNECTION variable
kill -INT $$
fi
TZ=$(mmdblookup -f $GEOIP2_DATABASE -i $RUSER_WAN_IP location 2>/dev/null | awk '/time_zone/ {getline; print $0}' | awk -F '"' '{print $2}')
# If TZ is empty, then use server default time zone setting as a default value
if [[ $(echo $TZ | wc -w) -eq 0 ]]; then
export TZ=$(timedatectl | sed '/Time zone:/!d' | awk '{print $3}')
else
export TZ
fi