correctly detect PE drivers

binref · Dec 3, 2024 · 86bd910 · 86bd910
1 parent 3651d7f
commit 86bd910
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/refinery/lib/mime.py b/refinery/lib/mime.py
@@ -270,7 +270,12 @@ def __init__(self, data, default='bin', decompress=True):
         if self.description == 'Microsoft OOXML':
             extension = 'docx'
         if extension == 'exe':
-            extension = 'dll' if '(DLL)' in self.description else 'exe'
+            if '(DLL)' in self.description:
+                extension = 'dll'
+            elif '(native)' in self.description:
+                extension = 'sys'
+            else:
+                extension = 'exe'
         elif extension in ('gz', 'gzip', 'bz2') and decompress:
             if extension == 'bz2':
                 import bz2