#!/usr/bin/perl

# Universal script to launch rdp client with ssh tunnel to CSO 

# supported systems 
# Mac OSX: MS remote desktop v2 or v8
# linux:   rdesktop

# $version = "  Version: 0.0  2014-08-26";
$version = "  Version: 0.1  2014-09-26";

# S. Radford

# Portions adapted from tunnelopen.pl
# Copyright (c) 2008 Thomas A. Fine
# Redistribution is permitted, in whole or in part, for commercial and
# non-commercial purposes, provided only that this copyright notice
# remains.

use Socket;

$timeout=30;
$startport=12345;
$maxtries=50;
$options="";
$user="";
$localhost="localhost";
$command="sleep $timeout";

$service = "rdp";
$ssh_host = "kilauea.caltech.edu";
$host="localhost";

$rdpport=3389;

$width  = $templatewidth  = 1024;
$height = $templateheight =  768;
$geometry = $width."x".$height;

# determine os type, available remote desktop clients

$ostype=$^O;
if ($debug) { print STDERR $ostype, "\n"; }     # $debug not defined yet

if ($ostype eq "darwin") {

# application versions and paths
  $rdpver=2;
  $rdpapp2 = "/Applications/Remote Desktop Connection.app";
  $rdpapp8 = "/Applications/Microsoft Remote Desktop.app";

# template files
  $rdpconn="$ENV{HOME}/Documents/RDC Connections/";
  $rdptemplate = "template$$.rdp";
  $rdptmp="tmp$$.rdp";
  $templatehost="rdphost";
  $templateport="98765";
  $templateuser="rdpuser";

} elsif ($ostype eq "linux") {

  if (length (`which rdesktop 2>/dev/null`) > 0) {
    $rdpapp = "rdesktop";
  } else {
    print STDERR "No supported remote desktop client available \n";
    exit(-1);
  }
  if ($debug) { print STDERR $rdpapp, "\n"; }

} else {
  print STDERR "Unsupported system ", $ostype, "\n";
  exit(-1);
}

# parse options

while (substr($ARGV[0],0,1) eq "-") {
  $opt=shift(@ARGV);

  if ($opt eq "-l" || $opt eq "--login") {
    $user = shift(@ARGV) . "@";
  }
  elsif ($opt eq "-ll" ) {
    $user= getpwuid($<) . "@";
  }
  elsif ($opt eq "-b" || $opt eq "--debug") {
    $debug=1;
  }
  elsif ($opt eq "-t" || $opt eq "--timeout") {
    if (length($command) == 0) {
      &usage;
      exit(-1);
    }
    $timeout=shift(@ARGV);
    if ($timeout !~ /^[0-9]+[hmd]?$/) {
      &usage;
      exit(-1);
    }
    if ($timeout =~ /[hmd]$/) {
      $u=substr($timeout,length($timeout)-1,1);
      $timeout=substr($timeout,0,length($timeout)-1);
      if ($u eq "m") { $timeout *= 60; }
      if ($u eq "h") { $timeout *= 3600; }
      if ($u eq "d") { $timeout *= 86400; }
    }
    $command="sleep $timeout";
    ++$customtimeout;
  }
  elsif ($opt eq "-v") {
    $rdpver=shift(@ARGV);
  }
  elsif ($opt eq "-g" || $opt eq "--geometry") {
    $geometry=shift(@ARGV);
    if (index(lc($geometry),"x") > 0) {
      ($width, $height) = split('x', lc($geometry), 2) ;
    } else {
      $width = $geometry ;
    }
    $geometry = $width."x".$height;
    if ($width !~ /^[0-9]/ || $height !~ /^[0-9]/) {
      &usage;
      exit(-1);
    }
  }
  elsif ($opt eq "-C" || $opt eq "--comp") {
    $options .= "-C ";
  }
  elsif ($opt eq "-h" || $opt eq "--help") {
    &usage;
    exit(0);
  }
}

