[people/pmueller/ipfire-2.x.git] / src / initscripts / helper / aws-setup

#!/bin/bash

. /etc/sysconfig/rc
. ${rc_functions}

get() {
	local file="${1}"

	wget -qO - "http://169.254.169.254/latest/${file}"
}

to_address() {
	local n="${1}"

	local o1=$(( (n & 0xff000000) >> 24 ))
	local o2=$(( (n & 0xff0000) >> 16 ))
	local o3=$(( (n & 0xff00) >> 8 ))
	local o4=$(( (n & 0xff) ))

	printf "%d.%d.%d.%d\n" "${o1}" "${o2}" "${o3}" "${o4}"
}

to_integer() {
	local address="${1}"

	local integer=0

	local i
	for i in ${address//\./ }; do
		integer=$(( (integer << 8) + i ))
	done

	printf "%d\n" "${integer}"
}

prefix2netmask() {
	local prefix=${1}

	local zeros=$(( 32 - prefix ))
	local netmask=0

	local i
	for (( i=0; i<${zeros}; i++ )); do
		netmask=$(( (netmask << 1) ^ 1 ))
	done

	to_address "$(( netmask ^ 0xffffffff ))"
}

find_interface() {
	local mac="${1}"

	local path
	for path in /sys/class/net/*; do
		local address="$(<${path}/address)"

		if [ "${mac}" = "${address}" ]; then
			basename "${path}"
			return 0
		fi
	done

	return 1
}

import_aws_configuration() {
	local instance_id="$(get meta-data/instance-id)"

	boot_mesg "Importing AWS configuration for instance ${instance_id}..."

	# Store instance ID
	echo "${instance_id}" > /var/run/aws-instance-id

	# Initialise system settings
	local hostname=$(get meta-data/local-hostname)

	# Set hostname
	if ! grep -q "^HOSTNAME=" /var/ipfire/main/settings; then
		echo "HOSTNAME=${hostname%%.*}" >> /var/ipfire/main/settings
	fi

	# Set domainname
	if ! grep -q "^DOMAINNAME=" /var/ipfire/main/settings; then
		echo "DOMAINNAME=${hostname#*.}" >> /var/ipfire/main/settings
	fi

	# Create setup user
	if ! getent passwd setup &>/dev/null; then
		useradd setup -s /usr/bin/run-setup -g nobody -m

		# Unlock the account
		usermod -p "x" setup
	fi

	# Import SSH keys for setup user
	local line
	for line in $(get "meta-data/public-keys/"); do
		local key_no="${line%=*}"

		local key="$(get meta-data/public-keys/${key_no}/openssh-key)"
		if [ -n "${key}" ] && ! grep -q "^${key}$" "/home/setup/.ssh/authorized_keys" 2>/dev/null; then
			mkdir -p "/home/setup/.ssh"
			chmod 700 "/home/setup/.ssh"
			chown setup.nobody "/home/setup/.ssh"

			echo "${key}" >> "/home/setup/.ssh/authorized_keys"
			chmod 600 "/home/setup/.ssh/authorized_keys"
			chown setup.nobody "/home/setup/.ssh/authorized_keys"
		fi
	done

	# Download user-data
	local user_data="$(get user-data)"

	# Import any DNS server settings
	eval $(/usr/local/bin/readhash <(grep -E "^DNS([0-9])=" /var/ipfire/ethernet/settings 2>/dev/null))

	# Import network configuration
	# After this, no network connectivity will be available from this script due to the
	# renaming of the network interfaces for which they have to be shut down
	local config_type=1
	: > /var/ipfire/ethernet/settings

	local mac
	for mac in $(get meta-data/network/interfaces/macs/); do
		# Remove trailing slash
		mac="${mac//\//}"

		local device_number="$(get "meta-data/network/interfaces/macs/${mac}/device-number")"
		local interface_id="$(get "meta-data/network/interfaces/macs/${mac}/interface-id")"

		# First IPv4 address
		local ipv4_address="$(get "meta-data/network/interfaces/macs/${mac}/local-ipv4s" | head -n1)"
		local ipv4_address_num="$(to_integer "${ipv4_address}")"

		# Get VPC subnet
		local vpc="$(get "meta-data/network/interfaces/macs/${mac}/vpc-ipv4-cidr-block")"
		local vpc_netaddress="${vpc%/*}"
		local vpc_netaddress_num="$(to_integer "${vpc_netaddress}")"

		# Get subnet size
		local subnet="$(get "meta-data/network/interfaces/macs/${mac}/subnet-ipv4-cidr-block")"

		local prefix="${subnet#*/}"
		local netmask="$(prefix2netmask "${prefix}")"
		local netmask_num="$(to_integer "${netmask}")"

		# Calculate the network and broadcast addresses
		local netaddress="${subnet%/*}"
		local netaddress_num="$(to_integer "${netaddress}")"
		local broadcast="$(to_address $(( ipv4_address_num | (0xffffffff ^ netmask_num) )))"

		case "${device_number}" in
			# RED
			0)
				local interface_name="red0"

				# The gateway is always the first IP address in the subnet
				local gateway="$(to_address $(( netaddress_num + 1 )))"

				# The AWS internal DNS service is available on the second IP address of the VPC
				local dns1="$(to_address $(( vpc_netaddress_num + 2 )))"
				local dns2=

				(
					echo "RED_TYPE=STATIC"
					echo "RED_DEV=${interface_name}"
					echo "RED_MACADDR=${mac}"
					echo "RED_DESCRIPTION='${interface_id}'"
					echo "RED_ADDRESS=${ipv4_address}"
					echo "RED_NETMASK=${netmask}"
					echo "RED_NETADDRESS=${netaddress}"
					echo "RED_BROADCAST=${broadcast}"
					echo "DEFAULT_GATEWAY=${gateway}"
					echo "DNS1=${DNS1:-${dns1}}"
					echo "DNS2=${DNS2:-${dns2}}"
				) >> /var/ipfire/ethernet/settings

				# Import aliases for RED
				for alias in $(get "meta-data/network/interfaces/macs/${mac}/local-ipv4s" | tail -n +2); do
					echo "${alias},on,"
				done > /var/ipfire/ethernet/aliases
				;;

			# GREEN
			1)
				local interface_name="green0"

				(
					echo "GREEN_DEV=${interface_name}"
					echo "GREEN_MACADDR=${mac}"
					echo "GREEN_DESCRIPTION='${interface_id}'"
					echo "GREEN_ADDRESS=${ipv4_address}"
					echo "GREEN_NETMASK=${netmask}"
					echo "GREEN_NETADDRESS=${netaddress}"
					echo "GREEN_BROADCAST=${broadcast}"
				) >> /var/ipfire/ethernet/settings
				;;

			# ORANGE
			2)
				local interface_name="orange0"
				config_type=2

				(
					echo "ORANGE_DEV=${interface_name}"
					echo "ORANGE_MACADDR=${mac}"
					echo "ORANGE_DESCRIPTION='${interface_id}'"
					echo "ORANGE_ADDRESS=${ipv4_address}"
					echo "ORANGE_NETMASK=${netmask}"
					echo "ORANGE_NETADDRESS=${netaddress}"
					echo "ORANGE_BROADCAST=${broadcast}"
				) >> /var/ipfire/ethernet/settings
				;;
		esac

		# Rename interface
		local interface="$(find_interface "${mac}")"

		if [ -n "${interface}" ] && [ -n "${interface_name}" ] && [ "${interface}" != "${interface_name}" ]; then
			ip link set "${interface}" down
			ip link set "${interface}" name "${interface_name}"
		fi
	done

	# Save CONFIG_TYPE
	echo "CONFIG_TYPE=${config_type}" >> /var/ipfire/ethernet/settings

	# Actions performed only on the very first start
	if [ ! -e "/var/ipfire/main/firstsetup_ok" ]; then
		# Enable SSH
		sed -e "s/ENABLE_SSH=.*/ENABLE_SSH=on/g" -i /var/ipfire/remote/settings

		# Disable SSH password authentication
		sed -e "s/^ENABLE_SSH_PASSWORDS=.*/ENABLE_SSH_PASSWORDS=off/" -i /var/ipfire/remote/settings

		# Enable SSH key authentication
		sed -e "s/^ENABLE_SSH_KEYS=.*/ENABLE_SSH_KEYS=on/" -i /var/ipfire/remote/settings

		# Apply SSH settings
		/usr/local/bin/sshctrl

		# Mark SSH to start immediately (but not right now)
		touch /var/ipfire/remote/enablessh
		chown nobody:nobody /var/ipfire/remote/enablessh

		# Firewall rules for SSH and WEBIF
		(
			echo "1,ACCEPT,INPUTFW,ON,std_net_src,ALL,ipfire,RED1,,TCP,,,ON,,,cust_srv,SSH,,,,,,,,,,,00:00,00:00,,AUTO,,dnat,,,,,second"
			echo "2,ACCEPT,INPUTFW,ON,std_net_src,ALL,ipfire,RED1,,TCP,,,ON,,,TGT_PORT,444,,,,,,,,,,,00:00,00:00,,AUTO,,dnat,,,,,second"
		) >> /var/ipfire/firewall/input

		# This script has now completed the first steps of setup
		touch /var/ipfire/main/firstsetup_ok

		# Save user-data script to be executed later
		if [ "${user_data:0:2}" = "#!" ]; then
			echo "${user_data}" > /tmp/aws-user-data.script
			chmod 700 /tmp/aws-user-data.script
		fi
	fi

	# All done
	echo_ok
}

