mk-job.aix

#!/bin/ksh
# Copyright (C) 2020 tribe29 GmbH - License: GNU General Public License v2
# This file is part of Checkmk (https://checkmk.com). It is subject to the terms and
# conditions defined in the file COPYING, which is part of this source code package.

export MK_VARDIR=/var/lib/check_mk_agent
TMPDIR=${TMPDIR:-/tmp}

help() {
    echo "Usage: mk-job IDENT PROGRAM [ARGS...]"
    echo ""
    echo "Execute PROGRAM as subprocess while measuring performance information"
    echo "about the running process and writing it to an output file. This file"
    echo "can be monitored using Check_MK. The Check_MK Agent will forward the"
    echo "information of all job files to the monitoring server."
    echo ""
    echo "This file is being distributed with the Check_MK Agent."
}

if [ $# -lt 2 ]; then
    help >&2
    exit 1
fi

MYSELF=$(whoami)
OUTPUT_PATH="${MK_VARDIR}/job/${MYSELF}"
IDENT="${1}"
TMP_FILE="${TMPDIR}/${IDENT}.$$"
RUNNING_FILE="${OUTPUT_PATH}/${IDENT}.$$running"
COMPLETED_FILE="${OUTPUT_PATH}/${IDENT}"

shift

if [ ! -d "${OUTPUT_PATH}" ]; then
    if [ "${MYSELF}" = root ]; then
        mkdir -p "${OUTPUT_PATH}"
    else
        echo "ERROR: Missing output directory ${OUTPUT_PATH} for non-root user '${MYSELF}'." >&2
        exit 1
    fi
fi

if ! type "${1}" >/dev/null 2>&1; then
    echo -e "ERROR: Cannot run ${1}. Command not found.\n" >&2
    help >&2
    exit 1
fi

echo "start_time $(perl -e 'print time')" >"${TMP_FILE}" 2>/dev/null
cp "${TMP_FILE}" "${RUNNING_FILE}" 2>/dev/null

if [ ! -w "${RUNNING_FILE}" ]; then
    # Looks like we are lacking the permissions to create this file..
    # In this scenario no mk-job status file is created. We simply execute the command
    rm "${TMP_FILE}" 2>/dev/null
    exec "$@"
    exit $?
fi

/usr/bin/time -p "$@" 2>>"${TMP_FILE}" # the -p option writes to stderr, strangely...
RC=$?
echo "exit_code ${RC}" >>"${TMP_FILE}"

rm "${RUNNING_FILE}"
mv "${TMP_FILE}" "${COMPLETED_FILE}"
exit $RC