ipc/glue/ScopedXREEmbed.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "ScopedXREEmbed.h"
   8
   9 #include "base/command_line.h"
  10 #include "base/string_util.h"
  11
  12 #include "nsIFile.h"
  13
  14 #include "nsCOMPtr.h"
  15 #include "nsServiceManagerUtils.h"
  16 #include "nsString.h"
  17 #include "nsXULAppAPI.h"
  18
  19 using mozilla::ipc::ScopedXREEmbed;
  20
  21 ScopedXREEmbed::ScopedXREEmbed() : mShouldKillEmbedding(false) { NS_LogInit(); }
  22
  23 ScopedXREEmbed::~ScopedXREEmbed() {
  24   Stop();
  25   NS_LogTerm();
  26 }
  27
  28 void ScopedXREEmbed::SetAppDir(const nsACString& aPath) {
  29   bool flag;
  30   nsresult rv =
  31       XRE_GetFileFromPath(aPath.BeginReading(), getter_AddRefs(mAppDir));
  32   if (NS_FAILED(rv) || NS_FAILED(mAppDir->Exists(&flag)) || !flag) {
  33     NS_WARNING("Invalid application directory passed to content process.");
  34     mAppDir = nullptr;
  35   }
  36 }
  37
  38 void ScopedXREEmbed::Start() {
  39   nsCOMPtr<nsIFile> localFile;
  40   nsresult rv = XRE_GetBinaryPath(getter_AddRefs(localFile));
  41   if (NS_FAILED(rv)) return;
  42
  43   nsCOMPtr<nsIFile> parent;
  44   rv = localFile->GetParent(getter_AddRefs(parent));
  45   if (NS_FAILED(rv)) return;
  46
  47   localFile = parent;
  48   NS_ENSURE_TRUE_VOID(localFile);
  49
  50 #ifdef OS_MACOSX
  51   if (XRE_IsContentProcess()) {
  52     // We're an XPCOM-using subprocess.  Walk out of
  53     // [subprocess].app/Contents/MacOS to the real GRE dir.
  54     rv = localFile->GetParent(getter_AddRefs(parent));
  55     if (NS_FAILED(rv)) return;
  56
  57     localFile = parent;
  58     NS_ENSURE_TRUE_VOID(localFile);
  59
  60     rv = localFile->GetParent(getter_AddRefs(parent));
  61     if (NS_FAILED(rv)) return;
  62
  63     localFile = parent;
  64     NS_ENSURE_TRUE_VOID(localFile);
  65
  66     rv = localFile->GetParent(getter_AddRefs(parent));
  67     if (NS_FAILED(rv)) return;
  68
  69     localFile = parent;
  70     NS_ENSURE_TRUE_VOID(localFile);
  71
  72     rv = localFile->SetNativeLeafName("Resources"_ns);
  73     if (NS_FAILED(rv)) {
  74       return;
  75     }
  76   }
  77 #endif
  78
  79   if (mAppDir)
  80     rv = XRE_InitEmbedding2(localFile, mAppDir, nullptr);
  81   else
  82     rv = XRE_InitEmbedding2(localFile, localFile, nullptr);
  83   if (NS_FAILED(rv)) return;
  84
  85   mShouldKillEmbedding = true;
  86 }
  87
  88 void ScopedXREEmbed::Stop() {
  89   if (mShouldKillEmbedding) {
  90     XRE_TermEmbedding();
  91     mShouldKillEmbedding = false;
  92   }
  93 }