Make local regitry processing Python3 compatible

mandiant · Oct 19, 2019 · 0695f0e · 0695f0e
1 parent 59daa71
commit 0695f0e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/ShimCacheParser.py b/ShimCacheParser.py
@@ -792,9 +792,12 @@ def get_local_data():
     global g_verbose
 
     try:
-        import _winreg as reg
+        if sys.version_info >= (3,0,0):
+            import winreg as reg
+        else:
+            import _winreg as reg
     except ImportError:
-        print("[-] \'winreg.py\' not found... Is this a Windows system?")
+        print("[-] 'winreg' not found... Is this a Windows system?")
         sys.exit(1)
 
     hReg = reg.ConnectRegistry(None, reg.HKEY_LOCAL_MACHINE)