Subversion Repositories camp_sysinfo_client_3

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

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

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

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

my $clientName = '';

my $serialNumber = '';

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

                  '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 .= "\@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-'};

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

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

         $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';

}

# 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 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 );

}

   } 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 );

      }

sub processParameters {

   while ( my $parameter = shift ) {

      if ( $parameter eq '-v' ) {

         print "$main::VERSION\n";

         exit;

      }

   } # while

}

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';

}