adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

perf(sandbox): optimize code encryption handling

Browse Source
This commit is contained in:
Eternity
2026-01-26 12:22:54 +08:00
parent 0fd8a122fb
commit 1fc04c37d3
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
sandbox/app/core/encryption.py
View File

@@ -12,9 +12,10 @@ def encrypt_code(code: bytes, key: bytes) -> str:
Returns: Returns:
Base64 encoded encrypted code Base64 encoded encrypted code
""" """
key_length = len(key)
encrypted_code = bytearray(len(code)) encrypted_code = bytearray(len(code))
for i in range(len(code)): for i in range(len(code)):
encrypted_code[i] = code[i] ^ key[i % 64] encrypted_code[i] = code[i] ^ key[i % key_length]
encoded_code = base64.b64encode(encrypted_code).decode("utf-8") encoded_code = base64.b64encode(encrypted_code).decode("utf-8")
return encoded_code return encoded_code

2
sandbox/app/core/runners/python/prescript.py
View File

@@ -17,7 +17,7 @@ sys.excepthook = excepthook
# Load security library if available # Load security library if available
lib = ctypes.CDLL("./libpython.so") lib = ctypes.CDLL("./libpython.so")
lib.init_seccomp.argtypes = [ctypes.c_uint32, ctypes.c_uint32, ctypes.c_bool] lib.init_seccomp.argtypes = [ctypes.c_uint32, ctypes.c_uint32, ctypes.c_bool]
lib.init_seccomp.restype = None lib.init_seccomp.restype = None # TODO: raise error info
# Get running path # Get running path
running_path = sys.argv[1] running_path = sys.argv[1]