case "${reason}" in
	PREINIT)
		# Bring up the interface
		ip link set "${interface}" up
		;;

	BOUND|RENEW|REBIND|REBOOT)
		# Remove any previous IP addresses
		ip addr flush dev "${interface}"

		# Add (or re-add) the new IP address
		ip addr add "${new_ip_address}/${new_subnet_mask}" dev "${interface}"

		# Add the default route
		ip route add default via "${new_routers}"

		# Import AWS configuration
		import_aws_configuration
		;;

	EXPIRE|FAIL|RELEASE|STOP)
		# Remove all IP addresses
		ip addr flush dev "${interface}"
		;;

	*)
		echo "Unhandled reason: ${reason}" >&2
		exit 2
		;;
esac

# Terminate
exit 0
Commit	Line	Data
bd3bcb45 MT	1	#!/bin/bash
	2
	3	. /etc/sysconfig/rc
	4	. ${rc_functions}
	5
	6	get() {
	7	local file="${1}"
	8
4e4c122c	9	wget -qO - "http://169.254.169.254/latest/${file}"
bd3bcb45 MT	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
c7141f04 MT	50	find_interface() {
	51	local mac="${1}"
	52
	53	local path
	54	for path in /sys/class/net/*; do
	55	local address="$(<${path}/address)"
	56
	57	if [ "${mac}" = "${address}" ]; then
	58	basename "${path}"
	59	return 0
	60	fi
	61	done
	62
	63	return 1
	64	}
	65
bd3bcb45	66	import_aws_configuration() {
4e4c122c	67	local instance_id="$(get meta-data/instance-id)"
bd3bcb45 MT	68
	69	boot_mesg "Importing AWS configuration for instance ${instance_id}..."
	70
eb7ccf87 MT	71	# Store instance ID
	72	echo "${instance_id}" > /var/run/aws-instance-id
	73
bd3bcb45	74	# Initialise system settings
4e4c122c	75	local hostname=$(get meta-data/local-hostname)
bd3bcb45	76
3273ff48 MT	77	# Set hostname
	78	if ! grep -q "^HOSTNAME=" /var/ipfire/main/settings; then
	79	echo "HOSTNAME=${hostname%%.*}" >> /var/ipfire/main/settings
	80	fi
	81
	82	# Set domainname
	83	if ! grep -q "^DOMAINNAME=" /var/ipfire/main/settings; then
	84	echo "DOMAINNAME=${hostname#*.}" >> /var/ipfire/main/settings
bd3bcb45 MT	85	fi
bd3bcb45 MT	86
1413006b MT	87	# Create setup user
1413006b MT	88	if ! getent passwd setup &>/dev/null; then
e2d9fe7e	89	useradd setup -s /usr/bin/run-setup -g nobody -m
87d7fc66 MT	90
	91	# Unlock the account
	92	usermod -p "x" setup
1413006b	93	fi
2e42a9ea	94
53eb4918 MT	95	# Import SSH keys for setup user
53eb4918 MT	96	local line
4e4c122c	97	for line in $(get "meta-data/public-keys/"); do
53eb4918 MT	98	local key_no="${line%=*}"
53eb4918 MT	99
4e4c122c	100	local key="$(get meta-data/public-keys/${key_no}/openssh-key)"
53eb4918 MT	101	if [ -n "${key}" ] && ! grep -q "^${key}$" "/home/setup/.ssh/authorized_keys" 2>/dev/null; then
	102	mkdir -p "/home/setup/.ssh"
	103	chmod 700 "/home/setup/.ssh"
6a7e6b44	104	chown setup.nobody "/home/setup/.ssh"
53eb4918 MT	105
	106	echo "${key}" >> "/home/setup/.ssh/authorized_keys"
	107	chmod 600 "/home/setup/.ssh/authorized_keys"
6a7e6b44	108	chown setup.nobody "/home/setup/.ssh/authorized_keys"
53eb4918	109	fi
2e42a9ea MT	110	done
2e42a9ea MT	111
4e4c122c MT	112	# Download user-data
	113	local user_data="$(get user-data)"
	114
bd3bcb45 MT	115	# Import any DNS server settings
	116	eval $(/usr/local/bin/readhash <(grep -E "^DNS([0-9])=" /var/ipfire/ethernet/settings 2>/dev/null))
	117
	118	# Import network configuration
2e42a9ea MT	119	# After this, no network connectivity will be available from this script due to the
2e42a9ea MT	120	# renaming of the network interfaces for which they have to be shut down
0f224ad7 MT	121	local config_type=1
0f224ad7 MT	122	: > /var/ipfire/ethernet/settings
bd3bcb45	123
3273ff48	124	local mac
4e4c122c	125	for mac in $(get meta-data/network/interfaces/macs/); do
bd3bcb45 MT	126	# Remove trailing slash
	127	mac="${mac//\//}"
	128
4e4c122c MT	129	local device_number="$(get "meta-data/network/interfaces/macs/${mac}/device-number")"
4e4c122c MT	130	local interface_id="$(get "meta-data/network/interfaces/macs/${mac}/interface-id")"
bd3bcb45 MT	131
bd3bcb45 MT	132	# First IPv4 address
4e4c122c	133	local ipv4_address="$(get "meta-data/network/interfaces/macs/${mac}/local-ipv4s" \| head -n1)"
bd3bcb45 MT	134	local ipv4_address_num="$(to_integer "${ipv4_address}")"
bd3bcb45 MT	135
607240e2	136	# Get VPC subnet
4e4c122c	137	local vpc="$(get "meta-data/network/interfaces/macs/${mac}/vpc-ipv4-cidr-block")"
607240e2 MT	138	local vpc_netaddress="${vpc%/*}"
	139	local vpc_netaddress_num="$(to_integer "${vpc_netaddress}")"
	140
bd3bcb45	141	# Get subnet size
4e4c122c	142	local subnet="$(get "meta-data/network/interfaces/macs/${mac}/subnet-ipv4-cidr-block")"
bd3bcb45 MT	143
	144	local prefix="${subnet#*/}"
	145	local netmask="$(prefix2netmask "${prefix}")"
	146	local netmask_num="$(to_integer "${netmask}")"
	147
	148	# Calculate the network and broadcast addresses
	149	local netaddress="${subnet%/*}"
	150	local netaddress_num="$(to_integer "${netaddress}")"
	151	local broadcast="$(to_address $(( ipv4_address_num \| (0xffffffff ^ netmask_num) )))"
	152
bd3bcb45 MT	153	case "${device_number}" in
	154	# RED
	155	0)
