From 832a906b27feccd9916d38b6a890b047781acce2 Mon Sep 17 00:00:00 2001
From: sikyu <sikyu.geek@gmail.com>
Date: Tue, 30 May 2017 23:09:29 -0500
Subject: [PATCH 1/2] Handle 32-bit Python on 64-bit Windows

Also, reg path changed from WinVNC4 to vncserver
---
 .gitignore         | 3 +++
 WindowsRegistry.py | 8 +++++++-
 vncpasswd.py       | 4 ++--
 3 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/.gitignore b/.gitignore
index 0d20b64..830963a 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,4 @@
 *.pyc
+.idea/misc.xml
+*.xml
+*.iml
diff --git a/WindowsRegistry.py b/WindowsRegistry.py
index 0ea0a44..d9a3863 100755
--- a/WindowsRegistry.py
+++ b/WindowsRegistry.py
@@ -1,6 +1,7 @@
 import sys
 import _winreg as wreg
 import cPickle as pickle
+import platform
 
 class WindowsRegistry:
     """ A class to simplify read/write access to the Windows Registry """
@@ -58,7 +59,12 @@ def __init__(self, company="RealVNC", project="WinVNC4", create=0):
         i=0
         while not_opened and i<len(rights)-1:
             try:
-                self.key = wreg.OpenKey(wreg.HKEY_LOCAL_MACHINE, self.keyname,0, rights[i])
+                if platform.machine().endswith('64') and platform.architecture()[0] == '32bit':
+                    print('yes')
+                    print(rights[i])
+                    self.key = wreg.OpenKey(wreg.HKEY_LOCAL_MACHINE, self.keyname, 0, rights[i] | wreg.KEY_WOW64_64KEY)
+                else:
+                    self.key = wreg.OpenKey(wreg.HKEY_LOCAL_MACHINE, self.keyname,0, rights[i])
                 not_opened = False
                 self.right = rights[i]
                 #print "rights = %05x" % rights[i]
diff --git a/vncpasswd.py b/vncpasswd.py
index 9d69f00..f55c4c5 100755
--- a/vncpasswd.py
+++ b/vncpasswd.py
@@ -125,7 +125,7 @@ def main():
     if ( args.filename == None and args.passwd == None and (args.registry == False or not platform.system().startswith('Windows')) ):
         parser.error('Error: No password file or password passed\n')
     if ( args.registry and args.decrypt and platform.system().startswith('Windows')):
-        reg = wreg.WindowsRegistry("RealVNC", "WinVNC4")
+        reg = wreg.WindowsRegistry("RealVNC", "vncserver")
         ( args.passwd, key_type) = reg.getval("Password")
     elif not platform.system().startswith('Windows'):
 	print 'Cannot read from Windows Registry on a %s system' % platform.system()
@@ -159,7 +159,7 @@ def main():
     if ( args.filename != None and not args.decrypt ):
         do_file_out(args.filename, crypted, args.hex)
     if ( args.registry and not args.decrypt and platform.system().startswith('Windows')):
-        reg = wreg.WindowsRegistry("RealVNC", "WinVNC4")
+        reg = wreg.WindowsRegistry("RealVNC", "vncserver")
         reg.setval('Password', crypted, wreg.WindowsRegistry.REG_BINARY)
     elif not platform.system().startswith('Windows'):
 	print 'Cannot write to Windows Registry on a %s system' % platform.system()

From 9f73ffa76978324b4b77ac7de5a18d7ea10f4901 Mon Sep 17 00:00:00 2001
From: sikyu <sikyu.geek@gmail.com>
Date: Wed, 31 May 2017 23:28:05 -0500
Subject: [PATCH 2/2] Handle both vncserver and WinVNC keys

---
 WindowsRegistry.py |  2 --
 vncpasswd.py       | 21 +++++++++++++++++++--
 2 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/WindowsRegistry.py b/WindowsRegistry.py
index d9a3863..020dfcb 100755
--- a/WindowsRegistry.py
+++ b/WindowsRegistry.py
@@ -60,8 +60,6 @@ def __init__(self, company="RealVNC", project="WinVNC4", create=0):
         while not_opened and i<len(rights)-1:
             try:
                 if platform.machine().endswith('64') and platform.architecture()[0] == '32bit':
-                    print('yes')
-                    print(rights[i])
                     self.key = wreg.OpenKey(wreg.HKEY_LOCAL_MACHINE, self.keyname, 0, rights[i] | wreg.KEY_WOW64_64KEY)
                 else:
                     self.key = wreg.OpenKey(wreg.HKEY_LOCAL_MACHINE, self.keyname,0, rights[i])
diff --git a/vncpasswd.py b/vncpasswd.py
index f55c4c5..4c2857d 100755
--- a/vncpasswd.py
+++ b/vncpasswd.py
@@ -125,7 +125,7 @@ def main():
     if ( args.filename == None and args.passwd == None and (args.registry == False or not platform.system().startswith('Windows')) ):
         parser.error('Error: No password file or password passed\n')
     if ( args.registry and args.decrypt and platform.system().startswith('Windows')):
-        reg = wreg.WindowsRegistry("RealVNC", "vncserver")
+        reg = get_realvnc_key()
         ( args.passwd, key_type) = reg.getval("Password")
     elif not platform.system().startswith('Windows'):
 	print 'Cannot read from Windows Registry on a %s system' % platform.system()
@@ -159,7 +159,7 @@ def main():
     if ( args.filename != None and not args.decrypt ):
         do_file_out(args.filename, crypted, args.hex)
     if ( args.registry and not args.decrypt and platform.system().startswith('Windows')):
-        reg = wreg.WindowsRegistry("RealVNC", "vncserver")
+        reg = get_realvnc_key()
         reg.setval('Password', crypted, wreg.WindowsRegistry.REG_BINARY)
     elif not platform.system().startswith('Windows'):
 	print 'Cannot write to Windows Registry on a %s system' % platform.system()
@@ -168,5 +168,22 @@ def main():
     print "%scrypted Bin Pass= '%s'" % ( prefix, crypted )
     print "%scrypted Hex Pass= '%s'" % ( prefix, crypted.encode('hex') )
 
+
+def get_realvnc_key():
+    reg = None
+
+    for k in ['vncserver', 'WinVNC4',]:
+        try:
+            reg = wreg.WindowsRegistry('RealVNC', k)
+            break
+        except WindowsError as e:
+            if 'The system cannot find the file specified' in str(e):
+                pass
+            else:
+                raise e
+
+    return reg
+
+
 if __name__ == '__main__':
 	main()