]> git.saurik.com Git - wxWidgets.git/blobdiff - src/msw/registry.cpp
backported fix for proper tree ctrl refershing after changing item colour/font from 2.2
[wxWidgets.git] / src / msw / registry.cpp
index fe8897463ec947a417234378e63dfade2c0fef94..34e7c891a51efc34d8acec68f80f7ce97b581c76 100644 (file)
@@ -2,7 +2,7 @@
 // Name:        msw/registry.cpp
 // Purpose:     implementation of registry classes and functions
 // Author:      Vadim Zeitlin
-// Modified by: 
+// Modified by:
 // Created:     03.04.98
 // RCS-ID:      $Id$
 // Copyright:   (c) 1998 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
@@ -27,7 +27,6 @@
 #include  "wx/string.h"
 #include  "wx/intl.h"
 #include  "wx/log.h"
-#include  "wx/config.h"    // for wxExpandEnvVars
 
 #ifndef __WIN16__
 
@@ -43,7 +42,7 @@
 #include  <stdlib.h>      // for _MAX_PATH
 
 #ifndef _MAX_PATH
-       #define _MAX_PATH 512
+    #define _MAX_PATH 512
 #endif
 
 // our header
@@ -65,25 +64,25 @@ static struct
   const wxChar *szName;
   const wxChar *szShortName;
 }
