1 : /* ***** BEGIN LICENSE BLOCK *****
2 : * Version: MPL 1.1/GPL 2.0/LGPL 2.1
3 : *
4 : * The contents of this file are subject to the Mozilla Public License Version
5 : * 1.1 (the "License"); you may not use this file except in compliance with
6 : * the License. You may obtain a copy of the License at
7 : * http://www.mozilla.org/MPL/
8 : *
9 : * Software distributed under the License is distributed on an "AS IS" basis,
10 : * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
11 : * for the specific language governing rights and limitations under the
12 : * License.
13 : *
14 : * The Original Code is Mozilla Plugin App.
15 : *
16 : * The Initial Developer of the Original Code is
17 : * Ben Turner <bent.mozilla@gmail.com>.
18 : * Portions created by the Initial Developer are Copyright (C) 2009
19 : * the Initial Developer. All Rights Reserved.
20 : *
21 : * Contributor(s):
22 : *
23 : * Alternatively, the contents of this file may be used under the terms of
24 : * either the GNU General Public License Version 2 or later (the "GPL"), or
25 : * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
26 : * in which case the provisions of the GPL or the LGPL are applicable instead
27 : * of those above. If you wish to allow use of your version of this file only
28 : * under the terms of either the GPL or the LGPL, and not to allow others to
29 : * use your version of this file under the terms of the MPL, indicate your
30 : * decision by deleting the provisions above and replace them with the notice
31 : * and other provisions required by the GPL or the LGPL. If you do not delete
32 : * the provisions above, a recipient may use your version of this file under
33 : * the terms of any one of the MPL, the GPL or the LGPL.
34 : *
35 : * ***** END LICENSE BLOCK ***** */
36 :
37 : #include "ScopedXREEmbed.h"
38 :
39 : #include "base/command_line.h"
40 : #include "base/string_util.h"
41 :
42 : #include "nsIFile.h"
43 : #include "nsILocalFile.h"
44 :
45 : #include "nsCOMPtr.h"
46 : #include "nsServiceManagerUtils.h"
47 : #include "nsStringGlue.h"
48 : #include "nsXULAppAPI.h"
49 :
50 : using mozilla::ipc::ScopedXREEmbed;
51 :
52 1 : ScopedXREEmbed::ScopedXREEmbed()
53 1 : : mShouldKillEmbedding(false)
54 : {
55 1 : NS_LogInit();
56 1 : }
57 :
58 0 : ScopedXREEmbed::~ScopedXREEmbed()
59 : {
60 0 : Stop();
61 0 : NS_LogTerm();
62 0 : }
63 :
64 : void
65 0 : ScopedXREEmbed::Start()
66 : {
67 0 : std::string path;
68 : #if defined(OS_WIN)
69 : path = WideToUTF8(CommandLine::ForCurrentProcess()->program());
70 : #elif defined(OS_POSIX)
71 0 : path = CommandLine::ForCurrentProcess()->argv()[0];
72 : #else
73 : # error Sorry
74 : #endif
75 :
76 0 : nsCOMPtr<nsILocalFile> localFile;
77 0 : nsresult rv = XRE_GetBinaryPath(path.c_str(), getter_AddRefs(localFile));
78 0 : if (NS_FAILED(rv))
79 : return;
80 :
81 0 : nsCOMPtr<nsIFile> parent;
82 0 : rv = localFile->GetParent(getter_AddRefs(parent));
83 0 : if (NS_FAILED(rv))
84 : return;
85 :
86 0 : localFile = do_QueryInterface(parent);
87 0 : NS_ENSURE_TRUE(localFile,);
88 :
89 : #ifdef OS_MACOSX
90 : if (XRE_GetProcessType() == GeckoProcessType_Content) {
91 : // We're an XPCOM-using subprocess. Walk out of
92 : // [subprocess].app/Contents/MacOS to the real GRE dir.
93 : rv = localFile->GetParent(getter_AddRefs(parent));
94 : if (NS_FAILED(rv))
95 : return;
96 :
97 : localFile = do_QueryInterface(parent);
98 : NS_ENSURE_TRUE(localFile,);
99 :
100 : rv = localFile->GetParent(getter_AddRefs(parent));
101 : if (NS_FAILED(rv))
102 : return;
103 :
104 : localFile = do_QueryInterface(parent);
105 : NS_ENSURE_TRUE(localFile,);
106 :
107 : rv = localFile->GetParent(getter_AddRefs(parent));
108 : if (NS_FAILED(rv))
109 : return;
110 :
111 : localFile = do_QueryInterface(parent);
112 : NS_ENSURE_TRUE(localFile,);
113 : }
114 : #endif
115 :
116 0 : rv = XRE_InitEmbedding2(localFile, localFile, nsnull);
117 0 : if (NS_FAILED(rv))
118 : return;
119 :
120 0 : mShouldKillEmbedding = true;
121 : }
122 :
123 : void
124 0 : ScopedXREEmbed::Stop()
125 : {
126 0 : if (mShouldKillEmbedding) {
127 0 : XRE_TermEmbedding();
128 0 : mShouldKillEmbedding = false;
129 : }
130 0 : }
|