/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ /* ***** 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) 2000 * the Initial Developer. All Rights Reserved. * * Contributor(s): * Scott Collins * * 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 ***** */ #define PL_ARENA_CONST_ALIGN_MASK 7 #include "nsICategoryManager.h" #include "nsCategoryManager.h" #include "plarena.h" #include "prio.h" #include "prprf.h" #include "prlock.h" #include "nsCOMPtr.h" #include "nsTHashtable.h" #include "nsClassHashtable.h" #include "nsIFactory.h" #include "nsIStringEnumerator.h" #include "nsSupportsPrimitives.h" #include "nsServiceManagerUtils.h" #include "nsIObserver.h" #include "nsIObserverService.h" #include "nsReadableUtils.h" #include "nsCRT.h" #include "nsQuickSort.h" #include "nsEnumeratorUtils.h" #include "nsIProxyObjectManager.h" #include "nsThreadUtils.h" #include "mozilla/Services.h" #include "mozilla/FunctionTimer.h" using namespace mozilla; class nsIComponentLoaderManager; /* CategoryDatabase contains 0 or more 1-1 mappings of string to Category each Category contains 0 or more 1-1 mappings of string keys to string values In other words, the CategoryDatabase is a tree, whose root is a hashtable. Internal nodes (or Categories) are hashtables. Leaf nodes are strings. The leaf strings are allocated in an arena, because we assume they're not going to change much ;) */ #define NS_CATEGORYMANAGER_ARENA_SIZE (1024 * 8) // pulled in from nsComponentManager.cpp char* ArenaStrdup(const char* s, PLArenaPool* aArena); // // BaseStringEnumerator is subclassed by EntryEnumerator and // CategoryEnumerator // class BaseStringEnumerator : public nsISimpleEnumerator, private nsIUTF8StringEnumerator { public: NS_DECL_ISUPPORTS NS_DECL_NSISIMPLEENUMERATOR NS_DECL_NSIUTF8STRINGENUMERATOR protected: // Callback function for NS_QuickSort to sort mArray static int SortCallback(const void *, const void *, void *); BaseStringEnumerator() : mArray(nsnull), mCount(0), mSimpleCurItem(0), mStringCurItem(0) { } // A virtual destructor is needed here because subclasses of // BaseStringEnumerator do not implement their own Release() method. virtual ~BaseStringEnumerator() { if (mArray) delete[] mArray; } void Sort(); const char** mArray; PRUint32 mCount; PRUint32 mSimpleCurItem; PRUint32 mStringCurItem; }; NS_IMPL_ISUPPORTS2(BaseStringEnumerator, nsISimpleEnumerator, nsIUTF8StringEnumerator) NS_IMETHODIMP BaseStringEnumerator::HasMoreElements(PRBool *_retval) { *_retval = (mSimpleCurItem < mCount); return NS_OK; } NS_IMETHODIMP BaseStringEnumerator::GetNext(nsISupports **_retval) { if (mSimpleCurItem >= mCount) return NS_ERROR_FAILURE; nsSupportsDependentCString* str = new nsSupportsDependentCString(mArray[mSimpleCurItem++]); if (!str) return NS_ERROR_OUT_OF_MEMORY; *_retval = str; NS_ADDREF(*_retval); return NS_OK; } NS_IMETHODIMP BaseStringEnumerator::HasMore(PRBool *_retval) { *_retval = (mStringCurItem < mCount); return NS_OK; } NS_IMETHODIMP BaseStringEnumerator::GetNext(nsACString& _retval) { if (mStringCurItem >= mCount) return NS_ERROR_FAILURE; _retval = nsDependentCString(mArray[mStringCurItem++]); return NS_OK; } int BaseStringEnumerator::SortCallback(const void *e1, const void *e2, void * /*unused*/) { char const *const *s1 = reinterpret_cast(e1); char const *const *s2 = reinterpret_cast(e2); return strcmp(*s1, *s2); } void BaseStringEnumerator::Sort() { NS_QuickSort(mArray, mCount, sizeof(mArray[0]), SortCallback, nsnull); } // // EntryEnumerator is the wrapper that allows nsICategoryManager::EnumerateCategory // class EntryEnumerator : public BaseStringEnumerator { public: static EntryEnumerator* Create(nsTHashtable& aTable); private: static PLDHashOperator enumfunc_createenumerator(CategoryLeaf* aLeaf, void* userArg); }; PLDHashOperator EntryEnumerator::enumfunc_createenumerator(CategoryLeaf* aLeaf, void* userArg) { EntryEnumerator* mythis = static_cast(userArg); if (aLeaf->nonpValue) mythis->mArray[mythis->mCount++] = aLeaf->GetKey(); return PL_DHASH_NEXT; } EntryEnumerator* EntryEnumerator::Create(nsTHashtable& aTable) { EntryEnumerator* enumObj = new EntryEnumerator(); if (!enumObj) return nsnull; enumObj->mArray = new char const* [aTable.Count()]; if (!enumObj->mArray) { delete enumObj; return nsnull; } aTable.EnumerateEntries(enumfunc_createenumerator, enumObj); enumObj->Sort(); return enumObj; } // // CategoryNode implementations // CategoryNode* CategoryNode::Create(PLArenaPool* aArena) { CategoryNode* node = new(aArena) CategoryNode(); if (!node) return nsnull; if (!node->mTable.Init()) { delete node; return nsnull; } return node; } CategoryNode::~CategoryNode() { } void* CategoryNode::operator new(size_t aSize, PLArenaPool* aArena) { void* p; PL_ARENA_ALLOCATE(p, aArena, aSize); return p; } NS_METHOD CategoryNode::GetLeaf(const char* aEntryName, char** _retval) { MutexAutoLock lock(mLock); nsresult rv = NS_ERROR_NOT_AVAILABLE; CategoryLeaf* ent = mTable.GetEntry(aEntryName); // we only want the non-persistent value if (ent && ent->nonpValue) { *_retval = NS_strdup(ent->nonpValue); if (*_retval) rv = NS_OK; } return rv; } NS_METHOD CategoryNode::AddLeaf(const char* aEntryName, const char* aValue, PRBool aPersist, PRBool aReplace, char** _retval, PLArenaPool* aArena) { MutexAutoLock lock(mLock); CategoryLeaf* leaf = mTable.GetEntry(aEntryName); nsresult rv = NS_OK; if (leaf) { //if the entry was found, aReplace must be specified if (!aReplace && (leaf->nonpValue || (aPersist && leaf->pValue ))) rv = NS_ERROR_INVALID_ARG; } else { const char* arenaEntryName = ArenaStrdup(aEntryName, aArena); if (!arenaEntryName) { rv = NS_ERROR_OUT_OF_MEMORY; } else { leaf = mTable.PutEntry(arenaEntryName); if (!leaf) rv = NS_ERROR_OUT_OF_MEMORY; } } if (NS_SUCCEEDED(rv)) { const char* arenaValue = ArenaStrdup(aValue, aArena); if (!arenaValue) { rv = NS_ERROR_OUT_OF_MEMORY; } else { if (_retval) { const char *toDup = leaf->nonpValue ? leaf->nonpValue : leaf->pValue; if (toDup) { *_retval = ToNewCString(nsDependentCString(toDup)); if (!*_retval) return NS_ERROR_OUT_OF_MEMORY; } else { *_retval = nsnull; } } leaf->nonpValue = arenaValue; if (aPersist) leaf->pValue = arenaValue; } } return rv; } NS_METHOD CategoryNode::DeleteLeaf(const char* aEntryName, PRBool aDontPersist) { // we don't throw any errors, because it normally doesn't matter // and it makes JS a lot cleaner MutexAutoLock lock(mLock); if (aDontPersist) { // we can just remove the entire hash entry without introspection mTable.RemoveEntry(aEntryName); } else { // if we are keeping the persistent value, we need to look at // the contents of the current entry CategoryLeaf* leaf = mTable.GetEntry(aEntryName); if (leaf) { if (leaf->pValue) { leaf->nonpValue = nsnull; } else { // if there is no persistent value, just remove the entry mTable.RawRemoveEntry(leaf); } } } return NS_OK; } NS_METHOD CategoryNode::Enumerate(nsISimpleEnumerator **_retval) { NS_ENSURE_ARG_POINTER(_retval); MutexAutoLock lock(mLock); EntryEnumerator* enumObj = EntryEnumerator::Create(mTable); if (!enumObj) return NS_ERROR_OUT_OF_MEMORY; *_retval = enumObj; NS_ADDREF(*_retval); return NS_OK; } struct persistent_userstruct { PRFileDesc* fd; const char* categoryName; PRBool success; }; PLDHashOperator enumfunc_pentries(CategoryLeaf* aLeaf, void* userArg) { persistent_userstruct* args = static_cast(userArg); PLDHashOperator status = PL_DHASH_NEXT; if (aLeaf->pValue) { if (PR_fprintf(args->fd, "%s,%s,%s\n", args->categoryName, aLeaf->GetKey(), aLeaf->pValue) == (PRUint32) -1) { args->success = PR_FALSE; status = PL_DHASH_STOP; } } return status; } PRBool CategoryNode::WritePersistentEntries(PRFileDesc* fd, const char* aCategoryName) { persistent_userstruct args = { fd, aCategoryName, PR_TRUE }; MutexAutoLock lock(mLock); mTable.EnumerateEntries(enumfunc_pentries, &args); return args.success; } // // CategoryEnumerator class // class CategoryEnumerator : public BaseStringEnumerator { public: static CategoryEnumerator* Create(nsClassHashtable& aTable); private: static PLDHashOperator enumfunc_createenumerator(const char* aStr, CategoryNode* aNode, void* userArg); }; CategoryEnumerator* CategoryEnumerator::Create(nsClassHashtable& aTable) { CategoryEnumerator* enumObj = new CategoryEnumerator(); if (!enumObj) return nsnull; enumObj->mArray = new const char* [aTable.Count()]; if (!enumObj->mArray) { delete enumObj; return nsnull; } aTable.EnumerateRead(enumfunc_createenumerator, enumObj); return enumObj; } PLDHashOperator CategoryEnumerator::enumfunc_createenumerator(const char* aStr, CategoryNode* aNode, void* userArg) { CategoryEnumerator* mythis = static_cast(userArg); // if a category has no entries, we pretend it doesn't exist if (aNode->Count()) mythis->mArray[mythis->mCount++] = aStr; return PL_DHASH_NEXT; } // // nsCategoryManager implementations // NS_IMPL_THREADSAFE_ISUPPORTS1(nsCategoryManager, nsICategoryManager) nsCategoryManager* nsCategoryManager::Create() { nsCategoryManager* manager = new nsCategoryManager(); if (!manager) return nsnull; PL_INIT_ARENA_POOL(&(manager->mArena), "CategoryManagerArena", NS_CATEGORYMANAGER_ARENA_SIZE); // this never fails if (!manager->mTable.Init()) { delete manager; return nsnull; } return manager; } nsCategoryManager::~nsCategoryManager() { // the hashtable contains entries that must be deleted before the arena is // destroyed, or else you will have PRLocks undestroyed and other Really // Bad Stuff (TM) mTable.Clear(); PL_FinishArenaPool(&mArena); } inline CategoryNode* nsCategoryManager::get_category(const char* aName) { CategoryNode* node; if (!mTable.Get(aName, &node)) { return nsnull; } return node; } void nsCategoryManager::NotifyObservers( const char *aTopic, const char *aCategoryName, const char *aEntryName ) { if (mSuppressNotifications) return; nsCOMPtr observerService = mozilla::services::GetObserverService(); if (!observerService) return; nsCOMPtr obsProxy; NS_GetProxyForObject(NS_PROXY_TO_MAIN_THREAD, NS_GET_IID(nsIObserverService), observerService, NS_PROXY_ASYNC, getter_AddRefs(obsProxy)); if (!obsProxy) return; if (aEntryName) { nsCOMPtr entry (do_CreateInstance (NS_SUPPORTS_CSTRING_CONTRACTID)); if (!entry) return; nsresult rv = entry->SetData(nsDependentCString(aEntryName)); if (NS_FAILED(rv)) return; obsProxy->NotifyObservers(entry, aTopic, NS_ConvertUTF8toUTF16(aCategoryName).get()); } else { obsProxy->NotifyObservers(this, aTopic, NS_ConvertUTF8toUTF16(aCategoryName).get()); } } NS_IMETHODIMP nsCategoryManager::GetCategoryEntry( const char *aCategoryName, const char *aEntryName, char **_retval ) { NS_ENSURE_ARG_POINTER(aCategoryName); NS_ENSURE_ARG_POINTER(aEntryName); NS_ENSURE_ARG_POINTER(_retval); nsresult status = NS_ERROR_NOT_AVAILABLE; CategoryNode* category; { MutexAutoLock lock(mLock); category = get_category(aCategoryName); } if (category) { status = category->GetLeaf(aEntryName, _retval); } return status; } NS_IMETHODIMP nsCategoryManager::AddCategoryEntry( const char *aCategoryName, const char *aEntryName, const char *aValue, PRBool aPersist, PRBool aReplace, char **_retval ) { NS_ENSURE_ARG_POINTER(aCategoryName); NS_ENSURE_ARG_POINTER(aEntryName); NS_ENSURE_ARG_POINTER(aValue); // Before we can insert a new entry, we'll need to // find the |CategoryNode| to put it in... CategoryNode* category; { MutexAutoLock lock(mLock); category = get_category(aCategoryName); if (!category) { // That category doesn't exist yet; let's make it. category = CategoryNode::Create(&mArena); char* categoryName = ArenaStrdup(aCategoryName, &mArena); mTable.Put(categoryName, category); } } if (!category) return NS_ERROR_OUT_OF_MEMORY; // We will need the return value of AddLeaf even if the called doesn't want it char *oldEntry = nsnull; nsresult rv = category->AddLeaf(aEntryName, aValue, aPersist, aReplace, &oldEntry, &mArena); if (NS_SUCCEEDED(rv)) { if (oldEntry) { NotifyObservers(NS_XPCOM_CATEGORY_ENTRY_REMOVED_OBSERVER_ID, aCategoryName, oldEntry); } NotifyObservers(NS_XPCOM_CATEGORY_ENTRY_ADDED_OBSERVER_ID, aCategoryName, aEntryName); if (_retval) *_retval = oldEntry; else if (oldEntry) nsMemory::Free(oldEntry); } return rv; } NS_IMETHODIMP nsCategoryManager::DeleteCategoryEntry( const char *aCategoryName, const char *aEntryName, PRBool aDontPersist) { NS_ENSURE_ARG_POINTER(aCategoryName); NS_ENSURE_ARG_POINTER(aEntryName); /* Note: no errors are reported since failure to delete probably won't hurt you, and returning errors seriously inconveniences JS clients */ CategoryNode* category; { MutexAutoLock lock(mLock); category = get_category(aCategoryName); } if (!category) return NS_OK; nsresult rv = category->DeleteLeaf(aEntryName, aDontPersist); if (NS_SUCCEEDED(rv)) { NotifyObservers(NS_XPCOM_CATEGORY_ENTRY_REMOVED_OBSERVER_ID, aCategoryName, aEntryName); } return rv; } NS_IMETHODIMP nsCategoryManager::DeleteCategory( const char *aCategoryName ) { NS_ENSURE_ARG_POINTER(aCategoryName); // the categories are arena-allocated, so we don't // actually delete them. We just remove all of the // leaf nodes. CategoryNode* category; { MutexAutoLock lock(mLock); category = get_category(aCategoryName); } if (category) { category->Clear(); NotifyObservers(NS_XPCOM_CATEGORY_CLEARED_OBSERVER_ID, aCategoryName, nsnull); } return NS_OK; } NS_IMETHODIMP nsCategoryManager::EnumerateCategory( const char *aCategoryName, nsISimpleEnumerator **_retval ) { NS_ENSURE_ARG_POINTER(aCategoryName); NS_ENSURE_ARG_POINTER(_retval); CategoryNode* category; { MutexAutoLock lock(mLock); category = get_category(aCategoryName); } if (!category) { return NS_NewEmptyEnumerator(_retval); } return category->Enumerate(_retval); } NS_IMETHODIMP nsCategoryManager::EnumerateCategories(nsISimpleEnumerator **_retval) { NS_ENSURE_ARG_POINTER(_retval); MutexAutoLock lock(mLock); CategoryEnumerator* enumObj = CategoryEnumerator::Create(mTable); if (!enumObj) return NS_ERROR_OUT_OF_MEMORY; *_retval = enumObj; NS_ADDREF(*_retval); return NS_OK; } struct writecat_struct { PRFileDesc* fd; PRBool success; }; PLDHashOperator enumfunc_categories(const char* aKey, CategoryNode* aCategory, void* userArg) { writecat_struct* args = static_cast(userArg); PLDHashOperator result = PL_DHASH_NEXT; if (!aCategory->WritePersistentEntries(args->fd, aKey)) { args->success = PR_FALSE; result = PL_DHASH_STOP; } return result; } NS_METHOD nsCategoryManager::WriteCategoryManagerToRegistry(PRFileDesc* fd) { writecat_struct args = { fd, PR_TRUE }; MutexAutoLock lock(mLock); mTable.EnumerateRead(enumfunc_categories, &args); if (!args.success) { return NS_ERROR_UNEXPECTED; } return NS_OK; } NS_METHOD nsCategoryManager::SuppressNotifications(PRBool aSuppress) { mSuppressNotifications = aSuppress; return NS_OK; } class nsCategoryManagerFactory : public nsIFactory { public: nsCategoryManagerFactory() { } NS_DECL_ISUPPORTS NS_DECL_NSIFACTORY }; NS_IMPL_ISUPPORTS1(nsCategoryManagerFactory, nsIFactory) NS_IMETHODIMP nsCategoryManagerFactory::CreateInstance( nsISupports* aOuter, const nsIID& aIID, void** aResult ) { NS_ENSURE_ARG_POINTER(aResult); *aResult = 0; nsresult status = NS_OK; if ( aOuter ) status = NS_ERROR_NO_AGGREGATION; else { nsCategoryManager* raw_category_manager = nsCategoryManager::Create(); nsCOMPtr new_category_manager = raw_category_manager; if ( new_category_manager ) status = new_category_manager->QueryInterface(aIID, aResult); else status = NS_ERROR_OUT_OF_MEMORY; } return status; } NS_IMETHODIMP nsCategoryManagerFactory::LockFactory( PRBool ) { // Not implemented... return NS_OK; } nsresult NS_CategoryManagerGetFactory( nsIFactory** aFactory ) { // assert(aFactory); nsresult status; *aFactory = 0; nsIFactory* new_factory = static_cast(new nsCategoryManagerFactory); if (new_factory) { *aFactory = new_factory; NS_ADDREF(*aFactory); status = NS_OK; } else status = NS_ERROR_OUT_OF_MEMORY; return status; } /* * CreateServicesFromCategory() * * Given a category, this convenience functions enumerates the category and * creates a service of every CID or ContractID registered under the category. * If observerTopic is non null and the service implements nsIObserver, * this will attempt to notify the observer with the origin, observerTopic string * as parameter. */ NS_COM nsresult NS_CreateServicesFromCategory(const char *category, nsISupports *origin, const char *observerTopic) { NS_TIME_FUNCTION_FMT("NS_CreateServicesFromCategory: %s (%s)", category, observerTopic ? observerTopic : "(no topic)"); nsresult rv = NS_OK; int nFailed = 0; nsCOMPtr categoryManager = do_GetService("@mozilla.org/categorymanager;1", &rv); if (!categoryManager) return rv; nsCOMPtr enumerator; rv = categoryManager->EnumerateCategory(category, getter_AddRefs(enumerator)); if (NS_FAILED(rv)) return rv; nsCOMPtr entry; while (NS_SUCCEEDED(enumerator->GetNext(getter_AddRefs(entry)))) { // From here on just skip any error we get. nsCOMPtr catEntry = do_QueryInterface(entry, &rv); if (NS_FAILED(rv)) { nFailed++; continue; } nsCAutoString entryString; rv = catEntry->GetData(entryString); if (NS_FAILED(rv)) { nFailed++; continue; } nsXPIDLCString contractID; rv = categoryManager->GetCategoryEntry(category,entryString.get(), getter_Copies(contractID)); if (NS_FAILED(rv)) { nFailed++; continue; } nsCOMPtr instance = do_GetService(contractID, &rv); if (NS_FAILED(rv)) { nFailed++; continue; } NS_TIME_FUNCTION_MARK("service: %s", nsPromiseFlatCString(contractID).get()); if (observerTopic) { // try an observer, if it implements it. nsCOMPtr observer = do_QueryInterface(instance, &rv); if (NS_SUCCEEDED(rv) && observer) observer->Observe(origin, observerTopic, EmptyString().get()); NS_TIME_FUNCTION_MARK(" & observe %s", observerTopic); } } return (nFailed ? NS_ERROR_FAILURE : NS_OK); }