Handle both vncserver and WinVNC keys

Author: sikyu

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])

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()