[npwoods] 18766: Regedit fix and enhancement

Regedit fix and enhancement
1.  Fixed a bug in suggestions if the selected key cycles
2.  Implemented "Copy Key Name"
Modified: trunk/reactos/subsys/system/regedit/En.rc
Modified: trunk/reactos/subsys/system/regedit/childwnd.c
Modified: trunk/reactos/subsys/system/regedit/framewnd.c
Modified: trunk/reactos/subsys/system/regedit/main.h
  _____  

Modified: trunk/reactos/subsys/system/regedit/En.rc
--- trunk/reactos/subsys/system/regedit/En.rc	2005-10-24 23:44:07 UTC
(rev 18765)
+++ trunk/reactos/subsys/system/regedit/En.rc	2005-10-24 23:46:55 UTC
(rev 18766)
@@ -138,7 +138,7 @@

         MENUITEM "&Delete",                         ID_TREE_DELETE
         MENUITEM "&Rename",                         ID_TREE_RENAME
         MENUITEM SEPARATOR
-        MENUITEM "&Copy Key Name",
ID_EDIT_COPYKEYNAME, GRAYED
+        MENUITEM "&Copy Key Name",                  ID_EDIT_COPYKEYNAME
   END
 END
 
  _____  

Modified: trunk/reactos/subsys/system/regedit/childwnd.c
--- trunk/reactos/subsys/system/regedit/childwnd.c	2005-10-24
23:44:07 UTC (rev 18765)
+++ trunk/reactos/subsys/system/regedit/childwnd.c	2005-10-24
23:46:55 UTC (rev 18766)
@@ -143,6 +143,11 @@

                 TreeView_DeleteItem(pChildWnd->hTreeWnd, hSelection);
         }
         break;
+    case ID_EDIT_COPYKEYNAME:
+        hSelection = TreeView_GetSelection(pChildWnd->hTreeWnd);
+        keyPath = GetItemPath(pChildWnd->hTreeWnd, hSelection,
&hRootKey);
+        CopyKeyName(hWnd, hRootKey, keyPath);
+        break;
     case ID_EDIT_NEW_KEY:
         CreateNewKey(pChildWnd->hTreeWnd,
TreeView_GetSelection(pChildWnd->hTreeWnd));
         break;
@@ -178,6 +183,8 @@
  *  Key suggestion
  */
 
+#define MIN(a,b)	((a < b) ? (a) : (b))
+
 static void SuggestKeys(HKEY hRootKey, LPCTSTR pszKeyPath, LPTSTR
pszSuggestions,
 	size_t iSuggestionsLength)
 {
@@ -201,7 +208,9 @@
 			if (RegQueryStringValue(hRootKey, pszKeyPath,
NULL,
 				szBuffer, sizeof(szBuffer) /
sizeof(szBuffer[0])) == ERROR_SUCCESS)
 			{
-				if (szBuffer[0] != '\0')
+				/* Sanity check this key; it cannot be
empty, nor can it be a
+				 * loop back */
+				if ((szBuffer[0] != '\0') &&
_tcsicmp(szBuffer, pszKeyPath))
 				{
 					if (RegOpenKey(hRootKey,
szBuffer, &hOtherKey) == ERROR_SUCCESS)
 					{
@@ -211,7 +220,7 @@
 	    				iSuggestionsLength -= i;
 
 						lstrcpyn(pszSuggestions,
szBuffer, iSuggestionsLength);
-						i =
_tcslen(pszSuggestions) + 1;
+						i =
MIN(_tcslen(pszSuggestions) + 1, iSuggestionsLength);
 						pszSuggestions += i;
 						iSuggestionsLength -= i;
 						RegCloseKey(hOtherKey);
@@ -223,7 +232,7 @@
 				}
 			}
 		}
-		while(bFound);
+		while(bFound && (iSuggestionsLength > 0));
 
 		/* Check CLSID key */
 		if (RegOpenKey(hRootKey, pszKeyPath, &hSubKey) ==
ERROR_SUCCESS)
@@ -237,7 +246,7 @@
 				iSuggestionsLength -= i;
 
 				lstrcpyn(pszSuggestions, szBuffer,
iSuggestionsLength);
-				i = _tcslen(pszSuggestions) + 1;
+				i = MIN(_tcslen(pszSuggestions) + 1,
iSuggestionsLength);
 				pszSuggestions += i;
 				iSuggestionsLength -= i;
 			}
  _____  

