/* -*- Mode: C++; tab-width: 4; 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 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 of 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 ***** * * This Original Code has been modified by IBM Corporation. * Modifications made by IBM described herein are * Copyright (c) International Business Machines * Corporation, 2000 * * Modifications to Mozilla code or documentation * identified per MPL Section 3.3 * * Date Modified by Description of modification * 04/20/2000 IBM Corp. Added PR_CALLBACK for Optlink use in OS2 */ #include #include "nscore.h" #include "nsISupports.h" #include "nspr.h" #include "nsCRT.h" // for atoll // Arena used by component manager for storing contractid string, dll // location strings and small objects // CAUTION: Arena align mask needs to be defined before including plarena.h // currently from nsComponentManager.h #define PL_ARENA_CONST_ALIGN_MASK 7 #define NS_CM_BLOCK_SIZE (1024 * 8) #include "nsAutoLock.h" #include "nsCOMPtr.h" #include "nsComponentManager.h" #include "nsDirectoryService.h" #include "nsDirectoryServiceDefs.h" #include "nsCategoryManager.h" #include "nsCategoryManagerUtils.h" #include "nsIEnumerator.h" #include "xptiprivate.h" #include "nsIConsoleService.h" #include "nsIModule.h" #include "nsIObserverService.h" #include "nsISimpleEnumerator.h" #include "nsIStringEnumerator.h" #include "nsXPCOM.h" #include "nsXPCOMPrivate.h" #include "nsISupportsPrimitives.h" #include "nsIClassInfo.h" #include "nsLocalFile.h" #include "nsReadableUtils.h" #include "nsString.h" #include "nsXPIDLString.h" #include "prcmon.h" #include "xptinfo.h" // this after nsISupports, to pick up IID so that xpt stuff doesn't try to define it itself... #include "nsThreadUtils.h" #include "prthread.h" #include "private/pprthred.h" #include "nsTArray.h" #include "prio.h" #include "nsInt64.h" #include "nsManifestLineReader.h" #include NEW_H // for placement new #ifdef XP_BEOS #include #include #endif #include "prlog.h" NS_COM PRLogModuleInfo* nsComponentManagerLog = nsnull; #if 0 || defined (DEBUG_timeless) #define SHOW_DENIED_ON_SHUTDOWN #define SHOW_CI_ON_EXISTING_SERVICE #endif // Bloated registry buffer size to improve startup performance -- needs to // be big enough to fit the entire file into memory or it'll thrash. // 512K is big enough to allow for some future growth in the registry. #define BIG_REGISTRY_BUFLEN (512*1024) // Common Key Names const char classIDKeyName[]="classID"; const char classesKeyName[]="contractID"; const char componentsKeyName[]="components"; const char xpcomComponentsKeyName[]="software/mozilla/XPCOM/components"; const char xpcomKeyName[]="software/mozilla/XPCOM"; // Common Value Names const char classIDValueName[]="ClassID"; const char classNameValueName[]="ClassName"; const char componentCountValueName[]="ComponentsCount"; const char componentTypeValueName[]="ComponentType"; const char contractIDValueName[]="ContractID"; const char fileSizeValueName[]="FileSize"; const char inprocServerValueName[]="InprocServer"; const char lastModValueName[]="LastModTimeStamp"; const char nativeComponentType[]="application/x-mozilla-native"; const char staticComponentType[]="application/x-mozilla-static"; const char versionValueName[]="VersionString"; const static char XPCOM_ABSCOMPONENT_PREFIX[] = "abs:"; const static char XPCOM_RELCOMPONENT_PREFIX[] = "rel:"; const static char XPCOM_GRECOMPONENT_PREFIX[] = "gre:"; static const char gIDFormat[] = "{%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}"; #define NS_EMPTY_IID \ { \ 0x00000000, \ 0x0000, \ 0x0000, \ {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00} \ } NS_DEFINE_CID(kEmptyCID, NS_EMPTY_IID); NS_DEFINE_CID(kCategoryManagerCID, NS_CATEGORYMANAGER_CID); #define UID_STRING_LENGTH 39 // Set to true from NS_ShutdownXPCOM. extern PRBool gXPCOMShuttingDown; static void GetIDString(const nsID& aCID, char buf[UID_STRING_LENGTH]) { PR_snprintf(buf, UID_STRING_LENGTH, gIDFormat, aCID.m0, (PRUint32) aCID.m1, (PRUint32) aCID.m2, (PRUint32) aCID.m3[0], (PRUint32) aCID.m3[1], (PRUint32) aCID.m3[2], (PRUint32) aCID.m3[3], (PRUint32) aCID.m3[4], (PRUint32) aCID.m3[5], (PRUint32) aCID.m3[6], (PRUint32) aCID.m3[7]); } nsresult nsGetServiceFromCategory::operator()(const nsIID& aIID, void** aInstancePtr) const { nsresult rv; nsXPIDLCString value; nsCOMPtr catman; nsComponentManagerImpl *compMgr = nsComponentManagerImpl::gComponentManager; if (!compMgr) { rv = NS_ERROR_NOT_INITIALIZED; goto error; } if (!mCategory || !mEntry) { // when categories have defaults, use that for null mEntry rv = NS_ERROR_NULL_POINTER; goto error; } rv = compMgr->nsComponentManagerImpl::GetService(kCategoryManagerCID, NS_GET_IID(nsICategoryManager), getter_AddRefs(catman)); if (NS_FAILED(rv)) goto error; /* find the contractID for category.entry */ rv = catman->GetCategoryEntry(mCategory, mEntry, getter_Copies(value)); if (NS_FAILED(rv)) goto error; if (!value) { rv = NS_ERROR_SERVICE_NOT_AVAILABLE; goto error; } rv = compMgr-> nsComponentManagerImpl::GetServiceByContractID(value, aIID, aInstancePtr); if (NS_FAILED(rv)) { error: *aInstancePtr = 0; } if (mErrorPtr) *mErrorPtr = rv; return rv; } //////////////////////////////////////////////////////////////////////////////// // Arena helper functions //////////////////////////////////////////////////////////////////////////////// char * ArenaStrndup(const char *s, PRUint32 len, PLArenaPool *arena) { void *mem; // Include trailing null in the len PL_ARENA_ALLOCATE(mem, arena, len+1); if (mem) memcpy(mem, s, len+1); return static_cast(mem); } char* ArenaStrdup(const char *s, PLArenaPool *arena) { return ArenaStrndup(s, strlen(s), arena); } //////////////////////////////////////////////////////////////////////////////// // Hashtable Callbacks //////////////////////////////////////////////////////////////////////////////// PRBool nsFactoryEntry_Destroy(nsHashKey *aKey, void *aData, void* closure); static PLDHashNumber factory_HashKey(PLDHashTable *aTable, const void *aKey) { const nsCID *cidp = reinterpret_cast(aKey); return cidp->m0; } static PRBool factory_MatchEntry(PLDHashTable *aTable, const PLDHashEntryHdr *aHdr, const void *aKey) { const nsFactoryTableEntry* entry = static_cast(aHdr); const nsCID *cidp = reinterpret_cast(aKey); return (entry->mFactoryEntry->mCid).Equals(*cidp); } static void factory_ClearEntry(PLDHashTable *aTable, PLDHashEntryHdr *aHdr) { nsFactoryTableEntry* entry = static_cast(aHdr); // nsFactoryEntry is arena allocated. So we don't delete it. // We call the destructor by hand. entry->mFactoryEntry->~nsFactoryEntry(); PL_DHashClearEntryStub(aTable, aHdr); } static const PLDHashTableOps factory_DHashTableOps = { PL_DHashAllocTable, PL_DHashFreeTable, factory_HashKey, factory_MatchEntry, PL_DHashMoveEntryStub, factory_ClearEntry, PL_DHashFinalizeStub, }; static void contractID_ClearEntry(PLDHashTable *aTable, PLDHashEntryHdr *aHdr) { nsContractIDTableEntry* entry = static_cast(aHdr); if (entry->mFactoryEntry->mLoaderType == NS_LOADER_TYPE_INVALID && entry->mFactoryEntry->mCid.Equals(kEmptyCID)) { // this object is owned by the hash. // nsFactoryEntry is arena allocated. So we don't delete it. // We call the destructor by hand. entry->mFactoryEntry->~nsFactoryEntry(); } // contractIDs are arena allocated. No need to free them. PL_DHashClearEntryStub(aTable, aHdr); } static const PLDHashTableOps contractID_DHashTableOps = { PL_DHashAllocTable, PL_DHashFreeTable, PL_DHashStringKey, PL_DHashMatchStringKey, PL_DHashMoveEntryStub, contractID_ClearEntry, PL_DHashFinalizeStub, }; //////////////////////////////////////////////////////////////////////////////// // Hashtable Enumeration //////////////////////////////////////////////////////////////////////////////// typedef NS_CALLBACK(EnumeratorConverter)(PLDHashTable *table, const PLDHashEntryHdr *hdr, void *data, nsISupports **retval); class PLDHashTableEnumeratorImpl : public nsIBidirectionalEnumerator, public nsISimpleEnumerator { public: NS_DECL_ISUPPORTS NS_DECL_NSIENUMERATOR NS_DECL_NSIBIDIRECTIONALENUMERATOR NS_DECL_NSISIMPLEENUMERATOR PLDHashTableEnumeratorImpl(PLDHashTable *table, EnumeratorConverter converter, void *converterData); PRInt32 Count() { return mCount; } private: PLDHashTableEnumeratorImpl(); /* no implementation */ ~PLDHashTableEnumeratorImpl(); void ReleaseElements(); nsTArray mElements; PRInt32 mCount, mCurrent; PRMonitor* mMonitor; struct Closure { PRBool succeeded; EnumeratorConverter converter; void *data; PLDHashTableEnumeratorImpl *impl; }; static PLDHashOperator Enumerator(PLDHashTable *table, PLDHashEntryHdr *hdr, PRUint32 number, void *data); }; // static PLDHashOperator PLDHashTableEnumeratorImpl::Enumerator(PLDHashTable *table, PLDHashEntryHdr *hdr, PRUint32 number, void *data) { Closure *c = reinterpret_cast(data); nsISupports *converted; if (NS_FAILED(c->converter(table, hdr, c->data, &converted)) || !c->impl->mElements.AppendElement(converted)) { c->succeeded = PR_FALSE; return PL_DHASH_STOP; } c->succeeded = PR_TRUE; return PL_DHASH_NEXT; } PLDHashTableEnumeratorImpl::PLDHashTableEnumeratorImpl(PLDHashTable *table, EnumeratorConverter converter, void *converterData) : mCurrent(0) { mMonitor = nsAutoMonitor::NewMonitor("PLDHashTableEnumeratorImpl"); NS_ASSERTION(mMonitor, "NULL Monitor"); nsAutoMonitor mon(mMonitor); Closure c = { PR_FALSE, converter, converterData, this }; mCount = PL_DHashTableEnumerate(table, Enumerator, &c); if (!c.succeeded) { ReleaseElements(); mCount = 0; } } NS_IMPL_ISUPPORTS3(PLDHashTableEnumeratorImpl, nsIBidirectionalEnumerator, nsIEnumerator, nsISimpleEnumerator) PLDHashTableEnumeratorImpl::~PLDHashTableEnumeratorImpl() { ReleaseElements(); // Destroy the Lock if (mMonitor) nsAutoMonitor::DestroyMonitor(mMonitor); } void PLDHashTableEnumeratorImpl::ReleaseElements() { for (PRInt32 i = 0; i < mCount; i++) { NS_IF_RELEASE(mElements[i]); } } NS_IMETHODIMP PL_NewDHashTableEnumerator(PLDHashTable *table, EnumeratorConverter converter, void *converterData, PLDHashTableEnumeratorImpl **retval) { PLDHashTableEnumeratorImpl *impl = new PLDHashTableEnumeratorImpl(table, converter, converterData); if (!impl) return NS_ERROR_OUT_OF_MEMORY; NS_ADDREF(impl); if (impl->Count() == -1) { // conversion failed NS_RELEASE(impl); return NS_ERROR_FAILURE; } *retval = impl; return NS_OK; } NS_IMETHODIMP PLDHashTableEnumeratorImpl::First() { if (!mCount) return NS_ERROR_FAILURE; mCurrent = 0; return NS_OK; } NS_IMETHODIMP PLDHashTableEnumeratorImpl::Last() { if (!mCount) return NS_ERROR_FAILURE; mCurrent = mCount - 1; return NS_OK; } NS_IMETHODIMP PLDHashTableEnumeratorImpl::Prev() { if (!mCurrent) return NS_ERROR_FAILURE; mCurrent--; return NS_OK; } NS_IMETHODIMP PLDHashTableEnumeratorImpl::Next() { // If empty or we're past the end, or we are at the end return error if (!mCount || (mCurrent == mCount) || (++mCurrent == mCount)) return NS_ERROR_FAILURE; return NS_OK; } NS_IMETHODIMP PLDHashTableEnumeratorImpl::CurrentItem(nsISupports **retval) { if (!mCount || mCurrent == mCount) return NS_ERROR_FAILURE; *retval = mElements[mCurrent]; if (*retval) NS_ADDREF(*retval); return NS_OK; } NS_IMETHODIMP PLDHashTableEnumeratorImpl::IsDone() { if (!mCount || (mCurrent == mCount)) return NS_OK; return NS_ENUMERATOR_FALSE; } NS_IMETHODIMP PLDHashTableEnumeratorImpl::HasMoreElements(PRBool *_retval) { if (!mCount || (mCurrent >= mCount - 1)) *_retval = PR_FALSE; else *_retval = PR_TRUE; return NS_OK; } NS_IMETHODIMP PLDHashTableEnumeratorImpl::GetNext(nsISupports **_retval) { nsresult rv = Next(); if (NS_FAILED(rv)) return rv; return CurrentItem(_retval); } static NS_IMETHODIMP ConvertFactoryEntryToCID(PLDHashTable *table, const PLDHashEntryHdr *hdr, void *data, nsISupports **retval) { nsresult rv; nsCOMPtr wrapper; nsComponentManagerImpl *cm = static_cast(data); rv = cm->CreateInstanceByContractID(NS_SUPPORTS_ID_CONTRACTID, nsnull, NS_GET_IID(nsISupportsID), getter_AddRefs(wrapper)); NS_ENSURE_SUCCESS(rv, rv); const nsFactoryTableEntry *entry = reinterpret_cast(hdr); if (entry) { nsFactoryEntry *fe = entry->mFactoryEntry; wrapper->SetData(&fe->mCid); *retval = wrapper; NS_ADDREF(*retval); return NS_OK; } *retval = nsnull; return rv; } static NS_IMETHODIMP ConvertContractIDKeyToString(PLDHashTable *table, const PLDHashEntryHdr *hdr, void *data, nsISupports **retval) { nsresult rv; nsCOMPtr wrapper; nsComponentManagerImpl *cm = static_cast(data); rv = cm->CreateInstanceByContractID(NS_SUPPORTS_CSTRING_CONTRACTID, nsnull, NS_GET_IID(nsISupportsCString), getter_AddRefs(wrapper)); NS_ENSURE_SUCCESS(rv, rv); const nsContractIDTableEntry *entry = reinterpret_cast(hdr); wrapper->SetData(nsDependentCString(entry->mContractID, entry->mContractIDLen)); *retval = wrapper; NS_ADDREF(*retval); return NS_OK; } // this is safe to call during InitXPCOM static nsresult GetLocationFromDirectoryService(const char* prop, nsIFile** aDirectory) { nsCOMPtr directoryService; nsDirectoryService::Create(nsnull, NS_GET_IID(nsIProperties), getter_AddRefs(directoryService)); if (!directoryService) return NS_ERROR_FAILURE; return directoryService->Get(prop, NS_GET_IID(nsIFile), (void**)aDirectory); } //////////////////////////////////////////////////////////////////////////////// // nsComponentManagerImpl //////////////////////////////////////////////////////////////////////////////// nsComponentManagerImpl::nsComponentManagerImpl() : mMon(NULL), mShuttingDown(NS_SHUTDOWN_NEVERHAPPENED), mLoaderData(4), mRegistryDirty(PR_FALSE) { mFactories.ops = nsnull; mContractIDs.ops = nsnull; } #define CONTRACTID_HASHTABLE_INITIAL_SIZE 2048 #define AUTOREGENTRY_HASHTABLE_INITIAL_SIZE 256 nsresult nsComponentManagerImpl::Init(nsStaticModuleInfo const *aStaticModules, PRUint32 aStaticModuleCount) { PR_ASSERT(mShuttingDown != NS_SHUTDOWN_INPROGRESS); if (mShuttingDown == NS_SHUTDOWN_INPROGRESS) return NS_ERROR_FAILURE; mShuttingDown = NS_SHUTDOWN_NEVERHAPPENED; if (nsComponentManagerLog == nsnull) { nsComponentManagerLog = PR_NewLogModule("nsComponentManager"); } // Initialize our arena PL_INIT_ARENA_POOL(&mArena, "ComponentManagerArena", NS_CM_BLOCK_SIZE); if (!mFactories.ops) { if (!PL_DHashTableInit(&mFactories, &factory_DHashTableOps, 0, sizeof(nsFactoryTableEntry), 1024)) { mFactories.ops = nsnull; return NS_ERROR_OUT_OF_MEMORY; } // Minimum alpha uses k=2 because nsFactoryTableEntry saves two // words compared to what a chained hash table requires. PL_DHashTableSetAlphaBounds(&mFactories, 0.875, PL_DHASH_MIN_ALPHA(&mFactories, 2)); } if (!mContractIDs.ops) { if (!PL_DHashTableInit(&mContractIDs, &contractID_DHashTableOps, 0, sizeof(nsContractIDTableEntry), CONTRACTID_HASHTABLE_INITIAL_SIZE)) { mContractIDs.ops = nsnull; return NS_ERROR_OUT_OF_MEMORY; } // Minimum alpha uses k=1 because nsContractIDTableEntry saves one // word compared to what a chained hash table requires. #if 0 PL_DHashTableSetAlphaBounds(&mContractIDs, 0.875, PL_DHASH_MIN_ALPHA(&mContractIDs, 1)); #endif } if (!mAutoRegEntries.Init(AUTOREGENTRY_HASHTABLE_INITIAL_SIZE)) return NS_ERROR_OUT_OF_MEMORY; if (mMon == nsnull) { mMon = nsAutoMonitor::NewMonitor("nsComponentManagerImpl"); if (mMon == nsnull) return NS_ERROR_OUT_OF_MEMORY; } GetLocationFromDirectoryService(NS_XPCOM_COMPONENT_DIR, getter_AddRefs(mComponentsDir)); if (!mComponentsDir) return NS_ERROR_OUT_OF_MEMORY; nsCAutoString componentDescriptor; nsresult rv = mComponentsDir->GetNativePath(componentDescriptor); if (NS_FAILED(rv)) return rv; mComponentsOffset = componentDescriptor.Length(); GetLocationFromDirectoryService(NS_GRE_COMPONENT_DIR, getter_AddRefs(mGREComponentsDir)); if (mGREComponentsDir) { nsresult rv = mGREComponentsDir->GetNativePath(componentDescriptor); if (NS_FAILED(rv)) { NS_WARNING("No GRE component manager"); return rv; } mGREComponentsOffset = componentDescriptor.Length(); } GetLocationFromDirectoryService(NS_XPCOM_COMPONENT_REGISTRY_FILE, getter_AddRefs(mRegistryFile)); if(!mRegistryFile) { NS_WARNING("No Component Registry file was found in the directory service"); return NS_ERROR_FAILURE; } PR_LOG(nsComponentManagerLog, PR_LOG_DEBUG, ("nsComponentManager: Initialized.")); rv = mNativeModuleLoader.Init(); if (NS_FAILED(rv)) return rv; rv = mStaticModuleLoader.Init(aStaticModules, aStaticModuleCount); if (NS_FAILED(rv)) return rv; return NS_OK; } nsresult nsComponentManagerImpl::Shutdown(void) { PR_ASSERT(mShuttingDown == NS_SHUTDOWN_NEVERHAPPENED); if (mShuttingDown != NS_SHUTDOWN_NEVERHAPPENED) return NS_ERROR_FAILURE; mShuttingDown = NS_SHUTDOWN_INPROGRESS; // Shutdown the component manager PR_LOG(nsComponentManagerLog, PR_LOG_DEBUG, ("nsComponentManager: Beginning Shutdown.")); // Write out our component data file. if (mRegistryDirty) { nsresult rv = WritePersistentRegistry(); if (NS_FAILED(rv)) { PR_LOG(nsComponentManagerLog, PR_LOG_ERROR, ("nsComponentManager: Could not write out persistent registry.")); #ifdef DEBUG printf("Could not write out persistent registry!\n"); #endif } } mAutoRegEntries.Clear(); // Release all cached factories if (mContractIDs.ops) { PL_DHashTableFinish(&mContractIDs); mContractIDs.ops = nsnull; } if (mFactories.ops) { PL_DHashTableFinish(&mFactories); mFactories.ops = nsnull; } mLoaderData.Clear(); // Free staticm modules mStaticModuleLoader.ReleaseModules(); // Unload libraries mNativeModuleLoader.UnloadLibraries(); // delete arena for strings and small objects PL_FinishArenaPool(&mArena); mComponentsDir = 0; mCategoryManager = 0; mShuttingDown = NS_SHUTDOWN_COMPLETE; PR_LOG(nsComponentManagerLog, PR_LOG_DEBUG, ("nsComponentManager: Shutdown complete.")); return NS_OK; } nsComponentManagerImpl::~nsComponentManagerImpl() { PR_LOG(nsComponentManagerLog, PR_LOG_DEBUG, ("nsComponentManager: Beginning destruction.")); if (mShuttingDown != NS_SHUTDOWN_COMPLETE) Shutdown(); if (mMon) { nsAutoMonitor::DestroyMonitor(mMon); } PR_LOG(nsComponentManagerLog, PR_LOG_DEBUG, ("nsComponentManager: Destroyed.")); } NS_IMPL_THREADSAFE_ISUPPORTS7(nsComponentManagerImpl, nsIComponentManager, nsIServiceManager, nsISupportsWeakReference, nsIInterfaceRequestor, nsIComponentRegistrar, nsIServiceManagerObsolete, nsIComponentManagerObsolete) nsresult nsComponentManagerImpl::GetInterface(const nsIID & uuid, void **result) { NS_WARNING("This isn't supported"); // fall through to QI as anything QIable is a superset of what can be // got via the GetInterface() return QueryInterface(uuid, result); } //////////////////////////////////////////////////////////////////////////////// // nsComponentManagerImpl: Platform methods //////////////////////////////////////////////////////////////////////////////// #define PERSISTENT_REGISTRY_VERSION_MINOR 5 #define PERSISTENT_REGISTRY_VERSION_MAJOR 0 static PRBool ReadSectionHeader(nsManifestLineReader& reader, const char *token) { while (1) { if (*reader.LinePtr() == '[') { char* p = reader.LinePtr() + (reader.LineLength() - 1); if (*p != ']') break; *p = 0; char* values[1]; int lengths[1]; if (2 != reader.ParseLine(values, lengths, 1)) break; // ignore the leading '[' if (0 != PL_strcmp(values[0]+1, token)) break; return PR_TRUE; } if (!reader.NextLine()) break; } return PR_FALSE; } nsresult nsComponentManagerImpl::ReadPersistentRegistry() { NS_ASSERTION(mComponentsDir, "nsComponentManager not initialized."); nsresult rv; // populate Category Manager. need to get this early so that we don't get // skipped by 'goto out' mCategoryManager = do_GetService(NS_CATEGORYMANAGER_CONTRACTID, &rv); if (NS_FAILED(rv)) return rv; nsAutoMonitor mon(mMon); nsManifestLineReader reader; PRFileDesc* fd = nsnull; // Set From Init if (!mRegistryFile) { return NS_ERROR_FILE_NOT_FOUND; } nsCOMPtr file; mRegistryFile->Clone(getter_AddRefs(file)); if (!file) return NS_ERROR_OUT_OF_MEMORY; nsCOMPtr localFile(do_QueryInterface(file)); rv = localFile->OpenNSPRFileDesc(PR_RDONLY, 0444, &fd); if (NS_FAILED(rv)) return rv; PRInt64 fileSize; rv = localFile->GetFileSize(&fileSize); if (NS_FAILED(rv)) { PR_Close(fd); return rv; } PRInt32 flen = nsInt64(fileSize); if (flen == 0) { PR_Close(fd); NS_WARNING("Persistent Registry Empty!"); return NS_OK; // ERROR CONDITION } char* registry = new char[flen+1]; if (!registry) goto out; if (flen > PR_Read(fd, registry, flen)) { rv = NS_ERROR_FAILURE; goto out; } registry[flen] = '\0'; reader.Init(registry, flen); if (ReadSectionHeader(reader, "HEADER")) goto out; if (!reader.NextLine()) goto out; char* values[6]; int lengths[6]; // VersionLiteral,major,minor if (3 != reader.ParseLine(values, lengths, 3)) goto out; // VersionLiteral if (!nsDependentCString(values[0], lengths[0]).EqualsLiteral("Version")) goto out; // major if (PERSISTENT_REGISTRY_VERSION_MAJOR != atoi(values[1])) goto out; // minor if (PERSISTENT_REGISTRY_VERSION_MINOR != atoi(values[2])) goto out; if (ReadSectionHeader(reader, "COMPONENTS")) goto out; while (1) { if (!reader.NextLine()) break; //name,last_modification_date[,optionaldata] int parts = reader.ParseLine(values, lengths, 3); if (2 != parts) break; PRInt64 a = nsCRT::atoll(values[1]); nsCOMPtr lf; rv = FileForRegistryLocation(nsDependentCString(values[0], lengths[0]), getter_AddRefs(lf)); if (NS_FAILED(rv)) continue; nsCOMPtr lfhash(do_QueryInterface(lf)); if (!lf) { NS_ERROR("nsLocalFile does not implement nsIHashable"); continue; } mAutoRegEntries.Put(lfhash, a); } if (ReadSectionHeader(reader, "CLASSIDS")) goto out; while (1) { if (!reader.NextLine()) break; // cid,contract_id,type,class_name,inproc_server if (5 != reader.ParseLine(values, lengths, 5)) break; nsCID aClass; if (!aClass.Parse(values[0])) continue; LoaderType loadertype = AddLoaderType(values[2]); if (loadertype == NS_LOADER_TYPE_INVALID) { NS_ERROR("Could not create LoaderType"); continue; } void *mem; PL_ARENA_ALLOCATE(mem, &mArena, sizeof(nsFactoryEntry)); if (!mem) { rv = NS_ERROR_OUT_OF_MEMORY; goto out; } nsFactoryEntry *entry = new (mem) nsFactoryEntry(aClass, loadertype, values[4]); if (!entry->mLocationKey) { rv = NS_ERROR_OUT_OF_MEMORY; goto out; } nsFactoryTableEntry* factoryTableEntry = static_cast (PL_DHashTableOperate(&mFactories, &aClass, PL_DHASH_ADD)); if (!factoryTableEntry) { rv = NS_ERROR_OUT_OF_MEMORY; goto out; } factoryTableEntry->mFactoryEntry = entry; } if (ReadSectionHeader(reader, "CONTRACTIDS")) goto out; while (1) { if (!reader.NextLine()) break; //contractID,cid if (2 != reader.ParseLine(values, lengths, 2)) break; nsCID aClass; if (!aClass.Parse(values[1])) continue; //need to find the location for this cid. nsFactoryEntry *cidEntry = GetFactoryEntry(aClass); if (!cidEntry || cidEntry->mLoaderType == NS_LOADER_TYPE_INVALID) continue; //what should we really do? nsContractIDTableEntry* contractIDTableEntry = static_cast (PL_DHashTableOperate(&mContractIDs, values[0], PL_DHASH_ADD)); if (!contractIDTableEntry) { continue; } if (!contractIDTableEntry->mContractID) { char *contractID = ArenaStrndup(values[0], lengths[0], &mArena); if (!contractID) { rv = NS_ERROR_OUT_OF_MEMORY; goto out; } contractIDTableEntry->mContractID = contractID; contractIDTableEntry->mContractIDLen = lengths[0]; } contractIDTableEntry->mFactoryEntry = cidEntry; } if (ReadSectionHeader(reader, "CATEGORIES")) goto out; mCategoryManager->SuppressNotifications(PR_TRUE); while (1) { if (!reader.NextLine()) break; //type,name,value if (3 != reader.ParseLine(values, lengths, 3)) break; mCategoryManager->AddCategoryEntry(values[0], values[1], values[2], PR_TRUE, PR_TRUE, 0); } mCategoryManager->SuppressNotifications(PR_FALSE); mRegistryDirty = PR_FALSE; out: if (fd) PR_Close(fd); if (registry) delete [] registry; return rv; } struct PersistentWriterArgs { PRFileDesc *mFD; nsTArray *mLoaderData; }; static PLDHashOperator ContractIDWriter(PLDHashTable *table, PLDHashEntryHdr *hdr, PRUint32 number, void *arg) { char *contractID = ((nsContractIDTableEntry*)hdr)->mContractID; nsFactoryEntry *factoryEntry = ((nsContractIDTableEntry*)hdr)->mFactoryEntry; // for now, we only save out the top most parent. while (factoryEntry->mParent) factoryEntry = factoryEntry->mParent; if (factoryEntry->mLoaderType == NS_LOADER_TYPE_INVALID) return PL_DHASH_NEXT; PRFileDesc* fd = ((PersistentWriterArgs*)arg)->mFD; char cidString[UID_STRING_LENGTH]; GetIDString(factoryEntry->mCid, cidString); PR_fprintf(fd, "%s,%s\n", contractID, cidString); // what if this fails? return PL_DHASH_NEXT; } static PLDHashOperator ClassIDWriter(PLDHashTable *table, PLDHashEntryHdr *hdr, PRUint32 number, void *arg) { nsFactoryEntry *factoryEntry = ((nsFactoryTableEntry*)hdr)->mFactoryEntry; PRFileDesc* fd = ((PersistentWriterArgs*)arg)->mFD; nsTArray *loaderData = ((PersistentWriterArgs*)arg)->mLoaderData; // for now, we only save out the top most parent. while (factoryEntry->mParent) factoryEntry = factoryEntry->mParent; if (factoryEntry->mLoaderType == NS_LOADER_TYPE_INVALID) { return PL_DHASH_NEXT; } char cidString[UID_STRING_LENGTH]; GetIDString(factoryEntry->mCid, cidString); char *contractID = nsnull, *className = nsnull; nsCOMPtr classInfo = do_QueryInterface(factoryEntry->mFactory); if (classInfo) { classInfo->GetContractID(&contractID); classInfo->GetClassDescription(&className); } const char *loaderName; switch (factoryEntry->mLoaderType) { case NS_LOADER_TYPE_STATIC: loaderName = staticComponentType; break; case NS_LOADER_TYPE_NATIVE: loaderName = nativeComponentType; break; default: loaderName = loaderData->ElementAt(factoryEntry->mLoaderType).type.get(); } const char* location = factoryEntry->mLocationKey; // cid,contract_id,type,class_name,inproc_server PR_fprintf(fd, "%s,%s,%s,%s,%s\n", cidString, (contractID ? contractID : ""), (loaderName ? loaderName : ""), (className ? className : ""), (location ? location : "")); if (contractID) NS_Free(contractID); if (className) NS_Free(className); return PL_DHASH_NEXT; } static PLDHashOperator AutoRegEntryWriter(nsIHashable *aKey, PRInt64 &aTimestamp, void* aClosure) { PRFileDesc* fd = (PRFileDesc*) aClosure; nsCOMPtr lf(do_QueryInterface(aKey)); nsCAutoString location; nsComponentManagerImpl::gComponentManager-> RegistryLocationForFile(lf, location); PR_fprintf(fd, "%s,%lld\n", location.get(), (PRInt64) aTimestamp); return PL_DHASH_NEXT; } nsresult nsComponentManagerImpl::WritePersistentRegistry() { if (!mRegistryFile) return NS_ERROR_FAILURE; // this should have been set by Init(). nsCOMPtr file; mRegistryFile->Clone(getter_AddRefs(file)); if (!file) return NS_ERROR_OUT_OF_MEMORY; nsCOMPtr localFile(do_QueryInterface(file)); nsCAutoString originalLeafName; localFile->GetNativeLeafName(originalLeafName); nsCAutoString leafName; leafName.Assign(originalLeafName + NS_LITERAL_CSTRING(".tmp")); localFile->SetNativeLeafName(leafName); PRFileDesc* fd = nsnull; // Owner and group can setup components, everyone else should be able to see but not poison them. nsresult rv = localFile->OpenNSPRFileDesc(PR_WRONLY | PR_CREATE_FILE | PR_TRUNCATE, 0664, &fd); if (NS_FAILED(rv)) return rv; if (PR_fprintf(fd, "Generated File. Do not edit.\n") == (PRUint32) -1) { rv = NS_ERROR_UNEXPECTED; goto out; } if (PR_fprintf(fd, "\n[HEADER]\nVersion,%d,%d\n", PERSISTENT_REGISTRY_VERSION_MAJOR, PERSISTENT_REGISTRY_VERSION_MINOR) == (PRUint32) -1) { rv = NS_ERROR_UNEXPECTED; goto out; } if (PR_fprintf(fd, "\n[COMPONENTS]\n") == (PRUint32) -1) { rv = NS_ERROR_UNEXPECTED; goto out; } mAutoRegEntries.Enumerate(AutoRegEntryWriter, (void*)fd); PersistentWriterArgs args; args.mFD = fd; args.mLoaderData = &mLoaderData; if (PR_fprintf(fd, "\n[CLASSIDS]\n") == (PRUint32) -1) { rv = NS_ERROR_UNEXPECTED; goto out; } PL_DHashTableEnumerate(&mFactories, ClassIDWriter, (void*)&args); if (PR_fprintf(fd, "\n[CONTRACTIDS]\n") == (PRUint32) -1) { rv = NS_ERROR_UNEXPECTED; goto out; } PL_DHashTableEnumerate(&mContractIDs, ContractIDWriter, (void*)&args); if (PR_fprintf(fd, "\n[CATEGORIES]\n") == (PRUint32) -1) { rv = NS_ERROR_UNEXPECTED; goto out; } NS_ASSERTION(mCategoryManager, "nsComponentManager used initialized"); rv = mCategoryManager->WriteCategoryManagerToRegistry(fd); out: PR_Close(fd); // don't create the file is there was a problem???? NS_ENSURE_SUCCESS(rv, rv); if (!mRegistryFile) return NS_ERROR_NOT_INITIALIZED; PRBool exists; if(NS_FAILED(mRegistryFile->Exists(&exists))) return PR_FALSE; if(exists && NS_FAILED(mRegistryFile->Remove(PR_FALSE))) return PR_FALSE; nsCOMPtr parent; mRegistryFile->GetParent(getter_AddRefs(parent)); rv = localFile->MoveToNative(parent, originalLeafName); mRegistryDirty = PR_FALSE; return rv; } //////////////////////////////////////////////////////////////////////////////// // Hash Functions //////////////////////////////////////////////////////////////////////////////// nsresult nsComponentManagerImpl::HashContractID(const char *aContractID, PRUint32 aContractIDLen, nsFactoryEntry *fe) { if(!aContractID || !aContractIDLen) return NS_ERROR_NULL_POINTER; NS_ABORT_IF_FALSE(PR_InMonitor(mMon), "called from outside mMon"); nsContractIDTableEntry* contractIDTableEntry = static_cast (PL_DHashTableOperate(&mContractIDs, aContractID, PL_DHASH_ADD)); if (!contractIDTableEntry) return NS_ERROR_OUT_OF_MEMORY; NS_ASSERTION(!contractIDTableEntry->mContractID || !strcmp(contractIDTableEntry->mContractID, aContractID), "contractid conflict"); if (!contractIDTableEntry->mContractID) { char *contractID = ArenaStrndup(aContractID, aContractIDLen, &mArena); if (!contractID) return NS_ERROR_OUT_OF_MEMORY; contractIDTableEntry->mContractID = contractID; contractIDTableEntry->mContractIDLen = aContractIDLen; } contractIDTableEntry->mFactoryEntry = fe; return NS_OK; } nsFactoryEntry * nsComponentManagerImpl::GetFactoryEntry(const char *aContractID, PRUint32 aContractIDLen) { nsFactoryEntry *fe = nsnull; { nsAutoMonitor mon(mMon); nsContractIDTableEntry* contractIDTableEntry = static_cast (PL_DHashTableOperate(&mContractIDs, aContractID, PL_DHASH_LOOKUP)); if (PL_DHASH_ENTRY_IS_BUSY(contractIDTableEntry)) { fe = contractIDTableEntry->mFactoryEntry; } } //exit monitor return fe; } nsFactoryEntry * nsComponentManagerImpl::GetFactoryEntry(const nsCID &aClass) { nsFactoryEntry *entry = nsnull; { nsAutoMonitor mon(mMon); nsFactoryTableEntry* factoryTableEntry = static_cast (PL_DHashTableOperate(&mFactories, &aClass, PL_DHASH_LOOKUP)); if (PL_DHASH_ENTRY_IS_BUSY(factoryTableEntry)) { entry = factoryTableEntry->mFactoryEntry; } } // exit monitor return entry; } /** * FindFactory() * * Given a classID, this finds the factory for this CID by first searching the * local CID<->factory mapping. Next it searches for a Dll that implements * this classID and calls LoadFactory() to create the factory. * * Again, no attempt is made at storing the factory. */ NS_IMETHODIMP nsComponentManagerImpl::FindFactory(const nsCID &aClass, nsIFactory **aFactory) { PR_ASSERT(aFactory != nsnull); nsFactoryEntry *entry = GetFactoryEntry(aClass); if (!entry) return NS_ERROR_FACTORY_NOT_REGISTERED; return entry->GetFactory(aFactory); } nsresult nsComponentManagerImpl::FindFactory(const char *contractID, PRUint32 aContractIDLen, nsIFactory **aFactory) { PR_ASSERT(aFactory != nsnull); nsFactoryEntry *entry = GetFactoryEntry(contractID, aContractIDLen); if (!entry) return NS_ERROR_FACTORY_NOT_REGISTERED; return entry->GetFactory(aFactory); } /** * GetClassObject() * * Given a classID, this finds the singleton ClassObject that implements the CID. * Returns an interface of type aIID off the singleton classobject. */ NS_IMETHODIMP nsComponentManagerImpl::GetClassObject(const nsCID &aClass, const nsIID &aIID, void **aResult) { nsresult rv; nsCOMPtr factory; #ifdef PR_LOGGING if (PR_LOG_TEST(nsComponentManagerLog, PR_LOG_DEBUG)) { char *buf = aClass.ToString(); PR_LogPrint("nsComponentManager: GetClassObject(%s)", buf); if (buf) NS_Free(buf); } #endif PR_ASSERT(aResult != nsnull); rv = FindFactory(aClass, getter_AddRefs(factory)); if (NS_FAILED(rv)) return rv; rv = factory->QueryInterface(aIID, aResult); PR_LOG(nsComponentManagerLog, PR_LOG_WARNING, ("\t\tGetClassObject() %s", NS_SUCCEEDED(rv) ? "succeeded" : "FAILED")); return rv; } NS_IMETHODIMP nsComponentManagerImpl::GetClassObjectByContractID(const char *contractID, const nsIID &aIID, void **aResult) { NS_ENSURE_ARG_POINTER(aResult); NS_ENSURE_ARG_POINTER(contractID); nsresult rv; nsCOMPtr factory; #ifdef PR_LOGGING if (PR_LOG_TEST(nsComponentManagerLog, PR_LOG_DEBUG)) { PR_LogPrint("nsComponentManager: GetClassObject(%s)", contractID); } #endif rv = FindFactory(contractID, strlen(contractID), getter_AddRefs(factory)); if (NS_FAILED(rv)) return rv; rv = factory->QueryInterface(aIID, aResult); PR_LOG(nsComponentManagerLog, PR_LOG_WARNING, ("\t\tGetClassObject() %s", NS_SUCCEEDED(rv) ? "succeeded" : "FAILED")); return rv; } /** * ContractIDToClassID() * * Mapping function from a ContractID to a classID. Directly talks to the registry. * */ NS_IMETHODIMP nsComponentManagerImpl::ContractIDToClassID(const char *aContractID, nsCID *aClass) { NS_ENSURE_ARG_POINTER(aContractID); NS_ENSURE_ARG_POINTER(aClass); nsresult rv = NS_ERROR_FACTORY_NOT_REGISTERED; nsFactoryEntry *fe = GetFactoryEntry(aContractID, strlen(aContractID)); if (fe) { *aClass = fe->mCid; rv = NS_OK; } #ifdef PR_LOGGING if (PR_LOG_TEST(nsComponentManagerLog, PR_LOG_WARNING)) { char *buf = 0; if (NS_SUCCEEDED(rv)) buf = aClass->ToString(); PR_LOG(nsComponentManagerLog, PR_LOG_WARNING, ("nsComponentManager: ContractIDToClassID(%s)->%s", aContractID, NS_SUCCEEDED(rv) ? buf : "[FAILED]")); if (buf) NS_Free(buf); } #endif return rv; } /** * CLSIDToContractID() * * Translates a classID to a {ContractID, Class Name}. Does direct registry * access to do the translation. * * NOTE: Since this isn't heavily used, we aren't caching this. */ NS_IMETHODIMP nsComponentManagerImpl::CLSIDToContractID(const nsCID &aClass, char* *aClassName, char* *aContractID) { NS_WARNING("Need to implement CLSIDToContractID"); nsresult rv = NS_ERROR_FACTORY_NOT_REGISTERED; #ifdef PR_LOGGING if (PR_LOG_TEST(nsComponentManagerLog, PR_LOG_WARNING)) { char *buf = aClass.ToString(); PR_LOG(nsComponentManagerLog, PR_LOG_WARNING, ("nsComponentManager: CLSIDToContractID(%s)->%s", buf, NS_SUCCEEDED(rv) ? *aContractID : "[FAILED]")); if (buf) NS_Free(buf); } #endif return rv; } /** * CreateInstance() * * Create an instance of an object that implements an interface and belongs * to the implementation aClass using the factory. The factory is immediately * released and not held onto for any longer. */ NS_IMETHODIMP nsComponentManagerImpl::CreateInstance(const nsCID &aClass, nsISupports *aDelegate, const nsIID &aIID, void **aResult) { // test this first, since there's no point in creating a component during // shutdown -- whether it's available or not would depend on the order it // occurs in the list if (gXPCOMShuttingDown) { // When processing shutdown, don't process new GetService() requests #ifdef SHOW_DENIED_ON_SHUTDOWN nsXPIDLCString cid, iid; cid.Adopt(aClass.ToString()); iid.Adopt(aIID.ToString()); fprintf(stderr, "Creating new instance on shutdown. Denied.\n" " CID: %s\n IID: %s\n", cid.get(), iid.get()); #endif /* SHOW_DENIED_ON_SHUTDOWN */ return NS_ERROR_UNEXPECTED; } if (aResult == nsnull) { return NS_ERROR_NULL_POINTER; } *aResult = nsnull; nsFactoryEntry *entry = GetFactoryEntry(aClass); if (!entry) return NS_ERROR_FACTORY_NOT_REGISTERED; #ifdef SHOW_CI_ON_EXISTING_SERVICE if (entry->mServiceObject) { nsXPIDLCString cid; cid.Adopt(aClass.ToString()); nsCAutoString message; message = NS_LITERAL_CSTRING("You are calling CreateInstance \"") + cid + NS_LITERAL_CSTRING("\" when a service for this CID already exists!"); NS_ERROR(message.get()); } #endif nsIFactory *factory = nsnull; nsresult rv = entry->GetFactory(&factory); if (NS_SUCCEEDED(rv)) { rv = factory->CreateInstance(aDelegate, aIID, aResult); if (NS_SUCCEEDED(rv) && !*aResult) { NS_ERROR("Factory did not return an object but returned success!"); rv = NS_ERROR_SERVICE_NOT_FOUND; } NS_RELEASE(factory); } else { // Translate error values rv = NS_ERROR_FACTORY_NOT_REGISTERED; } #ifdef PR_LOGGING if (PR_LOG_TEST(nsComponentManagerLog, PR_LOG_WARNING)) { char *buf = aClass.ToString(); PR_LOG(nsComponentManagerLog, PR_LOG_WARNING, ("nsComponentManager: CreateInstance(%s) %s", buf, NS_SUCCEEDED(rv) ? "succeeded" : "FAILED")); if (buf) NS_Free(buf); } #endif return rv; } /** * CreateInstanceByContractID() * * A variant of CreateInstance() that creates an instance of the object that * implements the interface aIID and whose implementation has a contractID aContractID. * * This is only a convenience routine that turns around can calls the * CreateInstance() with classid and iid. */ NS_IMETHODIMP nsComponentManagerImpl::CreateInstanceByContractID(const char *aContractID, nsISupports *aDelegate, const nsIID &aIID, void **aResult) { NS_ENSURE_ARG_POINTER(aContractID); // test this first, since there's no point in creating a component during // shutdown -- whether it's available or not would depend on the order it // occurs in the list if (gXPCOMShuttingDown) { // When processing shutdown, don't process new GetService() requests #ifdef SHOW_DENIED_ON_SHUTDOWN nsXPIDLCString iid; iid.Adopt(aIID.ToString()); fprintf(stderr, "Creating new instance on shutdown. Denied.\n" " ContractID: %s\n IID: %s\n", aContractID, iid.get()); #endif /* SHOW_DENIED_ON_SHUTDOWN */ return NS_ERROR_UNEXPECTED; } if (aResult == nsnull) { return NS_ERROR_NULL_POINTER; } *aResult = nsnull; nsFactoryEntry *entry = GetFactoryEntry(aContractID, strlen(aContractID)); if (!entry) return NS_ERROR_FACTORY_NOT_REGISTERED; #ifdef SHOW_CI_ON_EXISTING_SERVICE if (entry->mServiceObject) { nsCAutoString message; message = NS_LITERAL_CSTRING("You are calling CreateInstance \"") + nsDependentCString(aContractID) + NS_LITERAL_CSTRING("\" when a service for this CID already exists! " "Add it to abusedContracts to track down the service consumer."); NS_ERROR(message.get()); } #endif nsIFactory *factory = nsnull; nsresult rv = entry->GetFactory(&factory); if (NS_SUCCEEDED(rv)) { rv = factory->CreateInstance(aDelegate, aIID, aResult); if (NS_SUCCEEDED(rv) && !*aResult) { NS_ERROR("Factory did not return an object but returned success!"); rv = NS_ERROR_SERVICE_NOT_FOUND; } NS_RELEASE(factory); } else { // Translate error values rv = NS_ERROR_FACTORY_NOT_REGISTERED; } PR_LOG(nsComponentManagerLog, PR_LOG_WARNING, ("nsComponentManager: CreateInstanceByContractID(%s) %s", aContractID, NS_SUCCEEDED(rv) ? "succeeded" : "FAILED")); return rv; } // Service Manager Impl static PLDHashOperator FreeServiceFactoryEntryEnumerate(PLDHashTable *aTable, PLDHashEntryHdr *aHdr, PRUint32 aNumber, void *aData) { nsFactoryTableEntry* entry = static_cast(aHdr); if (!entry->mFactoryEntry) return PL_DHASH_NEXT; nsFactoryEntry* factoryEntry = entry->mFactoryEntry; factoryEntry->mServiceObject = nsnull; return PL_DHASH_NEXT; } static PLDHashOperator FreeServiceContractIDEntryEnumerate(PLDHashTable *aTable, PLDHashEntryHdr *aHdr, PRUint32 aNumber, void *aData) { nsContractIDTableEntry* entry = static_cast(aHdr); if (!entry->mFactoryEntry) return PL_DHASH_NEXT; nsFactoryEntry* factoryEntry = entry->mFactoryEntry; factoryEntry->mServiceObject = nsnull; return PL_DHASH_NEXT; } nsresult nsComponentManagerImpl::FreeServices() { NS_ASSERTION(gXPCOMShuttingDown, "Must be shutting down in order to free all services"); if (!gXPCOMShuttingDown) return NS_ERROR_FAILURE; if (mContractIDs.ops) { PL_DHashTableEnumerate(&mContractIDs, FreeServiceContractIDEntryEnumerate, nsnull); } if (mFactories.ops) { PL_DHashTableEnumerate(&mFactories, FreeServiceFactoryEntryEnumerate, nsnull); } return NS_OK; } // This should only ever be called within the monitor! nsComponentManagerImpl::PendingServiceInfo* nsComponentManagerImpl::AddPendingService(const nsCID& aServiceCID, PRThread* aThread) { PendingServiceInfo* newInfo = mPendingServices.AppendElement(); if (newInfo) { newInfo->cid = &aServiceCID; newInfo->thread = aThread; } return newInfo; } // This should only ever be called within the monitor! void nsComponentManagerImpl::RemovePendingService(const nsCID& aServiceCID) { PRUint32 pendingCount = mPendingServices.Length(); for (PRUint32 index = 0; index < pendingCount; ++index) { const PendingServiceInfo& info = mPendingServices.ElementAt(index); if (info.cid->Equals(aServiceCID)) { mPendingServices.RemoveElementAt(index); return; } } } // This should only ever be called within the monitor! PRThread* nsComponentManagerImpl::GetPendingServiceThread(const nsCID& aServiceCID) const { PRUint32 pendingCount = mPendingServices.Length(); for (PRUint32 index = 0; index < pendingCount; ++index) { const PendingServiceInfo& info = mPendingServices.ElementAt(index); if (info.cid->Equals(aServiceCID)) { return info.thread; } } return nsnull; } NS_IMETHODIMP nsComponentManagerImpl::GetService(const nsCID& aClass, const nsIID& aIID, void* *result) { // test this first, since there's no point in returning a service during // shutdown -- whether it's available or not would depend on the order it // occurs in the list if (gXPCOMShuttingDown) { // When processing shutdown, don't process new GetService() requests #ifdef SHOW_DENIED_ON_SHUTDOWN nsXPIDLCString cid, iid; cid.Adopt(aClass.ToString()); iid.Adopt(aIID.ToString()); fprintf(stderr, "Getting service on shutdown. Denied.\n" " CID: %s\n IID: %s\n", cid.get(), iid.get()); #endif /* SHOW_DENIED_ON_SHUTDOWN */ return NS_ERROR_UNEXPECTED; } nsAutoMonitor mon(mMon); nsFactoryEntry* entry = nsnull; nsFactoryTableEntry* factoryTableEntry = static_cast (PL_DHashTableOperate(&mFactories, &aClass, PL_DHASH_LOOKUP)); if (PL_DHASH_ENTRY_IS_BUSY(factoryTableEntry)) { entry = factoryTableEntry->mFactoryEntry; } if (entry && entry->mServiceObject) { nsCOMPtr supports = entry->mServiceObject; mon.Exit(); return supports->QueryInterface(aIID, result); } PRThread* currentPRThread = PR_GetCurrentThread(); NS_ASSERTION(currentPRThread, "This should never be null!"); // Needed to optimize the event loop below. nsIThread* currentThread = nsnull; PRThread* pendingPRThread; while ((pendingPRThread = GetPendingServiceThread(aClass))) { if (pendingPRThread == currentPRThread) { NS_ERROR("Recursive GetService!"); return NS_ERROR_NOT_AVAILABLE; } mon.Exit(); if (!currentThread) { currentThread = NS_GetCurrentThread(); NS_ASSERTION(currentThread, "This should never be null!"); } // This will process a single event or yield the thread if no event is // pending. if (!NS_ProcessNextEvent(currentThread, PR_FALSE)) { PR_Sleep(PR_INTERVAL_NO_WAIT); } mon.Enter(); } if (currentThread) { // If we have a currentThread then we must have waited on another thread // to create the service. Grab it now if that succeeded. if (!entry) { factoryTableEntry = static_cast (PL_DHashTableOperate(&mFactories, &aClass, PL_DHASH_LOOKUP)); if (PL_DHASH_ENTRY_IS_BUSY(factoryTableEntry)) { entry = factoryTableEntry->mFactoryEntry; } } // It's still possible that the other thread failed to create the // service so we're not guaranteed to have an entry or service yet. if (entry && entry->mServiceObject) { nsCOMPtr supports = entry->mServiceObject; mon.Exit(); return supports->QueryInterface(aIID, result); } } #ifdef DEBUG PendingServiceInfo* newInfo = #endif AddPendingService(aClass, currentPRThread); NS_ASSERTION(newInfo, "Failed to add info to the array!"); nsCOMPtr service; // We need to not be holding the service manager's monitor while calling // CreateInstance, because it invokes user code which could try to re-enter // the service manager: mon.Exit(); nsresult rv = CreateInstance(aClass, nsnull, aIID, getter_AddRefs(service)); mon.Enter(); #ifdef DEBUG pendingPRThread = GetPendingServiceThread(aClass); NS_ASSERTION(pendingPRThread == currentPRThread, "Pending service array has been changed!"); #endif RemovePendingService(aClass); if (NS_FAILED(rv)) return rv; if (!entry) { // second hash lookup for GetService nsFactoryTableEntry* factoryTableEntry = static_cast (PL_DHashTableOperate(&mFactories, &aClass, PL_DHASH_LOOKUP)); if (PL_DHASH_ENTRY_IS_BUSY(factoryTableEntry)) { entry = factoryTableEntry->mFactoryEntry; } NS_ASSERTION(entry, "we should have a factory entry since CI succeeded - we should not get here"); if (!entry) return NS_ERROR_FAILURE; } NS_ASSERTION(!entry->mServiceObject, "Created two instances of a service!"); entry->mServiceObject = service; *result = service.get(); if (!*result) { NS_ERROR("Factory did not return an object but returned success!"); return NS_ERROR_SERVICE_NOT_FOUND; } NS_ADDREF(static_cast((*result))); return rv; } NS_IMETHODIMP nsComponentManagerImpl::RegisterService(const nsCID& aClass, nsISupports* aService) { nsAutoMonitor mon(mMon); // check to see if we have a factory entry for the service nsFactoryEntry *entry = GetFactoryEntry(aClass); if (!entry) { void *mem; PL_ARENA_ALLOCATE(mem, &mArena, sizeof(nsFactoryEntry)); if (!mem) return NS_ERROR_OUT_OF_MEMORY; entry = new (mem) nsFactoryEntry(aClass, (nsIFactory*) nsnull); nsFactoryTableEntry* factoryTableEntry = static_cast (PL_DHashTableOperate(&mFactories, &aClass, PL_DHASH_ADD)); if (!factoryTableEntry) return NS_ERROR_OUT_OF_MEMORY; factoryTableEntry->mFactoryEntry = entry; } else { if (entry->mServiceObject) return NS_ERROR_FAILURE; } entry->mServiceObject = aService; return NS_OK; } NS_IMETHODIMP nsComponentManagerImpl::UnregisterService(const nsCID& aClass) { nsresult rv = NS_OK; nsFactoryEntry* entry = nsnull; nsAutoMonitor mon(mMon); nsFactoryTableEntry* factoryTableEntry = static_cast (PL_DHashTableOperate(&mFactories, &aClass, PL_DHASH_LOOKUP)); if (PL_DHASH_ENTRY_IS_BUSY(factoryTableEntry)) { entry = factoryTableEntry->mFactoryEntry; } if (!entry || !entry->mServiceObject) return NS_ERROR_SERVICE_NOT_AVAILABLE; entry->mServiceObject = nsnull; return rv; } NS_IMETHODIMP nsComponentManagerImpl::RegisterService(const char* aContractID, nsISupports* aService) { NS_ENSURE_ARG_POINTER(aContractID); nsAutoMonitor mon(mMon); // check to see if we have a factory entry for the service PRUint32 contractIDLen = strlen(aContractID); nsFactoryEntry *entry = GetFactoryEntry(aContractID, contractIDLen); if (!entry) { void *mem; PL_ARENA_ALLOCATE(mem, &mArena, sizeof(nsFactoryEntry)); if (!mem) return NS_ERROR_OUT_OF_MEMORY; entry = new (mem) nsFactoryEntry(kEmptyCID, (nsIFactory*) nsnull); nsContractIDTableEntry* contractIDTableEntry = static_cast (PL_DHashTableOperate(&mContractIDs, aContractID, PL_DHASH_ADD)); if (!contractIDTableEntry) { delete entry; return NS_ERROR_OUT_OF_MEMORY; } if (!contractIDTableEntry->mContractID) { char *contractID = ArenaStrndup(aContractID, contractIDLen, &mArena); if (!contractID) return NS_ERROR_OUT_OF_MEMORY; contractIDTableEntry->mContractID = contractID; contractIDTableEntry->mContractIDLen = contractIDLen; } contractIDTableEntry->mFactoryEntry = entry; } else { if (entry->mServiceObject) return NS_ERROR_FAILURE; } entry->mServiceObject = aService; return NS_OK; } NS_IMETHODIMP nsComponentManagerImpl::IsServiceInstantiated(const nsCID & aClass, const nsIID& aIID, PRBool *result) { // Now we want to get the service if we already got it. If not, we don't want // to create an instance of it. mmh! // test this first, since there's no point in returning a service during // shutdown -- whether it's available or not would depend on the order it // occurs in the list if (gXPCOMShuttingDown) { // When processing shutdown, don't process new GetService() requests #ifdef SHOW_DENIED_ON_SHUTDOWN nsXPIDLCString cid, iid; cid.Adopt(aClass.ToString()); iid.Adopt(aIID.ToString()); fprintf(stderr, "Checking for service on shutdown. Denied.\n" " CID: %s\n IID: %s\n", cid.get(), iid.get()); #endif /* SHOW_DENIED_ON_SHUTDOWN */ return NS_ERROR_UNEXPECTED; } nsresult rv = NS_ERROR_SERVICE_NOT_AVAILABLE; nsFactoryEntry* entry = nsnull; nsFactoryTableEntry* factoryTableEntry = static_cast (PL_DHashTableOperate(&mFactories, &aClass, PL_DHASH_LOOKUP)); if (PL_DHASH_ENTRY_IS_BUSY(factoryTableEntry)) { entry = factoryTableEntry->mFactoryEntry; } if (entry && entry->mServiceObject) { nsCOMPtr service; rv = entry->mServiceObject->QueryInterface(aIID, getter_AddRefs(service)); *result = (service!=nsnull); } return rv; } NS_IMETHODIMP nsComponentManagerImpl::IsServiceInstantiatedByContractID(const char *aContractID, const nsIID& aIID, PRBool *result) { // Now we want to get the service if we already got it. If not, we don't want // to create an instance of it. mmh! // test this first, since there's no point in returning a service during // shutdown -- whether it's available or not would depend on the order it // occurs in the list if (gXPCOMShuttingDown) { // When processing shutdown, don't process new GetService() requests #ifdef SHOW_DENIED_ON_SHUTDOWN nsXPIDLCString iid; iid.Adopt(aIID.ToString()); fprintf(stderr, "Checking for service on shutdown. Denied.\n" " ContractID: %s\n IID: %s\n", aContractID, iid.get()); #endif /* SHOW_DENIED_ON_SHUTDOWN */ return NS_ERROR_UNEXPECTED; } nsresult rv = NS_ERROR_SERVICE_NOT_AVAILABLE; nsFactoryEntry *entry = nsnull; { nsAutoMonitor mon(mMon); nsContractIDTableEntry* contractIDTableEntry = static_cast (PL_DHashTableOperate(&mContractIDs, aContractID, PL_DHASH_LOOKUP)); if (PL_DHASH_ENTRY_IS_BUSY(contractIDTableEntry)) { entry = contractIDTableEntry->mFactoryEntry; } } // exit monitor if (entry && entry->mServiceObject) { nsCOMPtr service; rv = entry->mServiceObject->QueryInterface(aIID, getter_AddRefs(service)); *result = (service!=nsnull); } return rv; } NS_IMETHODIMP nsComponentManagerImpl::UnregisterService(const char* aContractID) { nsresult rv = NS_OK; nsAutoMonitor mon(mMon); nsFactoryEntry *entry = nsnull; nsContractIDTableEntry* contractIDTableEntry = static_cast (PL_DHashTableOperate(&mContractIDs, aContractID, PL_DHASH_LOOKUP)); if (PL_DHASH_ENTRY_IS_BUSY(contractIDTableEntry)) { entry = contractIDTableEntry->mFactoryEntry; } if (!entry || !entry->mServiceObject) return NS_ERROR_SERVICE_NOT_AVAILABLE; entry->mServiceObject = nsnull; return rv; } NS_IMETHODIMP nsComponentManagerImpl::GetServiceByContractID(const char* aContractID, const nsIID& aIID, void* *result) { // test this first, since there's no point in returning a service during // shutdown -- whether it's available or not would depend on the order it // occurs in the list if (gXPCOMShuttingDown) { // When processing shutdown, don't process new GetService() requests #ifdef SHOW_DENIED_ON_SHUTDOWN nsXPIDLCString iid; iid.Adopt(aIID.ToString()); fprintf(stderr, "Getting service on shutdown. Denied.\n" " ContractID: %s\n IID: %s\n", aContractID, iid.get()); #endif /* SHOW_DENIED_ON_SHUTDOWN */ return NS_ERROR_UNEXPECTED; } nsAutoMonitor mon(mMon); nsFactoryEntry *entry = nsnull; nsContractIDTableEntry* contractIDTableEntry = static_cast (PL_DHashTableOperate(&mContractIDs, aContractID, PL_DHASH_LOOKUP)); if (!PL_DHASH_ENTRY_IS_BUSY(contractIDTableEntry)) return NS_ERROR_FACTORY_NOT_REGISTERED; entry = contractIDTableEntry->mFactoryEntry; NS_ASSERTION(entry, "This should never be null!"); if (entry->mServiceObject) { nsCOMPtr serviceObject = entry->mServiceObject; // We need to not be holding the service manager's monitor while calling // QueryInterface, because it invokes user code which could try to re-enter // the service manager, or try to grab some other lock/monitor/condvar // and deadlock, e.g. bug 282743. mon.Exit(); return serviceObject->QueryInterface(aIID, result); } PRThread* currentPRThread = PR_GetCurrentThread(); NS_ASSERTION(currentPRThread, "This should never be null!"); // Needed to optimize the event loop below. nsIThread* currentThread = nsnull; PRThread* pendingPRThread; while ((pendingPRThread = GetPendingServiceThread(entry->mCid))) { if (pendingPRThread == currentPRThread) { NS_ERROR("Recursive GetService!"); return NS_ERROR_NOT_AVAILABLE; } mon.Exit(); if (!currentThread) { currentThread = NS_GetCurrentThread(); NS_ASSERTION(currentThread, "This should never be null!"); } // This will process a single event or yield the thread if no event is // pending. if (!NS_ProcessNextEvent(currentThread, PR_FALSE)) { PR_Sleep(PR_INTERVAL_NO_WAIT); } mon.Enter(); } if (currentThread && entry->mServiceObject) { // If we have a currentThread then we must have waited on another thread // to create the service. Grab it now if that succeeded. nsCOMPtr serviceObject = entry->mServiceObject; mon.Exit(); return serviceObject->QueryInterface(aIID, result); } #ifdef DEBUG PendingServiceInfo* newInfo = #endif AddPendingService(entry->mCid, currentPRThread); NS_ASSERTION(newInfo, "Failed to add info to the array!"); nsCOMPtr service; // We need to not be holding the service manager's monitor while calling // CreateInstance, because it invokes user code which could try to re-enter // the service manager: mon.Exit(); nsresult rv = CreateInstanceByContractID(aContractID, nsnull, aIID, getter_AddRefs(service)); mon.Enter(); #ifdef DEBUG pendingPRThread = GetPendingServiceThread(entry->mCid); NS_ASSERTION(pendingPRThread == currentPRThread, "Pending service array has been changed!"); #endif RemovePendingService(entry->mCid); if (NS_FAILED(rv)) return rv; NS_ASSERTION(!entry->mServiceObject, "Created two instances of a service!"); entry->mServiceObject = service; *result = service.get(); NS_ADDREF(static_cast(*result)); return rv; } NS_IMETHODIMP nsComponentManagerImpl::GetService(const nsCID& aClass, const nsIID& aIID, nsISupports* *result, nsIShutdownListener* shutdownListener) { return GetService(aClass, aIID, (void**)result); } NS_IMETHODIMP nsComponentManagerImpl::GetService(const char* aContractID, const nsIID& aIID, nsISupports* *result, nsIShutdownListener* shutdownListener) { return GetServiceByContractID(aContractID, aIID, (void**)result); } NS_IMETHODIMP nsComponentManagerImpl::ReleaseService(const nsCID& aClass, nsISupports* service, nsIShutdownListener* shutdownListener) { NS_IF_RELEASE(service); return NS_OK; } NS_IMETHODIMP nsComponentManagerImpl::ReleaseService(const char* aContractID, nsISupports* service, nsIShutdownListener* shutdownListener) { NS_IF_RELEASE(service); return NS_OK; } NS_IMETHODIMP nsComponentManagerImpl::RegistryLocationForSpec(nsIFile *aSpec, char **aRegistryName) { nsCAutoString location; nsresult rv = RegistryLocationForFile(aSpec, location); if (NS_SUCCEEDED(rv)) { *aRegistryName = ToNewCString(location); if (!*aRegistryName) return NS_ERROR_OUT_OF_MEMORY; } return rv; } nsresult nsComponentManagerImpl::RegistryLocationForFile(nsIFile* aFile, nsCString& aRegistryName) { nsresult rv; if (!mComponentsDir) return NS_ERROR_NOT_INITIALIZED; // First check to see if this component is in the application // components directory PRBool containedIn; mComponentsDir->Contains(aFile, PR_TRUE, &containedIn); nsCAutoString nativePathString; if (containedIn){ rv = aFile->GetNativePath(nativePathString); if (NS_FAILED(rv)) return rv; aRegistryName.Assign(NS_LITERAL_CSTRING(XPCOM_RELCOMPONENT_PREFIX) + Substring(nativePathString, mComponentsOffset + 1)); return NS_OK; } // Next check to see if this component is in the GRE // components directory mGREComponentsDir->Contains(aFile, PR_TRUE, &containedIn); if (containedIn){ rv = aFile->GetNativePath(nativePathString); if (NS_FAILED(rv)) return rv; aRegistryName.Assign(NS_LITERAL_CSTRING(XPCOM_GRECOMPONENT_PREFIX) + Substring(nativePathString, mGREComponentsOffset + 1)); return NS_OK; } /* absolute names include volume info on Mac, so persistent descriptor */ rv = aFile->GetNativePath(nativePathString); if (NS_FAILED(rv)) return rv; aRegistryName.Assign(NS_LITERAL_CSTRING(XPCOM_ABSCOMPONENT_PREFIX) + nativePathString); return NS_OK; } NS_IMETHODIMP nsComponentManagerImpl::SpecForRegistryLocation(const char *aLocation, nsIFile **aSpec) { return FileForRegistryLocation(nsDependentCString(aLocation), (nsILocalFile**) aSpec); } nsresult nsComponentManagerImpl::FileForRegistryLocation(const nsCString &aLocation, nsILocalFile **aSpec) { // i18n: assuming aLocation is encoded for the current locale nsresult rv; const nsDependentCSubstring prefix = Substring(aLocation, 0, 4); /* abs:/full/path/to/libcomponent.so */ if (prefix.EqualsLiteral(XPCOM_ABSCOMPONENT_PREFIX)) { nsLocalFile* file = new nsLocalFile; if (!file) return NS_ERROR_FAILURE; rv = file->InitWithNativePath(Substring(aLocation, 4)); file->QueryInterface(NS_GET_IID(nsILocalFile), (void**)aSpec); return rv; } if (prefix.EqualsLiteral(XPCOM_RELCOMPONENT_PREFIX)) { if (!mComponentsDir) return NS_ERROR_NOT_INITIALIZED; nsILocalFile* file = nsnull; rv = mComponentsDir->Clone((nsIFile**)&file); if (NS_FAILED(rv)) return rv; rv = file->AppendRelativeNativePath(Substring(aLocation, 4)); *aSpec = file; return rv; } if (prefix.EqualsLiteral(XPCOM_GRECOMPONENT_PREFIX)) { if (!mGREComponentsDir) return NS_ERROR_NOT_INITIALIZED; nsILocalFile* file = nsnull; rv = mGREComponentsDir->Clone((nsIFile**)&file); if (NS_FAILED(rv)) return rv; rv = file->AppendRelativeNativePath(Substring(aLocation, 4)); *aSpec = file; return rv; } *aSpec = nsnull; return NS_ERROR_INVALID_ARG; } /** * RegisterFactory() * * Register a factory to be responsible for creation of implementation of * classID aClass. Plus creates as association of aClassName and aContractID * to the classID. If replace is PR_TRUE, we replace any existing registrations * with this one. * * Once registration is complete, we add the class to the factories cache * that we maintain. The factories cache is the ONLY place where these * registrations are ever kept. * * The other RegisterFunctions create a loader mapping and persistent * location, but we just slam it into the cache here. And we don't call the * loader's OnRegister function, either. */ NS_IMETHODIMP nsComponentManagerImpl::RegisterFactory(const nsCID &aClass, const char *aClassName, const char *aContractID, nsIFactory *aFactory, PRBool aReplace) { nsAutoMonitor mon(mMon); #ifdef PR_LOGGING if (PR_LOG_TEST(nsComponentManagerLog, PR_LOG_WARNING)) { char *buf = aClass.ToString(); PR_LOG(nsComponentManagerLog, PR_LOG_WARNING, ("nsComponentManager: RegisterFactory(%s, %s)", buf, (aContractID ? aContractID : "(null)"))); if (buf) NS_Free(buf); } #endif nsFactoryEntry *entry = nsnull; nsFactoryTableEntry* factoryTableEntry = static_cast (PL_DHashTableOperate(&mFactories, &aClass, PL_DHASH_ADD)); if (!factoryTableEntry) return NS_ERROR_OUT_OF_MEMORY; if (PL_DHASH_ENTRY_IS_BUSY(factoryTableEntry)) { entry = factoryTableEntry->mFactoryEntry; } if (entry && !aReplace) { // Already registered PR_LOG(nsComponentManagerLog, PR_LOG_ERROR, ("\t\tFactory already registered.")); return NS_ERROR_FACTORY_EXISTS; } void *mem; PL_ARENA_ALLOCATE(mem, &mArena, sizeof(nsFactoryEntry)); if (!mem) return NS_ERROR_OUT_OF_MEMORY; entry = new (mem) nsFactoryEntry(aClass, aFactory, entry); factoryTableEntry->mFactoryEntry = entry; // Update the ContractID->CLSID Map if (aContractID) { nsresult rv = HashContractID(aContractID, strlen(aContractID), entry); if (NS_FAILED(rv)) { PR_LOG(nsComponentManagerLog, PR_LOG_WARNING, ("\t\tFactory register succeeded. " "Hashing contractid (%s) FAILED.", aContractID)); return rv; } } PR_LOG(nsComponentManagerLog, PR_LOG_WARNING, ("\t\tFactory register succeeded contractid=%s.", aContractID ? aContractID : "")); return NS_OK; } NS_IMETHODIMP nsComponentManagerImpl::RegisterComponent(const nsCID &aClass, const char *aClassName, const char *aContractID, const char *aPersistentDescriptor, PRBool aReplace, PRBool aPersist) { NS_ENSURE_ARG_POINTER(aPersistentDescriptor); return RegisterComponentCommon(aClass, aClassName, aContractID, aContractID ? strlen(aContractID) : 0, aPersistentDescriptor, strlen(aPersistentDescriptor), aReplace, aPersist, nativeComponentType); } NS_IMETHODIMP nsComponentManagerImpl::RegisterComponentWithType(const nsCID &aClass, const char *aClassName, const char *aContractID, nsIFile *aSpec, const char *aLocation, PRBool aReplace, PRBool aPersist, const char *aType) { NS_ENSURE_ARG_POINTER(aLocation); return RegisterComponentCommon(aClass, aClassName, aContractID, aContractID ? strlen(aContractID) : 0, aLocation, strlen(aLocation), aReplace, aPersist, aType); } /* * Register a component, using whatever they stuck in the nsIFile. */ NS_IMETHODIMP nsComponentManagerImpl::RegisterComponentSpec(const nsCID &aClass, const char *aClassName, const char *aContractID, nsIFile *aLibrarySpec, PRBool aReplace, PRBool aPersist) { nsCAutoString registryName; nsresult rv = RegistryLocationForFile(aLibrarySpec, registryName); if (NS_FAILED(rv)) return rv; rv = RegisterComponentWithType(aClass, aClassName, aContractID, aLibrarySpec, registryName.get(), aReplace, aPersist, nativeComponentType); return rv; } NS_IMETHODIMP nsComponentManagerImpl::RegisterComponentLib(const nsCID &aClass, const char *aClassName, const char *aContractID, const char *aDllName, PRBool aReplace, PRBool aPersist) { // deprecated and obsolete. return NS_ERROR_NOT_IMPLEMENTED; } /* * Add a component to the known universe of components. * * Once we enter this function, we own aRegistryName, and must free it * or hand it to nsFactoryEntry. Common exit point ``out'' helps keep us * sane. */ nsresult nsComponentManagerImpl::RegisterComponentCommon(const nsCID &aClass, const char *aClassName, const char *aContractID, PRUint32 aContractIDLen, const char *aRegistryName, PRUint32 aRegistryNameLen, PRBool aReplace, PRBool aPersist, const char *aType) { nsresult rv; nsAutoMonitor mon(mMon); nsFactoryEntry *entry = GetFactoryEntry(aClass); // Normalize proid and classname const char *contractID = (aContractID && *aContractID) ? aContractID : nsnull; #ifdef PR_LOGGING if (PR_LOG_TEST(nsComponentManagerLog, PR_LOG_WARNING)) { char *buf = aClass.ToString(); PR_LOG(nsComponentManagerLog, PR_LOG_WARNING, ("nsComponentManager: RegisterComponentCommon(%s, %s, %s, %s)", buf, contractID ? contractID : "(null)", aRegistryName, aType)); if (buf) NS_Free(buf); } #endif if (entry && !aReplace) { PR_LOG(nsComponentManagerLog, PR_LOG_ERROR, ("\t\tFactory already registered.")); return NS_ERROR_FACTORY_EXISTS; } int typeIndex = GetLoaderType(aType); if (typeIndex == NS_LOADER_TYPE_INVALID) return NS_ERROR_OUT_OF_MEMORY; if (entry) { entry->ReInit(typeIndex, aRegistryName); } else { // Arena allocate the nsFactoryEntry void *mem; PL_ARENA_ALLOCATE(mem, &mArena, sizeof(nsFactoryEntry)); if (!mem) return NS_ERROR_OUT_OF_MEMORY; mRegistryDirty = PR_TRUE; entry = new (mem) nsFactoryEntry(aClass, typeIndex, aRegistryName); if (!entry->mLocationKey) return NS_ERROR_OUT_OF_MEMORY; nsFactoryTableEntry* factoryTableEntry = static_cast (PL_DHashTableOperate(&mFactories, &aClass, PL_DHASH_ADD)); if (!factoryTableEntry) return NS_ERROR_OUT_OF_MEMORY; factoryTableEntry->mFactoryEntry = entry; } // Update the ContractID->CLSID Map if (contractID) { rv = HashContractID(contractID, aContractIDLen, entry); if (NS_FAILED(rv)) { PR_LOG(nsComponentManagerLog, PR_LOG_ERROR, ("\t\tHashContractID(%s) FAILED\n", contractID)); return rv; } } return NS_OK; } nsIModuleLoader* nsComponentManagerImpl::LoaderForType(LoaderType aType) { NS_ASSERTION(aType != NS_LOADER_TYPE_STATIC, "Static component loader is special"); if (aType == NS_LOADER_TYPE_INVALID) return nsnull; if (aType == NS_LOADER_TYPE_NATIVE) return &mNativeModuleLoader; NS_ASSERTION(aType >= 0 && aType < mLoaderData.Length(), "LoaderType out of range"); if (!mLoaderData[aType].loader) { nsCOMPtr loader; loader = do_GetServiceFromCategory("module-loader", mLoaderData[aType].type.get()); if (!loader) return nsnull; loader.swap(mLoaderData[aType].loader); } return mLoaderData[aType].loader; } void nsComponentManagerImpl::GetAllLoaders() { NS_ASSERTION(mCategoryManager, "nsComponentManager used uninitialized"); nsCOMPtr loaderEnum; mCategoryManager->EnumerateCategory("module-loader", getter_AddRefs(loaderEnum)); nsCOMPtr loaderStrings(do_QueryInterface(loaderEnum)); if (loaderStrings) { PRBool hasMore; while (NS_SUCCEEDED(loaderStrings->HasMore(&hasMore)) && hasMore) { nsCAutoString loaderType; if (NS_FAILED(loaderStrings->GetNext(loaderType))) continue; // We depend on the loader being created. Add the loader type and // create the loader object too. (void) LoaderForType(AddLoaderType(loaderType.get())); } } } // Convert a loader type string into an index into the component data // array. Empty loader types are converted to NATIVE. LoaderType nsComponentManagerImpl::GetLoaderType(const char *typeStr) { if (!typeStr || !*typeStr) { // Empty type strings are NATIVE return NS_LOADER_TYPE_NATIVE; } if (!strcmp(typeStr, staticComponentType)) return NS_LOADER_TYPE_STATIC; if (!strcmp(typeStr, nativeComponentType)) return NS_LOADER_TYPE_NATIVE; const nsDependentCString type(typeStr); for (unsigned int i = 0; i < mLoaderData.Length(); ++i) { if (mLoaderData[i].type == type) return i; } return NS_LOADER_TYPE_INVALID; } // Add a loader type if not already known. Out the typeIndex // if the loader type is either added or already there. LoaderType nsComponentManagerImpl::AddLoaderType(const char *typeStr) { LoaderType typeIndex = GetLoaderType(typeStr); if (typeIndex != NS_LOADER_TYPE_INVALID) return typeIndex; // Add the loader type nsLoaderdata *elem = mLoaderData.AppendElement(); if (!elem) return NS_LOADER_TYPE_INVALID; elem->type.Assign(typeStr); return mLoaderData.Length() - 1; } typedef struct { const nsCID* cid; nsIFactory* factory; } UnregisterConditions; static PLDHashOperator DeleteFoundCIDs(PLDHashTable *aTable, PLDHashEntryHdr *aHdr, PRUint32 aNumber, void *aData) { nsContractIDTableEntry* entry = static_cast(aHdr); if (!entry->mFactoryEntry) return PL_DHASH_NEXT; UnregisterConditions* data = (UnregisterConditions*)aData; nsFactoryEntry* factoryEntry = entry->mFactoryEntry; if (data->cid->Equals(factoryEntry->mCid) && data->factory == factoryEntry->mFactory.get()) return PL_DHASH_REMOVE; return PL_DHASH_NEXT; } void nsComponentManagerImpl::DeleteContractIDEntriesByCID(const nsCID* aClass, nsIFactory* factory) { UnregisterConditions aData; aData.cid = aClass; aData.factory = factory; PL_DHashTableEnumerate(&mContractIDs, DeleteFoundCIDs, (void*)&aData); } nsresult nsComponentManagerImpl::UnregisterFactory(const nsCID &aClass, nsIFactory *aFactory) { #ifdef PR_LOGGING if (PR_LOG_TEST(nsComponentManagerLog, PR_LOG_WARNING)) { char *buf = aClass.ToString(); PR_LOG(nsComponentManagerLog, PR_LOG_WARNING, ("nsComponentManager: UnregisterFactory(%s)", buf)); if (buf) NS_Free(buf); } #endif nsFactoryEntry *old; // first delete all contract id entries that are registered with this cid. DeleteContractIDEntriesByCID(&aClass, aFactory); // next check to see if there is a CID registered nsresult rv = NS_ERROR_FACTORY_NOT_REGISTERED; old = GetFactoryEntry(aClass); if (old && (old->mFactory.get() == aFactory)) { nsAutoMonitor mon(mMon); PL_DHashTableOperate(&mFactories, &aClass, PL_DHASH_REMOVE); rv = NS_OK; } PR_LOG(nsComponentManagerLog, PR_LOG_WARNING, ("\t\tUnregisterFactory() %s", NS_SUCCEEDED(rv) ? "succeeded" : "FAILED")); return rv; } NS_IMETHODIMP nsComponentManagerImpl::UnregisterComponent(const nsCID &aClass, const char *registryName) { NS_ERROR("Don't call nsIComponentManagerObsolete.unregisterComponent"); return NS_ERROR_NOT_IMPLEMENTED; } NS_IMETHODIMP nsComponentManagerImpl::UnregisterComponentSpec(const nsCID &aClass, nsIFile *aLibrarySpec) { NS_ERROR("Don't call nsIComponentManagerObsolete.unregisterComponentSpec"); return NS_ERROR_NOT_IMPLEMENTED; } NS_IMETHODIMP nsComponentManagerImpl::FreeLibraries(void) { return NS_ERROR_NOT_IMPLEMENTED; } /** * AutoRegister(RegistrationInstant, const char *directory) * * Given a directory in the following format, this will ensure proper registration * of all components. No default directory is looked at. * * Directory and fullname are what NSPR will accept. For eg. * WIN y:/home/dp/mozilla/dist/bin * UNIX /home/dp/mozilla/dist/bin * MAC /Hard drive/mozilla/dist/apprunner * * This will take care not loading already registered dlls, finding and * registering new dlls, re-registration of modified dlls * */ NS_IMETHODIMP nsComponentManagerImpl::AutoRegister(PRInt32 when, nsIFile *inDirSpec) { return AutoRegister(inDirSpec); } nsresult nsComponentManagerImpl::AutoRegisterImpl(nsIFile *inDirSpec, nsCOMArray &aLeftovers, nsTArray &aDeferred) { NS_ASSERTION(inDirSpec, "inDirSpec must not be null"); nsresult rv; PRBool isDir; rv = inDirSpec->IsDirectory(&isDir); if (NS_FAILED(rv)) return rv; if (isDir) return AutoRegisterDirectory(inDirSpec, aLeftovers, aDeferred); nsCOMPtr lf(do_QueryInterface(inDirSpec)); if (!lf) return NS_NOINTERFACE; rv = AutoRegisterComponent(lf, aDeferred); if (NS_FAILED(rv)) aLeftovers.AppendObject(lf); return rv; } static const char kNL[] = "\r\n"; nsresult nsComponentManagerImpl::AutoRegisterDirectory(nsIFile *inDirSpec, nsCOMArray &aLeftovers, nsTArray &aDeferred) { nsresult rv; nsCOMPtr componentsList; inDirSpec->Clone(getter_AddRefs(componentsList)); if (componentsList) { nsCOMPtr lfComponentsList = do_QueryInterface(componentsList); lfComponentsList->AppendNative(NS_LITERAL_CSTRING("components.list")); PRFileDesc* fd; if (NS_SUCCEEDED(lfComponentsList->OpenNSPRFileDesc(PR_RDONLY, 0400, &fd))) { rv = AutoRegisterComponentsList(inDirSpec, fd, aLeftovers, aDeferred); PR_Close(fd); return rv; } } nsCOMPtr entries; rv = inDirSpec->GetDirectoryEntries(getter_AddRefs(entries)); if (NS_FAILED(rv)) return rv; PRBool hasMore; nsCOMPtr elem; while (NS_SUCCEEDED(entries->HasMoreElements(&hasMore)) && hasMore) { entries->GetNext(getter_AddRefs(elem)); nsCOMPtr lf(do_QueryInterface(elem)); if (!lf) { NS_ERROR("Directory enumerator misbehaving"); continue; } PRBool isDir; rv = lf->IsDirectory(&isDir); if (NS_FAILED(rv)) continue; if (isDir) AutoRegisterDirectory(lf, aLeftovers, aDeferred); else { rv = AutoRegisterComponent(lf, aDeferred); if (NS_FAILED(rv)) aLeftovers.AppendObject(lf); } } return NS_OK; } nsresult nsComponentManagerImpl::AutoRegisterComponentsList(nsIFile* inDir, PRFileDesc* fd, nsCOMArray& aLeftovers, nsTArray& aDeferred) { PRFileInfo info; if (PR_SUCCESS != PR_GetOpenFileInfo(fd, &info)) return NS_ErrorAccordingToNSPR(); nsAutoArrayPtr buf(new char[info.size + 1]); if (!buf) return NS_ERROR_OUT_OF_MEMORY; PRInt32 read = 0; while (read < info.size) { PRInt32 n = PR_Read(fd, buf + read, info.size - read); if (n < 0) return NS_ErrorAccordingToNSPR(); read += n; if (n == 0) break; } buf[read] = '\0'; char* c = buf; while (char *token = NS_strtok(kNL, &c)) { if (token[0] == '#') continue; nsCOMPtr component; inDir->Clone(getter_AddRefs(component)); if (!component) return NS_ERROR_OUT_OF_MEMORY; nsCOMPtr lfcomponent = do_QueryInterface(component); lfcomponent->AppendNative(nsDependentCString(token)); nsresult rv = AutoRegisterComponent(lfcomponent, aDeferred); if (NS_FAILED(rv)) aLeftovers.AppendObject(lfcomponent); } return NS_OK; } nsresult nsComponentManagerImpl::AutoRegisterComponent(PRInt32 unused, nsIFile *component) { nsCOMPtr lf(do_QueryInterface(component)); if (!lf) return NS_NOINTERFACE; GetAllLoaders(); nsTArray deferred; nsresult rv = AutoRegisterComponent(lf, deferred); if (deferred.Length()) return NS_ERROR_FACTORY_REGISTER_AGAIN; return rv; } nsresult nsComponentManagerImpl::AutoRegisterComponent(nsILocalFile* aComponentFile, nsTArray &aDeferred, LoaderType minLoader) { nsresult rv; NS_ASSERTION(minLoader < GetLoaderCount(), "Bad minLoader"); nsCAutoString registryLocation; rv = RegistryLocationForFile(aComponentFile, registryLocation); if (NS_FAILED(rv)) return rv; const nsDependentCSubstring extension = StringTail(registryLocation, 4); if (extension.LowerCaseEqualsLiteral(".dat") || extension.LowerCaseEqualsLiteral(".xpt")) return NS_OK; nsCOMPtr lfhash(do_QueryInterface(aComponentFile)); if (!lfhash) { NS_ERROR("localfile not implementing nsIHashable!"); return NS_NOINTERFACE; } PRInt64 modTime = 0; PRInt64 cachedModTime; // If it's in the cache, it should be valid. // The cache file is removed if files are modified. if (mAutoRegEntries.Get(lfhash, &cachedModTime)) { #ifdef DEBUG if (NS_SUCCEEDED(aComponentFile->GetLastModifiedTime(&modTime)) && cachedModTime == modTime) { return NS_OK; } #else return NS_OK; #endif } const char *registryType = nsnull; nsCOMPtr module; if (minLoader == NS_LOADER_TYPE_NATIVE) { rv = mNativeModuleLoader.LoadModule(aComponentFile, getter_AddRefs(module)); if (NS_SUCCEEDED(rv)) { if (!module) { NS_ERROR("Module loader succeeded without returning a module"); return NS_ERROR_FAILURE; } registryType = nativeComponentType; } minLoader = 0; } if (!registryType) { for (; minLoader < GetLoaderCount(); ++minLoader) { nsIModuleLoader* loader = LoaderForType(minLoader); if (!loader) continue; rv = loader->LoadModule(aComponentFile, getter_AddRefs(module)); if (NS_SUCCEEDED(rv)) { if (!module) { NS_ERROR("Module loader succeeded without returning a module."); return NS_ERROR_FAILURE; } registryType = StringForLoaderType(minLoader); break; } } if (!registryType) { return NS_ERROR_FAILURE; } } rv = module->RegisterSelf(this, aComponentFile, registryLocation.get(), registryType); if (NS_ERROR_FACTORY_REGISTER_AGAIN == rv) { DeferredModule *d = aDeferred.AppendElement(); if (!d) return NS_ERROR_OUT_OF_MEMORY; d->type = registryType; d->file = aComponentFile; d->location = registryLocation; d->module = module; d->modTime = modTime; return NS_OK; } if (NS_SUCCEEDED(rv) && modTime != 0) mAutoRegEntries.Put(lfhash, modTime); return rv; } void nsComponentManagerImpl::LoadLeftoverComponents( nsCOMArray &aLeftovers, nsTArray &aDeferred, LoaderType minLoader) { NS_ASSERTION(minLoader >= GetLoaderCount(), "Corrupted minLoader"); GetAllLoaders(); // If there aren't any new loaders found since we tried this last, there's // nothing left to do. if (GetLoaderCount() == minLoader) return; LoaderType curLoader = GetLoaderCount(); for (PRInt32 i = 0; i < aLeftovers.Count(); ) { nsresult rv = AutoRegisterComponent(aLeftovers[i], aDeferred, minLoader); if (NS_SUCCEEDED(rv)) { aLeftovers.RemoveObjectAt(i); } else { ++i; } } if (aLeftovers.Count()) // recursively try this again until there are no new loaders found LoadLeftoverComponents(aLeftovers, aDeferred, curLoader); } void nsComponentManagerImpl::LoadDeferredModules(nsTArray &aDeferred) { // We keep looping through deferred components until one of // 1) they're all gone // 2) we loop through and none of them succeed PRUint32 lastCount = PR_UINT32_MAX; while (aDeferred.Length() > 0 && lastCount > aDeferred.Length()) { lastCount = aDeferred.Length(); for (PRInt32 i = 0; i < aDeferred.Length(); ) { DeferredModule &d = aDeferred[i]; nsresult rv = d.module->RegisterSelf(this, d.file, d.location.get(), d.type); if (NS_SUCCEEDED(rv) && d.modTime != 0) { nsCOMPtr lfhash(do_QueryInterface(d.file)); if (lfhash) mAutoRegEntries.Put(lfhash, d.modTime); } if (rv != NS_ERROR_FACTORY_REGISTER_AGAIN) { aDeferred.RemoveElementAt(i); } else { ++i; } } } } NS_IMETHODIMP nsComponentManagerImpl::AutoUnregisterComponent(PRInt32 /* unused */, nsIFile *component) { nsresult rv; GetAllLoaders(); nsCAutoString location; rv = RegistryLocationForFile(component, location); if (NS_FAILED(rv)) return rv; nsCOMPtr lf(do_QueryInterface(component)); if (!lf) return NS_NOINTERFACE; nsCOMPtr module; rv = mNativeModuleLoader.LoadModule(lf, getter_AddRefs(module)); if (NS_FAILED(rv)) { for (LoaderType i = 0; i < mLoaderData.Length(); ++i) { nsIModuleLoader* loader = LoaderForType(i); if (!loader) continue; if (NS_SUCCEEDED(loader->LoadModule(lf, getter_AddRefs(module))) && module) { break; } } } if (!module) return NS_ERROR_FAILURE; rv = module->UnregisterSelf(this, lf, location.get()); nsCOMPtr lfhash(do_QueryInterface(lf)); mAutoRegEntries.Remove(lfhash); return rv; } NS_IMETHODIMP nsComponentManagerImpl::IsRegistered(const nsCID &aClass, PRBool *aRegistered) { if (!aRegistered) { NS_ERROR("null ptr"); return NS_ERROR_NULL_POINTER; } *aRegistered = (nsnull != GetFactoryEntry(aClass)); return NS_OK; } NS_IMETHODIMP nsComponentManagerImpl::EnumerateCLSIDs(nsIEnumerator** aEnumerator) { NS_ASSERTION(aEnumerator != nsnull, "null ptr"); if (!aEnumerator) { return NS_ERROR_NULL_POINTER; } *aEnumerator = nsnull; nsresult rv; PLDHashTableEnumeratorImpl *aEnum; rv = PL_NewDHashTableEnumerator(&mFactories, ConvertFactoryEntryToCID, (void*)this, &aEnum); if (NS_FAILED(rv)) return rv; *aEnumerator = static_cast(aEnum); return NS_OK; } NS_IMETHODIMP nsComponentManagerImpl::EnumerateContractIDs(nsIEnumerator** aEnumerator) { NS_ASSERTION(aEnumerator != nsnull, "null ptr"); if (!aEnumerator) { return NS_ERROR_NULL_POINTER; } *aEnumerator = nsnull; nsresult rv; PLDHashTableEnumeratorImpl *aEnum; rv = PL_NewDHashTableEnumerator(&mContractIDs, ConvertContractIDKeyToString, (void*)this, &aEnum); if (NS_FAILED(rv)) return rv; *aEnumerator = static_cast(aEnum); return NS_OK; } // nsIComponentRegistrar static void RegisterStaticModule(const char *key, nsIModule* module, nsTArray &deferred) { nsresult rv = module-> RegisterSelf(nsComponentManagerImpl::gComponentManager, nsnull, key, staticComponentType); if (NS_ERROR_FACTORY_REGISTER_AGAIN == rv) { DeferredModule *d = deferred.AppendElement(); if (d) { d->type = staticComponentType; d->location = key; } } } static void ReportLoadFailure(nsIFile* aFile, nsIConsoleService* aCS) { nsAutoString message; aFile->GetPath(message); message.Insert(NS_LITERAL_STRING("Failed to load XPCOM component: "), 0); aCS->LogStringMessage(message.get()); } NS_IMETHODIMP nsComponentManagerImpl::AutoRegister(nsIFile *aSpec) { nsresult rv; if (!mCategoryManager) { mCategoryManager = do_GetService(NS_CATEGORYMANAGER_CONTRACTID, &rv); if (NS_FAILED(rv)) return rv; } GetAllLoaders(); // Notify observers of xpcom autoregistration start NS_CreateServicesFromCategory(NS_XPCOM_AUTOREGISTRATION_OBSERVER_ID, aSpec, "start"); nsCOMArray leftovers; nsTArray deferred; if (!aSpec) { mStaticModuleLoader.EnumerateModules(RegisterStaticModule, deferred); // Builtin component loaders (xpconnect!) can be static modules. // Set them up now, so that JS components don't go into // the leftovers list. GetAllLoaders(); } LoaderType curLoader = GetLoaderCount(); if (aSpec) { rv = AutoRegisterImpl(aSpec, leftovers, deferred); } else { // register static components, then GRE components, then // NS_XPCOM_COMPONENT_DIR, then NS_XPCOM_COMPONENT_DIR_LIST PRBool equals = PR_FALSE; if (mGREComponentsDir && NS_SUCCEEDED(mGREComponentsDir->Equals(mComponentsDir, &equals)) && !equals) { rv = AutoRegisterImpl(mGREComponentsDir, leftovers, deferred); if (NS_FAILED(rv)) { NS_WARNING("Couldn't register mGREComponentsDir"); } } rv = AutoRegisterImpl(mComponentsDir, leftovers, deferred); if (NS_FAILED(rv)) { NS_WARNING("Couldn't register mComponentsDir"); } nsCOMPtr dirList; rv = nsDirectoryService::gService->Get(NS_XPCOM_COMPONENT_DIR_LIST, NS_GET_IID(nsISimpleEnumerator), getter_AddRefs(dirList)); if (NS_SUCCEEDED(rv) && dirList) { PRBool hasMore; nsCOMPtr elem; while (NS_SUCCEEDED(dirList->HasMoreElements(&hasMore)) && hasMore) { dirList->GetNext(getter_AddRefs(elem)); nsCOMPtr dir(do_QueryInterface(elem)); if (dir) { AutoRegisterImpl(dir, leftovers, deferred); } } } rv = NS_OK; } if (NS_SUCCEEDED(rv)) { // We may have found new component loaders in the meantime, try to go // back and load the leftovers. if (leftovers.Count()) LoadLeftoverComponents(leftovers, deferred, curLoader); // If a module said its loading should be deferred, register it now if (deferred.Length()) LoadDeferredModules(deferred); nsCOMPtr cs(do_GetService(NS_CONSOLESERVICE_CONTRACTID)); if (cs) { for (PRInt32 i = 0; i < leftovers.Count(); ++i) { ReportLoadFailure(leftovers[i], cs); } } } // Notify observers of xpcom autoregistration completion NS_CreateServicesFromCategory(NS_XPCOM_AUTOREGISTRATION_OBSERVER_ID, aSpec, "end"); if (mRegistryDirty) WritePersistentRegistry(); return rv; } NS_IMETHODIMP nsComponentManagerImpl::AutoUnregister(nsIFile *aSpec) { // unregistering a complete directory is not implmeneted yet...FIX if (aSpec == nsnull) return NS_ERROR_NOT_IMPLEMENTED; PRBool directory; aSpec->IsDirectory(&directory); if (directory) return NS_ERROR_NOT_IMPLEMENTED; return AutoUnregisterComponent(0, aSpec); } NS_IMETHODIMP nsComponentManagerImpl::RegisterFactory(const nsCID & aClass, const char *aClassName, const char *aContractID, nsIFactory *aFactory) { return RegisterFactory(aClass, aClassName, aContractID, aFactory, PR_TRUE); } NS_IMETHODIMP nsComponentManagerImpl::RegisterFactoryLocation(const nsCID & aClass, const char *aClassName, const char *aContractID, nsIFile *aFile, const char *loaderStr, const char *aType) { nsCAutoString registryName; if (!loaderStr) { nsresult rv = RegistryLocationForFile(aFile, registryName); if (NS_FAILED(rv)) return rv; } nsresult rv; rv = RegisterComponentWithType(aClass, aClassName, aContractID, aFile, (loaderStr ? loaderStr : registryName.get()), PR_TRUE, PR_TRUE, (aType ? aType : nativeComponentType)); return rv; } NS_IMETHODIMP nsComponentManagerImpl::UnregisterFactoryLocation(const nsCID & aClass, nsIFile *aFile) { return UnregisterComponentSpec(aClass, aFile); } NS_IMETHODIMP nsComponentManagerImpl::IsCIDRegistered(const nsCID & aClass, PRBool *_retval) { return IsRegistered(aClass, _retval); } NS_IMETHODIMP nsComponentManagerImpl::IsContractIDRegistered(const char *aClass, PRBool *_retval) { NS_ENSURE_ARG_POINTER(aClass); nsFactoryEntry *entry = GetFactoryEntry(aClass, strlen(aClass)); if (entry) *_retval = PR_TRUE; else *_retval = PR_FALSE; return NS_OK; } NS_IMETHODIMP nsComponentManagerImpl::EnumerateCIDs(nsISimpleEnumerator **aEnumerator) { NS_ASSERTION(aEnumerator != nsnull, "null ptr"); if (!aEnumerator) return NS_ERROR_NULL_POINTER; *aEnumerator = nsnull; nsresult rv; PLDHashTableEnumeratorImpl *aEnum; rv = PL_NewDHashTableEnumerator(&mFactories, ConvertFactoryEntryToCID, (void*)this, &aEnum); if (NS_FAILED(rv)) return rv; *aEnumerator = static_cast(aEnum); return NS_OK; } NS_IMETHODIMP nsComponentManagerImpl::EnumerateContractIDs(nsISimpleEnumerator **aEnumerator) { NS_ASSERTION(aEnumerator != nsnull, "null ptr"); if (!aEnumerator) return NS_ERROR_NULL_POINTER; *aEnumerator = nsnull; nsresult rv; PLDHashTableEnumeratorImpl *aEnum; rv = PL_NewDHashTableEnumerator(&mContractIDs, ConvertContractIDKeyToString, (void*)this, &aEnum); if (NS_FAILED(rv)) return rv; *aEnumerator = static_cast(aEnum); return NS_OK; } NS_IMETHODIMP nsComponentManagerImpl::CIDToContractID(const nsCID & aClass, char **_retval) { return CLSIDToContractID(aClass, nsnull, _retval); } NS_IMETHODIMP nsComponentManagerImpl::ContractIDToCID(const char *aContractID, nsCID * *_retval) { *_retval = (nsCID*) nsMemory::Alloc(sizeof(nsCID)); if (!*_retval) return NS_ERROR_OUT_OF_MEMORY; nsresult rv = ContractIDToClassID(aContractID, *_retval); if (NS_FAILED(rv)) { nsMemory::Free(*_retval); *_retval = nsnull; } return rv; } //////////////////////////////////////////////////////////////////////////////// // nsFactoryEntry //////////////////////////////////////////////////////////////////////////////// nsFactoryEntry::nsFactoryEntry(const nsCID &aClass, LoaderType aLoaderType, const char *aLocationKey, nsFactoryEntry *aParent) : mCid(aClass), mLoaderType(aLoaderType), mLocationKey( ArenaStrdup(aLocationKey, &nsComponentManagerImpl::gComponentManager->mArena)), mParent(aParent) { } void nsFactoryEntry::ReInit(LoaderType aLoaderType, const char *aLocationKey) { mLoaderType = aLoaderType; if (!mLocationKey || strcmp(mLocationKey, aLocationKey)) { mLocationKey = ArenaStrdup(aLocationKey, &nsComponentManagerImpl::gComponentManager->mArena); } } nsresult nsFactoryEntry::GetFactory(nsIFactory **aFactory) { if (!mFactory) { nsresult rv; if (mLoaderType == NS_LOADER_TYPE_INVALID) return NS_ERROR_FAILURE; nsCOMPtr module; if (mLoaderType == NS_LOADER_TYPE_STATIC) { rv = nsComponentManagerImpl::gComponentManager-> mStaticModuleLoader. GetModuleFor(mLocationKey, getter_AddRefs(module)); } else { nsCOMPtr moduleFile; rv = nsComponentManagerImpl::gComponentManager-> FileForRegistryLocation(nsDependentCString(mLocationKey), getter_AddRefs(moduleFile)); NS_ENSURE_SUCCESS(rv, rv); nsIModuleLoader* loader = nsComponentManagerImpl::gComponentManager-> LoaderForType(mLoaderType); if (!loader) return NS_ERROR_FAILURE; rv = loader->LoadModule(moduleFile, getter_AddRefs(module)); } if (NS_FAILED(rv)) return rv; if (!module) { NS_ERROR("Module returned success but no module!"); return NS_ERROR_FAILURE; } rv = module-> GetClassObject(nsComponentManagerImpl::gComponentManager, mCid, NS_GET_IID(nsIFactory), getter_AddRefs(mFactory)); if (NS_FAILED(rv)) return rv; NS_ASSERTION(mFactory, "Loader says it succeeded; got null factory!"); if (!mFactory) return NS_ERROR_UNEXPECTED; } *aFactory = mFactory.get(); NS_ADDREF(*aFactory); return NS_OK; } nsFactoryEntry::~nsFactoryEntry() { // nsFactoryEntry is arena-allocated. So we don't delete it; // call the destructor by hand. if (mParent) mParent->~nsFactoryEntry(); } //////////////////////////////////////////////////////////////////////////////// // Static Access Functions //////////////////////////////////////////////////////////////////////////////// NS_COM nsresult NS_GetComponentManager(nsIComponentManager* *result) { if (nsComponentManagerImpl::gComponentManager == nsnull) { // XPCOM needs initialization. nsresult rv = NS_InitXPCOM2(nsnull, nsnull, nsnull); if (NS_FAILED(rv)) return rv; } *result = static_cast (nsComponentManagerImpl::gComponentManager); NS_IF_ADDREF(*result); return NS_OK; } NS_COM nsresult NS_GetServiceManager(nsIServiceManager* *result) { nsresult rv = NS_OK; if (nsComponentManagerImpl::gComponentManager == nsnull) { // XPCOM needs initialization. rv = NS_InitXPCOM2(nsnull, nsnull, nsnull); } if (NS_FAILED(rv)) return rv; *result = static_cast (nsComponentManagerImpl::gComponentManager); NS_IF_ADDREF(*result); return NS_OK; } NS_COM nsresult NS_GetComponentRegistrar(nsIComponentRegistrar* *result) { nsresult rv = NS_OK; if (nsComponentManagerImpl::gComponentManager == nsnull) { // XPCOM needs initialization. rv = NS_InitXPCOM2(nsnull, nsnull, nsnull); } if (NS_FAILED(rv)) return rv; *result = static_cast (nsComponentManagerImpl::gComponentManager); NS_IF_ADDREF(*result); return NS_OK; }