programs/regedit/edit.c - wine - Git at Google

 /*
  * Registry editing UI functions.
  *
  * Copyright (C) 2003 Dimitrie O. Paun
  *
  * 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  */

 #define WIN32_LEAN_AND_MEAN     /* Exclude rarely-used stuff from Windows headers */

 #include <windows.h>
 #include <tchar.h>
 #include <commctrl.h>
 #include <commdlg.h>
 #include <cderr.h>
 #include <stdlib.h>
 #include <stdio.h>
 #include <shellapi.h>
 #include <shlwapi.h>

 #include "main.h"
 #include "regproc.h"
 #include "resource.h"

 static const TCHAR* editValueName;
 static TCHAR* stringValueData;
 static BOOL isDecimal;

 struct edit_params
 {
     HKEY    hKey;
     LPCTSTR lpszValueName;
     void   *pData;
     LONG    cbData;
 };

 static INT vmessagebox(HWND hwnd, INT buttons, INT titleId, INT resId, va_list ap)
 {
     TCHAR title[256];
     TCHAR errfmt[1024];
     TCHAR errstr[1024];

     if (!LoadString(hInst, titleId, title, COUNT_OF(title)))
         lstrcpy(title, "Error");

     if (!LoadString(hInst, resId, errfmt, COUNT_OF(errfmt)))
         lstrcpy(errfmt, "Unknown error string!");

     _vsntprintf(errstr, COUNT_OF(errstr), errfmt, ap);

     return MessageBox(hwnd, errstr, title, buttons);
 }

 static INT messagebox(HWND hwnd, INT buttons, INT titleId, INT resId, ...)
 {
     va_list ap;
     INT result;

     va_start(ap, resId);
     result = vmessagebox(hwnd, buttons, titleId, resId, ap);
     va_end(ap);

     return result;
 }

 void error(HWND hwnd, INT resId, ...)
 {
     va_list ap;

     va_start(ap, resId);
     vmessagebox(hwnd, MB_OK | MB_ICONERROR, IDS_ERROR, resId, ap);
     va_end(ap);
 }

 static void error_code_messagebox(HWND hwnd, DWORD error_code)
 {
     LPTSTR lpMsgBuf;
     DWORD status;
     TCHAR title[256];
     static const TCHAR fallback[] = TEXT("Error displaying error message.\n");
     if (!LoadString(hInst, IDS_ERROR, title, COUNT_OF(title)))
         lstrcpy(title, TEXT("Error"));
     status = FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
                            NULL, error_code, 0, (LPTSTR)&lpMsgBuf, 0, NULL);
     if (!status)
         lpMsgBuf = (LPTSTR)fallback;
     MessageBox(hwnd, lpMsgBuf, title, MB_OK | MB_ICONERROR);
     if (lpMsgBuf != fallback)
         LocalFree(lpMsgBuf);
 }

 static BOOL change_dword_base(HWND hwndDlg, BOOL toHex)
 {
     TCHAR buf[128];
     DWORD val;

     if (!GetDlgItemText(hwndDlg, IDC_VALUE_DATA, buf, COUNT_OF(buf))) return FALSE;
     if (!_stscanf(buf, toHex ? "%ld" : "%lx", &val)) return FALSE;
     wsprintf(buf, toHex ? "%lx" : "%ld", val);
     return SetDlgItemText(hwndDlg, IDC_VALUE_DATA, buf);
 }

 static INT_PTR CALLBACK modify_dlgproc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
 {
     TCHAR* valueData;
     HWND hwndValue;
     int len;

     switch(uMsg) {
     case WM_INITDIALOG:
         SetDlgItemText(hwndDlg, IDC_VALUE_NAME, editValueName);
         SetDlgItemText(hwndDlg, IDC_VALUE_DATA, stringValueData);
 	CheckRadioButton(hwndDlg, IDC_DWORD_HEX, IDC_DWORD_DEC, isDecimal ? IDC_DWORD_DEC : IDC_DWORD_HEX);
         return TRUE;
     case WM_COMMAND:
         switch (LOWORD(wParam)) {
         case IDC_DWORD_HEX:
 	    if (isDecimal && change_dword_base(hwndDlg, TRUE)) isDecimal = FALSE;
 	break;
         case IDC_DWORD_DEC:
 	    if (!isDecimal && change_dword_base(hwndDlg, FALSE)) isDecimal = TRUE;
 	break;
         case IDOK:
             if ((hwndValue = GetDlgItem(hwndDlg, IDC_VALUE_DATA))) {
                 len = GetWindowTextLength(hwndValue);
                 if ((valueData = HeapReAlloc(GetProcessHeap(), 0, stringValueData, (len + 1) * sizeof(TCHAR)))) {
                     stringValueData = valueData;
                     if (!GetWindowText(hwndValue, stringValueData, len + 1))
                         *stringValueData = 0;
                 }
             }
             /* Fall through */
         case IDCANCEL:
             EndDialog(hwndDlg, wParam);
             return TRUE;
         }
     }
     return FALSE;
 }

 static INT_PTR CALLBACK bin_modify_dlgproc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
 {
     struct edit_params *params;
     LPBYTE pData;
     LONG cbData;
     LONG lRet;

     switch(uMsg) {
     case WM_INITDIALOG:
         params = (struct edit_params *)lParam;
         SetWindowLongPtr(hwndDlg, DWLP_USER, (ULONG_PTR)params);
         if (params->lpszValueName)
             SetDlgItemText(hwndDlg, IDC_VALUE_NAME, params->lpszValueName);
         else
             SetDlgItemText(hwndDlg, IDC_VALUE_NAME, g_pszDefaultValueName);
         SendDlgItemMessage(hwndDlg, IDC_VALUE_DATA, HEM_SETDATA, (WPARAM)params->cbData, (LPARAM)params->pData);
         return TRUE;
     case WM_COMMAND:
         switch (LOWORD(wParam)) {
         case IDOK:
             params = (struct edit_params *)GetWindowLongPtr(hwndDlg, DWLP_USER);
             cbData = SendDlgItemMessage(hwndDlg, IDC_VALUE_DATA, HEM_GETDATA, 0, 0);
             pData = HeapAlloc(GetProcessHeap(), 0, cbData);

             if (pData)
             {
                 SendDlgItemMessage(hwndDlg, IDC_VALUE_DATA, HEM_GETDATA, (WPARAM)cbData, (LPARAM)pData);
                 lRet = RegSetValueEx(params->hKey, params->lpszValueName, 0, REG_BINARY, pData, cbData);
             }
             else
                 lRet = ERROR_OUTOFMEMORY;

             if (lRet == ERROR_SUCCESS)
                 EndDialog(hwndDlg, 1);
             else
             {
                 error_code_messagebox(hwndDlg, lRet);
                 EndDialog(hwndDlg, 0);
             }
             return TRUE;
         case IDCANCEL:
             EndDialog(hwndDlg, 0);
             return TRUE;
         }
     }
     return FALSE;
 }

 static BOOL check_value(HWND hwnd, HKEY hKey, LPCTSTR valueName)
 {
     LONG lRet = RegQueryValueEx(hKey, valueName ? valueName : _T(""), 0, NULL, 0, NULL);
     if(lRet != ERROR_SUCCESS) return FALSE;
     return TRUE;
 }

 static LPTSTR read_value(HWND hwnd, HKEY hKey, LPCTSTR valueName, DWORD *lpType, LONG *len)
 {
     DWORD valueDataLen;
     LPTSTR buffer = NULL;
     LONG lRet;

     lRet = RegQueryValueEx(hKey, valueName ? valueName : _T(""), 0, lpType, 0, &valueDataLen);
     if (lRet != ERROR_SUCCESS) {
         if (lRet == ERROR_FILE_NOT_FOUND && !valueName) { /* no default value here, make it up */
             if (len) *len = 1;
             if (lpType) *lpType = REG_SZ;
             buffer = HeapAlloc(GetProcessHeap(), 0, 1);
             *buffer = '\0';
             return buffer;
         }
         error(hwnd, IDS_BAD_VALUE, valueName);
         goto done;
     }
     if ( *lpType == REG_DWORD ) valueDataLen = sizeof(DWORD);
     if (!(buffer = HeapAlloc(GetProcessHeap(), 0, valueDataLen))) {
         error(hwnd, IDS_TOO_BIG_VALUE, valueDataLen);
         goto done;
     }
     lRet = RegQueryValueEx(hKey, valueName, 0, 0, (LPBYTE)buffer, &valueDataLen);
     if (lRet != ERROR_SUCCESS) {
         error(hwnd, IDS_BAD_VALUE, valueName);
         goto done;
     }

     if(len) *len = valueDataLen;
     return buffer;

 done:
     HeapFree(GetProcessHeap(), 0, buffer);
     return NULL;
 }

 BOOL CreateKey(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, LPTSTR keyName)
 {
     BOOL result = FALSE;
     LONG lRet = ERROR_SUCCESS;
     HKEY retKey;
     TCHAR newKey[MAX_NEW_KEY_LEN - 4];
     int keyNum;
     HKEY hKey;

     lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_CREATE_SUB_KEY, &hKey);
     if (lRet != ERROR_SUCCESS) {
 	error_code_messagebox(hwnd, lRet);
 	goto done;
     }

     if (!LoadString(GetModuleHandle(0), IDS_NEWKEY, newKey, COUNT_OF(newKey))) goto done;

     /* try to find out a name for the newly create key (max 100 times) */
     for (keyNum = 1; keyNum < 100; keyNum++) {
 	wsprintf(keyName, newKey, keyNum);
 	lRet = RegOpenKey(hKey, keyName, &retKey);
 	if (lRet != ERROR_SUCCESS) break;
 	RegCloseKey(retKey);
     }
     if (lRet == ERROR_SUCCESS) goto done;

     lRet = RegCreateKey(hKey, keyName, &retKey);
     if (lRet != ERROR_SUCCESS) {
 	error_code_messagebox(hwnd, lRet);
 	goto done;
     }

     result = TRUE;

 done:
     RegCloseKey(retKey);
     return result;
 }

 BOOL ModifyValue(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, LPCTSTR valueName)
 {
     BOOL result = FALSE;
     DWORD type;
     LONG lRet;
     HKEY hKey;
     LONG len;

     lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ | KEY_SET_VALUE, &hKey);
     if (lRet != ERROR_SUCCESS) {
 	error_code_messagebox(hwnd, lRet);
 	return FALSE;
     }

     editValueName = valueName ? valueName : g_pszDefaultValueName;
     if(!(stringValueData = read_value(hwnd, hKey, valueName, &type, &len))) goto done;

     if ( (type == REG_SZ) || (type == REG_EXPAND_SZ) ) {
         if (DialogBox(0, MAKEINTRESOURCE(IDD_EDIT_STRING), hwnd, modify_dlgproc) == IDOK) {
             lRet = RegSetValueEx(hKey, valueName, 0, type, (LPBYTE)stringValueData, lstrlen(stringValueData) + 1);
             if (lRet == ERROR_SUCCESS) result = TRUE;
             else error_code_messagebox(hwnd, lRet);
         }
     } else if ( type == REG_DWORD ) {
 	wsprintf(stringValueData, isDecimal ? "%ld" : "%lx", *((DWORD*)stringValueData));
 	if (DialogBox(0, MAKEINTRESOURCE(IDD_EDIT_DWORD), hwnd, modify_dlgproc) == IDOK) {
 	    DWORD val;
 	    if (_stscanf(stringValueData, isDecimal ? "%ld" : "%lx", &val)) {
 		lRet = RegSetValueEx(hKey, valueName, 0, type, (BYTE*)&val, sizeof(val));
 		if (lRet == ERROR_SUCCESS) result = TRUE;
 		else error_code_messagebox(hwnd, lRet);
 	    }
 	}
     } else if ( type == REG_BINARY ) {
         struct edit_params params;
         params.hKey = hKey;
         params.lpszValueName = valueName;
         params.pData = stringValueData;
         params.cbData = len;
         result = DialogBoxParam(NULL, MAKEINTRESOURCE(IDD_EDIT_BINARY), hwnd,
             bin_modify_dlgproc, (LPARAM)&params);
     } else {
         error(hwnd, IDS_UNSUPPORTED_TYPE, type);
     }

 done:
     HeapFree(GetProcessHeap(), 0, stringValueData);
     stringValueData = NULL;
     RegCloseKey(hKey);
     return result;
 }

 BOOL DeleteKey(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath)
 {
     BOOL result = FALSE;
     LONG lRet;
     HKEY hKey;

     lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ|KEY_SET_VALUE, &hKey);
     if (lRet != ERROR_SUCCESS) {
 	error_code_messagebox(hwnd, lRet);
 	return FALSE;
     }

     if (messagebox(hwnd, MB_YESNO | MB_ICONEXCLAMATION, IDS_DELETE_BOX_TITLE, IDS_DELETE_BOX_TEXT, keyPath) != IDYES)
 	goto done;

     lRet = SHDeleteKey(hKeyRoot, keyPath);
     if (lRet != ERROR_SUCCESS) {
 	error(hwnd, IDS_BAD_KEY, keyPath);
 	goto done;
     }
     result = TRUE;

 done:
     RegCloseKey(hKey);
     return result;
 }

 BOOL DeleteValue(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, LPCTSTR valueName)
 {
     BOOL result = FALSE;
     LONG lRet;
     HKEY hKey;
     LPCSTR visibleValueName = valueName ? valueName : g_pszDefaultValueName;

     lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ | KEY_SET_VALUE, &hKey);
     if (lRet != ERROR_SUCCESS) return FALSE;

     if (messagebox(hwnd, MB_YESNO | MB_ICONEXCLAMATION, IDS_DELETE_BOX_TITLE, IDS_DELETE_BOX_TEXT, visibleValueName) != IDYES)
 	goto done;

     lRet = RegDeleteValue(hKey, valueName ? valueName : "");
     if (lRet != ERROR_SUCCESS && valueName) {
         error(hwnd, IDS_BAD_VALUE, valueName);
     }
     if (lRet != ERROR_SUCCESS) goto done;
     result = TRUE;

 done:
     RegCloseKey(hKey);
     return result;
 }

 BOOL CreateValue(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, DWORD valueType, LPTSTR valueName)
 {
     LONG lRet = ERROR_SUCCESS;
     TCHAR newValue[256];
     DWORD valueDword = 0;
     BOOL result = FALSE;
     int valueNum;
     HKEY hKey;

     lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ | KEY_SET_VALUE, &hKey);
     if (lRet != ERROR_SUCCESS) {
 	error_code_messagebox(hwnd, lRet);
 	return FALSE;
     }

     if (!LoadString(GetModuleHandle(0), IDS_NEWVALUE, newValue, COUNT_OF(newValue))) goto done;

     /* try to find out a name for the newly create key (max 100 times) */
     for (valueNum = 1; valueNum < 100; valueNum++) {
 	wsprintf(valueName, newValue, valueNum);
 	lRet = RegQueryValueEx(hKey, valueName, 0, 0, 0, 0);
 	if (lRet == ERROR_FILE_NOT_FOUND) break;
     }
     if (lRet != ERROR_FILE_NOT_FOUND) {
 	error_code_messagebox(hwnd, lRet);
 	goto done;
     }

     lRet = RegSetValueEx(hKey, valueName, 0, valueType, (BYTE*)&valueDword, sizeof(DWORD));
     if (lRet != ERROR_SUCCESS) {
 	error_code_messagebox(hwnd, lRet);
 	goto done;
     }
     result = TRUE;

 done:
     RegCloseKey(hKey);
     return result;
 }

 BOOL RenameValue(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, LPCTSTR oldName, LPCTSTR newName)
 {
     LPTSTR value = NULL;
     DWORD type;
     LONG len, lRet;
     BOOL result = FALSE;
     HKEY hKey;

     if (!oldName) return FALSE;
     if (!newName) return FALSE;

     lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ | KEY_SET_VALUE, &hKey);
     if (lRet != ERROR_SUCCESS) {
 	error_code_messagebox(hwnd, lRet);
 	return FALSE;
     }
     /* check if value already exists */
     if (check_value(hwnd, hKey, newName)) goto done;
     value = read_value(hwnd, hKey, oldName, &type, &len);
     if(!value) goto done;
     lRet = RegSetValueEx(hKey, newName, 0, type, (BYTE*)value, len);
     if (lRet != ERROR_SUCCESS) {
 	error_code_messagebox(hwnd, lRet);
 	goto done;
     }
     lRet = RegDeleteValue(hKey, oldName);
     if (lRet != ERROR_SUCCESS) {
 	RegDeleteValue(hKey, newName);
 	error_code_messagebox(hwnd, lRet);
 	goto done;
     }
     result = TRUE;

 done:
     HeapFree(GetProcessHeap(), 0, value);
     RegCloseKey(hKey);
     return result;
 }


 BOOL RenameKey(HWND hwnd, HKEY hRootKey, LPCTSTR keyPath, LPCTSTR newName)
 {
     LPTSTR parentPath = 0;
     LPCTSTR srcSubKey = 0;
     HKEY parentKey = 0;
     HKEY destKey = 0;
     BOOL result = FALSE;
     LONG lRet;
     DWORD disposition;

     if (!keyPath || !newName) return FALSE;

     if (!strrchr(keyPath, '\\')) {
 	parentKey = hRootKey;
 	srcSubKey = keyPath;
     } else {
 	parentPath = strdup(keyPath);
 	srcSubKey = strrchr(parentPath, '\\') + 1;
 	*((LPTSTR)srcSubKey - 1) = 0;
 	lRet = RegOpenKeyEx(hRootKey, parentPath, 0, KEY_READ | KEY_CREATE_SUB_KEY, &parentKey);
 	if (lRet != ERROR_SUCCESS) {
 	    error_code_messagebox(hwnd, lRet);
 	    goto done;
 	}
     }

     /* The following fails if the old name is the same as the new name. */
     if (!strcmp(srcSubKey, newName)) goto done;

     lRet = RegCreateKeyEx(parentKey, newName, 0, NULL, REG_OPTION_NON_VOLATILE,
         KEY_WRITE, NULL /* FIXME */, &destKey, &disposition);
     if (disposition == REG_OPENED_EXISTING_KEY)
         lRet = ERROR_FILE_EXISTS; /* FIXME: we might want a better error message than this */
     if (lRet != ERROR_SUCCESS) {
         error_code_messagebox(hwnd, lRet);
         goto done;
     }

     /* FIXME: SHCopyKey does not copy the security attributes */
     lRet = SHCopyKey(parentKey, srcSubKey, destKey, 0);
     if (lRet != ERROR_SUCCESS) {
         RegCloseKey(destKey);
         RegDeleteKey(parentKey, newName);
         error_code_messagebox(hwnd, lRet);
         goto done;
     }

     lRet = SHDeleteKey(hRootKey, keyPath);
     if (lRet != ERROR_SUCCESS) {
         error_code_messagebox(hwnd, lRet);
         goto done;
     }

     result = TRUE;

 done:
     RegCloseKey(destKey);
     if (parentKey) {
         RegCloseKey(parentKey);
         free(parentPath);
     }
     return result;
 }
	/*
	* Registry editing UI functions.
	*
	* Copyright (C) 2003 Dimitrie O. Paun
	*
	* 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	*/

	#define WIN32_LEAN_AND_MEAN /* Exclude rarely-used stuff from Windows headers */

	#include <windows.h>
	#include <tchar.h>
	#include <commctrl.h>
	#include <commdlg.h>
	#include <cderr.h>
	#include <stdlib.h>
	#include <stdio.h>
	#include <shellapi.h>
	#include <shlwapi.h>

	#include "main.h"
	#include "regproc.h"
	#include "resource.h"

	static const TCHAR* editValueName;
	static TCHAR* stringValueData;
	static BOOL isDecimal;

	struct edit_params
	{
	HKEY hKey;
	LPCTSTR lpszValueName;
	void *pData;
	LONG cbData;
	};

	static INT vmessagebox(HWND hwnd, INT buttons, INT titleId, INT resId, va_list ap)
	{
	TCHAR title[256];
	TCHAR errfmt[1024];
	TCHAR errstr[1024];

	if (!LoadString(hInst, titleId, title, COUNT_OF(title)))
	lstrcpy(title, "Error");

	if (!LoadString(hInst, resId, errfmt, COUNT_OF(errfmt)))
	lstrcpy(errfmt, "Unknown error string!");

	_vsntprintf(errstr, COUNT_OF(errstr), errfmt, ap);

	return MessageBox(hwnd, errstr, title, buttons);
	}

	static INT messagebox(HWND hwnd, INT buttons, INT titleId, INT resId, ...)
	{
	va_list ap;
	INT result;

	va_start(ap, resId);
	result = vmessagebox(hwnd, buttons, titleId, resId, ap);
	va_end(ap);

	return result;
	}

	void error(HWND hwnd, INT resId, ...)
	{
	va_list ap;

	va_start(ap, resId);
	vmessagebox(hwnd, MB_OK \| MB_ICONERROR, IDS_ERROR, resId, ap);
	va_end(ap);
	}

	static void error_code_messagebox(HWND hwnd, DWORD error_code)
	{
	LPTSTR lpMsgBuf;
	DWORD status;
	TCHAR title[256];
	static const TCHAR fallback[] = TEXT("Error displaying error message.\n");
	if (!LoadString(hInst, IDS_ERROR, title, COUNT_OF(title)))
	lstrcpy(title, TEXT("Error"));
	status = FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER \| FORMAT_MESSAGE_FROM_SYSTEM,
	NULL, error_code, 0, (LPTSTR)&lpMsgBuf, 0, NULL);
	if (!status)
	lpMsgBuf = (LPTSTR)fallback;
	MessageBox(hwnd, lpMsgBuf, title, MB_OK \| MB_ICONERROR);
	if (lpMsgBuf != fallback)
	LocalFree(lpMsgBuf);
	}

	static BOOL change_dword_base(HWND hwndDlg, BOOL toHex)
	{
	TCHAR buf[128];
	DWORD val;

	if (!GetDlgItemText(hwndDlg, IDC_VALUE_DATA, buf, COUNT_OF(buf))) return FALSE;
	if (!_stscanf(buf, toHex ? "%ld" : "%lx", &val)) return FALSE;
	wsprintf(buf, toHex ? "%lx" : "%ld", val);
	return SetDlgItemText(hwndDlg, IDC_VALUE_DATA, buf);
	}

	static INT_PTR CALLBACK modify_dlgproc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
	{
	TCHAR* valueData;
	HWND hwndValue;
	int len;

	switch(uMsg) {
	case WM_INITDIALOG:
	SetDlgItemText(hwndDlg, IDC_VALUE_NAME, editValueName);
	SetDlgItemText(hwndDlg, IDC_VALUE_DATA, stringValueData);
	CheckRadioButton(hwndDlg, IDC_DWORD_HEX, IDC_DWORD_DEC, isDecimal ? IDC_DWORD_DEC : IDC_DWORD_HEX);
	return TRUE;
	case WM_COMMAND:
	switch (LOWORD(wParam)) {
	case IDC_DWORD_HEX:
	if (isDecimal && change_dword_base(hwndDlg, TRUE)) isDecimal = FALSE;
	break;
	case IDC_DWORD_DEC:
	if (!isDecimal && change_dword_base(hwndDlg, FALSE)) isDecimal = TRUE;
	break;
	case IDOK:
	if ((hwndValue = GetDlgItem(hwndDlg, IDC_VALUE_DATA))) {
	len = GetWindowTextLength(hwndValue);
	if ((valueData = HeapReAlloc(GetProcessHeap(), 0, stringValueData, (len + 1) * sizeof(TCHAR)))) {
	stringValueData = valueData;
	if (!GetWindowText(hwndValue, stringValueData, len + 1))
	*stringValueData = 0;
	}
	}
	/* Fall through */
	case IDCANCEL:
	EndDialog(hwndDlg, wParam);
	return TRUE;
	}
	}
	return FALSE;
	}

	static INT_PTR CALLBACK bin_modify_dlgproc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
	{
	struct edit_params *params;
	LPBYTE pData;
	LONG cbData;
	LONG lRet;

	switch(uMsg) {
	case WM_INITDIALOG:
	params = (struct edit_params *)lParam;
	SetWindowLongPtr(hwndDlg, DWLP_USER, (ULONG_PTR)params);
	if (params->lpszValueName)
	SetDlgItemText(hwndDlg, IDC_VALUE_NAME, params->lpszValueName);
	else
	SetDlgItemText(hwndDlg, IDC_VALUE_NAME, g_pszDefaultValueName);
	SendDlgItemMessage(hwndDlg, IDC_VALUE_DATA, HEM_SETDATA, (WPARAM)params->cbData, (LPARAM)params->pData);
	return TRUE;
	case WM_COMMAND:
	switch (LOWORD(wParam)) {
	case IDOK:
	params = (struct edit_params *)GetWindowLongPtr(hwndDlg, DWLP_USER);
	cbData = SendDlgItemMessage(hwndDlg, IDC_VALUE_DATA, HEM_GETDATA, 0, 0);
	pData = HeapAlloc(GetProcessHeap(), 0, cbData);

	if (pData)
	{
	SendDlgItemMessage(hwndDlg, IDC_VALUE_DATA, HEM_GETDATA, (WPARAM)cbData, (LPARAM)pData);
	lRet = RegSetValueEx(params->hKey, params->lpszValueName, 0, REG_BINARY, pData, cbData);
	}
	else
	lRet = ERROR_OUTOFMEMORY;

	if (lRet == ERROR_SUCCESS)
	EndDialog(hwndDlg, 1);
	else
	{
	error_code_messagebox(hwndDlg, lRet);
	EndDialog(hwndDlg, 0);
	}
	return TRUE;
	case IDCANCEL:
	EndDialog(hwndDlg, 0);
	return TRUE;
	}
	}
	return FALSE;
	}

	static BOOL check_value(HWND hwnd, HKEY hKey, LPCTSTR valueName)
	{
	LONG lRet = RegQueryValueEx(hKey, valueName ? valueName : _T(""), 0, NULL, 0, NULL);
	if(lRet != ERROR_SUCCESS) return FALSE;
	return TRUE;
	}

	static LPTSTR read_value(HWND hwnd, HKEY hKey, LPCTSTR valueName, DWORD lpType, LONG len)
	{
	DWORD valueDataLen;
	LPTSTR buffer = NULL;
	LONG lRet;

	lRet = RegQueryValueEx(hKey, valueName ? valueName : _T(""), 0, lpType, 0, &valueDataLen);
	if (lRet != ERROR_SUCCESS) {
	if (lRet == ERROR_FILE_NOT_FOUND && !valueName) { /* no default value here, make it up */
	if (len) *len = 1;
	if (lpType) *lpType = REG_SZ;
	buffer = HeapAlloc(GetProcessHeap(), 0, 1);
	*buffer = '\0';
	return buffer;
	}
	error(hwnd, IDS_BAD_VALUE, valueName);
	goto done;
	}
	if ( *lpType == REG_DWORD ) valueDataLen = sizeof(DWORD);
	if (!(buffer = HeapAlloc(GetProcessHeap(), 0, valueDataLen))) {
	error(hwnd, IDS_TOO_BIG_VALUE, valueDataLen);
	goto done;
	}
	lRet = RegQueryValueEx(hKey, valueName, 0, 0, (LPBYTE)buffer, &valueDataLen);
	if (lRet != ERROR_SUCCESS) {
	error(hwnd, IDS_BAD_VALUE, valueName);
	goto done;
	}

	if(len) *len = valueDataLen;
	return buffer;

	done:
	HeapFree(GetProcessHeap(), 0, buffer);
	return NULL;
	}

	BOOL CreateKey(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, LPTSTR keyName)
	{
	BOOL result = FALSE;
	LONG lRet = ERROR_SUCCESS;
	HKEY retKey;
	TCHAR newKey[MAX_NEW_KEY_LEN - 4];
	int keyNum;
	HKEY hKey;

	lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_CREATE_SUB_KEY, &hKey);
	if (lRet != ERROR_SUCCESS) {
	error_code_messagebox(hwnd, lRet);
	goto done;
	}

	if (!LoadString(GetModuleHandle(0), IDS_NEWKEY, newKey, COUNT_OF(newKey))) goto done;

	/* try to find out a name for the newly create key (max 100 times) */
	for (keyNum = 1; keyNum < 100; keyNum++) {
	wsprintf(keyName, newKey, keyNum);
	lRet = RegOpenKey(hKey, keyName, &retKey);
	if (lRet != ERROR_SUCCESS) break;
	RegCloseKey(retKey);
	}
	if (lRet == ERROR_SUCCESS) goto done;

	lRet = RegCreateKey(hKey, keyName, &retKey);
	if (lRet != ERROR_SUCCESS) {
	error_code_messagebox(hwnd, lRet);
	goto done;
	}

	result = TRUE;

	done:
	RegCloseKey(retKey);
	return result;
	}

	BOOL ModifyValue(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, LPCTSTR valueName)
	{
	BOOL result = FALSE;
	DWORD type;
	LONG lRet;
	HKEY hKey;
	LONG len;

	lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ \| KEY_SET_VALUE, &hKey);
	if (lRet != ERROR_SUCCESS) {
	error_code_messagebox(hwnd, lRet);
	return FALSE;
	}

	editValueName = valueName ? valueName : g_pszDefaultValueName;
	if(!(stringValueData = read_value(hwnd, hKey, valueName, &type, &len))) goto done;

	if ( (type == REG_SZ) \|\| (type == REG_EXPAND_SZ) ) {
	if (DialogBox(0, MAKEINTRESOURCE(IDD_EDIT_STRING), hwnd, modify_dlgproc) == IDOK) {
	lRet = RegSetValueEx(hKey, valueName, 0, type, (LPBYTE)stringValueData, lstrlen(stringValueData) + 1);
	if (lRet == ERROR_SUCCESS) result = TRUE;
	else error_code_messagebox(hwnd, lRet);
	}
	} else if ( type == REG_DWORD ) {
	wsprintf(stringValueData, isDecimal ? "%ld" : "%lx", ((DWORD)stringValueData));
	if (DialogBox(0, MAKEINTRESOURCE(IDD_EDIT_DWORD), hwnd, modify_dlgproc) == IDOK) {
	DWORD val;
	if (_stscanf(stringValueData, isDecimal ? "%ld" : "%lx", &val)) {
	lRet = RegSetValueEx(hKey, valueName, 0, type, (BYTE*)&val, sizeof(val));
	if (lRet == ERROR_SUCCESS) result = TRUE;
	else error_code_messagebox(hwnd, lRet);
	}
	}
	} else if ( type == REG_BINARY ) {
	struct edit_params params;
	params.hKey = hKey;
	params.lpszValueName = valueName;
	params.pData = stringValueData;
	params.cbData = len;
	result = DialogBoxParam(NULL, MAKEINTRESOURCE(IDD_EDIT_BINARY), hwnd,
	bin_modify_dlgproc, (LPARAM)&params);
	} else {
	error(hwnd, IDS_UNSUPPORTED_TYPE, type);
	}

	done:
	HeapFree(GetProcessHeap(), 0, stringValueData);
	stringValueData = NULL;
	RegCloseKey(hKey);
	return result;
	}

	BOOL DeleteKey(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath)
	{
	BOOL result = FALSE;
	LONG lRet;
	HKEY hKey;

	lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ\|KEY_SET_VALUE, &hKey);
	if (lRet != ERROR_SUCCESS) {
	error_code_messagebox(hwnd, lRet);
	return FALSE;
	}

	if (messagebox(hwnd, MB_YESNO \| MB_ICONEXCLAMATION, IDS_DELETE_BOX_TITLE, IDS_DELETE_BOX_TEXT, keyPath) != IDYES)
	goto done;

	lRet = SHDeleteKey(hKeyRoot, keyPath);
	if (lRet != ERROR_SUCCESS) {
	error(hwnd, IDS_BAD_KEY, keyPath);
	goto done;
	}
	result = TRUE;

	done:
	RegCloseKey(hKey);
	return result;
	}

	BOOL DeleteValue(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, LPCTSTR valueName)
	{
	BOOL result = FALSE;
	LONG lRet;
	HKEY hKey;
	LPCSTR visibleValueName = valueName ? valueName : g_pszDefaultValueName;

	lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ \| KEY_SET_VALUE, &hKey);
	if (lRet != ERROR_SUCCESS) return FALSE;

	if (messagebox(hwnd, MB_YESNO \| MB_ICONEXCLAMATION, IDS_DELETE_BOX_TITLE, IDS_DELETE_BOX_TEXT, visibleValueName) != IDYES)
	goto done;

	lRet = RegDeleteValue(hKey, valueName ? valueName : "");
	if (lRet != ERROR_SUCCESS && valueName) {
	error(hwnd, IDS_BAD_VALUE, valueName);
	}
	if (lRet != ERROR_SUCCESS) goto done;
	result = TRUE;

	done:
	RegCloseKey(hKey);
	return result;
	}

	BOOL CreateValue(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, DWORD valueType, LPTSTR valueName)
	{
	LONG lRet = ERROR_SUCCESS;
	TCHAR newValue[256];
	DWORD valueDword = 0;
	BOOL result = FALSE;
	int valueNum;
	HKEY hKey;

	lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ \| KEY_SET_VALUE, &hKey);
	if (lRet != ERROR_SUCCESS) {
	error_code_messagebox(hwnd, lRet);
	return FALSE;
	}

	if (!LoadString(GetModuleHandle(0), IDS_NEWVALUE, newValue, COUNT_OF(newValue))) goto done;

	/* try to find out a name for the newly create key (max 100 times) */
	for (valueNum = 1; valueNum < 100; valueNum++) {
	wsprintf(valueName, newValue, valueNum);
	lRet = RegQueryValueEx(hKey, valueName, 0, 0, 0, 0);
	if (lRet == ERROR_FILE_NOT_FOUND) break;
	}
	if (lRet != ERROR_FILE_NOT_FOUND) {
	error_code_messagebox(hwnd, lRet);
	goto done;
	}

	lRet = RegSetValueEx(hKey, valueName, 0, valueType, (BYTE*)&valueDword, sizeof(DWORD));
	if (lRet != ERROR_SUCCESS) {
	error_code_messagebox(hwnd, lRet);
	goto done;
	}
	result = TRUE;

	done:
	RegCloseKey(hKey);
	return result;
	}

	BOOL RenameValue(HWND hwnd, HKEY hKeyRoot, LPCTSTR keyPath, LPCTSTR oldName, LPCTSTR newName)
	{
	LPTSTR value = NULL;
	DWORD type;
	LONG len, lRet;
	BOOL result = FALSE;
	HKEY hKey;

	if (!oldName) return FALSE;
	if (!newName) return FALSE;

	lRet = RegOpenKeyEx(hKeyRoot, keyPath, 0, KEY_READ \| KEY_SET_VALUE, &hKey);
	if (lRet != ERROR_SUCCESS) {
	error_code_messagebox(hwnd, lRet);
	return FALSE;
	}
	/* check if value already exists */
	if (check_value(hwnd, hKey, newName)) goto done;
	value = read_value(hwnd, hKey, oldName, &type, &len);
	if(!value) goto done;
	lRet = RegSetValueEx(hKey, newName, 0, type, (BYTE*)value, len);
	if (lRet != ERROR_SUCCESS) {
	error_code_messagebox(hwnd, lRet);
	goto done;
	}
	lRet = RegDeleteValue(hKey, oldName);
	if (lRet != ERROR_SUCCESS) {
	RegDeleteValue(hKey, newName);
	error_code_messagebox(hwnd, lRet);
	goto done;
	}
	result = TRUE;

	done:
	HeapFree(GetProcessHeap(), 0, value);
	RegCloseKey(hKey);
	return result;
	}


	BOOL RenameKey(HWND hwnd, HKEY hRootKey, LPCTSTR keyPath, LPCTSTR newName)
	{
	LPTSTR parentPath = 0;
	LPCTSTR srcSubKey = 0;
	HKEY parentKey = 0;
	HKEY destKey = 0;
	BOOL result = FALSE;
	LONG lRet;
	DWORD disposition;

	if (!keyPath \|\| !newName) return FALSE;

	if (!strrchr(keyPath, '\\')) {
	parentKey = hRootKey;
	srcSubKey = keyPath;
	} else {
	parentPath = strdup(keyPath);
	srcSubKey = strrchr(parentPath, '\\') + 1;
	*((LPTSTR)srcSubKey - 1) = 0;
	lRet = RegOpenKeyEx(hRootKey, parentPath, 0, KEY_READ \| KEY_CREATE_SUB_KEY, &parentKey);
	if (lRet != ERROR_SUCCESS) {
	error_code_messagebox(hwnd, lRet);
	goto done;
	}
	}

	/* The following fails if the old name is the same as the new name. */
	if (!strcmp(srcSubKey, newName)) goto done;

	lRet = RegCreateKeyEx(parentKey, newName, 0, NULL, REG_OPTION_NON_VOLATILE,
	KEY_WRITE, NULL /* FIXME */, &destKey, &disposition);
	if (disposition == REG_OPENED_EXISTING_KEY)
	lRet = ERROR_FILE_EXISTS; /* FIXME: we might want a better error message than this */
	if (lRet != ERROR_SUCCESS) {
	error_code_messagebox(hwnd, lRet);
	goto done;
	}

	/* FIXME: SHCopyKey does not copy the security attributes */
	lRet = SHCopyKey(parentKey, srcSubKey, destKey, 0);
	if (lRet != ERROR_SUCCESS) {
	RegCloseKey(destKey);
	RegDeleteKey(parentKey, newName);
	error_code_messagebox(hwnd, lRet);
	goto done;
	}

	lRet = SHDeleteKey(hRootKey, keyPath);
	if (lRet != ERROR_SUCCESS) {
	error_code_messagebox(hwnd, lRet);
	goto done;
	}

	result = TRUE;

	done:
	RegCloseKey(destKey);
	if (parentKey) {
	RegCloseKey(parentKey);
	free(parentPath);
	}
	return result;
	}