048c44e151
runtime, they should no-op, not bring an install script to its knees. r=hewitt, sr=dveditz git-svn-id: svn://10.0.0.236/trunk@127603 18797224-902f-48f8-a5cc-f745e15eee43
902 lines
23 KiB
C++
902 lines
23 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* ***** BEGIN LICENSE BLOCK *****
|
|
* Version: NPL 1.1/GPL 2.0/LGPL 2.1
|
|
*
|
|
* 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 the Initial Developer are Copyright (C) 1998
|
|
* the Initial Developer. All Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
*
|
|
* Alternatively, the contents of this file may be used under the terms of
|
|
* either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
* in which case the provisions of the GPL or the LGPL are applicable instead
|
|
* of those above. If you wish to allow use of your version of this file only
|
|
* under the terms of either the GPL or the LGPL, and not to allow others to
|
|
* use your version of this file under the terms of the NPL, indicate your
|
|
* decision by deleting the provisions above and replace them with the notice
|
|
* and other provisions required by the GPL or the LGPL. If you do not delete
|
|
* the provisions above, a recipient may use your version of this file under
|
|
* the terms of any one of the NPL, the GPL or the LGPL.
|
|
*
|
|
* ***** END LICENSE BLOCK ***** */
|
|
|
|
#include "nsWinReg.h"
|
|
#include "nsWinRegItem.h"
|
|
#include <windows.h> /* is this needed? */
|
|
#include "nsReadableUtils.h"
|
|
|
|
/* Public Methods */
|
|
|
|
MOZ_DECL_CTOR_COUNTER(nsWinReg)
|
|
|
|
nsWinReg::nsWinReg(nsInstall* suObj)
|
|
{
|
|
MOZ_COUNT_CTOR(nsWinReg);
|
|
|
|
mInstallObject = suObj;
|
|
mRootKey = (PRInt32)HKEY_CLASSES_ROOT;
|
|
}
|
|
|
|
nsWinReg::~nsWinReg()
|
|
{
|
|
MOZ_COUNT_DTOR(nsWinReg);
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::SetRootKey(PRInt32 key)
|
|
{
|
|
mRootKey = key;
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::CreateKey(const nsAString& subkey, const nsAString& classname, PRInt32* aReturn)
|
|
{
|
|
nsWinRegItem* wi = new nsWinRegItem(this, mRootKey, NS_WIN_REG_CREATE, subkey, classname, NS_LITERAL_STRING("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)
|
|
*aReturn = mInstallObject->ScheduleForInstall(wi);
|
|
|
|
return 0;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::DeleteKey(const nsAString& subkey, PRInt32* aReturn)
|
|
{
|
|
nsWinRegItem* wi = new nsWinRegItem(this, mRootKey, NS_WIN_REG_DELETE, subkey, NS_LITERAL_STRING("null"), NS_LITERAL_STRING("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)
|
|
*aReturn = mInstallObject->ScheduleForInstall(wi);
|
|
|
|
return 0;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::DeleteValue(const nsAString& subkey, const nsAString& valname, PRInt32* aReturn)
|
|
{
|
|
nsWinRegItem* wi = new nsWinRegItem(this, mRootKey, NS_WIN_REG_DELETE_VAL, subkey, valname, NS_LITERAL_STRING("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)
|
|
*aReturn = mInstallObject->ScheduleForInstall(wi);
|
|
|
|
return 0;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::SetValueString(const nsAString& subkey, const nsAString& valname, const nsAString& 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)
|
|
*aReturn = 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::EnumValueNames(const nsString& aSubkey, PRInt32 aIndex, nsString &aReturn)
|
|
{
|
|
char namebuf[MAX_BUF];
|
|
HKEY root;
|
|
HKEY newkey;
|
|
LONG result;
|
|
DWORD namesize = sizeof(namebuf);
|
|
char subkeyCString[MAX_BUF];
|
|
unsigned short returnBuf[MAX_BUF];
|
|
PRInt32 rv = NS_ERROR_FAILURE;
|
|
|
|
subkeyCString[0] = 0;
|
|
returnBuf[0] = 0;
|
|
namebuf[0] = 0;
|
|
|
|
::WideCharToMultiByte(CP_ACP, 0,
|
|
aSubkey.get(), -1,
|
|
subkeyCString, sizeof subkeyCString, NULL, NULL);
|
|
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx( root, subkeyCString, 0, KEY_READ, &newkey );
|
|
|
|
if ( ERROR_SUCCESS == result ) {
|
|
result = RegEnumValue( newkey, aIndex, namebuf, &namesize, nsnull, 0, 0, 0 );
|
|
RegCloseKey( newkey );
|
|
|
|
if ( ERROR_SUCCESS == result ) {
|
|
if ( ::MultiByteToWideChar(CP_ACP, 0, namebuf, -1, returnBuf, MAX_BUF) )
|
|
{
|
|
aReturn.Assign(returnBuf);
|
|
rv = NS_OK;
|
|
}
|
|
}
|
|
}
|
|
|
|
return rv;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::EnumKeys(const nsString& aSubkey, PRInt32 aIndex, nsString &aReturn)
|
|
{
|
|
char keybuf[MAX_BUF];
|
|
HKEY root;
|
|
HKEY newkey;
|
|
LONG result;
|
|
DWORD type = REG_SZ;
|
|
char subkeyCString[MAX_BUF];
|
|
unsigned short returnBuf[MAX_BUF];
|
|
PRInt32 rv = NS_ERROR_FAILURE;
|
|
|
|
subkeyCString[0] = 0;
|
|
returnBuf[0] = 0;
|
|
keybuf[0] = 0;
|
|
|
|
::WideCharToMultiByte(CP_ACP, 0,
|
|
aSubkey.get(), -1,
|
|
subkeyCString, sizeof subkeyCString, NULL, NULL);
|
|
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx( root, subkeyCString, 0, KEY_READ, &newkey );
|
|
|
|
if ( ERROR_SUCCESS == result ) {
|
|
result = RegEnumKey( newkey, aIndex, keybuf, sizeof keybuf );
|
|
RegCloseKey( newkey );
|
|
|
|
if ( ERROR_SUCCESS == result ) {
|
|
if ( ::MultiByteToWideChar(CP_ACP, 0, keybuf, -1, returnBuf, MAX_BUF) )
|
|
{
|
|
aReturn.Assign(returnBuf);
|
|
rv = NS_OK;
|
|
}
|
|
}
|
|
}
|
|
|
|
return rv;
|
|
}
|
|
|
|
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)
|
|
*aReturn = 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::KeyExists(const nsString& subkey,
|
|
PRInt32* aReturn)
|
|
{
|
|
*aReturn = NativeKeyExists(subkey);
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::ValueExists(const nsString& subkey,
|
|
const nsString& valname,
|
|
PRInt32* aReturn)
|
|
{
|
|
*aReturn = NativeValueExists(subkey, valname);
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::IsKeyWritable(const nsString& subkey,
|
|
PRInt32* aReturn)
|
|
{
|
|
*aReturn = NativeIsKeyWritable(subkey);
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::PrepareCreateKey(PRInt32 root,
|
|
const nsString& subkey,
|
|
PRInt32* aReturn)
|
|
{
|
|
SetRootKey(root);
|
|
if(NativeIsKeyWritable(subkey))
|
|
*aReturn = nsInstall::SUCCESS;
|
|
else
|
|
*aReturn = nsInstall::KEY_ACCESS_DENIED;
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::PrepareDeleteKey(PRInt32 root,
|
|
const nsString& subkey,
|
|
PRInt32* aReturn)
|
|
{
|
|
SetRootKey(root);
|
|
if(NativeKeyExists(subkey))
|
|
{
|
|
if(NativeIsKeyWritable(subkey))
|
|
*aReturn = nsInstall::SUCCESS;
|
|
else
|
|
*aReturn = nsInstall::KEY_ACCESS_DENIED;
|
|
}
|
|
else
|
|
*aReturn = nsInstall::KEY_DOES_NOT_EXIST;
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::PrepareDeleteValue(PRInt32 root,
|
|
const nsString& subkey,
|
|
const nsString& valname,
|
|
PRInt32* aReturn)
|
|
{
|
|
SetRootKey(root);
|
|
if(NativeValueExists(subkey, valname))
|
|
{
|
|
if(NativeIsKeyWritable(subkey))
|
|
*aReturn = nsInstall::SUCCESS;
|
|
else
|
|
*aReturn = nsInstall::KEY_ACCESS_DENIED;
|
|
}
|
|
else
|
|
*aReturn = nsInstall::VALUE_DOES_NOT_EXIST;
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::PrepareSetValueString(PRInt32 root,
|
|
const nsString& subkey,
|
|
PRInt32* aReturn)
|
|
{
|
|
SetRootKey(root);
|
|
if(NativeIsKeyWritable(subkey))
|
|
*aReturn = nsInstall::SUCCESS;
|
|
else
|
|
*aReturn = nsInstall::KEY_ACCESS_DENIED;
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::PrepareSetValueNumber(PRInt32 root,
|
|
const nsString& subkey,
|
|
PRInt32* aReturn)
|
|
{
|
|
SetRootKey(root);
|
|
if(NativeIsKeyWritable(subkey))
|
|
*aReturn = nsInstall::SUCCESS;
|
|
else
|
|
*aReturn = nsInstall::KEY_ACCESS_DENIED;
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::PrepareSetValue(PRInt32 root,
|
|
const nsString& subkey,
|
|
PRInt32* aReturn)
|
|
{
|
|
SetRootKey(root);
|
|
if(NativeIsKeyWritable(subkey))
|
|
*aReturn = nsInstall::SUCCESS;
|
|
else
|
|
*aReturn = nsInstall::KEY_ACCESS_DENIED;
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
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);
|
|
if ( PR_TRUE == NativeKeyExists(subkey) )
|
|
*aReturn = NativeDeleteKey(subkey);
|
|
else {
|
|
NS_WARNING("Trying to delete a key that doesn't exist anymore. Possible causes include calling deleteKey for the same key multiple times, or key+subkey are not unique");
|
|
*aReturn = ERROR_SUCCESS;
|
|
}
|
|
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::NativeKeyExists(const nsString& subkey)
|
|
{
|
|
HKEY root, newkey;
|
|
LONG result;
|
|
PRInt32 keyExists = PR_FALSE;
|
|
char* subkeyCString = ToNewCString(subkey);
|
|
|
|
#ifdef WIN32
|
|
root = (HKEY)mRootKey;
|
|
result = RegOpenKeyEx(root, subkeyCString, 0, KEY_READ, &newkey);
|
|
switch(result)
|
|
{
|
|
case ERROR_SUCCESS:
|
|
RegCloseKey(newkey);
|
|
keyExists = PR_TRUE;
|
|
break;
|
|
|
|
case ERROR_FILE_NOT_FOUND:
|
|
case ERROR_ACCESS_DENIED:
|
|
default:
|
|
break;
|
|
}
|
|
#endif
|
|
|
|
if (subkeyCString) Recycle(subkeyCString);
|
|
|
|
return keyExists;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::NativeValueExists(const nsString& subkey, const nsString& valname)
|
|
{
|
|
HKEY root;
|
|
HKEY newkey;
|
|
LONG result;
|
|
PRInt32 valueExists = PR_FALSE;
|
|
DWORD length = _MAXKEYVALUE_;
|
|
unsigned char valbuf[_MAXKEYVALUE_];
|
|
char* subkeyCString = ToNewCString(subkey);
|
|
char* valnameCString = ToNewCString(valname);
|
|
|
|
#ifdef WIN32
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx(root, subkeyCString, 0, KEY_READ, &newkey);
|
|
switch(result)
|
|
{
|
|
case ERROR_SUCCESS:
|
|
result = RegQueryValueEx(newkey,
|
|
valnameCString,
|
|
0,
|
|
NULL,
|
|
valbuf,
|
|
&length);
|
|
switch(result)
|
|
{
|
|
case ERROR_SUCCESS:
|
|
valueExists = PR_TRUE;
|
|
break;
|
|
|
|
case ERROR_FILE_NOT_FOUND:
|
|
case ERROR_ACCESS_DENIED:
|
|
default:
|
|
break;
|
|
}
|
|
RegCloseKey(newkey);
|
|
break;
|
|
|
|
case ERROR_FILE_NOT_FOUND:
|
|
case ERROR_ACCESS_DENIED:
|
|
default:
|
|
break;
|
|
}
|
|
#endif
|
|
|
|
if (subkeyCString) Recycle(subkeyCString);
|
|
if (valnameCString) Recycle(valnameCString);
|
|
|
|
return valueExists;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::NativeIsKeyWritable(const nsString& subkey)
|
|
{
|
|
HKEY root;
|
|
HKEY newkey;
|
|
LONG result;
|
|
char* subkeyCString;
|
|
nsString subkeyParent = subkey;
|
|
PRInt32 index;
|
|
PRInt32 rv = PR_FALSE;
|
|
|
|
#ifdef WIN32
|
|
/* In order to check to see if a key is writable (user has write access
|
|
* to the key), we need to open the key with KEY_WRITE access. In order
|
|
* to open a key, the key must first exist. If the key passed does not
|
|
* exist, we need to search for one of its parent key that _does_ exist.
|
|
* This do{} loop will search for an existing parent key. */
|
|
do
|
|
{
|
|
rv = NativeKeyExists(subkeyParent);
|
|
if(!rv)
|
|
{
|
|
index = subkeyParent.RFindChar('\\', -1, -1);
|
|
if(index > 0)
|
|
/* delete everything from the '\\' found to the end of the string */
|
|
subkeyParent.SetLength(index);
|
|
else
|
|
/* key does not exist and no parent key found */
|
|
break;
|
|
}
|
|
}while(!rv);
|
|
|
|
if(rv)
|
|
{
|
|
rv = PR_FALSE;
|
|
subkeyCString = ToNewCString(subkeyParent);
|
|
if(!subkeyCString)
|
|
result = nsInstall::OUT_OF_MEMORY;
|
|
else
|
|
{
|
|
root = (HKEY)mRootKey;
|
|
result = RegOpenKeyEx(root, subkeyCString, 0, KEY_WRITE, &newkey);
|
|
switch(result)
|
|
{
|
|
case ERROR_SUCCESS:
|
|
RegCloseKey(newkey);
|
|
rv = PR_TRUE;
|
|
break;
|
|
|
|
case ERROR_FILE_NOT_FOUND:
|
|
case ERROR_ACCESS_DENIED:
|
|
default:
|
|
break;
|
|
}
|
|
if(subkeyCString)
|
|
Recycle(subkeyCString);
|
|
}
|
|
}
|
|
#endif
|
|
return rv;
|
|
}
|
|
|
|
PRInt32
|
|
nsWinReg::NativeCreateKey(const nsString& subkey, const nsString& classname)
|
|
{
|
|
HKEY root, newkey;
|
|
LONG result;
|
|
ULONG disposition;
|
|
char* subkeyCString = ToNewCString(subkey);
|
|
char* classnameCString = ToNewCString(classname);
|
|
|
|
#ifdef WIN32
|
|
root = (HKEY)mRootKey;
|
|
result = RegCreateKeyEx(root, subkeyCString, 0, classnameCString, REG_OPTION_NON_VOLATILE, KEY_WRITE, 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 = ToNewCString(subkey);
|
|
|
|
#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 = ToNewCString(subkey);
|
|
char* valnameCString = ToNewCString(valname);
|
|
|
|
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 = ToNewCString(subkey);
|
|
char* valnameCString = ToNewCString(valname);
|
|
char* valueCString = ToNewCString(value);
|
|
|
|
length = value.Length();
|
|
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx( root, subkeyCString, 0, KEY_WRITE, &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 = ToNewCString(subkey);
|
|
char* valnameCString = ToNewCString(valname);
|
|
|
|
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(type == REG_SZ)
|
|
aReturn->AssignWithConversion((char*)valbuf);
|
|
}
|
|
|
|
if(ERROR_ACCESS_DENIED == result)
|
|
result = nsInstall::ACCESS_DENIED;
|
|
|
|
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 = ToNewCString(subkey);
|
|
char* valnameCString = ToNewCString(valname);
|
|
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx( root, subkeyCString, 0, KEY_WRITE, &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 = ToNewCString(subkey);
|
|
char* valnameCString = ToNewCString(valname);
|
|
|
|
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(type == REG_DWORD)
|
|
*aReturn = valbuf;
|
|
}
|
|
|
|
if(ERROR_ACCESS_DENIED == result)
|
|
result = nsInstall::ACCESS_DENIED;
|
|
|
|
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 = ToNewCString(subkey);
|
|
char* valnameCString = ToNewCString(valname);
|
|
|
|
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx( root, subkeyCString, 0, KEY_WRITE, &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 = ToNewCString(subkey);
|
|
char* valnameCString = ToNewCString(valname);
|
|
|
|
root = (HKEY) mRootKey;
|
|
result = RegOpenKeyEx( root, subkeyCString, 0, KEY_READ, &newkey );
|
|
|
|
if(ERROR_SUCCESS == result)
|
|
{
|
|
result = RegQueryValueEx( newkey, valnameCString, nsnull, &type, valbuf, &length );
|
|
|
|
if ( ERROR_SUCCESS == result ) {
|
|
data = new nsString;
|
|
data->AssignWithConversion((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
|
|
}
|
|
|