#!/sbin/sh
#
#
# CDDL HEADER START
#
# The contents of this file are subject to the terms of the
# Common Development and Distribution License (the "License").
# You may not use this file except in compliance with the License.
#
# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
# or http://www.opensolaris.org/os/licensing.
# See the License for the specific language governing permissions
# and limitations under the License.
#
# When distributing Covered Code, include this CDDL HEADER in each
# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
# If applicable, add the following below this CDDL HEADER, with the
# fields enclosed by brackets "[]" replaced with your own identifying
# information: Portions Copyright [yyyy] [name of copyright owner]
#
# CDDL HEADER END
#
#

#
# Copyright (c) 2015, 2016, Oracle and/or its affiliates. All rights reserved.
#

. /lib/svc/share/smf_include.sh
. /lib/svc/share/net_include.sh

typeset -r OVS_OVSDB_FMRI=svc:/application/openvswitch/ovsdb-server:default
typeset -r OVS_VSWITCHD_FMRI=svc:/application/openvswitch/vswitch-server:default

typeset -r OVS_USRLIB_DIR=/usr/lib/ovs
typeset -r OVS_VARLIB_DIR=/var/lib/ovs
typeset -r OVS_TMP_DIR=/var/run/ovs
typeset -r OVS_LOG_DIR=/var/log/ovs

typeset -r OVS_USER=_ovs
typeset -r OVS_GROUP=_ovs

typeset -r OVS_SBIN_DIR=/usr/sbin
typeset -r OVSDB_TOOL=${OVS_SBIN_DIR}/ovsdb-tool
typeset -r OVS_VSCTL=${OVS_SBIN_DIR}/ovs-vsctl

typeset -r OVS_VSWITCHD=ovs-vswitchd
typeset -r OVSDB_SERVER=ovsdb-server
typeset -r OVS_CLEAN=ovs-clean.py
typeset -r OVSDB_SERVER_PATH=${OVS_USRLIB_DIR}/${OVSDB_SERVER}
typeset -r OVS_VSWITCHD_PATH=${OVS_USRLIB_DIR}/${OVS_VSWITCHD}
typeset -r OVS_CLEAN_PATH=${OVS_USRLIB_DIR}/${OVS_CLEAN}

typeset -r OVSDB_REMOTE=${OVS_TMP_DIR}/db.sock

typeset -r PFEXEC=/usr/bin/pfexec
typeset -r MKDIR=/usr/bin/mkdir
typeset -r CHOWN=/usr/bin/chown
typeset -r PKILL=/usr/bin/pkill
typeset -r IPADM=/usr/sbin/ipadm

errlog () {
	echo $1 >&2
}

create_ovs_tempdir() {
	if [[ ! -d ${OVS_TMP_DIR} ]]; then
		${PFEXEC} ${MKDIR} -m 775 ${OVS_TMP_DIR} || exit $SMF_EXIT_ERR_CONFIG
                ${PFEXEC} ${CHOWN} ${OVS_USER}:${OVS_GROUP} ${OVS_TMP_DIR}
	fi
}

start_ovsdb_server() {
	typeset -i OVSDB_INIT=0
	typeset -r OVSDB_DATABASE=${OVS_VARLIB_DIR}/etc/conf.db

	create_ovs_tempdir

	if [[ ! -f ${OVSDB_DATABASE} ]]; then
		typeset -r OVSDB_SCHEMA=${OVS_USRLIB_DIR}/share/vswitch.ovsschema

		echo "Creating ${OVSDB_DATABASE} from ${OVSDB_SCHEMA}"
		${OVSDB_TOOL} create ${OVSDB_DATABASE} ${OVSDB_SCHEMA}
		if [ $? -ne 0 ]; then
			errlog "Error creating database, exiting"
			return 1
		fi
		OVSDB_INIT=1
	fi

	typeset -r OVSDB_LOGFILE=${OVS_LOG_DIR}/ovsdb-server.log
	typeset -r OVSDB_PIDFILE=${OVS_TMP_DIR}/ovsdb-server.pid

	${OVSDB_SERVER_PATH} ${OVSDB_DATABASE} \
	    -vconsole:emer -vsyslog:off -vfile:warn \
	    --remote=punix:${OVSDB_REMOTE} \
	    --remote=db:Open_vSwitch,Open_vSwitch,manager_options \
	    --private-key=db:Open_vSwitch,SSL,private_key \
	    --certificate=db:Open_vSwitch,SSL,certificate \
	    --bootstrap-ca-cert=db:Open_vSwitch,SSL,ca_cert \
	    --no-chdir --log-file=${OVSDB_LOGFILE} \
	    --pidfile=${OVSDB_PIDFILE} \
	    --detach
	if [ $? -ne 0 ]; then
		errlog "${OVSDB_SERVER_PATH} failed with $?"
		exit $SMF_EXIT_ERR_FATAL
	fi

	if [ ${OVSDB_INIT} -ne 0 ]; then
		echo "Initializing OVSDB database"
		${OVS_VSCTL} --no-wait init
		if [ $? -ne 0 ]; then
			errlog "${OVSDB_VSCTL} failed with $?"
			exit $SMF_EXIT_ERR_FATAL
		fi
	fi
}