if ($ostype eq "darwin") {

  die ( "Unknown (to this script) RDC version\n" )
    unless ($rdpver eq 2 || $rdpver eq 8) ;

  if ($rdpver == 2) { unless (-e $rdpapp2) { $rdpver=8; } }

  if ($rdpver == 2) {
    die "Application not found: $rdpapp2\n" unless (-e $rdpapp2) ;
    $rdpapp = $rdpapp2 ; 
  } elsif ($rdpver == 8) {
    die "Application not found: $rdpapp8\n" unless (-e $rdpapp8) ;
    $rdpapp = $rdpapp8 ; 
  }
}

if ($debug) { print STDERR "app: $rdpapp \n"; }

if ($#ARGV == -1 || $#ARGV > 0) {
  &usage;
  exit(0);
}

if ($user eq "") { $user= getpwuid($<) . "@"; }

$target=shift(@ARGV);
if ($debug) { print STDERR "target: ", $target, "\n"; }

# parse target to extract username, ip address
($host,$port)=split(':',$target);
if (index($host,'@') > 0) {
  ($servuser,$host)=split('@',$host);
#  $servuser .= '@';
}

if ($port>0) { $remoteport=$port; }
else { $remoteport=$rdpport; }

if ($debug) { print STDERR "host: ", $host, "\n"; }
if ($debug) { print STDERR "port: ", $port, "\n"; }
if ($debug) { print STDERR "servuser: ", $servuser, "\n"; }

# if (length($ssh_host) == 0) { $ssh_host=$host; }

$proto = getprotobyname('tcp');
if (!socket(S,PF_INET,SOCK_STREAM,$proto)) {
  print STDERR "Can't create socket: $!\n";
  exit(-1);
}

for ($i=$startport; $i<$startport+$maxtries; ++$i) {
  $here=sockaddr_in($i,inet_aton($localhost));
  if (bind(S, $here)) {
    if ($debug) { print STDERR "using local port $i\n"; }
    $localport=$i;
    #this shouldn't cause TIME_WAIT, because it isn't connected or listening
    close(S);
    last;
  }
}
if ($localport == 0) {
  print STDERR "Can't find an available intermediate port!\n";
  exit(-1);
}

if ($ostype eq "darwin") {

# Prepare temporary RDC parameter file from template
# create directory and templates if necessary

  if ($debug) { print STDERR "template: $rdpconn$rdptemplate\n", 
    "tmp: $rdpconn$rdptmp\n"; }

  unless (-d "$rdpconn") {
    if ($debug) { print STDERR "Creating RDP template $rdpconn\n"; }
    mkdir "$rdpconn" or die "Cannot create $rdpconn\n"; 
  }  

  unless (-e "$rdpconn$rdptemplate") {
    if ($debug) { print STDERR "Creating RDP template $rdpconn$rdptemplate\n"; }
    &template;
  }  

  open(INF, "<", "$rdpconn$rdptemplate") or die "Cannot open $rdpconn$rdptemplate\n $!";
  open(OUF, ">", "$rdpconn$rdptmp")      or die "Cannot open $rdpconn$rdptmp\n $!";

  # copy template to configuration file
  # substitute $localhost, $localport, $servuser, and geometry for placeholders
  while (<INF>) {
    s/$templatehost/$localhost/g;
    s/$templateport/$localport/g;
    s/$templateuser/$servuser/g;
    s/$templatewidth/$width/g;
    s/$templateheight/$height/g;
    print OUF;
  }
  close(OUF);
  close(IUF);

  # remove template
  unlink "$rdpconn$rdptemplate"
    or warn "Could not unlink $rdpconn$rdptemplate:\n $!";

}

# open ssh tunnel

if ($debug) { print STDERR 
  "ssh -L $localport:$host:$remoteport -f $options $user$ssh_host $command\n"; }
print "ssh $user$ssh_host\n";
system("ssh -L $localport:$host:$remoteport -f $options $user$ssh_host $command");

