Subversion Repositories camp_sysinfo_client_3

#!/usr/bin/env perl

use warnings;

my $seedFile = 'sysinfo-client.seed';

my $sysinfo2 = '/etc/sysinfo/sysinfo.conf';

my $sysinfo3 = '/etc/camp/sysinfo-client/sysinfo-client.conf';

my $configPath = '/etc/camp/sysinfo-client';

my $configFile = 'sysinfo-client.conf';

my $clientName = '';

my $serialNumber = '';

my $hostname;

my @moduleDirs = ( '/opt/camp/sysinfo-client/modules', '/etc/camp/sysinfo-client/modules' );

my @scriptDirs = ( '/opt/camp/sysinfo-client/scripts', '/etc/camp/sysinfo-client/scripts' );

my $transports = {}; # holds transportation mechanisms

# the following are keys which are specific to the different transport channels

my %sendTypes = ( 

                  'SendEmail' =>    { 'sendScript' => 'sendEmailScript',

                                      'keys' => 

                                      [

                                        'mailTo',

                                        'mailSubject',

                                        'mailCC',

                                        'mailBCC',

                                        'mailServer',

                                        'mailFrom',

                                        'logFile',

                                        'otherCLParams',

                                        'tls',

                                        'smtpUser',

                                        'smtpPass',

                                        'sendEmailScriptLocation'

                                      ],

                                    },

                  'HTTP Upload' =>  { 'sendScript' => 'upload_http',

                                      'keys' => 

                                      [

                                        'URL',

                                        'key for report',

                                        'key for date',

                                        'key for hostname',

                                        'key for client',

                                        'key for serial number'

                                      ]

                                    }

                );

sub showConf {

   my $conf;

   $conf .= "\$clientName = '" . $clientName . "';\n";

   $conf .= "\$serialNumber = '" . ( defined( $serialNumber ) ? $serialNumber : '' ) . "';\n";

   $conf .= "\$hostname = '" . $hostname . "';\n";

   $conf .= "\@moduleDirs = ('" . join( "','", @moduleDirs ). "');\n";

   $conf .= "\@scriptDirs = ('" . join( "','", @scriptDirs ). "');\n";

   $conf .= &transportsToConfig();

   return $conf;

}

sub transportsToConfig {

   my $config;

   foreach my $priority ( sort { $a <=> $b } keys %$transports ) {

      my $name = $$transports{ $priority }{'-name-'};

      $config .= "# Tranport $name at priority $priority\n";

      my $thisOne = $$transports{ $priority };

      foreach my $key ( sort keys %$thisOne ) {

         # if they are wanting variable substitution, surround with double quote

         # otherwise, use single quote

         if ( $$thisOne{$key} =~ m/(\$hostname)|(\$reportDate)|(\$clientName)|(\$serialNumber)/ ) {

            $config .= "\$\$transports{$priority}{'$key'} = \"$$thisOne{$key}\";\n";

         } else {

            $config .= "\$\$transports{$priority}{'$key'} = '$$thisOne{$key}';\n";

         }

      } # foreach

   } # foreach

   return $config;

} # transportsToConfig

# prompt the user for a response, then allow them to enter it

# the first response is considered the default and is printed

# in all caps if more than one exists

# first answer is used if they simply press the Enter

# key. The response is returned all lower case if more than one

# exists.

# it is assumed 

sub getAnswer {

   my ( $prompt, @answers ) = @_;

   $answers[0] = '' unless defined( $answers[0] );

   my $default = $answers[0];

   my $onlyOneAnswer = scalar( @answers ) == 1;

   print $prompt . '[ ';

   $answers[0] = uc $answers[0] unless $onlyOneAnswer;

   print join( ' | ', @answers ) . ' ]: ';

   $thisAnswer = <>;

   chomp $thisAnswer;

   $thisAnswer = $default unless $thisAnswer;

   return $thisAnswer;

}

sub yesno {

   my $prompt = shift;

   my $answer = &getAnswer( $prompt, ('yes','no' ) );

   return lc( substr( $answer, 0, 1 ) ) eq 'y';

}

sub findSendEmail {

   my $currentLocation = shift;

   my @possibles = ( '/opt/sendEmail/sendEmail', '/opt/sendEmail/sendEmail.pl' );

   return $currentLocation if ( $currentLocation && -x $currentLocation );

   # well, we didn't find it, so look around some more

   # we install it in /opt, so try there

   foreach my $current ( @possibles ) {

      return $current if -x $current;

   }

   if ( &yesno( "You are asking for sendEmail, but I don't see it on the system\nWould you like me to automatically download and install" ) ) {

      $path = `perl getSendEmail.pl`;

      chomp $path;

      return $path;

   }

   return '';

}

sub userConfirmation {

   my $temp;

   $clientName = &getAnswer( "Client Name ", ($clientName) );

   $serialNumber = &getAnswer( "Serial Number ", ($serialNumber) );

   $hostname = &getAnswer( "Host Name ", ($hostname) );

   $temp = join( ',',@moduleDirs );

   $temp = &getAnswer( "Locations to search for modules (comma separated) ", ($temp) );

   @moduleDirs = split( ',', $temp );

   $temp = join( ',',@scriptDirs );

   $temp = &getAnswer( "Locations to search for scripts (comma separated) ", ($temp) );

   @scriptDirs = split( ',', $temp );

}

