ansys-pyensight-core 0.9.2__py3-none-any.whl → 0.9.3__py3-none-any.whl

ansys/pyensight/core/__init__.py

@@ -5,7 +5,7 @@ except ModuleNotFoundError:  # pragma: no cover
 __version__ = importlib_metadata.version(__name__.replace(".", "-"))
 
 VERSION = __version__
-DEFAULT_ANSYS_VERSION = "242"
+DEFAULT_ANSYS_VERSION = "251"
 
 # Ansys version number that this release is associated with
 __ansys_version__ = DEFAULT_ANSYS_VERSION

ansys/pyensight/core/common.py

@@ -1,5 +1,6 @@
 """ This module provides a list of common utilities shared between different PyEnSight modules."""
 
+import os
 import random
 import re
 import socket
@@ -214,3 +215,19 @@ def pull_image(docker_client: "DockerClient", image_name: str) -> None:
             docker_client.images.pull(image_name)
     except Exception:
         raise RuntimeError(f"Can't pull Docker image: {image_name}")
+
+
+def _is_within_directory(directory, target):
+    """Check if target is inside of the input directory."""
+    abs_directory = os.path.abspath(directory)
+    abs_target = os.path.abspath(target)
+    return os.path.commonprefix([abs_directory, abs_target]) == abs_directory
+
+
+def safe_extract(tar, path=".", members=None, *, numeric_owner=False):
+    """Utility to check tar extraction to avoid bandit check issue."""
+    for member in tar.getmembers():
+        member_path = os.path.join(path, member.name)
+        if not _is_within_directory(path, member_path):
+            raise Exception("Attempted Path Traversal in Tar File")
+    tar.extractall(path, members, numeric_owner=numeric_owner)

ansys/pyensight/core/dockerlauncher.py

@@ -325,6 +325,7 @@ class DockerLauncher(Launcher):
                         detach=True,
                         auto_remove=True,
                         remove=True,
+                        user="ensight:ensight",
                     )
             else:
                 logging.debug(f"Running container {self._image_name} with cmd {enshell_cmd}\n")
@@ -340,6 +341,7 @@ class DockerLauncher(Launcher):
                         detach=True,
                         auto_remove=True,
                         remove=True,
+                        user="ensight:ensight",
                     )
                 logging.debug(f"_container = {str(self._container)}\n")
         else:
@@ -357,6 +359,7 @@ class DockerLauncher(Launcher):
                         detach=True,
                         auto_remove=True,
                         remove=True,
+                        user="ensight:ensight",
                     )
             else:  # pragma: no cover
                 logging.debug(
@@ -373,6 +376,7 @@ class DockerLauncher(Launcher):
                         detach=True,
                         auto_remove=True,
                         remove=True,
+                        user="ensight:ensight",
                     )
                 # logging.debug(f"_container = {str(self._container)}\n")
         logging.debug("Container started.\n")