#!/usr/bin/perl
use strict;
use English;
use Getopt::Long;
use File::Basename;

my ($foo, $help, @computers, $file, $wd, $run, $nodel, $user, $pass, $script);
Getopt::Long::Configure ('bundling');
GetOptions ('h|help' => \$help, 'c|computer=s' => \@computers, 'f|file=s' => \$file, 'w|working_directory=s' => \$wd, 'r|run_command=s' => \$run, 'D' => \$nodel, 'u|user=s' => \$user, 'p|password=s' => \$pass,'<>' => \&doitall);

if ($help)
{
	print "\n";
	print "psScript.pl [-c computer [-c computer2 ...] | -f file][-w directory][-r command][-D] script\n";
	print "-c computer \t Computer Name or IP Address to run on\n";
	print "-f file \t Path of file containing muliple computer names or IP addresses (one per line)\n";
	print "-w directory \t working directory to copy script to\n";
	print "-r command \t Command to run (default is cmd.exe /c <script>)\n";
	print "-D \t Don't delete script after running\n";
	print "script \t Path of script to run remotely\n";
	exit(0);
}
	
sub doitall ()
{
	$script = shift;
	if(defined $file)
	{
		my $ip;
		open FILE, "<", $file or die $!;
		my @ip_list = <FILE>;
		close(FILE);
		foreach $ip (@ip_list)
		{
			doit($ip);
		}
	}
	elsif(defined @computers)
	{
		my $computer;
		foreach $computer (@computers)
		{
			unless( $computer =~ /^#/ )
			{
				print "Running on '$computer'\n";
				doit($computer);
			}
		}
	}
	else
	{
		print 'You need to specify a computer name or file!';
	}
	print 'Done.';
	return;
}


sub doit ()
{
	my $comp = shift;
	my $cmd;
	my ($script_filename, $script_dir, $script_suffix) = fileparse($script, qr/\.[^.]*/);
	my $script_basename = $script_filename . $script_suffix;
	
	
	$cmd = "net use \\\\$comp\\c\$ ";
	if(defined $user)
	{
		$cmd .= " \/user:$user";
		if(defined $pass)
		{
			$cmd .= " $pass";
		}
		else
		{
			$cmd .= " *";
		}
	}
	print "1) Setting up the share:\n\t" . $cmd . "\n";
	$foo = `$cmd`; # net use
	unless($CHILD_ERROR == 0){print "ERROR: net use failed.\n"; exit(1);}
	
	$cmd = "xcopy \/Z \/E \/H \/I \/-Y \"$script\" \\\\$comp\\c\$\\";
	if(defined $wd)
	{
		$cmd .= "$wd\\";
	}
	else
	{
		if($script_basename =~ /^(boot.ini|autoexec.bat|.*[.]sys|ntldr|ntdetect.com)$/i )
		{
			print "ERROR: Windows needs that file to boot! Aborting.\n";
			exit(1);
		}
	}
	print "2) Copying the script:\n\t" . $cmd . "\n\t";
	system($cmd); # xcopy
	unless($CHILD_ERROR == 0)
	{
		print "ERROR: Xcopy failed.\n";
		$cmd = "net use \\\\$comp\\c\$ /delete";
		print "Cleanup: Deleting the share:\n\t" . $cmd . "\n";
		system($cmd); # net use /delete
		exit(1);
	}
	
	$cmd = "psexec.exe \\\\$comp";
	if(defined $user)
	{
		$cmd .= " -u $user";
		if(defined $pass)
		{
			$cmd .= " -p $pass";
		}
	}
	if(defined $run)
	{
		$cmd .= " $run";
	}
	else
	{
		$cmd .= " cmd.exe \/c C:\\";
		if(defined $wd){ $cmd .= "$wd\\";}
		$cmd .= $script_basename; 
	}
	print "3) Running the script:\n\t" . $cmd . "\n\t";
	system($cmd); # psexec
	unless($CHILD_ERROR == 0){ print "ERROR: Psexec Failed.\n"; }
	
	unless(defined $nodel)
	{
		$cmd = "del \"\\\\$comp\\c\$\\";
		if(defined $wd){$cmd .= "$wd\\";}
		$cmd .= $script_basename . '"';
		print "Cleanup: Deleting the script:\n\t" . $cmd . "\n";
		system($cmd); # del
		unless($CHILD_ERROR == 0){ print "ERROR: Del Failed.\n"; }
	}
	
	$cmd = "net use \\\\$comp\\c\$ /delete";
	print "Cleanup: Deleting the share:\n\t" . $cmd . "\n";
	$foo = `$cmd`; # net use /delete
	unless($CHILD_ERROR == 0){ print "ERROR: net use \/delete Failed.\n" }
	return;
}