# simply attempts to detect the operating system so we can do OS

# specific actions at the end.   

sub getOperatingSystem {

   my @OSTypes = ( 'ipfire','debian' );

   my $OS = `uname -a`;

   foreach $osType ( @OSTypes ) {

      return $osType if $OS =~ m/$osType/i;

   }

   return '';

} # getOperatingSystem

# special purpose for IPFire routers

sub ipFire {

   my @BACKUP_DIRS = ( '/etc/camp' );

   `ln -s /opt/camp/sysinfo/sysinfo-client /etc/fcron.daily/sysinfo.cron` if &yesno( 'Add link to fcron.daily' );

   # now, check for backup directories not in include.user

   open BACKUP, '</var/ipfire/backup/include.user';

   while ( $line = <BACKUP> ) {

      chomp $line;

      for (my $i = 0; $i < @BACKUP_DIRS; $i++ ) {

         if ( $BACKUP_DIRS[$i] eq $line ) {

            $BACKUP_DIRS[$i] = '';

            last;

         } # if

      }# for

   } # while

   close BACKUP;

   # if any remain, append them to include.user

   open BACKUP, '>>/var/ipfire/backup/include.user';

   foreach my $backupDir ( @BACKUP_DIRS ) {

      print BACKUP "$backupDir\n" if $backupDir;

   }

   close BACKUP;

   # set all modules with ipfire or unix in the name to run

   my $moduleDir = $moduleDirs[0];

   opendir $moduleDir, $moduleDir;

   my @modules = grep { /^((ipfire)|(unix))/ } readdir $moduleDir;

   closedir $moduleDir;

   foreach my $module ( @modules) {

      `chmod 0700 $moduleDir/$module`;

   }

   print "All IPFire specific modules have been enabled\n";

}

# some debian specific things

sub debian {

   if ( &yesno( 'Add link to cron.daily' ) ) {

      `rm /etc/cron.daily/sysinfo` if -e '/etc/cron.daily/sysinfo';

      `ln -s /opt/camp/sysinfo-client/sysinfo-client /etc/cron.daily/sysinfo`;

   }

   if ( `dpkg --get-selections | grep  sysinfo-client | grep install` 

      && &yesno ('It looks like sysinfo version 2 is installed via apt, should I remove it' ) ) {

      if ( &yesno( 'Back up old configuration? ') ) {

         `tar -czvf /root/sysinfo.2.config.tgz /etc/sysinfo`;

         print "Old configuration copied to /root/sysinfo.2.config.tgz\n";

      }

      `apt-get -y --purge remove sysinfo-client`;

      print "There may be some unused packages now. You can remove them with\napt-get autoremove\n\n";

   }

   if ( &yesno( 'Would you like a link to /usr/local/bin/sysinfo-client' ) ) {

      `rm /usr/local/bin/sysinfo-client` if -e '/usr/local/bin/sysinfo-client';

      `ln -s  /opt/camp/sysinfo/sysinfo-client /usr/local/bin/sysinfo-client`;

   }

   # set all modules with debian or unix in the name to run

   foreach my $directory ( @moduleDirs ) {

      opendir ( $dh, $directory ) or die "Could not open directory [$directory]\n";

      my @modules = grep { /^((dpkg)|(unix))/ } readdir $dh;

      closedir $dh;

      foreach my $module (@modules) {

         `chmod 0700 $directory/$module`;

         `chown root:root $directory/$module`;

      }

   }

   print "All debian specific modules have been enabled\n";

}

sub setUpTransport {

   my ($priority, $transport, $fields, $type ) = @_;

   $priority = getAnswer( 'Priority', $priority );

   $$transport{'sendScript'} = $$fields{'sendScript'} unless $$transport{'sendScript'};

   $$transport{'-name-'} = $type unless $$transport{'-name-'};

   my $allKeys = $$fields{'keys'};

   foreach my $key ( @$allKeys ) {

      if ( $key eq 'sendEmailScriptLocation' && ! -e $$transport{$key} ) {

         $temp = &findSendEmail( $$transport{$key} );

         $$transport{$key} = $temp if $temp;

      }

      $$transport{$key} = &getAnswer( "$key ", $$transport{$key} ? $$transport{$key} : '' );

   }

   return ( $priority, $transport );

}

sub showAllTransports {

   foreach my $priority ( sort { $a <=> $b } keys %$transports ) {

      print '='x40 . "\n";

      &showTransports( $priority, $$transports{ $priority } );

      print '='x40 . "\n";

   }

}   

sub showTransports {

   my ( $priority,$thisOne )  = @_;

   print $$thisOne{'-name-'} . " has priority $priority\n";

   foreach my $key ( sort keys %$thisOne ) {

      next if $key =~ m/^-.*-$/;

      print "$key = $$thisOne{$key}\n";

   }

}

