OLD | NEW |
| (Empty) |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "services/shell/runner/host/native_application_support.h" | |
6 | |
7 #include <stddef.h> | |
8 | |
9 #include "base/command_line.h" | |
10 #include "base/files/file_path.h" | |
11 #include "base/files/file_util.h" | |
12 #include "base/logging.h" | |
13 #include "mojo/edk/embedder/entrypoints.h" | |
14 #include "mojo/public/c/system/thunks.h" | |
15 | |
16 namespace shell { | |
17 | |
18 namespace { | |
19 | |
20 template <typename Thunks> | |
21 bool SetThunks(Thunks (*make_thunks)(), | |
22 const char* function_name, | |
23 base::NativeLibrary library) { | |
24 typedef size_t (*SetThunksFn)(const Thunks* thunks); | |
25 SetThunksFn set_thunks = reinterpret_cast<SetThunksFn>( | |
26 base::GetFunctionPointerFromNativeLibrary(library, function_name)); | |
27 if (!set_thunks) | |
28 return false; | |
29 Thunks thunks = make_thunks(); | |
30 size_t expected_size = set_thunks(&thunks); | |
31 if (expected_size > sizeof(Thunks)) { | |
32 LOG(ERROR) << "Invalid app library: expected " << function_name | |
33 << " to return thunks of size: " << expected_size; | |
34 return false; | |
35 } | |
36 return true; | |
37 } | |
38 | |
39 } // namespace | |
40 | |
41 base::NativeLibrary LoadNativeApplication(const base::FilePath& app_path) { | |
42 DVLOG(2) << "Loading Mojo app in process from library: " << app_path.value(); | |
43 | |
44 base::NativeLibraryLoadError error; | |
45 base::NativeLibrary app_library = base::LoadNativeLibrary(app_path, &error); | |
46 LOG_IF(ERROR, !app_library) | |
47 << "Failed to load app library (path: " << app_path.value() | |
48 << " reason: " << error.ToString() << ")"; | |
49 return app_library; | |
50 } | |
51 | |
52 bool RunNativeApplication(base::NativeLibrary app_library, | |
53 mojom::ServiceRequest request) { | |
54 // Tolerate |app_library| being null, to make life easier for callers. | |
55 if (!app_library) | |
56 return false; | |
57 | |
58 // Thunks aren't needed/used in component build, since the thunked methods | |
59 // just live in their own dynamically loaded library. | |
60 #if !defined(COMPONENT_BUILD) | |
61 if (!SetThunks(&mojo::edk::MakeSystemThunks, "MojoSetSystemThunks", | |
62 app_library)) { | |
63 LOG(ERROR) << "MojoSetSystemThunks not found"; | |
64 return false; | |
65 } | |
66 | |
67 #if !defined(OS_WIN) | |
68 // On Windows, initializing base::CommandLine with null parameters gets the | |
69 // process's command line from the OS. Other platforms need it to be passed | |
70 // in. This needs to be passed in before the app initializes the command line, | |
71 // which is done as soon as it loads. | |
72 typedef void (*InitCommandLineArgs)(int, const char* const*); | |
73 InitCommandLineArgs init_command_line_args = | |
74 reinterpret_cast<InitCommandLineArgs>( | |
75 base::GetFunctionPointerFromNativeLibrary(app_library, | |
76 "InitCommandLineArgs")); | |
77 if (init_command_line_args) { | |
78 int argc = 0; | |
79 base::CommandLine* cmd_line = base::CommandLine::ForCurrentProcess(); | |
80 const char** argv = new const char*[cmd_line->argv().size()]; | |
81 for (auto& arg : cmd_line->argv()) | |
82 argv[argc++] = arg.c_str(); | |
83 init_command_line_args(argc, argv); | |
84 } | |
85 #endif | |
86 | |
87 #endif // !defined(COMPONENT_BUILD) | |
88 | |
89 typedef MojoResult (*ServiceMainFunction)(MojoHandle); | |
90 ServiceMainFunction main_function = reinterpret_cast<ServiceMainFunction>( | |
91 base::GetFunctionPointerFromNativeLibrary(app_library, "ServiceMain")); | |
92 if (!main_function) { | |
93 LOG(ERROR) << "ServiceMain not found"; | |
94 return false; | |
95 } | |
96 // |ServiceMain()| takes ownership of the service handle. | |
97 MojoHandle handle = request.PassMessagePipe().release().value(); | |
98 MojoResult result = main_function(handle); | |
99 if (result != MOJO_RESULT_OK) { | |
100 LOG(ERROR) << "ServiceMain returned error (result: " << result << ")"; | |
101 } | |
102 return true; | |
103 } | |
104 | |
105 } // namespace shell | |
OLD | NEW |