change shm name format

HXSecurity · Jun 6, 2022 · 5cc1083 · 5cc1083
1 parent 6cf2eb2
commit 5cc1083
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/dongtai_agent_python/setting/setting.py b/dongtai_agent_python/setting/setting.py
@@ -87,7 +87,7 @@ def init_os_environ(self):
 
     def set_shm(self, name):
         if self.shm is None:
-            self.shm = SharedMemoryDict(name)
+            self.shm = SharedMemoryDict('dongtai-shm-python-' + name)
 
     @property
     def paused(self):

diff --git a/dongtai_agent_python/utils/shm/shm.py b/dongtai_agent_python/utils/shm/shm.py
@@ -26,11 +26,10 @@ def __init__(self, name, size=1024):
 
     @lock(_lock)
     def get_or_create(self, size):
-        name = 'dongtai-shm-' + self.name
         try:
-            return SharedMemory(name=name)
+            return SharedMemory(name=self.name)
         except FileNotFoundError:
-            return SharedMemory(name=name, create=True, size=size)
+            return SharedMemory(name=self.name, create=True, size=size)
 
     def init_memory(self):
         memory_is_empty = (bytes(self.mem_block.buf).split(NULL_BYTE, 1)[0] == b'')