528 lines
13 KiB
C++
528 lines
13 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/*
|
|
* The contents of this file are subject to the Netscape Public
|
|
* License Version 1.1 (the "License"); you may not use this file
|
|
* except in compliance with the License. You may obtain a copy of
|
|
* the License at http://www.mozilla.org/NPL/
|
|
*
|
|
* Software distributed under the License is distributed on an "AS
|
|
* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
|
|
* implied. See the License for the specific language governing
|
|
* rights and limitations under the License.
|
|
*
|
|
* The Original Code is mozilla.org code.
|
|
*
|
|
* The Initial Developer of the Original Code is Netscape
|
|
* Communications Corporation. Portions created by Netscape are
|
|
* Copyright (C) 1998 Netscape Communications Corporation. All
|
|
* Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
*/
|
|
|
|
#include "nsWinReg.h"
|
|
#include "nsWinRegItem.h"
|
|
#include <windows.h> /* is this needed? */
|
|
|
|
/* Public Methods */
|
|
|
|
nsWinReg::nsWinReg(nsInstall* suObj)
|
|
{
|
|
mInstallObject = suObj;
|
|
mRootKey = (PRInt32)HKEY_CLASSES_ROOT;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::SetRootKey(PRInt32 key)
|
|
{
|
|
mRootKey = key;
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::CreateKey(const nsString& subkey, const nsString& classname, PRInt32* aReturn)
|
|
{
|
|
nsWinRegItem* wi = new nsWinRegItem(this, mRootKey, NS_WIN_REG_CREATE, subkey, classname, "null", aReturn);
|
|
|
|
if(wi == nsnull)
|
|
{
|
|
*aReturn = nsInstall::OUT_OF_MEMORY;
|
|
return NS_OK;
|
|
}
|
|
|
|
if(*aReturn != nsInstall::SUCCESS)
|
|
{
|
|
if(wi)
|
|
{
|
|
delete wi;
|
|
return NS_OK;
|
|
}
|
|
}
|
|
|
|
if(mInstallObject)
|
|
mInstallObject->ScheduleForInstall(wi);
|
|
|
|
return 0;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::DeleteKey(const nsString& subkey, PRInt32* aReturn)
|
|
{
|
|
nsWinRegItem* wi = new nsWinRegItem(this, mRootKey, NS_WIN_REG_DELETE, subkey, "null", "null", aReturn);
|
|
|
|
if(wi == nsnull)
|
|
{
|
|
*aReturn = nsInstall::OUT_OF_MEMORY;
|
|
return NS_OK;
|
|
}
|
|
|
|
if(*aReturn != nsInstall::SUCCESS)
|
|
{
|
|
if(wi)
|
|
{
|
|
delete wi;
|
|
return NS_OK;
|
|
}
|
|
}
|
|
|
|
if(mInstallObject)
|
|
mInstallObject->ScheduleForInstall(wi);
|
|
|
|
return 0;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::DeleteValue(const nsString& subkey, const nsString& valname, PRInt32* aReturn)
|
|
{
|
|
nsWinRegItem* wi = new nsWinRegItem(this, mRootKey, NS_WIN_REG_DELETE_VAL, subkey, valname, "null", aReturn);
|
|
|
|
if(wi == nsnull)
|
|
{
|
|
*aReturn = nsInstall::OUT_OF_MEMORY;
|
|
return NS_OK;
|
|
}
|
|
|
|
if(*aReturn != nsInstall::SUCCESS)
|
|
{
|
|
if(wi)
|
|
{
|
|
delete wi;
|
|
return NS_OK;
|
|
}
|
|
}
|
|
|
|
if(mInstallObject)
|
|
mInstallObject->ScheduleForInstall(wi);
|
|
|
|
return 0;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::SetValueString(const nsString& subkey, const nsString& valname, const nsString& value, PRInt32* aReturn)
|
|
{
|
|
nsWinRegItem* wi = new nsWinRegItem(this, mRootKey, NS_WIN_REG_SET_VAL_STRING, subkey, valname, value, aReturn);
|
|
|
|
if(wi == nsnull)
|
|
{
|
|
*aReturn = nsInstall::OUT_OF_MEMORY;
|
|
return NS_OK;
|
|
}
|
|
|
|
if(*aReturn != nsInstall::SUCCESS)
|
|
{
|
|
if(wi)
|
|
{
|
|
delete wi;
|
|
return NS_OK;
|
|
}
|
|
}
|
|
|
|
if(mInstallObject)
|
|
mInstallObject->ScheduleForInstall(wi);
|
|
|
|
return 0;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::GetValueString(const nsString& subkey, const nsString& valname, nsString* aReturn)
|
|
{
|
|
NativeGetValueString(subkey, valname, aReturn);
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::SetValueNumber(const nsString& subkey, const nsString& valname, PRInt32 value, PRInt32* aReturn)
|
|
{
|
|
nsWinRegItem* wi = new nsWinRegItem(this, mRootKey, NS_WIN_REG_SET_VAL_NUMBER, subkey, valname, value, aReturn);
|
|
|
|
if(wi == nsnull)
|
|
{
|
|
*aReturn = nsInstall::OUT_OF_MEMORY;
|
|
return NS_OK;
|
|
}
|
|
|
|
if(*aReturn != nsInstall::SUCCESS)
|
|
{
|
|
if(wi)
|
|
{
|
|
delete wi;
|
|
return NS_OK;
|
|
}
|
|
}
|
|
|
|
if(mInstallObject)
|
|
mInstallObject->ScheduleForInstall(wi);
|
|
|
|
return 0;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::GetValueNumber(const nsString& subkey, const nsString& valname, PRInt32* aReturn)
|
|
{
|
|
NativeGetValueNumber(subkey, valname, aReturn);
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::SetValue(const nsString& subkey, const nsString& valname, nsWinRegValue* value, PRInt32* aReturn)
|
|
{
|
|
// fix: need to figure out what to do with nsWinRegValue class.
|
|
//
|
|
// nsWinRegItem* wi = new nsWinRegItem(this, mRootKey, NS_WIN_REG_SET_VAL, subkey, valname, (nsWinRegValue*)value, aReturn);
|
|
//
|
|
// if(wi == nsnull)
|
|
// {
|
|
// return NS_OK;
|
|
// }
|
|
// mInstallObject->ScheduleForInstall(wi);
|
|
return 0;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::GetValue(const nsString& subkey, const nsString& valname, nsWinRegValue** aReturn)
|
|
{
|
|
// fix:
|
|
return NS_OK;
|
|
}
|
|
|
|
nsInstall* nsWinReg::InstallObject()
|
|
{
|
|
return mInstallObject;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::FinalCreateKey(PRInt32 root, const nsString& subkey, const nsString& classname, PRInt32* aReturn)
|
|
{
|
|
SetRootKey(root);
|
|
*aReturn = NativeCreateKey(subkey, classname);
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::FinalDeleteKey(PRInt32 root, const nsString& subkey, PRInt32* aReturn)
|
|
{
|
|
SetRootKey(root);
|
|
*aReturn = NativeDeleteKey(subkey);
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::FinalDeleteValue(PRInt32 root, const nsString& subkey, const nsString& valname, PRInt32* aReturn)
|
|
{
|
|
SetRootKey(root);
|
|
*aReturn = NativeDeleteValue(subkey, valname);
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::FinalSetValueString(PRInt32 root, const nsString& subkey, const nsString& valname, const nsString& value, PRInt32* aReturn)
|
|
{
|
|
SetRootKey(root);
|
|
*aReturn = NativeSetValueString(subkey, valname, value);
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::FinalSetValueNumber(PRInt32 root, const nsString& subkey, const nsString& valname, PRInt32 value, PRInt32* aReturn)
|
|
{
|
|
SetRootKey(root);
|
|
*aReturn = NativeSetValueNumber(subkey, valname, value);
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::FinalSetValue(PRInt32 root, const nsString& subkey, const nsString& valname, nsWinRegValue* value, PRInt32* aReturn)
|
|
{
|
|
SetRootKey(root);
|
|
*aReturn = NativeSetValue(subkey, valname, value);
|
|
return NS_OK;
|
|
}
|
|
|
|
|
|
/* Private Methods */
|
|
|
|
PRInt32
|
|
nsWinReg::NativeCreateKey(const nsString& subkey, const nsString& classname)
|
|
{
|
|
HKEY root, newkey;
|
|
LONG result;
|
|
ULONG disposition;
|
|
char* subkeyCString = subkey.ToNewCString();
|
|
char* classnameCString = classname.ToNewCString();
|
|
|
|
#ifdef WIN32
|
|
root = (HKEY)mRootKey;
|
|
result = RegCreateKeyEx(root, subkeyCString, 0, classnameCString, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, nsnull, &newkey, &disposition);
|
|
|
|
if(ERROR_SUCCESS == result)
|
|
{
|
|
RegCloseKey( newkey );
|
|
}
|
|
#endif
|
|
|
|
if (subkeyCString) Recycle(subkeyCString);
|
|
if (classnameCString) Recycle(classnameCString);
|
|
|
|
return result;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::NativeDeleteKey(const nsString& subkey)
|
|
{
|
|
HKEY root;
|
|
LONG result;
|
|
char* subkeyCString = subkey.ToNewCString();
|
|
|
|
#ifdef WIN32
|
|
root = (HKEY) mRootKey;
|
|
result = RegDeleteKey( root, subkeyCString );
|
|
#endif
|
|
|
|
if (subkeyCString) Recycle(subkeyCString);
|
|
|
|
return result;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::NativeDeleteValue(const nsString& subkey, const nsString& valname)
|
|
{
|
|
#if defined (WIN32) || defined (XP_OS2)
|
|
HKEY root, newkey;
|
|
LONG result;
|
|
char* subkeyCString = subkey.ToNewCString();
|
|
char* valnameCString = valname.ToNewCString();
|
|
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx( root, subkeyCString, 0, KEY_WRITE, &newkey);
|
|
|
|
if ( ERROR_SUCCESS == result )
|
|
{
|
|
result = RegDeleteValue( newkey, valnameCString );
|
|
RegCloseKey( newkey );
|
|
}
|
|
|
|
if (subkeyCString) Recycle(subkeyCString);
|
|
if (valnameCString) Recycle(valnameCString);
|
|
|
|
return result;
|
|
#else
|
|
return ERROR_INVALID_PARAMETER;
|
|
#endif
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::NativeSetValueString(const nsString& subkey, const nsString& valname, const nsString& value)
|
|
{
|
|
HKEY root;
|
|
HKEY newkey;
|
|
LONG result;
|
|
DWORD length;
|
|
|
|
char* subkeyCString = subkey.ToNewCString();
|
|
char* valnameCString = valname.ToNewCString();
|
|
char* valueCString = value.ToNewCString();
|
|
|
|
length = value.Length();
|
|
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx( root, subkeyCString, 0, KEY_ALL_ACCESS, &newkey);
|
|
|
|
if(ERROR_SUCCESS == result)
|
|
{
|
|
result = RegSetValueEx( newkey, valnameCString, 0, REG_SZ, (unsigned char*)valueCString, length );
|
|
RegCloseKey( newkey );
|
|
}
|
|
|
|
if (subkeyCString) Recycle(subkeyCString);
|
|
if (valnameCString) Recycle(valnameCString);
|
|
if (valueCString) Recycle(valueCString);
|
|
|
|
return result;
|
|
}
|
|
|
|
#define STRBUFLEN 255
|
|
|
|
void
|
|
nsWinReg::NativeGetValueString(const nsString& subkey, const nsString& valname, nsString* aReturn)
|
|
{
|
|
unsigned char valbuf[_MAXKEYVALUE_];
|
|
HKEY root;
|
|
HKEY newkey;
|
|
LONG result;
|
|
DWORD type = REG_SZ;
|
|
DWORD length = _MAXKEYVALUE_;
|
|
char* subkeyCString = subkey.ToNewCString();
|
|
char* valnameCString = valname.ToNewCString();
|
|
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx( root, subkeyCString, 0, KEY_READ, &newkey );
|
|
|
|
if ( ERROR_SUCCESS == result ) {
|
|
result = RegQueryValueEx( newkey, valnameCString, nsnull, &type, valbuf, &length );
|
|
|
|
RegCloseKey( newkey );
|
|
}
|
|
|
|
if(ERROR_SUCCESS == result && type == REG_SZ)
|
|
{
|
|
*aReturn = (char*)valbuf;
|
|
}
|
|
|
|
if (subkeyCString) Recycle(subkeyCString);
|
|
if (valnameCString) Recycle(valnameCString);
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::NativeSetValueNumber(const nsString& subkey, const nsString& valname, PRInt32 value)
|
|
{
|
|
HKEY root;
|
|
HKEY newkey;
|
|
LONG result;
|
|
|
|
char* subkeyCString = subkey.ToNewCString();
|
|
char* valnameCString = valname.ToNewCString();
|
|
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx( root, subkeyCString, 0, KEY_ALL_ACCESS, &newkey);
|
|
|
|
if(ERROR_SUCCESS == result)
|
|
{
|
|
result = RegSetValueEx( newkey, valnameCString, 0, REG_DWORD, (LPBYTE)&value, sizeof(PRInt32));
|
|
RegCloseKey( newkey );
|
|
}
|
|
|
|
if (subkeyCString) Recycle(subkeyCString);
|
|
if (valnameCString) Recycle(valnameCString);
|
|
|
|
return result;
|
|
}
|
|
|
|
void
|
|
nsWinReg::NativeGetValueNumber(const nsString& subkey, const nsString& valname, PRInt32* aReturn)
|
|
{
|
|
PRInt32 valbuf;
|
|
PRInt32 valbuflen;
|
|
HKEY root;
|
|
HKEY newkey;
|
|
LONG result;
|
|
DWORD type = REG_DWORD;
|
|
DWORD length = _MAXKEYVALUE_;
|
|
char* subkeyCString = subkey.ToNewCString();
|
|
char* valnameCString = valname.ToNewCString();
|
|
|
|
valbuflen = sizeof(PRInt32);
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx( root, subkeyCString, 0, KEY_READ, &newkey );
|
|
|
|
if ( ERROR_SUCCESS == result ) {
|
|
result = RegQueryValueEx( newkey, valnameCString, nsnull, &type, (LPBYTE)&valbuf, (LPDWORD)&valbuflen);
|
|
|
|
RegCloseKey( newkey );
|
|
}
|
|
|
|
if(ERROR_SUCCESS == result && type == REG_DWORD)
|
|
{
|
|
*aReturn = valbuf;
|
|
}
|
|
|
|
if (subkeyCString) Recycle(subkeyCString);
|
|
if (valnameCString) Recycle(valnameCString);
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::NativeSetValue(const nsString& subkey, const nsString& valname, nsWinRegValue* value)
|
|
{
|
|
#if defined (WIN32) || defined (XP_OS2)
|
|
HKEY root;
|
|
HKEY newkey;
|
|
LONG result;
|
|
DWORD length;
|
|
DWORD type;
|
|
unsigned char* data;
|
|
char* subkeyCString = subkey.ToNewCString();
|
|
char* valnameCString = valname.ToNewCString();
|
|
|
|
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx( root, subkeyCString, 0, KEY_ALL_ACCESS, &newkey );
|
|
|
|
if(ERROR_SUCCESS == result)
|
|
{
|
|
type = (DWORD)value->type;
|
|
data = (unsigned char*)value->data;
|
|
length = (DWORD)value->data_length;
|
|
|
|
result = RegSetValueEx( newkey, valnameCString, 0, type, data, length);
|
|
RegCloseKey( newkey );
|
|
}
|
|
|
|
if (subkeyCString) Recycle(subkeyCString);
|
|
if (valnameCString) Recycle(valnameCString);
|
|
|
|
return result;
|
|
#else
|
|
return ERROR_INVALID_PARAMETER;
|
|
#endif
|
|
}
|
|
|
|
nsWinRegValue*
|
|
nsWinReg::NativeGetValue(const nsString& subkey, const nsString& valname)
|
|
{
|
|
#if defined (WIN32) || defined (XP_OS2)
|
|
unsigned char valbuf[STRBUFLEN];
|
|
HKEY root;
|
|
HKEY newkey;
|
|
LONG result;
|
|
DWORD length=STRBUFLEN;
|
|
DWORD type;
|
|
nsString* data;
|
|
nsWinRegValue* value = nsnull;
|
|
char* subkeyCString = subkey.ToNewCString();
|
|
char* valnameCString = valname.ToNewCString();
|
|
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx( root, subkeyCString, 0, KEY_ALL_ACCESS, &newkey );
|
|
|
|
if(ERROR_SUCCESS == result)
|
|
{
|
|
result = RegQueryValueEx( newkey, valnameCString, nsnull, &type, valbuf, &length );
|
|
|
|
if ( ERROR_SUCCESS == result ) {
|
|
data = new nsString((char*)valbuf);
|
|
length = data->Length();
|
|
value = new nsWinRegValue(type, (void*)data, length);
|
|
}
|
|
|
|
RegCloseKey( newkey );
|
|
}
|
|
|
|
if (subkeyCString) Recycle(subkeyCString);
|
|
if (valnameCString) Recycle(valnameCString);
|
|
|
|
return value;
|
|
#else
|
|
return nsnull;
|
|
#endif
|
|
}
|
|
|