feat: strip inputs

tdagent/tools/get_domain_information.py

@@ -71,7 +71,7 @@ def get_geolocation(ip: str) -> dict[str, Any] | str:
     """
     try:
         return requests.get(
-            f"https://geolocation-db.com/json/{ip}",
+            f"https://geolocation-db.com/json/{ip.strip()}",
             timeout=1,
         ).json()
     except Exception as e:  # noqa: BLE001
@@ -174,7 +174,7 @@ def enumerate_dns(domain_name: str) -> dict[str, Any] | None:
     enumeration = {}
     for record_type in _DNS_RECORD_TYPES:
         try:
-            record = _request_dns_record(domain_name, record_type, timeout=1)
+            record = _request_dns_record(domain_name.strip(), record_type, timeout=1)
             if record:
                 enumeration[record_type] = record
         except Exception as e:  # noqa: BLE001, PERF203
@@ -242,7 +242,9 @@ def scrap_subdomains_for_domain(domain_name: str) -> list[str]:
     except FileNotFoundError:
         return []
 
-    potential_subdomains = [f"{subdomain}.{domain_name}" for subdomain in subdomains]
+    potential_subdomains = [
+        f"{subdomain}.{domain_name.strip()}" for subdomain in subdomains
+    ]
     with ThreadPoolExecutor(max_workers=None) as executor:
         results = executor.map(resolve_subdomain_ipv4, potential_subdomains)
         return [domain for domain in results if domain]
@@ -311,7 +313,7 @@ def retrieve_ioc_from_threatfox(potentially_ioc: str) -> str:
     )
     data = {
         "query": "search_ioc",
-        "search_term": potentially_ioc,
+        "search_term": potentially_ioc.strip(),
     }
     json_data = json.dumps(data)
     try:

tdagent/tools/retrieve_from_mitre_attack.py

@@ -29,12 +29,15 @@ def get_stix_object_of_attack_id(
         or as a custom dictionary following the structure defined by the relevant
         Pydantic model, depending on the 'stix_format' flag.
     """
-    lift = attack_client()
-    return lift.get_object_by_attack_id(
-        object_type=object_type,
-        attack_id=attack_id,
-        stix_format=False,
-    )[0]
+    try:
+        lift = attack_client()
+        return lift.get_object_by_attack_id(
+            object_type=object_type.strip(),
+            attack_id=attack_id.strip(),
+            stix_format=False,
+        )[0]
+    except Exception as e:  # noqa: BLE001
+        return {"Exception": str(e)}
 
 
 gr_get_stix_of_attack_id = gr.Interface(