regedit: Remove commented out block.
[wine.git] / programs / regedit / listview.c
blobdc3152d3e398a9ba34d9d51e38a838c926daacfa
1 /*
2 * Regedit listviews
4 * Copyright (C) 2002 Robert Dickenson <robd@reactos.org>
5 * Copyright (C) 2008 Alexander N. Sørnes <alex@thehandofagony.com>
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
22 #include <windows.h>
23 #include <winternl.h>
24 #include <commctrl.h>
25 #include <stdlib.h>
26 #include <stdio.h>
28 #include "main.h"
30 #include "wine/unicode.h"
31 static INT Image_String;
32 static INT Image_Binary;
34 typedef struct tagLINE_INFO
36 DWORD dwValType;
37 LPWSTR name;
38 void* val;
39 size_t val_len;
40 } LINE_INFO;
42 /*******************************************************************************
43 * Global and Local Variables:
46 static WNDPROC g_orgListWndProc;
47 static DWORD g_columnToSort = ~0U;
48 static BOOL g_invertSort = FALSE;
49 static LPWSTR g_valueName;
50 static LPWSTR g_currentPath;
51 static HKEY g_currentRootKey;
52 static WCHAR g_szValueNotSet[64];
54 #define MAX_LIST_COLUMNS (IDS_LIST_COLUMN_LAST - IDS_LIST_COLUMN_FIRST + 1)
55 static int default_column_widths[MAX_LIST_COLUMNS] = { 200, 175, 400 };
56 static int column_alignment[MAX_LIST_COLUMNS] = { LVCFMT_LEFT, LVCFMT_LEFT, LVCFMT_LEFT };
58 LPWSTR GetItemText(HWND hwndLV, UINT item)
60 LPWSTR newStr, curStr;
61 unsigned int maxLen = 128;
62 if (item == 0) return NULL; /* first item is ALWAYS a default */
64 curStr = HeapAlloc(GetProcessHeap(), 0, maxLen * sizeof(WCHAR));
65 if (!curStr) return NULL;
66 do {
67 ListView_GetItemTextW(hwndLV, item, 0, curStr, maxLen);
68 if (lstrlenW(curStr) < maxLen - 1) return curStr;
69 maxLen *= 2;
70 newStr = HeapReAlloc(GetProcessHeap(), 0, curStr, maxLen * sizeof(WCHAR));
71 if (!newStr) break;
72 curStr = newStr;
73 } while (TRUE);
74 HeapFree(GetProcessHeap(), 0, curStr);
75 return NULL;
78 LPCWSTR GetValueName(HWND hwndLV)
80 INT item;
82 if (g_valueName != LPSTR_TEXTCALLBACKW)
83 HeapFree(GetProcessHeap(), 0, g_valueName);
84 g_valueName = NULL;
86 item = SendMessageW(hwndLV, LVM_GETNEXTITEM, -1, MAKELPARAM(LVNI_FOCUSED, 0));
87 if (item == -1) return NULL;
89 g_valueName = GetItemText(hwndLV, item);
91 return g_valueName;
94 /* convert '\0' separated string list into ',' separated string list */
95 static void MakeMULTISZDisplayable(LPWSTR multi)
99 for (; *multi; multi++)
101 if (*(multi+1))
103 *multi = ',';
104 multi++;
106 } while (*multi);
109 /*******************************************************************************
110 * Local module support methods
112 void format_value_data(HWND hwndLV, int index, DWORD type, void *data, DWORD size)
114 switch (type)
116 case REG_SZ:
117 case REG_EXPAND_SZ:
118 ListView_SetItemTextW(hwndLV, index, 2, data ? data : g_szValueNotSet);
119 break;
120 case REG_DWORD:
121 case REG_DWORD_BIG_ENDIAN:
123 DWORD value = *(DWORD *)data;
124 WCHAR buf[64];
125 WCHAR format[] = {'0','x','%','0','8','x',' ','(','%','u',')',0};
126 if (type == REG_DWORD_BIG_ENDIAN)
127 value = RtlUlongByteSwap(value);
128 wsprintfW(buf, format, value, value);
129 ListView_SetItemTextW(hwndLV, index, 2, buf);
130 break;
132 case REG_MULTI_SZ:
133 MakeMULTISZDisplayable(data);
134 ListView_SetItemTextW(hwndLV, index, 2, data);
135 break;
136 case REG_BINARY:
137 case REG_NONE:
138 default:
140 unsigned int i;
141 BYTE *pData = data;
142 WCHAR *strBinary = HeapAlloc(GetProcessHeap(), 0, size * sizeof(WCHAR) * 3 + sizeof(WCHAR));
143 WCHAR format[] = {'%','0','2','X',' ',0};
144 for (i = 0; i < size; i++)
145 wsprintfW( strBinary + i*3, format, pData[i] );
146 strBinary[size * 3] = 0;
147 ListView_SetItemTextW(hwndLV, index, 2, strBinary);
148 HeapFree(GetProcessHeap(), 0, strBinary);
149 break;
154 int AddEntryToList(HWND hwndLV, WCHAR *Name, DWORD dwValType, void *ValBuf, DWORD dwCount, int pos)
156 LVITEMW item = { 0 };
157 LINE_INFO* linfo;
158 int index;
160 linfo = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(LINE_INFO) + dwCount);
161 linfo->dwValType = dwValType;
162 linfo->val_len = dwCount;
163 CopyMemory(&linfo[1], ValBuf, dwCount);
165 if (Name)
167 linfo->name = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(Name) + 1) * sizeof(WCHAR));
168 lstrcpyW(linfo->name, Name);
169 } else
171 linfo->name = NULL;
174 item.mask = LVIF_TEXT | LVIF_PARAM | LVIF_STATE | LVIF_IMAGE;
175 item.iItem = (pos == -1) ? SendMessageW(hwndLV, LVM_GETITEMCOUNT, 0, 0) : pos;
176 item.stateMask = LVIS_FOCUSED | LVIS_SELECTED;
177 item.pszText = Name ? Name : LPSTR_TEXTCALLBACKW;
178 item.cchTextMax = Name ? lstrlenW(item.pszText) : 0;
180 switch (dwValType)
182 case REG_SZ:
183 case REG_EXPAND_SZ:
184 case REG_MULTI_SZ:
185 item.iImage = Image_String;
186 break;
187 default:
188 item.iImage = Image_Binary;
189 break;
191 item.lParam = (LPARAM)linfo;
193 if ((index = ListView_InsertItemW(hwndLV, &item)) != -1)
194 format_value_data(hwndLV, index, dwValType, ValBuf, dwCount);
195 return index;
198 static BOOL InitListViewImageList(HWND hWndListView)
200 HIMAGELIST himl;
201 HICON hicon;
202 INT cx = GetSystemMetrics(SM_CXSMICON);
203 INT cy = GetSystemMetrics(SM_CYSMICON);
205 himl = ImageList_Create(cx, cy, ILC_MASK, 0, 2);
206 if (!himl)
207 return FALSE;
209 hicon = LoadImageW(hInst, MAKEINTRESOURCEW(IDI_STRING),
210 IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR);
211 Image_String = ImageList_AddIcon(himl, hicon);
213 hicon = LoadImageW(hInst, MAKEINTRESOURCEW(IDI_BIN),
214 IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR);
215 Image_Binary = ImageList_AddIcon(himl, hicon);
217 SendMessageW( hWndListView, LVM_SETIMAGELIST, LVSIL_SMALL, (LPARAM) himl );
219 /* fail if some of the icons failed to load */
220 if (ImageList_GetImageCount(himl) < 2)
221 return FALSE;
223 return TRUE;
226 static BOOL CreateListColumns(HWND hWndListView)
228 WCHAR szText[50];
229 int index;
230 LVCOLUMNW lvC;
232 /* Create columns. */
233 lvC.mask = LVCF_FMT | LVCF_WIDTH | LVCF_TEXT | LVCF_SUBITEM;
234 lvC.pszText = szText;
236 /* Load the column labels from the resource file. */
237 for (index = 0; index < MAX_LIST_COLUMNS; index++) {
238 lvC.iSubItem = index;
239 lvC.cx = default_column_widths[index];
240 lvC.fmt = column_alignment[index];
241 LoadStringW(hInst, IDS_LIST_COLUMN_FIRST + index, szText, sizeof(szText)/sizeof(WCHAR));
242 if (ListView_InsertColumnW(hWndListView, index, &lvC) == -1) return FALSE;
244 return TRUE;
247 /* OnGetDispInfo - processes the LVN_GETDISPINFO notification message. */
249 static void OnGetDispInfo(NMLVDISPINFOW* plvdi)
251 static WCHAR buffer[200];
252 static WCHAR reg_szT[] = {'R','E','G','_','S','Z',0},
253 reg_expand_szT[] = {'R','E','G','_','E','X','P','A','N','D','_','S','Z',0},
254 reg_binaryT[] = {'R','E','G','_','B','I','N','A','R','Y',0},
255 reg_dwordT[] = {'R','E','G','_','D','W','O','R','D',0},
256 reg_dword_big_endianT[] = {'R','E','G','_','D','W','O','R','D','_',
257 'B','I','G','_','E','N','D','I','A','N',0},
258 reg_multi_szT[] = {'R','E','G','_','M','U','L','T','I','_','S','Z',0},
259 reg_linkT[] = {'R','E','G','_','L','I','N','K',0},
260 reg_resource_listT[] = {'R','E','G','_','R','E','S','O','U','R','C','E','_','L','I','S','T',0},
261 reg_noneT[] = {'R','E','G','_','N','O','N','E',0},
262 emptyT[] = {0};
264 plvdi->item.pszText = NULL;
265 plvdi->item.cchTextMax = 0;
267 switch (plvdi->item.iSubItem) {
268 case 0:
269 plvdi->item.pszText = g_pszDefaultValueName;
270 break;
271 case 1:
273 DWORD data_type = ((LINE_INFO *)plvdi->item.lParam)->dwValType;
275 switch (data_type) {
276 case REG_SZ:
277 plvdi->item.pszText = reg_szT;
278 break;
279 case REG_EXPAND_SZ:
280 plvdi->item.pszText = reg_expand_szT;
281 break;
282 case REG_BINARY:
283 plvdi->item.pszText = reg_binaryT;
284 break;
285 case REG_DWORD:
286 plvdi->item.pszText = reg_dwordT;
287 break;
288 case REG_DWORD_BIG_ENDIAN:
289 plvdi->item.pszText = reg_dword_big_endianT;
290 break;
291 case REG_MULTI_SZ:
292 plvdi->item.pszText = reg_multi_szT;
293 break;
294 case REG_LINK:
295 plvdi->item.pszText = reg_linkT;
296 break;
297 case REG_RESOURCE_LIST:
298 plvdi->item.pszText = reg_resource_listT;
299 break;
300 case REG_NONE:
301 plvdi->item.pszText = reg_noneT;
302 break;
303 default:
305 WCHAR fmt[] = {'0','x','%','x',0};
306 wsprintfW(buffer, fmt, data_type);
307 plvdi->item.pszText = buffer;
308 break;
311 break;
313 case 2:
314 plvdi->item.pszText = g_szValueNotSet;
315 break;
316 case 3:
317 plvdi->item.pszText = emptyT;
318 break;
322 static int CALLBACK CompareFunc(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort)
324 LINE_INFO*l, *r;
325 l = (LINE_INFO*)lParam1;
326 r = (LINE_INFO*)lParam2;
327 if (!l->name) return -1;
328 if (!r->name) return +1;
330 if (g_columnToSort == ~0U)
331 g_columnToSort = 0;
333 if (g_columnToSort == 1)
334 return g_invertSort ? (int)r->dwValType - (int)l->dwValType : (int)l->dwValType - (int)r->dwValType;
335 if (g_columnToSort == 2) {
336 /* FIXME: Sort on value */
337 return 0;
339 return g_invertSort ? lstrcmpiW(r->name, l->name) : lstrcmpiW(l->name, r->name);
342 HWND StartValueRename(HWND hwndLV)
344 int item;
346 item = SendMessageW(hwndLV, LVM_GETNEXTITEM, -1, MAKELPARAM(LVNI_FOCUSED | LVNI_SELECTED, 0));
347 if (item < 1) { /* cannot rename default key */
348 MessageBeep(MB_ICONHAND);
349 return 0;
351 return (HWND)SendMessageW(hwndLV, LVM_EDITLABELW, item, 0);
354 static BOOL _CmdWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
356 switch (LOWORD(wParam)) {
357 /* case ID_FILE_OPEN: */
358 /* break; */
359 default:
360 return FALSE;
362 return TRUE;
365 static LRESULT CALLBACK ListWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
367 switch (message) {
368 case WM_COMMAND:
369 if (!_CmdWndProc(hWnd, message, wParam, lParam)) {
370 return CallWindowProcW(g_orgListWndProc, hWnd, message, wParam, lParam);
372 break;
373 case WM_NOTIFY_REFLECT:
374 switch (((LPNMHDR)lParam)->code) {
376 case LVN_BEGINLABELEDITW:
377 if (!((NMLVDISPINFOW *)lParam)->item.iItem)
378 return 1;
379 return 0;
380 case LVN_GETDISPINFOW:
381 OnGetDispInfo((NMLVDISPINFOW*)lParam);
382 break;
383 case LVN_COLUMNCLICK:
384 if (g_columnToSort == ((LPNMLISTVIEW)lParam)->iSubItem)
385 g_invertSort = !g_invertSort;
386 else {
387 g_columnToSort = ((LPNMLISTVIEW)lParam)->iSubItem;
388 g_invertSort = FALSE;
391 SendMessageW(hWnd, LVM_SORTITEMS, (WPARAM)hWnd, (LPARAM)CompareFunc);
392 break;
393 case LVN_ENDLABELEDITW: {
394 LPNMLVDISPINFOW dispInfo = (LPNMLVDISPINFOW)lParam;
395 LPWSTR oldName = GetItemText(hWnd, dispInfo->item.iItem);
396 LONG ret;
398 if (!oldName) return -1; /* cannot rename a default value */
399 ret = RenameValue(hWnd, g_currentRootKey, g_currentPath, oldName, dispInfo->item.pszText);
400 if (ret)
402 dispInfo->item.iSubItem = 0;
403 SendMessageW(hWnd, LVM_SETITEMTEXTW, dispInfo->item.iItem, (LPARAM)&dispInfo->item);
405 HeapFree(GetProcessHeap(), 0, oldName);
406 return 0;
408 case LVN_DELETEITEM: {
409 NMLISTVIEW *nmlv = (NMLISTVIEW *)lParam;
410 LINE_INFO *info = (LINE_INFO *)nmlv->lParam;
412 HeapFree(GetProcessHeap(), 0, info->name);
413 HeapFree(GetProcessHeap(), 0, info);
415 break;
416 case NM_RETURN: {
417 int cnt = SendMessageW(hWnd, LVM_GETNEXTITEM, -1, MAKELPARAM(LVNI_FOCUSED | LVNI_SELECTED, 0));
418 if (cnt != -1)
419 SendMessageW(hFrameWnd, WM_COMMAND, ID_EDIT_MODIFY, 0);
421 break;
422 case NM_SETFOCUS:
423 g_pChildWnd->nFocusPanel = 1;
424 break;
425 case NM_DBLCLK: {
426 NMITEMACTIVATE* nmitem = (LPNMITEMACTIVATE)lParam;
427 LVHITTESTINFO info;
429 /* if (nmitem->hdr.hwndFrom != hWnd) break; unnecessary because of WM_NOTIFY_REFLECT */
430 /* if (nmitem->hdr.idFrom != IDW_LISTVIEW) break; */
431 /* if (nmitem->hdr.code != ???) break; */
432 info.pt.x = nmitem->ptAction.x;
433 info.pt.y = nmitem->ptAction.y;
434 if (SendMessageW(hWnd, LVM_HITTEST, 0, (LPARAM)&info) != -1) {
435 LVITEMW item;
437 item.state = 0;
438 item.stateMask = LVIS_FOCUSED | LVIS_SELECTED;
439 SendMessageW(hWnd, LVM_SETITEMSTATE, (UINT)-1, (LPARAM)&item);
441 item.state = LVIS_FOCUSED | LVIS_SELECTED;
442 item.stateMask = LVIS_FOCUSED | LVIS_SELECTED;
443 SendMessageW(hWnd, LVM_SETITEMSTATE, info.iItem, (LPARAM)&item);
445 SendMessageW(hFrameWnd, WM_COMMAND, ID_EDIT_MODIFY, 0);
448 break;
450 default:
451 return 0; /* shouldn't call default ! */
453 break;
454 case WM_CONTEXTMENU: {
455 int cnt = SendMessageW(hWnd, LVM_GETNEXTITEM, -1, MAKELPARAM(LVNI_SELECTED, 0));
456 TrackPopupMenu(GetSubMenu(hPopupMenus, cnt == -1 ? PM_NEW_VALUE : PM_MODIFY_VALUE),
457 TPM_RIGHTBUTTON, (short)LOWORD(lParam), (short)HIWORD(lParam),
458 0, hFrameWnd, NULL);
459 break;
461 default:
462 return CallWindowProcW(g_orgListWndProc, hWnd, message, wParam, lParam);
464 return 0;
468 HWND CreateListView(HWND hwndParent, UINT id)
470 RECT rcClient;
471 HWND hwndLV;
472 WCHAR ListView[] = {'L','i','s','t',' ','V','i','e','w',0};
474 /* prepare strings */
475 LoadStringW(hInst, IDS_REGISTRY_VALUE_NOT_SET, g_szValueNotSet, COUNT_OF(g_szValueNotSet));
477 /* Get the dimensions of the parent window's client area, and create the list view control. */
478 GetClientRect(hwndParent, &rcClient);
479 hwndLV = CreateWindowExW(WS_EX_CLIENTEDGE, WC_LISTVIEWW, ListView,
480 WS_VISIBLE | WS_CHILD | WS_TABSTOP | LVS_REPORT | LVS_EDITLABELS,
481 0, 0, rcClient.right, rcClient.bottom,
482 hwndParent, ULongToHandle(id), hInst, NULL);
483 if (!hwndLV) return NULL;
484 SendMessageW(hwndLV, LVM_SETUNICODEFORMAT, TRUE, 0);
485 SendMessageW(hwndLV, LVM_SETEXTENDEDLISTVIEWSTYLE, 0, LVS_EX_FULLROWSELECT);
487 /* Initialize the image list */
488 if (!InitListViewImageList(hwndLV)) goto fail;
489 if (!CreateListColumns(hwndLV)) goto fail;
490 g_orgListWndProc = (WNDPROC) SetWindowLongPtrW(hwndLV, GWLP_WNDPROC, (LPARAM)ListWndProc);
491 return hwndLV;
492 fail:
493 DestroyWindow(hwndLV);
494 return NULL;
497 BOOL RefreshListView(HWND hwndLV, HKEY hKeyRoot, LPCWSTR keyPath, LPCWSTR highlightValue)
499 BOOL result = FALSE;
500 DWORD max_sub_key_len;
501 DWORD max_val_name_len, valNameLen;
502 DWORD max_val_size, valSize;
503 DWORD val_count, index, valType;
504 WCHAR* valName = 0;
505 BYTE* valBuf = 0;
506 HKEY hKey = 0;
507 LONG errCode;
508 LVITEMW item;
510 if (!hwndLV) return FALSE;
512 SendMessageW(hwndLV, WM_SETREDRAW, FALSE, 0);
514 errCode = RegOpenKeyExW(hKeyRoot, keyPath, 0, KEY_READ, &hKey);
515 if (errCode != ERROR_SUCCESS) goto done;
517 g_columnToSort = ~0U;
518 SendMessageW(hwndLV, LVM_DELETEALLITEMS, 0, 0);
520 /* get size information and resize the buffers if necessary */
521 errCode = RegQueryInfoKeyW(hKey, NULL, NULL, NULL, NULL, &max_sub_key_len, NULL,
522 &val_count, &max_val_name_len, &max_val_size, NULL, NULL);
523 if (errCode != ERROR_SUCCESS) goto done;
525 /* account for the terminator char */
526 max_val_name_len++;
527 max_val_size++;
529 valName = HeapAlloc(GetProcessHeap(), 0, max_val_name_len * sizeof(WCHAR));
530 if (!valName)
531 goto done;
532 valBuf = HeapAlloc(GetProcessHeap(), 0, max_val_size);
533 if (!valBuf)
534 goto done;
536 valSize = max_val_size;
537 if (RegQueryValueExW(hKey, NULL, NULL, &valType, valBuf, &valSize) == ERROR_FILE_NOT_FOUND) {
538 AddEntryToList(hwndLV, NULL, REG_SZ, NULL, 0, -1);
540 for(index = 0; index < val_count; index++) {
541 valNameLen = max_val_name_len;
542 valSize = max_val_size;
543 valType = 0;
544 errCode = RegEnumValueW(hKey, index, valName, &valNameLen, NULL, &valType, valBuf, &valSize);
545 if (errCode != ERROR_SUCCESS) goto done;
546 valBuf[valSize] = 0;
547 AddEntryToList(hwndLV, valName[0] ? valName : NULL, valType, valBuf, valSize, -1);
550 memset(&item, 0, sizeof(item));
551 if (!highlightValue)
553 item.state = item.stateMask = LVIS_FOCUSED;
554 SendMessageW(hwndLV, LVM_SETITEMSTATE, 0, (LPARAM)&item);
557 SendMessageW(hwndLV, LVM_SORTITEMS, (WPARAM)hwndLV, (LPARAM)CompareFunc);
559 g_currentRootKey = hKeyRoot;
560 if (keyPath != g_currentPath) {
561 HeapFree(GetProcessHeap(), 0, g_currentPath);
562 g_currentPath = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(keyPath) + 1) * sizeof(WCHAR));
563 if (!g_currentPath) goto done;
564 lstrcpyW(g_currentPath, keyPath);
567 result = TRUE;
569 done:
570 HeapFree(GetProcessHeap(), 0, valBuf);
571 HeapFree(GetProcessHeap(), 0, valName);
572 SendMessageW(hwndLV, WM_SETREDRAW, TRUE, 0);
573 if (hKey) RegCloseKey(hKey);
575 return result;