Modified: trunk/reactos/subsys/system/regedit/framewnd.c
--- trunk/reactos/subsys/system/regedit/framewnd.c	2005-10-24
23:44:07 UTC (rev 18765)
+++ trunk/reactos/subsys/system/regedit/framewnd.c	2005-10-24
23:46:55 UTC (rev 18766)
@@ -470,34 +470,43 @@

     return TRUE;
 }
 
-static BOOL CopyKeyName(HWND hWnd, LPCTSTR keyName)
+BOOL CopyKeyName(HWND hWnd, HKEY hRootKey, LPCTSTR keyName)
 {
-    BOOL result;
+    BOOL bClipboardOpened = FALSE;
+    BOOL bSuccess = FALSE;
+    TCHAR szBuffer[512];
+    HGLOBAL hGlobal;
+    LPTSTR s;
 
-    result = OpenClipboard(hWnd);
-    if (result) {
-        result = EmptyClipboard();
-        if (result) {
+    if (!OpenClipboard(hWnd))
+        goto done;
+    bClipboardOpened = TRUE;
 
-            /*HANDLE hClipData;*/
-            /*hClipData = SetClipboardData(UINT uFormat, HANDLE
hMem);*/
+    if (!EmptyClipboard())
+        goto done;
 
-        } else {
-            /* error emptying clipboard*/
-            /* DWORD dwError = GetLastError(); */
-            ;
-        }
-        if (!CloseClipboard()) {
-            /* error closing clipboard*/
-            /* DWORD dwError = GetLastError(); */
-            ;
-        }
-    } else {
-        /* error opening clipboard*/
-        /* DWORD dwError = GetLastError(); */
-        ;
-    }
-    return result;
+    if (!RegKeyGetName(szBuffer, sizeof(szBuffer) /
sizeof(szBuffer[0]), hRootKey, keyName))
+        goto done;
+
+    hGlobal = GlobalAlloc(GMEM_MOVEABLE, (_tcslen(szBuffer) + 1) *
sizeof(TCHAR));
+    if (!hGlobal)
+        goto done;
+
+    s = GlobalLock(hGlobal);
+    _tcscpy(s, szBuffer);
+    GlobalUnlock(hGlobal);
+
+#ifdef UNICODE
+    SetClipboardData(CF_UNICODETEXT, hGlobal);
+#else
+    SetClipboardData(CF_TEXT, hGlobal);
+#endif
+    bSuccess = TRUE;
+
+done:
+    if (bClipboardOpened)
+        CloseClipboard();
+    return bSuccess;
 }
 
 static BOOL CreateNewValue(HKEY hRootKey, LPCTSTR pszKeyPath, DWORD
dwType)
@@ -860,7 +869,7 @@
         break;
     }
     case ID_EDIT_COPYKEYNAME:
-        CopyKeyName(hWnd, _T(""));
+        CopyKeyName(hWnd, hKeyRoot, keyPath);
         break;
     case ID_EDIT_PERMISSIONS:
         if(keyPath != NULL && _tcslen(keyPath) > 0)
  _____  

Modified: trunk/reactos/subsys/system/regedit/main.h
--- trunk/reactos/subsys/system/regedit/main.h	2005-10-24 23:44:07 UTC
(rev 18765)
+++ trunk/reactos/subsys/system/regedit/main.h	2005-10-24 23:46:55 UTC
(rev 18766)
@@ -89,6 +89,7 @@

 extern LRESULT CALLBACK FrameWndProc(HWND, UINT, WPARAM, LPARAM);
 extern void SetupStatusBar(HWND hWnd, BOOL bResize);
 extern void UpdateStatusBar(void);
+extern BOOL CopyKeyName(HWND hWnd, HKEY hRootKey, LPCTSTR keyName);
 
 /* listview.c */
 extern HWND CreateListView(HWND hwndParent, int id);