2 * Registry editing UI functions.
4 * Copyright (C) 2003 Dimitrie O. Paun
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #define WIN32_LEAN_AND_MEAN /* Exclude rarely-used stuff from Windows headers */
33 #include "wine/unicode.h"
38 static const WCHAR* editValueName;
39 static WCHAR* stringValueData;
40 static BOOL isDecimal;
45 LPCWSTR lpszValueName;
50 static INT vmessagebox(HWND hwnd, INT buttons, INT titleId, INT resId, va_list ap)
52 static const WCHAR errorW[] = {'E','r','r','o','r',0};
53 static const WCHAR unknownW[] = {'U','n','k','n','o','w','n',' ','e','r','r','o','r',' ','s','t','r','i','n','g','!',0};
59 if (!LoadStringW(hInst, titleId, title, COUNT_OF(title))) lstrcpyW(title, errorW);
60 if (!LoadStringW(hInst, resId, errfmt, COUNT_OF(errfmt))) lstrcpyW(errfmt, unknownW);
62 vsnprintfW(errstr, COUNT_OF(errstr), errfmt, ap);
64 return MessageBoxW(hwnd, errstr, title, buttons);
67 static INT messagebox(HWND hwnd, INT buttons, INT titleId, INT resId, ...)
73 result = vmessagebox(hwnd, buttons, titleId, resId, ap);
79 void error(HWND hwnd, INT resId, ...)
84 vmessagebox(hwnd, MB_OK | MB_ICONERROR, IDS_ERROR, resId, ap);
88 static void error_code_messagebox(HWND hwnd, DWORD error_code)
93 static WCHAR fallback[] = {'E','r','r','o','r',' ','d','i','s','p','l','a','y','i','n','g',' ','e','r','r','o','r',' ','m','e','s','s','a','g','e','.','\n',0};
94 static const WCHAR title_error[] = {'E','r','r','o','r',0};
96 if (!LoadStringW(hInst, IDS_ERROR, title, COUNT_OF(title)))
97 lstrcpyW(title, title_error);
98 status = FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
99 NULL, error_code, 0, (LPWSTR)&lpMsgBuf, 0, NULL);
102 MessageBoxW(hwnd, lpMsgBuf, title, MB_OK | MB_ICONERROR);
103 if (lpMsgBuf != fallback)
107 static BOOL change_dword_base(HWND hwndDlg, BOOL toHex)
109 static const WCHAR percent_u[] = {'%','u',0};
110 static const WCHAR percent_x[] = {'%','x',0};
115 if (!GetDlgItemTextW(hwndDlg, IDC_VALUE_DATA, buf, COUNT_OF(buf))) return FALSE;
116 if (!swscanf(buf, toHex ? percent_u : percent_x, &val)) return FALSE;
117 wsprintfW(buf, toHex ? percent_x : percent_u, val);
118 return SetDlgItemTextW(hwndDlg, IDC_VALUE_DATA, buf);
121 static INT_PTR CALLBACK modify_dlgproc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
129 SetDlgItemTextW(hwndDlg, IDC_VALUE_NAME, editValueName);
130 SetDlgItemTextW(hwndDlg, IDC_VALUE_DATA, stringValueData);
131 CheckRadioButton(hwndDlg, IDC_DWORD_HEX, IDC_DWORD_DEC, isDecimal ? IDC_DWORD_DEC : IDC_DWORD_HEX);
134 switch (LOWORD(wParam)) {
136 if (isDecimal && change_dword_base(hwndDlg, TRUE)) isDecimal = FALSE;
139 if (!isDecimal && change_dword_base(hwndDlg, FALSE)) isDecimal = TRUE;
142 if ((hwndValue = GetDlgItem(hwndDlg, IDC_VALUE_DATA))) {
143 len = GetWindowTextLengthW(hwndValue);
144 if ((valueData = HeapReAlloc(GetProcessHeap(), 0, stringValueData, (len + 1) * sizeof(WCHAR)))) {
145 stringValueData = valueData;
146 if (!GetWindowTextW(hwndValue, stringValueData, len + 1))
147 *stringValueData = 0;
152 EndDialog(hwndDlg, wParam);
159 static INT_PTR CALLBACK bin_modify_dlgproc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
161 struct edit_params *params;
168 params = (struct edit_params *)lParam;
169 SetWindowLongPtrW(hwndDlg, DWLP_USER, (ULONG_PTR)params);
170 if (params->lpszValueName)
171 SetDlgItemTextW(hwndDlg, IDC_VALUE_NAME, params->lpszValueName);
173 SetDlgItemTextW(hwndDlg, IDC_VALUE_NAME, g_pszDefaultValueName);
174 SendDlgItemMessageW(hwndDlg, IDC_VALUE_DATA, HEM_SETDATA, (WPARAM)params->cbData, (LPARAM)params->pData);
177 switch (LOWORD(wParam)) {
179 params = (struct edit_params *)GetWindowLongPtrW(hwndDlg, DWLP_USER);
180 cbData = SendDlgItemMessageW(hwndDlg, IDC_VALUE_DATA, HEM_GETDATA, 0, 0);
181 pData = HeapAlloc(GetProcessHeap(), 0, cbData);
185 SendDlgItemMessageW(hwndDlg, IDC_VALUE_DATA, HEM_GETDATA, (WPARAM)cbData, (LPARAM)pData);
186 lRet = RegSetValueExW(params->hKey, params->lpszValueName, 0, REG_BINARY, pData, cbData);
189 lRet = ERROR_OUTOFMEMORY;
191 if (lRet == ERROR_SUCCESS)
192 EndDialog(hwndDlg, 1);
195 error_code_messagebox(hwndDlg, lRet);
196 EndDialog(hwndDlg, 0);
200 EndDialog(hwndDlg, 0);
207 static BOOL check_value(HWND hwnd, HKEY hKey, LPCWSTR valueName)
210 LONG lRet = RegQueryValueExW(hKey, valueName ? valueName : &empty, 0, NULL, 0, NULL);
211 if(lRet != ERROR_SUCCESS) return FALSE;
215 static LPWSTR read_value(HWND hwnd, HKEY hKey, LPCWSTR valueName, DWORD *lpType, LONG *len)
218 LPWSTR buffer = NULL;
222 lRet = RegQueryValueExW(hKey, valueName ? valueName : &empty, 0, lpType, 0, &valueDataLen);
223 if (lRet != ERROR_SUCCESS) {
224 if (lRet == ERROR_FILE_NOT_FOUND && !valueName) { /* no default value here, make it up */
226 if (lpType) *lpType = REG_SZ;
227 buffer = HeapAlloc(GetProcessHeap(), 0, sizeof(WCHAR));
231 error(hwnd, IDS_BAD_VALUE, valueName);
234 if ( *lpType == REG_DWORD ) valueDataLen = sizeof(DWORD);
235 if (!(buffer = HeapAlloc(GetProcessHeap(), 0, valueDataLen+sizeof(WCHAR)))) {
236 error(hwnd, IDS_TOO_BIG_VALUE, valueDataLen);
239 lRet = RegQueryValueExW(hKey, valueName, 0, 0, (LPBYTE)buffer, &valueDataLen);
240 if (lRet != ERROR_SUCCESS) {
241 error(hwnd, IDS_BAD_VALUE, valueName);
244 if((valueDataLen % sizeof(WCHAR)) == 0)
245 buffer[valueDataLen / sizeof(WCHAR)] = 0;
246 if(len) *len = valueDataLen;
250 HeapFree(GetProcessHeap(), 0, buffer);
254 BOOL CreateKey(HWND hwnd, HKEY hKeyRoot, LPCWSTR keyPath, LPWSTR keyName)
257 LONG lRet = ERROR_SUCCESS;
259 WCHAR newKey[MAX_NEW_KEY_LEN - 4];
263 lRet = RegOpenKeyExW(hKeyRoot, keyPath, 0, KEY_CREATE_SUB_KEY, &hKey);
264 if (lRet != ERROR_SUCCESS) {
265 error_code_messagebox(hwnd, lRet);
269 if (!LoadStringW(GetModuleHandleW(0), IDS_NEWKEY, newKey, COUNT_OF(newKey))) goto done;
271 /* try to find out a name for the newly create key (max 100 times) */
272 for (keyNum = 1; keyNum < 100; keyNum++) {
273 wsprintfW(keyName, newKey, keyNum);
274 lRet = RegOpenKeyW(hKey, keyName, &retKey);
275 if (lRet != ERROR_SUCCESS) break;
278 if (lRet == ERROR_SUCCESS) goto done;
280 lRet = RegCreateKeyW(hKey, keyName, &retKey);
281 if (lRet != ERROR_SUCCESS) {
282 error_code_messagebox(hwnd, lRet);
293 BOOL ModifyValue(HWND hwnd, HKEY hKeyRoot, LPCWSTR keyPath, LPCWSTR valueName)
301 lRet = RegOpenKeyExW(hKeyRoot, keyPath, 0, KEY_READ | KEY_SET_VALUE, &hKey);
302 if (lRet != ERROR_SUCCESS) {
303 error_code_messagebox(hwnd, lRet);
307 editValueName = valueName ? valueName : g_pszDefaultValueName;
308 if(!(stringValueData = read_value(hwnd, hKey, valueName, &type, &len))) goto done;
310 if ( (type == REG_SZ) || (type == REG_EXPAND_SZ) ) {
311 if (DialogBoxW(0, MAKEINTRESOURCEW(IDD_EDIT_STRING), hwnd, modify_dlgproc) == IDOK) {
312 lRet = RegSetValueExW(hKey, valueName, 0, type, (LPBYTE)stringValueData, (lstrlenW(stringValueData) + 1) * sizeof(WCHAR));
313 if (lRet == ERROR_SUCCESS) result = TRUE;
314 else error_code_messagebox(hwnd, lRet);
316 } else if ( type == REG_DWORD ) {
317 const WCHAR u[] = {'%','u',0};
318 const WCHAR x[] = {'%','x',0};
319 wsprintfW(stringValueData, isDecimal ? u : x, *((DWORD*)stringValueData));
320 if (DialogBoxW(0, MAKEINTRESOURCEW(IDD_EDIT_DWORD), hwnd, modify_dlgproc) == IDOK) {
322 CHAR* valueA = GetMultiByteString(stringValueData);
323 if (_stscanf(valueA, isDecimal ? "%u" : "%x", &val)) {
324 lRet = RegSetValueExW(hKey, valueName, 0, type, (BYTE*)&val, sizeof(val));
325 if (lRet == ERROR_SUCCESS) result = TRUE;
326 else error_code_messagebox(hwnd, lRet);
328 HeapFree(GetProcessHeap(), 0, valueA);
330 } else if ( type == REG_BINARY ) {
331 struct edit_params params;
333 params.lpszValueName = valueName;
334 params.pData = stringValueData;
336 result = DialogBoxParamW(NULL, MAKEINTRESOURCEW(IDD_EDIT_BINARY), hwnd,
337 bin_modify_dlgproc, (LPARAM)¶ms);
338 } else if ( type == REG_MULTI_SZ ) {
339 WCHAR char1 = '\r', char2 = '\n';
340 WCHAR *tmpValueData = NULL;
343 for ( i = 0, count = 0; i < len - 1; i++)
344 if ( !stringValueData[i] && stringValueData[i + 1] )
346 tmpValueData = HeapAlloc( GetProcessHeap(), 0, ( len + count ) * sizeof(WCHAR));
347 if ( !tmpValueData ) goto done;
349 for ( i = 0, j = 0; i < len - 1; i++)
351 if ( !stringValueData[i] && stringValueData[i + 1])
353 tmpValueData[j++] = char1;
354 tmpValueData[j++] = char2;
357 tmpValueData[j++] = stringValueData[i];
359 tmpValueData[j] = stringValueData[i];
360 HeapFree( GetProcessHeap(), 0, stringValueData);
361 stringValueData = tmpValueData;
364 if (DialogBoxW(0, MAKEINTRESOURCEW(IDD_EDIT_MULTI_STRING), hwnd, modify_dlgproc) == IDOK)
366 len = lstrlenW( stringValueData );
367 tmpValueData = HeapAlloc( GetProcessHeap(), 0, (len + 2) * sizeof(WCHAR));
368 if ( !tmpValueData ) goto done;
370 for ( i = 0, j = 0; i < len - 1; i++)
372 if ( stringValueData[i] == char1 && stringValueData[i + 1] == char2)
374 if ( tmpValueData[j - 1] != 0)
375 tmpValueData[j++] = 0;
379 tmpValueData[j++] = stringValueData[i];
381 tmpValueData[j++] = stringValueData[i];
382 tmpValueData[j++] = 0;
383 tmpValueData[j++] = 0;
384 HeapFree( GetProcessHeap(), 0, stringValueData);
385 stringValueData = tmpValueData;
387 lRet = RegSetValueExW(hKey, valueName, 0, type, (LPBYTE)stringValueData, j * sizeof(WCHAR));
388 if (lRet == ERROR_SUCCESS) result = TRUE;
389 else error_code_messagebox(hwnd, lRet);
392 error(hwnd, IDS_UNSUPPORTED_TYPE, type);
396 HeapFree(GetProcessHeap(), 0, stringValueData);
397 stringValueData = NULL;
402 BOOL DeleteKey(HWND hwnd, HKEY hKeyRoot, LPCWSTR keyPath)
408 lRet = RegOpenKeyExW(hKeyRoot, keyPath, 0, KEY_READ|KEY_SET_VALUE, &hKey);
409 if (lRet != ERROR_SUCCESS) {
410 error_code_messagebox(hwnd, lRet);
414 if (messagebox(hwnd, MB_YESNO | MB_ICONEXCLAMATION, IDS_DELETE_BOX_TITLE, IDS_DELETE_BOX_TEXT, keyPath) != IDYES)
417 lRet = SHDeleteKeyW(hKeyRoot, keyPath);
418 if (lRet != ERROR_SUCCESS) {
419 error(hwnd, IDS_BAD_KEY, keyPath);
429 BOOL DeleteValue(HWND hwnd, HKEY hKeyRoot, LPCWSTR keyPath, LPCWSTR valueName, BOOL showMessageBox)
434 LPCWSTR visibleValueName = valueName ? valueName : g_pszDefaultValueName;
437 lRet = RegOpenKeyExW(hKeyRoot, keyPath, 0, KEY_READ | KEY_SET_VALUE, &hKey);
438 if (lRet != ERROR_SUCCESS) return FALSE;
442 if (messagebox(hwnd, MB_YESNO | MB_ICONEXCLAMATION, IDS_DELETE_BOX_TITLE, IDS_DELETE_BOX_TEXT, visibleValueName) != IDYES)
446 lRet = RegDeleteValueW(hKey, valueName ? valueName : &empty);
447 if (lRet != ERROR_SUCCESS && valueName) {
448 error(hwnd, IDS_BAD_VALUE, valueName);
450 if (lRet != ERROR_SUCCESS) goto done;
458 BOOL CreateValue(HWND hwnd, HKEY hKeyRoot, LPCWSTR keyPath, DWORD valueType, LPWSTR valueName)
460 LONG lRet = ERROR_SUCCESS;
462 DWORD valueDword = 0;
467 lRet = RegOpenKeyExW(hKeyRoot, keyPath, 0, KEY_READ | KEY_SET_VALUE, &hKey);
468 if (lRet != ERROR_SUCCESS) {
469 error_code_messagebox(hwnd, lRet);
473 if (!LoadStringW(GetModuleHandleW(0), IDS_NEWVALUE, newValue, COUNT_OF(newValue))) goto done;
475 /* try to find out a name for the newly create key (max 100 times) */
476 for (valueNum = 1; valueNum < 100; valueNum++) {
477 wsprintfW(valueName, newValue, valueNum);
478 lRet = RegQueryValueExW(hKey, valueName, 0, 0, 0, 0);
479 if (lRet == ERROR_FILE_NOT_FOUND) break;
481 if (lRet != ERROR_FILE_NOT_FOUND) {
482 error_code_messagebox(hwnd, lRet);
486 lRet = RegSetValueExW(hKey, valueName, 0, valueType, (BYTE*)&valueDword, sizeof(DWORD));
487 if (lRet != ERROR_SUCCESS) {
488 error_code_messagebox(hwnd, lRet);
498 BOOL RenameValue(HWND hwnd, HKEY hKeyRoot, LPCWSTR keyPath, LPCWSTR oldName, LPCWSTR newName)
506 if (!oldName) return FALSE;
507 if (!newName) return FALSE;
509 lRet = RegOpenKeyExW(hKeyRoot, keyPath, 0, KEY_READ | KEY_SET_VALUE, &hKey);
510 if (lRet != ERROR_SUCCESS) {
511 error_code_messagebox(hwnd, lRet);
514 /* check if value already exists */
515 if (check_value(hwnd, hKey, newName)) goto done;
516 value = read_value(hwnd, hKey, oldName, &type, &len);
517 if(!value) goto done;
518 lRet = RegSetValueExW(hKey, newName, 0, type, (BYTE*)value, len);
519 if (lRet != ERROR_SUCCESS) {
520 error_code_messagebox(hwnd, lRet);
523 lRet = RegDeleteValueW(hKey, oldName);
524 if (lRet != ERROR_SUCCESS) {
525 RegDeleteValueW(hKey, newName);
526 error_code_messagebox(hwnd, lRet);
532 HeapFree(GetProcessHeap(), 0, value);
538 BOOL RenameKey(HWND hwnd, HKEY hRootKey, LPCWSTR keyPath, LPCWSTR newName)
540 LPWSTR parentPath = 0;
541 LPCWSTR srcSubKey = 0;
548 if (!keyPath || !newName) return FALSE;
550 if (!strrchrW(keyPath, '\\')) {
551 parentKey = hRootKey;
554 LPWSTR srcSubKey_copy;
556 parentPath = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(keyPath)+1)*sizeof(WCHAR));
557 lstrcpyW(parentPath, keyPath);
558 srcSubKey_copy = strrchrW(parentPath, '\\');
560 srcSubKey = srcSubKey_copy + 1;
561 lRet = RegOpenKeyExW(hRootKey, parentPath, 0, KEY_READ | KEY_CREATE_SUB_KEY, &parentKey);
562 if (lRet != ERROR_SUCCESS) {
563 error_code_messagebox(hwnd, lRet);
568 /* The following fails if the old name is the same as the new name. */
569 if (!lstrcmpW(srcSubKey, newName)) goto done;
571 lRet = RegCreateKeyExW(parentKey, newName, 0, NULL, REG_OPTION_NON_VOLATILE,
572 KEY_WRITE, NULL /* FIXME */, &destKey, &disposition);
573 if (disposition == REG_OPENED_EXISTING_KEY)
574 lRet = ERROR_FILE_EXISTS; /* FIXME: we might want a better error message than this */
575 if (lRet != ERROR_SUCCESS) {
576 error_code_messagebox(hwnd, lRet);
580 /* FIXME: SHCopyKey does not copy the security attributes */
581 lRet = SHCopyKeyW(parentKey, srcSubKey, destKey, 0);
582 if (lRet != ERROR_SUCCESS) {
583 RegCloseKey(destKey);
584 RegDeleteKeyW(parentKey, newName);
585 error_code_messagebox(hwnd, lRet);
589 lRet = SHDeleteKeyW(hRootKey, keyPath);
590 if (lRet != ERROR_SUCCESS) {
591 error_code_messagebox(hwnd, lRet);
598 RegCloseKey(destKey);
600 RegCloseKey(parentKey);
601 HeapFree(GetProcessHeap(), 0, parentPath);