c7141f04 MT	156	local interface_name="red0"
c7141f04 MT	157
c86fd963 MT	158	# The gateway is always the first IP address in the subnet
	159	local gateway="$(to_address $(( netaddress_num + 1 )))"
	160
	161	# The AWS internal DNS service is available on the second IP address of the VPC
	162	local dns1="$(to_address $(( vpc_netaddress_num + 2 )))"
	163	local dns2=
	164
bd3bcb45 MT	165	(
bd3bcb45 MT	166	echo "RED_TYPE=STATIC"
c7141f04	167	echo "RED_DEV=${interface_name}"
bd3bcb45 MT	168	echo "RED_MACADDR=${mac}"
	169	echo "RED_DESCRIPTION='${interface_id}'"
	170	echo "RED_ADDRESS=${ipv4_address}"
	171	echo "RED_NETMASK=${netmask}"
	172	echo "RED_NETADDRESS=${netaddress}"
	173	echo "RED_BROADCAST=${broadcast}"
	174	echo "DEFAULT_GATEWAY=${gateway}"
	175	echo "DNS1=${DNS1:-${dns1}}"
	176	echo "DNS2=${DNS2:-${dns2}}"
	177	) >> /var/ipfire/ethernet/settings
	178
	179	# Import aliases for RED
4e4c122c	180	for alias in $(get "meta-data/network/interfaces/macs/${mac}/local-ipv4s" \| tail -n +2); do
bd3bcb45 MT	181	echo "${alias},on,"
	182	done > /var/ipfire/ethernet/aliases
	183	;;
	184
	185	# GREEN
	186	1)
