/* -*- 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 Communicator client 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): * Roland Mainz * Benjamin Smedberg * Ben Goodger * * Alternatively, the contents of this file may be used under the terms of * either the GNU General Public License Version 2 or later (the "GPL"), or * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), * in which case the provisions of the GPL or the LGPL are applicable instead * of those above. If you wish to allow use of your version of this file only * under the terms of either the GPL or the LGPL, and not to allow others to * use your version of this file under the terms of the MPL, indicate your * decision by deleting the provisions above and replace them with the notice * and other provisions required by the GPL or the LGPL. If you do not delete * the provisions above, a recipient may use your version of this file under * the terms of any one of the MPL, the GPL or the LGPL. * * ***** END LICENSE BLOCK ***** */ #include "nsAppRunner.h" #include "nsBuildID.h" #ifdef XP_MACOSX #include "MacLaunchHelper.h" #endif #ifdef XP_OS2 #include "private/pprthred.h" #endif #include "plevent.h" #include "prmem.h" #include "prnetdb.h" #include "prprf.h" #include "prproces.h" #include "prenv.h" #include "nsIAppShellService.h" #include "nsIAppStartup.h" #include "nsIAppStartupNotifier.h" #include "nsIArray.h" #include "nsICategoryManager.h" #include "nsIChromeRegistry.h" #include "nsICommandLineRunner.h" #include "nsIComponentManager.h" #include "nsIComponentRegistrar.h" #include "nsIContentHandler.h" #include "nsIDialogParamBlock.h" #include "nsIDOMWindow.h" #include "nsIEventQueueService.h" #include "nsIExtensionManager.h" #include "nsIIOService.h" #include "nsIObserverService.h" #include "nsINativeAppSupport.h" #include "nsIProcess.h" #include "nsIServiceManager.h" #include "nsISupportsPrimitives.h" #include "nsITimelineService.h" #include "nsIToolkitProfile.h" #include "nsIToolkitProfileService.h" #include "nsIURI.h" #include "nsIWindowCreator.h" #include "nsIWindowMediator.h" #include "nsIWindowWatcher.h" #include "nsIXULAppInfo.h" #include "nsCRT.h" #include "nsCOMPtr.h" #include "nsDirectoryServiceDefs.h" #include "nsDirectoryServiceUtils.h" #include "nsNetUtil.h" #include "nsXPCOM.h" #include "nsXPIDLString.h" #include "nsXPFEComponentsCID.h" #include "nsAppDirectoryServiceDefs.h" #include "nsXULAppAPI.h" #include "nsXREDirProvider.h" #include "nsINIParser.h" #include "InstallCleanupDefines.h" #include #ifdef XP_UNIX #include #include #endif #ifdef XP_WIN #include #include #endif #ifdef XP_OS2 #include #endif #ifdef XP_MACOSX #include "nsILocalFileMac.h" #include "nsCommandLineServiceMac.h" #endif // for X remote support #ifdef MOZ_ENABLE_XREMOTE #include "nsXRemoteClientCID.h" #include "nsIXRemoteClient.h" #include "nsIXRemoteService.h" #endif #ifdef NS_TRACE_MALLOC #include "nsTraceMalloc.h" #endif #if defined(DEBUG) && defined(XP_WIN32) #include #endif #if defined (XP_MACOSX) #include #endif extern "C" void ShowOSAlert(const char* aMessage); #define HELP_SPACER_1 "\t" #define HELP_SPACER_2 "\t\t" #define HELP_SPACER_4 "\t\t\t\t" #ifdef DEBUG #include "prlog.h" #endif #ifdef MOZ_JPROF #include "jprof.h" #endif // on x86 linux, the current builds of some popular plugins (notably // flashplayer and real) expect a few builtin symbols from libgcc // which were available in some older versions of gcc. However, // they're _NOT_ available in newer versions of gcc (eg 3.1), so if // we want those plugin to work with a gcc-3.1 built binary, we need // to provide these symbols. MOZ_ENABLE_OLD_ABI_COMPAT_WRAPPERS defaults // to true on x86 linux, and false everywhere else. // // The fact that the new and free operators are mismatched // mirrors the way the original functions in egcs 1.1.2 worked. #ifdef MOZ_ENABLE_OLD_ABI_COMPAT_WRAPPERS extern "C" { # ifndef HAVE___BUILTIN_VEC_NEW void *__builtin_vec_new(size_t aSize, const std::nothrow_t &aNoThrow) throw() { return ::operator new(aSize, aNoThrow); } # endif # ifndef HAVE___BUILTIN_VEC_DELETE void __builtin_vec_delete(void *aPtr, const std::nothrow_t &) throw () { if (aPtr) { free(aPtr); } } # endif # ifndef HAVE___BUILTIN_NEW void *__builtin_new(int aSize) { return malloc(aSize); } # endif # ifndef HAVE___BUILTIN_DELETE void __builtin_delete(void *aPtr) { free(aPtr); } # endif # ifndef HAVE___PURE_VIRTUAL void __pure_virtual(void) { extern void __cxa_pure_virtual(void); __cxa_pure_virtual(); } # endif } #endif #ifdef _BUILD_STATIC_BIN #include "nsStaticComponent.h" nsresult PR_CALLBACK app_getModuleInfo(nsStaticModuleInfo **info, PRUint32 *count); #endif #if defined(XP_UNIX) || defined(XP_BEOS) extern void InstallUnixSignalHandlers(const char *ProgramName); #endif int gArgc; char **gArgv; static int gRestartArgc; static char **gRestartArgv; #if defined(XP_BEOS) #include #include class nsBeOSApp : public BApplication { public: nsBeOSApp(sem_id sem) : BApplication(GetAppSig()), init(sem) { } void ReadyToRun(void) { release_sem(init); } static int32 Main(void *args) { nsBeOSApp *app = new nsBeOSApp((sem_id)args); if (nsnull == app) return B_ERROR; return app->Run(); } private: char *GetAppSig(void) { app_info appInfo; BFile file; BAppFileInfo appFileInfo; image_info info; int32 cookie = 0; static char sig[B_MIME_TYPE_LENGTH]; sig[0] = 0; if (get_next_image_info(0, &cookie, &info) != B_OK || file.SetTo(info.name, B_READ_ONLY) != B_OK || appFileInfo.SetTo(&file) != B_OK || appFileInfo.GetSignature(sig) != B_OK) { return "application/x-vnd.Mozilla"; } return sig; } sem_id init; }; static nsresult InitializeBeOSApp(void) { nsresult rv = NS_OK; sem_id initsem = create_sem(0, "beapp init"); if (initsem < B_OK) return NS_ERROR_FAILURE; thread_id tid = spawn_thread(nsBeOSApp::Main, "BApplication", B_NORMAL_PRIORITY, (void *)initsem); if (tid < B_OK || B_OK != resume_thread(tid)) rv = NS_ERROR_FAILURE; if (B_OK != acquire_sem(initsem)) rv = NS_ERROR_FAILURE; if (B_OK != delete_sem(initsem)) rv = NS_ERROR_FAILURE; return rv; } #endif // XP_BEOS #if defined(MOZ_WIDGET_GTK) || defined(MOZ_WIDGET_GTK2) #include #endif //MOZ_WIDGET_GTK || MOZ_WIDGET_GTK2 #if defined(MOZ_WIDGET_QT) #include #endif static PRBool strimatch(const char* lowerstr, const char* mixedstr) { while(*lowerstr) { if (!*mixedstr) return PR_FALSE; // mixedstr is shorter if (tolower(*mixedstr) != *lowerstr) return PR_FALSE; // no match ++lowerstr; ++mixedstr; } if (*mixedstr) return PR_FALSE; // lowerstr is shorter return PR_TRUE; } enum ArgResult { ARG_NONE = 0, ARG_FOUND = 1, ARG_BAD = 2 // you wanted a param, but there isn't one }; static void RemoveArg(char **argv) { do { *argv = *(argv + 1); ++argv; } while (*argv); --gArgc; } /** * Check for a commandline flag. If the flag takes a parameter, the * parameter is returned in aParam. Flags may be in the form -arg or * --arg (or /arg on win32/OS2). * * @param aArg the parameter to check. Must be lowercase. * @param if non-null, the -arg will be stored in this pointer. This is *not* * allocated, but rather a pointer to the argv data. */ static ArgResult CheckArg(const char* aArg, const char **aParam = nsnull) { char **curarg = gArgv + 1; // skip argv[0] while (*curarg) { char *arg = curarg[0]; if (arg[0] == '-' #if defined(XP_WIN) || defined(XP_OS2) || *arg == '/' #endif ) { ++arg; if (*arg == '-') ++arg; if (strimatch(aArg, arg)) { RemoveArg(curarg); if (!aParam) { return ARG_FOUND; } if (*curarg) { if (**curarg == '-' #if defined(XP_WIN) || defined(XP_OS2) || **curarg == '/' #endif ) return ARG_BAD; *aParam = *curarg; RemoveArg(curarg); return ARG_FOUND; } return ARG_BAD; } } ++curarg; } return ARG_NONE; } /** * The nsXULAppInfo object implements nsIFactory so that it can be its own * singleton. */ class nsXULAppInfo : public nsIXULAppInfo, public nsIFactory { public: NS_DECL_ISUPPORTS_INHERITED NS_DECL_NSIXULAPPINFO NS_DECL_NSIFACTORY }; NS_IMPL_QUERY_INTERFACE2(nsXULAppInfo, nsIXULAppInfo, nsIFactory) NS_IMETHODIMP_(nsrefcnt) nsXULAppInfo::AddRef() { return 1; } NS_IMETHODIMP_(nsrefcnt) nsXULAppInfo::Release() { return 1; } NS_IMETHODIMP nsXULAppInfo::GetVendor(nsACString& aResult) { aResult.Assign(gAppData->appVendor ? gAppData->appVendor : ""); return NS_OK; } NS_IMETHODIMP nsXULAppInfo::GetName(nsACString& aResult) { aResult.Assign(gAppData->appName); return NS_OK; } NS_IMETHODIMP nsXULAppInfo::GetID(nsID** aResult) { *aResult = (nsID*) NS_Alloc(sizeof(nsID)); if (!*aResult) return NS_ERROR_OUT_OF_MEMORY; **aResult = gAppData->id; return NS_OK; } NS_IMETHODIMP nsXULAppInfo::GetVersion(nsACString& aResult) { aResult.Assign(gAppData->appVersion ? gAppData->appVersion : ""); return NS_OK; } NS_IMETHODIMP nsXULAppInfo::GetAppBuildID(nsACString& aResult) { aResult.Assign(gAppData->appBuildID ? gAppData->appBuildID : ""); return NS_OK; } NS_IMETHODIMP nsXULAppInfo::GetGeckoBuildID(nsACString& aResult) { aResult.Assign(BUILD_ID); return NS_OK; } NS_IMETHODIMP nsXULAppInfo::CreateInstance(nsISupports* aOuter, REFNSIID aIID, void** aResult) { NS_ENSURE_NO_AGGREGATION(aOuter); return QueryInterface(aIID, aResult); } NS_IMETHODIMP nsXULAppInfo::LockFactory(PRBool aLock) { return NS_OK; } static const nsXULAppInfo kAppInfo; /** * Because we're starting/stopping XPCOM several times in different scenarios, * this class is a stack-based critter that makes sure that XPCOM is shut down * during early returns. */ class ScopedXPCOMStartup { public: ScopedXPCOMStartup() : mServiceManager(nsnull) { } ~ScopedXPCOMStartup(); nsresult Initialize(); nsresult DoAutoreg(); nsresult RegisterProfileService(nsIToolkitProfileService* aProfileService); nsresult InitEventQueue(); nsresult SetWindowCreator(nsINativeAppSupport* native); private: nsIServiceManager* mServiceManager; }; ScopedXPCOMStartup::~ScopedXPCOMStartup() { if (mServiceManager) { gDirServiceProvider->DoShutdown(); NS_ShutdownXPCOM(mServiceManager); mServiceManager = nsnull; } } // {95d89e3e-a169-41a3-8e56-719978e15b12} static const nsCID kAppInfoCID = { 0x95d89e3e, 0xa169, 0x41a3, { 0x8e, 0x56, 0x71, 0x99, 0x78, 0xe1, 0x5b, 0x12 } }; nsresult ScopedXPCOMStartup::Initialize() { NS_ASSERTION(gDirServiceProvider, "Should not get here!"); nsresult rv; rv = NS_InitXPCOM2(&mServiceManager, gDirServiceProvider->GetAppDir(), gDirServiceProvider); if (NS_FAILED(rv)) { NS_ERROR("Couldn't start xpcom!"); mServiceManager = nsnull; } else { nsCOMPtr reg = do_QueryInterface(mServiceManager); NS_ASSERTION(reg, "Service Manager doesn't QI to Registrar."); reg->RegisterFactory(kAppInfoCID, "nsXULAppInfo", XULAPPINFO_SERVICE_CONTRACTID, NS_CONST_CAST(nsXULAppInfo*,&kAppInfo)); } return rv; } // {0C4A446C-EE82-41f2-8D04-D366D2C7A7D4} static const nsCID kNativeAppSupportCID = { 0xc4a446c, 0xee82, 0x41f2, { 0x8d, 0x4, 0xd3, 0x66, 0xd2, 0xc7, 0xa7, 0xd4 } }; // {5F5E59CE-27BC-47eb-9D1F-B09CA9049836} static const nsCID kProfileServiceCID = { 0x5f5e59ce, 0x27bc, 0x47eb, { 0x9d, 0x1f, 0xb0, 0x9c, 0xa9, 0x4, 0x98, 0x36 } }; nsresult ScopedXPCOMStartup::RegisterProfileService(nsIToolkitProfileService* aProfileService) { NS_ASSERTION(mServiceManager, "Not initialized!"); nsCOMPtr factory = do_QueryInterface(aProfileService); NS_ASSERTION(factory, "Supposed to be an nsIFactory!"); nsCOMPtr reg (do_QueryInterface(mServiceManager)); if (!reg) return NS_ERROR_NO_INTERFACE; return reg->RegisterFactory(kProfileServiceCID, "Toolkit Profile Service", NS_PROFILESERVICE_CONTRACTID, factory); } nsresult ScopedXPCOMStartup::InitEventQueue() { NS_TIMELINE_ENTER("init event service"); nsresult rv; nsCOMPtr eventQService(do_GetService(NS_EVENTQUEUESERVICE_CONTRACTID, &rv)); NS_ENSURE_SUCCESS(rv, rv); rv = eventQService->CreateThreadEventQueue(); NS_TIMELINE_LEAVE("init event service"); return rv; } nsresult ScopedXPCOMStartup::DoAutoreg() { #ifdef DEBUG // _Always_ autoreg if we're in a debug build, under the assumption // that people are busily modifying components and will be angry if // their changes aren't noticed. nsCOMPtr registrar (do_QueryInterface(mServiceManager)); NS_ASSERTION(registrar, "Where's the component registrar?"); registrar->AutoRegister(nsnull); #endif return NS_OK; } /** * This is a little factory class that serves as a singleton-service-factory * for the nativeappsupport object. */ class nsSingletonFactory : public nsIFactory { public: NS_DECL_ISUPPORTS NS_DECL_NSIFACTORY nsSingletonFactory(nsISupports* aSingleton); ~nsSingletonFactory() { } private: nsCOMPtr mSingleton; }; nsSingletonFactory::nsSingletonFactory(nsISupports* aSingleton) : mSingleton(aSingleton) { NS_ASSERTION(mSingleton, "Singleton was null!"); } NS_IMPL_ISUPPORTS1(nsSingletonFactory, nsIFactory) NS_IMETHODIMP nsSingletonFactory::CreateInstance(nsISupports* aOuter, const nsIID& aIID, void* *aResult) { NS_ENSURE_NO_AGGREGATION(aOuter); return mSingleton->QueryInterface(aIID, aResult); } NS_IMETHODIMP nsSingletonFactory::LockFactory(PRBool) { return NS_OK; } /** * Set our windowcreator on the WindowWatcher service. */ nsresult ScopedXPCOMStartup::SetWindowCreator(nsINativeAppSupport* native) { nsresult rv; nsCOMPtr registrar (do_QueryInterface(mServiceManager)); NS_ASSERTION(registrar, "Where's the component registrar?"); nsCOMPtr nativeFactory = new nsSingletonFactory(native); NS_ENSURE_TRUE(nativeFactory, NS_ERROR_OUT_OF_MEMORY); rv = registrar->RegisterFactory(kNativeAppSupportCID, "Native App Support", NS_NATIVEAPPSUPPORT_CONTRACTID, nativeFactory); NS_ENSURE_SUCCESS(rv, rv); nsCOMPtr creator (do_GetService(NS_APPSTARTUP_CONTRACTID)); if (!creator) return NS_ERROR_UNEXPECTED; nsCOMPtr wwatch (do_GetService(NS_WINDOWWATCHER_CONTRACTID, &rv)); NS_ENSURE_SUCCESS(rv, rv); return wwatch->SetWindowCreator(creator); } static void DumpArbitraryHelp() { nsresult rv; nsXREDirProvider dirProvider; dirProvider.Initialize(nsnull); ScopedXPCOMStartup xpcom; xpcom.Initialize(); xpcom.DoAutoreg(); nsCOMPtr cmdline (do_CreateInstance("@mozilla.org/toolkit/command-line;1")); if (!cmdline) return; nsCString text; rv = cmdline->GetHelpText(text); if (NS_SUCCEEDED(rv)) printf("%s", text.get()); } // English text needs to go into a dtd file. // But when this is called we have no components etc. These strings must either be // here, or in a native resource file. static void DumpHelp() { printf("Usage: %s [ options ... ] [URL]\n", gArgv[0]); printf(" where options include:\n"); printf("\n"); #ifdef MOZ_WIDGET_GTK /* insert gtk options above moz options, like any other gtk app * * note: this isn't a very cool way to do things -- i'd rather get * these straight from a user's gtk version -- but it seems to be * what most gtk apps do. -dr */ printf("GTK options\n"); printf("%s--gdk-debug=FLAGS%sGdk debugging flags to set\n", HELP_SPACER_1, HELP_SPACER_2); printf("%s--gdk-no-debug=FLAGS%sGdk debugging flags to unset\n", HELP_SPACER_1, HELP_SPACER_2); printf("%s--gtk-debug=FLAGS%sGtk+ debugging flags to set\n", HELP_SPACER_1, HELP_SPACER_2); printf("%s--gtk-no-debug=FLAGS%sGtk+ debugging flags to unset\n", HELP_SPACER_1, HELP_SPACER_2); printf("%s--gtk-module=MODULE%sLoad an additional Gtk module\n", HELP_SPACER_1, HELP_SPACER_2); printf("%s-install%sInstall a private colormap\n", HELP_SPACER_1, HELP_SPACER_2); /* end gtk toolkit options */ #endif /* MOZ_WIDGET_GTK */ #if MOZ_WIDGET_XLIB printf("Xlib options\n"); printf("%s-display=DISPLAY%sX display to use\n", HELP_SPACER_1, HELP_SPACER_2); printf("%s-visual=VISUALID%sX visual to use\n", HELP_SPACER_1, HELP_SPACER_2); printf("%s-install_colormap%sInstall own colormap\n", HELP_SPACER_1, HELP_SPACER_2); printf("%s-sync%sMake X calls synchronous\n", HELP_SPACER_1, HELP_SPACER_2); printf("%s-no-xshm%sDon't use X shared memory extension\n", HELP_SPACER_1, HELP_SPACER_2); /* end xlib toolkit options */ #endif /* MOZ_WIDGET_XLIB */ #ifdef MOZ_X11 printf("X11 options\n"); printf("%s--display=DISPLAY%sX display to use\n", HELP_SPACER_1, HELP_SPACER_2); printf("%s--sync%sMake X calls synchronous\n", HELP_SPACER_1, HELP_SPACER_2); printf("%s--no-xshm%sDon't use X shared memory extension\n", HELP_SPACER_1, HELP_SPACER_2); printf("%s--xim-preedit=STYLE\n", HELP_SPACER_1); printf("%s--xim-status=STYLE\n", HELP_SPACER_1); #endif #ifdef XP_UNIX printf("%s--g-fatal-warnings%sMake all warnings fatal\n", HELP_SPACER_1, HELP_SPACER_2); printf("\nMozilla options\n"); #endif printf("%s-height %sSet height of startup window to .\n",HELP_SPACER_1,HELP_SPACER_2); printf("%s-h or -help%sPrint this message.\n",HELP_SPACER_1,HELP_SPACER_2); printf("%s-width %sSet width of startup window to .\n",HELP_SPACER_1,HELP_SPACER_2); printf("%s-v or -version%sPrint %s version.\n",HELP_SPACER_1,HELP_SPACER_2, gAppData->appName); printf("%s-P %sStart with .\n",HELP_SPACER_1,HELP_SPACER_2); printf("%s-ProfileManager%sStart with profile manager.\n",HELP_SPACER_1,HELP_SPACER_2); printf("%s-UILocale %sStart with resources as UI Locale.\n",HELP_SPACER_1,HELP_SPACER_2); printf("%s-contentLocale %sStart with resources as content Locale.\n",HELP_SPACER_1,HELP_SPACER_2); #ifdef XP_WIN printf("%s-console%sStart Mozilla with a debugging console.\n",HELP_SPACER_1,HELP_SPACER_2); #endif #ifdef MOZ_ENABLE_XREMOTE printf("%s-remote %sExecute in an already running\n" "%sMozilla process. For more info, see:\n" "\n%shttp://www.mozilla.org/unix/remote.html\n\n", HELP_SPACER_1,HELP_SPACER_1,HELP_SPACER_4,HELP_SPACER_2); #endif // this works, but only after the components have registered. so if you drop in a new command line handler, -help // won't not until the second run. // out of the bug, because we ship a component.reg file, it works correctly. DumpArbitraryHelp(); } // don't modify aAppDir directly... clone it first static int VerifyInstallation(nsIFile* aAppDir) { static const char lastResortMessage[] = "A previous install did not complete correctly. Finishing install."; // Maximum allowed / used length of alert message is 255 chars, due to restrictions on Mac. // Please make sure that file contents and fallback_alert_text are at most 255 chars. char message[256]; PRInt32 numRead = 0; const char *messageToShow = lastResortMessage; nsresult rv; nsCOMPtr messageFile; rv = aAppDir->Clone(getter_AddRefs(messageFile)); if (NS_SUCCEEDED(rv)) { messageFile->AppendNative(NS_LITERAL_CSTRING("res")); messageFile->AppendNative(CLEANUP_MESSAGE_FILENAME); PRFileDesc* fd = 0; nsCOMPtr lf (do_QueryInterface(messageFile)); if (lf) { rv = lf->OpenNSPRFileDesc(PR_RDONLY, 0664, &fd); if (NS_SUCCEEDED(rv)) { numRead = PR_Read(fd, message, sizeof(message)-1); if (numRead > 0) { message[numRead] = 0; messageToShow = message; } } } } ShowOSAlert(messageToShow); nsCOMPtr cleanupUtility; aAppDir->Clone(getter_AddRefs(cleanupUtility)); if (!cleanupUtility) return 1; cleanupUtility->AppendNative(CLEANUP_UTIL); ScopedXPCOMStartup xpcom; rv = xpcom.Initialize(); if (NS_FAILED(rv)) return 1; { // extra scoping needed to release things before xpcom shutdown //Create the process framework to run the cleanup utility nsCOMPtr cleanupProcess (do_CreateInstance(NS_PROCESS_CONTRACTID)); rv = cleanupProcess->Init(cleanupUtility); if (NS_FAILED(rv)) return 1; rv = cleanupProcess->Run(PR_FALSE,nsnull, 0, nsnull); if (NS_FAILED(rv)) return 1; } return 0; } #ifdef DEBUG_warren #ifdef XP_WIN #define _CRTDBG_MAP_ALLOC #include #endif #endif #if defined(FREEBSD) // pick up fpsetmask prototype. #include #endif static void DumpVersion() { printf("%s %s %s, %s\n", gAppData->appVendor, gAppData->appName, gAppData->appVersion, gAppData->copyright); } #ifdef MOZ_ENABLE_XREMOTE // use int here instead of a PR type since it will be returned // from main - just to keep types consistent static int HandleRemoteArgument(const char* remote) { nsresult rv; ArgResult ar; const char *profile = 0; nsCAutoString program(gAppData->appName); ToLowerCase(program); const char *username = getenv("LOGNAME"); ar = CheckArg("p", &profile); if (ar == ARG_BAD) { PR_fprintf(PR_STDERR, "Error: argument -p requires a profile name\n"); return 1; } const char *temp = nsnull; ar = CheckArg("a", &temp); if (ar == ARG_BAD) { PR_fprintf(PR_STDERR, "Error: argument -a requires an application name\n"); return 1; } else if (ar == ARG_FOUND) { program.Assign(temp); } ar = CheckArg("u", &username); if (ar == ARG_BAD) { PR_fprintf(PR_STDERR, "Error: argument -u requires a username\n"); return 1; } // start XPCOM ScopedXPCOMStartup xpcom; rv = xpcom.Initialize(); NS_ENSURE_SUCCESS(rv, 1); { // scope the comptr so we don't hold on to XPCOM objects beyond shutdown // try to get the X remote client nsCOMPtr client (do_CreateInstance(NS_XREMOTECLIENT_CONTRACTID)); NS_ENSURE_TRUE(client, 1); // try to init - connects to the X server and stuff rv = client->Init(); if (NS_FAILED(rv)) { PR_fprintf(PR_STDERR, "Error: Failed to connect to X server.\n"); return 1; } nsXPIDLCString response; PRBool success = PR_FALSE; rv = client->SendCommand(program.get(), username, profile, remote, getter_Copies(response), &success); // did the command fail? if (NS_FAILED(rv)) { PR_fprintf(PR_STDERR, "Error: Failed to send command: %s\n", response ? response.get() : "No response included"); return 1; } if (!success) { PR_fprintf(PR_STDERR, "Error: No running window found\n"); return 2; } client->Shutdown(); } return 0; } #endif // MOZ_ENABLE_XREMOTE #if defined(XP_UNIX) && !defined(XP_MACOSX) char gBinaryPath[MAXPATHLEN]; static PRBool GetBinaryPath(const char* argv0) { struct stat fileStat; // on unix, there is no official way to get the path of the current binary. // instead of using the MOZILLA_FIVE_HOME hack, which doesn't scale to // multiple applications, we will try a series of techniques: // // 1) look for /proc//exe which is a symlink to the executable on newer // Linux kernels // 2) use realpath() on argv[0], which works unless we're loaded from the // PATH // 3) manually walk through the PATH and look for ourself // 4) give up // #ifdef __linux__ #if 0 int r = readlink("/proc/self/exe", gBinaryPath, MAXPATHLEN); // apparently, /proc/self/exe can sometimes return weird data... check it if (r > 0 && r < MAXPATHLEN && stat(gBinaryPath, &fileStat) == 0) return PR_TRUE; #endif if (realpath(argv0, gBinaryPath) && stat(gBinaryPath, &fileStat) == 0) return PR_TRUE; const char *path = getenv("PATH"); if (!path) return PR_FALSE; char *pathdup = strdup(path); if (!pathdup) return PR_FALSE; PRBool found = PR_FALSE; char *newStr = pathdup; char *token; while ( (token = nsCRT::strtok(newStr, ":", &newStr)) ) { sprintf(gBinaryPath, "%s/%s", token, argv0); if (stat(gBinaryPath, &fileStat) == 0) { found = PR_TRUE; break; } } free(pathdup); return found; } #endif #define NS_ERROR_LAUNCHED_CHILD_PROCESS NS_ERROR_GENERATE_FAILURE(NS_ERROR_MODULE_PROFILE, 200) static nsresult LaunchChild(nsINativeAppSupport* aNative) { aNative->Quit(); // release DDE mutex, if we're holding it // We need to use platform-specific hackery to find the // path of this executable. This is copied, with some modifications, from // nsGREDirServiceProvider.cpp #ifdef XP_WIN // We must shorten the path to a 8.3 path, since that's all _execv can // handle (otherwise segments after any spaces that might exist in the path // will be converted into parameters, and really weird things will happen) char exePath[MAXPATHLEN]; if (!::GetModuleFileName(0, exePath, MAXPATHLEN) || !::GetShortPathName(exePath, exePath, sizeof(exePath))) return NS_ERROR_FAILURE; gRestartArgv[0] = (char*)exePath; #elif defined(XP_MACOSX) // Works even if we're not bundled. CFBundleRef appBundle = CFBundleGetMainBundle(); if (!appBundle) return NS_ERROR_FAILURE; CFURLRef bundleURL = CFBundleCopyExecutableURL(appBundle); if (!bundleURL) return NS_ERROR_FAILURE; FSRef fileRef; if (!CFURLGetFSRef(bundleURL, &fileRef)) { CFRelease(bundleURL); return NS_ERROR_FAILURE; } char exePath[MAXPATHLEN]; if (FSRefMakePath(&fileRef, exePath, MAXPATHLEN)) { CFRelease(bundleURL); return NS_ERROR_FAILURE; } CFRelease(bundleURL); #elif defined(XP_UNIX) char* exePath = gBinaryPath; #elif defined(XP_OS2) PPIB ppib; PTIB ptib; char exePath[MAXPATHLEN]; DosGetInfoBlocks( &ptib, &ppib); DosQueryModuleName( ppib->pib_hmte, MAXPATHLEN, exePath); #elif defined(XP_BEOS) int32 cookie = 0; image_info info; if(get_next_image_info(0, &cookie, &info) != B_OK) return NS_ERROR_FAILURE; char *exePath = info.name; #elif #error Oops, you need platform-specific code here #endif // restart this process by exec'ing it into the current process // if supported by the platform. otherwise, use nspr ;-) #if defined(XP_WIN) || defined(XP_OS2) if (_execv(exePath, gRestartArgv) == -1) return NS_ERROR_FAILURE; #elif defined(XP_MACOSX) LaunchChildMac(gRestartArgc, gRestartArgv); #elif defined(XP_UNIX) if (execv(exePath, gRestartArgv) == -1) return NS_ERROR_FAILURE; #else PRProcess* process = PR_CreateProcess(exePath, gRestartArgv, nsnull, nsnull); if (!process) return NS_ERROR_FAILURE; PRInt32 exitCode; PRStatus failed = PR_WaitProcess(process, &exitCode); if (failed || exitCode) return NS_ERROR_FAILURE; #endif return NS_ERROR_LAUNCHED_CHILD_PROCESS; } static const char kProfileManagerURL[] = "chrome://mozapps/content/profile/profileSelection.xul"; static nsresult ShowProfileManager(nsIToolkitProfileService* aProfileSvc, nsINativeAppSupport* aNative) { nsresult rv; nsCOMPtr lf; { ScopedXPCOMStartup xpcom; rv = xpcom.Initialize(); NS_ENSURE_SUCCESS(rv, rv); rv = xpcom.RegisterProfileService(aProfileSvc); rv |= xpcom.DoAutoreg(); rv |= xpcom.InitEventQueue(); rv |= xpcom.SetWindowCreator(aNative); NS_ENSURE_SUCCESS(rv, NS_ERROR_FAILURE); { //extra scoping is needed so we release these components before xpcom shutdown nsCOMPtr windowWatcher (do_GetService(NS_WINDOWWATCHER_CONTRACTID)); nsCOMPtr ioParamBlock (do_CreateInstance(NS_DIALOGPARAMBLOCK_CONTRACTID)); nsCOMPtr dlgArray (do_CreateInstance(NS_ARRAY_CONTRACTID)); NS_ENSURE_TRUE(windowWatcher && ioParamBlock && dlgArray, NS_ERROR_FAILURE); ioParamBlock->SetObjects(dlgArray); nsCOMPtr appStartup (do_GetService(NS_APPSTARTUP_CONTRACTID)); NS_ENSURE_TRUE(appStartup, NS_ERROR_FAILURE); appStartup->EnterLastWindowClosingSurvivalArea(); nsCOMPtr newWindow; rv = windowWatcher->OpenWindow(nsnull, kProfileManagerURL, "_blank", "centerscreen,chrome,modal,titlebar", ioParamBlock, getter_AddRefs(newWindow)); appStartup->ExitLastWindowClosingSurvivalArea(); NS_ENSURE_SUCCESS(rv, rv); aProfileSvc->Flush(); PRInt32 dialogConfirmed; rv = ioParamBlock->GetInt(0, &dialogConfirmed); if (NS_FAILED(rv) || dialogConfirmed == 0) return NS_ERROR_ABORT; nsCOMPtr lock; rv = dlgArray->QueryElementAt(0, NS_GET_IID(nsIProfileLock), getter_AddRefs(lock)); NS_ENSURE_SUCCESS(rv, rv); rv = lock->GetDirectory(getter_AddRefs(lf)); NS_ENSURE_SUCCESS(rv, rv); lock->Unlock(); } } nsCAutoString path; lf->GetNativePath(path); static char kEnvVar[MAXPATHLEN]; sprintf(kEnvVar, "XRE_PROFILE_PATH=%s", path.get()); PR_SetEnv(kEnvVar); PRBool offline = PR_FALSE; aProfileSvc->GetStartOffline(&offline); if (offline) { PR_SetEnv("XRE_START_OFFLINE=1"); } return LaunchChild(aNative); } static nsresult ImportProfiles(nsIToolkitProfileService* aPService, nsINativeAppSupport* aNative) { nsresult rv; PR_SetEnv("XRE_IMPORT_PROFILES=1"); // try to import old-style profiles { // scope XPCOM ScopedXPCOMStartup xpcom; rv = xpcom.Initialize(); if (NS_SUCCEEDED(rv)) { xpcom.DoAutoreg(); xpcom.RegisterProfileService(aPService); nsCOMPtr migrator (do_GetService(NS_PROFILEMIGRATOR_CONTRACTID)); if (migrator) { migrator->Import(); } } } aPService->Flush(); return LaunchChild(aNative); } // Pick a profile. We need to end up with a profile lock. // // 1) check for -profile // 2) check for -P // 3) check for -ProfileManager // 4) use the default profile, if there is one // 5) if there are *no* profiles, set up profile-migration // 6) display the profile-manager UI static PRBool gDoMigration = PR_FALSE; static nsresult SelectProfile(nsIProfileLock* *aResult, nsINativeAppSupport* aNative, PRBool* aStartOffline) { nsresult rv; ArgResult ar; const char* arg; *aResult = nsnull; *aStartOffline = PR_FALSE; arg = PR_GetEnv("XRE_START_OFFLINE"); if ((arg && *arg) || CheckArg("offline")) *aStartOffline = PR_TRUE; arg = PR_GetEnv("XRE_PROFILE_PATH"); if (arg && *arg) { nsCOMPtr lf; rv = NS_NewNativeLocalFile(nsDependentCString(arg), PR_TRUE, getter_AddRefs(lf)); NS_ENSURE_SUCCESS(rv, rv); return NS_LockProfilePath(lf, aResult); } if (CheckArg("migration")) gDoMigration = PR_TRUE; ar = CheckArg("profile", &arg); if (ar == ARG_BAD) { PR_fprintf(PR_STDERR, "Error: argument -profile requires a path\n"); return NS_ERROR_FAILURE; } if (ar) { nsCOMPtr lf; rv = NS_GetFileFromPath(arg, getter_AddRefs(lf)); NS_ENSURE_SUCCESS(rv, rv); return NS_LockProfilePath(lf, aResult); } nsCOMPtr profileSvc; rv = NS_NewToolkitProfileService(getter_AddRefs(profileSvc)); NS_ENSURE_SUCCESS(rv, rv); ar = CheckArg("createprofile", &arg); if (ar == ARG_BAD) { PR_fprintf(PR_STDERR, "Error: argument -createprofile requires a profile name\n"); return NS_ERROR_FAILURE; } if (ar) { nsCOMPtr profile; const char* delim = strchr(arg, ' '); if (delim) { nsCOMPtr lf; rv = NS_NewNativeLocalFile(nsDependentCString(delim + 1), PR_TRUE, getter_AddRefs(lf)); if (NS_FAILED(rv)) { PR_fprintf(PR_STDERR, "Error: profile path not valid."); return rv; } rv = profileSvc->CreateProfile(lf, nsDependentCSubstring(arg, delim), getter_AddRefs(profile)); } else { rv = profileSvc->CreateProfile(nsnull, nsDependentCString(arg), getter_AddRefs(profile)); } if (NS_SUCCEEDED(rv)) { rv = NS_ERROR_ABORT; PR_fprintf(PR_STDERR, "Success: created profile '%s'\n", arg); } profileSvc->Flush(); // XXXben need to ensure prefs.js exists here so the tinderboxes will // not go orange. nsCOMPtr prefsJSFile; profile->GetRootDir(getter_AddRefs(prefsJSFile)); prefsJSFile->AppendNative(NS_LITERAL_CSTRING("prefs.js")); PRBool exists; prefsJSFile->Exists(&exists); if (!exists) prefsJSFile->Create(nsIFile::NORMAL_FILE_TYPE, 0644); return rv; } PRUint32 count; rv = profileSvc->GetProfileCount(&count); NS_ENSURE_SUCCESS(rv, rv); arg = PR_GetEnv("XRE_IMPORT_PROFILES"); if (!count && (!arg || !*arg)) { return ImportProfiles(profileSvc, aNative); } ar = CheckArg("p", &arg); if (ar == ARG_BAD) { return ShowProfileManager(profileSvc, aNative); } if (ar) { nsCOMPtr profile; rv = profileSvc->GetProfileByName(nsDependentCString(arg), getter_AddRefs(profile)); if (NS_SUCCEEDED(rv)) { rv = profile->Lock(aResult); if (NS_SUCCEEDED(rv)) return NS_OK; } return ShowProfileManager(profileSvc, aNative); } if (CheckArg("profilemanager")) { return ShowProfileManager(profileSvc, aNative); } if (!count) { gDoMigration = PR_TRUE; // create a default profile nsCOMPtr profile; nsresult rv = profileSvc->CreateProfile(nsnull, // choose a default dir for us NS_LITERAL_CSTRING("default"), getter_AddRefs(profile)); if (NS_SUCCEEDED(rv)) { profileSvc->Flush(); rv = profile->Lock(aResult); if (NS_SUCCEEDED(rv)) return NS_OK; } } PRBool useDefault = PR_TRUE; if (count > 1) profileSvc->GetStartWithLastProfile(&useDefault); if (useDefault) { nsCOMPtr profile; // GetSelectedProfile will auto-select the only profile if there's just one profileSvc->GetSelectedProfile(getter_AddRefs(profile)); if (profile) { rv = profile->Lock(aResult); if (NS_SUCCEEDED(rv)) return NS_OK; } } return ShowProfileManager(profileSvc, aNative); } #define FILE_COMPATIBILITY_INFO NS_LITERAL_CSTRING("compatibility.ini") static void GetVersion(nsIFile* aProfileDir, char* aVersion, int aVersionLength) { aVersion[0] = '\0'; nsCOMPtr file; aProfileDir->Clone(getter_AddRefs(file)); if (!file) return; file->AppendNative(FILE_COMPATIBILITY_INFO); nsINIParser parser; nsCOMPtr localFile(do_QueryInterface(file)); nsresult rv = parser.Init(localFile); if (NS_FAILED(rv)) return; parser.GetString("Compatibility", "LastVersion", aVersion, aVersionLength); } static void BuildVersion(nsCString &aBuf) { aBuf.Assign(gAppData->appVersion); aBuf.Append('_'); aBuf.Append(gAppData->appBuildID); aBuf.Append('/'); aBuf.AppendLiteral(GRE_BUILD_ID); } static void WriteVersion(nsIFile* aProfileDir, const nsCSubstring &version) { nsCOMPtr file; aProfileDir->Clone(getter_AddRefs(file)); if (!file) return; file->AppendNative(FILE_COMPATIBILITY_INFO); nsCOMPtr lf = do_QueryInterface(file); PRFileDesc *fd = nsnull; lf->OpenNSPRFileDesc(PR_WRONLY | PR_CREATE_FILE | PR_TRUNCATE, 0600, &fd); if (!fd) { NS_ERROR("could not create output stream"); return; } nsCAutoString buf; buf.AssignLiteral("[Compatibility]\r\nLastVersion="); buf.Append(version); buf.AppendLiteral(NS_LINEBREAK); PR_Write(fd, buf.get(), buf.Length()); PR_Close(fd); } static PRBool ComponentsListChanged(nsIFile* aProfileDir) { nsCOMPtr file; aProfileDir->Clone(getter_AddRefs(file)); if (!file) return PR_TRUE; file->AppendNative(NS_LITERAL_CSTRING(".autoreg")); PRBool exists = PR_FALSE; file->Exists(&exists); return exists; } static void RemoveComponentRegistries(nsIFile* aProfileDir) { nsCOMPtr file; aProfileDir->Clone(getter_AddRefs(file)); if (!file) return; file->AppendNative(NS_LITERAL_CSTRING("compreg.dat")); file->Remove(PR_FALSE); file->SetNativeLeafName(NS_LITERAL_CSTRING("xpti.dat")); file->Remove(PR_FALSE); file->SetNativeLeafName(NS_LITERAL_CSTRING(".autoreg")); file->Remove(PR_FALSE); } const nsXREAppData* gAppData = nsnull; #if defined(XP_OS2) // because we use early returns, we use a stack-based helper to un-set the OS2 FP handler class ScopedFPHandler { private: EXCEPTIONREGISTRATIONRECORD excpreg; public: ScopedFPHandler() { PR_OS2_SetFloatExcpHandler(&excpreg); } ~ScopedFPHandler() { PR_OS2_UnsetFloatExcpHandler(&excpreg); } }; #endif int xre_main(int argc, char* argv[], const nsXREAppData* aAppData) { nsresult rv; NS_TIMELINE_MARK("enter main"); #if defined(DEBUG) && defined(XP_WIN32) // Disable small heap allocator to get heapwalk() giving us // accurate heap numbers. Win2k non-debug does not use small heap allocator. // Win2k debug seems to be still using it. // http://msdn.microsoft.com/library/default.asp?url=/library/en-us/vclib/html/_crt__set_sbh_threshold.asp _set_sbh_threshold(0); #endif #if defined(XP_UNIX) || defined(XP_BEOS) InstallUnixSignalHandlers(argv[0]); #endif // Unbuffer stdout, needed for tinderbox tests. setbuf(stdout, 0); #if defined(FREEBSD) // Disable all SIGFPE's on FreeBSD, as it has non-IEEE-conformant fp // trap behavior that trips up on floating-point tests performed by // the JS engine. See bugzilla bug 9967 details. fpsetmask(0); #endif #if defined(XP_UNIX) && !defined(XP_MACOSX) if (!GetBinaryPath(argv[0])) return 1; #endif gArgc = argc; gArgv = argv; NS_ASSERTION(aAppData, "must specify XUL app data"); gAppData = aAppData; gRestartArgc = argc; gRestartArgv = (char**) malloc(sizeof(char*) * (argc + 1)); if (!gRestartArgv) return 1; int i; for (i = 0; i < argc; ++i) { gRestartArgv[i] = argv[i]; } gRestartArgv[argc] = nsnull; #if defined(XP_OS2) ULONG ulMaxFH = 0; LONG ulReqCount = 0; DosSetRelMaxFH(&ulReqCount, &ulMaxFH); if (ulMaxFH < 256) { DosSetMaxFH(256); } ScopedFPHandler handler(); #endif /* XP_OS2 */ #if defined(XP_BEOS) if (NS_OK != InitializeBeOSApp()) return 1; #endif #ifdef _BUILD_STATIC_BIN // Initialize XPCOM's module info table NSGetStaticModuleInfo = app_getModuleInfo; #endif // Handle -help and -version command line arguments. // They should return quickly, so we deal with them here. if (CheckArg("h") || CheckArg("help") || CheckArg("?")) { DumpHelp(); return 0; } if (CheckArg("v") || CheckArg("version")) { DumpVersion(); return 0; } #ifdef NS_TRACE_MALLOC gArgc = argc = NS_TraceMallocStartupArgs(argc, argv); #endif nsXREDirProvider dirProvider; { nsCOMPtr xulAppDir; const char *appDataFile; if (CheckArg("app", &appDataFile) == ARG_FOUND) { nsCOMPtr lf; NS_GetFileFromPath(appDataFile, getter_AddRefs(lf)); lf->GetParent(getter_AddRefs(xulAppDir)); } rv = dirProvider.Initialize(xulAppDir); if (NS_FAILED(rv)) return 1; } // Check for -register, which registers chrome and then exits immediately. if (CheckArg("register")) { ScopedXPCOMStartup xpcom; rv = xpcom.Initialize(); NS_ENSURE_SUCCESS(rv, 1); { nsCOMPtr chromeReg (do_GetService("@mozilla.org/chrome/chrome-registry;1")); NS_ENSURE_TRUE(chromeReg, 1); chromeReg->CheckForNewChrome(); if (gAppData->flags & NS_XRE_ENABLE_EXTENSION_MANAGER) { nsCOMPtr em (do_GetService("@mozilla.org/extensions/manager;1")); NS_ENSURE_TRUE(em, 1); em->Register(); } } return 0; } #if defined(MOZ_WIDGET_GTK) || defined(MOZ_WIDGET_GTK2) // setup for private colormap. Ideally we'd like to do this // in nsAppShell::Create, but we need to get in before gtk // has been initialized to make sure everything is running // consistently. if (CheckArg("install")) gdk_rgb_set_install(TRUE); // Initialize GTK+1/2 here for splash #if defined(MOZ_WIDGET_GTK) gtk_set_locale(); #endif gtk_init(&gArgc, &gArgv); gtk_widget_set_default_visual(gdk_rgb_get_visual()); gtk_widget_set_default_colormap(gdk_rgb_get_cmap()); #endif /* MOZ_WIDGET_GTK || MOZ_WIDGET_GTK2 */ #if defined(MOZ_WIDGET_QT) QAppliction qapp(argc, argv); #endif // #if defined(MOZ_WIDGET_XLIB) // XXXtimeless fix me! How do we get a Display from here to nsAppShell.cpp ? // #endif // Call the code to install our handler #ifdef MOZ_JPROF setupProfilingStuff(); #endif // Try to allocate "native app support." nsCOMPtr nativeApp; rv = NS_CreateNativeAppSupport(getter_AddRefs(nativeApp)); if (NS_FAILED(rv)) return 1; PRBool canRun = PR_FALSE; rv = nativeApp->Start(&canRun); if (NS_FAILED(rv) || !canRun) { return 1; } //---------------------------------------------------------------- // We need to check if a previous installation occured and // if so, make sure it finished and cleaned up correctly. // // If there is an xpicleanup.dat file left around, that means the // previous installation did not finish correctly. We must cleanup // before a valid mozilla can run. // // Show the user a platform-specific Alert message, then spawn the // xpicleanup utility, then exit. //---------------------------------------------------------------- nsCOMPtr registryFile; rv = dirProvider.GetAppDir()->Clone(getter_AddRefs(registryFile)); if (NS_SUCCEEDED(rv)) { registryFile->AppendNative(CLEANUP_REGISTRY); PRBool exists; rv = registryFile->Exists(&exists); if (NS_SUCCEEDED(rv) && exists) { return VerifyInstallation(dirProvider.GetAppDir()); } } #ifdef MOZ_ENABLE_XREMOTE // handle -remote now that xpcom is fired up const char* xremotearg; ArgResult ar = CheckArg("remote", &xremotearg); if (ar == ARG_BAD) { PR_fprintf(PR_STDERR, "Error: -remote requires an argument\n"); return 1; } if (ar) { return HandleRemoteArgument(xremotearg); } #endif nsCOMPtr profileLock; PRBool startOffline = PR_FALSE; rv = SelectProfile(getter_AddRefs(profileLock), nativeApp, &startOffline); if (rv == NS_ERROR_LAUNCHED_CHILD_PROCESS || rv == NS_ERROR_ABORT) return 0; if (NS_FAILED(rv)) return 1; nsCOMPtr lf; rv = profileLock->GetDirectory(getter_AddRefs(lf)); NS_ENSURE_SUCCESS(rv, 1); rv = dirProvider.SetProfileDir(lf); NS_ENSURE_SUCCESS(rv, 1); //////////////////////// NOW WE HAVE A PROFILE //////////////////////// PRBool upgraded = PR_FALSE; PRBool componentsListChanged = PR_FALSE; // Check for version compatibility with the last version of the app this // profile was started with. The format of the version stamp is defined // by the BuildVersion function. char lastVersion[MAXPATHLEN]; GetVersion(lf, lastVersion, MAXPATHLEN); // Build the version stamp for the running application. nsCAutoString version; BuildVersion(version); // Every time a profile is loaded by a build with a different version, // it updates the compatibility.ini file saying what version last wrote // the compreg.dat. On subsequent launches if the version matches, // there is no need for re-registration. If the user loads the same // profile in different builds the component registry must be // re-generated to prevent mysterious component loading failures. // if (version.Equals(lastVersion)) { componentsListChanged = ComponentsListChanged(lf); if (componentsListChanged) { // Remove compreg.dat and xpti.dat, forcing component re-registration, // with the new list of additional components directories specified // in "components.ini" which we have just discovered changed since the // last time the application was run. RemoveComponentRegistries(lf); } // Nothing need be done for the normal startup case. } else { // Remove compreg.dat and xpti.dat, forcing component re-registration // with the default set of components (this disables any potentially // troublesome incompatible XPCOM components). RemoveComponentRegistries(lf); // Tell the Extension Manager it should check for incompatible // Extensions and re-write the Components manifest ("components.ini") // with a list of XPCOM components for compatible extensions upgraded = PR_TRUE; // Write out version WriteVersion(lf, version); } PRBool needsRestart = PR_FALSE; // Allows the user to forcefully bypass the restart process at their // own risk. Useful for debugging or for tinderboxes where child // processes can be problematic. PRBool noRestart = PR_FALSE; { // Start the real application ScopedXPCOMStartup xpcom; rv = xpcom.Initialize(); NS_ENSURE_SUCCESS(rv, 1); rv = xpcom.DoAutoreg(); rv |= xpcom.InitEventQueue(); rv |= xpcom.SetWindowCreator(nativeApp); NS_ENSURE_SUCCESS(rv, 1); { if (startOffline) { nsCOMPtr io (do_GetService("@mozilla.org/network/io-service;1")); NS_ENSURE_TRUE(io, 1); io->SetOffline(PR_TRUE); } { NS_TIMELINE_ENTER("startupNotifier"); nsCOMPtr startupNotifier (do_CreateInstance(NS_APPSTARTUPNOTIFIER_CONTRACTID, &rv)); NS_ENSURE_SUCCESS(rv, 1); startupNotifier->Observe(nsnull, APPSTARTUP_TOPIC, nsnull); NS_TIMELINE_LEAVE("startupNotifier"); } nsCOMPtr appStartup (do_GetService(NS_APPSTARTUP_CONTRACTID)); NS_ENSURE_TRUE(appStartup, 1); // So we can open and close windows during startup appStartup->EnterLastWindowClosingSurvivalArea(); // Profile Migration if (gAppData->flags & NS_XRE_ENABLE_PROFILE_MIGRATOR && gDoMigration) { gDoMigration = PR_FALSE; nsCOMPtr pm (do_CreateInstance(NS_PROFILEMIGRATOR_CONTRACTID)); if (pm) pm->Migrate(&dirProvider); } dirProvider.DoStartup(); nsCOMPtr cmdLine (do_CreateInstance("@mozilla.org/toolkit/command-line;1")); NS_ENSURE_TRUE(cmdLine, 1); nsCOMPtr workingDir; rv = NS_GetSpecialDirectory(NS_OS_CURRENT_WORKING_DIR, getter_AddRefs(workingDir)); NS_ENSURE_SUCCESS(rv, 1); rv = cmdLine->Init(gArgc, gArgv, workingDir, nsICommandLine::STATE_INITIAL_LAUNCH); NS_ENSURE_SUCCESS(rv, 1); /* Special-case services that need early access to the command line. */ nsCOMPtr chromeObserver (do_GetService("@mozilla.org/chrome/chrome-registry;1")); if (chromeObserver) { chromeObserver->Observe(cmdLine, "command-line-startup", nsnull); } NS_TIMELINE_ENTER("appStartup->CreateHiddenWindow"); rv = appStartup->CreateHiddenWindow(); NS_TIMELINE_LEAVE("appStartup->CreateHiddenWindow"); NS_ENSURE_SUCCESS(rv, 1); // Extension Compatibility Checking and Startup if (gAppData->flags & NS_XRE_ENABLE_EXTENSION_MANAGER) { nsCOMPtr em(do_GetService("@mozilla.org/extensions/manager;1")); NS_ENSURE_TRUE(em, 1); if (CheckArg("install-global-extension") || CheckArg("install-global-theme") || CheckArg("list-global-items") || CheckArg("lock-item") || CheckArg("unlock-item")) { // Do the required processing and then shut down. em->HandleCommandLineArgs(cmdLine); return 0; } char* noEMRestart = PR_GetEnv("NO_EM_RESTART"); noRestart = noEMRestart && !strcmp("1", noEMRestart); if (!noRestart && upgraded) { rv = em->CheckForMismatches(&needsRestart); NS_ASSERTION(NS_SUCCEEDED(rv), "Oops, looks like you have a extensions.rdf file generated by a buggy nightly build. Please remove it!"); if (NS_FAILED(rv)) { needsRestart = PR_FALSE; upgraded = PR_FALSE; } } if (noRestart || (!upgraded || !needsRestart)) em->Start(cmdLine, componentsListChanged, &needsRestart); } if (noRestart || (!upgraded && !needsRestart)) { // clear out any environment variables which may have been set // during the relaunch process now that we know we won't be relaunching. PR_SetEnv("XRE_PROFILE_PATH="); PR_SetEnv("XRE_START_OFFLINE="); PR_SetEnv("XRE_IMPORT_PROFILES="); #ifdef XP_MACOSX // we re-initialize the command-line service and do appleevents munging // after we are sure that we're not restarting cmdLine = do_CreateInstance("@mozilla.org/toolkit/command-line;1"); NS_ENSURE_TRUE(cmdLine, 1); rv = InitializeMacCommandLine(gArgc, gArgv); NS_ENSURE_SUCCESS(rv, 1); rv = cmdLine->Init(gArgc, gArgv, workingDir, nsICommandLine::STATE_INITIAL_LAUNCH); NS_ENSURE_SUCCESS(rv, 1); // Kick off the prebinding update now that we know we won't be // relaunching. UpdatePrebinding(); #endif rv = cmdLine->Run(); NS_ENSURE_SUCCESS(rv, 1); nsCOMPtr windowMediator (do_GetService(NS_WINDOWMEDIATOR_CONTRACTID, &rv)); NS_ENSURE_SUCCESS(rv, 1); // Make sure there exists at least 1 window. nsCOMPtr windowEnumerator; rv = windowMediator->GetEnumerator(nsnull, getter_AddRefs(windowEnumerator)); NS_ENSURE_SUCCESS(rv, 1); PRBool more; windowEnumerator->HasMoreElements(&more); if (more) { #ifndef XP_MACOSX appStartup->ExitLastWindowClosingSurvivalArea(); #endif #ifdef MOZ_ENABLE_XREMOTE // if we have X remote support and we have our one window up and // running start listening for requests on the proxy window. nsCOMPtr remoteService; remoteService = do_GetService(NS_IXREMOTESERVICE_CONTRACTID); if (remoteService) remoteService->Startup(aAppData->appName); #endif /* MOZ_ENABLE_XREMOTE */ // enable win32 DDE responses and Mac appleevents responses nativeApp->Enable(); // Start main event loop NS_TIMELINE_ENTER("appStartup->Run"); rv = appStartup->Run(); NS_TIMELINE_LEAVE("appStartup->Run"); NS_ASSERTION(NS_SUCCEEDED(rv), "failed to run appstartup"); #ifdef MOZ_ENABLE_XREMOTE // shut down the x remote proxy window if (remoteService) remoteService->Shutdown(); #endif /* MOZ_ENABLE_XREMOTE */ } #ifdef MOZ_TIMELINE // Make sure we print this out even if timeline is runtime disabled if (NS_FAILED(NS_TIMELINE_LEAVE("main1"))) NS_TimelineForceMark("...main1"); #endif } else { // Upgrade condition (build id changes), but the restart hint was // not set by the Extension Manager. This is because the compatibility // resolution for Extensions is different than for the component // registry - major milestone vs. build id. needsRestart = PR_TRUE; } } profileLock->Unlock(); } // Restart the app after XPCOM has been shut down cleanly. if (!noRestart && needsRestart) { nsCAutoString path; lf->GetNativePath(path); static char kEnvVar[MAXPATHLEN]; sprintf(kEnvVar, "XRE_PROFILE_PATH=%s", path.get()); PR_SetEnv(kEnvVar); return LaunchChild(nativeApp) == NS_ERROR_LAUNCHED_CHILD_PROCESS ? 0 : 1; } return NS_FAILED(rv) ? 1 : 0; }