#!/usr/bin/perl

#############################################################################
# smbmounts.pl: Setup Automounter to mount via SMB for federated experiments
# $Id: smbmount.Linux.pl,v 1.1 2008-05-18 00:09:31 faber Exp $
#

$SHARE=shift || "USERS";
$ADDR=shift || `cat /usr/local/etc/emulab/bossnode`;
$PUSER=shift || "jhickey";
$PNAME=shift || "emulab-ops";
$FSTYPE=shift || 'cifs';
$FSTAB="/etc/fstab";
$HOMEROOT="/users";
$PROJROOT="/proj";

my $share = 0;

chomp $ADDR;

# ADDR is probably a DNS name (and we assume so if it has letters.  mount cifs
# wants an IP address, so this hex dereferences the DNS name and converts the
# first address into the string representation of the IP.  It is, perhaps, a
# but terse.
if ($ADDR =~ /[a-z]/ ) {
    my @addrs = (gethostbyname($ADDR))[4];
    $ADDR = join(".", unpack("C4", @addrs[0]))
	if @addrs;
}


$TMCC="/usr/local/etc/emulab/tmcc";
$UMOUNT="/bin/umount";
$MOUNT="/bin/mount";
$MKDIR="/bin/mkdir";
$CP="/bin/cp";

print "Unmounting all nfs and cifs filesystems.\n";
system("$UMOUNT -a -f -t nfs,cifs"); # or die("Failed to unmount NFS");

# Backup the fstab so we can run multiple times
if (! -f "$FSTAB.bak") {
	system("$CP $FSTAB $FSTAB.bak") && die("Unable to backup $FSTAB");
} else {
	system("$CP $FSTAB.bak $FSTAB") && die("Unable to restore $FSTAB");
}

# open(TM, "$TMCC accounts windows |") or die("Failed to execute TMCC");
open(TM, "/usr/local/federation/etc/userconf")or die("Failed to open userconf");
open(FSTAB,">>$FSTAB") or die("Cannot Open File $FSTAB");

print FSTAB "\n# SMB Configuration Generated by smbmount.pl\n";

while(<TM>) {
    /ADDUSER/ && do {
	/LOGIN=(\S+)/ && do { $user = $1; };
	/WPSWD=(\S+)/ && do { $pswd = $1; };
	/UID=(\S+)/ && do { $uid = $1; };
	/GID=(\S+)/ && do { $gid = $1; };
	/HOMEDIR=(\S+)/ && do { $homedir = $1; };
	my $ids = $FSTYPE == 'smbfs' ? ",uid=$uid,gid=$gid" : "";

	#We should be even more careful about creating these files securely,
	#confirming ownership, etc., but this is run in an restricted
	#environment.
	open(PWDFILE, ">/tmp/$user.cifs_creds") || 
	    warn "Can't create credentials for $user:$!\n";
	chmod(0600, "/tmp/$user.cifs_creds") == 1 || 
	    warn "Credential file /tmp/$user.cifs_creds may have " .
		"bad permissions:$!\n";
	print PWDFILE "username=$user\npassword=$pswd\n";
	close(PWDFILE);

	print FSTAB "//$SHARE/$user\t";
	print FSTAB "$homedir\t$FSTYPE\t";
	print FSTAB "auto,rw,credentials=/tmp/$user.cifs_creds$ids,ip=$ADDR\t0\t0\n";

	#
	# If we are PUSER, the user assigned to mount the project
	# directory, then we add in the mount for that.  The share is
	# called proj-$PNAME where PNAME is the name of the project at
	# the mothership.  We need a second map for this since we will
	# be mounting under /proj.  We should really not use share in
	# federation.
	#

	if($user =~ m/$PUSER/i) {
	    print FSTAB "//$SHARE/proj-$PNAME\t";
	    print FSTAB "$PROJROOT/$PNAME\t$FSTYPE\t";
	    print FSTAB "rw,credentials=/tmp/$user.cifs_creds$ids,ip=$ADDR\t0\t0\n";
	    mkdir("/$PROJROOT/$PNAME") unless -d "/$PROJROOT/$PNAME";
	    if ( $share) {
		print FSTAB "//$SHARE/share\t";
		print FSTAB "/share\t$FSTYPE\t";
		print FSTAB "rw,credentials=/tmp/$user.cifs_creds$ids,ip=$ADDR\t0\t0\n";

		mkdir("/share") unless -d "/share";
	    }
	}
    }
}

close(FSTAB);

print("Mounting the $FSTYPE versions of everything\n");
system("$MOUNT -a -t $FSTYPE");