From dd7f19990a067ecc76a643a4ded64e3bb2a4b880 Mon Sep 17 00:00:00 2001 From: Yan Date: Wed, 13 Nov 2024 11:47:34 -0700 Subject: [PATCH] fix library extrusion --- pwnshop/__main__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pwnshop/__main__.py b/pwnshop/__main__.py index 3753a02..09079aa 100644 --- a/pwnshop/__main__.py +++ b/pwnshop/__main__.py @@ -213,6 +213,8 @@ def handle_apply(args): if keep_source: shutil.copy2(challenge.src_path, os.path.join(out_dir, os.path.basename(challenge.src_path))) shutil.copy2(challenge.bin_path, os.path.join(out_dir, os.path.basename(challenge.bin_path))) + if os.path.exists(challenge.lib_path): + shutil.copytree(challenge.lib_path, os.path.join(out_dir, os.path.basename(challenge.lib_path)), dirs_exist_ok=True) #if pdb: # with open(f"{args.out.name.replace('.exe', '.pdb')}", 'wb') as f: