#!/bin/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 2009 Sun Microsystems, Inc.  All rights reserved.
# Use is subject to license terms.
# Copyright 2015 Nexenta Systems, Inc. All rights reserved.
#

proc smf_present {
	test -r /etc/svc/volatile/repository_door && \
	    test ! -f /etc/svc/volatile/repository_door
}

proc smf_clear_env {
	unset \
		SMF_FMRI \
		SMF_METHOD \
		SMF_RESTARTER \
		SMF_ZONENAME
}

# smf_console
#
#   Use as "echo message 2>&1 | smf_console".  If SMF_MSGLOG_REDIRECT is
#   unset, message will be displayed to console.  SMF_MSGLOG_REDIRECT is
#   reserved for future use.
#
proc smf_console {
	/usr/bin/tee ${SMF_MSGLOG_REDIRECT:-/dev/msglog}
}

# smf_zonename
#
#  Prints the name of this zone.

proc smf_zonename {
	echo ${SMF_ZONENAME:=`/sbin/zonename`}
}

# smf_is_globalzone
#
#  Returns zero (success) if this is the global zone.  1 otherwise.
#
proc smf_is_globalzone {
	test ${SMF_ZONENAME:=`/sbin/zonename`} = "global" && return 0
	return 1
}

# smf_is_nonglobalzone
#
#  Returns zero (success) if this is not the global zone.  1 otherwise.
#
proc smf_is_nonglobalzone {
	test ${SMF_ZONENAME:=`/sbin/zonename`} != "global" && return 0
	return 1
}

# smf_configure_ip
#
#  Returns zero (success) if this zone needs IP to be configured i.e.
#  the global zone or has an exclusive stack.  1 otherwise.
#
proc smf_configure_ip {
	test ${SMF_ZONENAME:=`/sbin/zonename`} = "global" -o \
	 $(/sbin/zonename -t) = exclusive && return 0
	return 1
}

# smf_dont_configure_ip
#
#  Inverse of smf_configure_ip
#
proc smf_dont_configure_ip {
	test ${SMF_ZONENAME:=`/sbin/zonename`} != "global" -a \
	 $(/sbin/zonename -t) = shared && return 0
	return 1
}

# smf_dont_configure_vt
#
#  Returns zero (success) if vt functionality is not to be configured,
#  1 otherwise.
#
proc smf_dont_configure_vt {
	test ${SMF_ZONENAME:=`/sbin/zonename`} != "global" && return 0
	/usr/lib/vtinfo > /dev/null 2>&1
	return $?
}

# smf_is_system_labeled
#
#  Returns zero (success) if system is labeled (aka Trusted Extensions).
#  1 otherwise.
#
proc smf_is_system_labeled {
	test ! -x /bin/plabel && return 1
	/bin/plabel > /dev/null 2>&1
	return $?
}

# smf_netstrategy
#   -> (_INIT_NET_IF, _INIT_NET_STRATEGY)
#
#   Sets _INIT_NET_IF to the name for the network-booted
#   interface if we are booting from the network.  _INIT_NET_STRATEGY is
#   assigned the value of the current network configuration strategy.
#   Valid values for _INIT_NET_STRATEGY are "none", "dhcp", and "rarp".
#
#   The network boot strategy for a zone is always "none".
#
proc smf_netstrategy {
	if smf_is_nonglobalzone {"
		_INIT_NET_STRATEGY="none"" export _INIT_NET_STRATEGY
		return 0
	}

	set -- $(/sbin/netstrategy)
	if test $Status -eq 0 {
		test $1 = "nfs"" && \
			_INIT_NET_IF=$2" export _INIT_NET_IF"
		_INIT_NET_STRATEGY=$3" export _INIT_NET_STRATEGY
	} else {
		return 1
	}
}

#
# smf_kill_contract CONTRACT SIGNAL WAIT TIMEOUT
#
#   To be called from stop methods of non-transient services.
#   Sends SIGNAL to the service contract CONTRACT.  If the
#   WAIT argument is non-zero, smf_kill_contract will wait
#   until the contract is empty before returning, or until
#   TIMEOUT expires.
#
#   Example, send SIGTERM to contract 200:
#
#       smf_kill_contract 200 TERM 
#
#   Since killing a contract with pkill(1) is not atomic,
#   smf_kill_contract will continue to send SIGNAL to CONTRACT
#   every second until the contract is empty.  This will catch
#   races between fork(2) and pkill(1).
#
#   Note that time in this routine is tracked (after being input
#   via TIMEOUT) in 10ths of a second.  This is because we want
#   to sleep for short periods of time, and expr(1) is too dumb
#   to do non-integer math.
#
#   Returns 1 if the contract is invalid.
#   Returns 2 if WAIT is "1", TIMEOUT is > 0, and TIMEOUT expires.
#   Returns 0 on success.
#
proc smf_kill_contract {

	setvar time_waited = '0'
	setvar time_to_wait = "$4"

	test -z $time_to_wait && setvar time_to_wait = '0'

	# convert to 10ths.
	setvar time_to_wait = $(/usr/bin/expr $time_to_wait '*' 10)

	# Verify contract id is valid using pgrep
	/usr/bin/pgrep -c $1 > /dev/null 2>&1
	setvar ret = "$Status"
	if test $ret -gt 1  {
		echo "Error, invalid contract \"$1\"" >&2
		return 1
	}

	# Return if contract is already empty.
	test $ret -eq 1 && return 0

	# Kill contract.
	/usr/bin/pkill -$2 -c $1
	if test $Status -gt 1  {
		echo "Error, could not kill contract \"$1\"" >&2
		return 1
	}

	# Return if WAIT is not set or is "0"
	test -z $3 && return 0
	test $3 -eq 0 && return 0
 
	# If contract does not empty, keep killing the contract to catch
	# any child processes missed because they were forking
	/usr/bin/pgrep -c $1 > /dev/null 2>&1
	while test $Status -eq 0 {
		# Return 2 if TIMEOUT was passed, and it has expired
		test $time_to_wait -gt 0 -a $time_waited -ge $time_to_wait && \
		    return 2

		#
		# At five second intervals, issue the kill again.  Note that
		# the sleep time constant (in tenths) must be a factor of 50
		# for the remainder trick to work.  i.e. sleeping 2 tenths is
		# fine, but 27 tenths is not.
		#
		setvar remainder = $(/usr/bin/expr $time_waited % 50)
		if test $time_waited -gt 0 -a $remainder -eq 0 {
			/usr/bin/pkill -$2 -c $1
		}

		# Wait two tenths, and go again.
		/usr/bin/sleep 0.2
		setvar time_waited = $(/usr/bin/expr $time_waited + 2)
		/usr/bin/pgrep -c $1 > /dev/null 2>&1
	}

	return 0
}

#
# smf(5) method and monitor exit status definitions
#   SMF_EXIT_ERR_OTHER, although not defined, encompasses all non-zero
#   exit status values.
#
setvar SMF_EXIT_OK = '0'
setvar SMF_EXIT_ERR_FATAL = '95'
setvar SMF_EXIT_ERR_CONFIG = '96'
setvar SMF_EXIT_MON_DEGRADE = '97'
setvar SMF_EXIT_MON_OFFLINE = '98'
setvar SMF_EXIT_ERR_NOSMF = '99'
setvar SMF_EXIT_ERR_PERM = '100'