OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 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 <cstring> | |
6 #include <string> | |
7 | |
8 #include "base/at_exit.h" | |
9 #include "base/memory/scoped_ptr.h" | |
10 #include "base/strings/utf_string_conversions.h" | |
11 #include "build/build_config.h" | |
12 #include "chrome/test/chromedriver/chrome/log.h" | |
13 #include "chrome/test/chromedriver/chromedriver.h" | |
14 #include "chrome/test/chromedriver/command_executor.h" | |
15 #include "chrome/test/chromedriver/command_executor_impl.h" | |
16 | |
17 #if defined(OS_WIN) | |
18 #include <windows.h> | |
19 #endif | |
20 | |
21 #if defined(OS_WIN) | |
22 #define EXPORT __declspec(dllexport) | |
23 #else | |
24 #define EXPORT __attribute__((visibility("default"))) | |
25 #endif | |
26 | |
27 namespace { | |
28 | |
29 base::AtExitManager* g_at_exit = NULL; | |
30 Log* g_log = NULL; | |
31 | |
32 } // namespace | |
33 | |
34 extern "C" { | |
35 | |
36 // Synchronously executes the given command. Thread safe. | |
37 // Every call dynamically allocates a |response| string, which must be | |
38 // deallocated by calling |Free|. This |response| string is not guaranteed | |
39 // to be null terminated, and may contain null characters within. You | |
40 // should use the given response size to construct the string. | |
41 void EXPORT ExecuteCommand( | |
42 const char* command, | |
43 unsigned int command_size, | |
44 char** response, | |
45 unsigned int* response_size) { | |
46 std::string command_str(command, command_size); | |
47 std::string json; | |
48 ExecuteCommand(command_str, &json); | |
49 *response = new char[json.length()]; | |
50 std::memcpy(*response, json.c_str(), json.length()); | |
51 *response_size = json.length(); | |
52 } | |
53 | |
54 void EXPORT Free(char* p) { | |
55 delete [] p; | |
56 } | |
57 | |
58 } // extern "C" | |
59 | |
60 #if defined(OS_WIN) | |
61 BOOL APIENTRY DllMain(HMODULE module, DWORD reason, void* reserved) { | |
62 if (reason == DLL_PROCESS_ATTACH) { | |
63 g_at_exit = new base::AtExitManager(); | |
64 g_log = new Logger(); | |
65 Init(scoped_ptr<CommandExecutor>(new CommandExecutorImpl(g_log))); | |
66 } | |
67 if (reason == DLL_PROCESS_DETACH) { | |
68 // If |reserved| is not null, the process is terminating and | |
69 // ChromeDriver's threads have already been signaled to exit. | |
70 // Just leak and let the OS clean it up. | |
71 if (reserved) | |
72 return TRUE; | |
73 Shutdown(); | |
74 delete g_log; | |
75 delete g_at_exit; | |
76 } | |
77 return TRUE; | |
78 } | |
79 #else | |
80 void __attribute__((constructor)) OnLoad(void) { | |
81 g_at_exit = new base::AtExitManager(); | |
82 g_log = new Logger(); | |
83 Init(scoped_ptr<CommandExecutor>(new CommandExecutorImpl(g_log))); | |
84 } | |
85 void __attribute__((destructor)) OnUnload(void) { | |
86 Shutdown(); | |
87 delete g_log; | |
88 delete g_at_exit; | |
89 } | |
90 #endif | |
OLD | NEW |