[BACK]Return to update_trango.pl CVS log [TXT][DIR] Up to [local] / trango / Net-Telnet-Trango / scripts

Diff for /trango/Net-Telnet-Trango/scripts/update_trango.pl between version 1.26 and 1.30

version 1.26, 2007/02/02 18:08:56 version 1.30, 2007/02/06 23:00:31
Line 1 
Line 1 
 #!/usr/bin/perl  #!/usr/bin/perl
 # $RedRiver: update_trango.pl,v 1.25 2007/02/02 17:50:09 andrew Exp $  # $RedRiver: update_trango.pl,v 1.29 2007/02/05 16:37:20 mike Exp $
 ########################################################################  ########################################################################
 # update_trango.pl *** Updates trango hosts with a new firmware  # update_trango.pl *** Updates trango hosts with a new firmware
 #  #
Line 10 
Line 10 
   
 use YAML qw/ LoadFile /;  use YAML qw/ LoadFile /;
 use Net::TFTP;  use Net::TFTP;
 use lib '.';  
 use Net::Telnet::Trango;  use Net::Telnet::Trango;
 use RedRiver::Wireless;  
   
 my $config_file = shift || 'update_trango.yaml';  my $config_file = shift || 'update_trango.yaml';
 my $max_tries = 3;  my $max_tries = 3;
