From ef65b92b9a61ce5ba55bbbbb9fa23e7fd32224eb Mon Sep 17 00:00:00 2001
From: Dobin <dobin.rutishauser@gmail.com>
Date: Sun, 18 Feb 2024 11:45:07 +0000
Subject: [PATCH] fix: close PE's

---
 pehelper.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pehelper.py b/pehelper.py
index 85a35e7..e3f8dbb 100644
--- a/pehelper.py
+++ b/pehelper.py
@@ -18,6 +18,7 @@ def extract_code_from_exe(exe_file: FilePath) -> bytes:
     logger.info("    > 0x{:X} Code Size: {}  (raw code section size: {})".format(
         section.VirtualAddress,
         len(data), section.SizeOfRawData))
+    pe.close()
     return data
 
 
@@ -28,7 +29,7 @@ def write_code_section(exe_file: FilePath, new_data: bytes):
     with open(exe_file, 'r+b') as f:
         f.seek(file_offset)
         f.write(new_data)
-
+    pe.close()
 
 def get_code_section(pe: pefile.PE) -> pefile.SectionStructure:
     entrypoint = pe.OPTIONAL_HEADER.AddressOfEntryPoint