#!/bin/sh
# Copyright (c) 2003 Andy Barclay (Unixpeople)
#
# Permission is hereby granted, free of charge, to any person obtaining a
# copy of this software and associated documentation files (the "Software"),
# to deal in the Software without restriction, including without limitation
# the rights to use, copy, modify, merge, publish, distribute, sublicense,
# and/or sell copies of the Software, and to permit persons to whom the
# Software is furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
# IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
# CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
# TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#

# This cgi script will allow the user to update dns zone(s)

# ensure that /usr/local/bin is first on the path so we get the
# good version of nsupdate (sun machines....)
PATH=/usr/local/bin:$PATH

ZONES="unixpeople.internal
	vmprivate.com
	254.168.192.in-addr.arpa
	253.168.192.in-addr.arpa
	37.168.192.in-addr.arpa."

RRTYPES="A CNAME PTR NS MX TXT"
DEFTTL=86400
NEWAWK=/usr/xpg4/bin/awk

# choose format based on version of nsupdate
# this is an invalid option
nsupdate --version 2>/tmp/nsupdate.$$
if grep -- "-y" /tmp/nsupdate.$$ >/dev/null
then
	# its the newer version
	DNSKEY=dnskeys/dynupdates
else
	# its the old (sun stock) version
	DNSKEY=dnskeys:dynupdates
fi
rm /tmp/nsupdate.$$

# Need the new awk
if [ "`uname -s`" = "SunOS" ] && [ -x $NEWAWK ]
then
	AWK=$NEWAWK
else
	AWK=awk
fi

# this is used to build dynamic variables
selecteditems=1
exec 2>/tmp/errors

# source the parse function
. ./parse.sh

reverseIP()
{
	echo $1 |$AWK 'split($1,ip,".") {print ip[4] "." ip[3] "." ip[2] "." ip[1]}'
}

footer()
{
	echo "</BODY>"
	echo "</HTML>"
}

displayzone()
{
	if [ -z "$1" ]
	then
		echo "You didn't select a zone to view<BR>"
                echo "Go back and select an zone first"
                footer
                exit 1
	fi

	zone=$1
	rrtype=$2

        # print the top of the table
        cat <<EOT
<FORM method=POST action="/cgi-bin/dns.cgi">
<input type="hidden" name="zone" value="$zone">
<TABLE border="1" rules=all cellpadding=5>
<TR>
        <TH>Select</TH>
        <TH>Domain Name</TH>
        <TH>TTL</TH>
        <TH>Type</TH>
        <TH>Data</TH>
</TR>
EOT
	nameserver=`dig +short "$zone" in soa | cut -d " " -f 1`
	dig @"$nameserver" "$zone" in axfr |egrep -v "^;|^$" |
		sort -u >/tmp/zone.$$
	if [ -n "$rrtype" ]
	then
		cat /tmp/zone.$$  |
			$AWK -v rrtype=$rrtype '
				{if ($4==rrtype) {print}}' >/tmp/zone2.$$
		mv /tmp/zone2.$$ /tmp/zone.$$
	fi
	cat /tmp/zone.$$ |while read rrname ttl class rrtype rrdata
	do
		echo $rrdata >/tmp/rrdata$selecteditems.$$
        cat <<EOT
		<TR>
		<TD><INPUT type="checkbox" name="select$selecteditems" value="$rrname"></TD>
		<input type="hidden" name="rrname$selecteditems" value="$rrname">
		<input type="hidden" name="ttl$selecteditems" value="$ttl">
		<input type="hidden" name="class$selecteditems" value="$class">
		<input type="hidden" name="rrtype$selecteditems" value="$rrtype">
		<input type="hidden" name="rrdatafile$selecteditems" value="/tmp/rrdata$selecteditems.$$">
        	<TD>$rrname</TD>
        	<TD>$ttl</TD>
        	<TD>$rrtype</TD>
        	<TD>$rrdata</TD>
</TR>
EOT
	selecteditems=`expr $selecteditems + 1`
	done
        cat <<EOT
<FORM method=POST action="/cgi-bin/dns.cgi">
</TABLE>
<P>
<! input type="Submit" name=action value="Add a Host">
<! input type="Submit" name=action value="Add a Resource Record">
<input type="Submit" name=action value="Edit Selected Resource Record">
<input type="Submit" name=action value="Delete Selected Resource Records">
<input type="Submit" name=action value="Choose a different Zone">
<input type="reset" value="Reset Form">
</FORM>
EOT
	rm /tmp/zone.$$
}

