lents_adutils 0.1.0__tar.gz

lents_adutils-0.1.0/PKG-INFO

@@ -0,0 +1,14 @@
+Metadata-Version: 2.4
+Name: lents_adutils
+Version: 0.1.0
+Summary: 
+Author: Antonio Lentini
+Author-email: 111151429+lentscode@users.noreply.github.com
+Requires-Python: >=3.14
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.14
+Requires-Dist: requests (>=2.34.2,<3.0.0)
+Description-Content-Type: text/markdown
+
+# Utils for Attack&Defense CTFs
+

lents_adutils-0.1.0/README.md

@@ -0,0 +1 @@
+# Utils for Attack&Defense CTFs

lents_adutils-0.1.0/pyproject.toml

@@ -0,0 +1,24 @@
+[project]
+name = "lents_adutils"
+version = "0.1.0"
+description = ""
+authors = [
+    {name = "Antonio Lentini",email = "111151429+lentscode@users.noreply.github.com"}
+]
+readme = "README.md"
+requires-python = ">=3.14"
+dependencies = [
+    "requests (>=2.34.2,<3.0.0)"
+]
+
+[tool.poetry]
+packages = [{include = "adutils", from = "src"}]
+
+[build-system]
+requires = ["poetry-core>=2.0.0,<3.0.0"]
+build-backend = "poetry.core.masonry.api"
+
+[dependency-groups]
+dev = [
+    "black (>=26.5.1,<27.0.0)"
+]

lents_adutils-0.1.0/src/adutils/__init__.py

@@ -0,0 +1 @@
+from adutils.exploit import Exploit

lents_adutils-0.1.0/src/adutils/exploit.py

@@ -0,0 +1,81 @@
+from abc import ABC, abstractmethod
+from concurrent.futures import ThreadPoolExecutor
+from requests import put, get
+from requests.exceptions import ConnectTimeout
+from time import sleep
+from json import loads
+
+
+class Exploit(ABC):
+    def __init__(
+        self,
+        port: int,
+        team_token: str,
+        n_teams: int = 80,
+        sleep_interval: int = 60,
+        ignore: list[int] = [],
+        verbose: bool = True,
+        submit_url: str = "http://10.10.0.1/flags",
+        submit_timeout: int = 5,
+        flag_ids_url: str = "http://10.10.0.1:8081/flagIds",
+    ):
+        self.port = port
+        self.team_token = team_token
+        self.n_teams = n_teams
+        self.sleep_interval = sleep_interval
+        self.ignore = ignore
+        self.flags: list[str] = []
+        self.verbose = verbose
+        self.submit_url = submit_url
+        self.submit_timeout = submit_timeout
+        self.flag_ids_url = flag_ids_url
+        self.flag_ids: dict = {}
+
+    def generate_ip(self, team_id: int) -> str:
+        return f"10.80.{team_id}.1"
+
+    def get_flag_ids(self):
+        response = get(self.flag_ids_url)
+        self.flag_ids = loads(response.content)
+
+    def flagout(self, flag: str):
+        if self.verbose:
+            print(f"Found flag = {flag}")
+        self.flags.append(flag)
+
+    @abstractmethod
+    def run(self, ip: str):
+        pass
+
+    def submit_flags(self):
+        try:
+            put(
+                self.submit_url,
+                headers={"X-Team-Token": self.team_token},
+                json=self.flags,
+                timeout=self.submit_timeout,
+            )
+        except ConnectTimeout as e:
+            print(f"[submit_flags] TIMEOUT: {e}")
+
+    def start(self):
+        while True:
+            self.get_flag_ids()
+            with ThreadPoolExecutor() as executor:
+                ips: list[str] = []
+                for i in range(self.n_teams):
+                    if i in self.ignore:
+                        continue
+
+                    ip = self.generate_ip(i)
+                    ips.append(ip)
+                list(executor.map(self.run, ips))
+
+            if self.verbose:
+                print(
+                    f"Submitting {len(self.flags)} flags to {self.submit_url} with token {self.team_token}"
+                )
+            self.submit_flags()
+            if self.verbose:
+                print(f"Sleeping for {self.sleep_interval} before next cycle")
+            sleep(self.sleep_interval)