=================================================================== RCS file: /cvs/trango/Net-Telnet-Trango/scripts/update_trango.pl,v retrieving revision 1.22 retrieving revision 1.26 diff -u -r1.22 -r1.26 --- trango/Net-Telnet-Trango/scripts/update_trango.pl 2007/02/01 17:58:33 1.22 +++ trango/Net-Telnet-Trango/scripts/update_trango.pl 2007/02/02 18:08:56 1.26 @@ -1,45 +1,36 @@ #!/usr/bin/perl -# $RedRiver: update_trango.pl,v 1.21 2007/02/01 17:08:44 mike Exp $ +# $RedRiver: update_trango.pl,v 1.25 2007/02/02 17:50:09 andrew Exp $ ######################################################################## -# update_trango.pl *** Updates trango foxes with a new firmware +# update_trango.pl *** Updates trango hosts with a new firmware # # 2005.11.15 #*#*# andrew fresh ######################################################################## use strict; use warnings; -use YAML qw/ Dump LoadFile /; +use YAML qw/ LoadFile /; use Net::TFTP; use lib '.'; use Net::Telnet::Trango; +use RedRiver::Wireless; my $config_file = shift || 'update_trango.yaml'; my $max_tries = 3; - my $l = Mylogger->new( { log_prefix => 'UT' } ); - $l->sp("Reading config file '$config_file'"); my $conf = LoadFile($config_file); -#$l->sp(" Hardware Type: $conf->{'type'}"); -#$l->sp(" File Name: $conf->{'file_name'}"); -#$l->sp(" File Size: $conf->{'file_size'}"); -#$l->sp(" File Checksum: $conf->{'file_cksum'}"); -#$l->sp(" FW Version: $conf->{'ver'}"); -#$l->sp(" FW Checksum: $conf->{'cksum'}"); -#$l->sp(""); +$l->sp("Reading hosts"); +my $hosts = RedRiver::Wireless::Read_Hosts(); -my @foxes = ( -'10.100.1.2', -'10.100.2.2', -'10.100.3.2', -'10.100.4.2', -); -foreach my $fox (@foxes) { - $l->sp("Checking: $fox"); + +foreach my $host (@{ $hosts }) { + next if $host->{group} !~ /^Trango/; + $l->sp(""); + $l->sp("Checking: $host->{name}"); my $needs_reboot = 0; ## Connect and login. @@ -47,64 +38,108 @@ Timeout => 5, Errmode => 'return', ) or die "Couldn't make new connection: $!"; - $l->p("Connecting to $fox"); - unless ( $t->open($fox) ) { + $l->p("Connecting to $host->{name}"); + unless ( $t->open($host->{name}) ) { $l->sp("Error connecting: $!"); next; } + my $password = $host->{Telnet_Password} || $conf->{general}->{password}; + + if ($host->{group} !~ /Client$/) { + $l->p("Logging in"); + $t->login($password); + unless ($t->logged_in) { + $l->p('Failed!'); + $t->close; + next; + } + + $l->sp("Getting sudb"); + my $sudb = $t->sudb_view; + if ($sudb) { + foreach my $su (@{ $sudb }) { + $l->p("Getting su info $su->{suid}"); + my $su_info = $t->su_info( $su->{suid} ); + if ($su_info->{ip}) { + $l->sp("Adding host $su_info->{ip}"); + my $new_host = { + Telnet_Password => $host->{Telnet_Password}, + group => $host->{group} . '-Client', + name => $su_info->{ip}, + remarks => $su_info->{remarks}, + }; + push @{ $hosts }, $new_host; + } else { + $l->sp("Couldn't get su info for $su->{suid}"); + if ($su_info->{ERR}) { + $l->sp("ERR: $su_info->{ERR}"); + } + next; + } + } + } + } + foreach my $firmware_type ('Firmware', 'FPGA') { if (! exists $conf->{$firmware_type}) { $l->s("No configs for '$firmware_type'"); - $t->close(); + $t->close; next; } my $host_type = $t->host_type; - if (uc($conf->{'firmware_type'}) eq 'FPGA') { + if ($firmware_type eq 'FPGA') { $host_type =~ s/\s.*$//; } if (! exists $conf->{$firmware_type}->{$host_type}) { $l->sp("No '$firmware_type' config for type $host_type"); - $t->close(); + $t->close; next; } - unless ($t->logged_in) { + if (! $t->logged_in) { $l->p("Logging in"); - $t->login($conf->{general}->{'password'}) || die "Couldn't login: $!"; - $t->exit(); + $t->login($password); + unless ($t->logged_in) { + $l->p('Failed!'); + $t->close; + next; + } } foreach my $k (keys %{ $conf->{general} }) { $conf->{$firmware_type}->{$host_type}->{$k} ||= $conf->{general}->{$k}; } $conf->{$firmware_type}->{$host_type}->{firmware_type} ||= $firmware_type; - $conf->{$firmware_type}->{$host_type}->{type} ||= $host_type; + $conf->{$firmware_type}->{$host_type}->{type} = $host_type; - $l->sp("Updating: $host_type $firmware_type on $fox"); + $l->sp("$host_type $firmware_type"); $l->p("Sending commands"); ## Send commands - if ( upload($t, $conf->{$firmware_type}->{$host_type}) ) { + my $rc = upload($t, $conf->{$firmware_type}->{$host_type}); + if ($rc) { $l->sp("Successfull!"); $needs_reboot = 1; + } elsif (defined $rc) { + $l->sp("Already up to date"); } else { $l->sp("Failed"); + $t->bye; + next; } - $l->sp(""); } if ($needs_reboot) { - $l->sp("Rebooting $fox"); + $l->sp("Rebooting $host->{name}"); $t->reboot; } else { - $l->sp("Exiting $fox"); - $t->exit(); + $l->sp("Bye $host->{name}"); + $t->bye(); } - exit; } sub upload @@ -117,23 +152,38 @@ my $fw_type = $conf->{firmware_type}; my $ver = $t->ver; - print Dump $ver; - $l->p("Current version '" . $ver->{$fw_type . ' Version'} . "'"); - exit; + if (! ( + $ver->{$fw_type . ' Version'} && + $ver->{$fw_type . ' Checksum'} + )) { + $l->sp("Error getting current version numbers"); + return; + } + if ( $ver->{$fw_type . ' Version'} eq $conf->{'ver'} && $ver->{$fw_type . ' Checksum'} eq $conf->{'cksum'} ) { - $l->sp("Already updated!"); return 0; } + $l->sp("Config information:"); + $l->sp(" Hardware Type: $conf->{'type'}"); + $l->sp(" File Name: $conf->{'file_name'}"); + $l->sp(" File Size: $conf->{'file_size'}"); + $l->sp(" File Checksum: $conf->{'file_cksum'}"); + $l->sp(" Conf Version: $conf->{'ver'}"); + $l->sp(" Cur Version: $ver->{$fw_type . ' Version'}"); + $l->sp(" Conf Checksum: $conf->{'cksum'}"); + $l->sp(" Cur Checksum: $ver->{$fw_type . ' Checksum'}"); + + $l->sp("Updating"); my $try = 0; while (1) { if ($try >= $max_tries) { $l->sp("Couldn't update in $max_tries tries!"); - return undef; + return; } $try++; @@ -175,10 +225,10 @@ } $l->p("File length matches . . . "); - if ( uc($results->{'File Name'}) ne uc($conf->{'file_name'}) ) { + if ( uc($results->{'File Name'}) ne uc($file) ) { $l->sp( "File name '" . $results->{'File Name'} . - "' does not match config file '" . $conf->{'file_name'} . "'!" + "' does not match config file '" . $file . "'!" ); next; }