machine_setup/ansible/roles/jail/files/jail_netgraph_bridge.bash

136 lines
3.7 KiB
Bash
Raw Normal View History

#!/usr/local/bin/bash
#
set -euo pipefail
IFS=$'\n\t'
DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
function main {
if [ "$1" = "start" ]; then
shift 1
start_jail "${@}"
elif [ "$1" = "stop" ]; then
shift 1
stop_jail "${@}"
else
>&2 echo "Unrecognized command"
exit 1
fi
}
function start_jail {
host_interface_name="$1"
bridge_name="bridge_${host_interface_name}"
jail_interface_name=$(sanitize_interface_name "$2")
ip_range="$3"
2024-07-12 23:58:50 +00:00
local mac_address
mac_address=$(calculate_mac_address "$jail_interface_name")
assert_bridge "$host_interface_name" "$bridge_name" "$ip_range"
bridge_link_name=$(detect_available_link "${bridge_name}")
ngctl -d -f - <<EOF
mkpeer ${bridge_name}: eiface $bridge_link_name ether
2024-07-12 23:58:50 +00:00
msg ${bridge_name}:$bridge_link_name set $mac_address
name ${bridge_name}:$bridge_link_name $jail_interface_name
EOF
ifconfig $(ngctl msg "${jail_interface_name}:" getifname | grep Args | cut -d '"' -f 2) name "${jail_interface_name}" up
}
function stop_jail {
host_interface_name="$1"
bridge_name="bridge_${host_interface_name}"
jail_interface_name=$(sanitize_interface_name "$2")
if ng_exists "${jail_interface_name}:"; then
wait_for_interface_to_exist "${jail_interface_name}" 120
ngctl shutdown "${jail_interface_name}:"
fi
if ng_exists "${bridge_name}:"; then
num_remaining_hooks=$(ng_bridge_get_num_hooks "${bridge_name}:")
if [ $num_remaining_hooks -eq 1 ]; then
ngctl shutdown "${bridge_name}:"
ngctl shutdown "${host_interface_name}:"
fi
fi
}
function assert_bridge {
host_interface_name="$1"
bridge_name="$2"
ip_range="$3"
if ! ng_exists "${bridge_name}:"; then
ngctl -d -f - <<EOF
mkpeer . eiface hook ether
name .:hook $host_interface_name
EOF
ngctl -d -f - <<EOF
mkpeer ${host_interface_name}: bridge ether link0
name ${host_interface_name}:ether $bridge_name
EOF
ifconfig $(ngctl msg "${host_interface_name}:" getifname | grep Args | cut -d '"' -f 2) name "${host_interface_name}" "$ip_range" up
fi
}
function ng_exists {
ngctl status "${1}" >/dev/null 2>&1
}
function ng_bridge_get_num_hooks {
ngctl show "${1}" | grep -oE 'Num hooks: [0-9]+' | sed 's/Num hooks: //g'
}
function detect_available_link {
bridge_name="$1"
linknum=1
while true; do
link_name="link${linknum}"
if ! ng_exists "${bridge_name}:${link_name}"; then
echo "$link_name"
return
fi
(>&2 echo "$link_name failed on $bridge_name")
linknum=$((linknum + 1))
if [ "$linknum" -gt 90 ]; then
(>&2 echo "No available links on bridge $bridge_name")
exit 1
fi
done
}
function wait_for_interface_to_exist {
# Wait for a vnet interface to exist again as a jail is shutting
# down. If you delete the netgraph node before the interface
# device exists, then the interface device will persist in a
# broken state.
ifname="$1"
max_wait_seconds="$2"
start=$(date +%s)
while true; do
now=$(date +%s)
if [ $((now - start)) -gt $max_wait_seconds ]; then
(>&2 echo "Waited for at least $max_wait_seconds seconds but the interface $ifname did not appear.")
return 1;
fi
if ifconfig "$ifname" >/dev/null 2>&1; then
return 0;
fi
sleep 2
done
}
function sanitize_interface_name {
echo "${1:0:15}"
}
2024-07-12 23:58:50 +00:00
function calculate_mac_address {
local name="$1"
local source
source=$(md5 -r -s "$name" | awk '{print $1}')
echo "06:${source:0:2}:${source:2:2}:${source:4:2}:${source:6:2}:${source:8:2}"
}
main "${@}"