echo "Content-type: text/html"
echo ""
echo "<HTML>"
echo "<BODY>"

#create all all the params as variables
parse

case "$action" in
"View Zone")
	# query the zone and display it in a table
	[ "$rrtype" = "ALL" ] && rrtype=""
	displayzone $zone $rrtype
	;;
"Add This Record"|"Update This Record")
	#loop through all the records, adding them
	eval rrname=\$rrname$selecteditems
	eval rrtype=\$rrtype$selecteditems
	eval ttl=\$ttl$selecteditems
	eval class=\$class$selecteditems
	# AWB - because of the problems with data containing quotes
	# check to see if rdata was passed or not. if not, then a filename
	# was passed
	eval rrdata=\"\$rrdata$selecteditems\"
	if [ -z "$rrdata" ]
	then
		eval rrdatafile=\$rrdatafile$selecteditems
		rrdata=`cat $rrdatafile`
	fi
	# the file will exist whether its selected or not
	rm $rrdatafile 2>/dev/null
	eval pref=\$pref$selecteditems
	eval addptr=\$addptr$selecteditems
	eval existing=\"\$existing$selecteditems\"
	eval zone=\$zone$selecteditems
        # shit. its possible that the rrname is null
        # its not possible for rrtype to be null
        while [ -n "$rrtype" ]
        do
		#if there is an existing record, delete it
		if [ -n "$existing" ]
		then
			echo "update delete $existing" >/tmp/dnsdel.$$
			echo "" >>/tmp/dnsdel.$$
			nsupdate -k $DNSKEY /tmp/dnsdel.$$
			rm /tmp/dnsdel.$$
		fi
        	#add the record
                if [ -z "$rrname" ]
                then
                   rr="$zone $ttl $class $rrtype $pref $rrdata"
                else
                   rr="$rrname.$zone $ttl $class $rrtype $pref $rrdata"
                fi
		echo "update add $rr" >/tmp/dnsadd.$$
		echo "" >>/tmp/dnsadd.$$
		nsupdate -k $DNSKEY /tmp/dnsadd.$$
		rm /tmp/dnsadd.$$
		# add the PTR if requested
		if [ "$addptr" = "True" ]
		then
			#if there is an existing record, delete it
			revip=`reverseIP $rrdata`
			if [ -n "$existing" ]
			then
				# its an edit, so the name will be fqdn
				oldrevip=`echo $existing |$AWK '{print $NF}'`
				oldrevip=`reverseIP $oldrevip`.in-addr.arpa.
				oldrrname=`echo $existing |$AWK '{print $1}'`
				rr="$oldrevip $class PTR $oldrrname"
				echo "update delete $rr" >/tmp/dnsadd.$$
				echo "" >>/tmp/dnsadd.$$
				nsupdate -k $DNSKEY /tmp/dnsadd.$$
				rm /tmp/dnsadd.$$
			fi
			echo "update add $revip.in-addr.arpa. $ttl $class PTR $rrname.$zone." >/tmp/dnsadd.$$
			echo "" >>/tmp/dnsadd.$$
			nsupdate -k $DNSKEY /tmp/dnsadd.$$
			rm /tmp/dnsadd.$$
		fi
                selecteditems=`expr $selecteditems + 1`
                eval rrname=\$rrname$selecteditems
		eval rrtype=\$rrtype$selecteditems
		eval ttl=\$ttl$selecteditems
		eval class=\$class$selecteditems
		eval rrdata=\"\$rrdata$selecteditems\"
		eval zone=\$zone$selecteditems
		if [ -z "$rrdata" ]
		then
			eval rrdatafile=\$rrdatafile$selecteditems
			rrdata=`cat $rrdatafile`
		fi
		# the file will exist regardless of whether its selected
		rm $rrdatafile 2>/dev/null
		eval pref=\$pref$selecteditems
		eval addptr=\$addptr$selecteditems
		eval existing=\"\$existing$selecteditems\"
        done
	echo "done!"
	echo '<P><A HREF="/cgi-bin/dns.cgi">Return to zone list</A>'
	;;
"Add a Resource Record")
        if [ "$rrtype" = "ALL" ]
        then
                echo "You can't <B>add</B> an RR type of ALL<BR>"
                echo "Go back and choose a valid RR type"
                footer
                exit 1
        fi
	
	cat <<EOT
<FORM method=POST action="/cgi-bin/dns.cgi">
<TABLE border="1" rules=all cellpadding=5>
        <TH>Domain Name</TH>
        <TH>TTL</TH>
        <TH>Class</TH>
	<TH>Type</TH>