Line 22 
Line 20 
 $l->sp("Reading config file '$config_file'");  $l->sp("Reading config file '$config_file'");
 my $conf = LoadFile($config_file);  my $conf = LoadFile($config_file);
   
 $l->sp("Reading hosts");  my $hosts;
 my $hosts = RedRiver::Wireless::Read_Hosts();  if (@ARGV) {
       @{ $hosts } = map { { name => $_, group => 'Trango-Client' } } @ARGV
   } else {
       $hosts = parse_hosts($conf->{hosts});
   }
   
   #@{ $hosts } = grep { $_->{name} eq '10.100.7.2' } @{ $hosts };
   
   my $global_tries = $max_tries * 2;
   while ($global_tries > 0) {
     $global_tries--;
     my $processed = 0;
   
   
 foreach my $host (@{ $hosts }) {  foreach my $host (@{ $hosts }) {
   next if $host->{group} !~ /^Trango/;  
     if (! exists $host->{retry}) {
       $host->{tries} = 0;
       $host->{retry} = 1;
     }
   
     if ($host->{tries} >= $max_tries) {
       $host->{retry} = 0;
     }
   
     if ($host->{retry} <= 0) {
       next;
     }
   
     $host->{tries}++;
     $processed++;
   
   $l->sp("");    $l->sp("");
   $l->sp("Checking: $host->{name}");    $l->sp("Checking: $host->{name} (try $host->{tries})");
   my $needs_reboot = 0;    my $needs_reboot = 0;
   
   ## Connect and login.    ## Connect and login.
Line 46 
Line 70 
   
   my $password = $host->{Telnet_Password} || $conf->{general}->{password};    my $password = $host->{Telnet_Password} || $conf->{general}->{password};
   
   if ($host->{group} !~ /Client$/) {    $l->p("Logging in");
     $l->p("Logging in");    $t->login($password);
     $t->login($password);    unless ($t->logged_in) {
     unless ($t->logged_in) {      $l->p('Failed!');
       $l->p('Failed!');      $t->close;
       $t->close;      next;
       next;    }
     }  
   
     $l->sp("Getting sudb");    $l->sp("Getting sudb");
     my $sudb = $t->sudb_view;    my $sudb = $t->sudb_view;
     if ($sudb) {    if ($sudb) {
       foreach my $su (@{ $sudb }) {      foreach my $su (@{ $sudb }) {
         $l->p("Getting su info $su->{suid}");        $l->p("Getting su info $su->{suid}");
         my $su_info = $t->su_info( $su->{suid} );        my $su_info = $t->su_info( $su->{suid} );
         if ($su_info->{ip}) {        if ($su_info->{ip}) {
           $l->sp("Adding host $su_info->{ip}");          if (grep { $_->{name} eq $su_info->{'ip'} } @{ $hosts }) {
           my $new_host = {            $l->p("Already have $su_info->{ip}");
             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;            next;
         }          }
           $l->sp("Adding host $su_info->{ip}");
           my $new_host = {
             password => $host->{password},
             name     => $su_info->{ip},
             remarks  => $su_info->{remarks},
           };
           push @{ $hosts }, $new_host;
         } else {
           $l->sp("Couldn't get su info for $su->{suid}");
           $l->sp("ERR: " . $t->last_error);
       }        }
     }      }
   }    }
   
   foreach my $firmware_type ('Firmware', 'FPGA') {    foreach my $firmware_type ('Firmware', 'FPGA') {
   
     if (! exists $conf->{$firmware_type}) {      if (! exists $conf->{$firmware_type}) {
           $l->s("No configs for '$firmware_type'");        $l->s("No configs for '$firmware_type'");
           $t->close;        $t->close;
           next;        next;
         }      }
   
     my $host_type = $t->host_type;      my $host_type = $t->host_type;
     if ($firmware_type eq 'FPGA') {      if ($firmware_type eq 'FPGA') {
Line 95 
Line 117 
     }      }
   
     if (! exists $conf->{$firmware_type}->{$host_type}) {      if (! exists $conf->{$firmware_type}->{$host_type}) {
           $l->sp("No '$firmware_type' config for type $host_type");        $l->sp("No '$firmware_type' config for type $host_type");
           $t->close;        $t->close;
           next;        next;
         }      }
   
       if ($firmware_type eq 'Firmware' &&
         $t->firmware_version eq
         $conf->{$firmware_type}->{$host_type}->{ver}
       ) {
         $l->sp("Firmware already up to date");
         next;
       }
   
     if (! $t->logged_in) {      if (! $t->logged_in) {
       $l->p("Logging in");        $l->p("Logging in");
       $t->login($password);        $t->login($password);
Line 110 
Line 140 
       }        }
     }      }
   
         foreach my $k (keys %{ $conf->{general} }) {      foreach my $k (keys %{ $conf->{general} }) {
           $conf->{$firmware_type}->{$host_type}->{$k} ||= $conf->{general}->{$k};        $conf->{$firmware_type}->{$host_type}->{$k} ||= $conf->{general}->{$k};
     }      }
         $conf->{$firmware_type}->{$host_type}->{firmware_type} ||= $firmware_type;      $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("$host_type $firmware_type");      $l->sp("$host_type $firmware_type");
     $l->p("Sending commands");  
     ## Send commands      ## Send commands
     my $rc = upload($t, $conf->{$firmware_type}->{$host_type});      my $rc = upload($t, $conf->{$firmware_type}->{$host_type});
         if ($rc) {      if ($rc) {
       $l->sp("Successfull!");        $l->sp("Successfull!");
           $needs_reboot = 1;        $host->{retry}--;
         $needs_reboot = 1;
     } elsif (defined $rc) {      } elsif (defined $rc) {
           $l->sp("Already up to date");        $l->sp("Already up to date");
         $host->{retry}--;
     } else {      } else {
           $l->sp("Failed");        $l->sp("Failed");
           $t->bye;        $t->bye;
           next;        next;
     }      }
   
   }    }
Line 142 
Line 173 
   }    }
 }  }
   
     if (! $processed) {
       $l->sp("");
       $l->sp("Finished.  No more hosts.");
       last;
     }
   }
   
 sub upload  sub upload
 {  {
   my $t    = shift;    my $t    = shift;
Line 158 
Line 196 
     $ver->{$fw_type . ' Checksum'}      $ver->{$fw_type . ' Checksum'}
   )) {    )) {
     $l->sp("Error getting current version numbers");      $l->sp("Error getting current version numbers");
         return;      return;
   }    }
   
   if (    if (
Line 168 
Line 206 
     return 0;      return 0;
   }    }
   
     $l->sp("Updating $fw_type");
   $l->sp("Config information:");    $l->sp("Config information:");
   $l->sp("  Hardware Type: $conf->{'type'}");    $l->sp("  Hardware Type: $conf->{'type'}");
   $l->sp("  File Name:     $conf->{'file_name'}");    $l->sp("  File Name:     $conf->{'file_name'}");