if ($ostype eq "darwin") {

# open configuration file, system opens rdp app 
  $rdpapp  =~ s/ /\\ /g ;               # escape embedded spaces
  $rdpconn =~ s/ /\\ /g ;               # escape embedded spaces
  if ($debug) { print STDERR "open -a $rdpapp $rdpconn$rdptmp \n"; }
  system("open -a $rdpapp $rdpconn$rdptmp ");

  sleep 5;                               # wait for app to start
                                         # should test process id, etc.

# remove configuration file
  $rdpconn =~ s/\\ / /g ;                # un-escape embedded spaces
  unlink "$rdpconn$rdptmp"
    or warn "Could not unlink $rdpconn$rdptmp:\n $!";

} elsif ($ostype eq "linux") {

  if ($servuser ne "") {
    if ($debug) { print STDERR 
      "$rdpapp -g $geometry -u $servuser $localhost:$localport & \n"; }
    system("$rdpapp -g $geometry -u $servuser $localhost:$localport &");
  } else {
    if ($debug) { print STDERR 
      "$rdpapp -g $geometry $localhost:$localport & \n"; }
    system("$rdpapp -g $geometry $localhost:$localport &");
  }

} else {
  print STDERR "Unsupported system ", $ostype, "\n";
  &usage;
  exit(-1);
}

#
# END OF MAIN
#

sub usage {
  print <<EEE;
Usage: $0 [options] [rdcuser@]target.computer
  Parameters:
    rdcuser             (optional) username on target
    target.computer     name or ip address of target computer
  Options are:
  -C/--comp             Use ssh compression.
  -h/--help             Show this message.
  -g/--geometry WxH     Geometry. Default is 1024x768.
                          On linux, do not exceed native geometry of target.
  -l/--login user       Specifies the login username for the intermediate gateway.
			  If username is not specified it uses the current login name.
  -t/--timeout          Keeps the ssh tunnel running for at least timeout
			  seconds. Defaults to 30, to leave time for VNC
			  viewer setup. Can be set to longer values
			  but will automatically exit eventually. May also be 
                          specified in minutes, hours or days, by adding an m, h, or d
			  at the end of the number, e.g. "-t 3h"
  -v {2,8}              Version of MS RDC client. OSX only. Default is v2
EEE
  print $version, "\n";
}

