tyler.durden/pandorafms
1
0
Fork 0
mirror of https://github.com/pandorafms/pandorafms.git synced 2025-07-30 17:25:26 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'ent-9741-autodiscover-is-enabled-in-the-default-config-even-though-autodiscover-is-not-operating' into 'develop'

Browse Source 
ent-9741-autodiscover-is-enabled-in-the-default-config-even-though-autodiscover-is-not-operating

See merge request artica/pandorafms!5275
This commit is contained in:
Daniel Rodriguez 2022-12-05 15:29:24 +00:00
commit 66b0e7bab8
1 changed files with 56 additions and 55 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

111
pandora_agents/unix/plugins/autodiscover
View File

@ -12,32 +12,34 @@
# #
################################################### ###################################################
from sys import argv try:
from sys import path from sys import argv
from sys import stderr from sys import stderr
from sys import exit from sys import exit
from subprocess import Popen from subprocess import Popen
from subprocess import PIPE from subprocess import PIPE
from subprocess import DEVNULL from subprocess import DEVNULL
from subprocess import getstatusoutput from subprocess import getstatusoutput
import psutil import psutil
except ModuleNotFoundError as err:
print("{} error: {}. Exiting...".format(argv[0], err), file=stderr)
exit(1)
global module_list
module_list = [] module_list = []
version = "1.1" VERSION = "1.2"
######################################################################################### #########################################################################################
# Powershell class # Powershell class
######################################################################################### #########################################################################################
class PSCheck: class PSCheck:
@staticmethod @staticmethod
def check_service(servicename, option=False, memcpu=False): def check_service(servicename, option=False, memcpu=False):
"""Check services with powershell by parsing their DisplayName. Returns a dict\ """Check services with powershell by parsing their DisplayName. Returns a dict\
list with the name of the service and a boolean with its status.\n list with the name of the service and a boolean with its status.\n
Requires service name (case insensitive).""" Requires service name (case insensitive)."""
pscall = Popen(["powershell", "Get-Service", "-Name", "'*"+ str(servicename) + "*'", pscall = Popen(["powershell", "Get-Service", "-Name", "'*"+ str(servicename) + "*'",
"|", "Select-Object", "-ExpandProperty", "Name"], "|", "Select-Object", "-ExpandProperty", "Name"],
stdout=PIPE, stdin=DEVNULL, stderr=DEVNULL, universal_newlines=True) stdout=PIPE, stdin=DEVNULL, stderr=DEVNULL, universal_newlines=True)
result = pscall.communicate() result = pscall.communicate()
result = str(result[0]).strip().split("\n") result = str(result[0]).strip().split("\n")
@ -50,15 +52,15 @@ class PSCheck:
procname = PSCheck.get_serviceprocess(element) procname = PSCheck.get_serviceprocess(element)
# Get process status # Get process status
parstatus = PSCheck.getstatus(element) parstatus = PSCheck.getstatus(element)
if memcpu == True and parstatus == 1: if memcpu and parstatus == 1:
usage = get_memcpu(str(procname), str(element)) usage = get_memcpu(str(procname), str(element))
output += usage output += usage
# Generate module with name and status # Generate module with name and status
parent = service_module(str(element), parstatus) parent = service_module(str(element), parstatus)
output += parent output += parent
if option == True: if option:
children = PSCheck.getchildren(element, memcpu) children = PSCheck.getchildren(element, memcpu)
if type(children) == list and len(children) > 1: if isinstance(children, list) and len(children) > 1:
for child in children: for child in children:
output += child output += child
else: else:
@ -66,7 +68,6 @@ class PSCheck:
else: else:
next next
#if output != '':
if output and element and procname: if output and element and procname:
return ({"name" : element, "process" : procname, "modules": output}) return ({"name" : element, "process" : procname, "modules": output})
else: else:
@ -84,22 +85,22 @@ class PSCheck:
status = PSCheck.getstatus(child) status = PSCheck.getstatus(child)
kids += service_module(str(child), status, "Service " + str(servicename) + " - Status") kids += service_module(str(child), status, "Service " + str(servicename) + " - Status")
if status: if status:
if memcpu == True: if memcpu:
kidsusage = get_memcpu(str(child)) kidsusage = get_memcpu(str(child))
for usage in kidsusage: for usage in kidsusage:
kids += usage kids += usage
else: else:
next next
return (kids) return kids
@staticmethod @staticmethod
def getstatus(servicename): def getstatus(servicename):
"""Gets the status of a given Windows service""" """Gets the status of a given Windows service"""
running = Popen(["powershell", "Get-Service", "-Name '" + str(servicename) + running = Popen(["powershell", "Get-Service", "-Name '" + str(servicename) +
"' |", "Select-Object", "-ExpandProperty", "Status"], "' |", "Select-Object", "-ExpandProperty", "Status"],
stdout=PIPE, stdin=DEVNULL, stderr=DEVNULL, universal_newlines=True) stdout=PIPE, stdin=DEVNULL, stderr=DEVNULL, universal_newlines=True)
status = running.communicate()[0].strip() status = running.communicate()[0].strip()
return (int(status == "Running")) return int(status == "Running")
@staticmethod @staticmethod
def get_serviceprocess(servicename): def get_serviceprocess(servicename):
@ -108,7 +109,7 @@ class PSCheck:
srv_pid = service.pid() srv_pid = service.pid()
process = psutil.Process(srv_pid) process = psutil.Process(srv_pid)
proc_name = process.name() proc_name = process.name()
return (proc_name) return proc_name
######################################################################################### #########################################################################################
@ -124,13 +125,13 @@ def service_module(name, value, parent=None):
"module_parent" : parent, "module_parent" : parent,
}] }]
#print ("service_module END "+str(now(0,1))) #print ("service_module END "+str(now(0,1)))
return (module) return module
def get_memcpu (process, servicename): def get_memcpu (process, servicename=None):
"""Creates a module for Memory and CPU for a given process. Returns a list of dictionaries.""" """Creates a module for Memory and CPU for a given process. Returns a list of dictionaries."""
modules = [] modules = []
if process: if process:
if servicename != None: if servicename is not None:
parentname = servicename parentname = servicename
else: else:
parentname = process parentname = process
@ -147,7 +148,7 @@ def get_memcpu (process, servicename):
"unit" : "%", "unit" : "%",
"module_parent" : "Service "+ parentname + " - Status", "module_parent" : "Service "+ parentname + " - Status",
}] }]
return (modules) return modules
def proc_percentbyname(procname): ############# 03/03/2020 def proc_percentbyname(procname): ############# 03/03/2020
"""Gets Memory and CPU usage for a given process. Returns a list.""" """Gets Memory and CPU usage for a given process. Returns a list."""
@ -165,7 +166,7 @@ def proc_percentbyname(procname): ############# 03/03/2020
except psutil.NoSuchProcess: except psutil.NoSuchProcess:
next next
#print ("proc_percentbyname END "+str(now(0,1))) #print ("proc_percentbyname END "+str(now(0,1)))
return ([sum(memory),sum(cpu)]) return [sum(memory),sum(cpu)]
def win_service(servicelist, option=False, memcpu=False): def win_service(servicelist, option=False, memcpu=False):
"""Creates modules for Windows servers.""" """Creates modules for Windows servers."""
@ -173,7 +174,7 @@ def win_service(servicelist, option=False, memcpu=False):
for srvc in servicelist: for srvc in servicelist:
if srvc and len(srvc) > 2: if srvc and len(srvc) > 2:
output = PSCheck.check_service(srvc, option, memcpu) output = PSCheck.check_service(srvc, option, memcpu)
if output != None and output["modules"]: if output is not None and output["modules"]:
modules += PSCheck.check_service(srvc.strip(), option, memcpu)["modules"] modules += PSCheck.check_service(srvc.strip(), option, memcpu)["modules"]
module_list.append(srvc) module_list.append(srvc)
#winprocess = output["name"] #winprocess = output["name"]
@ -197,9 +198,9 @@ def lnx_service(services_list, memcpu=False):
if sysctl == 0: if sysctl == 0:
### Systemd available ### Systemd available
syscall = Popen(["systemctl", "show", "-pLoadState", "-pActiveState", srvc], stdout=PIPE, syscall = Popen(["systemctl", "show", "-pLoadState", "-pActiveState", srvc], stdout=PIPE,
stdin=DEVNULL, universal_newlines=True) stdin=DEVNULL, universal_newlines=True)
result = syscall.communicate() result = syscall.communicate()
srvstatus= result[0].strip().lower().split("\n") srvstatus = result[0].strip().lower().split("\n")
if srvstatus[0] == "loadstate=not-found": if srvstatus[0] == "loadstate=not-found":
next next
else: else:
@ -212,7 +213,7 @@ def lnx_service(services_list, memcpu=False):
elif sysctl != 0 and servic == 0: elif sysctl != 0 and servic == 0:
### Systemd not available, switch to service command ### Systemd not available, switch to service command
syscall = Popen(["service", srvc, "status"], stdout=PIPE, syscall = Popen(["service", srvc, "status"], stdout=PIPE,
stdin=DEVNULL, stderr=DEVNULL, universal_newlines=True) stdin=DEVNULL, stderr=DEVNULL, universal_newlines=True)
result = syscall.communicate()[0].lower() result = syscall.communicate()[0].lower()
if "is running" in result: if "is running" in result:
modules += service_module(srvc, 1) modules += service_module(srvc, 1)
@ -223,15 +224,15 @@ def lnx_service(services_list, memcpu=False):
else: else:
next next
else: else:
print ("No systemd or service commands available. Exiting...", file=stderr) print("No systemd or service commands available. Exiting...", file=stderr)
exit() exit()
if status: if status:
module_list.append(srvc) module_list.append(srvc)
if memcpu == True: if memcpu:
modules += get_memcpu(srvc, None) modules += get_memcpu(srvc, None)
for m in modules: for m in modules:
print_module (m, 1) print_module(m, 1)
######################################################################################### #########################################################################################
@ -339,7 +340,7 @@ def print_module(module, str_flag=False):
#### Print flag #### Print flag
if str_flag is not False: if str_flag is not False:
print (module_xml) print(module_xml)
return (module_xml) return (module_xml)
@ -362,7 +363,7 @@ def main():
"postfix", "mysqld", "postgres", "oracle", "mongod"] "postfix", "mysqld", "postgres", "oracle", "mongod"]
discover(OS, service_list) discover(OS, service_list)
else: else:
print ("OS not recognized. Exiting...", file=stderr) print("OS not recognized. Exiting...", file=stderr)
exit() exit()
def discover(osyst, servicelist): def discover(osyst, servicelist):
@ -383,24 +384,24 @@ def discover(osyst, servicelist):
elif osyst == "Linux": elif osyst == "Linux":
lnx_service(servicelist, memcpu) lnx_service(servicelist, memcpu)
else: else:
print ("\nPandora FMS Autodiscovery plugin v{}".format(version)) print("\nPandora FMS Autodiscovery plugin v{}".format(VERSION))
print ("Checks the status of the services in list and monitors CPU and Memory for each of them.\n") print("Checks the status of the services in list and monitors CPU and Memory for each of them.\n")
print ("Usage:") print("Usage:")
print ("{} [options] [--usage]".format(argv[0])) print("{} [options] [--usage]".format(argv[0]))
print ("--help") print("--help")
print ("\tPrints this help screen") print("\tPrints this help screen")
print ("--default") print("--default")
print ("\tRuns this tool with default monitoring.") print("\tRuns this tool with default monitoring.")
print ("\tServices monitored by default for {}:".format(osyst)) print("\tServices monitored by default for {}:".format(osyst))
print ("\t",", ".join(servicelist)) print("\t", ", ".join(servicelist))
print ("--list \"<srvc1,srvc2,srvc3>\"") print("--list \"<srvc1,srvc2,srvc3>\"")
print ("\tReplaces default services for a given list (comma-separated)") print("\tReplaces default services for a given list (comma-separated)")
if osyst == "Windows": if osyst == "Windows":
print ("\tEach element of the list will be treated as a regexp, but they must be over 2 characters.") print("\tEach element of the list will be treated as a regexp, but they must be over 2 characters.")
print ("\tElements under 2 characters will be discarded.") print("\tElements under 2 characters will be discarded.")
print ("--usage") print("--usage")
print ("\tAdds modules for CPU and Memory usage per service/process (optional, can take some time).\n") print("\tAdds modules for CPU and Memory usage per service/process (optional, can take some time).\n")
##### RUN #### ##### RUN ####
main() main()