Synnefo » snf-ganeti

#!/bin/bash

set -e

LIBDIR="@LOCALSTATEDIR@/lib"

DATADIR="${LIBDIR}/ganeti"

RUNDIR="@LOCALSTATEDIR@/run"

GNTRUNDIR="${RUNDIR}/ganeti"

LOGDIR="@LOCALSTATEDIR@/log"

GNTLOGDIR="${LOGDIR}/ganeti"

LOCKDIR="@LOCALSTATEDIR@/lock"

_fileset_owner() {

  case "$1" in

    masterd)

      echo "@GNTMASTERUSER@:@GNTMASTERDGROUP@"

      ;;

    confd)

      echo "@GNTCONFDUSER@:@GNTCONFDGROUP@"

      ;;

    rapi)

      echo "@GNTRAPIUSER@:@GNTRAPIGROUP@"

      ;;

    noded)

      echo "root:@GNTMASTERDGROUP@"

      ;;

    daemons)

      echo "@GNTMASTERUSER@:@GNTDAEMONSGROUP@"

      ;;

    masterd-confd)

      echo "@GNTMASTERUSER@:@GNTCONFDGROUP@"

      ;;

    *)

      echo "root:root"

      ;;

  esac

}

_ensure_file() {

  local file="$1"

  local perm="$2"

  local owner="$3"

  [[ -e "${file}" ]] || return 1

  chmod ${perm} "${file}"

  if ! [[ -z "${owner}" ]]; then

    chown ${owner} "${file}"

  fi

  return 0

}

_ensure_dir() {

  local dir="$1"

  local perm="$2"

  local owner="$3"

  [[ -d "${dir}" ]] || mkdir "${dir}"

  _ensure_file "${dir}" "${perm}" "${owner}"

}

_gather_files() {

  local path="$1"

  local perm="$2"

  local user="$3"

  local group="$4"

  shift 4

  find "${path}" -type f "(" "!" -perm ${perm} -or "(" "!" -user ${user} -or \

       "!" -group ${group} ")" ")" "$@"

}

_ensure_datadir() {

  local full_run="$1"

  _ensure_dir ${DATADIR} 0755 "$(_fileset_owner masterd)"

  _ensure_dir ${DATADIR}/queue 0700 "$(_fileset_owner masterd)"

  _ensure_dir ${DATADIR}/queue/archive 0700 "$(_fileset_owner masterd)"

  _ensure_dir ${DATADIR}/uidpool 0750 "$(_fileset_owner noded)"

  _ensure_dir ${DATADIR}/rapi 0750 "$(_fileset_owner rapi)"

  # We ignore these files if they don't exists (incomplete setup)

  _ensure_file ${DATADIR}/cluster-domain-secret 0640 \

               "$(_fileset_owner masterd)" || :

  _ensure_file ${DATADIR}/config.data 0640 "$(_fileset_owner masterd-confd)" || :

  _ensure_file ${DATADIR}/hmac.key 0440 "$(_fileset_owner confd)" || :

  _ensure_file ${DATADIR}/known_hosts 0644 "$(_fileset_owner masterd)" || :

  _ensure_file ${DATADIR}/rapi.pem 0440 "$(_fileset_owner rapi)" || :

  _ensure_file ${DATADIR}/rapi/users 0640 "$(_fileset_owner rapi)" || :

  _ensure_file ${DATADIR}/server.pem 0440 "$(_fileset_owner masterd)" || :

  _ensure_file ${DATADIR}/queue/serial 0600 "$(_fileset_owner masterd)" || :

  # To not change the utils.LockFile object

  touch ${DATADIR}/queue/lock

  _ensure_file ${DATADIR}/queue/lock 0600 "$(_fileset_owner masterd)"

  if ! [[ -z "${full_run}" ]]; then

    local queue_owner="$(_fileset_owner masterd)"

    local ssconf_owner="$(_fileset_owner noded)"

    _gather_files ${DATADIR}/queue 0600 @GNTMASTERUSER@ @GNTMASTERDGROUP@ | \

    while read path; do

      _ensure_file "$path" 0600 "$queue_owner"

    done

    _gather_files ${DATADIR} 0600 root @GNTMASTERDGROUP@ -name 'ssconf_*' | \

    while read path; do

      _ensure_file "$path" 0444 "$ssconf_owner"

    done

  fi

}

_ensure_rundir() {

  _ensure_dir ${GNTRUNDIR} 0775 "$(_fileset_owner daemons)"

  _ensure_dir ${GNTRUNDIR}/socket 0750 "$(_fileset_owner daemons)"

  _ensure_dir ${GNTRUNDIR}/bdev-cache 0755 "$(_fileset_owner noded)"

  _ensure_dir ${GNTRUNDIR}/instance-disks 0755 "$(_fileset_owner noded)"

  _ensure_dir ${GNTRUNDIR}/crypto 0700 "$(_fileset_owner noded)"

  _ensure_dir ${GNTRUNDIR}/import-export 0755 "$(_fileset_owner noded)"

  # We ignore this file if it don't exists (not yet start up)

  _ensure_file ${GNTRUNDIR}/socket/ganeti-master 0770 \

               "$(_fileset_owner daemons)" || :

}

_ensure_logdir() {

  _ensure_dir ${GNTLOGDIR} 0770 "$(_fileset_owner daemons)"

  _ensure_dir ${GNTLOGDIR}/os 0750 "$(_fileset_owner daemons)"

  # We ignore these files if they don't exists (incomplete setup)

  _ensure_file ${GNTLOGDIR}/master-daemon.log 0600 "$(_fileset_owner masterd)" || :

  _ensure_file ${GNTLOGDIR}/conf-daemon.log 0600 "$(_fileset_owner confd)" || :

  _ensure_file ${GNTLOGDIR}/node-daemon.log 0600 "$(_fileset_owner noded)" || :

  _ensure_file ${GNTLOGDIR}/rapi-daemon.log 0600 "$(_fileset_owner rapi)" || :

}

_ensure_lockdir() {

  _ensure_dir ${LOCKDIR} 1777 ""

}

_operate_while_hold() {

  local fn=$1

  local path=$2

  shift 2

  (cd ${path};

   ${fn} "$@")

}

main() {

  local full_run

  while getopts "f" OPTION; do

    case ${OPTION} in

      f) full_run=1 ;;

    esac

  done

  _operate_while_hold "_ensure_datadir" ${DATADIR} ${full_run}

  _operate_while_hold "_ensure_rundir" ${RUNDIR}

  _operate_while_hold "_ensure_logdir" ${LOGDIR}

  _operate_while_hold "_ensure_lockdir" @LOCALSTATEDIR@

}

main "$@"