Monitoring/ssh-wrapper-for-monitoring
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Added syslog logging and some command line arguments

Browse source 
-s:   Do NOT log to syslog
:     Log errors only
-v:   + executed command
-vv:  + matching regexp
-vvv: + command output & return code
This commit is contained in:
Sven Velt 2020-10-23 14:30:37 +02:00
parent be2ff6a2cd
commit f0a969d40f
1 changed files with 36 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
ssh-wrapper.py
View file

@ -1,10 +1,27 @@
#!/usr/bin/env python3
import argparse
import os
import re
import shlex
import subprocess
import sys
import syslog
parser = argparse.ArgumentParser()
parser.add_argument('--verbose', '-v', action='count', default=0)
parser.add_argument('--silent', action='store_true', default=False)
args = parser.parse_args()
if args.silent:
args.verbose = -1
if args.verbose >= 0:
syslog.openlog(
ident=sys.argv[0],
logoption=syslog.LOG_PID,
facility=syslog.LOG_LOCAL3 | syslog.LOG_ERR
)
allowed = [
##### System informations
@ -34,21 +51,36 @@ allowed = [
cmdline = os.getenv('SSH_ORIGINAL_COMMAND')
if not cmdline:
print('This is just a wrapper, no command specified!')
if args.verbose >= 0:
syslog.syslog('Called without SSH_ORIGINAL_COMMAND')
sys.exit(3)
for maybe in allowed:
if re.match(maybe, cmdline):
cmdline = shlex.split(cmdline)
if args.verbose >= 2:
syslog.syslog(syslog.LOG_INFO, 'Found command line >%s< with regexp >%s<' % ( cmdline, maybe ) )
cmdlinelist = shlex.split(cmdline)
try:
cmd = subprocess.Popen(cmdline, stdout=subprocess.PIPE)
cmd = subprocess.Popen(cmdlinelist, stdout=subprocess.PIPE)
except Exception as exc:
print('Could not execute plugin ("%s"): %s' % (' '.join(cmdline), exc))
print('Could not execute plugin ("%s"): %s' % cmdline)
if args.verbose >= 0:
syslog.syslog('Could not execute plugin >%s<' % cmdline)
sys.exit(3)
else:
print(cmd.communicate()[0].rstrip().decode('utf-8'))
(out, outerr) = cmd.communicate()
out = out.rstrip().decode('utf-8')
outerr = (outerr or b'').rstrip().decode('utf-8')
print(out)
if args.verbose >= 1:
syslog.syslog('Executed command line >%s<' % cmdline)
if args.verbose >= 3:
syslog.syslog('Output >%s<, Error >%s<' % (out, outerr))
sys.exit(cmd.returncode)
print('%s: No allowed command found!' % sys.argv[0])
if args.verbose >= 0:
syslog.syslog('No allowed command found for >%s<' % cmdline)
sys.exit(3)