X-Git-Url: https://git.rrq.au/?a=blobdiff_plain;f=functions;h=8f39d4014e9213a56c63f4dc03b228f94e73c5ce;hb=HEAD;hp=1d52d7439d9227bb0c86ebdea25272f8462a2839;hpb=df3ccb2170beff852f8ec7d421709c5e85bafc7d;p=rrq%2Foverlay-boot.git diff --git a/functions b/functions index 1d52d74..8f39d40 100644 --- a/functions +++ b/functions @@ -1,84 +1,212 @@ -# This file implements common functions for all boot methods +# This file implements common functions for all boot scripts +# Rerun with sudo if needed +[ $(id -u) = 0 ] || exec sudo $0 $@ + +export ACTION="$(basename $0)" + +# Function to write a message and exit with error code die() { echo "$*" >&2 exit 1 } +beginswith() { + [ "$1" != "${1#$2}" ] +} + +# Function to setup subhost name and log file +subhost_name() { + CONFIG="$1" + [ -r "$CONFIG" ] || die "Cannot use $CONFIG" + config NAME "$(basename $CONFIG .conf)" + config LOG /tmp/oly-$NAME.log +} + +# Function to set up all subhost configuration +subhost_config() { + + config BASE + BASE="$(cd $(dirname $CONFIG); realpath $BASE)" + [ -z "$BASE" ] && die "BASE is unset; bogus $CONFIG ?" + [ -d "$BASE" ] || die "$BASE is not a directory; bogus $CONFIG ?" + cd "$BASE" || die "$BASE is inaccessible" + + config CABLES "" + config LIVE "$BASE/live" + config UPPER "$BASE/root" + config WORK "$BASE/work" + config LOWER "/" + config START "networking ssh" + config PREMOUNT "$PROGRAMDIR/overlay-premount" + config POSTMOUNT "$PROGRAMDIR/overlay-postmount" + config INIT "$PROGRAMDIR/overlay-init" + config RAM_SIZE 50M +} + +# function to reverse the $* words +reverse() { + local OUT="" + for w in $* ; do OUT="$w $OUT" ; done + echo "${OUT% }" +} + # grab and set a configuration variable # $1 = variable, [ $2 = default .. error otherwise ] config() { - eval $1="$(sed "/^$1=.*/{s|^$1=||;b};d" $CONFIG)" - [ -z "$(eval echo "\$$1")" ] || return 0 - [ $# -lt 2 ] && die "Missing $1=... in $CONFIG" - eval $1="'$2'" - eval echo "$1=\$$1" + local V W + read V <&2 } -# Unless the subhost already has a private /etc/network/interfaces, -# install an "empty" one +# Install a default $1/etc/network/interfaces on the subhost root $1 setup_networking() { - [ -r $UPPER/etc/network/interfaces ] && return 0 - [ "$UPPER/etc/network" = "/etc/network" ] && exit 1 - mkdir -p $UPPER/etc/network - cat < $UPPER/etc/network/interfaces + [ -r $1/etc/network/interfaces ] && return 0 + mkdir -p $1/etc/network + cat <> $1/etc/network/interfaces # Generated for $NAME subhost auto lo iface lo inet loopback EOF - return 1 + for IF in $(ip netns exec $NAME ip link show | grep "^eth") ; do + cat <> $1/etc/network/interfaces + +auto eth$i +iface eth$i inet manual +EOF + done } # Setup the network namespace for the given $CABLES # $1=netns ( $2="br=mac" .. ) +# br is optional, mac is optional. +# If mac is .N then it's taken as vlan tag on prior outer interface +# (with ifup configuration) and the inner interface is left alone. setup_veth_cables() { local NETNS BR IF MAC C i ADD NETNS="$1" shift 1 i=0 - ADD=false - setup_networking || ADD=true for C in "$@" ; do IF=$NETNS$i MAC="${C#*=}" - [ -z "$MAC" ] || MAC="address $MAC" - ip link add $IF type veth peer name eth$i $MAC netns $NETNS - ip link set $IF up - $ADD && cat <> $UPPER/etc/network/interfaces - -auto eth$i -iface eth$i inet manual -EOF + if ip link show $IF > /dev/null 2>&1 ; then + : # The interface exists already (bad badness); let things fail + elif ifquery --state $IF >/dev/null 2>&1 ; then + # doesn't exist but has residue state; quiet cleanup + ifdown -f $IF > /dev/null 2>&1 + fi + if [ -z "$MAC" ] ; then + # set up veth with "random" mac address + ip link add $IF type veth peer name eth$i netns $NETNS + elif [ -z "${MAC%%.*}" ] ; then + # set up a host vlan with specified tag on previous eth + i=$((i-1)) + IF=$NETNS$i$MAC + ifup $IF + else + # set up veth with specified mac address + ip link add $IF type veth peer name eth$i address $MAC netns $NETNS + fi BR="${C%=*}" - [ -z "$BR" ] || brctl addif $BR $IF + if [ -z "$BR" ] ; then + ifup $IF || ip link set $IF up + else + ip link set $IF up + brctl addif $BR $IF + fi i=$((i+1)) done } -REAPER=$(dirname $(realpath $0))/reaper/reaper -# (name live system root work) -# Set up an overlay fmr $name on $live, with a new tmpfs on its /run, +# Set up an overlay for $name on $live, with a new tmpfs on its /run, # and "install" a "reaper" as the upcoming pid 1 setup_overlay() { - set -x - mkdir -p $4/run - mount -t tmpfs -osize=100M tmpfs $4/run - mkdir -p $4/run/lock - grep -q "$1 $2" /proc/mounts || \ - mount -t overlay -olowerdir=$3,upperdir=$4,workdir=$5 $1 $2 || \ - die "Cannot set up the overlay mount $2" - mount --bind $4/run $2/run - cp $REAPER $LIVE/.reaper -} + local NAME="$1" LIVE="$2" LOWER="$3" UPPER="$4" WORK="$5" + + echo setup_overlay "$NAME" "$LIVE" "$LOWER" "$UPPER" "$WORK" + + if grep -qE "^[^ ]+ $LIVE " /proc/mounts ; then + die "$LIVE already has a mount" + fi + + [ -d "$UPPER" ] || die "UPPER=$UPPER is not a directory" + [ -d "$LOWER" ] || die "LOWER=LOWPER is not a directory" + [ -d "$LIVE" ] || die "LOWER=LOWPER is not a directory" + [ -x "${PREMOUNT%% *}" ] || die "PREMOUNT=${PREMOUNT%% *} not executable" + [ -f "${PREMOUNT%% *}" ] || die "PREMOUNT='$PREMOUNT' is not a command" + [ -x "${POSTMOUNT%% *}" ] || \ + die "POSTMOUNT=${POSTMOUNT%% *} not executable" + [ -f "${POSTMOUNT%% *}" ] || \ + die "POSTMOUNT='$POSTMOUNT' is not a command" + # UPPER is the same as LOWER then skip the overlay mount + if [ "$UPPER" != "$LOWER" ] ; then + # sanity check + [ -d "$WORK" ] || die "WORK=$WORK is not a directory" -# Check if $SRV is "live" ; will + env CONFIG="$CONFIG" $PREMOUNT "$UPPER" + + OLY="-olowerdir=$3,upperdir=$UPPER,workdir=$5" + if ! mount -t overlay "$OLY" $1 $2 ; then + umount -R "$UPPER/dev" + umount "$UPPER/run" + die "Cannot set up the overlay mount $2" + fi + elif [ "$LIVE" != "$UPPER" ] ; then + # With UPPER = LOWER we rather make a bind mount to LIVE + env CONFIG="$CONFIG" $PREMOUNT "$UPPER" + mount --bind $UPPER $LIVE + fi + + grep ^SHARE= "$CONFIG" | while read A ; do + B="$(echo ${A#SHARE=})" + D="$(realpath "$B")" + [ "$D" = "$LOWER" ] && continue + if [ -d "$D" ] ; then + echo bind mount $D onto $LIVE$B + mkdir -p $LIVE$D + mount --bind $D $LIVE$B + fi + done + + env CONFIG="$CONFIG" $POSTMOUNT "LIVE" "$UPPER" +} + +# Find the "unshare" process for $1 and echo the its pid and the pids +# of its child processes. is_live() { - pgrep -f ".reaper $SRV" > /dev/null + local NAME=$1 + local USPID="$(pgrep -f "unshare.* $NAME ")" + [ -z "$USPID" ] && return 1 + echo "$USPID $(ps -hopid --ppid=$USPID)" } -start_services() { - for S in "$@" ; do - service $S start - done +# Find all overlay-boot processes and list their config files +list_running() { + pgrep -a overlay-boot | awk '{print $4}' +} + +# Start cgroup v2 cpuset accounting if enabled. +# Needs manual enabling, with: +# mount -t cgroup2 cgroup2 /sys/fs/cgroup +setup_cgroup2_accounting() { + local NAME="$1" ME="$2" + local ACCDIR="$(awk '$3 == "cgroup2" {print $2; exit}' /proc/mounts)" + [ -z "$ACCDIR" ] && return 0 + mkdir -p "$ACCDIR/$NAME" + echo "$ME" > $ACCDIR/$NAME/cgroup.procs }