Files
Mozilla/mozilla/xpinstall/src/nsWinProfile.cpp
dougt%netscape.com e8c9ffe47b Fixes to address memory allocation problems - Bug #8227
git-svn-id: svn://10.0.0.236/trunk@40016 18797224-902f-48f8-a5cc-f745e15eee43
1999-07-18 20:56:12 +00:00

147 lines
3.7 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.0 (the "NPL"); you may not use this file except in
* compliance with the NPL. You may obtain a copy of the NPL at
* http://www.mozilla.org/NPL/
*
* Software distributed under the NPL is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
* for the specific language governing rights and limitations under the
* NPL.
*
* The Initial Developer of this code under the NPL is Netscape
* Communications Corporation. Portions created by Netscape are
* Copyright (C) 1998 Netscape Communications Corporation. All Rights
* Reserved.
*/
#include "nsWinProfile.h"
#include "nsWinProfileItem.h"
#include "nspr.h"
#include <windows.h>
/* Public Methods */
nsWinProfile::nsWinProfile( nsInstall* suObj, const nsString& folder, const nsString& file )
{
mFilename = new nsString(folder);
if (mFilename)
{
if(mFilename->Last() != '\\')
{
mFilename->Append("\\");
}
mFilename->Append(file);
mInstallObject = suObj;
}
}
nsWinProfile::~nsWinProfile()
{
if (mFilename)
delete mFilename;
}
PRInt32
nsWinProfile::GetString(nsString section, nsString key, nsString* aReturn)
{
return NativeGetString(section, key, aReturn);
}
PRInt32
nsWinProfile::WriteString(nsString section, nsString key, nsString value, PRInt32* aReturn)
{
*aReturn = NS_OK;
nsWinProfileItem* wi = new nsWinProfileItem(this, section, key, value);
if(wi == nsnull)
return nsInstall::OUT_OF_MEMORY;
if (mInstallObject)
mInstallObject->ScheduleForInstall(wi);
return NS_OK;
}
nsString* nsWinProfile::GetFilename()
{
return mFilename;
}
nsInstall* nsWinProfile::InstallObject()
{
return mInstallObject;
}
PRInt32
nsWinProfile::FinalWriteString( nsString section, nsString key, nsString value )
{
/* do we need another security check here? */
return NativeWriteString(section, key, value);
}
/* Private Methods */
#define STRBUFLEN 255
PRInt32
nsWinProfile::NativeGetString(nsString section, nsString key, nsString* aReturn )
{
int numChars;
char valbuf[STRBUFLEN];
char* sectionCString;
char* keyCString;
char* filenameCString;
/* make sure conversions worked */
if(section.First() != '\0' && key.First() != '\0' && mFilename->First() != '\0')
{
sectionCString = section.ToNewCString();
keyCString = key.ToNewCString();
filenameCString = mFilename->ToNewCString();
numChars = GetPrivateProfileString(sectionCString, keyCString, "", valbuf, STRBUFLEN, filenameCString);
*aReturn = valbuf;
if (sectionCString) delete [] sectionCString;
if (keyCString) delete [] keyCString;
if (filenameCString) delete [] filenameCString;
}
return numChars;
}
PRInt32
nsWinProfile::NativeWriteString( nsString section, nsString key, nsString value )
{
char* sectionCString;
char* keyCString;
char* valueCString;
char* filenameCString;
int success = 0;
/* make sure conversions worked */
if(section.First() != '\0' && key.First() != '\0' && mFilename->First() != '\0')
{
sectionCString = section.ToNewCString();
keyCString = key.ToNewCString();
valueCString = value.ToNewCString();
filenameCString = mFilename->ToNewCString();
success = WritePrivateProfileString( sectionCString, keyCString, valueCString, filenameCString );
if (sectionCString) delete [] sectionCString;
if (keyCString) delete [] keyCString;
if (valueCString) delete [] valueCString;
if (filenameCString) delete [] filenameCString;
}
return success;
}