Subversion Repositories camp_sysinfo_client_3

#!/usr/bin/env perl

# v1.2.0 20161022 RWR

# moved makeConfig here so it is usable by install and configure

#

# v1.3.0 20190108 RWR

# added UUID and set defaults for config file

#

# v1.4.0 20190204 RWR

# added autoconvert from v1.2 configuration file to v1.3

# using dumper to set configuration file

#

# version 2.0 20190330 RWR

# changed it so all configs are YAML

#

# version 2.1 20191101 RWR

# changed --name-- tag to name

#

# version 2.2 20191105 RWR

# fixed where dmidecode missing caused exception

#

# version 2.2.1 20191112 RWR

# added timeStamp

package sysinfoconf;

our $VERSION = '2.2.1';

use warnings;

use strict;  

#use Data::Dumper;

eval( 'use YAML::Tiny' );

#use YAML::Tiny;

use File::Basename;

use Exporter;

our @ISA = qw( Exporter );

our @EXPORT = qw( $clientName $serialNumber $hostname @moduleDirs @scriptDirs 

                  $transports $sysinfo3 %sendTypes $binDir $moduleDir

                  $scriptDir $confDir $binName $confName

                  &showConf &transportsToConfig &getAnswer &yesno  

                  &writeConfig &processParameters $TESTING &checkDirectoryExists

                  &runCommand &setDryRun &enableModules &makeConfig &findConf &timeStamp

                );

our $TESTING = 0;

our $dryRun;

our $clientName = '';

our $serialNumber = '';

our $hostname = '';

my @installDirs = ( '/opt/camp/sysinfo-client', '/usr/local/opt/camp/sysinfo-client' );

my @confDirs =    ( '/etc/camp/sysinfo-client', '/usr/local/etc/camp/sysinfo-client' );

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

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

our $transports = {}; # holds transportation mechanisms

our $sysinfo3 = 'sysinfo-client.yaml';

our $binDir = '/opt/camp/sysinfo-client';

our $moduleDir = $binDir . '/modules';

our $scriptDir = $binDir . '/scripts';

our $confDir = '/etc/camp/sysinfo-client';

our $binName = 'sysinfo-client';

our $confName = 'sysinfo-client.yaml';

sub setDryRun {

   $dryRun = shift;

}

our %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'

                                      ]

                                    },

                  'Save Local' =>   { 'sendScript' => 'save_local',

                                      'keys' =>

                                      [

                                         'output directory'

                                      ]

                                   }

                );

#######################################################

#

# timeStamp

#

# return current system date as YYYY-MM-DD HH:MM:SS

#

#######################################################

sub timeStamp {

   my ($sec,$min,$hour,$mday,$mon,$year,$wday,$yday,$isdst)=localtime(time);

   return sprintf "%4d-%02d-%02d %02d:%02d:%02d",$year+1900,$mon+1,$mday,$hour,$min,$sec;

}

sub enableModules {

   my $moduleDir = shift;

   my %modules;

   if ( opendir( my $dh, "$moduleDir" ) ) {

      %modules = map{ $_ => { 'enabled' => -x "$moduleDir/$_" } } 

                 grep { ! /^\./ } 

                 readdir $dh;

      closedir( $dh );

      foreach my $filename ( keys %modules ) {

         if ( open FILE,"<$moduleDir/$_" ) {

            my @descript = grep{ /^# Description: (.*)/ } <FILE>;

            close FILE;

            chomp @descript;

            $modules{$filename}{'description'} = $descript[0];

         } # if

      } # foreach

   } # if

} # enableModules

sub showConf {

   my $configuration = shift;

   $configuration = Dump($configuration);

   return $configuration;

}

sub transportsToConfig {

   my $transports = shift;

   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 ) {

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

      } # foreach

   } # foreach

   return $config;

} # transportsToConfig

# read the configuration file passed in and return a reference to it

# to the calling routine