clean_vswitch() {
        typeset BRIDGENAME

	SVC_METHOD=$1
	
        ${OVS_VSCTL} list-br |
                while read BRIDGENAME; do
			${PFEXEC} ${IPADM} disable-if -t ${BRIDGENAME} >/dev/null 2>&1
                        ${OVS_CLEAN_PATH} delete-vnic ${BRIDGENAME} >/dev/null 2>&1
                        if [[ $? -ne 0 && "$1" = 'stop' ]]; then
                                errlog "Error $? removing ${BRIDGENAME} VNIC"
                        fi
                done
        ${OVS_CLEAN_PATH} reset-ofports
        if [ $? -ne 0 ]; then
        	errlog "Error $? resetting OF ports"
	fi

        ${OVS_CLEAN_PATH} remove-flows
        if [ $? -ne 0 ]; then
                errlog "Error $? removing OF flows"
        fi

        ${OVS_CLEAN_PATH} delete-etherstub ovs.etherstub0
        if [ $? -ne 0 ]; then
                errlog "Error $? deleting the OVS etherstub"
        fi
}    

stop_vswitch_server() {
        typeset BRIDGENAME

        $PKILL -x -u ${OVS_USER} -z `smf_zonename` ${OVS_VSWITCHD}
        if [[ $? -ne 0 && $? -ne 1 ]]; then
                errlog "pkill of ${OVS_VSWITCHD} failed with $?"
        fi

        clean_vswitch stop
}

start_vswitch_server() {
	typeset -r VSWITCHD_LOGFILE=${OVS_LOG_DIR}/ovs-vswitchd.log
	typeset -r VSWITCHD_PIDFILE=${OVS_TMP_DIR}/ovs-vswitchd.pid

	create_ovs_tempdir

        clean_vswitch start

	${PFEXEC} ${OVS_VSWITCHD_PATH} unix:${OVSDB_REMOTE} \
	    -vconsole:emer -vsyslog:off -vfile:warn --mlockall --no-chdir \
	    --log-file=${VSWITCHD_LOGFILE} \
	    --pidfile=${VSWITCHD_PIDFILE} \
	    --detach
	if [ $? -ne 0 ]; then
		errlog "${OVS_VSWITCHD} failed with $?"
		exit $SMF_EXIT_ERR_FATAL
	fi


	${OVS_VSCTL} list-br |
		while read BRIDGENAME; do
			${PFEXEC} ${IPADM} enable-if -t ${BRIDGENAME} >/dev/null 2>&1
		done
}

case "$1" in
'start')
	case "$SMF_FMRI" in
	"${OVS_OVSDB_FMRI}")
		start_ovsdb_server
		;;
	"${OVS_VSWITCHD_FMRI}")
		start_vswitch_server
		;;
	*)
		echo "$SMF_FMRI does not support start method"
		exit $SMF_EXIT_ERR_FATAL
		;;
	esac
	;;
'stop')
	case "$SMF_FMRI" in
	"${OVS_VSWITCHD_FMRI}")
		stop_vswitch_server
		;;
	*)
		echo "$SMF_FMRI does not support stop method"
		exit $SMF_EXIT_ERR_FATAL
		;;
	esac 
	;;
*)
	echo "Service must be invoked from within SMF"
	exit $SMF_EXIT_ERR_FATAL
	;;
esac

exit $SMF_EXIT_OK