Create temporary copy of cached file

Gallopsled · Feb 25, 2024 · b13f56b · b13f56b
1 parent 4c42d77
commit b13f56b
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pwnlib/asm.py b/pwnlib/asm.py
@@ -783,8 +783,13 @@ def asm(shellcode, vma = 0, extract = True, shared = False):
             if extract:
                 with open(cache_file, 'rb') as f:
                     return f.read()
-            else:
-                return cache_file
+
+            # Create a temporary copy of the cached file to avoid modification.
+            tmpdir = tempfile.mkdtemp(prefix = 'pwn-asm-')
+            atexit.register(shutil.rmtree, tmpdir)
+            step3 = os.path.join(tmpdir, 'step3')
+            shutil.copy(cache_file, step3)
+            return step3
 
     assembler = _assembler()
     linker    = _linker()