feat(feed): add AWS IP ranges feed connector (allowlist / known infra)

2026-07-13 12:08:48 +02:00
parent f6ff018f09
commit c42a9cef03
3 changed files with 171 additions and 0 deletions
@@ -0,0 +1,79 @@
+import json, os, sys, ssl, urllib.request, urllib.error
+
+URL = "https://ip-ranges.amazonaws.com/ip-ranges.json"
+
+
+def _cfg():
+    return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}"))
+
+
+def _inputs():
+    return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}"))
+
+
+def _ctx(cfg):
+    if cfg.get("insecure"):
+        c = ssl.create_default_context()
+        c.check_hostname = False
+        c.verify_mode = ssl.CERT_NONE
+        return c
+    return None
+
+
+def _get(url, cfg):
+    req = urllib.request.Request(
+        url, headers={"User-Agent": "Riposte-SOAR", "Accept": "application/json"}
+    )
+    with urllib.request.urlopen(req, timeout=120, context=_ctx(cfg)) as r:
+        return r.read()
+
+
+def _run(fn):
+    try:
+        print(json.dumps(fn(_cfg(), _inputs())))
+    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)
+
+
+def main(cfg, inputs):
+    region = inputs.get("region") or cfg.get("region")
+    service = inputs.get("service") or cfg.get("service")
+    raw = _get(URL, cfg)
+    data = json.loads(raw) if raw else {}
+
+    maxn = int(inputs.get("max_indicators") or 0)
+    out = []
+
+    def add(prefixes, field):
+        for p in prefixes:
+            if not isinstance(p, dict):
+                continue
+            cidr = p.get(field)
+            if not cidr:
+                continue
+            if region and p.get("region") != region:
+                continue
+            if service and p.get("service") != service:
+                continue
+            out.append({
+                "value": cidr,
+                "type": "cidr",
+                "region": p.get("region"),
+                "service": p.get("service"),
+                "provider": "aws",
+            })
+            if maxn and len(out) >= maxn:
+                return True
+        return False
+
+    if not add(data.get("prefixes", []), "ip_prefix"):
+        add(data.get("ipv6_prefixes", []), "ipv6_prefix")
+
+    return {"source": "aws:ip-ranges", "count": len(out), "indicators": out}
+
+
+_run(main)
@@ -0,0 +1,49 @@
+import json, os, sys, ssl, urllib.request, urllib.error
+
+URL = "https://ip-ranges.amazonaws.com/ip-ranges.json"
+
+
+def _cfg():
+    return json.loads(os.environ.get("INTEGRATION_SECRETS", "{}"))
+
+
+def _inputs():
+    return json.loads(os.environ.get("INTEGRATION_INPUTS", "{}"))
+
+
+def _ctx(cfg):
+    if cfg.get("insecure"):
+        c = ssl.create_default_context()
+        c.check_hostname = False
+        c.verify_mode = ssl.CERT_NONE
+        return c
+    return None
+
+
+def _get(url, cfg):
+    req = urllib.request.Request(
+        url, headers={"User-Agent": "Riposte-SOAR", "Accept": "application/json"}
+    )
+    with urllib.request.urlopen(req, timeout=60, context=_ctx(cfg)) as r:
+        return r.read()
+
+
+def _run(fn):
+    try:
+        print(json.dumps(fn(_cfg(), _inputs())))
+    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)
+
+
+def main(cfg, inputs):
+    raw = _get(URL, cfg)
+    data = json.loads(raw) if raw else {}
+    n = len(data.get("prefixes", [])) if isinstance(data, dict) else 0
+    return {"ok": True, "sample_count": n, "create_date": data.get("createDate")}
+
+
+_run(main)