diff --git a/integrations/harfanglab/scripts/quarantine_file.py b/integrations/harfanglab/scripts/quarantine_file.py new file mode 100644 index 0000000..0b1d7c2 --- /dev/null +++ b/integrations/harfanglab/scripts/quarantine_file.py @@ -0,0 +1,38 @@ +import json, os, sys, urllib.request, urllib.parse, urllib.error + + +def request(method, url, headers, body=None): + data = json.dumps(body).encode("utf-8") if body is not None else None + req = urllib.request.Request(url, data=data, headers=headers, method=method) + with urllib.request.urlopen(req, timeout=60) as resp: + raw = resp.read() + return json.loads(raw) if raw else {} + + +def main(): + secrets = json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + inputs = json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + base = secrets.get("url", "").rstrip("/") + headers = { + "Authorization": "Token " + secrets.get("api_token", ""), + "Accept": "application/json", + "Content-Type": "application/json", + } + params = {"values": [{"file_path": inputs.get("file_path")}]} + if inputs.get("comment"): + params["comment"] = inputs.get("comment") + body = { + "targets": {"agents": [inputs.get("agent_id")]}, + "actions": [{"value": "quarantineAdd", "params": params}], + } + print(json.dumps(request("POST", base + "/api/data/Job/", headers, body))) + + +try: + main() +except urllib.error.HTTPError as e: + print(json.dumps({"error": "HTTP " + str(e.code), "detail": e.read().decode("utf-8", "replace")})) + sys.exit(1) +except Exception as e: + print(json.dumps({"error": str(e)})) + sys.exit(1) diff --git a/integrations/harfanglab/scripts/quarantine_request_upload.py b/integrations/harfanglab/scripts/quarantine_request_upload.py new file mode 100644 index 0000000..3de48fe --- /dev/null +++ b/integrations/harfanglab/scripts/quarantine_request_upload.py @@ -0,0 +1,33 @@ +import json, os, sys, urllib.request, urllib.parse, urllib.error + + +def request(method, url, headers, body=None): + data = json.dumps(body).encode("utf-8") if body is not None else None + req = urllib.request.Request(url, data=data, headers=headers, method=method) + with urllib.request.urlopen(req, timeout=60) as resp: + raw = resp.read() + return json.loads(raw) if raw else {} + + +def main(): + secrets = json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + inputs = json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + base = secrets.get("url", "").rstrip("/") + headers = { + "Authorization": "Token " + secrets.get("api_token", ""), + "Accept": "application/json", + "Content-Type": "application/json", + } + item_id = inputs.get("item_id") + url = base + "/api/data/quarantine/item/" + str(item_id) + "/request_upload/" + print(json.dumps(request("POST", url, headers))) + + +try: + main() +except urllib.error.HTTPError as e: + print(json.dumps({"error": "HTTP " + str(e.code), "detail": e.read().decode("utf-8", "replace")})) + sys.exit(1) +except Exception as e: + print(json.dumps({"error": str(e)})) + sys.exit(1) diff --git a/integrations/harfanglab/scripts/quarantine_search.py b/integrations/harfanglab/scripts/quarantine_search.py new file mode 100644 index 0000000..4dd5627 --- /dev/null +++ b/integrations/harfanglab/scripts/quarantine_search.py @@ -0,0 +1,57 @@ +import json, os, sys, urllib.request, urllib.parse, urllib.error + + +def request(method, url, headers, body=None): + data = json.dumps(body).encode("utf-8") if body is not None else None + req = urllib.request.Request(url, data=data, headers=headers, method=method) + with urllib.request.urlopen(req, timeout=60) as resp: + raw = resp.read() + return json.loads(raw) if raw else {} + + +def qs(params): + clean = {} + for k, v in params.items(): + if v is None or v == "" or v == []: + continue + if isinstance(v, bool): + v = "true" if v else "false" + clean[k] = v + return ("?" + urllib.parse.urlencode(clean)) if clean else "" + + +def main(): + secrets = json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + inputs = json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + base = secrets.get("url", "").rstrip("/") + headers = { + "Authorization": "Token " + secrets.get("api_token", ""), + "Accept": "application/json", + "Content-Type": "application/json", + } + params = { + "limit": int(inputs.get("limit") or 100), + "offset": int(inputs.get("offset") or 0), + "search": inputs.get("search"), + "agent.hostname": inputs.get("agent_hostname"), + "agent.id": inputs.get("agent_id"), + } + h = str(inputs.get("hash") or "").strip().lower() + if h: + key = {32: "item_md5", 40: "item_sha1", 64: "item_sha256"}.get(len(h)) + if not key: + print(json.dumps({"error": "unrecognized hash length (expected md5, sha1 or sha256)"})) + sys.exit(1) + params[key] = h + out = request("GET", base + "/api/data/quarantine/item/" + qs(params), headers) + print(json.dumps(out)) + + +try: + main() +except urllib.error.HTTPError as e: + print(json.dumps({"error": "HTTP " + str(e.code), "detail": e.read().decode("utf-8", "replace")})) + sys.exit(1) +except Exception as e: + print(json.dumps({"error": str(e)})) + sys.exit(1) diff --git a/integrations/harfanglab/scripts/unquarantine_file.py b/integrations/harfanglab/scripts/unquarantine_file.py new file mode 100644 index 0000000..39d2e48 --- /dev/null +++ b/integrations/harfanglab/scripts/unquarantine_file.py @@ -0,0 +1,38 @@ +import json, os, sys, urllib.request, urllib.parse, urllib.error + + +def request(method, url, headers, body=None): + data = json.dumps(body).encode("utf-8") if body is not None else None + req = urllib.request.Request(url, data=data, headers=headers, method=method) + with urllib.request.urlopen(req, timeout=60) as resp: + raw = resp.read() + return json.loads(raw) if raw else {} + + +def main(): + secrets = json.loads(os.environ.get("INTEGRATION_SECRETS", "{}")) + inputs = json.loads(os.environ.get("INTEGRATION_INPUTS", "{}")) + base = secrets.get("url", "").rstrip("/") + headers = { + "Authorization": "Token " + secrets.get("api_token", ""), + "Accept": "application/json", + "Content-Type": "application/json", + } + body = { + "targets": {"agents": [inputs.get("agent_id")]}, + "actions": [{"value": "quarantineRestore", "params": [{"values": [{ + "local_id": inputs.get("local_id"), + "overwrite_existing": bool(inputs.get("overwrite_existing", True)), + }]}]}], + } + print(json.dumps(request("POST", base + "/api/data/Job/", headers, body))) + + +try: + main() +except urllib.error.HTTPError as e: + print(json.dumps({"error": "HTTP " + str(e.code), "detail": e.read().decode("utf-8", "replace")})) + sys.exit(1) +except Exception as e: + print(json.dumps({"error": str(e)})) + sys.exit(1)