Putty Session Generator

Category:	Win32 Stuff
Author/Contact Info	Daniel Schrock dschrock@gmail.com
Description:	This script will parse a config file and generate registry keys and shortcuts that you can add to your taskbar for easy access to saved putty sessions. Any and all comments/enhancements are welcome. I use this site extensively, so I figured its time I give something back. Although the script isn't perfect, it works well for me and can be easily modified to accomodate your personal setup. Enjoy.
The code: #!/usr/bin/perl -w # written by Daniel Schrock # # ver. 0.1 - 03/17/04 - initial testing and dev. # ver. 0.2 - 04/23/04 - First working version # # Requirements: # -ActiveState Perl 5.8.2 or higher # -Win32::TieRegistry module (use ppm or CPAN) # -Win32::Shortcut (use ppm or CPAN) use strict 'vars'; my $Registry; use Win32::TieRegistry( TiedRef => \$Registry, Delimiter=>"/", qw( REG +_SZ REG_EXPAND_SZ REG_DWORD REG_BINARY REG_MULTI_SZ KEY_READ KEY_WRIT +E KEY_ALL_ACCESS ),); use Win32::Shortcut; #my $print_output="notTRUE"; my $print_output="TRUE"; my $ignore='^\s#\|^\s\n\r'; my $base_dir= "c:/NoInstall"; my $user; my $netuser= "admin"; my $sshuser= $ARGV[0]\|\|""; my $tmpuser; my $server_list= "$base_dir/servers.txt"; my $folder; my $protocol; my $protover; my $version; my $session_name; my $server_name; my $app="putty.exe"; my $appdir="$base_dir/$app"; my $dest_dir="$base_dir/putty-sessions"; my $LINK; my $RegPath= "CUser/Software/SimonTatham/PuTTY/Sessions/"; my $hexport; my $protoport; my $compr; my $protocompr; &check_putty_exe; &check_putty_session_dir; open(SERVER_LIST,$server_list)\|\|die "Can't open $server_list: $!\n"; while (<SERVER_LIST>) { next if /$ignore/ ; chop; ( $folder, $server_name, $session_name, $protocol)= split (":"); ($hexport,$version,$compr)=create_port($protocol); $user=create_user($folder); &create_session($server_name, $session_name, $protocol, $user); &create_link($folder, $server_name, $session_name); } close(SERVER_LIST); exit; sub check_putty_exe { if (! -f "$appdir" ) { print "$appdir not found!\n\n Unable to continue!\n\n Please mov +e/copy $app into $base_dir and rerun this script\n"; exit 1; }else{ print "Excellent! Found $app in $base_dir\n"; } } sub check_putty_session_dir { if (! -d "$dest_dir" ) { print "$dest_dir not found.\nCreating $dest_dir\n"; mkdir $dest_dir \|\| die "Error! Can not create $dest_dir"; }else{ print "$dest_dir found.\nCreating a backup\n"; if ( -d "$dest_dir.bak" ) { print "Found previous backup.\nPlease remove and rerun the + script.\n"; exit 1; } rename $dest_dir, $dest_dir.'.bak' \|\| die "Couldn't backup $de +st_dir\n"; print "Backup created.\n$ $dest_dir.bak $\nCreating new sess +ion directory\n\n"; mkdir $dest_dir \|\| die "Error! Can not create $dest_dir"; print "New session directory created.\n$ $dest_dir $\n"; } } sub create_port { if ( $protocol eq "telnet" ) { $protoport= "0x00017"; $protover= "0x00003"; $protocompr= "0x00000"; }elsif ( $protocol eq "ssh2" ) { $protoport= "0x00016"; $protover= "0x00003"; $protocompr= "0x00001"; }elsif ( $protocol eq "ssh1" ) { $protoport= "0x00016"; $protover= "0x00002"; $protocompr= "0x00000"; } return ($protoport, $protover, $protocompr); } sub create_user { if ( $folder eq "network" \|\| $folder eq "frame" \|\| $folder eq "s2s +-vpn" ) { $tmpuser= $netuser; }else{ $tmpuser= $sshuser; } return($tmpuser); } sub create_link { if (! -d "$dest_dir/$folder" ) { mkdir "$dest_dir/$folder" \|\| die "Error! Can not create $dest_ +dir/$folder"; } $LINK=new Win32::Shortcut(); $LINK->{'Path'}="$appdir"; $LINK->{'Arguments'}="-load $session_name"; $LINK->{'WorkingDirectory'}="$base_dir"; $LINK->{'Description'}="$server_name"; $LINK->{'ShowCmd'}=SW_SHOWNORMAL; $LINK->{'IconLocation'}="$appdir"; $LINK->{'IconNumber'}=0; $LINK->Save("$dest_dir/$folder/$session_name.lnk"); $LINK->Close(); print "$dest_dir/$folder/$session_name.lnk created.\n" } sub create_session { $Registry->{"$RegPath"}= { "$session_name/" => { "/HostName" => "$server_name", "/Protocol" => "$protocol", "/PortNumber" => ["$hexport", REG_DWORD ], "/PingIntervalSecs" => [ "0x0001", REG_DWORD ], "/TCPNoDelay" => [ "0x0000", REG_DWORD ], "/UserName" => "$user", "/Compression" => [ "$compr", REG_DWORD ], "/Cipher" => "blowfish,aes,3des,WARN,des", "/SshProt" => [ "$version", REG_DWORD ], "/LinuxFunctionKeys" => [ "0x0002", REG_DWORD ], "/BLinkCur" => [ "0x0001", REG_DWORD ], "/ScrollbackLines" => [ "0x07d0", REG_DWORD ], }, } \|\| die "Can't create $RegPath/$session_name: $^E\n"; } The Config File:* ###CONSOLES consoles:server1-console.example.com:server1-console:telnet consoles:server2-console.example.com:server2-console:telnet consoles:server3-console.example.com:server3-console:telnet consoles:server4-console.example.com:server4-console:telnet consoles:router1-console.example.com:router1-console:telnet consoles:router2-console.example.com:router2-console:telnet ####CONSOLES ###DMZ dmz:ns01.example.com:ns01:ssh2 dmz:ns02.example.com:ns02:ssh2 dmz:web01.example.com:web01:ssh2 dmz:web02.example.com:web02:ssh2 ####DMZ ###FRAME frame:framerelay1.example.com:framerelay1:telnet frame:framerelay2.example.com:framerelay2:telnet ####FRAME ###INSIDE inside:server1.example.com:server1:ssh2 inside:server2.example.com:server2:ssh2 inside:server3.example.com:server3:ssh2 inside:server4.example.com:server4:ssh2 ####INSIDE ###NETWORK network:switch1.example.com:switch1:telnet network:switch2.example.com:switch2:telnet network:switch3.example.com:switch3:telnet network:router1.example.com:router1:ssh2 network:router2.example.com:router2:ssh2 network:bigip01.example.com:bigip01:ssh2 network:bigip02.example.com:bigip02:ssh2 network:pix525.example.com:pix525:ssh1 network:console-server.example.com:console-server:ssh1 ####NETWORK ###VPN s2s-vpn:pix506.example.com:pix506:ssh1 ####VPN ###ROUTE SERVERS route-servers:route-server.ip.att.net:route-server.ip.att.net:telnet route-servers:route-server.cerf.net:route-server.cerf.net:telnet route-servers:route-server.colt.net:route-server.colt.net:telnet route-servers:route-server.exodus.net:route-server.exodus.net:telnet route-servers:route-server.gblx.net:route-server.gblx.net:telnet route-servers:route-views.oregon-ix.net:route-views.oregon-ix.net:teln +et ####ROUTE SERVERS ###PERSONAL personal:192.168.0.1:pix:ssh1 personal:192.168.0.2:2924:telnet personal:192.168.0.200:server1:ssh2 personal:192.168.0.254:server2:ssh2 personal:192.168.0.253:server3:ssh2 ####PERSONAL After running, just create a new toolbar that points to the putty-sessions folder.

