ntdll: Work around a conflict between Wine and Solaris list.h.
[wine/multimedia.git] / programs / regedit / listview.c
blobc0119ff293ef4ed48e8e242885ba2ae207800162
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 <commctrl.h>
24 #include <stdlib.h>
25 #include <stdio.h>
27 #include "main.h"
29 #include "wine/unicode.h"
30 static INT Image_String;
31 static INT Image_Binary;
33 typedef struct tagLINE_INFO
35 DWORD dwValType;
36 LPWSTR name;
37 void* val;
38 size_t val_len;
39 } LINE_INFO;
41 /*******************************************************************************
42 * Global and Local Variables:
45 static WNDPROC g_orgListWndProc;
46 static DWORD g_columnToSort = ~0U;
47 static BOOL g_invertSort = FALSE;
48 static LPWSTR g_valueName;
49 static LPWSTR g_currentPath;
50 static HKEY g_currentRootKey;
51 static WCHAR g_szValueNotSet[64];
53 #define MAX_LIST_COLUMNS (IDS_LIST_COLUMN_LAST - IDS_LIST_COLUMN_FIRST + 1)
54 static int default_column_widths[MAX_LIST_COLUMNS] = { 200, 175, 400 };
55 static int column_alignment[MAX_LIST_COLUMNS] = { LVCFMT_LEFT, LVCFMT_LEFT, LVCFMT_LEFT };
57 LPWSTR GetItemText(HWND hwndLV, UINT item)
59 LPWSTR newStr, curStr;
60 unsigned int maxLen = 128;
61 if (item == 0) return NULL; /* first item is ALWAYS a default */
63 curStr = HeapAlloc(GetProcessHeap(), 0, maxLen * sizeof(WCHAR));
64 if (!curStr) return NULL;
65 do {
66 ListView_GetItemTextW(hwndLV, item, 0, curStr, maxLen);
67 if (lstrlenW(curStr) < maxLen - 1) return curStr;
68 maxLen *= 2;
69 newStr = HeapReAlloc(GetProcessHeap(), 0, curStr, maxLen * sizeof(WCHAR));
70 if (!newStr) break;
71 curStr = newStr;
72 } while (TRUE);
73 HeapFree(GetProcessHeap(), 0, curStr);
74 return NULL;
77 LPCWSTR GetValueName(HWND hwndLV)
79 INT item;
81 if (g_valueName != LPSTR_TEXTCALLBACKW)
82 HeapFree(GetProcessHeap(), 0, g_valueName);
83 g_valueName = NULL;
85 item = SendMessageW(hwndLV, LVM_GETNEXTITEM, -1, MAKELPARAM(LVNI_FOCUSED, 0));
86 if (item == -1) return NULL;
88 g_valueName = GetItemText(hwndLV, item);
90 return g_valueName;
93 /* convert '\0' separated string list into ',' separated string list */
94 static void MakeMULTISZDisplayable(LPWSTR multi)
98 for (; *multi; multi++)
100 if (*(multi+1))
102 *multi = ',';
103 multi++;
105 } while (*multi);
108 /*******************************************************************************
109 * Local module support methods
111 static void AddEntryToList(HWND hwndLV, LPWSTR Name, DWORD dwValType,
112 void* ValBuf, DWORD dwCount, BOOL bHighlight)
114 LINE_INFO* linfo;
115 LVITEMW item;
116 int index;
118 linfo = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(LINE_INFO) + dwCount);
119 linfo->dwValType = dwValType;
120 linfo->val_len = dwCount;
121 CopyMemory(&linfo[1], ValBuf, dwCount);
123 if (Name)
125 linfo->name = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(Name) + 1) * sizeof(WCHAR));
126 lstrcpyW(linfo->name, Name);
127 } else
129 linfo->name = NULL;
132 item.mask = LVIF_TEXT | LVIF_PARAM | LVIF_STATE | LVIF_IMAGE;
133 item.iItem = SendMessageW(hwndLV, LVM_GETITEMCOUNT, 0, 0);
134 item.iSubItem = 0;
135 item.state = 0;
136 item.stateMask = LVIS_FOCUSED | LVIS_SELECTED;
137 item.pszText = Name ? Name : LPSTR_TEXTCALLBACKW;
138 item.cchTextMax = Name ? lstrlenW(item.pszText) : 0;
139 if (bHighlight) {
140 item.stateMask = item.state = LVIS_FOCUSED | LVIS_SELECTED;
142 switch (dwValType)
144 case REG_SZ:
145 case REG_EXPAND_SZ:
146 case REG_MULTI_SZ:
147 item.iImage = Image_String;
148 break;
149 default:
150 item.iImage = Image_Binary;
151 break;
153 item.lParam = (LPARAM)linfo;
155 #if (_WIN32_IE >= 0x0300)
156 item.iIndent = 0;
157 #endif
159 index = ListView_InsertItemW(hwndLV, &item);
160 if (index != -1) {
161 switch (dwValType) {
162 case REG_SZ:
163 case REG_EXPAND_SZ:
164 if (ValBuf) {
165 ListView_SetItemTextW(hwndLV, index, 2, ValBuf);
166 } else {
167 ListView_SetItemTextW(hwndLV, index, 2, g_szValueNotSet);
169 break;
170 case REG_DWORD: {
171 WCHAR buf[64];
172 WCHAR format[] = {'0','x','%','0','8','x',' ','(','%','u',')',0};
173 wsprintfW(buf, format, *(DWORD*)ValBuf, *(DWORD*)ValBuf);
174 ListView_SetItemTextW(hwndLV, index, 2, buf);
176 break;
177 case REG_BINARY: {
178 unsigned int i;
179 LPBYTE pData = ValBuf;
180 LPWSTR strBinary = HeapAlloc(GetProcessHeap(), 0, dwCount * sizeof(WCHAR) * 3 + sizeof(WCHAR));
181 WCHAR format[] = {'%','0','2','X',' ',0};
182 for (i = 0; i < dwCount; i++)
183 wsprintfW( strBinary + i*3, format, pData[i] );
184 strBinary[dwCount * 3] = 0;
185 ListView_SetItemTextW(hwndLV, index, 2, strBinary);
186 HeapFree(GetProcessHeap(), 0, strBinary);
188 break;
189 case REG_MULTI_SZ:
190 MakeMULTISZDisplayable(ValBuf);
191 ListView_SetItemTextW(hwndLV, index, 2, ValBuf);
192 break;
193 default:
195 WCHAR szText[128];
196 LoadStringW(hInst, IDS_REGISTRY_VALUE_CANT_DISPLAY, szText, COUNT_OF(szText));
197 ListView_SetItemTextW(hwndLV, index, 2, szText);
198 break;
204 static BOOL InitListViewImageList(HWND hWndListView)
206 HIMAGELIST himl;
207 HICON hicon;
208 INT cx = GetSystemMetrics(SM_CXSMICON);
209 INT cy = GetSystemMetrics(SM_CYSMICON);
211 himl = ImageList_Create(cx, cy, ILC_MASK, 0, 2);
212 if (!himl)
213 return FALSE;
215 hicon = LoadImageW(hInst, MAKEINTRESOURCEW(IDI_STRING),
216 IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR);
217 Image_String = ImageList_AddIcon(himl, hicon);
219 hicon = LoadImageW(hInst, MAKEINTRESOURCEW(IDI_BIN),
220 IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR);
221 Image_Binary = ImageList_AddIcon(himl, hicon);
223 SendMessageW( hWndListView, LVM_SETIMAGELIST, LVSIL_SMALL, (LPARAM) himl );
225 /* fail if some of the icons failed to load */
226 if (ImageList_GetImageCount(himl) < 2)
227 return FALSE;
229 return TRUE;
232 static BOOL CreateListColumns(HWND hWndListView)
234 WCHAR szText[50];
235 int index;
236 LVCOLUMNW lvC;
238 /* Create columns. */
239 lvC.mask = LVCF_FMT | LVCF_WIDTH | LVCF_TEXT | LVCF_SUBITEM;
240 lvC.pszText = szText;
242 /* Load the column labels from the resource file. */
243 for (index = 0; index < MAX_LIST_COLUMNS; index++) {
244 lvC.iSubItem = index;
245 lvC.cx = default_column_widths[index];
246 lvC.fmt = column_alignment[index];
247 LoadStringW(hInst, IDS_LIST_COLUMN_FIRST + index, szText, sizeof(szText)/sizeof(WCHAR));
248 if (ListView_InsertColumnW(hWndListView, index, &lvC) == -1) return FALSE;
250 return TRUE;
253 /* OnGetDispInfo - processes the LVN_GETDISPINFO notification message. */
255 static void OnGetDispInfo(NMLVDISPINFOW* plvdi)
257 static WCHAR buffer[200];
258 static WCHAR reg_szT[] = {'R','E','G','_','S','Z',0},
259 reg_expand_szT[] = {'R','E','G','_','E','X','P','A','N','D','_','S','Z',0},
260 reg_binaryT[] = {'R','E','G','_','B','I','N','A','R','Y',0},
261 reg_dwordT[] = {'R','E','G','_','D','W','O','R','D',0},
262 reg_dword_big_endianT[] = {'R','E','G','_','D','W','O','R','D','_',
263 'B','I','G','_','E','N','D','I','A','N',0},
264 reg_multi_szT[] = {'R','E','G','_','M','U','L','T','I','_','S','Z',0},
265 reg_linkT[] = {'R','E','G','_','L','I','N','K',0},
266 reg_resource_listT[] = {'R','E','G','_','R','E','S','O','U','R','C','E','_','L','I','S','T',0},
267 reg_noneT[] = {'R','E','G','_','N','O','N','E',0},
268 emptyT[] = {0};
270 plvdi->item.pszText = NULL;
271 plvdi->item.cchTextMax = 0;
273 switch (plvdi->item.iSubItem) {
274 case 0:
275 plvdi->item.pszText = g_pszDefaultValueName;
276 break;
277 case 1:
278 switch (((LINE_INFO*)plvdi->item.lParam)->dwValType) {
279 case REG_SZ:
280 plvdi->item.pszText = reg_szT;
281 break;
282 case REG_EXPAND_SZ:
283 plvdi->item.pszText = reg_expand_szT;
284 break;
285 case REG_BINARY:
286 plvdi->item.pszText = reg_binaryT;
287 break;
288 case REG_DWORD:
289 plvdi->item.pszText = reg_dwordT;
290 break;
291 case REG_DWORD_BIG_ENDIAN:
292 plvdi->item.pszText = reg_dword_big_endianT;
293 break;
294 case REG_MULTI_SZ:
295 plvdi->item.pszText = reg_multi_szT;
296 break;
297 case REG_LINK:
298 plvdi->item.pszText = reg_linkT;
299 break;
300 case REG_RESOURCE_LIST:
301 plvdi->item.pszText = reg_resource_listT;
302 break;
303 case REG_NONE:
304 plvdi->item.pszText = reg_noneT;
305 break;
306 default:
308 WCHAR szUnknownFmt[64];
309 LoadStringW(hInst, IDS_REGISTRY_UNKNOWN_TYPE, szUnknownFmt, COUNT_OF(szUnknownFmt));
310 wsprintfW(buffer, szUnknownFmt, plvdi->item.lParam);
311 plvdi->item.pszText = buffer;
312 break;
315 break;
316 case 2:
317 plvdi->item.pszText = g_szValueNotSet;
318 break;
319 case 3:
320 plvdi->item.pszText = emptyT;
321 break;
325 static int CALLBACK CompareFunc(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort)
327 LINE_INFO*l, *r;
328 l = (LINE_INFO*)lParam1;
329 r = (LINE_INFO*)lParam2;
330 if (!l->name) return -1;
331 if (!r->name) return +1;
333 if (g_columnToSort == ~0U)
334 g_columnToSort = 0;
336 if (g_columnToSort == 1)
337 return g_invertSort ? (int)r->dwValType - (int)l->dwValType : (int)l->dwValType - (int)r->dwValType;
338 if (g_columnToSort == 2) {
339 /* FIXME: Sort on value */
340 return 0;
342 return g_invertSort ? lstrcmpiW(r->name, l->name) : lstrcmpiW(l->name, r->name);
345 HWND StartValueRename(HWND hwndLV)
347 int item;
349 item = SendMessageW(hwndLV, LVM_GETNEXTITEM, -1, MAKELPARAM(LVNI_FOCUSED | LVNI_SELECTED, 0));
350 if (item < 1) { /* cannot rename default key */
351 MessageBeep(MB_ICONHAND);
352 return 0;
354 return (HWND)SendMessageW(hwndLV, LVM_EDITLABELW, item, 0);
357 static BOOL _CmdWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
359 switch (LOWORD(wParam)) {
360 /* case ID_FILE_OPEN: */
361 /* break; */
362 default:
363 return FALSE;
365 return TRUE;
368 static LRESULT CALLBACK ListWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
370 switch (message) {
371 case WM_COMMAND:
372 if (!_CmdWndProc(hWnd, message, wParam, lParam)) {
373 return CallWindowProcW(g_orgListWndProc, hWnd, message, wParam, lParam);
375 break;
376 case WM_NOTIFY_REFLECT:
377 switch (((LPNMHDR)lParam)->code) {
379 case LVN_BEGINLABELEDITW:
380 if (!((NMLVDISPINFOW *)lParam)->item.iItem)
381 return 1;
382 return 0;
383 case LVN_GETDISPINFOW:
384 OnGetDispInfo((NMLVDISPINFOW*)lParam);
385 break;
386 case LVN_COLUMNCLICK:
387 if (g_columnToSort == ((LPNMLISTVIEW)lParam)->iSubItem)
388 g_invertSort = !g_invertSort;
389 else {
390 g_columnToSort = ((LPNMLISTVIEW)lParam)->iSubItem;
391 g_invertSort = FALSE;
394 SendMessageW(hWnd, LVM_SORTITEMS, (WPARAM)hWnd, (LPARAM)CompareFunc);
395 break;
396 case LVN_ENDLABELEDITW: {
397 LPNMLVDISPINFOW dispInfo = (LPNMLVDISPINFOW)lParam;
398 LPWSTR oldName = GetItemText(hWnd, dispInfo->item.iItem);
399 LONG ret;
400 if (!oldName) return -1; /* cannot rename a default value */
401 ret = RenameValue(hWnd, g_currentRootKey, g_currentPath, oldName, dispInfo->item.pszText);
402 if (ret)
404 RefreshListView(hWnd, g_currentRootKey, g_currentPath, dispInfo->item.pszText);
406 HeapFree(GetProcessHeap(), 0, oldName);
407 return 0;
409 case NM_RETURN: {
410 int cnt = SendMessageW(hWnd, LVM_GETNEXTITEM, -1, MAKELPARAM(LVNI_FOCUSED | LVNI_SELECTED, 0));
411 if (cnt != -1)
412 SendMessageW(hFrameWnd, WM_COMMAND, ID_EDIT_MODIFY, 0);
414 break;
415 case NM_DBLCLK: {
416 NMITEMACTIVATE* nmitem = (LPNMITEMACTIVATE)lParam;
417 LVHITTESTINFO info;
419 /* if (nmitem->hdr.hwndFrom != hWnd) break; unnecessary because of WM_NOTIFY_REFLECT */
420 /* if (nmitem->hdr.idFrom != IDW_LISTVIEW) break; */
421 /* if (nmitem->hdr.code != ???) break; */
422 #ifdef _MSC_VER
423 switch (nmitem->uKeyFlags) {
424 case LVKF_ALT: /* The ALT key is pressed. */
425 /* properties dialog box ? */
426 break;
427 case LVKF_CONTROL: /* The CTRL key is pressed. */
428 /* run dialog box for providing parameters... */
429 break;
430 case LVKF_SHIFT: /* The SHIFT key is pressed. */
431 break;
433 #endif
434 info.pt.x = nmitem->ptAction.x;
435 info.pt.y = nmitem->ptAction.y;
436 if (SendMessageW(hWnd, LVM_HITTEST, 0, (LPARAM)&info) != -1) {
437 LVITEMW item;
439 item.state = 0;
440 item.stateMask = LVIS_FOCUSED | LVIS_SELECTED;
441 SendMessageW(hWnd, LVM_SETITEMSTATE, (UINT)-1, (LPARAM)&item);
443 item.state = LVIS_FOCUSED | LVIS_SELECTED;
444 item.stateMask = LVIS_FOCUSED | LVIS_SELECTED;
445 SendMessageW(hWnd, LVM_SETITEMSTATE, info.iItem, (LPARAM)&item);
447 SendMessageW(hFrameWnd, WM_COMMAND, ID_EDIT_MODIFY, 0);
450 break;
452 default:
453 return 0; /* shouldn't call default ! */
455 break;
456 case WM_CONTEXTMENU: {
457 int cnt = SendMessageW(hWnd, LVM_GETNEXTITEM, -1, MAKELPARAM(LVNI_SELECTED, 0));
458 TrackPopupMenu(GetSubMenu(hPopupMenus, cnt == -1 ? PM_NEW : PM_MODIFYVALUE),
459 TPM_RIGHTBUTTON, (short)LOWORD(lParam), (short)HIWORD(lParam),
460 0, hFrameWnd, NULL);
461 break;
463 default:
464 return CallWindowProcW(g_orgListWndProc, hWnd, message, wParam, lParam);
466 return 0;
470 HWND CreateListView(HWND hwndParent, UINT id)
472 RECT rcClient;
473 HWND hwndLV;
474 WCHAR ListView[] = {'L','i','s','t',' ','V','i','e','w',0};
476 /* prepare strings */
477 LoadStringW(hInst, IDS_REGISTRY_VALUE_NOT_SET, g_szValueNotSet, COUNT_OF(g_szValueNotSet));
479 /* Get the dimensions of the parent window's client area, and create the list view control. */
480 GetClientRect(hwndParent, &rcClient);
481 hwndLV = CreateWindowExW(WS_EX_CLIENTEDGE, WC_LISTVIEWW, ListView,
482 WS_VISIBLE | WS_CHILD | WS_TABSTOP | LVS_REPORT | LVS_EDITLABELS,
483 0, 0, rcClient.right, rcClient.bottom,
484 hwndParent, ULongToHandle(id), hInst, NULL);
485 if (!hwndLV) return NULL;
486 SendMessageW(hwndLV, LVM_SETUNICODEFORMAT, TRUE, 0);
487 SendMessageW(hwndLV, LVM_SETEXTENDEDLISTVIEWSTYLE, 0, LVS_EX_FULLROWSELECT);
489 /* Initialize the image list */
490 if (!InitListViewImageList(hwndLV)) goto fail;
491 if (!CreateListColumns(hwndLV)) goto fail;
492 g_orgListWndProc = (WNDPROC) SetWindowLongPtrW(hwndLV, GWLP_WNDPROC, (LPARAM)ListWndProc);
493 return hwndLV;
494 fail:
495 DestroyWindow(hwndLV);
496 return NULL;
499 BOOL RefreshListView(HWND hwndLV, HKEY hKeyRoot, LPCWSTR keyPath, LPCWSTR highlightValue)
501 BOOL result = FALSE;
502 DWORD max_sub_key_len;
503 DWORD max_val_name_len, valNameLen;
504 DWORD max_val_size, valSize;
505 DWORD val_count, index, valType;
506 WCHAR* valName = 0;
507 BYTE* valBuf = 0;
508 HKEY hKey = 0;
509 LONG errCode;
510 INT count, i;
511 LVITEMW item;
513 if (!hwndLV) return FALSE;
515 SendMessageW(hwndLV, WM_SETREDRAW, FALSE, 0);
517 errCode = RegOpenKeyExW(hKeyRoot, keyPath, 0, KEY_READ, &hKey);
518 if (errCode != ERROR_SUCCESS) goto done;
520 count = SendMessageW(hwndLV, LVM_GETITEMCOUNT, 0, 0);
521 for (i = 0; i < count; i++) {
522 item.mask = LVIF_PARAM;
523 item.iItem = i;
524 SendMessageW( hwndLV, LVM_GETITEMW, 0, (LPARAM)&item );
525 HeapFree(GetProcessHeap(), 0, ((LINE_INFO*)item.lParam)->name);
526 HeapFree(GetProcessHeap(), 0, (void*)item.lParam);
528 g_columnToSort = ~0U;
529 SendMessageW( hwndLV, LVM_DELETEALLITEMS, 0, 0L );
531 /* get size information and resize the buffers if necessary */
532 errCode = RegQueryInfoKeyW(hKey, NULL, NULL, NULL, NULL, &max_sub_key_len, NULL,
533 &val_count, &max_val_name_len, &max_val_size, NULL, NULL);
534 if (errCode != ERROR_SUCCESS) goto done;
536 /* account for the terminator char */
537 max_val_name_len++;
538 max_val_size++;
540 valName = HeapAlloc(GetProcessHeap(), 0, max_val_name_len * sizeof(WCHAR));
541 if (!valName)
542 goto done;
543 valBuf = HeapAlloc(GetProcessHeap(), 0, max_val_size);
544 if (!valBuf)
545 goto done;
547 if (RegQueryValueExW(hKey, NULL, NULL, &valType, valBuf, &valSize) == ERROR_FILE_NOT_FOUND) {
548 AddEntryToList(hwndLV, NULL, REG_SZ, NULL, 0, !highlightValue);
550 for(index = 0; index < val_count; index++) {
551 BOOL bSelected = (valName == highlightValue); /* NOT a bug, we check for double NULL here */
552 valNameLen = max_val_name_len;
553 valSize = max_val_size;
554 valType = 0;
555 errCode = RegEnumValueW(hKey, index, valName, &valNameLen, NULL, &valType, valBuf, &valSize);
556 if (errCode != ERROR_SUCCESS) goto done;
557 valBuf[valSize] = 0;
558 if (highlightValue && !lstrcmpW(valName, highlightValue))
559 bSelected = TRUE;
560 AddEntryToList(hwndLV, valName[0] ? valName : NULL, valType, valBuf, valSize, bSelected);
562 SendMessageW(hwndLV, LVM_SORTITEMS, (WPARAM)hwndLV, (LPARAM)CompareFunc);
564 g_currentRootKey = hKeyRoot;
565 if (keyPath != g_currentPath) {
566 HeapFree(GetProcessHeap(), 0, g_currentPath);
567 g_currentPath = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(keyPath) + 1) * sizeof(WCHAR));
568 if (!g_currentPath) goto done;
569 lstrcpyW(g_currentPath, keyPath);
572 result = TRUE;
574 done:
575 HeapFree(GetProcessHeap(), 0, valBuf);
576 HeapFree(GetProcessHeap(), 0, valName);
577 SendMessageW(hwndLV, WM_SETREDRAW, TRUE, 0);
578 if (hKey) RegCloseKey(hKey);
580 return result;