Bommels05/Mozilla
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
cbb38ee9272c554731003310e1bfbcbb9d36fb8a
Mozilla/mozilla/xpinstall/src/nsWinReg.cpp
ssu%netscape.com 9f956e0118 bug 4933 - new image for the download/xpinstall dialog. 
bug 80237 - "destination directory" title is gone...
bug 77244 - Installers need to create Win32 Reg. key schema that Gecko Embedding browsers can follow
bug 78318 - crash with installer if you go back to change the directory selected
bug 56538 - Installer fails with error 5 installing on my Win2K box
bug 68053 - Win32 installer needs to log the installation process.
bug 81081 - Xpinstall dialog in the win32 installer requires touch up

sr=dveditz


git-svn-id: svn://10.0.0.236/trunk@95391 18797224-902f-48f8-a5cc-f745e15eee43
2001-05-18 10:22:54 +00:00

803 lines
20 KiB
C++
Raw Blame History

/* -*- 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 */
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 nsString& subkey, const nsString& classname, PRInt32* aReturn)
{
nsWinRegItem* wi = new nsWinRegItem(this, mRootKey, NS_WIN_REG_CREATE, subkey, classname, NS_ConvertASCIItoUCS2("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 nsString& subkey, PRInt32* aReturn)
{
nsWinRegItem* wi = new nsWinRegItem(this, mRootKey, NS_WIN_REG_DELETE, subkey, NS_ConvertASCIItoUCS2("null"), NS_ConvertASCIItoUCS2("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 nsString& subkey, const nsString& valname, PRInt32* aReturn)
{
nsWinRegItem* wi = new nsWinRegItem(this, mRootKey, NS_WIN_REG_DELETE_VAL, subkey, valname, NS_ConvertASCIItoUCS2("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 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)
*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::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);
*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::NativeKeyExists(const nsString& subkey)
{
HKEY root, newkey;
LONG result;
PRInt32 keyExists = PR_FALSE;
char* subkeyCString = subkey.ToNewCString();
#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 = subkey.ToNewCString();
char* valnameCString = valname.ToNewCString();
#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('\\', PR_FALSE, -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 = subkeyParent.ToNewCString();
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 = subkey.ToNewCString();
char* classnameCString = classname.ToNewCString();
#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 = 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_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 = 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(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 = subkey.ToNewCString();
char* valnameCString = valname.ToNewCString();
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 = 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(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 = subkey.ToNewCString();
char* valnameCString = valname.ToNewCString();
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 = 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 );
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
}
Reference in New Issue View Git Blame Copy Permalink