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 */
37 static const TCHAR* editValueName;
38 static TCHAR* stringValueData;
39 static BOOL isDecimal;
44 LPCTSTR lpszValueName;
49 static INT vmessagebox(HWND hwnd, INT buttons, INT titleId, INT resId, va_list ap)
55 if (!LoadString(hInst, titleId, title, COUNT_OF(title)))
56 lstrcpy(title, "Error");
58 if (!LoadString(hInst, resId, errfmt, COUNT_OF(errfmt)))
59 lstrcpy(errfmt, "Unknown error string!");
61 _vsntprintf(errstr, COUNT_OF(errstr), errfmt, ap);
63 return MessageBox(hwnd, errstr, title, buttons);
66 static INT messagebox(HWND hwnd, INT buttons, INT titleId, INT resId, ...)
72 result = vmessagebox(hwnd, buttons, titleId, resId, ap);
78 void error(HWND hwnd, INT resId, ...)
83 vmessagebox(hwnd, MB_OK | MB_ICONERROR, IDS_ERROR, resId, ap);
87 static void error_code_messagebox(HWND hwnd, DWORD error_code)
92 static TCHAR fallback[] = TEXT("Error displaying error message.\n");
93 if (!LoadString(hInst, IDS_ERROR, title, COUNT_OF(title)))
94 lstrcpy(title, TEXT("Error"));
95 status = FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
96 NULL, error_code, 0, (LPTSTR)&lpMsgBuf, 0, NULL);
99 MessageBox(hwnd, lpMsgBuf, title, MB_OK | MB_ICONERROR);
100 if (lpMsgBuf != fallback)
104 static BOOL change_dword_base(HWND hwndDlg, BOOL toHex)
109 if (!GetDlgItemText(hwndDlg, IDC_VALUE_DATA, buf, COUNT_OF(buf))) return FALSE;
110 if (!_stscanf(buf, toHex ? "%u" : "%x", &val)) return FALSE;
111 wsprintf(buf, toHex ? "%x" : "%u", val);
112 return SetDlgItemText(hwndDlg, IDC_VALUE_DATA, buf);
115 static INT_PTR CALLBACK modify_dlgproc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
123 SetDlgItemText(hwndDlg, IDC_VALUE_NAME, editValueName);
124 SetDlgItemText(hwndDlg, IDC_VALUE_DATA, stringValueData);
125 CheckRadioButton(hwndDlg, IDC_DWORD_HEX, IDC_DWORD_DEC, isDecimal ? IDC_DWORD_DEC : IDC_DWORD_HEX);
128 switch (LOWORD(wParam)) {
130 if (isDecimal && change_dword_base(hwndDlg, TRUE)) isDecimal = FALSE;
133 if (!isDecimal && change_dword_base(hwndDlg, FALSE)) isDecimal = TRUE;
136 if ((hwndValue = GetDlgItem(hwndDlg, IDC_VALUE_DATA))) {
137 len = GetWindowTextLength(hwndValue);
138 if ((valueData = HeapReAlloc(GetProcessHeap(), 0, stringValueData, (len + 1) * sizeof(TCHAR)))) {
139 stringValueData = valueData;
140 if (!GetWindowText(hwndValue, stringValueData, len + 1))
141 *stringValueData = 0;
146 EndDialog(hwndDlg, wParam);
153 static INT_PTR CALLBACK bin_modify_dlgproc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
155 struct edit_params *params;
162 params = (struct edit_params *)lParam;
163 SetWindowLongPtr(hwndDlg, DWLP_USER, (ULONG_PTR)params);
164 if (params->lpszValueName)
165 SetDlgItemText(hwndDlg, IDC_VALUE_NAME, params->lpszValueName);
167 SetDlgItemText(hwndDlg, IDC_VALUE_NAME, g_pszDefaultValueName);
168 SendDlgItemMessage(hwndDlg, IDC_VALUE_DATA, HEM_SETDATA, (WPARAM)params->cbData, (LPARAM)params->pData);
171 switch (LOWORD(wParam)) {
173 params = (struct edit_params *)GetWindowLongPtr(hwndDlg, DWLP_USER);
174 cbData = SendDlgItemMessage(hwndDlg, IDC_VALUE_DATA, HEM_GETDATA, 0, 0);
175 pData = HeapAlloc(GetProcessHeap(), 0, cbData);
179 SendDlgItemMessage(hwndDlg, IDC_VALUE_DATA, HEM_GETDATA, (WPARAM)cbData, (LPARAM)pData);
180 lRet = RegSetValueEx(params->hKey, params->lpszValueName, 0, REG_BINARY, pData, cbData);
183 lRet = ERROR_OUTOFMEMORY;
185 if (lRet == ERROR_SUCCESS)
186 EndDialog(hwndDlg, 1);
189 error_code_messagebox(hwndDlg, lRet);
190 EndDialog(hwndDlg, 0);
194 EndDialog(hwndDlg, 0);
201 static BOOL check_value(HWND hwnd, HKEY hKey, LPCTSTR valueName)
203 LONG lRet = RegQueryValueEx(hKey, valueName ? valueName : _T(""), 0, NULL, 0, NULL);
204 if(lRet != ERROR_SUCCESS) return FALSE;
208 static LPTSTR read_value(HWND hwnd, HKEY hKey, LPCTSTR valueName, DWORD *lpType, LONG *len)
211 LPTSTR buffer = NULL;
214 lRet = RegQueryValueEx(hKey, valueName ? valueName : _T(""), 0, lpType, 0, &valueDataLen);
215 if (lRet != ERROR_SUCCESS) {
216 if (lRet == ERROR_FILE_NOT_FOUND && !valueName) { /* no default value here, make it up */
218 if (lpType) *lpType = REG_SZ;
219 buffer = HeapAlloc(GetProcessHeap(), 0, 1);
223 error(hwnd, IDS_BAD_VALUE, valueName);
226 if ( *lpType == REG_DWORD ) valueDataLen = sizeof(DWORD);
227 if (!(buffer = HeapAlloc(GetProcessHeap(), 0, valueDataLen+1))) {
228 error(hwnd, IDS_TOO_BIG_VALUE, valueDataLen);
231 lRet = RegQueryValueEx(hKey, valueName, 0, 0, (LPBYTE)buffer, &valueDataLen);
232 if (lRet != ERROR_SUCCESS) {
233 error(hwnd, IDS_BAD_VALUE, valueName);
236 buffer[valueDataLen] = 0;
237 if(len) *len = valueDataLen;
241 HeapFree(GetProcessHeap(), 0, buffer);
245 BOOL CreateKey(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, LPTSTR keyName)
248 LONG lRet = ERROR_SUCCESS;
250 TCHAR newKey[MAX_NEW_KEY_LEN - 4];
254 lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_CREATE_SUB_KEY, &hKey);
255 if (lRet != ERROR_SUCCESS) {
256 error_code_messagebox(hwnd, lRet);
260 if (!LoadString(GetModuleHandle(0), IDS_NEWKEY, newKey, COUNT_OF(newKey))) goto done;
262 /* try to find out a name for the newly create key (max 100 times) */
263 for (keyNum = 1; keyNum < 100; keyNum++) {
264 wsprintf(keyName, newKey, keyNum);
265 lRet = RegOpenKey(hKey, keyName, &retKey);
266 if (lRet != ERROR_SUCCESS) break;
269 if (lRet == ERROR_SUCCESS) goto done;
271 lRet = RegCreateKey(hKey, keyName, &retKey);
272 if (lRet != ERROR_SUCCESS) {
273 error_code_messagebox(hwnd, lRet);
284 BOOL ModifyValue(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, LPCTSTR valueName)
292 lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ | KEY_SET_VALUE, &hKey);
293 if (lRet != ERROR_SUCCESS) {
294 error_code_messagebox(hwnd, lRet);
298 editValueName = valueName ? valueName : g_pszDefaultValueName;
299 if(!(stringValueData = read_value(hwnd, hKey, valueName, &type, &len))) goto done;
301 if ( (type == REG_SZ) || (type == REG_EXPAND_SZ) ) {
302 if (DialogBox(0, MAKEINTRESOURCE(IDD_EDIT_STRING), hwnd, modify_dlgproc) == IDOK) {
303 lRet = RegSetValueEx(hKey, valueName, 0, type, (LPBYTE)stringValueData, lstrlen(stringValueData) + 1);
304 if (lRet == ERROR_SUCCESS) result = TRUE;
305 else error_code_messagebox(hwnd, lRet);
307 } else if ( type == REG_DWORD ) {
308 wsprintf(stringValueData, isDecimal ? "%u" : "%x", *((DWORD*)stringValueData));
309 if (DialogBox(0, MAKEINTRESOURCE(IDD_EDIT_DWORD), hwnd, modify_dlgproc) == IDOK) {
311 if (_stscanf(stringValueData, isDecimal ? "%u" : "%x", &val)) {
312 lRet = RegSetValueEx(hKey, valueName, 0, type, (BYTE*)&val, sizeof(val));
313 if (lRet == ERROR_SUCCESS) result = TRUE;
314 else error_code_messagebox(hwnd, lRet);
317 } else if ( type == REG_BINARY ) {
318 struct edit_params params;
320 params.lpszValueName = valueName;
321 params.pData = stringValueData;
323 result = DialogBoxParam(NULL, MAKEINTRESOURCE(IDD_EDIT_BINARY), hwnd,
324 bin_modify_dlgproc, (LPARAM)¶ms);
325 } else if ( type == REG_MULTI_SZ ) {
326 TCHAR char1 = (TCHAR)'\r', char2 = (TCHAR)'\n';
327 TCHAR *tmpValueData = NULL;
330 for ( i = 0, count = 0; i < len - 1; i++)
331 if ( !stringValueData[i] && stringValueData[i + 1] )
333 tmpValueData = HeapAlloc( GetProcessHeap(), 0, ( len + count ) * sizeof(TCHAR));
334 if ( !tmpValueData ) goto done;
336 for ( i = 0, j = 0; i < len - 1; i++)
338 if ( !stringValueData[i] && stringValueData[i + 1])
340 tmpValueData[j++] = char1;
341 tmpValueData[j++] = char2;
344 tmpValueData[j++] = stringValueData[i];
346 tmpValueData[j] = stringValueData[i];
347 HeapFree( GetProcessHeap(), 0, stringValueData);
348 stringValueData = tmpValueData;
351 if (DialogBox(0, MAKEINTRESOURCE(IDD_EDIT_MULTI_STRING), hwnd, modify_dlgproc) == IDOK)
353 len = lstrlen( stringValueData );
354 tmpValueData = HeapAlloc( GetProcessHeap(), 0, (len + 2) * sizeof(TCHAR));
355 if ( !tmpValueData ) goto done;
357 for ( i = 0, j = 0; i < len - 1; i++)
359 if ( stringValueData[i] == char1 && stringValueData[i + 1] == char2)
361 if ( tmpValueData[j - 1] != 0)
362 tmpValueData[j++] = 0;
366 tmpValueData[j++] = stringValueData[i];
368 tmpValueData[j++] = stringValueData[i];
369 tmpValueData[j++] = 0;
370 tmpValueData[j++] = 0;
371 HeapFree( GetProcessHeap(), 0, stringValueData);
372 stringValueData = tmpValueData;
374 lRet = RegSetValueEx(hKey, valueName, 0, type, (LPBYTE)stringValueData, j * sizeof(TCHAR));
375 if (lRet == ERROR_SUCCESS) result = TRUE;
376 else error_code_messagebox(hwnd, lRet);
379 error(hwnd, IDS_UNSUPPORTED_TYPE, type);
383 HeapFree(GetProcessHeap(), 0, stringValueData);
384 stringValueData = NULL;
389 BOOL DeleteKey(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath)
395 lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ|KEY_SET_VALUE, &hKey);
396 if (lRet != ERROR_SUCCESS) {
397 error_code_messagebox(hwnd, lRet);
401 if (messagebox(hwnd, MB_YESNO | MB_ICONEXCLAMATION, IDS_DELETE_BOX_TITLE, IDS_DELETE_BOX_TEXT, keyPath) != IDYES)
404 lRet = SHDeleteKey(hKeyRoot, keyPath);
405 if (lRet != ERROR_SUCCESS) {
406 error(hwnd, IDS_BAD_KEY, keyPath);
416 BOOL DeleteValue(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, LPCTSTR valueName, BOOL showMessageBox)
421 LPCSTR visibleValueName = valueName ? valueName : g_pszDefaultValueName;
423 lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ | KEY_SET_VALUE, &hKey);
424 if (lRet != ERROR_SUCCESS) return FALSE;
427 if (messagebox(hwnd, MB_YESNO | MB_ICONEXCLAMATION, IDS_DELETE_BOX_TITLE, IDS_DELETE_BOX_TEXT, visibleValueName) != IDYES)
430 lRet = RegDeleteValue(hKey, valueName ? valueName : "");
431 if (lRet != ERROR_SUCCESS && valueName) {
432 error(hwnd, IDS_BAD_VALUE, valueName);
434 if (lRet != ERROR_SUCCESS) goto done;
442 BOOL CreateValue(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, DWORD valueType, LPTSTR valueName)
444 LONG lRet = ERROR_SUCCESS;
446 DWORD valueDword = 0;
451 lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ | KEY_SET_VALUE, &hKey);
452 if (lRet != ERROR_SUCCESS) {
453 error_code_messagebox(hwnd, lRet);
457 if (!LoadString(GetModuleHandle(0), IDS_NEWVALUE, newValue, COUNT_OF(newValue))) goto done;
459 /* try to find out a name for the newly create key (max 100 times) */
460 for (valueNum = 1; valueNum < 100; valueNum++) {
461 wsprintf(valueName, newValue, valueNum);
462 lRet = RegQueryValueEx(hKey, valueName, 0, 0, 0, 0);
463 if (lRet == ERROR_FILE_NOT_FOUND) break;
465 if (lRet != ERROR_FILE_NOT_FOUND) {
466 error_code_messagebox(hwnd, lRet);
470 lRet = RegSetValueEx(hKey, valueName, 0, valueType, (BYTE*)&valueDword, sizeof(DWORD));
471 if (lRet != ERROR_SUCCESS) {
472 error_code_messagebox(hwnd, lRet);
482 BOOL RenameValue(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, LPCTSTR oldName, LPCTSTR newName)
490 if (!oldName) return FALSE;
491 if (!newName) return FALSE;
493 lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ | KEY_SET_VALUE, &hKey);
494 if (lRet != ERROR_SUCCESS) {
495 error_code_messagebox(hwnd, lRet);
498 /* check if value already exists */
499 if (check_value(hwnd, hKey, newName)) goto done;
500 value = read_value(hwnd, hKey, oldName, &type, &len);
501 if(!value) goto done;
502 lRet = RegSetValueEx(hKey, newName, 0, type, (BYTE*)value, len);
503 if (lRet != ERROR_SUCCESS) {
504 error_code_messagebox(hwnd, lRet);
507 lRet = RegDeleteValue(hKey, oldName);
508 if (lRet != ERROR_SUCCESS) {
509 RegDeleteValue(hKey, newName);
510 error_code_messagebox(hwnd, lRet);
516 HeapFree(GetProcessHeap(), 0, value);
522 BOOL RenameKey(HWND hwnd, HKEY hRootKey, LPCTSTR keyPath, LPCTSTR newName)
524 LPTSTR parentPath = 0;
525 LPCTSTR srcSubKey = 0;
532 if (!keyPath || !newName) return FALSE;
534 if (!strrchr(keyPath, '\\')) {
535 parentKey = hRootKey;
538 LPTSTR srcSubKey_copy;
540 parentPath = strdup(keyPath);
541 srcSubKey_copy = strrchr(parentPath, '\\');
543 srcSubKey = srcSubKey_copy + 1;
544 lRet = RegOpenKeyEx(hRootKey, parentPath, 0, KEY_READ | KEY_CREATE_SUB_KEY, &parentKey);
545 if (lRet != ERROR_SUCCESS) {
546 error_code_messagebox(hwnd, lRet);
551 /* The following fails if the old name is the same as the new name. */
552 if (!strcmp(srcSubKey, newName)) goto done;
554 lRet = RegCreateKeyEx(parentKey, newName, 0, NULL, REG_OPTION_NON_VOLATILE,
555 KEY_WRITE, NULL /* FIXME */, &destKey, &disposition);
556 if (disposition == REG_OPENED_EXISTING_KEY)
557 lRet = ERROR_FILE_EXISTS; /* FIXME: we might want a better error message than this */
558 if (lRet != ERROR_SUCCESS) {
559 error_code_messagebox(hwnd, lRet);
563 /* FIXME: SHCopyKey does not copy the security attributes */
564 lRet = SHCopyKey(parentKey, srcSubKey, destKey, 0);
565 if (lRet != ERROR_SUCCESS) {
566 RegCloseKey(destKey);
567 RegDeleteKey(parentKey, newName);
568 error_code_messagebox(hwnd, lRet);
572 lRet = SHDeleteKey(hRootKey, keyPath);
573 if (lRet != ERROR_SUCCESS) {
574 error_code_messagebox(hwnd, lRet);
581 RegCloseKey(destKey);
583 RegCloseKey(parentKey);