sub doTransports {

   my ( $transports ) = @_;

   foreach my $priority ( sort { $a <=> $b } keys %$transports ) {

      if ( &yesno( $$transports{$priority}{'-name-'} . " has a priority of $priority, edit it?") ) {

         #print Dumper( $sendTypes{$$transports{$priority}{'-name-'}} );

         #die;

         my ( $newpriority,$temp ) = &setUpTransport( $priority, $$transports{$priority}, $sendTypes{$$transports{$priority}{'-name-'}} );

         if ( $newpriority != $priority ) {

            delete $$transports{$priority};

            $priority = $newpriority;

         }

         $$transports{$priority} = $temp;

      } # if

   }

   foreach my $priority ( sort { $a <=> $b } keys %$transports ) {

      print '='x40 . "\n";

      &showTransports( $priority, $$transports{ $priority } );

      print '='x40 . "\n";

   }

   while ( &yesno( "Would you like to add any other transport mechanisms?" ) ) {

      $newType = &getAnswer( 'Type of Transport? ', keys %sendTypes );

      my ( $priority,$temp ) = &setUpTransport( 99, {}, $sendTypes{$newType}, $newType );

      $$transports{$priority} = $temp;

   }

   foreach my $priority ( sort { $a <=> $b } keys %$transports ) {

      print '='x40 . "\n";

      &showTransports( $priority, $$transports{ $priority } );

      print '='x40 . "\n";

   }

}

sub writeConfig {

   my ($path,$filename,$content) = @_;

   my $return;

   `mkdir -p $path` unless -d $path;

   $filename = $path . '/' . $filename;

   if ( -e $filename ) {

      `cp $filename $filename.bak` if ( -e $filename );

      $return .= "Old config copied to $filename.bak\n";

   }

   open CONF,">$filename" or die "Could not write to $filename: $!\n";

   print CONF $content;

   close CONF;

   `chmod 600 $filename`;

   $return .= "Configuration successfully written to $filename\n";

   return $return;

}

sub convertSysinfo2 {

   my $client_name;

   my $iMailResults;

   my $mailTo;

   my $mailSubject;

   my $mailCC;

   my $mailBCC;

   my $mailServer;

   my $mailServerPort;

   my $mailFrom;

   my $SENDMAIL;

   my $clientName;

   my $serialNumber;

   my $hostname;

   my $transports = {}; # holds transportation mechanisms

   open SEED, "<$sysinfo2" or die "Could not open $sysinfo2: $!\n";

   my $temp = join( '', <SEED> );

   close SEED;

   eval( $temp );

   if ( $iMailResults ) {

      $temp = {};

      $$temp{'-name-'} = 'SendEmail';

      $$temp{'mailTo'} = $mailTo if $mailTo;

      $$temp{'$mailFrom'} = $mailFrom if $mailFrom;

      $$temp{'mailCC'} = $mailCC if $mailCC;

      $$temp{'mailServer'} = $mailServer if $mailServer;

      $$temp{'mailServer'} .= ":$mailServerPort" if $mailServerPort;

      $$transports{'1'} = $temp;

   }   

   $clientName = $client_name if ( $client_name );

   return ( $clientName,$hostname,$serialNumber,$transports );

}

sub loadConfig {

   # See if existing configuration. If so, offer to load it

   if ( -f $sysinfo3 && &yesno( 'I found an existing configuration, modify it ' ) ) {

      print "Loading defaults from existing config $sysinfo3\n";

      my $client_name;

      open SEED, "<$sysinfo3" or die "Could not open $sysinfo3: $!\n";

      my $temp = join( '', <SEED> );

      close SEED;

      eval( $temp );

   } else { # nope, so see if an old sysinfo2 or a seed file exists

      if ( -f $sysinfo2 && &yesno( 'Old sysinfo 2 config exists, load it for defaults' ) ) {

         print "Loading defaults from sysinfo2 config $sysinfo2\n";

         # NOTE: the return values are all placed into globals!!!!

         ( $clientName,$hostname,$serialNumber,$transports ) = &convertSysinfo2();

      }

      # seed files are expected to be in the correct format already

      if ( -f $seedFile  && &yesno( 'An installation seed file was found, load it' ) ) {

         print "Loading seed file $seedFile\n";

         open SEED, "<$seedFile" or die "Could not open $seedFile: $!\n";

         my $temp = join( '', <SEED> );

         close SEED;

         eval( $temp );

      }

   }

}

&loadConfig();

unless ( $hostname ) {

   $hostname = `hostname -f`;

   chomp $hostname;

}

&userConfirmation();

&doTransports( $transports );

print "The following is the configuration I'll write out\n" . '-'x40 . "\n";

print &showConf(  );

print '-'x40 . "\n";

print &writeConfig( $configPath, $configFile, &showConf() ) if &yesno( "Write this configuration" );

my $os = &getOperatingSystem();

if ( $os && &yesno( "I recognize this system, may I automatically set up a few things" ) ) {

   &ipFire() if $os eq 'ipfire';

   &debian() if $os eq 'debian';

}

print "sysinfo-client has been installed configured. You can return to\nthis screen at any time by running /opt/camp/sysinfo/config.pl\n";

1;