126 lines
3.0 KiB
Perl
126 lines
3.0 KiB
Perl
# Pandora FMS Agent WMI library.
|
|
package PandoraWMI;
|
|
use strict;
|
|
use warnings;
|
|
use DBI;
|
|
use DBD::WMI;
|
|
|
|
########################################################################################
|
|
# PandoraWMI class constructor.
|
|
########################################################################################
|
|
sub new ($) {
|
|
my $class = shift;
|
|
|
|
my $self = {};
|
|
bless $self, $class;
|
|
return $self;
|
|
}
|
|
|
|
# Return a WQL safe string
|
|
sub safe_wql_string ($) {
|
|
my $string = shift;
|
|
|
|
# Remove characters that may break the query
|
|
$string =~s/'//g;
|
|
|
|
return $string;
|
|
}
|
|
|
|
# Return an array of values from a WMI query
|
|
sub get_row ($@) {
|
|
my ($query, @values) = @_;
|
|
|
|
my $dbh = DBI->connect('dbi:WMI:');
|
|
return undef unless defined ($dbh);
|
|
|
|
my $sth = $dbh->prepare($query);
|
|
return undef unless defined ($sth);
|
|
|
|
$sth->execute(@values);
|
|
return undef unless defined ($sth);
|
|
|
|
while (my @row = $sth->fetchrow()) {
|
|
return @row;
|
|
}
|
|
|
|
$sth->finish();
|
|
return undef;
|
|
}
|
|
|
|
# Get CPU load percentage
|
|
sub get_cpu_usage ($) {
|
|
my $self = shift;
|
|
|
|
my ($load_percentage) = get_row ('SELECT LoadPercentage FROM Win32_Processor');
|
|
return $load_percentage;
|
|
}
|
|
|
|
# Get memory usage in MBytes
|
|
sub get_free_memory ($) {
|
|
my $self = shift;
|
|
|
|
my ($available_mbytes) = get_row ('SELECT AvailableMBytes FROM Win32_PerfRawData_PerfOS_Memory');
|
|
return $available_mbytes;
|
|
}
|
|
|
|
# Get memory usage percentage
|
|
sub get_free_memory_percentage ($) {
|
|
my $self = shift;
|
|
|
|
my ($free_memory, $total_memory) = get_row ('SELECT FreePhysicalMemory, TotalVisibleMemorySize FROM Win32_OperatingSystem');
|
|
return undef unless (defined ($free_memory) && defined ($total_memory));
|
|
|
|
return 0 unless ($total_memory > 0);
|
|
|
|
return $free_memory * 100.0 / $total_memory;
|
|
}
|
|
|
|
# Get process status (1 running, 0 not running)
|
|
sub get_process_status ($$) {
|
|
my ($self, $process_name) = @_;
|
|
|
|
my ($name) = get_row ("SELECT Name FROM Win32_Process WHERE Name='" . safe_wql_string ($process_name) . "'");
|
|
return 0 unless (defined ($name));
|
|
|
|
return 1;
|
|
}
|
|
|
|
# Get service status (1 running, 0 not running)
|
|
sub get_service_status ($$) {
|
|
my ($self, $service_name) = @_;
|
|
|
|
my ($state) = get_row ("SELECT State FROM Win32_Service WHERE Name='" . safe_wql_string ($service_name) . "'");
|
|
return 0 unless (defined ($state));
|
|
|
|
# Running!
|
|
return 1 if ($state eq 'Running');
|
|
|
|
return 0;
|
|
}
|
|
|
|
# Get free disk space in MBytes
|
|
sub get_free_disk_space ($$) {
|
|
my ($self, $device_id) = @_;
|
|
|
|
my ($free_space) = get_row ("SELECT FreeSpace FROM Win32_LogicalDisk WHERE DeviceID='" . safe_wql_string ($device_id) . "'");
|
|
return 0 unless (defined ($free_space));
|
|
|
|
# Convert to MBytes (1048576 = 1024 * 1024)
|
|
return $free_space / 1048576.0;
|
|
}
|
|
|
|
# Get free disk space percentage
|
|
sub get_free_disk_space_percentage ($$) {
|
|
my ($self, $device_id) = @_;
|
|
|
|
my ($size, $free_space) = get_row ("SELECT Size, FreeSpace FROM Win32_LogicalDisk WHERE DeviceID='" . safe_wql_string ($device_id) . "'");
|
|
return 0 unless (defined ($size) && defined ($free_space));
|
|
|
|
return 0 unless ($size > 0);
|
|
|
|
return $free_space * 100.0 / $size;
|
|
}
|
|
|
|
1;
|
|
__END__
|