sub readConfig {

   my $filename = shift;

   my $config;

   return $config unless -e $filename;

   if ( open CONF,"<$filename" ) {

      my $contents = join( '', <CONF> );

      close CONF;

      # now, look to see what kind of file this is.

      if ( $contents =~ m/^---(\s*#.*)?$/m ) {

         print "Reading $filename as YAML\n";

         #print "Contents are:\n\n=====================$contents\n=====================\n";

         # this is a yaml file

         #$contents = YAML::Tiny->read( $config );

         $config = Load( $contents );

      } elsif ( $contents =~ m/\$clientName/ ) {

         # this is old style

         print "Reading $filename as old school file\n";

         my $clientName = '';

         my $serialNumber = '';

         my $hostname = '';

         my @moduleDirs;

         my @scriptDirs;

         my $UUID = '';

         my $transports;

         # now, eval the information we just read.

         # NOTE: we must turn off strict while doing this, and we die if something breaks.

         no strict "vars";

         eval( $contents );

         use strict "vars";

         die "Error during eval: $@\n" if $@;

         $config->{'clientName'} = $clientName if $clientName;

         $config->{'serialNumber'} = $serialNumber if $serialNumber;

         $config->{'UUID'} = $UUID unless $config->{'UUID'};

         $config->{'hostname'} = $hostname if $hostname;

         $config->{'moduleDirs'} = [ @moduleDirs ] if @moduleDirs;

         $config->{'scriptDirs'} = [ @scriptDirs ] if @scriptDirs;

         $config->{'transports'} = $transports if $transports;

         foreach my $trans ( keys %{$config->{'transports'}} ) {

            if ( exists ( $config->{'transports'}->{$trans}->{'-name-'} ) ) {

               $config->{'transports'}->{$trans}->{'name'} = $config->{'transports'}->{$trans}->{'-name-'};

               delete( $config->{'transports'}->{$trans}->{'-name-'} );

            }

         }

      }

   } else {

         warn "Could not read config file $filename, skipped: $!\n";

   }

   return $config;

}

sub makeConfig {

   my @configFileNames = @_;

   my %config;

   foreach my $config ( @configFileNames ) {

      my $thisConfig = &readConfig( $config ) if $config && -e $config;

      # add the new config to %config, overwriting any existing keys which are duplicated

      @config{keys %$thisConfig} = values %$thisConfig;

   }

   # now, ensure the correct values are loaded in some areas

   unless ( $config{'hostname'} ) {

      $hostname = `hostname -f`;

      chomp $hostname;

      $config{'hostname'} = $hostname;

   }

   unless ( $config{'serialNumber'} ) {

      $serialNumber = `dmidecode -t 1 | grep 'Serial Number' | cut -d':' -f2` if `which dmidecode`;

      chomp $serialNumber;

      $serialNumber =~ s/\s//gi;

      $config{'serialNumber'} = $serialNumber;

   }

   unless ( $config{'UUID'} ) {

      my $UUID = `which dmidecode` ?  `dmidecode -t 1 | grep -i uuid | cut -d':' -f2` : '';

      $UUID =~ s/\s//gi;

      $config{'UUID'} = $UUID;

   }

   # ensure we have the default SaveLocal transport defined

   unless ( defined $config{'transports'}{'99'} ) {

      $config{'transports'}{'99'} = {

                         'name'=> 'SaveLocal',

                         'output directory' => '/tmp',

                         'sendScript' => 'save_local'

                        };

   }

   return \%config;

}

# 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 ) . ' ]: ';

   my $thisAnswer = <>;

   chomp $thisAnswer;

   $thisAnswer = $default unless $thisAnswer;

   return $thisAnswer;

}

sub yesno {

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

   $default = 'yes' unless $default;

   my $answer = &getAnswer( $prompt, $default eq 'yes' ? ('yes','no' ) : ('no', 'yes') );

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

}

# runs a system command. Also, if in testing mode, simply shows what

# would have been done.

sub runCommand {

   while ( my $command = shift ) {

      if ( $dryRun ) {

         print "$command\n";

      } else {

         `$command`;

      }

   }

   return 1;

} # runCommand

# checks if a directory exists and, if not, creates it

my %directories; # holds list of directories already created so no need to do an I/O

sub checkDirectoryExists {

   my ( $filename,$mod,$owner ) = @_;

   $mod = "0700" unless $mod;

   $owner = "root:root" unless $owner;

   print "Checking Directory for $filename with $mod and $owner\n" if $TESTING > 2;

   my ($fn, $dirname) = fileparse( $filename );

   print "\tParsing out $dirname and $filename\n" if $TESTING > 2;

   return '' if exists $directories{$dirname};

   if ( -d $dirname ) {

      $directories{$dirname} = 1;

      return '';

   }

   if ( &runCommand( "mkdir -p $dirname", "chmod $mod $dirname", "chown $owner $dirname" ) ) {

      $directories{$dirname} = 1;

   }

   return '';   

}

sub writeConfig {

   use File::Temp qw / tempfile /;

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

   if ( $filename ) { # they sent us a filename

      my $path;

      ($filename, $path ) = fileparse( $filename );

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

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

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

      unless ( $dryRun ) {

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

         print CONF $content;

         close CONF;

         `chmod 600 $filename`;

      }

   } else { # no path provided, so just create a temp file

      # we will create a temporary file and return the name

      # it is the calling programs responsiblity to remove the file

      my $fh;

      ($fh,$filename) = tempfile( UNLINK => 0 );

      print $fh $content;

      close $fh

   }

   return $filename;

}

sub processParameters {

   while ( my $parameter = shift ) {

      if ( $parameter eq '-v' ) {

         print "$VERSION\n";

         exit;

      }

   } # while

}

sub findConf {

   my $confName = shift;

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

      if ( -d $confDirs[ $i ] ) {

         return ( $confDirs[$i],  $confName );

      }

   }

   return ( '', $confName );

}

1;