recipes-google/bare-metal-setup/files/bare-metal-setup.sh.in

#!/bin/bash

# Statically setting the BM_MODE
BM_MODE="@BM_MODE@"

GPIO_CHIP="@GPIO_CHIP@"
GPIO_LINE="@GPIO_LINE@"
HIDE_BOOT_DRIVE="@HIDE_BOOT_DRIVE@"
SHOW_BOOT_DRIVE="@SHOW_BOOT_DRIVE@"
ENABLE_BM_FILE="@ENABLE_BM_FILE@"
ENABLE_CONSOLE_FILE="@ENABLE_CONSOLE_FILE@"
BMREADY_FILE="@BMREADY_FILE@"

READ_ONLY_CONSOLE_FLAG="/run/readonly-console.flag"

function override_obmc_console_in_bm_mode() {
   host_console_client_service="serial-to-host@.service"
   bmc_console_client_service="serial-to-bmc@.service"

   systemd_runtime_dir='/run/systemd/system'

   host_console_client_dir="${systemd_runtime_dir}/${host_console_client_service}.d"
   bmc_console_client_dir="${systemd_runtime_dir}/${bmc_console_client_service}.d"

   touch "${READ_ONLY_CONSOLE_FLAG}"

   mkdir -p "${host_console_client_dir}" "${bmc_console_client_dir}"

   echo "[Unit]
ConditionPathExists=${READ_ONLY_CONSOLE_FLAG}
[Service]
RestartPreventExitStatus=
RestartPreventExitStatus=SIGINT SIGQUIT SIGTSTP
SuccessExitStatus=
SuccessExitStatus=SIGINT SIGQUIT SIGTSTP
ExecStart=
ExecStart=-/sbin/agetty -8 -n -l /usr/libexec/readonly-obmc-console-client -i -N -R -L %I 115200 xterm
ExecStopPost=
ExecStopPost=-/bin/sh -c 'systemctl start --no-block serial-to-bmc@%i'
" > "${host_console_client_dir}/50-bm-override.conf"

   echo "[Service]
ExecStopPost=
ExecStopPost=/bin/bash -c \"if [ -f ${READ_ONLY_CONSOLE_FLAG} ]; then \
    systemctl start --no-block serial-to-host@%i; \
else \
    systemctl start --no-block serial-to-bmc@%i; \
fi\"
" > "${bmc_console_client_dir}/50-bm-override.conf"

}

function override_hostlogger_in_bm_mode() {
    systemd_runtime_dir='/run/systemd/system'
    hostlogger_service="hostlogger@.service"
    hostlogger_override_dir="${systemd_runtime_dir}/${hostlogger_service}.d"

    mkdir -p ${hostlogger_override_dir}
    echo "[Unit]
BindsTo=gbmc-bare-metal-active.target
Before=gbmc-bare-metal-active.target
[Install]
WantedBy=
WantedBy=gbmc-bare-metal-active.target
" > "${hostlogger_override_dir}/50-bm-override.conf"

    # Overriding Install section does not work for already installed service
    # Moving the service directly and recover it when turning off BM mode.
    # mv seems not happy with jffs + symlink.

    systemd_etc_dir='/etc/systemd/system'

    mkdir -p ${systemd_etc_dir}/gbmc-bare-metal-active.target.wants

    cp -P ${systemd_etc_dir}/multi-user.target.wants/hostlogger@*.service \
      ${systemd_etc_dir}/gbmc-bare-metal-active.target.wants/

    rm ${systemd_etc_dir}/multi-user.target.wants/hostlogger@*.service

    # looks like we need to stop from any existing transaction to really prevent
    # it from starting
    systemctl stop hostlogger@*.service
}

function recover_hostlogger_in_bm_mode() {
    systemd_etc_dir='/etc/systemd/system'

    mkdir -p ${systemd_etc_dir}/multi-user.target.wants

    # hostlogger already in multi-user target, return
    ls ${systemd_etc_dir}/multi-user.target.wants/hostlogger@* > /dev/null 2>&1 && return 1

    cp -P  ${systemd_etc_dir}/gbmc-bare-metal-active.target.wants/hostlogger@*.service \
      ${systemd_etc_dir}/multi-user.target.wants/ 2> /dev/null

    rm ${systemd_etc_dir}/gbmc-bare-metal-active.target.wants/hostlogger@*.service 2> /dev/null
}

function hide_boot_drive_in_bm_mode() {
    if [[ -n "${HIDE_BOOT_DRIVE}" ]]; then
        echo "hiding the boot drive with '${HIDE_BOOT_DRIVE}'" >&2
        ${HIDE_BOOT_DRIVE}
    else
        echo "hiding the boot drive with GPIO (${GPIO_CHIP}/${GPIO_LINE})" >&2
        gpioset --mode=signal --background "${GPIO_CHIP}" "${GPIO_LINE}"=0
    fi
}

function show_boot_drive() {
    if [[ -n "${SHOW_BOOT_DRIVE}" ]]; then
        echo "showing the boot drive with '${SHOW_BOOT_DRIVE}'" >&2
        ${SHOW_BOOT_DRIVE}
    else
        echo "showing the boot drive with GPIO (${GPIO_CHIP}/${GPIO_LINE})" >&2
        gpioset --mode=time -s 1 "${GPIO_CHIP}" "${GPIO_LINE}"=1
    fi
}

function main() {
    # In Bare Metal mode, power off the boot drive
    if [[ "$BM_MODE" -eq "1" || -f "$ENABLE_BM_FILE" ]]; then
        # Turn off the power to the boot drive
        # and keep asserting the power down
        if ! hide_boot_drive_in_bm_mode; then
            echo "Unable to hide cSSD/cnSSD - skip adding ${BMREADY_FILE}"
            exit 1
        fi
        echo "Successfully set the GPIO to hide cSSD/cnSSD creating ${BMREADY_FILE}"
        # Disable usb network
        ln -s /dev/null /run/systemd/system/google-usb-dynamic.service 2> /dev/null
        # Change the host console to read only, unless we require to keep the
        # console
        if [[ ! -f "$ENABLE_CONSOLE_FILE" ]]; then
            override_hostlogger_in_bm_mode
            override_obmc_console_in_bm_mode
        fi

        systemctl daemon-reload
        touch "${BMREADY_FILE}"
    else
        # daemon-reload only if we are really recoverying from bm mode
        recover_hostlogger_in_bm_mode && systemctl daemon-reload

        # Turn on the power to the boot drive
        echo "Powering on the cSSD/cnSSD as we are not in BM mode - remove ${BMREADY_FILE}"
        rm -f "${BMREADY_FILE}"
        show_boot_drive
    fi
}

main "$@"