EOT
	case $rrtype in
	A)
        	echo '<TH>IP Address</TH>'
		;;
	CNAME|PTR)
        	echo '<TH>Canonical Name</TH>'
		;;
	NS)
        	echo '<TH>Name Server</TH>'
		;;
	MX)
        	echo '<TH>Preference</TH>'
        	echo '<TH>Mail Server</TH>'
		;;
	TXT)
        	echo '<TH>Character String</TH>'
		;;
	esac
	cat <<EOT
<TR><TD><input type="hidden" name=zone1 value="$zone">
<input type="TEXT" name=rrname1 value="">.$zone
<TD><input type="TEXT" name=ttl1 value=$DEFTTL>
<input type="hidden" name=class1 value="IN">
<TD>IN
<input type="hidden" name=rrtype1 value="$rrtype">
<TD>$rrtype
EOT
	case $rrtype in
	A|CNAME|PTR|NS|TXT)
		echo '<TD><input type="TEXT" name=rrdata1 value="" SIZE=25></TD>'
		;;
	MX)
		echo '<TD><input type="TEXT" name=pref1 value=""></TD>'
		echo '<TD><input type="TEXT" name=rrdata1 value="" SIZE=25></TD>'
		;;
*)
	echo "WARNING: unhandled RR type $rrtype"
	footer
	exit 1
	;;
esac
	cat <<EOT
</TR></TABLE>
EOT
if [ "$rrtype" = "A" ]
then
	echo '<BR>'
	echo '<INPUT type="checkbox" name="addptr1" value="True" checked>'
	echo 'Select this box to create corresponding PTR Record as well'
fi
cat <<EOT
<P>
<input type="Submit" name=action value="Add This Record">
<input type="reset" value="Reset Form">
</FORM>
EOT
	;;
"Edit Selected Resource Record")
	#loop though the rrs until the end (null)
        eval rrname=\$rrname$selecteditems
        while [ -n "$rrname" ]
        do
                eval selected=\$select$selecteditems
                if [ -n "$selected" ]
                then
                        # found a selected rrname
                        break
                fi
                selecteditems=`expr $selecteditems + 1`
                eval rrname=\$rrname$selecteditems
        done
        if [ -z "$selected" ]
        then
                echo "You didn't select an RR to edit<BR>"
                echo "Go back and select an RR first"
                footer
                exit 1
        fi
        eval ttl=\$ttl$selecteditems
        eval class=\$class$selecteditems
        eval rrtype=\$rrtype$selecteditems
	eval rrdata=\$rrdata$selecteditems
	if [ -z "$rrdata" ]
	then
		eval rrdatafile=\$rrdatafile$selecteditems
		rrdata=`cat $rrdatafile`
	fi
	# the file will exist
	rm $rrdatafile 2>/dev/null
	# if its an MX, then we need to get the $pref
	if [ "$rrtype" = "MX" ]
	then
		pref=`echo $rrdata|$AWK '{print $1}'`
		rrdata=`echo $rrdata|sed -e "s/^.* //"`
	fi
	# display a form
	# AWB - changed from double quotes to single quotes
	# Thu Sep  5 11:27:28 PDT 2002
        cat <<EOT
        <FORM method=POST action="/cgi-bin/dns.cgi">
        <input type="hidden" name="existing1" value='$rrname $ttl $class $rrtype $rrdata'>

<TABLE border="1" rules=all cellpadding=5>
        <TH>Domain Name</TH>
        <TH>TTL</TH>
        <TH>Class</TH>
	<TH>Type</TH>
EOT
	case $rrtype in
	A)
        	echo '<TH>IP Address</TH>'
		;;
	CNAME|PTR)
        	echo '<TH>Canonical Name</TH>'
		;;
	NS)
        	echo '<TH>Name Server</TH>'
		;;
	MX)
        	echo '<TH>Preference</TH>'
        	echo '<TH>Mail Server</TH>'
		;;
	TXT)
        	echo '<TH>Character String</TH>'
		;;
	esac

	zone=`echo $rrname |sed -e "s/^[^.]*.//"`
	rrname=`echo $rrname|sed -e "s/.$zone$//"`
	cat <<EOT
<TR>
<TD><input type="TEXT" name=rrname1 value="$rrname">
$zone</TD>
<TD><input type="TEXT" name=ttl1 value=$ttl></TD>
<input type="hidden" name=class1 value="$class">
<TD>IN</TD>
<input type="hidden" name=rrtype1 value="$rrtype">
<input type="hidden" name=zone1 value="$zone">
<TD>$rrtype</TD>
EOT
	case $rrtype in
	A|CNAME|PTR|NS|TXT)
		# AWB - changed from double to single quotes
		# Thu Sep  5 11:29:07 PDT 2002
		cat <<EOT
