/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ /* ***** BEGIN LICENSE BLOCK ***** * Version: NPL 1.1/GPL 2.0/LGPL 2.1 * * The contents of this file are subject to the Netscape Public License * Version 1.1 (the "License"); you may not use this file except in * compliance with the License. You may obtain a copy of the License at * http://www.mozilla.org/NPL/ * * Software distributed under the License is distributed on an "AS IS" basis, * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License * for the specific language governing rights and limitations under the * License. * * The Original Code is mozilla.org code. * * The Initial Developer of the Original Code is * Netscape Communications Corporation. * Portions created by the Initial Developer are Copyright (C) 1998 * the Initial Developer. All Rights Reserved. * * Contributor(s): * IBM Corp. * * Alternatively, the contents of this file may be used under the terms of * either the GNU General Public License Version 2 or later (the "GPL"), or * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), * in which case the provisions of the GPL or the LGPL are applicable instead * of those above. If you wish to allow use of your version of this file only * under the terms of either the GPL or the LGPL, and not to allow others to * use your version of this file under the terms of the NPL, 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 NPL, the GPL or the LGPL. * * ***** END LICENSE BLOCK ***** */ #include "nsCOMPtr.h" #include "nsDirectoryService.h" #include "nsDirectoryServiceDefs.h" #include "nsLocalFile.h" #include "nsDebug.h" #include "nsStaticAtom.h" #if defined(XP_MAC) #include #include #include #include #include #elif defined(XP_WIN) #include #include #include #include #elif defined(XP_UNIX) || defined(XP_MACOSX) #include #include #include #include "prenv.h" #ifdef XP_MACOSX #include #include #include #include #include #include #include #include #endif #elif defined(XP_OS2) #define MAX_PATH _MAX_PATH #elif defined(XP_BEOS) #include #include #include #include #include #include #include #include "prenv.h" #endif #include "SpecialSystemDirectory.h" #include "nsAppFileLocationProvider.h" #if defined(XP_MAC) #define COMPONENT_REGISTRY_NAME NS_LITERAL_CSTRING("Component Registry") #define COMPONENT_DIRECTORY NS_LITERAL_CSTRING("Components") #else #define COMPONENT_REGISTRY_NAME NS_LITERAL_CSTRING("compreg.dat") #define COMPONENT_DIRECTORY NS_LITERAL_CSTRING("components") #endif #define XPTI_REGISTRY_NAME NS_LITERAL_CSTRING("xpti.dat") // define home directory // For Windows platform, We are choosing Appdata folder as HOME #if defined (XP_WIN) #define HOME_DIR NS_WIN_APPDATA_DIR #elif defined (XP_MAC) || defined (XP_MACOSX) #define HOME_DIR NS_OSX_HOME_DIR #elif defined (XP_UNIX) #define HOME_DIR NS_UNIX_HOME_DIR #elif defined (XP_OS2) #define HOME_DIR NS_OS2_HOME_DIR #elif defined (XP_BEOS) #define HOME_DIR NS_BEOS_HOME_DIR #endif //---------------------------------------------------------------------------------------- nsresult nsDirectoryService::GetCurrentProcessDirectory(nsILocalFile** aFile) //---------------------------------------------------------------------------------------- { NS_ENSURE_ARG_POINTER(aFile); *aFile = nsnull; // Set the component registry location: if (!mService) return NS_ERROR_FAILURE; nsresult rv; nsCOMPtr dirService; rv = nsDirectoryService::Create(nsnull, NS_GET_IID(nsIProperties), getter_AddRefs(dirService)); // needs to be around for life of product if (dirService) { nsCOMPtr aLocalFile; dirService->Get(NS_XPCOM_INIT_CURRENT_PROCESS_DIR, NS_GET_IID(nsILocalFile), getter_AddRefs(aLocalFile)); if (aLocalFile) { *aFile = aLocalFile; NS_ADDREF(*aFile); return NS_OK; } } nsLocalFile* localFile = new nsLocalFile; if (localFile == nsnull) return NS_ERROR_OUT_OF_MEMORY; NS_ADDREF(localFile); #ifdef XP_WIN char buf[MAX_PATH]; if ( ::GetModuleFileName(0, buf, sizeof(buf)) ) { // chop of the executable name by finding the rightmost backslash char* lastSlash = PL_strrchr(buf, '\\'); if (lastSlash) *(lastSlash + 1) = '\0'; localFile->InitWithNativePath(nsDependentCString(buf)); *aFile = localFile; return NS_OK; } #elif defined(XP_MAC) // get info for the the current process to determine the directory // its located in OSErr err; ProcessSerialNumber psn = {kNoProcess, kCurrentProcess}; ProcessInfoRec pInfo; FSSpec tempSpec; // initialize ProcessInfoRec before calling // GetProcessInformation() or die horribly. pInfo.processName = nil; pInfo.processAppSpec = &tempSpec; pInfo.processInfoLength = sizeof(ProcessInfoRec); err = GetProcessInformation(&psn, &pInfo); if (!err) { // create an FSSpec from the volume and dirid of the app. FSSpec appFSSpec; ::FSMakeFSSpec(pInfo.processAppSpec->vRefNum, pInfo.processAppSpec->parID, 0, &appFSSpec); nsCOMPtr localFileMac = do_QueryInterface((nsIFile*)localFile); if (localFileMac) { localFileMac->InitWithFSSpec(&appFSSpec); *aFile = localFile; return NS_OK; } } #elif defined(XP_MACOSX) // Works even if we're not bundled. CFBundleRef appBundle = CFBundleGetMainBundle(); if (appBundle != nsnull) { CFURLRef bundleURL = CFBundleCopyExecutableURL(appBundle); if (bundleURL != nsnull) { CFURLRef parentURL = CFURLCreateCopyDeletingLastPathComponent(kCFAllocatorDefault, bundleURL); if (parentURL) { // Pass PR_TRUE for the "resolveAgainstBase" arg to CFURLGetFileSystemRepresentation. // This will resolve the relative portion of the CFURL against it base, giving a full // path, which CFURLCopyFileSystemPath doesn't do. char buffer[PATH_MAX]; if (CFURLGetFileSystemRepresentation(parentURL, PR_TRUE, (UInt8 *)buffer, sizeof(buffer))) { #ifdef DEBUG_conrad printf("nsDirectoryService - CurrentProcessDir is: %s\n", buffer); #endif rv = localFile->InitWithNativePath(nsDependentCString(buffer)); if (NS_SUCCEEDED(rv)) *aFile = localFile; } CFRelease(parentURL); } CFRelease(bundleURL); } } NS_ASSERTION(*aFile, "nsDirectoryService - Could not determine CurrentProcessDir.\n"); if (*aFile) return NS_OK; #elif defined(XP_UNIX) // In the absence of a good way to get the executable directory let // us try this for unix: // - if MOZILLA_FIVE_HOME is defined, that is it // - else give the current directory char buf[MAXPATHLEN]; // The MOZ_DEFAULT_MOZILLA_FIVE_HOME variable can be set at configure time with // a --with-default-mozilla-five-home=foo autoconf flag. // // The idea here is to allow for builds that have a default MOZILLA_FIVE_HOME // regardless of the environment. This makes it easier to write apps that // embed mozilla without having to worry about setting up the environment // // We do this py putenv()ing the default value into the environment. Note that // we only do this if it is not already set. #ifdef MOZ_DEFAULT_MOZILLA_FIVE_HOME if (PR_GetEnv("MOZILLA_FIVE_HOME") == nsnull) { putenv("MOZILLA_FIVE_HOME=" MOZ_DEFAULT_MOZILLA_FIVE_HOME); } #endif char *moz5 = PR_GetEnv("MOZILLA_FIVE_HOME"); if (moz5) { localFile->InitWithNativePath(nsDependentCString(moz5)); localFile->Normalize(); *aFile = localFile; return NS_OK; } else { #if defined(DEBUG) static PRBool firstWarning = PR_TRUE; if(firstWarning) { // Warn that MOZILLA_FIVE_HOME not set, once. printf("Warning: MOZILLA_FIVE_HOME not set.\n"); firstWarning = PR_FALSE; } #endif /* DEBUG */ // Fall back to current directory. if (getcwd(buf, sizeof(buf))) { localFile->InitWithNativePath(nsDependentCString(buf)); *aFile = localFile; return NS_OK; } } #elif defined(XP_OS2) PPIB ppib; PTIB ptib; char buffer[CCHMAXPATH]; DosGetInfoBlocks( &ptib, &ppib); DosQueryModuleName( ppib->pib_hmte, CCHMAXPATH, buffer); *strrchr( buffer, '\\') = '\0'; // XXX DBCS misery localFile->InitWithNativePath(nsDependentCString(buffer)); *aFile = localFile; return NS_OK; #elif defined(XP_BEOS) char *moz5 = getenv("MOZILLA_FIVE_HOME"); if (moz5) { localFile->InitWithNativePath(nsDependentCString(moz5)); localFile->Normalize(); *aFile = localFile; return NS_OK; } else { static char buf[MAXPATHLEN]; int32 cookie = 0; image_info info; char *p; *buf = 0; if(get_next_image_info(0, &cookie, &info) == B_OK) { strcpy(buf, info.name); if((p = strrchr(buf, '/')) != 0) { *p = 0; localFile->InitWithNativePath(nsDependentCString(buf)); *aFile = localFile; return NS_OK; } } } #endif if (localFile) delete localFile; NS_ERROR("unable to get current process directory"); return NS_ERROR_FAILURE; } // GetCurrentProcessDirectory() nsIAtom* nsDirectoryService::sCurrentProcess = nsnull; nsIAtom* nsDirectoryService::sComponentRegistry = nsnull; nsIAtom* nsDirectoryService::sXPTIRegistry = nsnull; nsIAtom* nsDirectoryService::sComponentDirectory = nsnull; nsIAtom* nsDirectoryService::sGRE_Directory = nsnull; nsIAtom* nsDirectoryService::sGRE_ComponentDirectory = nsnull; nsIAtom* nsDirectoryService::sOS_DriveDirectory = nsnull; nsIAtom* nsDirectoryService::sOS_TemporaryDirectory = nsnull; nsIAtom* nsDirectoryService::sOS_CurrentProcessDirectory = nsnull; nsIAtom* nsDirectoryService::sOS_CurrentWorkingDirectory = nsnull; #if defined (XP_MACOSX) nsIAtom* nsDirectoryService::sDirectory = nsnull; nsIAtom* nsDirectoryService::sDesktopDirectory = nsnull; nsIAtom* nsDirectoryService::sTrashDirectory = nsnull; nsIAtom* nsDirectoryService::sStartupDirectory = nsnull; nsIAtom* nsDirectoryService::sShutdownDirectory = nsnull; nsIAtom* nsDirectoryService::sAppleMenuDirectory = nsnull; nsIAtom* nsDirectoryService::sControlPanelDirectory = nsnull; nsIAtom* nsDirectoryService::sExtensionDirectory = nsnull; nsIAtom* nsDirectoryService::sFontsDirectory = nsnull; nsIAtom* nsDirectoryService::sPreferencesDirectory = nsnull; nsIAtom* nsDirectoryService::sDocumentsDirectory = nsnull; nsIAtom* nsDirectoryService::sInternetSearchDirectory = nsnull; nsIAtom* nsDirectoryService::sUserLibDirectory = nsnull; nsIAtom* nsDirectoryService::sHomeDirectory = nsnull; nsIAtom* nsDirectoryService::sDefaultDownloadDirectory = nsnull; nsIAtom* nsDirectoryService::sUserDesktopDirectory = nsnull; nsIAtom* nsDirectoryService::sLocalDesktopDirectory = nsnull; nsIAtom* nsDirectoryService::sUserApplicationsDirectory = nsnull; nsIAtom* nsDirectoryService::sLocalApplicationsDirectory = nsnull; nsIAtom* nsDirectoryService::sUserDocumentsDirectory = nsnull; nsIAtom* nsDirectoryService::sLocalDocumentsDirectory = nsnull; nsIAtom* nsDirectoryService::sUserInternetPlugInDirectory = nsnull; nsIAtom* nsDirectoryService::sLocalInternetPlugInDirectory = nsnull; nsIAtom* nsDirectoryService::sUserFrameworksDirectory = nsnull; nsIAtom* nsDirectoryService::sLocalFrameworksDirectory = nsnull; nsIAtom* nsDirectoryService::sUserPreferencesDirectory = nsnull; nsIAtom* nsDirectoryService::sLocalPreferencesDirectory = nsnull; nsIAtom* nsDirectoryService::sPictureDocumentsDirectory = nsnull; nsIAtom* nsDirectoryService::sMovieDocumentsDirectory = nsnull; nsIAtom* nsDirectoryService::sMusicDocumentsDirectory = nsnull; nsIAtom* nsDirectoryService::sInternetSitesDirectory = nsnull; #elif defined (XP_WIN) nsIAtom* nsDirectoryService::sSystemDirectory = nsnull; nsIAtom* nsDirectoryService::sWindowsDirectory = nsnull; nsIAtom* nsDirectoryService::sHomeDirectory = nsnull; nsIAtom* nsDirectoryService::sDesktop = nsnull; nsIAtom* nsDirectoryService::sPrograms = nsnull; nsIAtom* nsDirectoryService::sControls = nsnull; nsIAtom* nsDirectoryService::sPrinters = nsnull; nsIAtom* nsDirectoryService::sPersonal = nsnull; nsIAtom* nsDirectoryService::sFavorites = nsnull; nsIAtom* nsDirectoryService::sStartup = nsnull; nsIAtom* nsDirectoryService::sRecent = nsnull; nsIAtom* nsDirectoryService::sSendto = nsnull; nsIAtom* nsDirectoryService::sBitbucket = nsnull; nsIAtom* nsDirectoryService::sStartmenu = nsnull; nsIAtom* nsDirectoryService::sDesktopdirectory = nsnull; nsIAtom* nsDirectoryService::sDrives = nsnull; nsIAtom* nsDirectoryService::sNetwork = nsnull; nsIAtom* nsDirectoryService::sNethood = nsnull; nsIAtom* nsDirectoryService::sFonts = nsnull; nsIAtom* nsDirectoryService::sTemplates = nsnull; nsIAtom* nsDirectoryService::sCommon_Startmenu = nsnull; nsIAtom* nsDirectoryService::sCommon_Programs = nsnull; nsIAtom* nsDirectoryService::sCommon_Startup = nsnull; nsIAtom* nsDirectoryService::sCommon_Desktopdirectory = nsnull; nsIAtom* nsDirectoryService::sAppdata = nsnull; nsIAtom* nsDirectoryService::sPrinthood = nsnull; nsIAtom* nsDirectoryService::sWinCookiesDirectory = nsnull; #elif defined (XP_UNIX) nsIAtom* nsDirectoryService::sLocalDirectory = nsnull; nsIAtom* nsDirectoryService::sLibDirectory = nsnull; nsIAtom* nsDirectoryService::sHomeDirectory = nsnull; #elif defined (XP_OS2) nsIAtom* nsDirectoryService::sSystemDirectory = nsnull; nsIAtom* nsDirectoryService::sOS2Directory = nsnull; nsIAtom* nsDirectoryService::sHomeDirectory = nsnull; nsIAtom* nsDirectoryService::sDesktopDirectory = nsnull; #elif defined (XP_BEOS) nsIAtom* nsDirectoryService::sSettingsDirectory = nsnull; nsIAtom* nsDirectoryService::sHomeDirectory = nsnull; nsIAtom* nsDirectoryService::sDesktopDirectory = nsnull; nsIAtom* nsDirectoryService::sSystemDirectory = nsnull; #endif nsDirectoryService* nsDirectoryService::mService = nsnull; nsDirectoryService::nsDirectoryService() : mHashtable(256, PR_TRUE) { } NS_METHOD nsDirectoryService::Create(nsISupports *outer, REFNSIID aIID, void **aResult) { NS_ENSURE_ARG_POINTER(aResult); if (!mService) { mService = new nsDirectoryService(); if (!mService) return NS_ERROR_OUT_OF_MEMORY; } return mService->QueryInterface(aIID, aResult); } static const nsStaticAtom directory_atoms[] = { { NS_XPCOM_CURRENT_PROCESS_DIR, &nsDirectoryService::sCurrentProcess }, { NS_XPCOM_COMPONENT_REGISTRY_FILE, &nsDirectoryService::sComponentRegistry }, { NS_XPCOM_COMPONENT_DIR, &nsDirectoryService::sComponentDirectory }, { NS_XPCOM_XPTI_REGISTRY_FILE, &nsDirectoryService::sXPTIRegistry }, { NS_GRE_DIR, &nsDirectoryService::sGRE_Directory }, { NS_GRE_COMPONENT_DIR, &nsDirectoryService::sGRE_ComponentDirectory }, { NS_OS_DRIVE_DIR, &nsDirectoryService::sOS_DriveDirectory }, { NS_OS_TEMP_DIR, &nsDirectoryService::sOS_TemporaryDirectory }, { NS_OS_CURRENT_PROCESS_DIR, &nsDirectoryService::sOS_CurrentProcessDirectory }, { NS_OS_CURRENT_WORKING_DIR, &nsDirectoryService::sOS_CurrentWorkingDirectory }, { NS_XPCOM_INIT_CURRENT_PROCESS_DIR, nsnull }, #if defined (XP_MACOSX) { NS_OS_SYSTEM_DIR, &nsDirectoryService::sDirectory }, { NS_MAC_DESKTOP_DIR, &nsDirectoryService::sDesktopDirectory }, { NS_MAC_TRASH_DIR, &nsDirectoryService::sTrashDirectory }, { NS_MAC_STARTUP_DIR, &nsDirectoryService::sStartupDirectory }, { NS_MAC_SHUTDOWN_DIR, &nsDirectoryService::sShutdownDirectory }, { NS_MAC_APPLE_MENU_DIR, &nsDirectoryService::sAppleMenuDirectory }, { NS_MAC_CONTROL_PANELS_DIR, &nsDirectoryService::sControlPanelDirectory }, { NS_MAC_EXTENSIONS_DIR, &nsDirectoryService::sExtensionDirectory }, { NS_MAC_FONTS_DIR, &nsDirectoryService::sFontsDirectory }, { NS_MAC_PREFS_DIR, &nsDirectoryService::sPreferencesDirectory }, { NS_MAC_DOCUMENTS_DIR, &nsDirectoryService::sDocumentsDirectory }, { NS_MAC_INTERNET_SEARCH_DIR, &nsDirectoryService::sInternetSearchDirectory }, { NS_MAC_USER_LIB_DIR, &nsDirectoryService::sUserLibDirectory }, { NS_OSX_HOME_DIR, &nsDirectoryService::sHomeDirectory }, { NS_OSX_DEFAULT_DOWNLOAD_DIR, &nsDirectoryService::sDefaultDownloadDirectory }, { NS_OSX_USER_DESKTOP_DIR, &nsDirectoryService::sUserDesktopDirectory }, { NS_OSX_LOCAL_DESKTOP_DIR, &nsDirectoryService::sLocalDesktopDirectory }, { NS_OSX_USER_APPLICATIONS_DIR, &nsDirectoryService::sUserApplicationsDirectory }, { NS_OSX_LOCAL_APPLICATIONS_DIR, &nsDirectoryService::sLocalApplicationsDirectory }, { NS_OSX_USER_DOCUMENTS_DIR, &nsDirectoryService::sUserDocumentsDirectory }, { NS_OSX_LOCAL_DOCUMENTS_DIR, &nsDirectoryService::sLocalDocumentsDirectory }, { NS_OSX_USER_INTERNET_PLUGIN_DIR, &nsDirectoryService::sUserInternetPlugInDirectory }, { NS_OSX_LOCAL_INTERNET_PLUGIN_DIR, &nsDirectoryService::sLocalInternetPlugInDirectory }, { NS_OSX_USER_FRAMEWORKS_DIR, &nsDirectoryService::sUserFrameworksDirectory }, { NS_OSX_LOCAL_FRAMEWORKS_DIR, &nsDirectoryService::sLocalFrameworksDirectory }, { NS_OSX_USER_PREFERENCES_DIR, &nsDirectoryService::sUserPreferencesDirectory }, { NS_OSX_LOCAL_PREFERENCES_DIR, &nsDirectoryService::sLocalPreferencesDirectory }, { NS_OSX_PICTURE_DOCUMENTS_DIR, &nsDirectoryService::sPictureDocumentsDirectory }, { NS_OSX_MOVIE_DOCUMENTS_DIR, &nsDirectoryService::sMovieDocumentsDirectory }, { NS_OSX_MUSIC_DOCUMENTS_DIR, &nsDirectoryService::sMusicDocumentsDirectory }, { NS_OSX_INTERNET_SITES_DIR, &nsDirectoryService::sInternetSitesDirectory }, #elif defined (XP_WIN) { NS_OS_SYSTEM_DIR, &nsDirectoryService::sSystemDirectory }, { NS_WIN_WINDOWS_DIR, &nsDirectoryService::sWindowsDirectory }, { NS_WIN_HOME_DIR, &nsDirectoryService::sHomeDirectory }, { NS_WIN_DESKTOP_DIR, &nsDirectoryService::sDesktop }, { NS_WIN_PROGRAMS_DIR, &nsDirectoryService::sPrograms }, { NS_WIN_CONTROLS_DIR, &nsDirectoryService::sControls }, { NS_WIN_PRINTERS_DIR, &nsDirectoryService::sPrinters }, { NS_WIN_PERSONAL_DIR, &nsDirectoryService::sPersonal }, { NS_WIN_FAVORITES_DIR, &nsDirectoryService::sFavorites }, { NS_WIN_STARTUP_DIR, &nsDirectoryService::sStartup }, { NS_WIN_RECENT_DIR, &nsDirectoryService::sRecent }, { NS_WIN_SEND_TO_DIR, &nsDirectoryService::sSendto }, { NS_WIN_BITBUCKET_DIR, &nsDirectoryService::sBitbucket }, { NS_WIN_STARTMENU_DIR, &nsDirectoryService::sStartmenu }, { NS_WIN_DESKTOP_DIRECTORY, &nsDirectoryService::sDesktopdirectory }, { NS_WIN_DRIVES_DIR, &nsDirectoryService::sDrives }, { NS_WIN_NETWORK_DIR, &nsDirectoryService::sNetwork }, { NS_WIN_NETHOOD_DIR, &nsDirectoryService::sNethood }, { NS_WIN_FONTS_DIR, &nsDirectoryService::sFonts }, { NS_WIN_TEMPLATES_DIR, &nsDirectoryService::sTemplates }, { NS_WIN_COMMON_STARTMENU_DIR, &nsDirectoryService::sCommon_Startmenu }, { NS_WIN_COMMON_PROGRAMS_DIR, &nsDirectoryService::sCommon_Programs }, { NS_WIN_COMMON_STARTUP_DIR, &nsDirectoryService::sCommon_Startup }, { NS_WIN_COMMON_DESKTOP_DIRECTORY, &nsDirectoryService::sCommon_Desktopdirectory }, { NS_WIN_APPDATA_DIR, &nsDirectoryService::sAppdata }, { NS_WIN_PRINTHOOD, &nsDirectoryService::sPrinthood }, { NS_WIN_COOKIES_DIR, &nsDirectoryService::sWinCookiesDirectory }, #elif defined (XP_UNIX) { NS_UNIX_LOCAL_DIR, &nsDirectoryService::sLocalDirectory }, { NS_UNIX_LIB_DIR, &nsDirectoryService::sLibDirectory }, { NS_UNIX_HOME_DIR, &nsDirectoryService::sHomeDirectory }, #elif defined (XP_OS2) { NS_OS_SYSTEM_DIR, &nsDirectoryService::sSystemDirectory }, { NS_OS2_DIR, &nsDirectoryService::sOS2Directory }, { NS_OS2_HOME_DIR, &nsDirectoryService::sHomeDirectory }, { NS_OS2_DESKTOP_DIR, &nsDirectoryService::sDesktopDirectory }, #elif defined (XP_BEOS) { NS_OS_SYSTEM_DIR, &nsDirectoryService::sSystemDirectory }, { NS_BEOS_SETTINGS_DIR, &nsDirectoryService::sSettingsDirectory }, { NS_BEOS_HOME_DIR, &nsDirectoryService::sHomeDirectory }, { NS_BEOS_DESKTOP_DIR, &nsDirectoryService::sDesktopDirectory }, #endif }; nsresult nsDirectoryService::Init() { nsresult rv; rv = NS_NewISupportsArray(getter_AddRefs(mProviders)); if (NS_FAILED(rv)) return rv; NS_RegisterStaticAtoms(directory_atoms, NS_ARRAY_LENGTH(directory_atoms)); // Let the list hold the only reference to the provider. nsAppFileLocationProvider *defaultProvider = new nsAppFileLocationProvider; if (!defaultProvider) return NS_ERROR_OUT_OF_MEMORY; // AppendElement returns PR_TRUE for success. rv = mProviders->AppendElement(defaultProvider) ? NS_OK : NS_ERROR_FAILURE; return rv; } PRBool nsDirectoryService::ReleaseValues(nsHashKey* key, void* data, void* closure) { nsISupports* value = (nsISupports*)data; NS_IF_RELEASE(value); return PR_TRUE; } nsDirectoryService::~nsDirectoryService() { // clear the global mService = nsnull; } NS_IMPL_THREADSAFE_ISUPPORTS4(nsDirectoryService, nsIProperties, nsIDirectoryService, nsIDirectoryServiceProvider, nsIDirectoryServiceProvider2) NS_IMETHODIMP nsDirectoryService::Undefine(const char* prop) { nsCStringKey key(prop); if (!mHashtable.Exists(&key)) return NS_ERROR_FAILURE; mHashtable.Remove (&key); return NS_OK; } NS_IMETHODIMP nsDirectoryService::GetKeys(PRUint32 *count, char ***keys) { return NS_ERROR_NOT_IMPLEMENTED; } struct FileData { FileData(const char* aProperty, const nsIID& aUUID) : property(aProperty), data(nsnull), persistent(PR_TRUE), uuid(aUUID) {} const char* property; nsISupports* data; PRBool persistent; const nsIID& uuid; }; static PRBool FindProviderFile(nsISupports* aElement, void *aData) { nsresult rv; FileData* fileData = (FileData*)aData; if (fileData->uuid.Equals(NS_GET_IID(nsISimpleEnumerator))) { // Not all providers implement this iface nsCOMPtr prov2 = do_QueryInterface(aElement); if (prov2) { rv = prov2->GetFiles(fileData->property, (nsISimpleEnumerator **)&fileData->data); if (NS_SUCCEEDED(rv) && fileData->data) { fileData->persistent = PR_FALSE; // Enumerators can never be peristent return PR_FALSE; } } } else { nsCOMPtr prov = do_QueryInterface(aElement); if (!prov) return PR_FALSE; rv = prov->GetFile(fileData->property, &fileData->persistent, (nsIFile **)&fileData->data); if (NS_SUCCEEDED(rv) && fileData->data) return PR_FALSE; } return PR_TRUE; } NS_IMETHODIMP nsDirectoryService::Get(const char* prop, const nsIID & uuid, void* *result) { nsCStringKey key(prop); nsCOMPtr value = dont_AddRef(mHashtable.Get(&key)); if (value) { nsCOMPtr cloneFile; nsCOMPtr cachedFile = do_QueryInterface(value); NS_ASSERTION(cachedFile, "nsIFile expected"); cachedFile->Clone(getter_AddRefs(cloneFile)); return cloneFile->QueryInterface(uuid, result); } // it is not one of our defaults, lets check any providers FileData fileData(prop, uuid); mProviders->EnumerateBackwards(FindProviderFile, &fileData); if (fileData.data) { if (fileData.persistent) { Set(prop, NS_STATIC_CAST(nsIFile*, fileData.data)); } nsresult rv = (fileData.data)->QueryInterface(uuid, result); NS_RELEASE(fileData.data); // addref occurs in FindProviderFile() return rv; } FindProviderFile(NS_STATIC_CAST(nsIDirectoryServiceProvider*, this), &fileData); if (fileData.data) { if (fileData.persistent) { Set(prop, NS_STATIC_CAST(nsIFile*, fileData.data)); } nsresult rv = (fileData.data)->QueryInterface(uuid, result); NS_RELEASE(fileData.data); // addref occurs in FindProviderFile() return rv; } return NS_ERROR_FAILURE; } NS_IMETHODIMP nsDirectoryService::Set(const char* prop, nsISupports* value) { nsCStringKey key(prop); if (mHashtable.Exists(&key) || value == nsnull) return NS_ERROR_FAILURE; nsCOMPtr ourFile; value->QueryInterface(NS_GET_IID(nsIFile), getter_AddRefs(ourFile)); if (ourFile) { nsCOMPtr cloneFile; ourFile->Clone (getter_AddRefs (cloneFile)); mHashtable.Put(&key, cloneFile); return NS_OK; } return NS_ERROR_FAILURE; } NS_IMETHODIMP nsDirectoryService::Has(const char *prop, PRBool *_retval) { *_retval = PR_FALSE; nsCOMPtr value; nsresult rv = Get(prop, NS_GET_IID(nsIFile), getter_AddRefs(value)); if (NS_FAILED(rv)) return rv; if (value) { *_retval = PR_TRUE; } return rv; } NS_IMETHODIMP nsDirectoryService::RegisterProvider(nsIDirectoryServiceProvider *prov) { nsresult rv; if (!prov) return NS_ERROR_FAILURE; if (!mProviders) return NS_ERROR_NOT_INITIALIZED; nsCOMPtr supports = do_QueryInterface(prov, &rv); if (NS_FAILED(rv)) return rv; // AppendElement returns PR_TRUE for success. return mProviders->AppendElement(supports) ? NS_OK : NS_ERROR_FAILURE; } NS_IMETHODIMP nsDirectoryService::UnregisterProvider(nsIDirectoryServiceProvider *prov) { nsresult rv; if (!prov) return NS_ERROR_FAILURE; if (!mProviders) return NS_ERROR_NOT_INITIALIZED; nsCOMPtr supports = do_QueryInterface(prov, &rv); if (NS_FAILED(rv)) return rv; // RemoveElement returns PR_TRUE for success. return mProviders->RemoveElement(supports) ? NS_OK : NS_ERROR_FAILURE; } // DO NOT ADD ANY LOCATIONS TO THIS FUNCTION UNTIL YOU TALK TO: dougt@netscape.com. // This is meant to be a place of xpcom or system specific file locations, not // application specific locations. If you need the later, register a callback for // your application. NS_IMETHODIMP nsDirectoryService::GetFile(const char *prop, PRBool *persistent, nsIFile **_retval) { nsCOMPtr localFile; nsresult rv = NS_ERROR_FAILURE; *_retval = nsnull; *persistent = PR_TRUE; nsIAtom* inAtom = NS_NewAtom(prop); // check to see if it is one of our defaults if (inAtom == nsDirectoryService::sCurrentProcess || inAtom == nsDirectoryService::sOS_CurrentProcessDirectory ) { rv = GetCurrentProcessDirectory(getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sComponentRegistry) { rv = GetCurrentProcessDirectory(getter_AddRefs(localFile)); if (!localFile) return NS_ERROR_FAILURE; localFile->AppendNative(COMPONENT_DIRECTORY); localFile->AppendNative(COMPONENT_REGISTRY_NAME); } else if (inAtom == nsDirectoryService::sXPTIRegistry) { rv = GetCurrentProcessDirectory(getter_AddRefs(localFile)); if (!localFile) return NS_ERROR_FAILURE; localFile->AppendNative(COMPONENT_DIRECTORY); localFile->AppendNative(XPTI_REGISTRY_NAME); } // Unless otherwise set, the core pieces of the GRE exist // in the current process directory. else if (inAtom == nsDirectoryService::sGRE_Directory) { rv = GetCurrentProcessDirectory(getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sGRE_ComponentDirectory) { rv = GetCurrentProcessDirectory(getter_AddRefs(localFile)); if (localFile) localFile->AppendNative(COMPONENT_DIRECTORY); } else if (inAtom == nsDirectoryService::sComponentDirectory) { rv = GetCurrentProcessDirectory(getter_AddRefs(localFile)); if (localFile) localFile->AppendNative(COMPONENT_DIRECTORY); } else if (inAtom == nsDirectoryService::sOS_DriveDirectory) { rv = GetSpecialSystemDirectory(OS_DriveDirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sOS_TemporaryDirectory) { rv = GetSpecialSystemDirectory(OS_TemporaryDirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sOS_CurrentProcessDirectory) { rv = GetSpecialSystemDirectory(OS_CurrentProcessDirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sOS_CurrentWorkingDirectory) { rv = GetSpecialSystemDirectory(OS_CurrentWorkingDirectory, getter_AddRefs(localFile)); } #if defined(XP_MACOSX) else if (inAtom == nsDirectoryService::sDirectory) { rv = GetOSXFolderType(kClassicDomain, kSystemFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sDesktopDirectory) { rv = GetOSXFolderType(kClassicDomain, kDesktopFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sTrashDirectory) { rv = GetOSXFolderType(kClassicDomain, kTrashFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sStartupDirectory) { rv = GetOSXFolderType(kClassicDomain, kStartupFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sShutdownDirectory) { rv = GetOSXFolderType(kClassicDomain, kShutdownFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sAppleMenuDirectory) { rv = GetOSXFolderType(kClassicDomain, kAppleMenuFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sControlPanelDirectory) { rv = GetOSXFolderType(kClassicDomain, kControlPanelFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sExtensionDirectory) { rv = GetOSXFolderType(kClassicDomain, kExtensionFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sFontsDirectory) { rv = GetOSXFolderType(kClassicDomain, kFontsFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sPreferencesDirectory) { rv = GetOSXFolderType(kClassicDomain, kPreferencesFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sDocumentsDirectory) { rv = GetOSXFolderType(kClassicDomain, kDocumentsFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sInternetSearchDirectory) { rv = GetOSXFolderType(kClassicDomain, kInternetSearchSitesFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sUserLibDirectory) { rv = GetOSXFolderType(kUserDomain, kDomainLibraryFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sHomeDirectory) { rv = GetOSXFolderType(kUserDomain, kDomainTopLevelFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sDefaultDownloadDirectory) { NS_NewLocalFile(nsString(), PR_TRUE, getter_AddRefs(localFile)); nsCOMPtr localMacFile(do_QueryInterface(localFile)); if (localMacFile) { OSErr err; ICInstance icInstance; err = ::ICStart(&icInstance, 'XPCM'); if (err == noErr) { ICAttr attrs; ICFileSpec icFileSpec; long size = kICFileSpecHeaderSize; err = ::ICGetPref(icInstance, kICDownloadFolder, &attrs, &icFileSpec, &size); if (err == noErr || (err == icTruncatedErr && size >= kICFileSpecHeaderSize)) { rv = localMacFile->InitWithFSSpec(&icFileSpec.fss); } ::ICStop(icInstance); } if NS_FAILED(rv) { // We got an error getting the DL folder from IC so try finding the user's Desktop folder rv = GetOSXFolderType(kUserDomain, kSystemDesktopFolderType, getter_AddRefs(localFile)); } } // Don't cache the DL directory as the user may change it while we're running. // Negligible perf hit as this directory is only requested for downloads *persistent = PR_FALSE; } else if (inAtom == nsDirectoryService::sUserDesktopDirectory) { rv = GetOSXFolderType(kUserDomain, kDesktopFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sLocalDesktopDirectory) { rv = GetOSXFolderType(kLocalDomain, kDesktopFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sUserApplicationsDirectory) { rv = GetOSXFolderType(kUserDomain, kApplicationsFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sLocalApplicationsDirectory) { rv = GetOSXFolderType(kLocalDomain, kApplicationsFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sUserDocumentsDirectory) { rv = GetOSXFolderType(kUserDomain, kDocumentsFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sLocalDocumentsDirectory) { rv = GetOSXFolderType(kLocalDomain, kDocumentsFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sUserInternetPlugInDirectory) { rv = GetOSXFolderType(kUserDomain, kInternetPlugInFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sLocalInternetPlugInDirectory) { rv = GetOSXFolderType(kLocalDomain, kInternetPlugInFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sUserFrameworksDirectory) { rv = GetOSXFolderType(kUserDomain, kFrameworksFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sLocalFrameworksDirectory) { rv = GetOSXFolderType(kLocalDomain, kFrameworksFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sUserPreferencesDirectory) { rv = GetOSXFolderType(kUserDomain, kPreferencesFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sLocalPreferencesDirectory) { rv = GetOSXFolderType(kLocalDomain, kPreferencesFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sPictureDocumentsDirectory) { rv = GetOSXFolderType(kUserDomain, kPictureDocumentsFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sMovieDocumentsDirectory) { rv = GetOSXFolderType(kUserDomain, kMovieDocumentsFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sMusicDocumentsDirectory) { rv = GetOSXFolderType(kUserDomain, kMusicDocumentsFolderType, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sInternetSitesDirectory) { rv = GetOSXFolderType(kUserDomain, kInternetSitesFolderType, getter_AddRefs(localFile)); } #elif defined (XP_WIN) else if (inAtom == nsDirectoryService::sSystemDirectory) { rv = GetSpecialSystemDirectory(Win_SystemDirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sWindowsDirectory) { rv = GetSpecialSystemDirectory(Win_WindowsDirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sHomeDirectory) { rv = GetSpecialSystemDirectory(Win_HomeDirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sDesktop) { rv = GetSpecialSystemDirectory(Win_Desktop, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sPrograms) { rv = GetSpecialSystemDirectory(Win_Programs, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sControls) { rv = GetSpecialSystemDirectory(Win_Controls, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sPrinters) { rv = GetSpecialSystemDirectory(Win_Printers, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sPersonal) { rv = GetSpecialSystemDirectory(Win_Personal, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sFavorites) { rv = GetSpecialSystemDirectory(Win_Favorites, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sStartup) { rv = GetSpecialSystemDirectory(Win_Startup, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sRecent) { rv = GetSpecialSystemDirectory(Win_Recent, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sSendto) { rv = GetSpecialSystemDirectory(Win_Sendto, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sBitbucket) { rv = GetSpecialSystemDirectory(Win_Bitbucket, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sStartmenu) { rv = GetSpecialSystemDirectory(Win_Startmenu, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sDesktopdirectory) { rv = GetSpecialSystemDirectory(Win_Desktopdirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sDrives) { rv = GetSpecialSystemDirectory(Win_Drives, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sNetwork) { rv = GetSpecialSystemDirectory(Win_Network, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sNethood) { rv = GetSpecialSystemDirectory(Win_Nethood, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sFonts) { rv = GetSpecialSystemDirectory(Win_Fonts, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sTemplates) { rv = GetSpecialSystemDirectory(Win_Templates, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sCommon_Startmenu) { rv = GetSpecialSystemDirectory(Win_Common_Startmenu, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sCommon_Programs) { rv = GetSpecialSystemDirectory(Win_Common_Programs, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sCommon_Startup) { rv = GetSpecialSystemDirectory(Win_Common_Startup, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sCommon_Desktopdirectory) { rv = GetSpecialSystemDirectory(Win_Common_Desktopdirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sAppdata) { rv = GetSpecialSystemDirectory(Win_Appdata, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sPrinthood) { rv = GetSpecialSystemDirectory(Win_Printhood, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sWinCookiesDirectory) { rv = GetSpecialSystemDirectory(Win_Cookies, getter_AddRefs(localFile)); } #elif defined (XP_UNIX) else if (inAtom == nsDirectoryService::sLocalDirectory) { rv = GetSpecialSystemDirectory(Unix_LocalDirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sLibDirectory) { rv = GetSpecialSystemDirectory(Unix_LibDirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sHomeDirectory) { rv = GetSpecialSystemDirectory(Unix_HomeDirectory, getter_AddRefs(localFile)); } #elif defined (XP_OS2) else if (inAtom == nsDirectoryService::sSystemDirectory) { rv = GetSpecialSystemDirectory(OS2_SystemDirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sOS2Directory) { rv = GetSpecialSystemDirectory(OS2_OS2Directory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sHomeDirectory) { rv = GetSpecialSystemDirectory(OS2_HomeDirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sDesktopDirectory) { rv = GetSpecialSystemDirectory(OS2_DesktopDirectory, getter_AddRefs(localFile)); } #elif defined (XP_BEOS) else if (inAtom == nsDirectoryService::sSettingsDirectory) { rv = GetSpecialSystemDirectory(BeOS_SettingsDirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sHomeDirectory) { rv = GetSpecialSystemDirectory(BeOS_HomeDirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sDesktopDirectory) { rv = GetSpecialSystemDirectory(BeOS_DesktopDirectory, getter_AddRefs(localFile)); } else if (inAtom == nsDirectoryService::sSystemDirectory) { rv = GetSpecialSystemDirectory(BeOS_SystemDirectory, getter_AddRefs(localFile)); } #endif NS_RELEASE(inAtom); if (localFile && NS_SUCCEEDED(rv)) return localFile->QueryInterface(NS_GET_IID(nsIFile), (void**)_retval); #ifdef DEBUG_dougt printf("Failed to find directory for key: %s\n", prop); #endif return rv; } NS_IMETHODIMP nsDirectoryService::GetFiles(const char *prop, nsISimpleEnumerator **_retval) { NS_ENSURE_ARG_POINTER(_retval); *_retval = nsnull; return NS_ERROR_FAILURE; }