[npwoods] 18077: 1. More merging from WINE regedit; imported refresh feature (which is wayyy slower than Windows regedit; need to optimize this)
26 Sep
2005
26 Sep
'05
3:19 a.m.
1. More merging from WINE regedit; imported refresh feature (which is
wayyy slower than Windows regedit; need to optimize this)
2. Using GetVersion() to determine whether to add HKEY_DYN_DATA
Modified: trunk/reactos/subsys/system/regedit/framewnd.c
Modified: trunk/reactos/subsys/system/regedit/main.h
Modified: trunk/reactos/subsys/system/regedit/treeview.c
_____
Modified: trunk/reactos/subsys/system/regedit/framewnd.c
--- trunk/reactos/subsys/system/regedit/framewnd.c 2005-09-25
22:16:05 UTC (rev 18076)
+++ trunk/reactos/subsys/system/regedit/framewnd.c 2005-09-26
03:18:53 UTC (rev 18077)
@@ -77,7 +77,7 @@
/* Update the status bar pane sizes */
nParts = -1;
- SendMessage(hStatusBar, SB_SETPARTS, 1, (long)&nParts);
+ SendMessage(hStatusBar, SB_SETPARTS, 1, (LPARAM)&nParts);
bInMenuLoop = TRUE;
SendMessage(hStatusBar, SB_SETTEXT, (WPARAM)0, (LPARAM)_T(""));
}
@@ -454,18 +454,6 @@
return result;
}
-BOOL RefreshView(HWND hWnd)
-{
- /* TODO:*/
- MessageBeep(-1);
- MessageBeep(MB_ICONASTERISK);
- MessageBeep(MB_ICONEXCLAMATION);
- MessageBeep(MB_ICONHAND);
- MessageBeep(MB_ICONQUESTION);
- MessageBeep(MB_OK);
- return TRUE;
-}
-
static BOOL CreateNewValue(HKEY hRootKey, LPCTSTR pszKeyPath, DWORD
dwType)
{
TCHAR szNewValueFormat[128];
@@ -689,7 +677,8 @@
break;
case ID_VIEW_REFRESH:
- RefreshView(hWnd);
+ RefreshTreeView(g_pChildWnd->hTreeWnd);
+ /*RefreshListView(g_pChildWnd->hListWnd, hKeyRoot, keyPath,
NULL); */
break;
/*case ID_OPTIONS_TOOLBAR:*/
/* toggle_child(hWnd, LOWORD(wParam), hToolBar);*/
_____
Modified: trunk/reactos/subsys/system/regedit/main.h
--- trunk/reactos/subsys/system/regedit/main.h 2005-09-25 22:16:05 UTC
(rev 18076)
+++ trunk/reactos/subsys/system/regedit/main.h 2005-09-26 03:18:53 UTC
(rev 18077)
@@ -37,6 +37,8 @@
#define PM_NEW 1
#define PM_TREECONTEXT 2
+#define MAX_NEW_KEY_LEN 128
+
extern HINSTANCE hInst;
/***********************************************************************
*******/
@@ -97,8 +99,11 @@
/* treeview.c */
extern HWND CreateTreeView(HWND hwndParent, LPTSTR pHostName, int id);
+extern BOOL RefreshTreeView(HWND hWndTV);
extern BOOL OnTreeExpanding(HWND hWnd, NMTREEVIEW* pnmtv);
extern LPCTSTR GetItemPath(HWND hwndTV, HTREEITEM hItem, HKEY*
phRootKey);
+extern HTREEITEM InsertNode(HWND hwndTV, HTREEITEM hItem, LPTSTR name);
+extern HWND StartKeyRename(HWND hwndTV);
extern BOOL CreateNewKey(HWND hwndTV, HTREEITEM hItem);
/* edit.c */
_____
Modified: trunk/reactos/subsys/system/regedit/treeview.c
--- trunk/reactos/subsys/system/regedit/treeview.c 2005-09-25
22:16:05 UTC (rev 18076)
+++ trunk/reactos/subsys/system/regedit/treeview.c 2005-09-26
03:18:53 UTC (rev 18077)
@@ -103,25 +103,197 @@
return pathBuffer;
}
-static HTREEITEM AddEntryToTree(HWND hwndTV, HTREEITEM hParent, LPTSTR
label, HKEY hKey, DWORD dwChildren, HTREEITEM insAfter )
+/* Add an entry to the tree. Only give hKey for root nodes (HKEY_
constants) */
+static HTREEITEM AddEntryToTree(HWND hwndTV, HTREEITEM hParent, LPTSTR
label, HKEY hKey, DWORD dwChildren)
{
TVITEM tvi;
TVINSERTSTRUCT tvins;
+ if (hKey) {
+ if (RegQueryInfoKey(hKey, 0, 0, 0, &dwChildren, 0, 0, 0, 0, 0,
0, 0) != ERROR_SUCCESS) {
+ dwChildren = 0;
+ }
+ }
+
tvi.mask = TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE |
TVIF_CHILDREN | TVIF_PARAM;
tvi.pszText = label;
- tvi.cchTextMax = _tcslen(tvi.pszText);
+ tvi.cchTextMax = lstrlen(tvi.pszText);
tvi.iImage = Image_Closed;
tvi.iSelectedImage = Image_Open;
tvi.cChildren = dwChildren;
tvi.lParam = (LPARAM)hKey;
tvins.u.item = tvi;
- tvins.hInsertAfter = insAfter;
+ tvins.hInsertAfter = (HTREEITEM)(hKey ? TVI_LAST : TVI_SORT);
tvins.hParent = hParent;
return TreeView_InsertItem(hwndTV, &tvins);
}
+static BOOL RefreshTreeItem(HWND hwndTV, HTREEITEM hItem)
+{
+ HKEY hRoot, hKey, hSubKey;
+ HTREEITEM childItem;
+ LPCTSTR KeyPath;
+ DWORD dwCount, dwIndex, dwMaxSubKeyLen;
+ LPSTR Name;
+ TVITEM tvItem;
+
+ KeyPath = GetItemPath(hwndTV, hItem, &hRoot);
+ if (*KeyPath) {
+ if (RegOpenKeyEx(hRoot, KeyPath, 0, KEY_READ, &hKey) !=
ERROR_SUCCESS) {
+ return FALSE;
+ }
+ } else {
+ hKey = hRoot;
+ }
+
+ if (RegQueryInfoKey(hKey, 0, 0, 0, &dwCount, &dwMaxSubKeyLen, 0, 0,
0, 0, 0, 0) != ERROR_SUCCESS) {
+ return FALSE;
+ }
+
+ /* Set the number of children again */
+ tvItem.mask = TVIF_CHILDREN;
+ tvItem.hItem = hItem;
+ tvItem.cChildren = dwCount;
+ if (!TreeView_SetItem(hwndTV, &tvItem)) {
+ return FALSE;
+ }
+
+ /* We don't have to bother with the rest if it's not expanded. */
+ if (TreeView_GetItemState(hwndTV, hItem, TVIS_EXPANDED) == 0) {
+ RegCloseKey(hKey);
+ return TRUE;
+ }
+
+ dwMaxSubKeyLen++; /* account for the \0 terminator */
+ if (!(Name = HeapAlloc(GetProcessHeap(), 0, dwMaxSubKeyLen *
sizeof(TCHAR)))) {
+ return FALSE;
+ }
+ tvItem.cchTextMax = dwMaxSubKeyLen;
+ if (!(tvItem.pszText = HeapAlloc(GetProcessHeap(), 0,
dwMaxSubKeyLen * sizeof(TCHAR)))) {
+ return FALSE;
+ }
+
+ /* Now go through all the children in the registry, and check if
any have to be added. */
+ for (dwIndex = 0; dwIndex < dwCount; dwIndex++) {
+ DWORD cName = dwMaxSubKeyLen, dwSubCount;
+ BOOL found;
+
+ found = FALSE;
+ if (RegEnumKeyEx(hKey, dwIndex, Name, &cName, 0, 0, 0, NULL) !=
ERROR_SUCCESS) {
+ continue;
+ }
+
+ /* Find the number of children of the node. */
+ dwSubCount = 0;
+ if (RegOpenKeyEx(hKey, Name, 0, KEY_QUERY_VALUE, &hSubKey) ==
ERROR_SUCCESS) {
+ if (RegQueryInfoKey(hSubKey, 0, 0, 0, &dwSubCount, 0, 0, 0,
0, 0, 0, 0) != ERROR_SUCCESS) {
+ dwSubCount = 0;
+ }
+ RegCloseKey(hSubKey);
+ }
+
+ /* Check if the node is already in there. */
+ for (childItem = TreeView_GetChild(hwndTV, hItem); childItem;
+ childItem = TreeView_GetNextSibling(hwndTV, childItem))
{
+ tvItem.mask = TVIF_TEXT;
+ tvItem.hItem = childItem;
+ if (!TreeView_GetItem(hwndTV, &tvItem)) {
+ return FALSE;
+ }
+
+ if (!strcmp(tvItem.pszText, Name)) {
+ found = TRUE;
+ break;
+ }
+ }
+
+ if (found == FALSE) {
+ AddEntryToTree(hwndTV, hItem, Name, NULL, dwSubCount);
+ }
+ }
+ HeapFree(GetProcessHeap(), 0, Name);
+ HeapFree(GetProcessHeap(), 0, tvItem.pszText);
+ RegCloseKey(hKey);
+
+ /* Now go through all the children in the tree, and check if any
have to be removed. */
+ childItem = TreeView_GetChild(hwndTV, hItem);
+ while (childItem) {
+ HTREEITEM nextItem = TreeView_GetNextSibling(hwndTV,
childItem);
+ if (RefreshTreeItem(hwndTV, childItem) == FALSE) {
+ TreeView_DeleteItem(hwndTV, childItem);
+ }
+ childItem = nextItem;
+ }
+
+ return TRUE;
+}
+
+BOOL RefreshTreeView(HWND hwndTV)
+{
+ HTREEITEM hItem;
+ HTREEITEM hSelectedItem;
+ HCURSOR hcursorOld;
+
+ hSelectedItem = TreeView_GetSelection(hwndTV);
+ hcursorOld = SetCursor(LoadCursor(NULL, IDC_WAIT));
+ SendMessage(hwndTV, WM_SETREDRAW, FALSE, 0);
+
+ hItem = TreeView_GetChild(hwndTV, TreeView_GetRoot(hwndTV));
+ while (hItem) {
+ RefreshTreeItem(hwndTV, hItem);
+ hItem = TreeView_GetNextSibling(hwndTV, hItem);
+ }
+
+ SendMessage(hwndTV, WM_SETREDRAW, TRUE, 0);
+ SetCursor(hcursorOld);
+
+ /* We reselect the currently selected node, this will prompt a
refresh of the listview. */
+ TreeView_SelectItem(hwndTV, hSelectedItem);
+ return TRUE;
+}
+
+HTREEITEM InsertNode(HWND hwndTV, HTREEITEM hItem, LPTSTR name)
+{
+ TCHAR buf[MAX_NEW_KEY_LEN];
+ HTREEITEM hNewItem = 0;
+ TVITEMEX item;
+
+ if (!hItem) hItem = TreeView_GetSelection(hwndTV);
+ if (!hItem) return FALSE;
+ if (TreeView_GetItemState(hwndTV, hItem, TVIS_EXPANDEDONCE)) {
+ hNewItem = AddEntryToTree(hwndTV, hItem, name, 0, 0);
+ } else {
+ item.mask = TVIF_CHILDREN | TVIF_HANDLE;
+ item.hItem = hItem;
+ if (!TreeView_GetItem(hwndTV, &item)) return FALSE;
+ item.cChildren = 1;
+ if (!TreeView_SetItem(hwndTV, &item)) return FALSE;
+ }
+ TreeView_Expand(hwndTV, hItem, TVE_EXPAND);
+ if (!hNewItem) {
+ for(hNewItem = TreeView_GetChild(hwndTV, hItem); hNewItem;
hNewItem = TreeView_GetNextSibling(hwndTV, hNewItem)) {
+ item.mask = TVIF_HANDLE | TVIF_TEXT;
+ item.hItem = hNewItem;
+ item.pszText = buf;
+ item.cchTextMax = COUNT_OF(buf);
+ if (!TreeView_GetItem(hwndTV, &item)) continue;
+ if (lstrcmp(name, item.pszText) == 0) break;
+ }
+ }
+ if (hNewItem) TreeView_SelectItem(hwndTV, hNewItem);
+
+ return hNewItem;
+}
+
+HWND StartKeyRename(HWND hwndTV)
+{
+ HTREEITEM hItem;
+
+ if(!(hItem = TreeView_GetSelection(hwndTV))) return 0;
+ return TreeView_EditLabel(hwndTV, hItem);
+}
+
static BOOL InitTreeViewItems(HWND hwndTV, LPTSTR pHostName)
{
TVITEM tvi;
@@ -131,7 +303,7 @@
tvi.mask = TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE |
TVIF_CHILDREN | TVIF_PARAM;
/* Set the text of the item. */
tvi.pszText = pHostName;
- tvi.cchTextMax = _tcslen(tvi.pszText);
+ tvi.cchTextMax = lstrlen(tvi.pszText);
/* Assume the item is not a parent item, so give it an image. */
tvi.iImage = Image_Root;
tvi.iSelectedImage = Image_Root;
@@ -144,12 +316,18 @@
/* Add the item to the tree view control. */
if (!(hRoot = TreeView_InsertItem(hwndTV, &tvins))) return FALSE;
- if (!AddEntryToTree(hwndTV, hRoot, _T("HKEY_CLASSES_ROOT"),
HKEY_CLASSES_ROOT, 1, TVI_LAST)) return FALSE;
- if (!AddEntryToTree(hwndTV, hRoot, _T("HKEY_CURRENT_USER"),
HKEY_CURRENT_USER, 1, TVI_LAST)) return FALSE;
- if (!AddEntryToTree(hwndTV, hRoot, _T("HKEY_LOCAL_MACHINE"),
HKEY_LOCAL_MACHINE, 1, TVI_LAST)) return FALSE;
- if (!AddEntryToTree(hwndTV, hRoot, _T("HKEY_USERS"), HKEY_USERS, 1,
TVI_LAST)) return FALSE;
- if (!AddEntryToTree(hwndTV, hRoot, _T("HKEY_CURRENT_CONFIG"),
HKEY_CURRENT_CONFIG, 1, TVI_LAST)) return FALSE;
+ if (!AddEntryToTree(hwndTV, hRoot, _T("HKEY_CLASSES_ROOT"),
HKEY_CLASSES_ROOT, 1)) return FALSE;
+ if (!AddEntryToTree(hwndTV, hRoot, _T("HKEY_CURRENT_USER"),
HKEY_CURRENT_USER, 1)) return FALSE;
+ if (!AddEntryToTree(hwndTV, hRoot, _T("HKEY_LOCAL_MACHINE"),
HKEY_LOCAL_MACHINE, 1)) return FALSE;
+ if (!AddEntryToTree(hwndTV, hRoot, _T("HKEY_USERS"), HKEY_USERS,
1)) return FALSE;
+ if (!AddEntryToTree(hwndTV, hRoot, _T("HKEY_CURRENT_CONFIG"),
HKEY_CURRENT_CONFIG, 1)) return FALSE;
+ if (GetVersion() & 0x80000000)
+ {
+ /* Win9x specific key */
+ if (!AddEntryToTree(hwndTV, hRoot, _T("HKEY_DYN_DATA"),
HKEY_DYN_DATA, 1)) return FALSE;
+ }
+
/* expand and select host name */
TreeView_Expand(hwndTV, hRoot, TVE_EXPAND);
TreeView_Select(hwndTV, hRoot, TVGN_CARET);
@@ -230,19 +408,17 @@
if (!Name) goto done;
for (dwIndex = 0; dwIndex < dwCount; dwIndex++) {
- DWORD cName = dwMaxSubKeyLen, dwSubCount;
- FILETIME LastWriteTime;
+ DWORD cName = dwMaxSubKeyLen, dwSubCount;
- errCode = RegEnumKeyEx(hNewKey, dwIndex, Name, &cName, 0, 0, 0,
&LastWriteTime);
- if (errCode != ERROR_SUCCESS) continue;
- errCode = RegOpenKeyEx(hNewKey, Name, 0, KEY_QUERY_VALUE,
&hKey);
- if (errCode == ERROR_SUCCESS) {
- errCode = RegQueryInfoKey(hKey, 0, 0, 0, &dwSubCount, 0, 0,
0, 0, 0, 0, 0);
- RegCloseKey(hKey);
- }
- if (errCode != ERROR_SUCCESS) dwSubCount = 0;
- /*printf("dwSubCount=%ld, Name=%s\n", dwSubCount, Name);*/
- AddEntryToTree(hwndTV, pnmtv->itemNew.hItem, Name, NULL,
dwSubCount, TVI_FIRST);
+ errCode = RegEnumKeyEx(hNewKey, dwIndex, Name, &cName, 0, 0, 0,
0);
+ if (errCode != ERROR_SUCCESS) continue;
+ errCode = RegOpenKeyEx(hNewKey, Name, 0, KEY_QUERY_VALUE,
&hKey);
+ if (errCode == ERROR_SUCCESS) {
+ errCode = RegQueryInfoKey(hKey, 0, 0, 0, &dwSubCount, 0, 0,
0, 0, 0, 0, 0);
+ RegCloseKey(hKey);
+ }
+ if (errCode != ERROR_SUCCESS) dwSubCount = 0;
+ AddEntryToTree(hwndTV, pnmtv->itemNew.hItem, Name, NULL,
dwSubCount);
}
SendMessage(hwndTV, TVM_SORTCHILDREN, 0,
(LPARAM)pnmtv->itemNew.hItem);
@@ -292,7 +468,7 @@
}
while(!hNewKey);
- hNewItem = AddEntryToTree(hwndTV, hItem, szNewKey, NULL, 0, NULL);
+ hNewItem = AddEntryToTree(hwndTV, hItem, szNewKey, NULL, 0);
if (!hNewItem)
goto done;
TreeView_EditLabel(hwndTV, hNewItem);
6989
days inactive
6989
days old
0 comments
1 participants
participants (1)
-
npwoods＠svn.reactos.com