<TD><input type="TEXT" name=rrdata1 value='$rrdata' SIZE=35></TD>
EOT
		;;
	MX)
		cat <<EOT
<TD><input type="TEXT" name=pref1 value="$pref"></TD>
<TD><input type="TEXT" name=rrdata1 value="$rrdata" SIZE=25></TD>
EOT
		;;
	*)
		echo "WARNING: unhandled RR type $rrtype"
		footer
		exit 1
		;;
	esac
	cat <<EOT
</TR></TABLE>
EOT
	if [ "$rrtype" = "A" ]
	then
	cat <<EOT
<BR>
<INPUT type="checkbox" name="addptr$selecteditems" value="True" checked>
Select this box to edit the corresponding PTR Record too
EOT
	fi
	cat <<EOT
	<P>
        <input type="Submit" name="action" value="Update This Record">
        <input type="reset" value="Reset Form">
        </FORM>
EOT
	;;
"Delete Selected Resource Records")
	#loop though the rrs until the end (null)
        eval rrname=\$rrname$selecteditems
        eval class=\$class$selecteditems
        eval rrtype=\$rrtype$selecteditems
	eval rrdata=\"\$rrdata$selecteditems\"
	if [ -z "$rrdata" ]
	then
		eval rrdatafile=\$rrdatafile$selecteditems
		rrdata=`cat $rrdatafile`
	fi
	# the file will exist whether its selected or not
	rm $rrdatafile 2>/dev/null
        eval selected=\$select$selecteditems
        while [ -n "$rrname" ]
        do
                if [ -n "$selected" ]
                then
			#echo select is $selected for item $rrname >>/tmp/foo
                        # found a selected rrname
			found=true
        		#delete the record
			#echo "update delete $rrname $class $rrtype $rrdata" >>/tmp/foo
			echo "update delete $rrname $class $rrtype $rrdata" \
			>/tmp/dnsadd.$$
			echo "" >>/tmp/dnsadd.$$
			nsupdate -k $DNSKEY /tmp/dnsadd.$$
			rm /tmp/dnsadd.$$
			# if its an A RR, then delete the PTR too
			if [ "$rrtype" = "A" ]
			then
				revip=`reverseIP $rrdata`
				echo "update delete $revip.in-addr.arpa. $class PTR $rrname" >/tmp/dnsadd.$$
				echo "" >>/tmp/dnsadd.$$
				nsupdate -k $DNSKEY /tmp/dnsadd.$$
				rm /tmp/dnsadd.$$
			fi
                fi
                selecteditems=`expr $selecteditems + 1`
        	eval rrname=\$rrname$selecteditems
        	eval class=\$class$selecteditems
        	eval rrtype=\$rrtype$selecteditems
		eval rrdata=\"\$rrdata$selecteditems\"
		if [ -z "$rrdata" ]
		then
			eval rrdatafile=\$rrdatafile$selecteditems
			rrdata=`cat $rrdatafile`
		fi
		# the file will exist whether its selected or not
		rm $rrdatafile 2>/dev/null
        		eval selected=\$select$selecteditems
        done
        if [ -z "$found" ]
        then
                echo "You didn't select an RR to delete<BR>"
                echo "Go back and select an RR first"
                footer
                exit 1
        fi
	echo "done!"
	echo '<P><A HREF="/cgi-bin/dns.cgi">Return to zone list</A>'
	;;
*)
	# build the zones html
	for i in $ZONES
	do
		zones="$zones <OPTION> $i"
	done
	# choose a domain
	cat <<EOT
<FORM method=POST action="/cgi-bin/dns.cgi">
Choose a zone to operate on:
<BR>
(Optionally limit the Records to the chosen type)
<P>
<SELECT NAME="zone" SIZE=10>
$zones
</SELECT>
<SELECT NAME=rrtype SIZE=10>
<OPTION SELECTED>ALL
EOT
	for i in $RRTYPES
	do
		echo "<OPTION>$i"
	done
	cat <<EOT
</SELECT>
<P>
<input type="Submit" name=action value="View Zone">
<! input type="Submit" name=action value="Add a Host">
<input type="Submit" name=action value="Add a Resource Record">
<input type="reset" value="Reset Form">
</FORM>
EOT
	;;
esac

footer
exit 0