зеркало из https://github.com/mozilla/gecko-dev.git
958 строки
25 KiB
C++
958 строки
25 KiB
C++
|
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
|
||
|
/* ***** BEGIN LICENSE BLOCK *****
|
||
|
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
|
||
|
*
|
||
|
* The contents of this file are subject to the Mozilla 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/MPL/
|
||
|
*
|
||
|
* 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 the new Mozilla toolkit.
|
||
|
*
|
||
|
* The Initial Developer of the Original Code is
|
||
|
* Benjamin Smedberg <bsmedberg@covad.net>
|
||
|
* Portions created by the Initial Developer are Copyright (C) 2004
|
||
|
* 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 MPL, 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 MPL, the GPL or the LGPL.
|
||
|
*
|
||
|
* ***** END LICENSE BLOCK ***** */
|
||
|
|
||
|
#include <stdio.h>
|
||
|
#include <stdlib.h>
|
||
|
#include "nsProfileLock.h"
|
||
|
|
||
|
#ifdef XP_WIN
|
||
|
#include <windows.h>
|
||
|
#include <shlobj.h>
|
||
|
#endif
|
||
|
#ifdef XP_UNIX
|
||
|
#include <unistd.h>
|
||
|
#endif
|
||
|
|
||
|
#include "nsIToolkitProfileService.h"
|
||
|
#include "nsIToolkitProfile.h"
|
||
|
#include "nsIFactory.h"
|
||
|
#include "nsILocalFile.h"
|
||
|
#include "nsISimpleEnumerator.h"
|
||
|
|
||
|
#ifdef XP_MACOSX
|
||
|
#include <CoreFoundation/CoreFoundation.h>
|
||
|
#include "nsILocalFileMac.h"
|
||
|
#endif
|
||
|
|
||
|
#include "nsAppDirectoryServiceDefs.h"
|
||
|
#include "nsXULAppAPI.h"
|
||
|
|
||
|
#include "nsINIParser.h"
|
||
|
#include "nsXREDirProvider.h"
|
||
|
#include "nsAppRunner.h"
|
||
|
#include "nsString.h"
|
||
|
#include "nsReadableUtils.h"
|
||
|
#include "nsNativeCharsetUtils.h"
|
||
|
|
||
|
|
||
|
class nsToolkitProfile : public nsIToolkitProfile
|
||
|
{
|
||
|
public:
|
||
|
NS_DECL_ISUPPORTS
|
||
|
NS_DECL_NSITOOLKITPROFILE
|
||
|
|
||
|
friend class nsToolkitProfileService;
|
||
|
nsCOMPtr<nsToolkitProfile> mNext;
|
||
|
nsToolkitProfile *mPrev;
|
||
|
|
||
|
~nsToolkitProfile() { }
|
||
|
|
||
|
private:
|
||
|
nsToolkitProfile(const nsACString& aName,
|
||
|
nsILocalFile* aRootDir,
|
||
|
nsILocalFile* aLocalDir,
|
||
|
nsToolkitProfile* aPrev);
|
||
|
|
||
|
friend class nsToolkitProfileLock;
|
||
|
|
||
|
nsCString mName;
|
||
|
nsCOMPtr<nsILocalFile> mRootDir;
|
||
|
nsCOMPtr<nsILocalFile> mLocalDir;
|
||
|
nsIProfileLock* mLock;
|
||
|
};
|
||
|
|
||
|
class nsToolkitProfileLock : public nsIProfileLock
|
||
|
{
|
||
|
public:
|
||
|
NS_DECL_ISUPPORTS
|
||
|
NS_DECL_NSIPROFILELOCK
|
||
|
|
||
|
nsresult Init(nsToolkitProfile* aProfile, nsIProfileUnlocker* *aUnlocker);
|
||
|
nsresult Init(nsILocalFile* aDirectory, nsILocalFile* aLocalDirectory,
|
||
|
nsIProfileUnlocker* *aUnlocker);
|
||
|
|
||
|
nsToolkitProfileLock() { }
|
||
|
~nsToolkitProfileLock();
|
||
|
|
||
|
private:
|
||
|
nsCOMPtr<nsToolkitProfile> mProfile;
|
||
|
nsCOMPtr<nsILocalFile> mDirectory;
|
||
|
nsCOMPtr<nsILocalFile> mLocalDirectory;
|
||
|
|
||
|
nsProfileLock mLock;
|
||
|
};
|
||
|
|
||
|
class nsToolkitProfileFactory : public nsIFactory
|
||
|
{
|
||
|
public:
|
||
|
NS_DECL_ISUPPORTS
|
||
|
NS_DECL_NSIFACTORY
|
||
|
};
|
||
|
|
||
|
class nsToolkitProfileService : public nsIToolkitProfileService
|
||
|
{
|
||
|
public:
|
||
|
NS_DECL_ISUPPORTS
|
||
|
NS_DECL_NSITOOLKITPROFILESERVICE
|
||
|
|
||
|
private:
|
||
|
friend class nsToolkitProfile;
|
||
|
friend class nsToolkitProfileFactory;
|
||
|
friend nsresult NS_NewToolkitProfileService(nsIToolkitProfileService**);
|
||
|
|
||
|
nsToolkitProfileService() :
|
||
|
mDirty(PR_FALSE),
|
||
|
mStartWithLast(PR_TRUE),
|
||
|
mStartOffline(PR_FALSE)
|
||
|
{
|
||
|
gService = this;
|
||
|
}
|
||
|
~nsToolkitProfileService()
|
||
|
{
|
||
|
gService = nsnull;
|
||
|
}
|
||
|
|
||
|
NS_HIDDEN_(nsresult) Init();
|
||
|
|
||
|
nsRefPtr<nsToolkitProfile> mFirst;
|
||
|
nsCOMPtr<nsIToolkitProfile> mChosen;
|
||
|
nsCOMPtr<nsILocalFile> mAppData;
|
||
|
nsCOMPtr<nsILocalFile> mTempData;
|
||
|
nsCOMPtr<nsILocalFile> mListFile;
|
||
|
PRBool mDirty;
|
||
|
PRBool mStartWithLast;
|
||
|
PRBool mStartOffline;
|
||
|
|
||
|
static nsToolkitProfileService *gService;
|
||
|
|
||
|
class ProfileEnumerator : public nsISimpleEnumerator
|
||
|
{
|
||
|
public:
|
||
|
NS_DECL_ISUPPORTS
|
||
|
NS_DECL_NSISIMPLEENUMERATOR
|
||
|
|
||
|
ProfileEnumerator(nsToolkitProfile *first)
|
||
|
{ mCurrent = first; }
|
||
|
private:
|
||
|
~ProfileEnumerator() { }
|
||
|
nsCOMPtr<nsToolkitProfile> mCurrent;
|
||
|
};
|
||
|
};
|
||
|
|
||
|
nsToolkitProfile::nsToolkitProfile(const nsACString& aName,
|
||
|
nsILocalFile* aRootDir,
|
||
|
nsILocalFile* aLocalDir,
|
||
|
nsToolkitProfile* aPrev) :
|
||
|
mPrev(aPrev),
|
||
|
mName(aName),
|
||
|
mRootDir(aRootDir),
|
||
|
mLocalDir(aLocalDir),
|
||
|
mLock(nsnull)
|
||
|
{
|
||
|
NS_ASSERTION(aRootDir, "No file!");
|
||
|
|
||
|
if (aPrev)
|
||
|
aPrev->mNext = this;
|
||
|
else
|
||
|
nsToolkitProfileService::gService->mFirst = this;
|
||
|
}
|
||
|
|
||
|
NS_IMPL_ISUPPORTS1(nsToolkitProfile, nsIToolkitProfile)
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfile::GetRootDir(nsILocalFile* *aResult)
|
||
|
{
|
||
|
NS_ADDREF(*aResult = mRootDir);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfile::GetLocalDir(nsILocalFile* *aResult)
|
||
|
{
|
||
|
NS_ADDREF(*aResult = mLocalDir);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfile::GetName(nsACString& aResult)
|
||
|
{
|
||
|
aResult = mName;
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfile::SetName(const nsACString& aName)
|
||
|
{
|
||
|
NS_ASSERTION(nsToolkitProfileService::gService,
|
||
|
"Where did my service go?");
|
||
|
|
||
|
mName = aName;
|
||
|
nsToolkitProfileService::gService->mDirty = PR_TRUE;
|
||
|
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfile::Remove(PRBool removeFiles)
|
||
|
{
|
||
|
NS_ASSERTION(nsToolkitProfileService::gService,
|
||
|
"Whoa, my service is gone.");
|
||
|
|
||
|
if (mLock)
|
||
|
return NS_ERROR_FILE_IS_LOCKED;
|
||
|
|
||
|
if (removeFiles) {
|
||
|
PRBool equals;
|
||
|
nsresult rv = mRootDir->Equals(mLocalDir, &equals);
|
||
|
if (NS_FAILED(rv))
|
||
|
return rv;
|
||
|
|
||
|
// The root dir might contain the temp dir, so remove
|
||
|
// the temp dir first.
|
||
|
if (!equals)
|
||
|
mLocalDir->Remove(PR_TRUE);
|
||
|
|
||
|
mRootDir->Remove(PR_TRUE);
|
||
|
}
|
||
|
|
||
|
if (mPrev)
|
||
|
mPrev->mNext = mNext;
|
||
|
else
|
||
|
nsToolkitProfileService::gService->mFirst = mNext;
|
||
|
|
||
|
if (mNext)
|
||
|
mNext->mPrev = mPrev;
|
||
|
|
||
|
mPrev = nsnull;
|
||
|
mNext = nsnull;
|
||
|
|
||
|
if (nsToolkitProfileService::gService->mChosen == this)
|
||
|
nsToolkitProfileService::gService->mChosen = nsnull;
|
||
|
|
||
|
nsToolkitProfileService::gService->mDirty = PR_TRUE;
|
||
|
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfile::Lock(nsIProfileUnlocker* *aUnlocker, nsIProfileLock* *aResult)
|
||
|
{
|
||
|
if (mLock) {
|
||
|
NS_ADDREF(*aResult = mLock);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
nsCOMPtr<nsToolkitProfileLock> lock = new nsToolkitProfileLock();
|
||
|
if (!lock) return NS_ERROR_OUT_OF_MEMORY;
|
||
|
|
||
|
nsresult rv = lock->Init(this, aUnlocker);
|
||
|
if (NS_FAILED(rv)) return rv;
|
||
|
|
||
|
NS_ADDREF(*aResult = lock);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMPL_ISUPPORTS1(nsToolkitProfileLock, nsIProfileLock)
|
||
|
|
||
|
nsresult
|
||
|
nsToolkitProfileLock::Init(nsToolkitProfile* aProfile, nsIProfileUnlocker* *aUnlocker)
|
||
|
{
|
||
|
nsresult rv;
|
||
|
rv = Init(aProfile->mRootDir, aProfile->mLocalDir, aUnlocker);
|
||
|
if (NS_SUCCEEDED(rv))
|
||
|
mProfile = aProfile;
|
||
|
|
||
|
return rv;
|
||
|
}
|
||
|
|
||
|
nsresult
|
||
|
nsToolkitProfileLock::Init(nsILocalFile* aDirectory, nsILocalFile* aLocalDirectory,
|
||
|
nsIProfileUnlocker* *aUnlocker)
|
||
|
{
|
||
|
nsresult rv;
|
||
|
|
||
|
rv = mLock.Lock(aDirectory, aUnlocker);
|
||
|
|
||
|
if (NS_SUCCEEDED(rv)) {
|
||
|
mDirectory = aDirectory;
|
||
|
mLocalDirectory = aLocalDirectory;
|
||
|
}
|
||
|
|
||
|
return rv;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileLock::GetDirectory(nsILocalFile* *aResult)
|
||
|
{
|
||
|
if (!mDirectory) {
|
||
|
NS_ERROR("Not initialized, or unlocked!");
|
||
|
return NS_ERROR_NOT_INITIALIZED;
|
||
|
}
|
||
|
|
||
|
NS_ADDREF(*aResult = mDirectory);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileLock::GetLocalDirectory(nsILocalFile* *aResult)
|
||
|
{
|
||
|
if (!mLocalDirectory) {
|
||
|
NS_ERROR("Not initialized, or unlocked!");
|
||
|
return NS_ERROR_NOT_INITIALIZED;
|
||
|
}
|
||
|
|
||
|
NS_ADDREF(*aResult = mLocalDirectory);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileLock::Unlock()
|
||
|
{
|
||
|
if (!mDirectory) {
|
||
|
NS_ERROR("Unlocking a never-locked nsToolkitProfileLock!");
|
||
|
return NS_ERROR_UNEXPECTED;
|
||
|
}
|
||
|
|
||
|
mLock.Unlock();
|
||
|
|
||
|
if (mProfile) {
|
||
|
mProfile->mLock = nsnull;
|
||
|
mProfile = nsnull;
|
||
|
}
|
||
|
mDirectory = nsnull;
|
||
|
mLocalDirectory = nsnull;
|
||
|
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
nsToolkitProfileLock::~nsToolkitProfileLock()
|
||
|
{
|
||
|
if (mDirectory) {
|
||
|
Unlock();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
nsToolkitProfileService*
|
||
|
nsToolkitProfileService::gService = nsnull;
|
||
|
|
||
|
NS_IMPL_ISUPPORTS1(nsToolkitProfileService,
|
||
|
nsIToolkitProfileService)
|
||
|
|
||
|
nsresult
|
||
|
nsToolkitProfileService::Init()
|
||
|
{
|
||
|
NS_ASSERTION(gDirServiceProvider, "No dirserviceprovider!");
|
||
|
nsresult rv;
|
||
|
|
||
|
rv = gDirServiceProvider->GetUserAppDataDirectory(getter_AddRefs(mAppData));
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
rv = gDirServiceProvider->GetUserLocalDataDirectory(getter_AddRefs(mTempData));
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
nsCOMPtr<nsIFile> listFile;
|
||
|
rv = mAppData->Clone(getter_AddRefs(listFile));
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
mListFile = do_QueryInterface(listFile);
|
||
|
NS_ENSURE_TRUE(listFile, NS_ERROR_NO_INTERFACE);
|
||
|
|
||
|
rv = mListFile->AppendNative(NS_LITERAL_CSTRING("profiles.ini"));
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
PRBool exists;
|
||
|
rv = mListFile->IsFile(&exists);
|
||
|
if (NS_FAILED(rv) || !exists) {
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
PRInt64 size;
|
||
|
rv = listFile->GetFileSize(&size);
|
||
|
if (NS_FAILED(rv) || !size) {
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
nsINIParser parser;
|
||
|
rv = parser.Init(mListFile);
|
||
|
// Init does not fail on parsing errors, only on OOM/really unexpected
|
||
|
// conditions.
|
||
|
if (NS_FAILED(rv))
|
||
|
return rv;
|
||
|
|
||
|
nsCAutoString buffer;
|
||
|
rv = parser.GetString("General", "StartWithLastProfile", buffer);
|
||
|
if (NS_SUCCEEDED(rv) && buffer.EqualsLiteral("0"))
|
||
|
mStartWithLast = PR_FALSE;
|
||
|
|
||
|
nsToolkitProfile* currentProfile = nsnull;
|
||
|
|
||
|
unsigned int c = 0;
|
||
|
for (c = 0; PR_TRUE; ++c) {
|
||
|
nsCAutoString profileID("Profile");
|
||
|
profileID.AppendInt(c);
|
||
|
|
||
|
rv = parser.GetString(profileID.get(), "IsRelative", buffer);
|
||
|
if (NS_FAILED(rv)) break;
|
||
|
|
||
|
PRBool isRelative = buffer.EqualsLiteral("1");
|
||
|
|
||
|
nsCAutoString filePath;
|
||
|
|
||
|
rv = parser.GetString(profileID.get(), "Path", filePath);
|
||
|
if (NS_FAILED(rv)) {
|
||
|
NS_ERROR("Malformed profiles.ini: Path= not found");
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
rv = parser.GetString(profileID.get(), "Name", buffer);
|
||
|
if (NS_FAILED(rv)) {
|
||
|
NS_ERROR("Malformed profiles.ini: Name= not found");
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
nsCOMPtr<nsILocalFile> rootDir;
|
||
|
rv = NS_NewNativeLocalFile(EmptyCString(), PR_TRUE,
|
||
|
getter_AddRefs(rootDir));
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
if (isRelative) {
|
||
|
rv = rootDir->SetRelativeDescriptor(mAppData, filePath);
|
||
|
} else {
|
||
|
rv = rootDir->SetPersistentDescriptor(filePath);
|
||
|
}
|
||
|
if (NS_FAILED(rv)) continue;
|
||
|
|
||
|
nsCOMPtr<nsILocalFile> localDir;
|
||
|
if (isRelative) {
|
||
|
rv = NS_NewNativeLocalFile(EmptyCString(), PR_TRUE,
|
||
|
getter_AddRefs(localDir));
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
rv = localDir->SetRelativeDescriptor(mTempData, filePath);
|
||
|
} else {
|
||
|
localDir = rootDir;
|
||
|
}
|
||
|
|
||
|
currentProfile = new nsToolkitProfile(buffer,
|
||
|
rootDir, localDir,
|
||
|
currentProfile);
|
||
|
NS_ENSURE_TRUE(currentProfile, NS_ERROR_OUT_OF_MEMORY);
|
||
|
|
||
|
rv = parser.GetString(profileID.get(), "Default", buffer);
|
||
|
if (NS_SUCCEEDED(rv) && buffer.EqualsLiteral("1"))
|
||
|
mChosen = currentProfile;
|
||
|
}
|
||
|
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::SetStartWithLastProfile(PRBool aValue)
|
||
|
{
|
||
|
if (mStartWithLast != aValue) {
|
||
|
mStartWithLast = aValue;
|
||
|
mDirty = PR_TRUE;
|
||
|
}
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::GetStartWithLastProfile(PRBool *aResult)
|
||
|
{
|
||
|
*aResult = mStartWithLast;
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::GetStartOffline(PRBool *aResult)
|
||
|
{
|
||
|
*aResult = mStartOffline;
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::SetStartOffline(PRBool aValue)
|
||
|
{
|
||
|
mStartOffline = aValue;
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::GetProfiles(nsISimpleEnumerator* *aResult)
|
||
|
{
|
||
|
*aResult = new ProfileEnumerator(this->mFirst);
|
||
|
if (!*aResult)
|
||
|
return NS_ERROR_OUT_OF_MEMORY;
|
||
|
|
||
|
NS_ADDREF(*aResult);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMPL_ISUPPORTS1(nsToolkitProfileService::ProfileEnumerator,
|
||
|
nsISimpleEnumerator)
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::ProfileEnumerator::HasMoreElements(PRBool* aResult)
|
||
|
{
|
||
|
*aResult = mCurrent ? PR_TRUE : PR_FALSE;
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::ProfileEnumerator::GetNext(nsISupports* *aResult)
|
||
|
{
|
||
|
if (!mCurrent) return NS_ERROR_FAILURE;
|
||
|
|
||
|
NS_ADDREF(*aResult = mCurrent);
|
||
|
|
||
|
mCurrent = mCurrent->mNext;
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::GetSelectedProfile(nsIToolkitProfile* *aResult)
|
||
|
{
|
||
|
if (!mChosen && mFirst && !mFirst->mNext) // only one profile
|
||
|
mChosen = mFirst;
|
||
|
|
||
|
if (!mChosen) return NS_ERROR_FAILURE;
|
||
|
|
||
|
NS_ADDREF(*aResult = mChosen);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::SetSelectedProfile(nsIToolkitProfile* aProfile)
|
||
|
{
|
||
|
if (mChosen != aProfile) {
|
||
|
mChosen = aProfile;
|
||
|
mDirty = PR_TRUE;
|
||
|
}
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::GetProfileByName(const nsACString& aName,
|
||
|
nsIToolkitProfile* *aResult)
|
||
|
{
|
||
|
nsToolkitProfile* curP = mFirst;
|
||
|
while (curP) {
|
||
|
if (curP->mName.Equals(aName)) {
|
||
|
NS_ADDREF(*aResult = curP);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
curP = curP->mNext;
|
||
|
}
|
||
|
|
||
|
return NS_ERROR_FAILURE;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::LockProfilePath(nsILocalFile* aDirectory,
|
||
|
nsILocalFile* aLocalDirectory,
|
||
|
nsIProfileLock* *aResult)
|
||
|
{
|
||
|
return NS_LockProfilePath(aDirectory, aLocalDirectory, nsnull, aResult);
|
||
|
}
|
||
|
|
||
|
nsresult
|
||
|
NS_LockProfilePath(nsILocalFile* aPath, nsILocalFile* aTempPath,
|
||
|
nsIProfileUnlocker* *aUnlocker, nsIProfileLock* *aResult)
|
||
|
{
|
||
|
nsCOMPtr<nsToolkitProfileLock> lock = new nsToolkitProfileLock();
|
||
|
if (!lock) return NS_ERROR_OUT_OF_MEMORY;
|
||
|
|
||
|
nsresult rv = lock->Init(aPath, aTempPath, aUnlocker);
|
||
|
if (NS_FAILED(rv)) return rv;
|
||
|
|
||
|
NS_ADDREF(*aResult = lock);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
static const char kTable[] =
|
||
|
{ 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n',
|
||
|
'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z',
|
||
|
'1', '2', '3', '4', '5', '6', '7', '8', '9', '0' };
|
||
|
|
||
|
static void SaltProfileName(nsACString& aName)
|
||
|
{
|
||
|
double fpTime;
|
||
|
LL_L2D(fpTime, PR_Now());
|
||
|
|
||
|
// use 1e-6, granularity of PR_Now() on the mac is seconds
|
||
|
srand((uint)(fpTime * 1e-6 + 0.5));
|
||
|
|
||
|
char salt[9];
|
||
|
|
||
|
int i;
|
||
|
for (i = 0; i < 8; ++i)
|
||
|
salt[i] = kTable[rand() % NS_ARRAY_LENGTH(kTable)];
|
||
|
|
||
|
salt[8] = '.';
|
||
|
|
||
|
aName.Insert(salt, 0, 9);
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::CreateProfile(nsILocalFile* aRootDir,
|
||
|
nsILocalFile* aLocalDir,
|
||
|
const nsACString& aName,
|
||
|
nsIToolkitProfile* *aResult)
|
||
|
{
|
||
|
nsresult rv;
|
||
|
|
||
|
rv = GetProfileByName(aName, aResult);
|
||
|
if (NS_SUCCEEDED(rv)) return rv;
|
||
|
|
||
|
nsCOMPtr<nsILocalFile> rootDir (aRootDir);
|
||
|
|
||
|
nsCAutoString dirName;
|
||
|
if (!rootDir) {
|
||
|
nsCOMPtr<nsIFile> file;
|
||
|
PRBool dummy;
|
||
|
rv = gDirServiceProvider->GetFile(NS_APP_USER_PROFILES_ROOT_DIR, &dummy,
|
||
|
getter_AddRefs(file));
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
rootDir = do_QueryInterface(file);
|
||
|
NS_ENSURE_TRUE(rootDir, NS_ERROR_UNEXPECTED);
|
||
|
|
||
|
dirName = aName;
|
||
|
SaltProfileName(dirName);
|
||
|
|
||
|
if (NS_IsNativeUTF8()) {
|
||
|
rootDir->AppendNative(dirName);
|
||
|
} else {
|
||
|
rootDir->Append(NS_ConvertUTF8toUTF16(dirName));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
nsCOMPtr<nsILocalFile> localDir (aLocalDir);
|
||
|
|
||
|
if (!localDir) {
|
||
|
if (aRootDir) {
|
||
|
localDir = aRootDir;
|
||
|
}
|
||
|
else {
|
||
|
nsCOMPtr<nsIFile> file;
|
||
|
PRBool dummy;
|
||
|
rv = gDirServiceProvider->GetFile(NS_APP_USER_PROFILES_LOCAL_ROOT_DIR,
|
||
|
&dummy, getter_AddRefs(file));
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
localDir = do_QueryInterface(file);
|
||
|
NS_ENSURE_TRUE(localDir, NS_ERROR_UNEXPECTED);
|
||
|
|
||
|
// use same salting
|
||
|
if (NS_IsNativeUTF8()) {
|
||
|
localDir->AppendNative(dirName);
|
||
|
} else {
|
||
|
localDir->Append(NS_ConvertUTF8toUTF16(dirName));
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
PRBool exists;
|
||
|
rv = rootDir->Exists(&exists);
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
if (exists) {
|
||
|
rv = rootDir->IsDirectory(&exists);
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
if (!exists)
|
||
|
return NS_ERROR_FILE_NOT_DIRECTORY;
|
||
|
}
|
||
|
else {
|
||
|
nsCOMPtr<nsIFile> profileDefaultsDir;
|
||
|
nsCOMPtr<nsIFile> profileDirParent;
|
||
|
nsAutoString profileDirName;
|
||
|
|
||
|
rv = rootDir->GetParent(getter_AddRefs(profileDirParent));
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
rv = rootDir->GetLeafName(profileDirName);
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
PRBool dummy;
|
||
|
rv = gDirServiceProvider->GetFile(NS_APP_PROFILE_DEFAULTS_50_DIR, &dummy,
|
||
|
getter_AddRefs(profileDefaultsDir));
|
||
|
|
||
|
if (NS_SUCCEEDED(rv))
|
||
|
rv = profileDefaultsDir->CopyTo(profileDirParent,
|
||
|
profileDirName);
|
||
|
if (NS_FAILED(rv)) {
|
||
|
// if copying failed, lets just ensure that the profile directory exists.
|
||
|
rv = rootDir->Create(nsIFile::DIRECTORY_TYPE, 0700);
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
}
|
||
|
rv = rootDir->SetPermissions(0700);
|
||
|
#ifndef ANDROID
|
||
|
// If the profile is on the sdcard, this will fail but its non-fatal
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
#endif
|
||
|
}
|
||
|
|
||
|
rv = localDir->Exists(&exists);
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
if (!exists) {
|
||
|
rv = localDir->Create(nsIFile::DIRECTORY_TYPE, 0700);
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
}
|
||
|
|
||
|
nsToolkitProfile* last = mFirst;
|
||
|
if (last) {
|
||
|
while (last->mNext)
|
||
|
last = last->mNext;
|
||
|
}
|
||
|
|
||
|
nsCOMPtr<nsIToolkitProfile> profile =
|
||
|
new nsToolkitProfile(aName, rootDir, localDir, last);
|
||
|
if (!profile) return NS_ERROR_OUT_OF_MEMORY;
|
||
|
|
||
|
NS_ADDREF(*aResult = profile);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::GetProfileCount(PRUint32 *aResult)
|
||
|
{
|
||
|
if (!mFirst)
|
||
|
*aResult = 0;
|
||
|
else if (! mFirst->mNext)
|
||
|
*aResult = 1;
|
||
|
else
|
||
|
*aResult = 2;
|
||
|
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileService::Flush()
|
||
|
{
|
||
|
// Errors during writing might cause unhappy semi-written files.
|
||
|
// To avoid this, write the entire thing to a buffer, then write
|
||
|
// that buffer to disk.
|
||
|
|
||
|
nsresult rv;
|
||
|
PRUint32 pCount = 0;
|
||
|
nsToolkitProfile *cur;
|
||
|
|
||
|
for (cur = mFirst; cur != nsnull; cur = cur->mNext)
|
||
|
++pCount;
|
||
|
|
||
|
PRUint32 length;
|
||
|
nsAutoArrayPtr<char> buffer (new char[100+MAXPATHLEN*pCount]);
|
||
|
|
||
|
NS_ENSURE_TRUE(buffer, NS_ERROR_OUT_OF_MEMORY);
|
||
|
|
||
|
char *end = buffer;
|
||
|
|
||
|
end += sprintf(end,
|
||
|
"[General]\n"
|
||
|
"StartWithLastProfile=%s\n\n",
|
||
|
mStartWithLast ? "1" : "0");
|
||
|
|
||
|
nsCAutoString path;
|
||
|
cur = mFirst;
|
||
|
pCount = 0;
|
||
|
|
||
|
while (cur) {
|
||
|
// if the profile dir is relative to appdir...
|
||
|
PRBool isRelative;
|
||
|
rv = mAppData->Contains(cur->mRootDir, PR_TRUE, &isRelative);
|
||
|
if (NS_SUCCEEDED(rv) && isRelative) {
|
||
|
// we use a relative descriptor
|
||
|
rv = cur->mRootDir->GetRelativeDescriptor(mAppData, path);
|
||
|
} else {
|
||
|
// otherwise, a persistent descriptor
|
||
|
rv = cur->mRootDir->GetPersistentDescriptor(path);
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
}
|
||
|
|
||
|
end += sprintf(end,
|
||
|
"[Profile%u]\n"
|
||
|
"Name=%s\n"
|
||
|
"IsRelative=%s\n"
|
||
|
"Path=%s\n",
|
||
|
pCount, cur->mName.get(),
|
||
|
isRelative ? "1" : "0", path.get());
|
||
|
|
||
|
if (mChosen == cur) {
|
||
|
end += sprintf(end, "Default=1\n");
|
||
|
}
|
||
|
|
||
|
end += sprintf(end, "\n");
|
||
|
|
||
|
cur = cur->mNext;
|
||
|
++pCount;
|
||
|
}
|
||
|
|
||
|
FILE* writeFile;
|
||
|
rv = mListFile->OpenANSIFileDesc("w", &writeFile);
|
||
|
NS_ENSURE_SUCCESS(rv, rv);
|
||
|
|
||
|
if (buffer) {
|
||
|
length = end - buffer;
|
||
|
|
||
|
if (fwrite(buffer, sizeof(char), length, writeFile) != length) {
|
||
|
fclose(writeFile);
|
||
|
return NS_ERROR_UNEXPECTED;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
fclose(writeFile);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
NS_IMPL_ISUPPORTS1(nsToolkitProfileFactory, nsIFactory)
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileFactory::CreateInstance(nsISupports* aOuter, const nsID& aIID,
|
||
|
void** aResult)
|
||
|
{
|
||
|
if (aOuter)
|
||
|
return NS_ERROR_NO_AGGREGATION;
|
||
|
|
||
|
nsCOMPtr<nsIToolkitProfileService> profileService =
|
||
|
nsToolkitProfileService::gService;
|
||
|
if (!profileService) {
|
||
|
nsresult rv = NS_NewToolkitProfileService(getter_AddRefs(profileService));
|
||
|
if (NS_FAILED(rv))
|
||
|
return rv;
|
||
|
}
|
||
|
return profileService->QueryInterface(aIID, aResult);
|
||
|
}
|
||
|
|
||
|
NS_IMETHODIMP
|
||
|
nsToolkitProfileFactory::LockFactory(PRBool aVal)
|
||
|
{
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
nsresult
|
||
|
NS_NewToolkitProfileFactory(nsIFactory* *aResult)
|
||
|
{
|
||
|
*aResult = new nsToolkitProfileFactory();
|
||
|
if (!*aResult)
|
||
|
return NS_ERROR_OUT_OF_MEMORY;
|
||
|
|
||
|
NS_ADDREF(*aResult);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
nsresult
|
||
|
NS_NewToolkitProfileService(nsIToolkitProfileService* *aResult)
|
||
|
{
|
||
|
nsToolkitProfileService* profileService = new nsToolkitProfileService();
|
||
|
if (!profileService)
|
||
|
return NS_ERROR_OUT_OF_MEMORY;
|
||
|
nsresult rv = profileService->Init();
|
||
|
if (NS_FAILED(rv)) {
|
||
|
NS_ERROR("nsToolkitProfileService::Init failed!");
|
||
|
delete profileService;
|
||
|
return rv;
|
||
|
}
|
||
|
|
||
|
NS_ADDREF(*aResult = profileService);
|
||
|
return NS_OK;
|
||
|
}
|
||
|
|
||
|
nsresult
|
||
|
XRE_GetFileFromPath(const char *aPath, nsILocalFile* *aResult)
|
||
|
{
|
||
|
#if defined(XP_MACOSX)
|
||
|
PRInt32 pathLen = strlen(aPath);
|
||
|
if (pathLen > MAXPATHLEN)
|
||
|
return NS_ERROR_INVALID_ARG;
|
||
|
|
||
|
CFURLRef fullPath =
|
||
|
CFURLCreateFromFileSystemRepresentation(NULL, (const UInt8 *) aPath,
|
||
|
pathLen, true);
|
||
|
if (!fullPath)
|
||
|
return NS_ERROR_FAILURE;
|
||
|
|
||
|
nsCOMPtr<nsILocalFile> lf;
|
||
|
nsresult rv = NS_NewNativeLocalFile(EmptyCString(), PR_TRUE,
|
||
|
getter_AddRefs(lf));
|
||
|
if (NS_SUCCEEDED(rv)) {
|
||
|
nsCOMPtr<nsILocalFileMac> lfMac = do_QueryInterface(lf, &rv);
|
||
|
if (NS_SUCCEEDED(rv)) {
|
||
|
rv = lfMac->InitWithCFURL(fullPath);
|
||
|
if (NS_SUCCEEDED(rv))
|
||
|
NS_ADDREF(*aResult = lf);
|
||
|
}
|
||
|
}
|
||
|
CFRelease(fullPath);
|
||
|
return rv;
|
||
|
|
||
|
#elif defined(XP_UNIX)
|
||
|
char fullPath[MAXPATHLEN];
|
||
|
|
||
|
if (!realpath(aPath, fullPath))
|
||
|
return NS_ERROR_FAILURE;
|
||
|
|
||
|
return NS_NewNativeLocalFile(nsDependentCString(fullPath), PR_TRUE,
|
||
|
aResult);
|
||
|
#elif defined(XP_OS2)
|
||
|
char fullPath[MAXPATHLEN];
|
||
|
|
||
|
if (!realpath(aPath, fullPath))
|
||
|
return NS_ERROR_FAILURE;
|
||
|
|
||
|
// realpath on OS/2 returns a unix-ized path, so re-native-ize
|
||
|
for (char* ptr = strchr(fullPath, '/'); ptr; ptr = strchr(ptr, '/'))
|
||
|
*ptr = '\\';
|
||
|
|
||
|
return NS_NewNativeLocalFile(nsDependentCString(fullPath), PR_TRUE,
|
||
|
aResult);
|
||
|
|
||
|
#elif defined(XP_WIN)
|
||
|
WCHAR fullPath[MAXPATHLEN];
|
||
|
|
||
|
if (!_wfullpath(fullPath, NS_ConvertUTF8toUTF16(aPath).get(), MAXPATHLEN))
|
||
|
return NS_ERROR_FAILURE;
|
||
|
|
||
|
return NS_NewLocalFile(nsDependentString(fullPath), PR_TRUE,
|
||
|
aResult);
|
||
|
|
||
|
#else
|
||
|
#error Platform-specific logic needed here.
|
||
|
#endif
|
||
|
}
|