mirror of
https://github.com/classilla/tenfourfox.git
synced 2024-09-10 11:54:40 +00:00
367 lines
10 KiB
C++
367 lines
10 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "nsXULAppAPI.h"
|
|
#include "mozilla/AppData.h"
|
|
#include "application.ini.h"
|
|
#include "nsXPCOMGlue.h"
|
|
#if defined(XP_WIN)
|
|
#include <windows.h>
|
|
#include <stdlib.h>
|
|
#include <io.h>
|
|
#include <fcntl.h>
|
|
#elif defined(XP_UNIX)
|
|
#include <sys/resource.h>
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#ifdef XP_MACOSX
|
|
#include "MacQuirks.h"
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
#include <stdarg.h>
|
|
#include <time.h>
|
|
|
|
#include "nsCOMPtr.h"
|
|
#include "nsIFile.h"
|
|
#include "nsStringGlue.h"
|
|
|
|
#ifdef XP_WIN
|
|
// we want a wmain entry point
|
|
#ifdef MOZ_ASAN
|
|
// ASAN requires firefox.exe to be built with -MD, and it's OK if we don't
|
|
// support Windows XP SP2 in ASAN builds.
|
|
#define XRE_DONT_SUPPORT_XPSP2
|
|
#endif
|
|
#include "nsWindowsWMain.cpp"
|
|
#if defined(_MSC_VER) && (_MSC_VER < 1900)
|
|
#define snprintf _snprintf
|
|
#endif
|
|
#define strcasecmp _stricmp
|
|
#endif
|
|
#include "BinaryPath.h"
|
|
|
|
#include "nsXPCOMPrivate.h" // for MAXPATHLEN and XPCOM_DLL
|
|
|
|
#include "mozilla/Telemetry.h"
|
|
#include "mozilla/WindowsDllBlocklist.h"
|
|
|
|
using namespace mozilla;
|
|
|
|
#ifdef XP_MACOSX
|
|
#define kOSXResourcesFolder "Resources"
|
|
#endif
|
|
#define kDesktopFolder "browser"
|
|
|
|
static void Output(const char *fmt, ... )
|
|
{
|
|
va_list ap;
|
|
va_start(ap, fmt);
|
|
|
|
#ifndef XP_WIN
|
|
vfprintf(stderr, fmt, ap);
|
|
#else
|
|
char msg[2048];
|
|
vsnprintf_s(msg, _countof(msg), _TRUNCATE, fmt, ap);
|
|
|
|
wchar_t wide_msg[2048];
|
|
MultiByteToWideChar(CP_UTF8,
|
|
0,
|
|
msg,
|
|
-1,
|
|
wide_msg,
|
|
_countof(wide_msg));
|
|
#if MOZ_WINCONSOLE
|
|
fwprintf_s(stderr, wide_msg);
|
|
#else
|
|
// Linking user32 at load-time interferes with the DLL blocklist (bug 932100).
|
|
// This is a rare codepath, so we can load user32 at run-time instead.
|
|
HMODULE user32 = LoadLibraryW(L"user32.dll");
|
|
if (user32) {
|
|
decltype(MessageBoxW)* messageBoxW =
|
|
(decltype(MessageBoxW)*) GetProcAddress(user32, "MessageBoxW");
|
|
if (messageBoxW) {
|
|
messageBoxW(nullptr, wide_msg, L"Firefox", MB_OK
|
|
| MB_ICONERROR
|
|
| MB_SETFOREGROUND);
|
|
}
|
|
FreeLibrary(user32);
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
va_end(ap);
|
|
}
|
|
|
|
/**
|
|
* Return true if |arg| matches the given argument name.
|
|
*/
|
|
static bool IsArg(const char* arg, const char* s)
|
|
{
|
|
if (*arg == '-')
|
|
{
|
|
if (*++arg == '-')
|
|
++arg;
|
|
return !strcasecmp(arg, s);
|
|
}
|
|
|
|
#if defined(XP_WIN)
|
|
if (*arg == '/')
|
|
return !strcasecmp(++arg, s);
|
|
#endif
|
|
|
|
return false;
|
|
}
|
|
|
|
XRE_GetFileFromPathType XRE_GetFileFromPath;
|
|
XRE_CreateAppDataType XRE_CreateAppData;
|
|
XRE_FreeAppDataType XRE_FreeAppData;
|
|
XRE_TelemetryAccumulateType XRE_TelemetryAccumulate;
|
|
XRE_StartupTimelineRecordType XRE_StartupTimelineRecord;
|
|
XRE_mainType XRE_main;
|
|
XRE_StopLateWriteChecksType XRE_StopLateWriteChecks;
|
|
|
|
static const nsDynamicFunctionLoad kXULFuncs[] = {
|
|
{ "XRE_GetFileFromPath", (NSFuncPtr*) &XRE_GetFileFromPath },
|
|
{ "XRE_CreateAppData", (NSFuncPtr*) &XRE_CreateAppData },
|
|
{ "XRE_FreeAppData", (NSFuncPtr*) &XRE_FreeAppData },
|
|
{ "XRE_TelemetryAccumulate", (NSFuncPtr*) &XRE_TelemetryAccumulate },
|
|
{ "XRE_StartupTimelineRecord", (NSFuncPtr*) &XRE_StartupTimelineRecord },
|
|
{ "XRE_main", (NSFuncPtr*) &XRE_main },
|
|
{ "XRE_StopLateWriteChecks", (NSFuncPtr*) &XRE_StopLateWriteChecks },
|
|
{ nullptr, nullptr }
|
|
};
|
|
|
|
static int do_main(int argc, char* argv[], nsIFile *xreDirectory)
|
|
{
|
|
nsCOMPtr<nsIFile> appini;
|
|
nsresult rv;
|
|
uint32_t mainFlags = 0;
|
|
|
|
// Allow firefox.exe to launch XULRunner apps via -app <application.ini>
|
|
// Note that -app must be the *first* argument.
|
|
const char *appDataFile = getenv("XUL_APP_FILE");
|
|
if (appDataFile && *appDataFile) {
|
|
rv = XRE_GetFileFromPath(appDataFile, getter_AddRefs(appini));
|
|
if (NS_FAILED(rv)) {
|
|
Output("Invalid path found: '%s'", appDataFile);
|
|
return 255;
|
|
}
|
|
}
|
|
else if (argc > 1 && IsArg(argv[1], "app")) {
|
|
if (argc == 2) {
|
|
Output("Incorrect number of arguments passed to -app");
|
|
return 255;
|
|
}
|
|
|
|
rv = XRE_GetFileFromPath(argv[2], getter_AddRefs(appini));
|
|
if (NS_FAILED(rv)) {
|
|
Output("application.ini path not recognized: '%s'", argv[2]);
|
|
return 255;
|
|
}
|
|
|
|
char appEnv[MAXPATHLEN];
|
|
snprintf(appEnv, MAXPATHLEN, "XUL_APP_FILE=%s", argv[2]);
|
|
if (putenv(appEnv)) {
|
|
Output("Couldn't set %s.\n", appEnv);
|
|
return 255;
|
|
}
|
|
argv[2] = argv[0];
|
|
argv += 2;
|
|
argc -= 2;
|
|
}
|
|
|
|
if (appini) {
|
|
nsXREAppData *appData;
|
|
rv = XRE_CreateAppData(appini, &appData);
|
|
if (NS_FAILED(rv)) {
|
|
Output("Couldn't read application.ini");
|
|
return 255;
|
|
}
|
|
// xreDirectory already has a refcount from NS_NewLocalFile
|
|
appData->xreDirectory = xreDirectory;
|
|
int result = XRE_main(argc, argv, appData, mainFlags);
|
|
XRE_FreeAppData(appData);
|
|
return result;
|
|
}
|
|
|
|
ScopedAppData appData(&sAppData);
|
|
nsCOMPtr<nsIFile> exeFile;
|
|
rv = mozilla::BinaryPath::GetFile(argv[0], getter_AddRefs(exeFile));
|
|
if (NS_FAILED(rv)) {
|
|
Output("Couldn't find the application directory.\n");
|
|
return 255;
|
|
}
|
|
|
|
nsCOMPtr<nsIFile> greDir;
|
|
exeFile->GetParent(getter_AddRefs(greDir));
|
|
#ifdef XP_MACOSX
|
|
greDir->SetNativeLeafName(NS_LITERAL_CSTRING(kOSXResourcesFolder));
|
|
#endif
|
|
nsCOMPtr<nsIFile> appSubdir;
|
|
greDir->Clone(getter_AddRefs(appSubdir));
|
|
appSubdir->Append(NS_LITERAL_STRING(kDesktopFolder));
|
|
|
|
SetStrongPtr(appData.directory, static_cast<nsIFile*>(appSubdir.get()));
|
|
// xreDirectory already has a refcount from NS_NewLocalFile
|
|
appData.xreDirectory = xreDirectory;
|
|
|
|
return XRE_main(argc, argv, &appData, mainFlags);
|
|
}
|
|
|
|
static bool
|
|
FileExists(const char *path)
|
|
{
|
|
#ifdef XP_WIN
|
|
wchar_t wideDir[MAX_PATH];
|
|
MultiByteToWideChar(CP_UTF8, 0, path, -1, wideDir, MAX_PATH);
|
|
DWORD fileAttrs = GetFileAttributesW(wideDir);
|
|
return fileAttrs != INVALID_FILE_ATTRIBUTES;
|
|
#else
|
|
return access(path, R_OK) == 0;
|
|
#endif
|
|
}
|
|
|
|
static nsresult
|
|
InitXPCOMGlue(const char *argv0, nsIFile **xreDirectory)
|
|
{
|
|
char exePath[MAXPATHLEN];
|
|
|
|
nsresult rv = mozilla::BinaryPath::Get(argv0, exePath);
|
|
if (NS_FAILED(rv)) {
|
|
Output("Couldn't find the application directory.\n");
|
|
return rv;
|
|
}
|
|
|
|
char *lastSlash = strrchr(exePath, XPCOM_FILE_PATH_SEPARATOR[0]);
|
|
if (!lastSlash || (size_t(lastSlash - exePath) > MAXPATHLEN -
|
|
sizeof(XPCOM_DLL) - 1))
|
|
return NS_ERROR_FAILURE;
|
|
|
|
strcpy(lastSlash + 1, XPCOM_DLL);
|
|
|
|
if (!FileExists(exePath)) {
|
|
Output("Could not find the Mozilla runtime.\n");
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
// We do this because of data in bug 771745
|
|
XPCOMGlueEnablePreload();
|
|
|
|
rv = XPCOMGlueStartup(exePath);
|
|
if (NS_FAILED(rv)) {
|
|
Output("Couldn't load XPCOM.\n");
|
|
return rv;
|
|
}
|
|
|
|
rv = XPCOMGlueLoadXULFunctions(kXULFuncs);
|
|
if (NS_FAILED(rv)) {
|
|
Output("Couldn't load XRE functions.\n");
|
|
return rv;
|
|
}
|
|
|
|
// This will set this thread as the main thread.
|
|
NS_LogInit();
|
|
|
|
// chop XPCOM_DLL off exePath
|
|
*lastSlash = '\0';
|
|
#ifdef XP_MACOSX
|
|
lastSlash = strrchr(exePath, XPCOM_FILE_PATH_SEPARATOR[0]);
|
|
strcpy(lastSlash + 1, kOSXResourcesFolder);
|
|
#endif
|
|
#ifdef XP_WIN
|
|
rv = NS_NewLocalFile(NS_ConvertUTF8toUTF16(exePath), false,
|
|
xreDirectory);
|
|
#else
|
|
rv = NS_NewNativeLocalFile(nsDependentCString(exePath), false,
|
|
xreDirectory);
|
|
#endif
|
|
|
|
return rv;
|
|
}
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
mozilla::TimeStamp start = mozilla::TimeStamp::Now();
|
|
|
|
#ifdef XP_MACOSX
|
|
TriggerQuirks();
|
|
#endif
|
|
|
|
int gotCounters;
|
|
#if defined(XP_UNIX)
|
|
struct rusage initialRUsage;
|
|
gotCounters = !getrusage(RUSAGE_SELF, &initialRUsage);
|
|
#elif defined(XP_WIN)
|
|
IO_COUNTERS ioCounters;
|
|
gotCounters = GetProcessIoCounters(GetCurrentProcess(), &ioCounters);
|
|
#else
|
|
#error "Unknown platform" // having this here keeps cppcheck happy
|
|
#endif
|
|
|
|
nsIFile *xreDirectory;
|
|
|
|
#ifdef HAS_DLL_BLOCKLIST
|
|
DllBlocklist_Initialize();
|
|
|
|
#ifdef DEBUG
|
|
// In order to be effective against AppInit DLLs, the blocklist must be
|
|
// initialized before user32.dll is loaded into the process (bug 932100).
|
|
if (GetModuleHandleA("user32.dll")) {
|
|
fprintf(stderr, "DLL blocklist was unable to intercept AppInit DLLs.\n");
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
nsresult rv = InitXPCOMGlue(argv[0], &xreDirectory);
|
|
if (NS_FAILED(rv)) {
|
|
return 255;
|
|
}
|
|
|
|
XRE_StartupTimelineRecord(mozilla::StartupTimeline::START, start);
|
|
|
|
if (gotCounters) {
|
|
#if defined(XP_WIN)
|
|
XRE_TelemetryAccumulate(mozilla::Telemetry::EARLY_GLUESTARTUP_READ_OPS,
|
|
int(ioCounters.ReadOperationCount));
|
|
XRE_TelemetryAccumulate(mozilla::Telemetry::EARLY_GLUESTARTUP_READ_TRANSFER,
|
|
int(ioCounters.ReadTransferCount / 1024));
|
|
IO_COUNTERS newIoCounters;
|
|
if (GetProcessIoCounters(GetCurrentProcess(), &newIoCounters)) {
|
|
XRE_TelemetryAccumulate(mozilla::Telemetry::GLUESTARTUP_READ_OPS,
|
|
int(newIoCounters.ReadOperationCount - ioCounters.ReadOperationCount));
|
|
XRE_TelemetryAccumulate(mozilla::Telemetry::GLUESTARTUP_READ_TRANSFER,
|
|
int((newIoCounters.ReadTransferCount - ioCounters.ReadTransferCount) / 1024));
|
|
}
|
|
#elif defined(XP_UNIX)
|
|
XRE_TelemetryAccumulate(mozilla::Telemetry::EARLY_GLUESTARTUP_HARD_FAULTS,
|
|
int(initialRUsage.ru_majflt));
|
|
struct rusage newRUsage;
|
|
if (!getrusage(RUSAGE_SELF, &newRUsage)) {
|
|
XRE_TelemetryAccumulate(mozilla::Telemetry::GLUESTARTUP_HARD_FAULTS,
|
|
int(newRUsage.ru_majflt - initialRUsage.ru_majflt));
|
|
}
|
|
#else
|
|
#error "Unknown platform" // having this here keeps cppcheck happy
|
|
#endif
|
|
}
|
|
|
|
int result = do_main(argc, argv, xreDirectory);
|
|
|
|
NS_LogTerm();
|
|
|
|
#ifdef XP_MACOSX
|
|
// Allow writes again. While we would like to catch writes from static
|
|
// destructors to allow early exits to use _exit, we know that there is
|
|
// at least one such write that we don't control (see bug 826029). For
|
|
// now we enable writes again and early exits will have to use exit instead
|
|
// of _exit.
|
|
XRE_StopLateWriteChecks();
|
|
#endif
|
|
|
|
return result;
|
|
}
|