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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21 #define WIN32_LEAN_AND_MEAN /* Exclude rarely-used stuff from Windows headers */
39 typedef enum _EDIT_MODE
46 static const TCHAR
* editValueName
;
47 static TCHAR
* stringValueData
;
48 static PVOID binValueData
;
49 static DWORD dwordValueData
;
50 static DWORD valueDataLen
;
51 static EDIT_MODE dwordEditMode
= EDIT_MODE_HEX
;
54 void error(HWND hwnd
, INT resId
, ...)
62 hInstance
= GetModuleHandle(0);
64 if (!LoadString(hInstance
, IDS_ERROR
, title
, COUNT_OF(title
)))
65 _tcscpy(title
, _T("Error"));
67 if (!LoadString(hInstance
, resId
, errfmt
, COUNT_OF(errfmt
)))
68 _tcscpy(errfmt
, _T("Unknown error string!"));
71 _vsntprintf(errstr
, COUNT_OF(errstr
), errfmt
, ap
);
74 MessageBox(hwnd
, errstr
, title
, MB_OK
| MB_ICONERROR
);
77 void warning(HWND hwnd
, INT resId
, ...)
85 hInstance
= GetModuleHandle(0);
87 if (!LoadString(hInstance
, IDS_WARNING
, title
, COUNT_OF(title
)))
88 _tcscpy(title
, _T("Warning"));
90 if (!LoadString(hInstance
, resId
, errfmt
, COUNT_OF(errfmt
)))
91 _tcscpy(errfmt
, _T("Unknown error string!"));
94 _vsntprintf(errstr
, COUNT_OF(errstr
), errfmt
, ap
);
97 MessageBox(hwnd
, errstr
, title
, MB_OK
| MB_ICONSTOP
);
100 INT_PTR CALLBACK
modify_string_dlgproc(HWND hwndDlg
, UINT uMsg
, WPARAM wParam
, LPARAM lParam
)
108 if(editValueName
&& _tcscmp(editValueName
, _T("")))
110 SetDlgItemText(hwndDlg
, IDC_VALUE_NAME
, editValueName
);
115 LoadString(hInst
, IDS_DEFAULT_VALUE_NAME
, buffer
, sizeof(buffer
)/sizeof(TCHAR
));
116 SetDlgItemText(hwndDlg
, IDC_VALUE_NAME
, buffer
);
118 SetDlgItemText(hwndDlg
, IDC_VALUE_DATA
, stringValueData
);
119 SetFocus(GetDlgItem(hwndDlg
, IDC_VALUE_DATA
));
122 switch (LOWORD(wParam
))
125 if ((hwndValue
= GetDlgItem(hwndDlg
, IDC_VALUE_DATA
)))
127 if ((len
= GetWindowTextLength(hwndValue
)))
131 if ((valueData
= HeapReAlloc(GetProcessHeap(), 0, stringValueData
, (len
+ 1) * sizeof(TCHAR
))))
133 stringValueData
= valueData
;
134 if (!GetWindowText(hwndValue
, stringValueData
, len
+ 1))
135 *stringValueData
= 0;
140 if ((valueData
= HeapAlloc(GetProcessHeap(), 0, (len
+ 1) * sizeof(TCHAR
))))
142 stringValueData
= valueData
;
143 if (!GetWindowText(hwndValue
, stringValueData
, len
+ 1))
144 *stringValueData
= 0;
151 *stringValueData
= 0;
154 EndDialog(hwndDlg
, IDOK
);
157 EndDialog(hwndDlg
, IDCANCEL
);
165 INT_PTR CALLBACK
modify_multi_string_dlgproc(HWND hwndDlg
, UINT uMsg
, WPARAM wParam
, LPARAM lParam
)
173 if(editValueName
&& _tcscmp(editValueName
, _T("")))
175 SetDlgItemText(hwndDlg
, IDC_VALUE_NAME
, editValueName
);
180 LoadString(hInst
, IDS_DEFAULT_VALUE_NAME
, buffer
, sizeof(buffer
)/sizeof(TCHAR
));
181 SetDlgItemText(hwndDlg
, IDC_VALUE_NAME
, buffer
);
183 SetDlgItemText(hwndDlg
, IDC_VALUE_DATA
, stringValueData
);
184 SetFocus(GetDlgItem(hwndDlg
, IDC_VALUE_DATA
));
187 switch (LOWORD(wParam
))
190 if ((hwndValue
= GetDlgItem(hwndDlg
, IDC_VALUE_DATA
)))
192 if ((len
= GetWindowTextLength(hwndValue
)))
196 if ((valueData
= HeapReAlloc(GetProcessHeap(), 0, stringValueData
, (len
+ 1) * sizeof(TCHAR
))))
198 stringValueData
= valueData
;
199 if (!GetWindowText(hwndValue
, stringValueData
, len
+ 1))
200 *stringValueData
= 0;
205 if ((valueData
= HeapAlloc(GetProcessHeap(), 0, (len
+ 1) * sizeof(TCHAR
))))
207 stringValueData
= valueData
;
208 if (!GetWindowText(hwndValue
, stringValueData
, len
+ 1))
209 *stringValueData
= 0;
216 *stringValueData
= 0;
219 EndDialog(hwndDlg
, IDOK
);
222 EndDialog(hwndDlg
, IDCANCEL
);
230 LRESULT CALLBACK
DwordEditSubclassProc(HWND hwnd
, UINT uMsg
, WPARAM wParam
, LPARAM lParam
)
234 oldwndproc
= (WNDPROC
)GetWindowLongPtr(hwnd
, GWL_USERDATA
);
239 if (dwordEditMode
== EDIT_MODE_DEC
)
241 if (isdigit(wParam
& 0xff))
250 else if (dwordEditMode
== EDIT_MODE_HEX
)
252 if (isxdigit(wParam
& 0xff))
267 return CallWindowProc(oldwndproc
, hwnd
, uMsg
, wParam
, lParam
);
271 INT_PTR CALLBACK
modify_dword_dlgproc(HWND hwndDlg
, UINT uMsg
, WPARAM wParam
, LPARAM lParam
)
276 TCHAR ValueString
[32];
283 dwordEditMode
= EDIT_MODE_HEX
;
285 /* subclass the edit control */
286 hwndValue
= GetDlgItem(hwndDlg
, IDC_VALUE_DATA
);
287 oldproc
= (WNDPROC
)GetWindowLongPtr(hwndValue
, GWL_WNDPROC
);
288 SetWindowLongPtr(hwndValue
, GWL_USERDATA
, (DWORD_PTR
)oldproc
);
289 SetWindowLongPtr(hwndValue
, GWL_WNDPROC
, (DWORD_PTR
)DwordEditSubclassProc
);
291 if(editValueName
&& _tcscmp(editValueName
, _T("")))
293 SetDlgItemText(hwndDlg
, IDC_VALUE_NAME
, editValueName
);
298 LoadString(hInst
, IDS_DEFAULT_VALUE_NAME
, buffer
, sizeof(buffer
)/sizeof(TCHAR
));
299 SetDlgItemText(hwndDlg
, IDC_VALUE_NAME
, buffer
);
301 CheckRadioButton (hwndDlg
, IDC_FORMAT_HEX
, IDC_FORMAT_DEC
, IDC_FORMAT_HEX
);
302 _stprintf (ValueString
, _T("%lx"), dwordValueData
);
303 SetDlgItemText(hwndDlg
, IDC_VALUE_DATA
, ValueString
);
304 SetFocus(GetDlgItem(hwndDlg
, IDC_VALUE_DATA
));
308 switch (LOWORD(wParam
))
311 if (HIWORD(wParam
) == BN_CLICKED
&& dwordEditMode
== EDIT_MODE_DEC
)
313 dwordEditMode
= EDIT_MODE_HEX
;
314 if ((hwndValue
= GetDlgItem(hwndDlg
, IDC_VALUE_DATA
)))
316 if ((len
= GetWindowTextLength(hwndValue
)))
318 if (!GetWindowText(hwndValue
, ValueString
, 32))
324 Value
= _tcstoul (ValueString
, &Remainder
, 10);
332 _stprintf (ValueString
, _T("%lx"), Value
);
333 SetDlgItemText(hwndDlg
, IDC_VALUE_DATA
, ValueString
);
339 if (HIWORD(wParam
) == BN_CLICKED
&& dwordEditMode
== EDIT_MODE_HEX
)
341 dwordEditMode
= EDIT_MODE_DEC
;
342 if ((hwndValue
= GetDlgItem(hwndDlg
, IDC_VALUE_DATA
)))
344 if ((len
= GetWindowTextLength(hwndValue
)))
346 if (!GetWindowText(hwndValue
, ValueString
, 32))
352 Value
= _tcstoul (ValueString
, &Remainder
, 16);
360 _stprintf (ValueString
, _T("%lu"), Value
);
361 SetDlgItemText(hwndDlg
, IDC_VALUE_DATA
, ValueString
);
367 if ((hwndValue
= GetDlgItem(hwndDlg
, IDC_VALUE_DATA
)))
369 if ((len
= GetWindowTextLength(hwndValue
)))
371 if (!GetWindowText(hwndValue
, ValueString
, 32))
373 EndDialog(hwndDlg
, IDCANCEL
);
377 Base
= (dwordEditMode
== EDIT_MODE_HEX
) ? 16 : 10;
378 dwordValueData
= _tcstoul (ValueString
, &Remainder
, Base
);
382 EndDialog(hwndDlg
, IDCANCEL
);
386 EndDialog(hwndDlg
, IDOK
);
390 EndDialog(hwndDlg
, IDCANCEL
);
398 INT_PTR CALLBACK
modify_binary_dlgproc(HWND hwndDlg
, UINT uMsg
, WPARAM wParam
, LPARAM lParam
)
405 if(editValueName
&& _tcscmp(editValueName
, _T("")))
407 SetDlgItemText(hwndDlg
, IDC_VALUE_NAME
, editValueName
);
412 LoadString(hInst
, IDS_DEFAULT_VALUE_NAME
, buffer
, sizeof(buffer
)/sizeof(TCHAR
));
413 SetDlgItemText(hwndDlg
, IDC_VALUE_NAME
, buffer
);
415 hwndValue
= GetDlgItem(hwndDlg
, IDC_VALUE_DATA
);
416 HexEdit_LoadBuffer(hwndValue
, binValueData
, valueDataLen
);
417 /* reset the hex edit control's font */
418 SendMessage(hwndValue
, WM_SETFONT
, 0, 0);
422 switch (LOWORD(wParam
))
425 if ((hwndValue
= GetDlgItem(hwndDlg
, IDC_VALUE_DATA
)))
427 len
= HexEdit_GetBufferSize(hwndValue
);
428 if (len
!= valueDataLen
&& len
> 0)
430 binValueData
= HeapReAlloc(GetProcessHeap(), 0, binValueData
, len
);
434 HexEdit_CopyBuffer(hwndValue
, binValueData
, len
);
438 EndDialog(hwndDlg
, IDOK
);
441 EndDialog(hwndDlg
, IDCANCEL
);
449 BOOL
ModifyValue(HWND hwnd
, HKEY hKey
, LPCTSTR valueName
, BOOL EditBin
)
458 editValueName
= valueName
;
460 lRet
= RegQueryValueEx(hKey
, valueName
, 0, &type
, 0, &valueDataLen
);
461 if (lRet
!= ERROR_SUCCESS
&& (!_tcscmp(valueName
, _T("")) || valueName
== NULL
))
463 lRet
= ERROR_SUCCESS
; /* Allow editing of (Default) values which don't exist */
466 stringValueData
= NULL
;
470 if (lRet
!= ERROR_SUCCESS
)
472 error(hwnd
, IDS_BAD_VALUE
, valueName
);
476 if (EditBin
== FALSE
&& ((type
== REG_SZ
) || (type
== REG_EXPAND_SZ
)))
478 if (valueDataLen
> 0)
480 if (!(stringValueData
= HeapAlloc(GetProcessHeap(), 0, valueDataLen
)))
482 error(hwnd
, IDS_TOO_BIG_VALUE
, valueDataLen
);
485 lRet
= RegQueryValueEx(hKey
, valueName
, 0, 0, (LPBYTE
)stringValueData
, &valueDataLen
);
486 if (lRet
!= ERROR_SUCCESS
)
488 error(hwnd
, IDS_BAD_VALUE
, valueName
);
494 stringValueData
= NULL
;
497 if (DialogBox(0, MAKEINTRESOURCE(IDD_EDIT_STRING
), hwnd
, modify_string_dlgproc
) == IDOK
)
501 lRet
= RegSetValueEx(hKey
, valueName
, 0, type
, (LPBYTE
)stringValueData
, (_tcslen(stringValueData
) + 1) * sizeof(TCHAR
));
505 lRet
= RegSetValueEx(hKey
, valueName
, 0, type
, NULL
, 0);
507 if (lRet
== ERROR_SUCCESS
)
511 else if (EditBin
== FALSE
&& type
== REG_MULTI_SZ
)
513 if (valueDataLen
> 0)
515 DWORD llen
, listlen
, nl_len
;
516 LPTSTR src
, lines
= NULL
;
518 if (!(stringValueData
= HeapAlloc(GetProcessHeap(), 0, valueDataLen
)))
520 error(hwnd
, IDS_TOO_BIG_VALUE
, valueDataLen
);
523 lRet
= RegQueryValueEx(hKey
, valueName
, 0, 0, (LPBYTE
)stringValueData
, &valueDataLen
);
524 if (lRet
!= ERROR_SUCCESS
)
526 error(hwnd
, IDS_BAD_VALUE
, valueName
);
530 /* convert \0 to \r\n */
531 src
= stringValueData
;
532 nl_len
= _tcslen(_T("\r\n")) * sizeof(TCHAR
);
533 listlen
= sizeof(TCHAR
);
534 lines
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, listlen
+ sizeof(TCHAR
));
535 while(*src
!= _T('\0'))
540 listlen
+= (llen
* sizeof(TCHAR
)) + nl_len
;
541 lines
= HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, lines
, listlen
);
543 _tcscat(lines
, _T("\r\n"));
546 HeapFree(GetProcessHeap(), 0, stringValueData
);
547 stringValueData
= lines
;
551 stringValueData
= NULL
;
554 if (DialogBox(0, MAKEINTRESOURCE(IDD_EDIT_MULTI_STRING
), hwnd
, modify_multi_string_dlgproc
) == IDOK
)
558 /* convert \r\n to \0 */
559 BOOL EmptyLines
= FALSE
;
560 LPTSTR src
, lines
, nl
;
561 DWORD linechars
, buflen
, c_nl
, dest
;
563 src
= stringValueData
;
564 buflen
= sizeof(TCHAR
);
565 lines
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, buflen
+ sizeof(TCHAR
));
566 c_nl
= _tcslen(_T("\r\n"));
568 while(*src
!= _T('\0'))
570 if((nl
= _tcsstr(src
, _T("\r\n"))))
572 linechars
= nl
- src
;
582 linechars
= _tcslen(src
);
586 buflen
+= ((linechars
+ 1) * sizeof(TCHAR
));
587 lines
= HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, lines
, buflen
);
588 memcpy((lines
+ dest
), src
, linechars
* sizeof(TCHAR
));
590 lines
[dest
++] = _T('\0');
596 src
+= linechars
+ (nl
!= NULL
? c_nl
: 0);
598 lines
[++dest
] = _T('\0');
602 warning(hwnd
, IDS_MULTI_SZ_EMPTY_STRING
);
605 lRet
= RegSetValueEx(hKey
, valueName
, 0, type
, (LPBYTE
)lines
, buflen
);
606 HeapFree(GetProcessHeap(), 0, lines
);
610 lRet
= RegSetValueEx(hKey
, valueName
, 0, type
, NULL
, 0);
612 if (lRet
== ERROR_SUCCESS
)
616 else if (EditBin
== FALSE
&& type
== REG_DWORD
)
618 lRet
= RegQueryValueEx(hKey
, valueName
, 0, 0, (LPBYTE
)&dwordValueData
, &valueDataLen
);
619 if (lRet
!= ERROR_SUCCESS
)
621 error(hwnd
, IDS_BAD_VALUE
, valueName
);
625 if (DialogBox(0, MAKEINTRESOURCE(IDD_EDIT_DWORD
), hwnd
, modify_dword_dlgproc
) == IDOK
)
627 lRet
= RegSetValueEx(hKey
, valueName
, 0, type
, (LPBYTE
)&dwordValueData
, sizeof(DWORD
));
628 if (lRet
== ERROR_SUCCESS
)
632 else if (EditBin
== TRUE
|| type
== REG_NONE
|| type
== REG_BINARY
)
636 int len_vname
= lstrlen(valueName
);
640 if(!(u_valuename
= HeapAlloc(GetProcessHeap(), 0, (len_vname
+ 1) * sizeof(WCHAR
))))
642 error(hwnd
, IDS_TOO_BIG_VALUE
, len_vname
);
645 /* convert the ansi value name to an unicode string */
646 MultiByteToWideChar(CP_ACP
, 0, valueName
, -1, u_valuename
, len_vname
+ 1);
647 valueDataLen
*= sizeof(WCHAR
);
654 if(!(binValueData
= HeapAlloc(GetProcessHeap(), 0, valueDataLen
)))
656 error(hwnd
, IDS_TOO_BIG_VALUE
, valueDataLen
);
660 /* force to use the unicode version, so editing strings in binary mode is correct */
661 lRet
= RegQueryValueExW(hKey
,
667 0, 0, (LPBYTE
)binValueData
, &valueDataLen
);
668 if (lRet
!= ERROR_SUCCESS
)
670 HeapFree(GetProcessHeap(), 0, binValueData
);
673 HeapFree(GetProcessHeap(), 0, u_valuename
);
675 error(hwnd
, IDS_BAD_VALUE
, valueName
);
684 if (DialogBox(0, MAKEINTRESOURCE(IDD_EDIT_BIN_DATA
), hwnd
, modify_binary_dlgproc
) == IDOK
)
686 /* force to use the unicode version, so editing strings in binary mode is correct */
687 lRet
= RegSetValueExW(hKey
,
693 0, type
, (LPBYTE
)binValueData
, valueDataLen
);
694 if (lRet
== ERROR_SUCCESS
)
697 if(binValueData
!= NULL
)
698 HeapFree(GetProcessHeap(), 0, binValueData
);
701 HeapFree(GetProcessHeap(), 0, u_valuename
);
706 error(hwnd
, IDS_UNSUPPORTED_TYPE
, type
);
711 HeapFree(GetProcessHeap(), 0, stringValueData
);
712 stringValueData
= NULL
;