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

#	Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T
#	  All Rights Reserved

# Portions of this source code were derived from Berkeley 4.3 BSD
# under license from the Regents of the University of California.

# set -xv
setvar YPXFR = '/usr/lib/netsvc/yp/ypxfr'
setvar MAKEPATH = '/usr/ccs/bin'
setvar maps = ""publickey publickey.byname""
setvar yproot_dir = '/var/yp'
setvar yproot_exe = '/usr/sbin/yp'
setvar hf = "/var/run/ypservers.$Pid"
setvar XFR = ${YPXFR}

setvar hosts_file = '/etc/hosts'
setvar hosts6_file = '/etc/inet/ipnodes'
setvar clientp = 'F'
setvar masterp = 'F'
setvar slavep = 'F'
setvar host = """"
setvar def_dom = """"
setvar master = """"
setvar got_host_list = 'F'
setvar first_time = 'T'
setvar non_interactive = 'F'
setvar exit_on_error = 'F'
setvar errors_in_setup = 'F'

proc enable_next_boot {
	/usr/sbin/svcadm disable -t $1
	test $Status = 0 || echo "ypinit: unable to temporarily disable $1"
	/usr/sbin/svccfg -s $1 \
	    setprop general/enabled = true
	test $Status = 0 || echo "ypinit: unable to enable $1 for next boot"
}

proc enable_this_boot {
	/usr/sbin/svcadm enable $1
	test $Status = 0 || echo "ypinit: unable to enable $1"
}

proc is_valid_ipaddr {
	test -n $(echo $1 | awk 'NF != 1 {exit} \
	    $1 !~ /[0-9]/ || /[;-~]/ || /!--/ || /\// {exit} \
	    $1 !~ /\./ {exit} {print}) || \
	test -n $(echo $1 | awk 'NF != 1 {exit} \
	    ($1 !~ /[0-9]/ && $1 !~ /[A-F]/ && \
	    $1 !~ /[a-f]/) || \
	    /[;-@]/ || /[G-\`]/ || /[g-~]/ || /!--/ || \
	    /\// {exit} \
	    $1 !~ /:/ {exit} {print})
}

setvar PATH = "/bin:/usr/bin:/usr/etc:/usr/sbin:$yproot_exe:$MAKEPATH:$PATH"
export PATH

# To do cleanup
trap '/usr/bin/rm -f $hf' 0 1 2 3 15

# Check out total number of arguments
match $Argc {
with 1	match $1 {
	with -c	setvar clientp = 'T'
	with -m	setvar masterp = 'T'
	with *	echo 'usage:'
		echo '	ypinit -c [server_name...]'
		echo '	ypinit -m'
		echo '	ypinit -s master_server'
		echo ""
		echo "\
where -c is used to set up a yp client, -m is used to build a master "
                echo "\
yp server data base, and -s is used for a slave data base."
		echo "\
master_server must be an existing reachable yp server."
		exit 1
	}

with 2	match $1 {
	with -s	setvar slavep = 'T'; setvar master = "$2";
		if shell { grep $master $hosts_file $hosts6_file > /dev/null }
		{
			echo ""
		} else {
			echo "server not found in $hosts_file or $hosts6_file"
			exit 1
		}

	# the case with more than one argument with the '-c' option
	# is a subject to enter non-interactive mode
	with -c	setvar clientp = 'T'; setvar non_interactive = 'T';
		
	with *	echo 'usage:'
		echo '	ypinit -c [server_name...]'
		echo '	ypinit -m'
		echo '	ypinit -s master_server'
		echo ""
		echo "\
where -c is used to set up a yp client, -m is used to build a master "
                echo "\
yp server data base, and -s is used for a slave data base."
		echo "\
master_server must be an existing reachable yp server."
		exit 1
	}
with *	match $1 {

	# the case with more than one argument with the '-c' option
	# is a subject to enter non-interactive mode
	with -c	setvar clientp = 'T'; setvar non_interactive = 'T';
		
	with *	echo 'usage:'
		echo '	ypinit -c [server_name...]'
		echo '	ypinit -m'
		echo '	ypinit -s master_server'
		echo ""
		echo "\
where -c is used to set up a yp client, -m is used to build a master "
                echo "\
yp server data base, and -s is used for a slave data base."
		echo "\
master_server must be an existing reachable yp server."
		exit 1
	}
}

if test $Status -ne 0
{
	echo "\
You have to be the superuser to run this.  Please log in as root."
	exit 1
}

setvar host = $(uname -n)

if test $Status -ne 0
{
	echo "Can't get local host's name.  Please check your path."
	exit 1
}

if test -z $host
{
	echo "The local host's name hasn't been set.  Please set it."
	exit 1
}

setvar def_dom = $(domainname)

if test $Status -ne 0
{
	echo "Can't get local host's domain name.  Please check your path."
	exit 1
}

if test -z $def_dom
{
	echo "The local host's domain name hasn't been set.  Please set it."
	exit 1
}

domainname $def_dom
setvar real_def_dom = "$def_dom"
#def_dom=`ypalias -d $def_dom`
setvar ypservers_map = $(ypalias ypservers)
setvar domain_dir = ""$yproot_dir""/""$def_dom""
setvar binding_dir = ""$yproot_dir""/binding/""$def_dom""
setvar binding_file = ""$yproot_dir""/binding/""$def_dom""/ypservers""

if test ! -d $yproot_dir -o -f $yproot_dir
{
    echo "\
The directory $yproot_dir doesn't exist.  Restore it from the distribution."
	exit 1
}

# add domainname and ypservers aliases to aliases file
echo ypservers $ypservers_map >> $yproot_dir/aliases
echo $real_def_dom $def_dom >> $yproot_dir/aliases
sort $yproot_dir/aliases | uniq > /var/run/.ypaliases; mv /var/run/.ypaliases $yproot_dir/aliases

if test ! -d "$yproot_dir"/binding
{
	mkdir "$yproot_dir"/binding
}

if test ! -d  $binding_dir
{
	mkdir  $binding_dir
}

if test $slavep = F
{
	if test $non_interactive = F
	{
		while test $got_host_list = F {
			touch $hf    # make sure file exists
			echo ""
			echo "\
In order for NIS to operate sucessfully, we have to construct a list of the "
			echo "\
NIS servers.  Please continue to add the names for YP servers in order of"
			echo "\
preference, one per line.  When you are done with the list, type a <control D>"
			echo "\
or a return on a line by itself."
			if test $masterp = T
			{
				echo $host > $hf
				echo "\tnext host to add:  $host"
			} elif test -f $binding_file
			{
				if test $first_time = T
				{
					for h in [$(cat $binding_file)]
					{
						echo $h >> $hf
						echo "\tnext host to add:  $h"
					}
				}
			}

			echo  "\tnext host to add:  \c"

			while read h ; test -n "$h"
			{
				#
				# Host should be in the v4 or v6 hosts file or
				# reasonably resemble an IP address.  We'll do a
				# sanity check that a v4 addr is one word consisting
				# of only numbers and the "." character,
				# which should guard against fully qualified
				# hostnames and most malformed entries.  IPv6
				# addresses can be numbers, hex letters, and have
				# at least one ":" character and possibly one or
				# more "." characters for embedded v4 addresses
				#
				if shell { grep $h $hosts_file $hosts6_file > /dev/null } || \
				    shell { test $clientp = T && $(is_valid_ipaddr $h) }
				{
					echo $h >> $hf
					echo  "\tnext host to add:  \c"
				} else {
					echo "host $h not found in $hosts_file or" \
					    "$hosts6_file.\nNot added to the list."
					echo ""
					echo  "Do you wish to abort [y/n: y]  \c"
					read cont_ok

					match $cont_ok {
					with n*	echo "\tnext host to add:  \c"
					with N*	echo "\tnext host to add:  \c"
					with *	exit 1
					}
				}
			}

			echo ""
			if test -s $hf
			{
				echo "The current list of yp servers looks like this:"
				echo ""
				cat $hf
				echo ""
				echo "Is this correct?  [y/n: y]  \c"
			} else {
				echo "You have not added any server information."
				echo ""
				echo "Do you still wish to exit? [y/n: y]  \c"
			}

			read hlist_ok

			match $hlist_ok {
			with n*	setvar got_host_list = 'F'
				setvar first_time = 'F'
				rm $hf
				echo "Let's try the whole thing again..."
			with N*	setvar got_host_list = 'F'
				setvar first_time = 'F'
				rm $hf
				echo "Let's try the whole thing again..."
			with *	setvar got_host_list = 'T'
			}
		}
	} else {
		shift
		 {
			if shell { grep $1 $hosts_file $hosts6_file > /dev/null } || \
			    shell { $(is_valid_ipaddr $1) }
			{
				echo $1 >> $hf
			} else {
				echo "host $1 not found in $hosts_file or" \
				    "$hosts6_file.\nNot added to the list."
				echo ""
			}
			shift
		}
	}

	if test -s $hf
	{
		cp  $hf $binding_file
	}
}

#
# Start client service on next boot, unless we're establishing a slave
# server, in which case the binding is needed now (or should be
# preserved).
#
if test $slavep = T
{
	enable_this_boot network/nis/client:default
} else {
	enable_next_boot network/nis/client:default
}

#
# As a client, our configuration is correct once a binding file is
# established, and so we can exit (making sure we're no longer a server,
# of course).
#
if test $clientp = T
{
	rm $hf
	/usr/sbin/svcadm disable network/nis/server:default
	/usr/sbin/svcadm disable network/nis/xfr:default
	/usr/sbin/svcadm disable network/nis/passwd:default
	/usr/sbin/svcadm disable network/nis/update:default
	exit 0
}

if test $slavep = T
{
	if test $host = $master
	{
		echo "\
The host specified should be a running master yp server, not this machine."
		exit 1
	}

	setvar maps = $(ypwhich -m | egrep $master$| awk '{ printf("%s ",$1) }' -)
	if test -z $maps
	{
		echo "Can't enumerate maps from $master. Please check that it is running."
		exit 1
	}
}

echo ""

echo "Installing the YP database will require that you answer a few questions."
echo "Questions will all be asked at the beginning of the procedure."
echo ""
echo "Do you want this procedure to quit on non-fatal errors? [y/n: n]  \c"
read doexit

match $doexit {
with y*	setvar exit_on_error = 'T'
with Y*	setvar exit_on_error = 'T'
with *	echo "\
OK, please remember to go back and redo manually whatever fails.  If you"
	echo "\
don't, some part of the system (perhaps the yp itself) won't work."
}

echo "The yp domain directory is $yproot_dir""/""$def_dom"

for dir in [$yproot_dir/$def_dom]
{

	if test -d $dir {
		echo  "Can we destroy the existing $dir and its contents? [y/n: n]  \c"
		read kill_old_dir

		match $kill_old_dir {
		with y*	rm -r -f $dir

			if test $Status  -ne 0
			{
			echo "Can't clean up old directory $dir.  Fatal error."
				exit 1
			}

		with Y*	rm -r -f $dir

			if test $Status  -ne 0
			{
			echo "Can't clean up old directory $dir.  Fatal error."
				exit 1
			}

		with *    echo "OK, please clean it up by hand and start again.  Bye"
			exit 0
		}
	}

	mkdir $dir

	if test $Status  -ne 0
	{
		echo "Can't make new directory $dir.  Fatal error."
		exit 1
	}

}

if test $slavep = T
{
	echo "\
There will be no further questions. The remainder of the procedure should take"
	echo "a few minutes, to copy the data bases from $master."

	for dom in  [$real_def_dom]
	{
		for map in [$maps]
		{
			echo "Transferring $map..."
			$XFR -h $master -c -d $dom $map

			if test $Status  -ne 0
			{
				setvar errors_in_setup = 'T'

				if test $exit_on_error = T
				{
					exit 1
				}
			}
		}
	}

	echo ""
	echo  "${host}'s nis data base has been set up\n"

	if test $errors_in_setup = T
	{
		echo " with errors.  Please remember"
		echo "to figure out what went wrong, and fix it."
	} else {
		echo " without any errors."
	}

	# enable slave services
	enable_this_boot network/nis/server:default

	enable_this_boot network/nis/client:default

	exit 0
} else {

	rm -f $yproot_dir/*.time

	echo "\
There will be no further questions. The remainder of the procedure should take"
	echo "5 to 10 minutes."

	echo "Building $yproot_dir/$def_dom/ypservers..."
	makedbm $hf $yproot_dir/$def_dom/$ypservers_map

	if test $Status  -ne 0
	{
		echo "\
Couldn't build yp data base $yproot_dir/$def_dom/$ypservers_map."
		setvar errors_in_setup = 'T'

		if test $exit_on_error = T
		{
			exit 1
		}
	}

	rm $hf

	setvar in_pwd = $(pwd)
	cd $yproot_dir
	echo  "Running \c"
	echo  $yproot_dir "\c"
	echo "/Makefile..."
	make NOPUSH=1

	if test $Status  -ne 0
	{
		echo "\
Error running Makefile."
		setvar errors_in_setup = 'T'
		
		if test $exit_on_error = T
		{
			exit 1
		}
	}

	cd $in_pwd
	echo ""
	echo  "\
$host has been set up as a yp master server\c"

	if test $errors_in_setup = T
	{
		echo " with errors.  Please remember"
		echo "to figure out what went wrong, and fix it."
	} else {
		echo " without any errors."
	}

	echo ""
	echo "\
If there are running slave yp servers, run yppush now for any data bases"
	echo "\
which have been changed.  If there are no running slaves, run ypinit on"
	echo "\
those hosts which are to be slave servers."

	# enable master services
	enable_this_boot network/nis/server:default
	enable_this_boot network/nis/xfr:default
	enable_this_boot network/nis/passwd:default
	enable_this_boot network/nis/update:default

	enable_this_boot network/nis/client:default
}