src/initscripts/helper/aws-setup

   1 #!/bin/bash
   2
   3 . /etc/sysconfig/rc
   4 . ${rc_functions}
   5
   6 get() {
   7         local file="${1}"
   8
   9         wget -qO - "http://169.254.169.254/latest/meta-data/${file}"
  10 }
  11
  12 to_address() {
  13         local n="${1}"
  14
  15         local o1=$(( (n & 0xff000000) >> 24 ))
  16         local o2=$(( (n & 0xff0000) >> 16 ))
  17         local o3=$(( (n & 0xff00) >> 8 ))
  18         local o4=$(( (n & 0xff) ))
  19
  20         printf "%d.%d.%d.%d\n" "${o1}" "${o2}" "${o3}" "${o4}"
  21 }
  22
  23 to_integer() {
  24         local address="${1}"
  25
  26         local integer=0
  27
  28         local i
  29         for i in ${address//\./ }; do
  30                 integer=$(( (integer << 8) + i ))
  31         done
  32
  33         printf "%d\n" "${integer}"
  34 }
  35
  36 prefix2netmask() {
  37         local prefix=${1}
  38
  39         local zeros=$(( 32 - prefix ))
  40         local netmask=0
  41
  42         local i
  43         for (( i=0; i<${zeros}; i++ )); do
  44                 netmask=$(( (netmask << 1) ^ 1 ))
  45         done
  46
  47         to_address "$(( netmask ^ 0xffffffff ))"
  48 }
  49
  50 import_aws_configuration() {
  51         local instance_id="$(get instance-id)"
  52
  53         boot_mesg "Importing AWS configuration for instance ${instance_id}..."
  54
  55         # Initialise system settings
  56         local hostname=$(get local-hostname)
  57
  58         # Set hostname
  59         if ! grep -q "^HOSTNAME=" /var/ipfire/main/settings; then
  60                 echo "HOSTNAME=${hostname%%.*}" >> /var/ipfire/main/settings
  61         fi
  62
  63         # Set domainname
  64         if ! grep -q "^DOMAINNAME=" /var/ipfire/main/settings; then
  65                 echo "DOMAINNAME=${hostname#*.}" >> /var/ipfire/main/settings
  66         fi
  67
  68         # Import any DNS server settings
  69         eval $(/usr/local/bin/readhash <(grep -E "^DNS([0-9])=" /var/ipfire/ethernet/settings 2>/dev/null))
  70
  71         # Import network configuration
  72         (
  73                 # RED + GREEN
  74                 echo "CONFIG_TYPE=1"
  75         ) > /var/ipfire/ethernet/settings
  76
  77         local mac
  78         for mac in $(get network/interfaces/macs/); do
  79                 # Remove trailing slash
  80                 mac="${mac//\//}"
  81
  82                 local device_number="$(get "network/interfaces/macs/${mac}/device-number")"
  83                 local interface_id="$(get "network/interfaces/macs/${mac}/interface-id")"
  84
  85                 # First IPv4 address
  86                 local ipv4_address="$(get "network/interfaces/macs/${mac}/local-ipv4s" | head -n1)"
  87                 local ipv4_address_num="$(to_integer "${ipv4_address}")"
  88
  89                 # Get VPC subnet
  90                 local vpc="$(get "network/interfaces/macs/${mac}/vpc-ipv4-cidr-block")"
  91                 local vpc_netaddress="${vpc%/*}"
  92                 local vpc_netaddress_num="$(to_integer "${vpc_netaddress}")"
  93
  94                 # Get subnet size
  95                 local subnet="$(get "network/interfaces/macs/${mac}/subnet-ipv4-cidr-block")"
  96
  97                 local prefix="${subnet#*/}"
  98                 local netmask="$(prefix2netmask "${prefix}")"
  99                 local netmask_num="$(to_integer "${netmask}")"
 100
 101                 # Calculate the network and broadcast addresses
 102                 local netaddress="${subnet%/*}"
 103                 local netaddress_num="$(to_integer "${netaddress}")"
 104                 local broadcast="$(to_address $(( ipv4_address_num | (0xffffffff ^ netmask_num) )))"
 105
 106                 case "${device_number}" in
 107                         # RED
 108                         0)
 109                                 # The gateway is always the first IP address in the subnet
 110                                 local gateway="$(to_address $(( netaddress_num + 1 )))"
 111
 112                                 # The AWS internal DNS service is available on the second IP address of the VPC
 113                                 local dns1="$(to_address $(( vpc_netaddress_num + 2 )))"
 114                                 local dns2=
 115
 116                                 (
 117                                         echo "RED_TYPE=STATIC"
 118                                         echo "RED_DEV=red0"
 119                                         echo "RED_MACADDR=${mac}"
 120                                         echo "RED_DESCRIPTION='${interface_id}'"
 121                                         echo "RED_ADDRESS=${ipv4_address}"
 122                                         echo "RED_NETMASK=${netmask}"
 123                                         echo "RED_NETADDRESS=${netaddress}"
 124                                         echo "RED_BROADCAST=${broadcast}"
 125                                         echo "DEFAULT_GATEWAY=${gateway}"
 126                                         echo "DNS1=${DNS1:-${dns1}}"
 127                                         echo "DNS2=${DNS2:-${dns2}}"
 128                                 ) >> /var/ipfire/ethernet/settings
 129
 130                                 # Import aliases for RED
 131                                 for alias in $(get "network/interfaces/macs/${mac}/local-ipv4s" | tail -n +2); do
 132                                         echo "${alias},on,"
 133                                 done > /var/ipfire/ethernet/aliases
 134                                 ;;
 135
 136                         # GREEN
 137                         1)
 138                                 (
 139                                         echo "GREEN_DEV=green0"
 140                                         echo "GREEN_MACADDR=${mac}"
 141                                         echo "GREEN_DESCRIPTION='${interface_id}'"
 142                                         echo "GREEN_ADDRESS=${ipv4_address}"
 143                                         echo "GREEN_NETMASK=${netmask}"
 144                                         echo "GREEN_NETADDRESS=${netaddress}"
 145                                         echo "GREEN_BROADCAST=${broadcast}"
 146                                 ) >> /var/ipfire/ethernet/settings
 147                                 ;;
 148                 esac
 149         done
 150
 151         # Import SSH keys
 152         local line
 153         for line in $(get "public-keys/"); do
 154                 local key_no="${line%=*}"
 155
 156                 local key="$(get public-keys/${key_no}/openssh-key)"
 157                 if [ -n "${key}" ] && ! grep -q "^${key}$" /root/.ssh/authorized_keys 2>/dev/null; then
 158                         mkdir -p /root/.ssh
 159                         chmod 700 /root/.ssh
 160
 161                         echo "${key}" >> /root/.ssh/authorized_keys
 162                         chmod 600 /root/.ssh/authorized_keys
 163                 fi
 164         done
 165
 166         # Actions performed only on the very first start
 167         if [ ! -e "/var/ipfire/main/firstsetup_ok" ]; then
 168                 # Enable SSH
 169                 sed -e "s/ENABLE_SSH=.*/ENABLE_SSH=on/g" -i /var/ipfire/remote/settings
 170
 171                 touch /var/ipfire/remote/enablessh
 172                 chown nobody:nobody /var/ipfire/remote/enablessh
 173
 174                 # Enable SSH key authentication
 175                 sed -e "s/^ENABLE_SSH_KEYS=.*/ENABLE_SSH_KEYS=on/" -i /var/ipfire/remote/settings
 176
 177                 # Firewall rules for SSH and WEBIF
 178                 (
 179                         echo "1,ACCEPT,INPUTFW,ON,std_net_src,ALL,ipfire,RED1,,TCP,,,ON,,,cust_srv,SSH,,,,,,,,,,,00:00,00:00,,AUTO,,dnat,,,,,second"
 180                         echo "2,ACCEPT,INPUTFW,ON,std_net_src,ALL,ipfire,RED1,,TCP,,,ON,,,TGT_PORT,444,,,,,,,,,,,00:00,00:00,,AUTO,,dnat,,,,,second"
 181                 ) >> /var/ipfire/firewall/input
 182
 183                 # This script has now completed the first steps of setup
 184                 touch /var/ipfire/main/firstsetup_ok
 185         fi
 186
 187         # All done
 188         echo_ok
 189 }
 190
 191 case "${reason}" in
 192         PREINIT)
 193                 # Bring up the interface
 194                 ip link set "${interface}" up
 195                 ;;
 196
 197         BOUND|RENEW|REBIND|REBOOT)
 198                 # Remove any previous IP addresses
 199                 ip addr flush dev "${interface}"
 200
 201                 # Add (or re-add) the new IP address
 202                 ip addr add "${new_ip_address}/${new_subnet_mask}" dev "${interface}"
 203
 204                 # Add the default route
 205                 ip route add default via "${new_routers}"
 206
 207                 # Import AWS configuration
 208                 import_aws_configuration
 209                 ;;
 210
 211         EXPIRE|FAIL|RELEASE|STOP)
 212                 # Remove all IP addresses
 213                 ip addr flush dev "${interface}"
 214                 ;;
 215
 216         *)
 217                 echo "Unhandled reason: ${reason}" >&2
 218                 exit 2
 219                 ;;
 220 esac
 221
 222 # Terminate
 223 exit 0