loadmon.pl: prints out memory usage and file handle usage for lanforge related processes

Signed-off-by: Jed Reynolds <jed@bitratchet.com>
2025-11-03 04:07:52 +00:00 · 2022-05-09 22:37:56 -07:00
parent abe3255297
commit 8f62543df8
1 changed files with 163 additions and 0 deletions
--- a/loadmon.pl
+++ b/loadmon.pl
@@ -0,0 +1,163 @@
+#!/usr/bin/perl -w
+#
+# Log the traffic generating processing processes that LANforge creates
+# This script will increase system logging usage (journalctl via logger)
+# so if you need to add runtime constraints for disk usage, please add these
+# runtime settings to you /etc/systemd/journald.conf file:
+#   SystemMaxUse=100M
+#   RuntimeMaxUse=1024M
+#   RuntimeKeepFree=500M
+#   RuntimeMaxFiles=5
+#   RuntimeMaxFileSize=256M
+
+use diagnostics;
+use warnings;
+use strict;
+#use Time::localtime;
+use POSIX;
+use Data::Dumper;
+
+package main;
+
+our $QQ=q{"};
+our $Q=q{'};
+our $LC=q({);
+our $RC=q(});
+
+our @prog_names = (
+    "btserver",
+    "curl",
+    "dhclient",
+    "dnsmasq",
+    "hostapd",
+    "l4helper",
+    "logchopper",
+    "nginx",
+    "perl",
+    "pipe_helper",
+    "wget",
+    "wpa_cli",
+    "wpa_supplicant",
+);
+our %monitor_map = ();
+
+## - - - Define loadmon - - - ##
+package loadmon;
+sub new {
+    my $class = shift;
+    my ( $basename ) = @_;
+    my $self = {
+        basename => $basename,
+        ra_pid_list => [],
+        total_mem => 0,
+        total_fh => 0,
+    };
+    bless $self, $class;
+    return $self;
+}
+
+sub monitor {
+    my $self = shift;
+    $self->{total_mem} = 0;
+    $self->{total_fh} = 0;
+
+    my $cmd = qq(pgrep -f $self->{basename});
+    # print "CMD[$cmd]\n";
+    my @lines = `$cmd`;
+    chomp(@lines);
+    # print Data::Dumper->Dump(\@lines);
+    $self->{ra_pid_list} = [];
+    splice @{$self->{ra_pid_list}}, 0, 0, @lines;
+    if ( scalar( @{$self->{ra_pid_list}} ) < 1) {
+        return;
+    }
+    # print Data::Dumper->Dump(['ra_pid_list', $self->{ra_pid_list}] ), "\n";
+    my $pidlist = join(" ", @{$self->{ra_pid_list}});
+    $cmd = qq(echo $pidlist | xargs ps -o rss -p | tail -n+2);
+    # print "CMD2: $cmd\n";
+    my @mem_lines=`$cmd`;
+    chomp(@mem_lines);
+    #print Data::Dumper->Dump(['mem_lines', \@mem_lines]), "\n";
+    for my $mem (@mem_lines) {
+        $self->{total_mem} += int($mem);
+    }
+
+    for my $pid (@{$self->{ra_pid_list}}) {
+        next unless ( -d "/proc/$pid/fd" );
+        $cmd = "ls /proc/$pid/fd | wc -l";
+        @lines=`$cmd`;
+        chomp(@lines);
+        # print Data::Dumper->Dump(['fh_lines', \@lines]), "\n";
+        if (@lines > 0 ) {
+            $self->{total_fh} += int($lines[0]);
+        }
+    }
+
+    #die("testing");
+}
+
+sub report {
+    my ($self, $fh) = @_;
+    my $num_pids = 0 + @{$self->{ra_pid_list}};
+
+    # print Data::Dumper->Dump(['report', $self] ), "\n";
+    if ($num_pids < 1) {
+        return;
+    }
+    if (!$fh) {
+        $fh = *STDOUT;
+    }
+    print $fh qq(${LC}"basename":"$self->{basename}",);
+    print $fh qq("num_pids":$num_pids,);
+    print $fh qq("total_mem_KB":$self->{total_mem},);
+    print $fh qq("total_fh":$self->{total_fh}${RC});
+}
+1;
+## - - - End loadmon - - - ##
+
+## - - - Define main - - - ##
+package main;
+
+sub print_totals {
+    my $fh = shift;
+    if (!$fh) {
+        $fh = *STDOUT;
+    }
+    my $tt_num_pids = 0;
+    my $tt_mem_kb = 0;
+    my $tt_fh = 0;
+    for my $name (@main::prog_names) {
+        my $monitor = $main::monitor_map{$name};
+        #print Data::Dumper->Dump(["mm_name", $monitor ]);
+        my $ra_pl = $monitor->{ra_pid_list};
+        #print Data::Dumper->Dump(["ra_pl", $ra_pl]);
+        $tt_num_pids += @$ra_pl;
+        if (defined $main::monitor_map{$name}->{total_mem_KB}) {
+            $tt_mem_kb += $main::monitor_map{$name}->{total_mem_KB};
+        }
+        $tt_fh += $main::monitor_map{$name}->{total_fh};
+    }
+    print $fh qq(${LC}"tt_num_pids":$tt_num_pids, "tt_mem_kb":$tt_mem_kb, "tt_fh":$tt_fh${RC});
+}
+
+## - - -
+#           M A I N
+## - - -
+for my $name (@main::prog_names) {
+    $monitor_map{$name} = loadmon->new($name);
+}
+
+
+while (1) {
+    print STDOUT '{';
+    for my $name (@main::prog_names) {
+        my $lmonitor = $monitor_map{$name};
+        # print "$name ";
+        $lmonitor->monitor();
+        $lmonitor->report(*STDOUT);
+    }
+    print_totals(*STDOUT);
+    print "}\n";
+    sleep(60);
+}
+#