c7141f04 MT	187	local interface_name="green0"
c7141f04 MT	188
bd3bcb45	189	(
c7141f04	190	echo "GREEN_DEV=${interface_name}"
bd3bcb45 MT	191	echo "GREEN_MACADDR=${mac}"
	192	echo "GREEN_DESCRIPTION='${interface_id}'"
	193	echo "GREEN_ADDRESS=${ipv4_address}"
	194	echo "GREEN_NETMASK=${netmask}"
	195	echo "GREEN_NETADDRESS=${netaddress}"
	196	echo "GREEN_BROADCAST=${broadcast}"
	197	) >> /var/ipfire/ethernet/settings
	198	;;
0f224ad7 MT	199
	200	# ORANGE
	201	2)
c7141f04	202	local interface_name="orange0"
0f224ad7 MT	203	config_type=2
	204
	205	(
c7141f04	206	echo "ORANGE_DEV=${interface_name}"
0f224ad7 MT	207	echo "ORANGE_MACADDR=${mac}"
	208	echo "ORANGE_DESCRIPTION='${interface_id}'"
	209	echo "ORANGE_ADDRESS=${ipv4_address}"
	210	echo "ORANGE_NETMASK=${netmask}"
	211	echo "ORANGE_NETADDRESS=${netaddress}"
	212	echo "ORANGE_BROADCAST=${broadcast}"
	213	) >> /var/ipfire/ethernet/settings
	214	;;
