Subversion Repositories camp_sysinfo_client_3

#!/usr/bin/env perl

use warnings;

use strict;  

# Description: Use smartctl to get disk information

our $VERSION = '1.0';

# R. W. Rodolico

# grabs smart readings from all drives in system

BEGIN {

   push @INC, shift;

}

use library;

exit 0 unless checkDate( 'w' ); # run this only on Mondays

sub trim {

 my $value = shift;

 $value =~ s/^\s+|\s+$//g;

 return $value;

}

sub getSmartReport {

   my $drive = shift;

   my @report = `smartctl -a /dev/$drive`;

   return @report;

}

sub getTotalWrites {

 my @report = @_;

 return -2 unless grep{ /(Solid State Device)|(SSD)/ } @report;

 my @temp = grep{ /^Sector Size:/ } @report;

 my $sectors = shift @temp;

#   print "The Value is [$sectors]\n"; die;

 if ( $sectors =~ m/^Sector Size\:\s+(\d+)\s+bytes/ ) {

    $sectors = $1;

    # print "Sectors is $sectors\n"; die;

    @temp =  grep{ /^241/ } @report;

    my $lbas = $temp[0];

    if ( $lbas =~ m/(\d+)\s*$/ ) {

       $lbas = $1;

       return $lbas * $sectors;

    } else {

       return -3;

    }

 } else { 

    return -4; 

 }

 return -1; # we could not find something

}

sub getInformation {

   my @report = @_;

   my %info;

   my %keys = ( 

                  'Model Family:'  => { 

                                          'tag' => 'Make',

                                          'regex' => '(.*)'

                                      },

                  'Device Model:'  => { 

                                          'tag' => 'Model',

                                          'regex' => '(.*)'

                                      },

                  'Serial Number:' => { 

                                          'tag' => 'Serial',

                                          'regex' => '(.*)'

                                      },

                  'User Capacity:' => { 

                                          'tag' => 'Capacity',

                                          'regex' => '([0-9,]+)'

                                      },

                  'Sector Size:'   => { 

                                          'tag' => 'Sector Size',

                                          'regex' => '(\d+)'

                                      },

                  'Rotation Rate:' => { 

                                          'tag' => 'Rotation',

                                          'regex' => '(.*)'

                                      }

               );

   foreach my $key ( keys %keys ) {

      my @temp = grep { /^$key/ } @report;

      if ( @temp ) {

         $temp[0] =~ m/^$key\s+(.*)$/;

         my $value = $1;

         $value =~ m/$keys{$key}->{'regex'}/;

         $info{$keys{$key}->{'tag'}} = $1;

      }

   }

   # remove comma's from capacity

   $info{'Capacity'} =~ s/,//g if $info{'Capacity'};

   return \%info;

}

# category we will use for all values found

# see sysinfo for a list of valid categories

my $CATEGORY = 'system';

# run the commands necessary to do whatever you want to do

# The library entered above has some helper routines

# validCommandOnSystem -- passed a name, returns the fully qualified path or '' if it does not exist

# cleanUp - passed a delimiter and a string, does the following (delimiter can be '')

#           chomps the string (removes trailing newlines)

#           removes all text BEFORE the delimiter, the delimiter, and any whitespace

#           thus, the string 'xxI Am x  a weird string' with a newline will become

#           'a weird string' with no newline

# now, return the tab delimited output (to STDOUT). $CATEGORY is the first

# item, name as recognized by sysinfo is the second and the value is

# the last one. For multiple entries, place on separate lines (ie, newline separated)

# check for commands we want to run

my %commands = ( 

                  'smartctl' => '',

                  'geom'     => '',

                  'lsblk'    => '',

               );

# check the commands for validity

foreach my $command ( keys %commands ) {

   $commands{$command} = &validCommandOnSystem( $command );

}

$commands{'getDrives'} = $commands{ 'lsblk' } ? $commands{'lsblk'} . " | grep disk | cut -d' ' -f1" :

                                  ( $commands{'geom'} ? $commands{'geom'} . " disk list | grep 'Geom name:' | cut -d':' -f 2" : '' );

# bail if we don't have the commands we need

exit 1 unless $commands{'getDrives'} and $commands{'smartctl'};

# Get all the drives on the system

my %allDrives = map { &trim($_) => 0 } `$commands{'getDrives'}`;

# output the drives and information as tab delimited,

foreach my $thisDrive ( sort keys %allDrives ) {

 print "$thisDrive\n";

 my @report = &getSmartReport( $thisDrive );

 my $writes = &getTotalWrites( @report );

 my $info = &getInformation( @report );

 $info->{'writes'} = $writes if $writes > 0;

 #print Dumper( $info );

 foreach my $key ( keys %$info ) {

    print "$CATEGORY\t$thisDrive $key\t" . $info->{$key} . "\n";

 }

}

exit 0;