prelude-sdk 2.6.41__tar.gz → 2.6.43__tar.gz

{prelude_sdk-2.6.41 → prelude_sdk-2.6.43}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prelude-sdk
-Version: 2.6.41
+Version: 2.6.43
 Summary: For interacting with the Prelude API
 Home-page: https://github.com/preludeorg
 Author: Prelude Research

{prelude_sdk-2.6.41 → prelude_sdk-2.6.43}/prelude_sdk/controllers/build_controller.py

@@ -75,8 +75,8 @@ class BuildController(HttpController):
     @verify_credentials
     def upload(self, test_id, filename, data, skip_compile=False):
         """Upload a test or attachment"""
-        if len(data) > 1000000:
-            raise ValueError(f"File size must be under 1MB ({filename})")
+        if len(data) > 3145728:
+            raise ValueError(f"File size must be under 3MB ({filename})")
 
         h = self.account.headers | {"Content-Type": "application/octet-stream"}
         query_params = ""

{prelude_sdk-2.6.41 → prelude_sdk-2.6.43}/prelude_sdk/controllers/iam_controller.py

@@ -1,3 +1,7 @@
+import hashlib
+import json
+import os
+
 from prelude_sdk.controllers.http_controller import HttpController
 from prelude_sdk.models.account import verify_credentials
 from prelude_sdk.models.codes import Control, Mode, Permission
@@ -142,10 +146,22 @@ class IAMAccountController(HttpController):
     def sign_up(self, company, email, name):
         """(NOT AVAIABLE IN PRODUCTION) Create a new user and account"""
         body = dict(company=company, email=email, name=name)
+        headers = dict(_product="py-sdk")
+        if token := os.getenv("SDKTOKEN"):
+            sig = hashlib.sha256(
+                f"{token}{json.dumps(body)}".encode("utf-8")
+            ).hexdigest()
+            headers["x-sdk-sig"] = sig
 
         res = self._session.post(
-            f"{self.account.hq}/iam/new_user_and_account", json=body, timeout=20
+            f"{self.account.hq}/iam/new_user_and_account",
+            headers=headers,
+            json=body,
+            timeout=20,
         )
+        if not res.ok:
+            raise Exception(res.text)
+
         data = res.json()
         if self.account.profile:
             self.account.keychain.configure_keychain(

{prelude_sdk-2.6.41 → prelude_sdk-2.6.43}/prelude_sdk/controllers/probe_controller.py

@@ -8,5 +8,8 @@ class ProbeController(HttpController):
 
     def download(self, name: str, dos: str):
         """Download a probe executable"""
-        res = self.get(f"{self.account.hq}/download/{name}", headers=dict(dos=dos))
+        res = self.get(
+            f"{self.account.hq}/download/{name}",
+            headers=dict(dos=dos, _product="py-sdk"),
+        )
         return res.text

{prelude_sdk-2.6.41 → prelude_sdk-2.6.43}/prelude_sdk/models/codes.py

@@ -191,6 +191,7 @@ class Control(Enum, metaclass=MissingItem):
     CROWDSTRIKE_VULN = 31
     DEFENDER_VULN = 32
     NETSKOPE = 33
+    CUSTOM = 34
 
     @classmethod
     def _missing_(cls, value):
@@ -329,6 +330,13 @@ class ControlCategory(Enum, metaclass=MissingItem):
     def _missing_(cls, value):
         return ControlCategory.INVALID
 
+    @property
+    def scm_category(self):
+        for k, v in SCMCategory.category_mapping().items():
+            if self in v:
+                return k
+        return SCMCategory.NONE
+
     @classmethod
     def mapping(cls):
         return {

{prelude_sdk-2.6.41 → prelude_sdk-2.6.43}/prelude_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prelude-sdk
-Version: 2.6.41
+Version: 2.6.43
 Summary: For interacting with the Prelude API
 Home-page: https://github.com/preludeorg
 Author: Prelude Research

{prelude_sdk-2.6.41 → prelude_sdk-2.6.43}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = prelude-sdk
-version = 2.6.41
+version = 2.6.43
 author = Prelude Research
 author_email = support@preludesecurity.com
 description = For interacting with the Prelude API