bd3bcb45	215	esac
c7141f04 MT	216
	217	# Rename interface
	218	local interface="$(find_interface "${mac}")"
	219
470e85c3	220	if [ -n "${interface}" ] && [ -n "${interface_name}" ] && [ "${interface}" != "${interface_name}" ]; then
c7141f04 MT	221	ip link set "${interface}" down
	222	ip link set "${interface}" name "${interface_name}"
	223	fi
bd3bcb45 MT	224	done
bd3bcb45 MT	225
0f224ad7 MT	226	# Save CONFIG_TYPE
	227	echo "CONFIG_TYPE=${config_type}" >> /var/ipfire/ethernet/settings
	228
bd3bcb45 MT	229	# Actions performed only on the very first start
bd3bcb45 MT	230	if [ ! -e "/var/ipfire/main/firstsetup_ok" ]; then
7fa83c2f	231	# Enable SSH
8b59ef08 MT	232	sed -e "s/ENABLE_SSH=.*/ENABLE_SSH=on/g" -i /var/ipfire/remote/settings
8b59ef08 MT	233
0cf70cae MT	234	# Disable SSH password authentication
	235	sed -e "s/^ENABLE_SSH_PASSWORDS=.*/ENABLE_SSH_PASSWORDS=off/" -i /var/ipfire/remote/settings
	236
