assemblyline-service-client 4.6.1.dev136__py3-none-any.whl → 4.6.1.dev138__py3-none-any.whl

assemblyline_service_client/VERSION

@@ -1 +1 @@
-4.6.1.dev136
+4.6.1.dev138

assemblyline_service_client/task_handler.py

@@ -12,11 +12,13 @@ from typing import Any, Optional
 
 import requests
 import yaml
+from assemblyline_core.server_base import ServerBase
+
 from assemblyline.common.digests import get_sha256_for_file
 from assemblyline.common.str_utils import StringTable
+from assemblyline.odm import SHA256_REGEX
 from assemblyline.odm.messages.task import Task as ServiceTask
 from assemblyline.odm.models.service import Service
-from assemblyline_core.server_base import ServerBase
 
 STATUSES = StringTable('STATUSES', [
     ('INITIALIZING', 0),
@@ -365,6 +367,12 @@ class TaskHandler(ServerBase):
         return task
 
     def download_file(self, sha256, sid) -> Optional[str]:
+        if not SHA256_REGEX.match(sha256):
+            # If the SHA256 is not valid, we cannot download the file
+            self.log.error(f"[{sid}] Invalid SHA256 provided: {sha256}")
+            self.status = STATUSES.ERROR_FOUND
+            return None
+
         self.status = STATUSES.DOWNLOADING_FILE
         received_file_sha256 = ''
         file_path = None

{assemblyline_service_client-4.6.1.dev136.dist-info → assemblyline_service_client-4.6.1.dev138.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: assemblyline-service-client
-Version: 4.6.1.dev136
+Version: 4.6.1.dev138
 Summary: Assemblyline 4 - Service client
 Home-page: https://github.com/CybercentreCanada/assemblyline-service-client/
 Author: CCCS Assemblyline development team

assemblyline_service_client-4.6.1.dev138.dist-info/RECORD

@@ -0,0 +1,8 @@
+assemblyline_service_client/VERSION,sha256=u_nQXjGiaWrxuDT-5BHni2PZfcXepixHJLMu7qoQ7Js,13
+assemblyline_service_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+assemblyline_service_client/task_handler.py,sha256=k6RFFNIO9W6lukWF076Jps31UzXTJHE2VQZWMN0947Q,23848
+assemblyline_service_client-4.6.1.dev138.dist-info/licenses/LICENCE.md,sha256=NSkYo9EH8h5oOkzg4VhjAHF4339MqPP2cQ8msTPgl-c,1396
+assemblyline_service_client-4.6.1.dev138.dist-info/METADATA,sha256=NTHQFZYrILv8uZo6CnEc2BWLPA4fQXapS1e1qZ1jmvA,1677
+assemblyline_service_client-4.6.1.dev138.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+assemblyline_service_client-4.6.1.dev138.dist-info/top_level.txt,sha256=clNWHvn8nw0kR15l5TASxWxIQvKGKe5Pso3kVPMiJv0,28
+assemblyline_service_client-4.6.1.dev138.dist-info/RECORD,,

assemblyline_service_client-4.6.1.dev136.dist-info/RECORD

@@ -1,8 +0,0 @@
-assemblyline_service_client/VERSION,sha256=JKSD1AXttAaPMmok1BPbJjwg_UrmIcDeQa2FSw4q3Ak,13
-assemblyline_service_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-assemblyline_service_client/task_handler.py,sha256=XB2vSF-Zmff0veX6JOFqVPex7pNkxsZCNuhtY3LvRcM,23547
-assemblyline_service_client-4.6.1.dev136.dist-info/licenses/LICENCE.md,sha256=NSkYo9EH8h5oOkzg4VhjAHF4339MqPP2cQ8msTPgl-c,1396
-assemblyline_service_client-4.6.1.dev136.dist-info/METADATA,sha256=ZcZJzbU0o-YNUqmE1gShsiTqnHooOkLl9ctZWmEx1HU,1677
-assemblyline_service_client-4.6.1.dev136.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-assemblyline_service_client-4.6.1.dev136.dist-info/top_level.txt,sha256=clNWHvn8nw0kR15l5TASxWxIQvKGKe5Pso3kVPMiJv0,28
-assemblyline_service_client-4.6.1.dev136.dist-info/RECORD,,