bashclub/check-opnsense
1
0
Fork 0
mirror of https://github.com/bashclub/check-opnsense.git synced 2025-11-04 05:02:28 +01:00
Code Issues Projects Releases Wiki Activity

Update opnsense_checkmk_agent.py

Browse Source 
v1.02
This commit is contained in:
Thorsten Spille
2023-05-10 16:52:08 +02:00
committed by GitHub
parent c4c8ea20d6
commit 74ea84fd8f
1 changed files with 19 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
opnsense_checkmk_agent.py
View File

@@ -27,7 +27,7 @@
## * smartdisk - install the mkp from https://github.com/bashclub/checkmk-smart plugins os-smart ## * smartdisk - install the mkp from https://github.com/bashclub/checkmk-smart plugins os-smart
## * squid - install the mkp from https://exchange.checkmk.com/p/squid and forwarder -> listen on loopback active ## * squid - install the mkp from https://exchange.checkmk.com/p/squid and forwarder -> listen on loopback active
__VERSION__ = "1.0.0" __VERSION__ = "1.0.2"
import sys import sys
import os import os
@@ -937,21 +937,18 @@ class checkmk_checker(object):
_con["local-id"] = _sas.get("local-id") _con["local-id"] = _sas.get("local-id")
_con["remote-id"] = _sas.get("remote-id") _con["remote-id"] = _sas.get("remote-id")
_sas_installed = False
if _sas.get("state") != "ESTABLISHED": if _sas.get("state") != "ESTABLISHED":
continue continue
_con["remote-host"] = _sas.get("remote-host") _con["remote-host"] = _sas.get("remote-host")
for _child in _sas.get("child-sas",{}).values(): for _child in _sas.get("child-sas",{}).values():
if _child.get("state") != "INSTALLED": if _child.get("state") != "INSTALLED":
continue continue
_sas_installed = True _phase2_up += 1
_install_time = max(1,int(_child.get("install-time","1"))) _install_time = max(1,int(_child.get("install-time","1")))
_con["bytes-received"] += int(int(_child.get("bytes-in","0")) /_install_time) _con["bytes-received"] += int(int(_child.get("bytes-in","0")) /_install_time)
_con["bytes-sent"] += int(int(_child.get("bytes-out","0")) /_install_time) _con["bytes-sent"] += int(int(_child.get("bytes-out","0")) /_install_time)
_con["life-time"] = max(_con["life-time"],_install_time) _con["life-time"] = max(_con["life-time"],_install_time)
_con["status"] = 0 if _con["status"] != 1 else 1 _con["status"] = 0 if _con["status"] != 1 else 1
if _sas_installed:
_phase2_up += 1
_required_phase2 = len(list(filter(lambda x: x.get("ikeid") == _ikeid,_phase2config))) _required_phase2 = len(list(filter(lambda x: x.get("ikeid") == _ikeid,_phase2config)))
@@ -1708,6 +1705,7 @@ if __name__ == "__main__":
_active_methods = [getattr(args,x,False) for x in ("start","stop","status","zabbix","nodaemon","debug","update","help")] _active_methods = [getattr(args,x,False) for x in ("start","stop","status","zabbix","nodaemon","debug","update","help")]
if SYSHOOK_METHOD and any(_active_methods) == False: if SYSHOOK_METHOD and any(_active_methods) == False:
log(f"using syshook {SYSHOOK_METHOD[0]}")
setattr(args,SYSHOOK_METHOD[0],True) setattr(args,SYSHOOK_METHOD[0],True)
if args.start: if args.start:
if _pid > 0: if _pid > 0:
@@ -1760,10 +1758,12 @@ if __name__ == "__main__":
if _github_req.status_code != 200: if _github_req.status_code != 200:
raise Exception("Github Error") raise Exception("Github Error")
_github_version = _github_req.json() _github_version = _github_req.json()
_github_last_modified = datetime.strptime(_github_req.headers.get("last-modified"),"%a, %d %b %Y %X %Z")
_new_script = base64.b64decode(_github_version.get("content")).decode("utf-8") _new_script = base64.b64decode(_github_version.get("content")).decode("utf-8")
_new_version = re.findall("^__VERSION__.*?\"([0-9.]*)\"",_new_script,re.M) _new_version = re.findall("^__VERSION__.*?\"([0-9.]*)\"",_new_script,re.M)
_new_version = _new_version[0] if _new_version else "0.0.0" _new_version = _new_version[0] if _new_version else "0.0.0"
_script_location = os.path.realpath(__file__) _script_location = os.path.realpath(__file__)
_current_last_modified = datetime.fromtimestamp(int(os.path.getmtime(_script_location)))
with (open(_script_location,"rb")) as _f: with (open(_script_location,"rb")) as _f:
_content = _f.read() _content = _f.read()
_current_sha = hashlib.sha1(f"blob {len(_content)}\0".encode("utf-8") + _content).hexdigest() _current_sha = hashlib.sha1(f"blob {len(_content)}\0".encode("utf-8") + _content).hexdigest()
@@ -1782,6 +1782,7 @@ if __name__ == "__main__":
try: try:
_answer = input(f"Update {_script_location} to {_new_version} (y/n) or show difference (d)? ") _answer = input(f"Update {_script_location} to {_new_version} (y/n) or show difference (d)? ")
except KeyboardInterrupt: except KeyboardInterrupt:
print("")
sys.exit(0) sys.exit(0)
if _answer in ("Y","y","yes","j","J"): if _answer in ("Y","y","yes","j","J"):
with open(_script_location,"wb") as _f: with open(_script_location,"wb") as _f:
@@ -1794,6 +1795,7 @@ if __name__ == "__main__":
try: try:
_answer = input(f"Daemon is running (pid:{_pid}), reload and restart (Y/N)? ") _answer = input(f"Daemon is running (pid:{_pid}), reload and restart (Y/N)? ")
except KeyboardInterrupt: except KeyboardInterrupt:
print("")
sys.exit(0) sys.exit(0)
if _answer in ("Y","y","yes","j","J"): if _answer in ("Y","y","yes","j","J"):
print("stopping Daemon") print("stopping Daemon")
@@ -1807,27 +1809,15 @@ if __name__ == "__main__":
pass pass
break break
elif _answer in ("D","d"): elif _answer in ("D","d"):
_matcher = difflib.Differ() for _line in difflib.unified_diff(_content.split("\n"),
_linenr = 0 _new_script.split("\n"),
_linenr_printed = False fromfile=f"Version: {__VERSION__}",
for _line in _matcher.compare(_new_script.split("\n"),_content.split("\n")): fromfiledate=_current_last_modified.isoformat(),
if _line.startswith("+"): tofile=f"Version: {_new_version}",
if not _linenr_printed: tofiledate=_github_last_modified.isoformat(),
print(f"@@ {_linenr}") n=1,
_linenr_printed = True lineterm=""):
print(_line) print(_line)
_linenr += 1
elif _line.startswith("-"):
if not _linenr_printed:
print(f"@@ {_linenr}")
_linenr_printed = True
print(_line)
elif _line.startswith("?"):
pass
else:
_linenr_printed = False
_linenr += 1
print(" ")
else: else:
break break
@@ -1857,6 +1847,7 @@ if __name__ == "__main__":
print("") print("")
else: else:
log("no arguments")
print("#"*35) print("#"*35)
print("checkmk_agent for opnsense") print("checkmk_agent for opnsense")
print(f"Version: {__VERSION__}") print(f"Version: {__VERSION__}")