virtual-dsm/src/check.sh

#!/usr/bin/env bash
set -u

[ ! -f "/run/qemu.pid" ] && echo "QEMU not running yet.." && exit 0
[ -f "/run/qemu.count" ] && echo "QEMU is shutting down.." && exit 1

file="/run/dsm.url"

if [ ! -f  "$file" ]; then

  # Retrieve IP from guest VM for Docker healthcheck
  RESPONSE=$(curl -s -m 16 -S http://127.0.0.1:2210/read?command=10 2>&1)

  if [[ ! "${RESPONSE}" =~ "\"success\"" ]] ; then
    echo "Failed to connect to guest: $RESPONSE" && exit 1
  fi

  # Retrieve the HTTP port number
  if [[ ! "${RESPONSE}" =~ "\"http_port\"" ]] ; then
    echo "Failed to parse response from guest: $RESPONSE" && exit 1
  fi

  rest=${RESPONSE#*http_port}
  rest=${rest#*:}
  rest=${rest%%,*}
  PORT=${rest%%\"*}

  [ -z "${PORT}" ] && echo "Guest has not set a portnumber yet.." && exit 1

  # Retrieve the IP address
  if [[ ! "${RESPONSE}" =~ "eth0" ]] ; then
    echo "Failed to parse response from guest: $RESPONSE" && exit 1
  fi

  rest=${RESPONSE#*eth0}
  rest=${rest#*ip}
  rest=${rest#*:}
  rest=${rest#*\"}
  IP=${rest%%\"*}

  [ -z "${IP}" ] && echo "Guest has not received an IP yet.." && exit 1

  echo "${IP}:${PORT}" > $file

fi

LOCATION=$(cat "$file")

if ! curl -m 20 -ILfSs "http://${LOCATION}/" > /dev/null; then
  rm -f $file
  echo "Failed to reach http://${LOCATION}"
  exit 1
fi

if [[ "$LOCATION" == "20.20"* ]]; then
  echo "Healthcheck OK"
else
  echo "Healthcheck OK ( ${LOCATION%:*} )"
fi

exit 0