#!/usr/bin/perl

# Launch ms rdp client with ssh tunnel
# S. Radford

# $version = "  Version: 0.0  2013-08";
# $version = "  Version: 0.1  2013-13";
# $version = "  Version: 0.2  2014-07-02";
# $version = "  Version: 0.3  2014-07-03";
# $version = "  Version: 0.3.1  2014-07-30";
# $version = "  Version: 0.3.2  2014-08-26";
$version = "  Version: 0.4  2014-09-29";

# 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";               # sleep doesn't exist on win hosts

$width  = 1024;
$height =  768;
$geometry = $width."x".$height;
$diskmode = 0;
$diskpath="";

# application versions and paths
$rdpport=3389;
$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";

while (substr($ARGV[0],0,1) eq "-") {
  $opt=shift(@ARGV);
  if ($opt eq "-b" || $opt eq "--debug") {
    $debug=1;
  }
  elsif ($opt eq "-C" || $opt eq "--comp") {
    $options .= "-C ";
  }
  elsif ($opt eq "-d" || $opt eq "--disk") {
    $diskmode=shift(@ARGV);
    if ($debug) { print "diskmode: " . $diskmode . "\n"; }
      if ($debug) { print "diskpath: " . $diskpath . "\n"; }
    if (not $diskmode ~~ [qw(0 1 2 3 )] )  { 
      $diskpath = $diskmode ;
      $diskmode = 4;
      if ($debug) { print "diskpath: " . $diskpath . "\n"; }
    }
    if ($debug) { print "diskmode: " . $diskmode . "\n"; }
  }
  elsif ($opt eq "-h" || $opt eq "--help") {
    &usage;
    exit(0);
  }
  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 "-l" || $opt eq "--login") {
    $user=shift(@ARGV);
    if (index($user,"\@") == 0) {
      $ssh_host=substr($user,1);
      $user="";
    } elsif (index($user,"@") > 0) {
      ($user,$ssh_host)=split('@',$user,2);
      $user .= "@";
    } else {
      $user .= "@";
    }
  }
  elsif ($opt eq "-p" || $opt eq "--port") {
    $ssh_port=shift(@ARGV);
    $options .= "-p $ssh_port ";
  }
  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);
  }
}

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);
}

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

if ($url !~ /:\/\//) {
  print STDERR "ERROR: argument must be in URL form.\n";
  &usage;
  exit(0);
}

($service,$rest)=split(/:\/\//,$url,2);
($hostport,$path)=split('/',$rest);
($host,$port)=split(':',$hostport);
if (index($host,'@') > 0) {
  ($servuser,$host)=split('@',$host);
#  $servuser .= '@';
}

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; }

if ($service eq "rdp") {
  if ($port>0) { $remoteport=$port; }
  else { $remoteport=$rdpport; }
} else {
  print STDERR "Unknown (to this script) service: $service\n";
  exit(-1);
}

$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 a port to bind to!\n";
  exit(-1);
}

if ($host eq $ssh_host) { $host="localhost"; }

# Prepare temporary RDC configuration 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 directory $rdpconn\n"; }
  mkdir "$rdpconn" or die "Cannot create $rdpconn\n"; 
}  

if ($debug) { print STDERR "Creating RDP configuration $rdpconn$rdptmp\n"; }
  &configuration; 

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 (length($command) == 0 || $timeout >= 60) { print "Using $service://$servuser\@$localhost:$localport/\n"; }

# open configuration file, then 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
if (! $debug) { 
  unlink "$rdpconn$rdptmp"
    or warn "Could not unlink $rdpconn$rdptmp:\n $!";
}

#
# END OF MAIN
#

sub usage {

  print <<EEE;
Usage: $0 [options] rdp://[rdcuser@]target.computer
  Parameters:
    rdcuser               (optional) username on target
    target.computer       name or ip address of target computer
  Options:
    -C/--comp             Use ssh compression.
    -d/--disk {0,1,2,3, /path} Share local disks. v2 only.
                          0: no; 1: all disks; 2: home folder;
                          3: Documents folder; /path: specific path
    -h/--help             Show this message.
    -g/--geometry WxH     Geometry. Default is 1024x768.
    -l/--login [user]\@gw  Specifies the username and intermediate gateway
                          for the ssh tunnel. If username is not specified it uses 
                          the current login name.
    -p/--port             Specifies an alternate ssh port, if the ssh
			  server doesn't listen on the default port (22).
    -t/--timeout          Keeps the tunnel running for at least timeout
			  seconds.  Defaults to 30, to leave time for
			  connection 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. Default is v2.
EEE
  print $version, "\n";
}

sub configuration {

# create rdp configuration file for osx clients
# embed $localhost, $localport, $servuser, geometry, and disk mode

  open(OUF, ">", "$rdpconn$rdptmp") or die "Cannot open $rdpconn$rdptmp\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>
EEE

#	<string>rdphost:98765</string>
    print OUF "\t<string>" . $localhost . ":" . $localport ."</string> \n";

    print OUF <<EEE;
	<key>DesktopSize</key>
	<dict>
		<key>DesktopHeight</key>
EEE

#		<integer>768</integer>
    print OUF "\t\t<integer>" . $height . "</integer> \n";
    print OUF "\t\t<key>DesktopWidth</key> \n";
#		<integer>1024</integer>
    print OUF "\t\t<integer>" . $width . "</integer> \n";

    print OUF <<EEE;
	</dict>
	<key>Display</key>
	<integer>0</integer>
	<key>Domain</key>
EEE

#	<string>rdphost</string>
    print OUF "\t<string>" . $host ."</string> \n";

    print OUF <<EEE;
	<key>DontWarnOnChange</key>
	<false/>
	<key>DontWarnOnDriveMount</key>
	<false/>
	<key>DontWarnOnQuit</key>
	<false/>
	<key>DriveRedirectionMode</key>
EEE

#	<integer>0</integer>
    print OUF "\t<integer>" . $diskmode ."</integer> \n";

    print OUF <<EEE;
	<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>
EEE

#	<string></string> 
    print OUF "\t<string>" . $diskpath ."</string> \n";

    print OUF <<EEE;
	<key>RedirectPrinter</key>
	<string></string>
	<key>RemoteApplication</key>
	<false/>
	<key>Themes</key>
	<true/>
	<key>UserName</key>
EEE

#	<string>rdpuser</string>
    print OUF "\t<string>" . $servuser ."</string> \n";

    print OUF <<EEE;
	<key>Wallpaper</key>
	<false/>
	<key>WorkingDirectory</key>
	<string></string>
</dict>
</plist>
EEE

  } else {                          # MS RD v8


    print OUF "desktopwidth:i:" . $width . "\n";
    print OUF "desktopheight:i:" . $height . "\n";

    print OUF "full address:s:" . $localhost . ":" . $localport . "\n";
    print OUF "username:s:". $servuser  . "\n";

# 2014-09-29 
# although these parameters are documented, they don't seem to work
    print OUF "redirectdrives:i:" . $diskmode . "\n";
    print OUF "drivestoredirect:s:" . $diskpath . "\n";

    print OUF <<EEE;
screen mode id:i:1
use multimon:i:1
session bpp:i:24
audiomode:i:0
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) ;

}