Line 178 
Line 217 
   $l->sp("  Conf Checksum: $conf->{'cksum'}");    $l->sp("  Conf Checksum: $conf->{'cksum'}");
   $l->sp("  Cur  Checksum: $ver->{$fw_type . ' Checksum'}");    $l->sp("  Cur  Checksum: $ver->{$fw_type . ' Checksum'}");
   
   $l->sp("Updating");  
   my $try = 0;    my $try = 0;
   while (1) {    while (1) {
     if ($try >= $max_tries) {      if ($try >= $max_tries) {
Line 187 
Line 225 
     }      }
     $try++;      $try++;
   
     #sysinfo($self->{'_host'});  
   
     $l->p("Enabling TFTPd");      $l->p("Enabling TFTPd");
     unless ($t->enable_tftpd) {      unless ($t->enable_tftpd) {
       $l->sp("Couldn't enable tftpd");        $l->sp("Couldn't enable tftpd");
       next;        next;
     }      }
   
     $l->p("Uploading file ($file)");      $l->p("Uploading file ($conf->{file_name})");
     # use tftp to push the file up      # use tftp to push the file up
     my $tftp = Net::TFTP->new($t->Host, Mode => 'octet');      my $tftp = Net::TFTP->new($t->Host, Mode => 'octet');
   
Line 207 
Line 243 
     $l->p("Checking upload ($conf->{'file_cksum'})");      $l->p("Checking upload ($conf->{'file_cksum'})");
     my $results = $t->tftpd;      my $results = $t->tftpd;
     # check the 'File Length' against ???      # check the 'File Length' against ???
       if (! (
         $results->{'File Checksum'} &&
         $results->{'File Length'}   &&
         $results->{'File Name'}
       )) {
         $l->sp("Unable to get results of upload");
         next;
       }
     if ( $results->{'File Checksum'} ne $conf->{'file_cksum'}) {      if ( $results->{'File Checksum'} ne $conf->{'file_cksum'}) {
       $l->sp(        $l->sp(
         "File checksum '" . $results->{'File Checksum'} .          "File checksum (" . $results->{'File Checksum'} .
         " does not match config file '" . $conf->{'file_cksum'} . "'!"          ") does not match config file (" . $conf->{'file_cksum'} . ")!"
       );        );
       next;        next;
     }      }
Line 218 
Line 262 
   
     if ($results->{'File Length'}   !~ /^$conf->{'file_size'} bytes/) {      if ($results->{'File Length'}   !~ /^$conf->{'file_size'} bytes/) {
       $l->sp(        $l->sp(
         "File length '" . $results->{'File Length'} .          "File length (" . $results->{'File Length'} .
         "does not match config file '" . $conf->{'file_size'} . " bytes'!"          ") does not match config file (" . $conf->{'file_size'} . " bytes)!"
       );        );
       next;        next;
     }      }
Line 227 
Line 271 
   
     if ( uc($results->{'File Name'}) ne uc($file) ) {      if ( uc($results->{'File Name'}) ne uc($file) ) {
       $l->sp(        $l->sp(
         "File name '" . $results->{'File Name'} .          "File name (" . $results->{'File Name'} .
         "' does not match config file '" . $file . "'!"          ") does not match config file (" . $file . ")!"
       );        );
       next;        next;
     }      }
     $l->p("File name  matches . . . ");      $l->p("File name matches . . . ");
   
     my $image_type = 'mainimage';      my $image_type = 'mainimage';
     if ($fw_type eq 'FPGA') {      if ($fw_type eq 'FPGA') {
Line 255 
Line 299 
       $l->sp("Saved checksum " . $results->{'Checksum'} . " does not match config file " .  $conf->{'cksum'} . "!");        $l->sp("Saved checksum " . $results->{'Checksum'} . " does not match config file " .  $conf->{'cksum'} . "!");
       next;        next;
     }      }
     $l->p("Uploaded checksum ($results->{'Checksum'}) " .  
           "matches ($conf->{'cksum'})");      $l->p("$fw_type saved checksum matches . . . ");
   
     $l->sp("Successfully updated!");  
     return 1;      return 1;
   }    }
 }  }
   
   sub parse_hosts
   {
       my $src = shift;
   
       my @hosts;
       foreach my $h (@{ $src }) {
           if ($h->{name} =~ /^(\d{1,3}\.\d{1,3}\.\d{1,3}\.)(\d{1,3})-(\d{1,3})/) {
               for ($2..$3) {
                   my %cur_host;
                   foreach my $k (keys %{ $h }) {
                       $cur_host{$k} = $h->{$k};
                   }
                   $cur_host{name} = $1 . $_;
                   if (! grep { $cur_host{name} eq $h->{name} } @hosts) {
                       push @hosts, \%cur_host;
                   }
               }
           } else {
               push @hosts, $h;
           }
       }
   
       return \@hosts;
   }
   
 package Mylogger;  package Mylogger;
   
 use Fcntl ':flock'; # import LOCK_* constants  use Fcntl ':flock'; # import LOCK_* constants
Line 332 
Line 400 
       $MYLOG = $self->{'MYLOG'};        $MYLOG = $self->{'MYLOG'};
     } else {      } else {
       unless ($MYLOG) {        unless ($MYLOG) {
                 open ($MYLOG, '>>', $self->{'log_path'} . '/' . $self->{'log_file'})          open ($MYLOG, '>>', $self->{'log_path'} . '/' . $self->{'log_file'})
           or die "Couldn't open logfile!\n";            or die "Couldn't open logfile!\n";
         my $ofh = select $MYLOG;          my $ofh = select $MYLOG;
         $|=1;          $|=1;
Legend:
Removed from v.1.26  
changed lines
  Added in v.1.30
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>