Files
NLnetLabs-krill/docker/entrypoint.sh

87 lines
3.3 KiB
Bash
Executable File

#!/bin/bash
# Prepare the environment and config file for the Krill daemon.
# This script supports several scenarios:
# A. The operator wants to run the Krill daemon using the default setup:
# Krill doesn't have a default API token value, we have to supply one.
# Generate one and announce it, if no KRILL_ADMIN_TOKEN env var was
# supplied by the operator.
#
# B: The operator wants to control the Krill daemon configuration themselves.
# They do this by Docker mounting their own krill.conf over the
# /var/krill/data/krill.conf path.
#
# C: The operator wants to run some other command in the container, e.g.
# krillc.
#
set -e
KRILL_CONF=/var/krill/data/krill.conf
KRILL_FQDN="${KRILL_FQDN:-localhost:3000}"
KRILL_ADMIN_TOKEN="${KRILL_ADMIN_TOKEN:-None}"
KRILL_AUTH_TOKEN="${KRILL_AUTH_TOKEN:-None}"
KRILL_LOG_LEVEL="${KRILL_LOG_LEVEL:-warn}"
KRILL_USE_TA="${KRILL_USE_TA:-false}"
MAGIC="# DO NOT TOUCH, THIS LINE IS MANAGED BY DOCKER KRILL"
LOG_PREFIX="docker-krill:"
log_warning() {
echo >&2 "${LOG_PREFIX} Warning! $*"
}
log_info() {
echo "${LOG_PREFIX} $*"
}
if [ "$1" == "krill" ]; then
# Does the operator want to use their own API token? If so they must
# supply the KRILL_ADMIN_TOKEN env var.
if [ "${KRILL_ADMIN_TOKEN}" == "None" ]; then
# Check if the deprecated env variable was used.
if [ "${KRILL_AUTH_TOKEN}" == "None" ]; then
# Deprecated env variable was not set either.
#
# Generate a unique hard to guess authorization token and export it
# so that the Krill daemon uses it (unless overridden by the Krill
# daemon config file). Only do this if the operator didn't already
# supply a token when launching the Docker container.
export KRILL_ADMIN_TOKEN=$(uuidgen)
else
# Deprecated env variable was set. Use it.
export KRILL_ADMIN_TOKEN=${KRILL_AUTH_TOKEN}
fi
fi
# Announce the token in the Docker logs so that clients can obtain it.
log_info "Securing Krill daemon with token ${KRILL_ADMIN_TOKEN}"
log_info "Configuring ${KRILL_CONF} .."
# If the config file was persisted and the container was recreated with
# different arguments to docker run there may still be some lines in the
# config file that we added before which are now no longer correct. Remove
# any lines that we added.
if ! sed -i "/.\\+${MAGIC}/d" ${KRILL_CONF} 2>/dev/null; then
log_warning "Cannot write to ${KRILL_CONF}. You can ignore this warning if you mounted your own config file over ${KRILL_CONF}."
else
# Persist the current log level being used to krill.conf, whether the
# default defined above or an override supplied by the user via the
# KRILL_LOG_LEVEL environment variable.
cat << EOF >> ${KRILL_CONF}
log_level = "${KRILL_LOG_LEVEL}" ${MAGIC}
EOF
log_info "Dumping ${KRILL_CONF} config file"
cat ${KRILL_CONF}
log_info "End of dump"
fi
fi
# Launch the command supplied either by the default CMD (krill) in the
# Dockerfile or that given by the operator when invoking Docker run. Use exec
# to ensure krill runs as PID 1 as required by Docker for proper signal
# handling. This also allows this Docker image to be used to run krillc
# instead of krill.
exec "$@"