sub template {

# create rdp templates for Mac OSX rdp clients

open(OUF, ">$rdpconn$rdptemplate") or die "Cannot open $rdpconn$rdptemplate\n $!";

  if ($rdpver == 2) {               # RDC v2
    print OUF <<EEE;
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
<plist version="1.0">
<dict>
	<key>AddToKeychain</key>
	<true/>
	<key>ApplicationPath</key>
	<string></string>
	<key>AudioRedirectionMode</key>
	<integer>0</integer>
	<key>AuthenticateLevel</key>
	<integer>0</integer>
	<key>AutoReconnect</key>
	<true/>
	<key>BitmapCaching</key>
	<true/>
	<key>ColorDepth</key>
	<integer>2</integer>
	<key>ConnectionString</key>
	<string>rdphost:98765</string>
	<key>DesktopSize</key>
	<dict>
		<key>DesktopHeight</key>
		<integer>768</integer>
		<key>DesktopWidth</key>
		<integer>1024</integer>
	</dict>
	<key>Display</key>
	<integer>0</integer>
	<key>Domain</key>
	<string>remotehost</string>
	<key>DontWarnOnChange</key>
	<false/>
	<key>DontWarnOnDriveMount</key>
	<false/>
	<key>DontWarnOnQuit</key>
	<false/>
	<key>DriveRedirectionMode</key>
	<integer>0</integer>
	<key>FontSmoothing</key>
	<true/>
	<key>FullWindowDrag</key>
	<false/>
	<key>HideMacDock</key>
	<false/>
	<key>KeyMappingTable</key>
	<dict>
		<key>UI_ALPHANUMERIC_KEY</key>
		<dict>
			<key>MacKeyCode</key>
			<integer>102</integer>
			<key>MacModifier</key>
			<integer>0</integer>
			<key>On</key>
			<true/>
		</dict>
		<key>UI_ALT_KEY</key>
		<dict>
			<key>MacKeyCode</key>
			<integer>4294967295</integer>
			<key>MacModifier</key>
			<integer>2048</integer>
			<key>On</key>
			<true/>
		</dict>
		<key>UI_CONTEXT_MENU_KEY</key>
		<dict>
			<key>MacKeyCode</key>
			<integer>120</integer>
			<key>MacModifier</key>
			<integer>2048</integer>
			<key>On</key>
			<true/>
		</dict>
		<key>UI_CONVERSION_KEY</key>
		<dict>
			<key>MacKeyCode</key>
			<integer>4294967295</integer>
			<key>MacModifier</key>
			<integer>0</integer>
			<key>On</key>
			<false/>
		</dict>
		<key>UI_HALF_FULL_WIDTH_KEY</key>
		<dict>
			<key>MacKeyCode</key>
			<integer>49</integer>
			<key>MacModifier</key>
			<integer>256</integer>
			<key>On</key>
			<true/>
		</dict>
		<key>UI_HIRAGANA_KEY</key>
		<dict>
			<key>MacKeyCode</key>
			<integer>104</integer>
			<key>MacModifier</key>
			<integer>0</integer>
			<key>On</key>
			<true/>
		</dict>
		<key>UI_NON_CONVERSION_KEY</key>
		<dict>
			<key>MacKeyCode</key>
			<integer>4294967295</integer>
			<key>MacModifier</key>
			<integer>0</integer>
			<key>On</key>
			<false/>
		</dict>
		<key>UI_NUM_LOCK_KEY</key>
		<dict>
			<key>MacKeyCode</key>
			<integer>71</integer>
			<key>MacModifier</key>
			<integer>0</integer>
			<key>On</key>
			<true/>
		</dict>
		<key>UI_PAUSE_BREAK_KEY</key>
		<dict>
			<key>MacKeyCode</key>
			<integer>99</integer>
			<key>MacModifier</key>
			<integer>2048</integer>
			<key>On</key>
			<true/>
		</dict>
		<key>UI_PRINT_SCREEN_KEY</key>
		<dict>
			<key>MacKeyCode</key>
			<integer>118</integer>
			<key>MacModifier</key>
			<integer>2048</integer>
			<key>On</key>
			<true/>
		</dict>
		<key>UI_SCROLL_LOCK_KEY</key>
		<dict>
			<key>MacKeyCode</key>
			<integer>107</integer>
			<key>MacModifier</key>
			<integer>0</integer>
			<key>On</key>
			<true/>
		</dict>
		<key>UI_SECONDARY_MOUSE_BUTTON</key>
		<dict>
			<key>MacKeyCode</key>
			<integer>256</integer>
			<key>MacModifier</key>
			<integer>4608</integer>
			<key>On</key>
			<true/>
		</dict>
		<key>UI_WINDOWS_START_KEY</key>
		<dict>
			<key>MacKeyCode</key>
			<integer>122</integer>
			<key>MacModifier</key>
			<integer>2048</integer>
			<key>On</key>
			<true/>
		</dict>
	</dict>
	<key>MenuAnimations</key>
	<false/>
	<key>PrinterRedirection</key>
	<true/>
	<key>RedirectFolder</key>
	<string></string>
	<key>RedirectPrinter</key>
	<string></string>
	<key>RemoteApplication</key>
	<false/>
	<key>Themes</key>
	<true/>
	<key>UserName</key>
	<string>rdpuser</string>
	<key>Wallpaper</key>
	<false/>
	<key>WorkingDirectory</key>
	<string></string>
</dict>
</plist>
EEE
  } else {                          # MS RD v8
    print OUF <<EEE;
screen mode id:i:1
desktopwidth:i:1024
desktopheight:i:768
use multimon:i:1
session bpp:i:24
full address:s:rdphost:98765
audiomode:i:0
username:s:rdpuser
disable wallpaper:i:0
disable full window drag:i:0
disable menu anims:i:0
disable themes:i:0
alternate shell:s:
shell working directory:s:
authentication level:i:2
connect to console:i:0
gatewayusagemethod:i:0
disable cursor setting:i:0
allow font smoothing:i:1
allow desktop composition:i:1
bookmarktype:i:3
use redirection server name:i:0
EEE
  }
  close (OUF) ;

}