Return path to cache file if not extract

Gallopsled · Feb 25, 2024 · 5724c39 · 5724c39
1 parent 3496d98
commit 5724c39
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pwnlib/asm.py b/pwnlib/asm.py
@@ -780,8 +780,11 @@ def asm(shellcode, vma = 0, extract = True, shared = False):
         asm_hash = sha1sumhex(_encode(shellcode) + _encode(hash_params))
         cache_file = os.path.join(cache_dir, asm_hash)
         if os.path.exists(cache_file):
-            with open(cache_file, 'rb') as f:
-                return f.read()
+            if extract:
+                with open(cache_file, 'rb') as f:
+                    return f.read()
+            else:
+                return cache_file
 
     assembler = _assembler()
     linker    = _linker()