check_load: fix taint mode1.4

1 files changed, 6 insertions, 1 deletions

diff --git a/check_load b/check_load
index 1690872..b55ce38 100755
--- a/check_load
+++ b/check_load
@@ -6,7 +6,7 @@ use warnings;
 use VServer;
 use Nagios::Plugin;
 
-our $VERSION = '1.3';
+our $VERSION = '1.4';
 
 sub init_nagios_plugin() {
     my $plugin = Nagios::Plugin->new(
@@ -77,6 +77,11 @@ sub get_load($$) {
 }
 
 
+# fix taint mode
+if ($0 =~ m/(\w[\w\/]*\w)/) {
+    $ENV{'NAGIOS_PLUGIN'} = $1;
+}
+
 my $n = init_nagios_plugin();
 $n->getopts();
 alarm $n->opts->timeout;