first sync

iris-web/source/app/iris_engine/reporter/ImageHandler.py

@@ -0,0 +1,64 @@
+#!/usr/bin/env python3
+#
+#  IRIS Source Code
+#  Copyright (C) 2021 - Airbus CyberSecurity (SAS)
+#  contact@dfir-iris.org
+#  Created by Lukas Zurschmiede @LukyLuke
+#
+#  This program is free software; you can redistribute it and/or
+#  modify it under the terms of the GNU Lesser General Public
+#  License as published by the Free Software Foundation; either
+#  version 3 of the License, or (at your option) any later version.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+#  Lesser General Public License for more details.
+#
+#  You should have received a copy of the GNU Lesser General Public License
+#  along with this program; if not, write to the Free Software Foundation,
+#  Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+import logging
+import os
+import shutil
+import uuid
+import re
+
+from pathlib import Path
+from docxtpl import DocxTemplate
+
+from docx_generator.globals.picture_globals import PictureGlobals
+
+from app.datamgmt.datastore.datastore_db import datastore_get_local_file_path
+
+
+class ImageHandler(PictureGlobals):
+    def __init__(self, template: DocxTemplate, base_path: str):
+        self._logger = logging.getLogger(__name__)
+        PictureGlobals.__init__(self, template, base_path)
+
+    def _process_remote(self, image_path: str) -> str:
+        """
+        Checks if the given Link is a datastore-link and if so, save the image locally for further processing.
+        :
+        A Datastore Links looks like this: https://localhost:4433/datastore/file/view/2?cid=1
+        """
+        res = re.search(r'datastore\/file\/view\/(\d+)\?cid=(\d+)', image_path)
+        if not res:
+            return super()._process_remote(image_path)
+
+        if image_path[:4] == 'http' and len(res.groups()) == 2:
+            file_id = res.groups(0)[0]
+            case_id = res.groups(0)[1]
+            has_error, dsf = datastore_get_local_file_path(file_id, case_id)
+
+            if has_error:
+                raise RenderingError(self._logger, f'File-ID {file_id} does not exist in Case {case_id}')
+            if not Path(dsf.file_local_name).is_file():
+                raise RenderingError(self._logger, f'File {dsf.file_local_name} does not exists on the server. Update or delete virtual entry')
+
+            file_ext = os.path.splitext(dsf.file_original_name)[1]
+            file_name = os.path.join(self._output_path, str(uuid.uuid4())) + file_ext
+            return_value = shutil.copy(dsf.file_local_name, file_name)
+            return return_value
+        return super()._process_remote(image_path)