38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
import paramiko
|
|
import sys
|
|
|
|
sys.stdout.reconfigure(encoding='utf-8')
|
|
key_path = r'C:\Users\Snarf\.ssh\id_ed25519_snarfnet'
|
|
|
|
c = paramiko.SSHClient()
|
|
c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
|
c.connect("140.44.4.71", username="root", key_filename=key_path, timeout=10)
|
|
|
|
py_patch = """
|
|
path = '/opt/opencanary/alert_watcher.py'
|
|
with open(path, 'r') as f:
|
|
text = f.read()
|
|
|
|
target = 'def send_alert(event):'
|
|
replacement = '''def send_alert(event):
|
|
src_host = event.get("src_host", "")
|
|
src_port = event.get("src_port", -1)
|
|
# Ignore empty or internal startup events
|
|
if not src_host or src_port == -1:
|
|
return'''
|
|
|
|
if target in text and 'if not src_host or src_port == -1:' not in text:
|
|
text = text.replace(target, replacement)
|
|
with open(path, 'w') as f:
|
|
f.write(text)
|
|
print("Patched alert_watcher.py with filter")
|
|
else:
|
|
print("Filter already present or target not found")
|
|
"""
|
|
c.exec_command(f'python3 -c "{py_patch}"')
|
|
c.exec_command("systemctl restart opencanary-watcher.service")
|
|
print("Restarted watcher service")
|
|
|
|
c.close()
|
|
|