077977ad9c
* Added 'ip_address_note' to the 'ip_addresses' table as there was no column convenient for flagging as DELETEd. * Added 'uuid' to Database->insert_or_update_file_locations() and ->insert_or_update_files(), and actually used it in all ->inser_or_update_X() methods. * Added 'delete' as a parameter to Database->insert_or_update_ip_addresses() to allow simple deletion of a referenced IP address. * Addressed a few 'undefined variable' errors. Signed-off-by: Digimer <digimer@alteeve.ca>
120 lines
4.6 KiB
Perl
Executable File
120 lines
4.6 KiB
Perl
Executable File
#!/usr/bin/perl
|
|
#
|
|
# This reports the total memory used by all processes with to passed-in program name.
|
|
#
|
|
# The size in bytes is returned. If '0' is reported, check the exit code to see why.
|
|
#
|
|
# This software was created by Alteeve's Niche! Inc. and has been released under the terms of the GNU GPL
|
|
# version 2.
|
|
#
|
|
# https://alteeve.com
|
|
#
|
|
# Exit Codes:
|
|
# 0 - Success
|
|
# 1 - No program name passed-in.
|
|
# 2 - No PIDs found for the passed-in program name.
|
|
# 3 - No PID returned by pgrep
|
|
#
|
|
|
|
use strict;
|
|
use warnings;
|
|
use Anvil::Tools;
|
|
|
|
my $THIS_FILE = ($0 =~ /^.*\/(.*)$/)[0];
|
|
my $running_directory = ($0 =~ /^(.*?)\/$THIS_FILE$/)[0];
|
|
if (($running_directory =~ /^\./) && ($ENV{PWD}))
|
|
{
|
|
$running_directory =~ s/^\./$ENV{PWD}/;
|
|
}
|
|
|
|
# Turn off buffering so that the pinwheel will display while waiting for the SSH call(s) to complete.
|
|
$| = 1;
|
|
|
|
my $anvil = Anvil::Tools->new({log_level => 2, log_secure => 1});
|
|
|
|
$anvil->data->{switches}{program} = "";
|
|
$anvil->Get->switches;
|
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => {
|
|
'switches::program' => $anvil->data->{switches}{program},
|
|
}});
|
|
|
|
$anvil->data->{memory}{total} = 0;
|
|
|
|
# If we weren't given a program, exit
|
|
if (not $anvil->data->{switches}{program})
|
|
{
|
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 0, key => "error_0065"});
|
|
print "#!error!#\n";
|
|
$anvil->nice_exit({exit_code => 1});
|
|
}
|
|
|
|
# Find the PID(s) of the program.
|
|
$anvil->data->{sys}{pids} = $anvil->System->pids({ignore_me => 1, program_name => $anvil->data->{switches}{program}});
|
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 1, list => { 'sys::pids' => $anvil->data->{sys}{pids} }});
|
|
|
|
my $pids_found = @{$anvil->data->{sys}{pids}};
|
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 1, list => { pids_found => $pids_found }});
|
|
|
|
if (not $pids_found)
|
|
{
|
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 1, level => 0, key => "error_0066", variables => { prorgram => $anvil->data->{switches}{program} }});
|
|
print $anvil->data->{switches}{program}." = ".$anvil->data->{memory}{total}."\n";
|
|
$anvil->nice_exit({exit_code => 2});
|
|
}
|
|
|
|
# Read in the smaps for each pid
|
|
foreach my $pid (sort {$a cmp $b} @{$anvil->data->{sys}{pids}})
|
|
{
|
|
my $smaps_path = "/proc/".$pid."/smaps";
|
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 1, list => { smaps_path => $smaps_path }});
|
|
|
|
# This will store the amount of RAM used by this specific PID.
|
|
$anvil->data->{memory}{pid}{$pid} = 0;
|
|
|
|
if (not -e $smaps_path)
|
|
{
|
|
# It is possible that the program just closed.
|
|
$anvil->Log->entry({source => $THIS_FILE, line => __LINE__, 'print' => 0, level => 1, key => "log_0433", variables => { pid => $pid }});
|
|
next;
|
|
}
|
|
|
|
# Read in the file.
|
|
my $body = $anvil->Storage->read_file({file => $smaps_path});
|
|
foreach my $line (split/\n/, $body)
|
|
{
|
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 3, list => { line => $line }});
|
|
if ($line =~ /^Private_Dirty:\s+(\d+) (.*B)$/)
|
|
{
|
|
my $size = $1;
|
|
my $type = $2;
|
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => {
|
|
type => $type,
|
|
size => $size,
|
|
}});
|
|
next if not $size;
|
|
next if $size =~ /\D/;
|
|
|
|
# This uses 'kB' for 'KiB' >_>
|
|
$type = lc($type);
|
|
$type =~ s/b$/ib/ if $type !~ /ib$/;
|
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => { type => $type }});
|
|
|
|
my $size_in_bytes = $anvil->Convert->human_readable_to_bytes({size => $size, type => $type, base2 => 1});
|
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => {
|
|
size_in_bytes => $anvil->Convert->add_commas({number => $size_in_bytes})." (".$anvil->Convert->bytes_to_human_readable({'bytes' => $size_in_bytes}).")",
|
|
}});
|
|
|
|
$anvil->data->{memory}{pid}{$pid} += $size_in_bytes;
|
|
$anvil->data->{memory}{total} += $size_in_bytes;
|
|
$anvil->Log->variables({source => $THIS_FILE, line => __LINE__, level => 2, list => {
|
|
"memory::pid::${pid}" => $anvil->Convert->add_commas({number => $anvil->data->{memory}{pid}{$pid}})." (".$anvil->Convert->bytes_to_human_readable({'bytes' => $anvil->data->{memory}{pid}{$pid}}).")",
|
|
'memory::total' => $anvil->Convert->add_commas({number => $anvil->data->{memory}{total}})." (".$anvil->Convert->bytes_to_human_readable({'bytes' => $anvil->data->{memory}{total}}).")",
|
|
}});
|
|
}
|
|
}
|
|
}
|
|
|
|
# Report and exit.
|
|
print $anvil->data->{switches}{program}." = ".$anvil->data->{memory}{total}." # ".$anvil->Convert->bytes_to_human_readable({'bytes' => $anvil->data->{memory}{total}})."\n";
|
|
$anvil->nice_exit({exit_code => 0});
|