daemon.sh

#!/bin/sh
# 
# Copyright (c) 2013 Baptiste Daroussin <bapt@FreeBSD.org>
# Copyright (c) 2013 Bryan Drewery <bdrewery@FreeBSD.org>
# All rights reserved.
# 
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
# 
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
# SUCH DAMAGE.

usage() {
	echo "poudriere daemon [options]

Options:
    -n        -- No daemonise
    -p        -- pidfile
    -k        -- kill the running daemon"

    exit 1
}

start_queue_reader() {
	queue_reader_main &
	QUEUE_READER_PID=$!
}

queue_reader_main() {
	# Read from the socket and then write the command
	# to the watchdir. This is done so non-privileged users
	# do not need write access to the real queue dir
	umask 0111 # Create rw-rw-rw
	nc -klU ${QUEUE_SOCKET} | while read name command; do
		echo "${command}" > ${WATCHDIR}/${name}
	done
}

stop_queue_reader() {
	if [ -n "${QUEUE_READER_PID}" ]; then
		kill ${QUEUE_READER_PID} 2>/dev/null || :
		wait ${QUEUE_READER_PID} 2>/dev/null || :
		unset QUEUE_READER_PID
	fi
	rm -f ${QUEUE_SOCKET}
}


SCRIPTPATH=`realpath $0`
SCRIPTPREFIX=`dirname ${SCRIPTPATH}`
PTNAME="default"
NODAEMONIZE=0
KILL=0

. ${SCRIPTPREFIX}/common.sh

if [ -z "${DAEMON_ARGS_PARSED}" ]; then
	[ $# -eq 0 ] && usage

	while getopts "knp:" FLAG; do
		case "${FLAG}" in
		n)
			NODAEMONIZE=1
			;;
		p)
			PIDFILE=${OPTARG}
			;;
		k)
			KILL=1
			;;
		esac
	done
	if [ ${KILL} -eq 1 ]; then
		pkill -15 -F ${PIDFILE} >/dev/null 2>&1 || exit 1
		exit 0
	fi

	if [ ${NODAEMONIZE} -eq 0 ]; then
		daemon -f -p ${PIDFILE} env -i PATH=${PATH} DAEMON_ARGS_PARSED=1 $0 || exit 1
		exit 0
	else
		pgrep -F ${PIDFILE} >/dev/null 2>&1 && err 1 "poudriere daemon is already running"
		echo "$$" > ${PIDFILE}
	fi
fi

# Start the queue reader
start_queue_reader

CLEANUP_HOOK=daemon_cleanup
daemon_cleanup() {
	stop_queue_reader
}

while :; do
	next=$(find ${WATCHDIR} -type f -depth 1 -print -quit 2>/dev/null)
	if [ -z "${next}" ]; then
		dirwatch ${WATCHDIR}
		continue
	fi
	POUDRIERE_ARGS=$(sed -n "s/^POUDRIERE_ARGS: //p" ${next})
	mkdir -p ${POUDRIERE_DATA}/logs/daemon
	poudriere ${POUDRIERE_ARGS} > ${POUDRIERE_DATA}/logs/daemon/${next##*/}.log
	rm -f ${next}
done