Comment on Putty Session Generator Select or Download Code

Replies are listed 'Best First'.
Re: Putty Session Generator by clscott (Friar) on Oct 21, 2004 at 17:14 UTC
Very cool ++; I think I will find this useful at work. Thanks for taking the the time to decode the Registry I don't understand what you mean by "create a new toolbar." A new toolbar in what application? -- Clayton	[reply]
Re^2: Putty Session Generator by d_jabsd (Acolyte) on Oct 21, 2004 at 20:03 UTC
in Windows XP/2k3... might work in 2k as well... just right click your taskbar, select Toolbars, then New Toolbar and find the putty-sessions folder.	[reply]
Re: Putty Session Generator by thor (Priest) on Oct 24, 2004 at 13:19 UTC
I think I might use this once I disect it and (possibly) clean it up. Please take the following as instructive and not as derisive. A couple of things that jumped out at me: `use strict 'vars';` Usually one needs a pretty good reason to enable only some of the strictitures. As it turns out, your code runs under full strict mode. You're not passing variables to subs. You might think that you are with calls like `$user=create_user($folder);`, but the `create_user` sub (and all the others) use `$folder` (and the rest of the variables) in a global sense. This is usually considered bad form. I might try using your script as a base for creating one that does the same thing, but using an XML file as a config so that you only have to specify the values that are not the default value. Plus, it would give me an excuse to learn perlXML...;). This is a great idea! thor `Feel the white light, the light within Be your own disciple, fan the sparks of will For all of us waiting, your kingdom will come`	[reply]
Re^2: Putty Session Generator by d_jabsd (Acolyte) on Oct 25, 2004 at 16:26 UTC
Thanks for your input, thor! I'm still trying to get a handle on the intricacies of Perl, so all advice is greatly appreciated. Just to clarify, I can replace `use strict 'vars';` with `use strict;` ? Could you explain item 2 a bit further, or point me to something that might help me fix it? Thanks!	[reply] [d/l] [select]
Re^3: Putty Session Generator by thor (Priest) on Oct 25, 2004 at 18:36 UTC
Just to clarify, I can replace use strict 'vars'; with use strict; ? For this script, you can. I find that coding under the full strictitures keeps me from doing things that I probably didn't mean, and is not that much harder than doing so without. Could you explain item 2 a bit further, or point me to something that might help me fix it? Sure. You're calling subs right now like `mysub($foo, $bar)` where the definition of mysub is something like `sub mysub { $foo ++; some_other_sub($bar); }` [download] It should read like this: `sub mysub { my ($foo, $bar) = @_; #do stuff with $foo and $bar }` [download] This allows you to follow another maxim that should be adhered to in general: declare your variables in the tightest scope that you can. That is to say that if you're using a while loop and some values only hold true for only that iteration of the while loop, you should declare them within the scope of that loop. For a concrete example from your code: `my ($folder, $server_name, $session_name, $protocol, $hexport, $version, $compr, $user); while (<SERVER_LIST>) { next if /$ignore/ ; chop; ( $folder, $server_name, $session_name, $protocol)= split (":"); ($hexport,$version,$compr)=create_port($protocol); $user=create_user($folder); &create_session($server_name, $session_name, $protocol, $user); &create_link($folder, $server_name, $session_name); }` [download] becomes (with some other differences sprinkled in) while (<SERVER_LIST>) { next if /$ignore/ ; # chomp is safer than chop...read the perldoc for # both of those functions (perldoc -f chop for chop chomp; my ($folder, $server_name, $session_name, $protocol)= split (":"); my ($hexport,$version,$compr)=create_port($protocol); my $user=create_user($folder); # the &sub syntax is not advised in most situations for reasons # that are slightly advanced, sub() suffices create_session($server_name, $session_name, $protocol, $user); create_link($folder, $server_name, $session_name); } [download] I didn't look too carefully at the code that is called by this block, but I'm pretty sure that the subs create_session and create_link would have to be altered in such a way as to not use global variables to store their information. I hope this helps. If you have more questions, feel free to ask. I'm more than happy to help! thor `Feel the white light, the light within Be your own disciple, fan the sparks of will For all of us waiting, your kingdom will come`	[reply] [d/l] [select]
Re^4: Putty Session Generator by ikegami (Patriarch) on Oct 25, 2004 at 18:57 UTC
Re^5: Putty Session Generator by thor (Priest) on Oct 25, 2004 at 20:22 UTC


Think about Loose Coupling
	PerlMonks