-aStdKeys[] = 
-{ 
-  { HKEY_CLASSES_ROOT,      _T("HKEY_CLASSES_ROOT"),      _T("HKCR") },
+aStdKeys[] =
+{
+  { HKEY_CLASSES_ROOT,      wxT("HKEY_CLASSES_ROOT"),      wxT("HKCR") },
 #ifdef  __WIN32__
-  { HKEY_CURRENT_USER,      _T("HKEY_CURRENT_USER"),      _T("HKCU") },
-  { HKEY_LOCAL_MACHINE,     _T("HKEY_LOCAL_MACHINE"),     _T("HKLM") },
-  { HKEY_USERS,             _T("HKEY_USERS"),             _T("HKU")  }, // short name?
-  { HKEY_PERFORMANCE_DATA,  _T("HKEY_PERFORMANCE_DATA"),  _T("HKPD") },
+  { HKEY_CURRENT_USER,      wxT("HKEY_CURRENT_USER"),      wxT("HKCU") },
+  { HKEY_LOCAL_MACHINE,     wxT("HKEY_LOCAL_MACHINE"),     wxT("HKLM") },
+  { HKEY_USERS,             wxT("HKEY_USERS"),             wxT("HKU")  }, // short name?
+  { HKEY_PERFORMANCE_DATA,  wxT("HKEY_PERFORMANCE_DATA"),  wxT("HKPD") },
 #if     WINVER >= 0x0400
-  { HKEY_CURRENT_CONFIG,    _T("HKEY_CURRENT_CONFIG"),    _T("HKCC") },
+  { HKEY_CURRENT_CONFIG,    wxT("HKEY_CURRENT_CONFIG"),    wxT("HKCC") },
 #ifndef __GNUWIN32__
-  { HKEY_DYN_DATA,          _T("HKEY_DYN_DATA"),          _T("HKDD") }, // short name?
+  { HKEY_DYN_DATA,          wxT("HKEY_DYN_DATA"),          wxT("HKDD") }, // short name?
 #endif  //GNUWIN32
 #endif  //WINVER >= 4.0
 #endif  //WIN32
 };
 
 // the registry name separator (perhaps one day MS will change it to '/' ;-)
-#define   REG_SEPARATOR     _T('\\')
+#define   REG_SEPARATOR     wxT('\\')
 
 // useful for Windows programmers: makes somewhat more clear all these zeroes
 // being passed to Windows APIs
@@ -92,12 +91,12 @@ aStdKeys[] =
 // ----------------------------------------------------------------------------
 // macros
 // ----------------------------------------------------------------------------
-// @ const_cast<> is not yet supported by all compilers
+
+// const_cast<> is not yet supported by all compilers
 #define CONST_CAST    ((wxRegKey *)this)->
 
-#if   !USE_MUTABLE
-  #define m_dwLastError   CONST_CAST m_dwLastError
-#endif
+// and neither is mutable which m_dwLastError should be
+#define m_dwLastError   CONST_CAST m_dwLastError
 
 // ----------------------------------------------------------------------------
 // non member functions
@@ -110,7 +109,7 @@ static inline void RemoveTrailingSeparator(wxString& str);
 static bool KeyExists(WXHKEY hRootKey, const wxChar *szKey);
 
 // combines value and key name (uses static buffer!)
-static const wxChar *GetFullName(const wxRegKey *pKey, 
+static const wxChar *GetFullName(const wxRegKey *pKey,
                                const wxChar *szValue = NULL);
 
 // ============================================================================
@@ -124,11 +123,11 @@ static const wxChar *GetFullName(const wxRegKey *pKey,
 const size_t wxRegKey::nStdKeys = WXSIZEOF(aStdKeys);
 
 // @@ should take a `StdKey key', but as it's often going to be used in loops
-//    it would require casts in user code.  
+//    it would require casts in user code.
 const wxChar *wxRegKey::GetStdKeyName(size_t key)
 {
   // return empty string if key is invalid
-  wxCHECK_MSG( key < nStdKeys, _T(""), _T("invalid key in wxRegKey::GetStdKeyName") );
+  wxCHECK_MSG( key < nStdKeys, wxT(""), wxT("invalid key in wxRegKey::GetStdKeyName") );
 
   return aStdKeys[key].szName;
 }
@@ -136,19 +135,19 @@ const wxChar *wxRegKey::GetStdKeyName(size_t key)
 const wxChar *wxRegKey::GetStdKeyShortName(size_t key)
 {
   // return empty string if key is invalid
-  wxCHECK( key < nStdKeys, _T("") );
+  wxCHECK( key < nStdKeys, wxT("") );
 
   return aStdKeys[key].szShortName;
 }
 
 wxRegKey::StdKey wxRegKey::ExtractKeyName(wxString& strKey)
 {
-  wxString strRoot = strKey.Left(REG_SEPARATOR);
+  wxString strRoot = strKey.BeforeFirst(REG_SEPARATOR);
 
   HKEY hRootKey = 0;
   size_t ui;
   for ( ui = 0; ui < nStdKeys; ui++ ) {
-    if ( strRoot.CmpNoCase(aStdKeys[ui].szName) == 0 || 
+    if ( strRoot.CmpNoCase(aStdKeys[ui].szName) == 0 ||
          strRoot.CmpNoCase(aStdKeys[ui].szShortName) == 0 ) {
       hRootKey = aStdKeys[ui].hkey;
       break;
@@ -156,7 +155,7 @@ wxRegKey::StdKey wxRegKey::ExtractKeyName(wxString& strKey)
   }
 
   if ( ui == nStdKeys ) {
-    wxFAIL_MSG(_T("invalid key prefix in wxRegKey::ExtractKeyName."));
+    wxFAIL_MSG(wxT("invalid key prefix in wxRegKey::ExtractKeyName."));
 
     hRootKey = HKEY_CLASSES_ROOT;
   }
@@ -175,8 +174,8 @@ wxRegKey::StdKey wxRegKey::GetStdKeyFromHkey(WXHKEY hkey)
     if ( (int) aStdKeys[ui].hkey == (int) hkey )
       return (StdKey)ui;
   }
-  
-  wxFAIL_MSG(_T("non root hkey passed to wxRegKey::GetStdKeyFromHkey."));
+
+  wxFAIL_MSG(wxT("non root hkey passed to wxRegKey::GetStdKeyFromHkey."));
 
   return HKCR;
 }
@@ -187,16 +186,16 @@ wxRegKey::StdKey wxRegKey::GetStdKeyFromHkey(WXHKEY hkey)
 
 wxRegKey::wxRegKey()
 {
-  m_hKey = 0;
   m_hRootKey = (WXHKEY) aStdKeys[HKCR].hkey;
-  m_dwLastError = 0;
+
+  Init();
 }
 
 wxRegKey::wxRegKey(const wxString& strKey) : m_strKey(strKey)
 {
   m_hRootKey  = (WXHKEY) aStdKeys[ExtractKeyName(m_strKey)].hkey;
-  m_hKey      = (WXHKEY) NULL;
-  m_dwLastError = 0;
+
+  Init();
 }
 
 // parent is a predefined (and preopened) key
@@ -204,8 +203,8 @@ wxRegKey::wxRegKey(StdKey keyParent, const wxString& strKey) : m_strKey(strKey)
 {
   RemoveTrailingSeparator(m_strKey);
   m_hRootKey  = (WXHKEY) aStdKeys[keyParent].hkey;
-  m_hKey      = (WXHKEY) NULL;
-  m_dwLastError = 0;
+
+  Init();
 }
 
 // parent is a normal regkey
@@ -213,7 +212,7 @@ wxRegKey::wxRegKey(const wxRegKey& keyParent, const wxString& strKey)
         : m_strKey(keyParent.m_strKey)
 {
   // combine our name with parent's to get the full name
-  if ( !m_strKey.IsEmpty() && 
+  if ( !m_strKey.IsEmpty() &&
        (strKey.IsEmpty() || strKey[0] != REG_SEPARATOR) ) {
       m_strKey += REG_SEPARATOR;
   }
@@ -222,8 +221,8 @@ wxRegKey::wxRegKey(const wxRegKey& keyParent, const wxString& strKey)
   RemoveTrailingSeparator(m_strKey);
 
   m_hRootKey  = keyParent.m_hRootKey;
-  m_hKey      = (WXHKEY) NULL;
-  m_dwLastError = 0;
+
+  Init();
 }
 
 // dtor closes the key releasing system resource
@@ -261,7 +260,14 @@ void wxRegKey::SetName(const wxRegKey& keyParent, const wxString& strKey)
   Close();
 
   // combine our name with parent's to get the full name
-  m_strKey = keyParent.m_strKey;
+
+  // NB: this method is called by wxRegConfig::SetPath() which is a performance
+  //     critical function and so it preallocates space for our m_strKey to
+  //     gain some speed - this is why we only use += here and not = which
+  //     would just free the prealloc'd buffer and would have to realloc it the
+  //     next line!
+  m_strKey.clear();
+  m_strKey += keyParent.m_strKey;
   if ( !strKey.IsEmpty() && strKey[0] != REG_SEPARATOR )
     m_strKey += REG_SEPARATOR;
   m_strKey += strKey;
@@ -294,7 +300,7 @@ bool wxRegKey::Exists() const
 wxString wxRegKey::GetName(bool bShortPrefix) const
 {
   StdKey key = GetStdKeyFromHkey((StdKey) m_hRootKey);
-  wxString str = bShortPrefix ? aStdKeys[key].szShortName 
+  wxString str = bShortPrefix ? aStdKeys[key].szShortName
                               : aStdKeys[key].szName;
   if ( !m_strKey.IsEmpty() )
     str << "\\" << m_strKey;
@@ -302,42 +308,52 @@ wxString wxRegKey::GetName(bool bShortPrefix) const
   return str;
 }
 
-#if defined( __GNUWIN32__ ) && !defined(wxUSE_NORLANDER_HEADERS)
-bool wxRegKey::GetKeyInfo(size_t* pnSubKeys,
-                          size_t* pnMaxKeyLen,
-                          size_t* pnValues,
-                          size_t* pnMaxValueLen) const
-#else
-bool wxRegKey::GetKeyInfo(ulong *pnSubKeys,
-                          ulong *pnMaxKeyLen,
-                          ulong *pnValues,
-                          ulong *pnMaxValueLen) const
-#endif
+bool wxRegKey::GetKeyInfo(size_t *pnSubKeys,
+                          size_t *pnMaxKeyLen,
+                          size_t *pnValues,
+                          size_t *pnMaxValueLen) const
 {
 #if defined(__WIN32__) && !defined(__TWIN32__)
+
+    // old gcc headers incorrectly prototype RegQueryInfoKey()
+#if defined(__GNUWIN32_OLD__) && !defined(__CYGWIN10__)
+    #define REG_PARAM   (size_t *)
+#else
+    #define REG_PARAM   (LPDWORD)
+#endif
+
+  // it might be unexpected to some that this function doesn't open the key
+  wxASSERT_MSG( IsOpened(), _T("key should be opened in GetKeyInfo") );
+
   m_dwLastError = ::RegQueryInfoKey
                   (
                     (HKEY) m_hKey,
                     NULL,           // class name
                     NULL,           // (ptr to) size of class name buffer
                     RESERVED,
+                    REG_PARAM
                     pnSubKeys,      // [out] number of subkeys
+                    REG_PARAM
                     pnMaxKeyLen,    // [out] max length of a subkey name
                     NULL,           // longest subkey class name
+                    REG_PARAM
                     pnValues,       // [out] number of values
+                    REG_PARAM
                     pnMaxValueLen,  // [out] max length of a value name
                     NULL,           // longest value data
                     NULL,           // security descriptor
                     NULL            // time of last modification
                   );
 
+#undef REG_PARAM
+
   if ( m_dwLastError != ERROR_SUCCESS ) {
-    wxLogSysError(m_dwLastError, _("can't get info about registry key '%s'"),
+    wxLogSysError(m_dwLastError, _("Can't get info about registry key '%s'"),
                   GetName().c_str());
     return FALSE;
   }
-  else
-    return TRUE;
+
+  return TRUE;
 #else // Win16
   wxFAIL_MSG("GetKeyInfo() not implemented");
 
@@ -358,7 +374,7 @@ bool wxRegKey::Open()
   HKEY tmpKey;
   m_dwLastError = RegOpenKey((HKEY) m_hRootKey, m_strKey, &tmpKey);
   if ( m_dwLastError != ERROR_SUCCESS ) {
-    wxLogSysError(m_dwLastError, _("can't open registry key '%s'"),
+    wxLogSysError(m_dwLastError, _("Can't open registry key '%s'"),
                   GetName().c_str());
     return FALSE;
   }
@@ -383,7 +399,7 @@ bool wxRegKey::Create(bool bOkIfExists)
   HKEY tmpKey;
   m_dwLastError = RegCreateKey((HKEY) m_hRootKey, m_strKey, &tmpKey);
   if ( m_dwLastError != ERROR_SUCCESS ) {
-    wxLogSysError(m_dwLastError, _("can't create registry key '%s'"),
+    wxLogSysError(m_dwLastError, _("Can't create registry key '%s'"),
                   GetName().c_str());
     return FALSE;
   }
@@ -399,21 +415,200 @@ bool wxRegKey::Close()
 {
   if ( IsOpened() ) {
     m_dwLastError = RegCloseKey((HKEY) m_hKey);
+    m_hKey = 0;
+
     if ( m_dwLastError != ERROR_SUCCESS ) {
-      wxLogSysError(m_dwLastError, _("can't close registry key '%s'"),
+      wxLogSysError(m_dwLastError, _("Can't close registry key '%s'"),
                     GetName().c_str());
 
-      m_hKey = 0;
       return FALSE;
     }
-    else {
-      m_hKey = 0;
-    }
   }
 
   return TRUE;
 }
 
+bool wxRegKey::RenameValue(const wxChar *szValueOld, const wxChar *szValueNew)
+{
+    bool ok = TRUE;
+    if ( HasValue(szValueNew) ) {
+        wxLogError(_("Registry value '%s' already exists."), szValueNew);
+
+        ok = FALSE;
+    }
+
+    if ( !ok ||
+         !CopyValue(szValueOld, *this, szValueNew) ||
+         !DeleteValue(szValueOld) ) {
+        wxLogError(_("Failed to rename registry value '%s' to '%s'."),
+                   szValueOld, szValueNew);
+
+        return FALSE;
+    }
+
+    return TRUE;
+}
+
+bool wxRegKey::CopyValue(const wxChar *szValue,
+                         wxRegKey& keyDst,
+                         const wxChar *szValueNew)
+{
+    if ( !szValueNew ) {
+        // by default, use the same name
+        szValueNew = szValue;
+    }
+
+    switch ( GetValueType(szValue) ) {
+        case Type_String:
+            {
+                wxString strVal;
+                return QueryValue(szValue, strVal) &&
+                       keyDst.SetValue(szValueNew, strVal);
+            }
+
+        case Type_Dword:
+        /* case Type_Dword_little_endian: == Type_Dword */
+            {
+                long dwVal;
+                return QueryValue(szValue, &dwVal) &&
+                       keyDst.SetValue(szValueNew, dwVal);
+            }
+
+        // these types are unsupported because I am not sure about how
+        // exactly they should be copied and because they shouldn't
+        // occur among the application keys (supposedly created with
+        // this class)
+#ifdef  __WIN32__
+        case Type_None:
+        case Type_Expand_String:
+        case Type_Binary:
+        case Type_Dword_big_endian:
+        case Type_Link:
+        case Type_Multi_String:
+        case Type_Resource_list:
+        case Type_Full_resource_descriptor:
+        case Type_Resource_requirements_list:
+#endif // Win32
+        default:
+            wxLogError(_("Can't copy values of unsupported type %d."),
+                       GetValueType(szValue));
+            return FALSE;
+    }
+}
+
+bool wxRegKey::Rename(const wxChar *szNewName)
+{
+    wxCHECK_MSG( !!m_strKey, FALSE, _T("registry hives can't be renamed") );
+
+    if ( !Exists() ) {
+        wxLogError(_("Registry key '%s' does not exist, cannot rename it."),
+                   GetFullName(this));
+
+        return FALSE;
+    }
+
+    // do we stay in the same hive?
+    bool inSameHive = !wxStrchr(szNewName, REG_SEPARATOR);
+
+    // construct the full new name of the key
+    wxRegKey keyDst;
+
+    if ( inSameHive ) {
+        // rename the key to the new name under the same parent
+        wxString strKey = m_strKey.BeforeLast(REG_SEPARATOR);
+        if ( !!strKey ) {
+            // don't add '\\' in the start if strFullNewName is empty
+            strKey += REG_SEPARATOR;
+        }
+
+        strKey += szNewName;
+
+        keyDst.SetName(GetStdKeyFromHkey(m_hRootKey), strKey);
+    }
+    else {
+        // this is the full name already
+        keyDst.SetName(szNewName);
+    }
+
+    bool ok = keyDst.Create(FALSE /* fail if alredy exists */);
+    if ( !ok ) {
+        wxLogError(_("Registry key '%s' already exists."),
+                   GetFullName(&keyDst));
+    }
+    else {
+        ok = Copy(keyDst) && DeleteSelf();
+    }
+
+    if ( !ok ) {
+        wxLogError(_("Failed to rename the registry key '%s' to '%s'."),
+                   GetFullName(this), GetFullName(&keyDst));
+    }
+    else {
+        m_hRootKey = keyDst.m_hRootKey;
+        m_strKey = keyDst.m_strKey;
+    }
+
+    return ok;
+}
+
+bool wxRegKey::Copy(const wxChar *szNewName)
+{
+    // create the new key first
+    wxRegKey keyDst(szNewName);
+    bool ok = keyDst.Create(FALSE /* fail if alredy exists */);
+    if ( ok ) {
+        ok = Copy(keyDst);
+
+        // we created the dest key but copying to it failed - delete it
+        if ( !ok ) {
+            (void)keyDst.DeleteSelf();
+        }
+    }
+
+    return ok;
+}
+
+bool wxRegKey::Copy(wxRegKey& keyDst)
+{
+    bool ok = TRUE;
+
+    // copy all sub keys to the new location
+    wxString strKey;
+    long lIndex;
+    bool bCont = GetFirstKey(strKey, lIndex);
+    while ( ok && bCont ) {
+        wxRegKey key(*this, strKey);
+        wxString keyName;
+        keyName << GetFullName(&keyDst) << REG_SEPARATOR << strKey;
+        ok = key.Copy((const wxChar*) keyName);
+
+        if ( ok )
+            bCont = GetNextKey(strKey, lIndex);
+    }
+
+    // copy all values
+    wxString strVal;
+    bCont = GetFirstValue(strVal, lIndex);
+    while ( ok && bCont ) {
+        ok = CopyValue(strVal, keyDst);
+
+        if ( !ok ) {
+            wxLogSysError(m_dwLastError,
+                          _("Failed to copy registry value '%s'"),
+                          strVal.c_str());
+        }
+        else {
+            bCont = GetNextValue(strVal, lIndex);
+        }
+    }
+
+    if ( !ok ) {
+        wxLogError(_("Failed to copy the contents of registry key '%s' to '%s'."), GetFullName(this), GetFullName(&keyDst));
+    }
+
+    return ok;
+}
+
 // ----------------------------------------------------------------------------
 // delete keys/values
 // ----------------------------------------------------------------------------
@@ -430,11 +625,10 @@ bool wxRegKey::DeleteSelf()
   // prevent a buggy program from erasing one of the root registry keys or an
   // immediate subkey (i.e. one which doesn't have '\\' inside) of any other
   // key except HKCR (HKCR has some "deleteable" subkeys)
-  if ( m_strKey.IsEmpty() || (m_hRootKey != HKCR &&
-       m_strKey.Find(REG_SEPARATOR) == wxNOT_FOUND) ) {
-      wxLogError(_("Registry key '%s' is needed for normal system operation,\n"
-                   "deleting it will leave your system in unusable state:\n"
-                   "operation aborted."), GetFullName(this));
+  if ( m_strKey.IsEmpty() ||
+       ((m_hRootKey != (WXHKEY) aStdKeys[HKCR].hkey) &&
+        (m_strKey.Find(REG_SEPARATOR) == wxNOT_FOUND)) ) {
+      wxLogError(_("Registry key '%s' is needed for normal system operation,\ndeleting it will leave your system in unusable state:\noperation aborted."), GetFullName(this));
 
       return FALSE;
   }
@@ -464,7 +658,7 @@ bool wxRegKey::DeleteSelf()
 
   m_dwLastError = RegDeleteKey((HKEY) m_hRootKey, m_strKey);
   if ( m_dwLastError != ERROR_SUCCESS ) {
-    wxLogSysError(m_dwLastError, _("can't delete key '%s'"),
+    wxLogSysError(m_dwLastError, _("Can't delete key '%s'"),
                   GetName().c_str());
     return FALSE;
   }
@@ -489,7 +683,7 @@ bool wxRegKey::DeleteValue(const wxChar *szValue)
 #if defined(__WIN32__) && !defined(__TWIN32__)
     m_dwLastError = RegDeleteValue((HKEY) m_hKey, WXSTRINGCAST szValue);
     if ( m_dwLastError != ERROR_SUCCESS ) {
-      wxLogSysError(m_dwLastError, _("can't delete value '%s' from key '%s'"),
+      wxLogSysError(m_dwLastError, _("Can't delete value '%s' from key '%s'"),
                     szValue, GetName().c_str());
       return FALSE;
     }
@@ -500,7 +694,7 @@ bool wxRegKey::DeleteValue(const wxChar *szValue)
     // just set the (default and unique) value of the key to ""
     m_dwLastError = RegSetValue((HKEY) m_hKey, NULL, REG_SZ, "", RESERVED);
     if ( m_dwLastError != ERROR_SUCCESS ) {
-      wxLogSysError(m_dwLastError, _("can't delete value of key '%s'"),
+      wxLogSysError(m_dwLastError, _("Can't delete value of key '%s'"),
                     GetName().c_str());
       return FALSE;
     }
@@ -518,14 +712,16 @@ bool wxRegKey::HasValue(const wxChar *szValue) const
 {
   // this function should be silent, so suppress possible messages from Open()
   wxLogNull nolog;
-  
+
   #ifdef  __WIN32__
-    if ( CONST_CAST Open() ) {
-      return RegQueryValueEx((HKEY) m_hKey, WXSTRINGCAST szValue, RESERVED,
-                             NULL, NULL, NULL) == ERROR_SUCCESS;
-    }
-    else
-      return FALSE;
+    if ( !CONST_CAST Open() )
+        return FALSE;
+
+    LONG dwRet = ::RegQueryValueEx((HKEY) m_hKey,
+                                   WXSTRINGCAST szValue,
+                                   RESERVED,
+                                   NULL, NULL, NULL);
+    return dwRet == ERROR_SUCCESS;
   #else   // WIN16
     // only unnamed value exists
     return IsEmpty(szValue);
@@ -537,7 +733,7 @@ bool wxRegKey::HasValues() const
 {
   // suppress possible messages from GetFirstValue()
   wxLogNull nolog;
-  
+
   // just call GetFirstValue with dummy parameters
   wxString str;
   long     l;
@@ -549,7 +745,7 @@ bool wxRegKey::HasSubkeys() const
 {
   // suppress possible messages from GetFirstKey()
   wxLogNull nolog;
-  
+
   // just call GetFirstKey with dummy parameters
   wxString str;
   long     l;
@@ -561,11 +757,11 @@ bool wxRegKey::HasSubKey(const wxChar *szKey) const
 {
   // this function should be silent, so suppress possible messages from Open()
   wxLogNull nolog;
-  
-  if ( CONST_CAST Open() )
-    return KeyExists(m_hKey, szKey);
-  else
+
+  if ( !CONST_CAST Open() )
     return FALSE;
+
+  return KeyExists(m_hKey, szKey);
 }
 
 wxRegKey::ValueType wxRegKey::GetValueType(const wxChar *szValue) const
@@ -578,7 +774,7 @@ wxRegKey::ValueType wxRegKey::GetValueType(const wxChar *szValue) const
     m_dwLastError = RegQueryValueEx((HKEY) m_hKey, WXSTRINGCAST szValue, RESERVED,
                                     &dwType, NULL, NULL);
     if ( m_dwLastError != ERROR_SUCCESS ) {
-      wxLogSysError(m_dwLastError, _("can't read value of key '%s'"),
+      wxLogSysError(m_dwLastError, _("Can't read value of key '%s'"),
                     GetName().c_str());
       return Type_None;
     }
@@ -603,7 +799,7 @@ bool wxRegKey::SetValue(const wxChar *szValue, long lValue)
       return TRUE;
   }
 
-  wxLogSysError(m_dwLastError, _("can't set value of '%s'"),
+  wxLogSysError(m_dwLastError, _("Can't set value of '%s'"),
                 GetFullName(this, szValue));
   return FALSE;
 #endif
@@ -617,14 +813,14 @@ bool wxRegKey::QueryValue(const wxChar *szValue, long *plValue) const
     m_dwLastError = RegQueryValueEx((HKEY) m_hKey, WXSTRINGCAST szValue, RESERVED,
                                     &dwType, pBuf, &dwSize);
     if ( m_dwLastError != ERROR_SUCCESS ) {
-      wxLogSysError(m_dwLastError, _("can't read value of key '%s'"),
+      wxLogSysError(m_dwLastError, _("Can't read value of key '%s'"),
                     GetName().c_str());
       return FALSE;
     }
     else {
       // check that we read the value of right type
-      wxASSERT_MSG( dwType == REG_DWORD, 
-                    _T("Type mismatch in wxRegKey::QueryValue().")  );
+      wxASSERT_MSG( IsNumericValue(szValue),
+                    wxT("Type mismatch in wxRegKey::QueryValue().")  );
 
       return TRUE;
     }
@@ -635,7 +831,9 @@ bool wxRegKey::QueryValue(const wxChar *szValue, long *plValue) const
 
 #endif  //Win32
 
-bool wxRegKey::QueryValue(const wxChar *szValue, wxString& strValue) const
+bool wxRegKey::QueryValue(const wxChar *szValue,
+                          wxString& strValue,
+                          bool raw) const
 {
   if ( CONST_CAST Open() ) {
     #ifdef  __WIN32__
@@ -644,14 +842,50 @@ bool wxRegKey::QueryValue(const wxChar *szValue, wxString& strValue) const
       m_dwLastError = RegQueryValueEx((HKEY) m_hKey, WXSTRINGCAST szValue, RESERVED,
                                       &dwType, NULL, &dwSize);
       if ( m_dwLastError == ERROR_SUCCESS ) {
-        RegString pBuf = (RegString)strValue.GetWriteBuf(dwSize);
-        m_dwLastError = RegQueryValueEx((HKEY) m_hKey, WXSTRINGCAST szValue, RESERVED,
-                                        &dwType, pBuf, &dwSize);
-        strValue.UngetWriteBuf();
+        if ( !dwSize ) {
+            // must treat this case specially as GetWriteBuf() doesn't like
+            // being called with 0 size
+            strValue.Empty();
+        }
+        else {
+            RegString pBuf = (RegString)strValue.GetWriteBuf(dwSize);
+            m_dwLastError = RegQueryValueEx((HKEY) m_hKey,
+                                            WXSTRINGCAST szValue,
+                                            RESERVED,
+                                            &dwType,
+                                            pBuf,
+                                            &dwSize);
+            strValue.UngetWriteBuf();
+
+            // expand the var expansions in the string unless disabled
+            if ( (dwType == REG_EXPAND_SZ) && !raw )
+            {
+                DWORD dwExpSize = ::ExpandEnvironmentStrings(strValue, NULL, 0);
+                bool ok = dwExpSize != 0;
+                if ( ok )
+                {
+                    wxString strExpValue;
+                    ok = ::ExpandEnvironmentStrings
+                           (
+                            strValue,
+                            strExpValue.GetWriteBuf(dwExpSize),
+                            dwExpSize
+                           ) != 0;
+                    strExpValue.UngetWriteBuf();
+                    strValue = strExpValue;
+                }
+
+                if ( !ok )
+                {
+                    wxLogLastError(_T("ExpandEnvironmentStrings"));
+                }
+            }
+        }
+
         if ( m_dwLastError == ERROR_SUCCESS ) {
           // check that it was the right type
-          wxASSERT_MSG( dwType == REG_SZ, 
-                        _T("Type mismatch in wxRegKey::QueryValue().") );
+          wxASSERT_MSG( !IsNumericValue(szValue),
+                        wxT("Type mismatch in wxRegKey::QueryValue().") );
 
           return TRUE;
         }
@@ -667,7 +901,7 @@ bool wxRegKey::QueryValue(const wxChar *szValue, wxString& strValue) const
     #endif  //WIN16/32
   }
 
-  wxLogSysError(m_dwLastError, _("can't read value of '%s'"),
+  wxLogSysError(m_dwLastError, _("Can't read value of '%s'"),
                 GetFullName(this, szValue));
   return FALSE;
 }
@@ -677,8 +911,8 @@ bool wxRegKey::SetValue(const wxChar *szValue, const wxString& strValue)
   if ( CONST_CAST Open() ) {
 #if defined( __WIN32__) && !defined(__TWIN32__)
       m_dwLastError = RegSetValueEx((HKEY) m_hKey, szValue, (DWORD) RESERVED, REG_SZ,
-                                    (RegString)strValue.c_str(), 
-                                    strValue.Len() + 1);
+                                    (RegString)strValue.c_str(),
+                                    (strValue.Len() + 1)*sizeof(wxChar));
       if ( m_dwLastError == ERROR_SUCCESS )
         return TRUE;
     #else   //WIN16
@@ -691,7 +925,7 @@ bool wxRegKey::SetValue(const wxChar *szValue, const wxString& strValue)
     #endif  //WIN16/32
   }
 
-  wxLogSysError(m_dwLastError, _("can't set value of '%s'"),
+  wxLogSysError(m_dwLastError, _("Can't set value of '%s'"),
                 GetFullName(this, szValue));
   return FALSE;
 }
@@ -732,8 +966,8 @@ bool wxRegKey::GetNextValue(wxString& strValueName, long& lIndex) const
 
     m_dwLastError = RegEnumValue((HKEY) m_hKey, lIndex++,
                                  szValueName, &dwValueLen,
-                                 RESERVED, 
-                                 NULL,            // [out] type 
+                                 RESERVED,
+                                 NULL,            // [out] type
                                  NULL,            // [out] buffer for value
                                  NULL);           // [i/o]  it's length
 
@@ -743,7 +977,7 @@ bool wxRegKey::GetNextValue(wxString& strValueName, long& lIndex) const
         lIndex = -1;
       }
       else {
-        wxLogSysError(m_dwLastError, _("can't enumerate values of key '%s'"),
+        wxLogSysError(m_dwLastError, _("Can't enumerate values of key '%s'"),
                       GetName().c_str());
       }
 
@@ -788,7 +1022,7 @@ bool wxRegKey::GetNextKey(wxString& strKeyName, long& lIndex) const
       lIndex = -1;
     }
     else {
-      wxLogSysError(m_dwLastError, _("can't enumerate subkeys of key '%s'"),
+      wxLogSysError(m_dwLastError, _("Can't enumerate subkeys of key '%s'"),
                     GetName().c_str());
     }
 
@@ -801,11 +1035,11 @@ bool wxRegKey::GetNextKey(wxString& strKeyName, long& lIndex) const
 
 // returns TRUE if the value contains a number (else it's some string)
 bool wxRegKey::IsNumericValue(const wxChar *szValue) const
-  { 
+  {
       ValueType type = GetValueType(szValue);
       switch ( type ) {
         case Type_Dword:
-        case Type_Dword_little_endian:
+        /* case Type_Dword_little_endian: == Type_Dword */
         case Type_Dword_big_endian:
             return TRUE;
 
@@ -817,8 +1051,13 @@ bool wxRegKey::IsNumericValue(const wxChar *szValue) const
 // ============================================================================
 // implementation of global private functions
 // ============================================================================
+
 bool KeyExists(WXHKEY hRootKey, const wxChar *szKey)
 {
+  // don't close this key itself for the case of empty szKey!
+  if ( wxIsEmpty(szKey) )
+    return TRUE;
+
   HKEY hkeyDummy;
   if ( RegOpenKey( (HKEY) hRootKey, szKey, &hkeyDummy) == ERROR_SUCCESS ) {
     RegCloseKey(hkeyDummy);
@@ -833,7 +1072,7 @@ const wxChar *GetFullName(const wxRegKey *pKey, const wxChar *szValue)
   static wxString s_str;
   s_str = pKey->GetName();
   if ( !wxIsEmpty(szValue) )
-    s_str << _T("\\") << szValue;
+    s_str << wxT("\\") << szValue;
 
   return s_str.c_str();
 }