7fa83c2f MT	237	# Enable SSH key authentication
	238	sed -e "s/^ENABLE_SSH_KEYS=.*/ENABLE_SSH_KEYS=on/" -i /var/ipfire/remote/settings
	239
8f2c3b49 MT	240	# Apply SSH settings
	241	/usr/local/bin/sshctrl
	242
693208bf MT	243	# Mark SSH to start immediately (but not right now)
	244	touch /var/ipfire/remote/enablessh
	245	chown nobody:nobody /var/ipfire/remote/enablessh
	246
bd3bcb45 MT	247	# Firewall rules for SSH and WEBIF
	248	(
	249	echo "1,ACCEPT,INPUTFW,ON,std_net_src,ALL,ipfire,RED1,,TCP,,,ON,,,cust_srv,SSH,,,,,,,,,,,00:00,00:00,,AUTO,,dnat,,,,,second"
	250	echo "2,ACCEPT,INPUTFW,ON,std_net_src,ALL,ipfire,RED1,,TCP,,,ON,,,TGT_PORT,444,,,,,,,,,,,00:00,00:00,,AUTO,,dnat,,,,,second"
	251	) >> /var/ipfire/firewall/input
	252
	253	# This script has now completed the first steps of setup
	254	touch /var/ipfire/main/firstsetup_ok
4e4c122c MT	255
	256	# Save user-data script to be executed later
	257	if [ "${user_data:0:2}" = "#!" ]; then
	258	echo "${user_data}" > /tmp/aws-user-data.script
	259	chmod 700 /tmp/aws-user-data.script
	260	fi
bd3bcb45 MT	261	fi
	262
	263	# All done
	264	echo_ok
	265	}
	266
	267	case "${reason}" in
	268	PREINIT)
	269	# Bring up the interface
	270	ip link set "${interface}" up
	271	;;
	272
	273	BOUND\|RENEW\|REBIND\|REBOOT)
	274	# Remove any previous IP addresses
	275	ip addr flush dev "${interface}"
	276
	277	# Add (or re-add) the new IP address
	278	ip addr add "${new_ip_address}/${new_subnet_mask}" dev "${interface}"
	279
	280	# Add the default route
	281	ip route add default via "${new_routers}"
	282
	283	# Import AWS configuration
	284	import_aws_configuration
	285	;;
	286
	287	EXPIRE\|FAIL\|RELEASE\|STOP)
	288	# Remove all IP addresses
	289	ip addr flush dev "${interface}"
	290	;;
	291
	292	*)
	293	echo "Unhandled reason: ${reason}" >&2
	294	exit 2
	295	;;
	296	esac
	297
	298	# Terminate
	299	exit 0