Do not truncate the ELF file when saving to the same path

Gallopsled · Jan 20, 2017 · 31cd98a · 31cd98a
1 parent 3e4a1bc
commit 31cd98a
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/pwnlib/elf/elf.py b/pwnlib/elf/elf.py
@@ -874,14 +874,8 @@ def save(self, path=None):
         """
         if path is None:
             path = self.path
-
-        old = self.stream.tell()
-
-        with open(path,'wb+') as fd:
-            self.stream.seek(0)
-            fd.write(self.get_data())
-
-        self.stream.seek(old)
+        data = self.get_data()
+        misc.write(path, data)
 
     def get_data(self):
         """get_data() -> bytes