]> git.ipfire.org Git - ipfire-2.x.git/blame - src/initscripts/helper/aws-setup
AWS: Use correct IP address for the internal DNS
[ipfire-2.x.git] / src / initscripts / helper / aws-setup
CommitLineData
bd3bcb45
MT
1#!/bin/bash
2
3. /etc/sysconfig/rc
4. ${rc_functions}
5
6get() {
7 local file="${1}"
8
9 wget -qO - "http://169.254.169.254/latest/meta-data/${file}"
10}
11
12to_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
23to_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
36prefix2netmask() {
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
50import_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
3273ff48 56 local hostname=$(get local-hostname)
bd3bcb45 57
3273ff48
MT
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
bd3bcb45
MT
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
3273ff48 77 local mac
bd3bcb45
MT
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
607240e2
MT
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
bd3bcb45
MT
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 # The gateway is always the first IP address in the subnet
107 local gateway="$(to_address $(( netaddress_num + 1 )))"
108
607240e2
MT
109 # The AWS internal DNS service is available on the second IP address of the VPC
110 local dns1="$(to_address $(( vpc_netaddress_num + 2 )))"
bd3bcb45
MT
111 local dns2=
112
113 case "${device_number}" in
114 # RED
115 0)
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 # DEBUG
152 cat /var/ipfire/ethernet/settings
153
154 # Import SSH keys
155 local line
156 for line in $(get "public-keys/"); do
157 local key_no="${line%=*}"
158
159 local key="$(get public-keys/${key_no}/openssh-key)"
160 if [ -n "${key}" ] && ! grep -q "^${key}$" /root/.ssh/authorized_keys 2>/dev/null; then
161 mkdir -p /root/.ssh
162 chmod 700 /root/.ssh
163
164 echo "${key}" >> /root/.ssh/authorized_keys
165 chmod 600 /root/.ssh/authorized_keys
166 fi
167 done
168
bd3bcb45
MT
169 # Actions performed only on the very first start
170 if [ ! -e "/var/ipfire/main/firstsetup_ok" ]; then
7fa83c2f 171 # Enable SSH
8b59ef08
MT
172 sed -e "s/ENABLE_SSH=.*/ENABLE_SSH=on/g" -i /var/ipfire/remote/settings
173
7fa83c2f
MT
174 touch /var/ipfire/remote/enablessh
175 chown nobody:nobody /var/ipfire/remote/enablessh
176
177 # Enable SSH key authentication
178 sed -e "s/^ENABLE_SSH_KEYS=.*/ENABLE_SSH_KEYS=on/" -i /var/ipfire/remote/settings
179
bd3bcb45
MT
180 # Firewall rules for SSH and WEBIF
181 (
182 echo "1,ACCEPT,INPUTFW,ON,std_net_src,ALL,ipfire,RED1,,TCP,,,ON,,,cust_srv,SSH,,,,,,,,,,,00:00,00:00,,AUTO,,dnat,,,,,second"
183 echo "2,ACCEPT,INPUTFW,ON,std_net_src,ALL,ipfire,RED1,,TCP,,,ON,,,TGT_PORT,444,,,,,,,,,,,00:00,00:00,,AUTO,,dnat,,,,,second"
184 ) >> /var/ipfire/firewall/input
185
186 # This script has now completed the first steps of setup
187 touch /var/ipfire/main/firstsetup_ok
188 fi
189
190 # All done
191 echo_ok
192}
193
194case "${reason}" in
195 PREINIT)
196 # Bring up the interface
197 ip link set "${interface}" up
198 ;;
199
200 BOUND|RENEW|REBIND|REBOOT)
201 # Remove any previous IP addresses
202 ip addr flush dev "${interface}"
203
204 # Add (or re-add) the new IP address
205 ip addr add "${new_ip_address}/${new_subnet_mask}" dev "${interface}"
206
207 # Add the default route
208 ip route add default via "${new_routers}"
209
210 # Import AWS configuration
211 import_aws_configuration
212 ;;
213
214 EXPIRE|FAIL|RELEASE|STOP)
215 # Remove all IP addresses
216 ip addr flush dev "${interface}"
217 ;;
218
219 *)
220 echo "Unhandled reason: ${reason}" >&2
221 exit 2
222 ;;
223esac
224
225# Terminate
226exit 0