/* * Regedit listviews * * Copyright (C) 2002 Robert Dickenson <robd@reactos.org> * Copyright (C) 2008 Alexander N. Sørnes <alex@thehandofagony.com> * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #include <windows.h> #include <winternl.h> #include <commctrl.h> #include <stdlib.h> #include "main.h" #include "wine/heap.h" static INT Image_String; static INT Image_Binary; /******************************************************************************* * Global and Local Variables: */ DWORD g_columnToSort = ~0U; BOOL g_invertSort = FALSE; WCHAR *g_currentPath; HKEY g_currentRootKey; static WCHAR g_szValueNotSet[64]; #define MAX_LIST_COLUMNS (IDS_LIST_COLUMN_LAST - IDS_LIST_COLUMN_FIRST + 1) static int default_column_widths[MAX_LIST_COLUMNS] = { 200, 175, 400 }; static int column_alignment[MAX_LIST_COLUMNS] = { LVCFMT_LEFT, LVCFMT_LEFT, LVCFMT_LEFT }; LPWSTR GetItemText(HWND hwndLV, UINT item) { WCHAR *curStr; unsigned int maxLen = 128; if (item == 0) return NULL; /* first item is ALWAYS a default */ curStr = heap_xalloc(maxLen * sizeof(WCHAR)); do { ListView_GetItemTextW(hwndLV, item, 0, curStr, maxLen); if (lstrlenW(curStr) < maxLen - 1) return curStr; maxLen *= 2; curStr = heap_xrealloc(curStr, maxLen * sizeof(WCHAR)); } while (TRUE); heap_free(curStr); return NULL; } WCHAR *GetValueName(HWND hwndLV) { INT item; item = SendMessageW(hwndLV, LVM_GETNEXTITEM, -1, MAKELPARAM(LVNI_FOCUSED, 0)); if (item == -1) return NULL; return GetItemText(hwndLV, item); } BOOL update_listview_path(const WCHAR *path) { heap_free(g_currentPath); g_currentPath = heap_xalloc((lstrlenW(path) + 1) * sizeof(WCHAR)); lstrcpyW(g_currentPath, path); return TRUE; } /* convert '\0' separated string list into ',' separated string list */ static void MakeMULTISZDisplayable(LPWSTR multi) { do { for (; *multi; multi++) ; if (*(multi+1)) { *multi = ','; multi++; } } while (*multi); } /******************************************************************************* * Local module support methods */ void format_value_data(HWND hwndLV, int index, DWORD type, void *data, DWORD size) { switch (type) { case REG_SZ: case REG_EXPAND_SZ: ListView_SetItemTextW(hwndLV, index, 2, data ? data : g_szValueNotSet); break; case REG_DWORD: case REG_DWORD_BIG_ENDIAN: { DWORD value = *(DWORD *)data; WCHAR buf[64]; WCHAR format[] = {'0','x','%','0','8','x',' ','(','%','u',')',0}; if (type == REG_DWORD_BIG_ENDIAN) value = RtlUlongByteSwap(value); wsprintfW(buf, format, value, value); ListView_SetItemTextW(hwndLV, index, 2, buf); break; } case REG_MULTI_SZ: MakeMULTISZDisplayable(data); ListView_SetItemTextW(hwndLV, index, 2, data); break; case REG_BINARY: case REG_NONE: default: { unsigned int i; BYTE *pData = data; WCHAR *strBinary = heap_xalloc(size * sizeof(WCHAR) * 3 + sizeof(WCHAR)); WCHAR format[] = {'%','0','2','X',' ',0}; for (i = 0; i < size; i++) wsprintfW( strBinary + i*3, format, pData[i] ); strBinary[size * 3] = 0; ListView_SetItemTextW(hwndLV, index, 2, strBinary); heap_free(strBinary); break; } } } int AddEntryToList(HWND hwndLV, WCHAR *Name, DWORD dwValType, void *ValBuf, DWORD dwCount, int pos) { LINE_INFO *linfo; LVITEMW item = { 0 }; int index; linfo = heap_xalloc(sizeof(LINE_INFO)); linfo->dwValType = dwValType; linfo->val_len = dwCount; if (Name) { linfo->name = heap_xalloc((lstrlenW(Name) + 1) * sizeof(WCHAR)); lstrcpyW(linfo->name, Name); } else linfo->name = NULL; if (ValBuf && dwCount) { linfo->val = heap_xalloc(dwCount); memcpy(linfo->val, ValBuf, dwCount); } else linfo->val = NULL; item.mask = LVIF_TEXT | LVIF_PARAM | LVIF_STATE | LVIF_IMAGE; item.iItem = (pos == -1) ? SendMessageW(hwndLV, LVM_GETITEMCOUNT, 0, 0) : pos; item.stateMask = LVIS_FOCUSED | LVIS_SELECTED; item.pszText = Name ? Name : LPSTR_TEXTCALLBACKW; item.cchTextMax = Name ? lstrlenW(item.pszText) : 0; switch (dwValType) { case REG_SZ: case REG_EXPAND_SZ: case REG_MULTI_SZ: item.iImage = Image_String; break; default: item.iImage = Image_Binary; break; } item.lParam = (LPARAM)linfo; if ((index = ListView_InsertItemW(hwndLV, &item)) != -1) format_value_data(hwndLV, index, dwValType, ValBuf, dwCount); return index; } static BOOL InitListViewImageList(HWND hWndListView) { HIMAGELIST himl; HICON hicon; INT cx = GetSystemMetrics(SM_CXSMICON); INT cy = GetSystemMetrics(SM_CYSMICON); himl = ImageList_Create(cx, cy, ILC_MASK, 0, 2); if (!himl) return FALSE; hicon = LoadImageW(hInst, MAKEINTRESOURCEW(IDI_STRING), IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR); Image_String = ImageList_AddIcon(himl, hicon); hicon = LoadImageW(hInst, MAKEINTRESOURCEW(IDI_BIN), IMAGE_ICON, cx, cy, LR_DEFAULTCOLOR); Image_Binary = ImageList_AddIcon(himl, hicon); SendMessageW( hWndListView, LVM_SETIMAGELIST, LVSIL_SMALL, (LPARAM) himl ); /* fail if some of the icons failed to load */ if (ImageList_GetImageCount(himl) < 2) return FALSE; return TRUE; } static BOOL CreateListColumns(HWND hWndListView) { WCHAR szText[50]; int index; LVCOLUMNW lvC; /* Create columns. */ lvC.mask = LVCF_FMT | LVCF_WIDTH | LVCF_TEXT | LVCF_SUBITEM; lvC.pszText = szText; /* Load the column labels from the resource file. */ for (index = 0; index < MAX_LIST_COLUMNS; index++) { lvC.iSubItem = index; lvC.cx = default_column_widths[index]; lvC.fmt = column_alignment[index]; LoadStringW(hInst, IDS_LIST_COLUMN_FIRST + index, szText, ARRAY_SIZE(szText)); if (ListView_InsertColumnW(hWndListView, index, &lvC) == -1) return FALSE; } return TRUE; } /* OnGetDispInfo - processes the LVN_GETDISPINFO notification message. */ void OnGetDispInfo(NMLVDISPINFOW *plvdi) { static WCHAR buffer[200]; static WCHAR reg_szT[] = {'R','E','G','_','S','Z',0}, reg_expand_szT[] = {'R','E','G','_','E','X','P','A','N','D','_','S','Z',0}, reg_binaryT[] = {'R','E','G','_','B','I','N','A','R','Y',0}, reg_dwordT[] = {'R','E','G','_','D','W','O','R','D',0}, reg_dword_big_endianT[] = {'R','E','G','_','D','W','O','R','D','_', 'B','I','G','_','E','N','D','I','A','N',0}, reg_multi_szT[] = {'R','E','G','_','M','U','L','T','I','_','S','Z',0}, reg_linkT[] = {'R','E','G','_','L','I','N','K',0}, reg_resource_listT[] = {'R','E','G','_','R','E','S','O','U','R','C','E','_','L','I','S','T',0}, reg_noneT[] = {'R','E','G','_','N','O','N','E',0}, emptyT[] = {0}; plvdi->item.pszText = NULL; plvdi->item.cchTextMax = 0; switch (plvdi->item.iSubItem) { case 0: plvdi->item.pszText = g_pszDefaultValueName; break; case 1: { DWORD data_type = ((LINE_INFO *)plvdi->item.lParam)->dwValType; switch (data_type) { case REG_SZ: plvdi->item.pszText = reg_szT; break; case REG_EXPAND_SZ: plvdi->item.pszText = reg_expand_szT; break; case REG_BINARY: plvdi->item.pszText = reg_binaryT; break; case REG_DWORD: plvdi->item.pszText = reg_dwordT; break; case REG_DWORD_BIG_ENDIAN: plvdi->item.pszText = reg_dword_big_endianT; break; case REG_MULTI_SZ: plvdi->item.pszText = reg_multi_szT; break; case REG_LINK: plvdi->item.pszText = reg_linkT; break; case REG_RESOURCE_LIST: plvdi->item.pszText = reg_resource_listT; break; case REG_NONE: plvdi->item.pszText = reg_noneT; break; default: { WCHAR fmt[] = {'0','x','%','x',0}; wsprintfW(buffer, fmt, data_type); plvdi->item.pszText = buffer; break; } } break; } case 2: plvdi->item.pszText = g_szValueNotSet; break; case 3: plvdi->item.pszText = emptyT; break; } } int CALLBACK CompareFunc(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort) { LINE_INFO*l, *r; l = (LINE_INFO*)lParam1; r = (LINE_INFO*)lParam2; if (!l->name) return -1; if (!r->name) return +1; if (g_columnToSort == ~0U) g_columnToSort = 0; if (g_columnToSort == 1) return g_invertSort ? (int)r->dwValType - (int)l->dwValType : (int)l->dwValType - (int)r->dwValType; if (g_columnToSort == 2) { /* FIXME: Sort on value */ return 0; } return g_invertSort ? lstrcmpiW(r->name, l->name) : lstrcmpiW(l->name, r->name); } HWND StartValueRename(HWND hwndLV) { int item; item = SendMessageW(hwndLV, LVM_GETNEXTITEM, -1, MAKELPARAM(LVNI_FOCUSED | LVNI_SELECTED, 0)); if (item < 1) { /* cannot rename default key */ MessageBeep(MB_ICONHAND); return 0; } return (HWND)SendMessageW(hwndLV, LVM_EDITLABELW, item, 0); } HWND CreateListView(HWND hwndParent, UINT id) { RECT rcClient; HWND hwndLV; WCHAR ListView[] = {'L','i','s','t',' ','V','i','e','w',0}; /* prepare strings */ LoadStringW(hInst, IDS_REGISTRY_VALUE_NOT_SET, g_szValueNotSet, ARRAY_SIZE(g_szValueNotSet)); /* Get the dimensions of the parent window's client area, and create the list view control. */ GetClientRect(hwndParent, &rcClient); hwndLV = CreateWindowExW(WS_EX_CLIENTEDGE, WC_LISTVIEWW, ListView, WS_VISIBLE | WS_CHILD | WS_TABSTOP | LVS_REPORT | LVS_EDITLABELS, 0, 0, rcClient.right, rcClient.bottom, hwndParent, ULongToHandle(id), hInst, NULL); if (!hwndLV) return NULL; SendMessageW(hwndLV, LVM_SETUNICODEFORMAT, TRUE, 0); SendMessageW(hwndLV, LVM_SETEXTENDEDLISTVIEWSTYLE, 0, LVS_EX_FULLROWSELECT); /* Initialize the image list */ if (!InitListViewImageList(hwndLV)) goto fail; if (!CreateListColumns(hwndLV)) goto fail; return hwndLV; fail: DestroyWindow(hwndLV); return NULL; } BOOL RefreshListView(HWND hwndLV, HKEY hKeyRoot, LPCWSTR keyPath, LPCWSTR highlightValue) { BOOL result = FALSE; DWORD max_sub_key_len; DWORD max_val_name_len, valNameLen; DWORD max_val_size, valSize; DWORD val_count, index, valType; WCHAR* valName = 0; BYTE* valBuf = 0; HKEY hKey = 0; LONG errCode; LVITEMW item; if (!hwndLV) return FALSE; SendMessageW(hwndLV, WM_SETREDRAW, FALSE, 0); errCode = RegOpenKeyExW(hKeyRoot, keyPath, 0, KEY_READ, &hKey); if (errCode != ERROR_SUCCESS) goto done; g_columnToSort = ~0U; SendMessageW(hwndLV, LVM_DELETEALLITEMS, 0, 0); /* get size information and resize the buffers if necessary */ errCode = RegQueryInfoKeyW(hKey, NULL, NULL, NULL, NULL, &max_sub_key_len, NULL, &val_count, &max_val_name_len, &max_val_size, NULL, NULL); if (errCode != ERROR_SUCCESS) goto done; /* account for the terminator char */ max_val_name_len++; max_val_size++; valName = heap_xalloc(max_val_name_len * sizeof(WCHAR)); valBuf = heap_xalloc(max_val_size); valSize = max_val_size; if (RegQueryValueExW(hKey, NULL, NULL, &valType, valBuf, &valSize) == ERROR_FILE_NOT_FOUND) { AddEntryToList(hwndLV, NULL, REG_SZ, NULL, 0, -1); } for(index = 0; index < val_count; index++) { valNameLen = max_val_name_len; valSize = max_val_size; valType = 0; errCode = RegEnumValueW(hKey, index, valName, &valNameLen, NULL, &valType, valBuf, &valSize); if (errCode != ERROR_SUCCESS) goto done; valBuf[valSize] = 0; AddEntryToList(hwndLV, valName[0] ? valName : NULL, valType, valBuf, valSize, -1); } memset(&item, 0, sizeof(item)); if (!highlightValue) { item.state = item.stateMask = LVIS_FOCUSED; SendMessageW(hwndLV, LVM_SETITEMSTATE, 0, (LPARAM)&item); } SendMessageW(hwndLV, LVM_SORTITEMS, (WPARAM)hwndLV, (LPARAM)CompareFunc); g_currentRootKey = hKeyRoot; if (keyPath != g_currentPath && !update_listview_path(keyPath)) goto done; result = TRUE; done: heap_free(valBuf); heap_free(valName); SendMessageW(hwndLV, WM_SETREDRAW, TRUE, 0); if (hKey) RegCloseKey(hKey); return result; }