mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-14 15:37:55 +00:00
504 lines
13 KiB
C++
504 lines
13 KiB
C++
/* ***** 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 libXUL embedding.
|
|
*
|
|
* The Initial Developer of the Original Code is
|
|
* Benjamin Smedberg <benjamin@smedbergs.us>
|
|
*
|
|
* Portions created by the Initial Developer are Copyright (C) 2005
|
|
* the Mozilla Foundation. All Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
*
|
|
* Alternatively, the contents of this file may be used under the terms of
|
|
* either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
* in which case the provisions of the GPL or the LGPL are applicable instead
|
|
* of those above. If you wish to allow use of your version of this file only
|
|
* under the terms of either the GPL or the LGPL, and not to allow others to
|
|
* use your version of this file under the terms of the MPL, indicate your
|
|
* decision by deleting the provisions above and replace them with the notice
|
|
* and other provisions required by the GPL or the LGPL. If you do not delete
|
|
* the provisions above, a recipient may use your version of this file under
|
|
* the terms of any one of the MPL, the GPL or the LGPL.
|
|
*
|
|
* ***** END LICENSE BLOCK ***** */
|
|
|
|
#ifdef MOZ_IPC
|
|
#include "base/basictypes.h"
|
|
#endif
|
|
|
|
#include "nsXULAppAPI.h"
|
|
|
|
#include <stdlib.h>
|
|
#if defined(MOZ_WIDGET_GTK2)
|
|
#include <glib.h>
|
|
#endif
|
|
|
|
#include "prenv.h"
|
|
|
|
#include "nsIAppShell.h"
|
|
#include "nsIAppStartupNotifier.h"
|
|
#include "nsIDirectoryService.h"
|
|
#include "nsILocalFile.h"
|
|
#include "nsIToolkitChromeRegistry.h"
|
|
#include "nsIToolkitProfile.h"
|
|
|
|
#if defined(OS_LINUX)
|
|
# define XP_LINUX
|
|
#endif
|
|
|
|
#ifdef XP_WIN
|
|
#include <process.h>
|
|
#endif
|
|
|
|
#include "nsAppDirectoryServiceDefs.h"
|
|
#include "nsAppRunner.h"
|
|
#include "nsAutoRef.h"
|
|
#include "nsDirectoryServiceDefs.h"
|
|
#include "nsExceptionHandler.h"
|
|
#include "nsStaticComponents.h"
|
|
#include "nsString.h"
|
|
#include "nsThreadUtils.h"
|
|
#include "nsWidgetsCID.h"
|
|
#include "nsXPFEComponentsCID.h"
|
|
#include "nsXREDirProvider.h"
|
|
|
|
#ifdef MOZ_IPC
|
|
#include "nsX11ErrorHandler.h"
|
|
#include "base/at_exit.h"
|
|
#include "base/command_line.h"
|
|
#include "base/message_loop.h"
|
|
#include "base/process_util.h"
|
|
#include "chrome/common/child_process.h"
|
|
|
|
#include "mozilla/ipc/GeckoChildProcessHost.h"
|
|
#include "mozilla/ipc/BrowserProcessSubThread.h"
|
|
#include "ScopedXREEmbed.h"
|
|
|
|
#include "mozilla/plugins/PluginThreadChild.h"
|
|
|
|
#include "mozilla/Monitor.h"
|
|
|
|
#ifdef MOZ_IPDL_TESTS
|
|
#include "mozilla/_ipdltest/IPDLUnitTests.h"
|
|
#include "mozilla/_ipdltest/IPDLUnitTestThreadChild.h"
|
|
|
|
using mozilla::_ipdltest::IPDLUnitTestThreadChild;
|
|
#endif // ifdef MOZ_IPDL_TESTS
|
|
|
|
using mozilla::ipc::GeckoChildProcessHost;
|
|
using mozilla::ipc::BrowserProcessSubThread;
|
|
using mozilla::ipc::ScopedXREEmbed;
|
|
|
|
using mozilla::plugins::PluginThreadChild;
|
|
|
|
using mozilla::Monitor;
|
|
using mozilla::MonitorAutoEnter;
|
|
|
|
using mozilla::startup::sChildProcessType;
|
|
#endif
|
|
|
|
static NS_DEFINE_CID(kAppShellCID, NS_APPSHELL_CID);
|
|
|
|
void
|
|
XRE_GetStaticComponents(nsStaticModuleInfo const **aStaticComponents,
|
|
PRUint32 *aComponentCount)
|
|
{
|
|
*aStaticComponents = kPStaticModules;
|
|
*aComponentCount = kStaticModuleCount;
|
|
}
|
|
|
|
nsresult
|
|
XRE_LockProfileDirectory(nsILocalFile* aDirectory,
|
|
nsISupports* *aLockObject)
|
|
{
|
|
nsCOMPtr<nsIProfileLock> lock;
|
|
|
|
nsresult rv = NS_LockProfilePath(aDirectory, nsnull, nsnull,
|
|
getter_AddRefs(lock));
|
|
if (NS_SUCCEEDED(rv))
|
|
NS_ADDREF(*aLockObject = lock);
|
|
|
|
return rv;
|
|
}
|
|
|
|
static nsStaticModuleInfo *sCombined;
|
|
static PRInt32 sInitCounter;
|
|
|
|
nsresult
|
|
XRE_InitEmbedding(nsILocalFile *aLibXULDirectory,
|
|
nsILocalFile *aAppDirectory,
|
|
nsIDirectoryServiceProvider *aAppDirProvider,
|
|
nsStaticModuleInfo const *aStaticComponents,
|
|
PRUint32 aStaticComponentCount)
|
|
{
|
|
// Initialize some globals to make nsXREDirProvider happy
|
|
static char* kNullCommandLine[] = { nsnull };
|
|
gArgv = kNullCommandLine;
|
|
gArgc = 0;
|
|
|
|
NS_ENSURE_ARG(aLibXULDirectory);
|
|
|
|
if (++sInitCounter > 1) // XXXbsmedberg is this really the right solution?
|
|
return NS_OK;
|
|
|
|
if (!aAppDirectory)
|
|
aAppDirectory = aLibXULDirectory;
|
|
|
|
nsresult rv;
|
|
|
|
new nsXREDirProvider; // This sets gDirServiceProvider
|
|
if (!gDirServiceProvider)
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
rv = gDirServiceProvider->Initialize(aAppDirectory, aLibXULDirectory,
|
|
aAppDirProvider);
|
|
if (NS_FAILED(rv))
|
|
return rv;
|
|
|
|
// Combine the toolkit static components and the app components.
|
|
PRUint32 combinedCount = kStaticModuleCount + aStaticComponentCount;
|
|
|
|
sCombined = new nsStaticModuleInfo[combinedCount];
|
|
if (!sCombined)
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
memcpy(sCombined, kPStaticModules,
|
|
sizeof(nsStaticModuleInfo) * kStaticModuleCount);
|
|
memcpy(sCombined + kStaticModuleCount, aStaticComponents,
|
|
sizeof(nsStaticModuleInfo) * aStaticComponentCount);
|
|
|
|
rv = NS_InitXPCOM3(nsnull, aAppDirectory, gDirServiceProvider,
|
|
sCombined, combinedCount);
|
|
if (NS_FAILED(rv))
|
|
return rv;
|
|
|
|
// We do not need to autoregister components here. The CheckUpdateFile()
|
|
// bits in NS_InitXPCOM3 check for an .autoreg file. If the app wants
|
|
// to autoregister every time (for instance, if it's debug), it can do
|
|
// so after we return from this function.
|
|
|
|
nsCOMPtr<nsIObserver> startupNotifier
|
|
(do_CreateInstance(NS_APPSTARTUPNOTIFIER_CONTRACTID));
|
|
if (!startupNotifier)
|
|
return NS_ERROR_FAILURE;
|
|
|
|
startupNotifier->Observe(nsnull, APPSTARTUP_TOPIC, nsnull);
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
void
|
|
XRE_NotifyProfile()
|
|
{
|
|
NS_ASSERTION(gDirServiceProvider, "XRE_InitEmbedding was not called!");
|
|
gDirServiceProvider->DoStartup();
|
|
}
|
|
|
|
void
|
|
XRE_TermEmbedding()
|
|
{
|
|
if (--sInitCounter != 0)
|
|
return;
|
|
|
|
NS_ASSERTION(gDirServiceProvider,
|
|
"XRE_TermEmbedding without XRE_InitEmbedding");
|
|
|
|
gDirServiceProvider->DoShutdown();
|
|
NS_ShutdownXPCOM(nsnull);
|
|
delete [] sCombined;
|
|
delete gDirServiceProvider;
|
|
}
|
|
|
|
const char*
|
|
XRE_ChildProcessTypeToString(GeckoProcessType aProcessType)
|
|
{
|
|
return (aProcessType < GeckoProcessType_End) ?
|
|
kGeckoProcessTypeString[aProcessType] : nsnull;
|
|
}
|
|
|
|
GeckoProcessType
|
|
XRE_StringToChildProcessType(const char* aProcessTypeString)
|
|
{
|
|
for (int i = 0;
|
|
i < (int) NS_ARRAY_LENGTH(kGeckoProcessTypeString);
|
|
++i) {
|
|
if (!strcmp(kGeckoProcessTypeString[i], aProcessTypeString)) {
|
|
return static_cast<GeckoProcessType>(i);
|
|
}
|
|
}
|
|
return GeckoProcessType_Invalid;
|
|
}
|
|
|
|
#ifdef MOZ_IPC
|
|
namespace mozilla {
|
|
namespace startup {
|
|
GeckoProcessType sChildProcessType = GeckoProcessType_Default;
|
|
}
|
|
}
|
|
|
|
static MessageLoop* sIOMessageLoop;
|
|
|
|
#if defined(MOZ_CRASHREPORTER)
|
|
// FIXME/bug 539522: this out-of-place function is stuck here because
|
|
// IPDL wants access to this crashreporter interface, and
|
|
// crashreporter is built in such a way to make that awkward
|
|
PRBool
|
|
XRE_GetMinidumpForChild(PRUint32 aChildPid, nsIFile** aDump)
|
|
{
|
|
return CrashReporter::GetMinidumpForChild(aChildPid, aDump);
|
|
}
|
|
|
|
PRBool
|
|
XRE_SetRemoteExceptionHandler(const char* aPipe/*= 0*/)
|
|
{
|
|
#if defined(XP_WIN)
|
|
return CrashReporter::SetRemoteExceptionHandler(nsDependentCString(aPipe));
|
|
#elif defined(OS_LINUX)
|
|
return CrashReporter::SetRemoteExceptionHandler();
|
|
#else
|
|
# error "OOP crash reporter unsupported on this platform"
|
|
#endif
|
|
}
|
|
#endif // if defined(MOZ_CRASHREPORTER)
|
|
|
|
nsresult
|
|
XRE_InitChildProcess(int aArgc,
|
|
char* aArgv[],
|
|
GeckoProcessType aProcess)
|
|
{
|
|
NS_ENSURE_ARG_MIN(aArgc, 2);
|
|
NS_ENSURE_ARG_POINTER(aArgv);
|
|
NS_ENSURE_ARG_POINTER(aArgv[0]);
|
|
|
|
sChildProcessType = aProcess;
|
|
|
|
gArgv = aArgv;
|
|
gArgc = aArgc;
|
|
|
|
SetupErrorHandling(aArgv[0]);
|
|
|
|
#if defined(MOZ_WIDGET_GTK2)
|
|
g_thread_init(NULL);
|
|
#endif
|
|
|
|
if (PR_GetEnv("MOZ_DEBUG_CHILD_PROCESS")) {
|
|
#ifdef OS_POSIX
|
|
printf("\n\nCHILDCHILDCHILDCHILD\n debug me @%d\n\n", getpid());
|
|
sleep(30);
|
|
#elif defined(OS_WIN)
|
|
printf("\n\nCHILDCHILDCHILDCHILD\n debug me @%d\n\n", _getpid());
|
|
Sleep(30000);
|
|
#endif
|
|
}
|
|
|
|
// child processes launched by GeckoChildProcessHost get this magic
|
|
// argument appended to their command lines
|
|
const char* const parentPIDString = aArgv[aArgc-1];
|
|
NS_ABORT_IF_FALSE(parentPIDString, "NULL parent PID");
|
|
--aArgc;
|
|
|
|
char* end = 0;
|
|
base::ProcessId parentPID = strtol(parentPIDString, &end, 10);
|
|
NS_ABORT_IF_FALSE(!*end, "invalid parent PID");
|
|
|
|
base::ProcessHandle parentHandle;
|
|
bool ok = base::OpenProcessHandle(parentPID, &parentHandle);
|
|
NS_ABORT_IF_FALSE(ok, "can't open handle to parent");
|
|
|
|
base::AtExitManager exitManager;
|
|
|
|
NS_LogInit();
|
|
|
|
int rv = XRE_InitCommandLine(aArgc, aArgv);
|
|
if (NS_FAILED(rv)) {
|
|
NS_LogTerm();
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
MessageLoopForIO mainMessageLoop;
|
|
|
|
{
|
|
ChildThread* mainThread;
|
|
|
|
switch (aProcess) {
|
|
case GeckoProcessType_Default:
|
|
NS_RUNTIMEABORT("This makes no sense");
|
|
break;
|
|
|
|
case GeckoProcessType_Plugin:
|
|
mainThread = new PluginThreadChild(parentHandle);
|
|
break;
|
|
|
|
case GeckoProcessType_IPDLUnitTest:
|
|
#ifdef MOZ_IPDL_TESTS
|
|
mainThread = new IPDLUnitTestThreadChild(parentHandle);
|
|
#else
|
|
NS_RUNTIMEABORT("rebuild with --enable-ipdl-tests");
|
|
#endif
|
|
break;
|
|
|
|
default:
|
|
NS_RUNTIMEABORT("Unknown main thread class");
|
|
}
|
|
|
|
ChildProcess process(mainThread);
|
|
|
|
// Do IPC event loop
|
|
sIOMessageLoop = MessageLoop::current();
|
|
|
|
sIOMessageLoop->Run();
|
|
|
|
sIOMessageLoop = nsnull;
|
|
}
|
|
|
|
NS_LogTerm();
|
|
return XRE_DeinitCommandLine();
|
|
}
|
|
|
|
MessageLoop*
|
|
XRE_GetIOMessageLoop()
|
|
{
|
|
if (sChildProcessType == GeckoProcessType_Default) {
|
|
NS_ASSERTION(!sIOMessageLoop, "Shouldn't be set on parent process!");
|
|
return BrowserProcessSubThread::GetMessageLoop(BrowserProcessSubThread::IO);
|
|
}
|
|
return sIOMessageLoop;
|
|
}
|
|
|
|
namespace {
|
|
|
|
class MainFunctionRunnable : public nsRunnable
|
|
{
|
|
public:
|
|
NS_DECL_NSIRUNNABLE
|
|
|
|
MainFunctionRunnable(MainFunction aFunction,
|
|
void* aData)
|
|
: mFunction(aFunction),
|
|
mData(aData)
|
|
{
|
|
NS_ASSERTION(aFunction, "Don't give me a null pointer!");
|
|
}
|
|
|
|
private:
|
|
MainFunction mFunction;
|
|
void* mData;
|
|
};
|
|
|
|
} /* anonymous namespace */
|
|
|
|
NS_IMETHODIMP
|
|
MainFunctionRunnable::Run()
|
|
{
|
|
mFunction(mData);
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult
|
|
XRE_InitParentProcess(int aArgc,
|
|
char* aArgv[],
|
|
MainFunction aMainFunction,
|
|
void* aMainFunctionData)
|
|
{
|
|
NS_ENSURE_ARG_MIN(aArgc, 1);
|
|
NS_ENSURE_ARG_POINTER(aArgv);
|
|
NS_ENSURE_ARG_POINTER(aArgv[0]);
|
|
|
|
int rv = XRE_InitCommandLine(aArgc, aArgv);
|
|
if (NS_FAILED(rv))
|
|
return NS_ERROR_FAILURE;
|
|
|
|
ScopedXREEmbed embed;
|
|
|
|
{
|
|
embed.Start();
|
|
|
|
nsCOMPtr<nsIAppShell> appShell(do_GetService(kAppShellCID));
|
|
NS_ENSURE_TRUE(appShell, NS_ERROR_FAILURE);
|
|
|
|
if (aMainFunction) {
|
|
nsCOMPtr<nsIRunnable> runnable =
|
|
new MainFunctionRunnable(aMainFunction, aMainFunctionData);
|
|
NS_ENSURE_TRUE(runnable, NS_ERROR_OUT_OF_MEMORY);
|
|
|
|
nsresult rv = NS_DispatchToCurrentThread(runnable);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
}
|
|
|
|
// Do event loop
|
|
if (NS_FAILED(appShell->Run())) {
|
|
NS_WARNING("Failed to run appshell");
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
}
|
|
|
|
return XRE_DeinitCommandLine();
|
|
}
|
|
|
|
#ifdef MOZ_IPDL_TESTS
|
|
//-----------------------------------------------------------------------------
|
|
// IPDL unit test
|
|
|
|
int
|
|
XRE_RunIPDLTest(int aArgc, char** aArgv)
|
|
{
|
|
if (aArgc < 2) {
|
|
fprintf(stderr, "TEST-UNEXPECTED-FAIL | <---> | insufficient #args, need at least 2\n");
|
|
return 1;
|
|
}
|
|
|
|
void* data = reinterpret_cast<void*>(aArgv[aArgc-1]);
|
|
|
|
nsresult rv =
|
|
XRE_InitParentProcess(
|
|
--aArgc, aArgv, mozilla::_ipdltest::IPDLUnitTestMain, data);
|
|
NS_ENSURE_SUCCESS(rv, 1);
|
|
|
|
return 0;
|
|
}
|
|
#endif // ifdef MOZ_IPDL_TESTS
|
|
|
|
nsresult
|
|
XRE_RunAppShell()
|
|
{
|
|
nsCOMPtr<nsIAppShell> appShell(do_GetService(kAppShellCID));
|
|
NS_ENSURE_TRUE(appShell, NS_ERROR_FAILURE);
|
|
|
|
return appShell->Run();
|
|
}
|
|
|
|
void
|
|
XRE_ShutdownChildProcess()
|
|
{
|
|
NS_ABORT_IF_FALSE(MessageLoopForUI::current(), "Wrong thread!");
|
|
|
|
MessageLoop* ioLoop = XRE_GetIOMessageLoop();
|
|
NS_ABORT_IF_FALSE(!!ioLoop, "Bad shutdown order");
|
|
|
|
ioLoop->PostTask(FROM_HERE, new MessageLoop::QuitTask());
|
|
}
|
|
|
|
#ifdef MOZ_X11
|
|
void
|
|
XRE_InstallX11ErrorHandler()
|
|
{
|
|
InstallX11ErrorHandler();
|
|
}
|
|
#endif
|
|
|
|
#endif // MOZ_IPC
|
|
|