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 "chromeos/dbus/dbus_client_bundle.h"
6
7 #include "base/command_line.h"
8 #include "base/strings/string_split.h"
9 #include "base/strings/string_util.h"
10 #include "chromeos/chromeos_switches.h"
11 #include "chromeos/dbus/bluetooth_adapter_client.h"
12 #include "chromeos/dbus/bluetooth_agent_manager_client.h"
13 #include "chromeos/dbus/bluetooth_device_client.h"
14 #include "chromeos/dbus/bluetooth_gatt_characteristic_client.h"
15 #include "chromeos/dbus/bluetooth_gatt_descriptor_client.h"
16 #include "chromeos/dbus/bluetooth_gatt_manager_client.h"
17 #include "chromeos/dbus/bluetooth_gatt_service_client.h"
18 #include "chromeos/dbus/bluetooth_input_client.h"
19 #include "chromeos/dbus/bluetooth_profile_manager_client.h"
20 #include "chromeos/dbus/cras_audio_client.h"
21 #include "chromeos/dbus/cras_audio_client_stub_impl.h"
22 #include "chromeos/dbus/cros_disks_client.h"
23 #include "chromeos/dbus/cryptohome_client.h"
24 #include "chromeos/dbus/debug_daemon_client.h"
25 #include "chromeos/dbus/easy_unlock_client.h"
26 #include "chromeos/dbus/fake_bluetooth_adapter_client.h"
27 #include "chromeos/dbus/fake_bluetooth_agent_manager_client.h"
28 #include "chromeos/dbus/fake_bluetooth_device_client.h"
29 #include "chromeos/dbus/fake_bluetooth_gatt_characteristic_client.h"
30 #include "chromeos/dbus/fake_bluetooth_gatt_descriptor_client.h"
31 #include "chromeos/dbus/fake_bluetooth_gatt_manager_client.h"
32 #include "chromeos/dbus/fake_bluetooth_gatt_service_client.h"
33 #include "chromeos/dbus/fake_bluetooth_input_client.h"
34 #include "chromeos/dbus/fake_bluetooth_profile_manager_client.h"
35 #include "chromeos/dbus/fake_cryptohome_client.h"
36 #include "chromeos/dbus/fake_debug_daemon_client.h"
37 #include "chromeos/dbus/fake_easy_unlock_client.h"
38 #include "chromeos/dbus/fake_gsm_sms_client.h"
39 #include "chromeos/dbus/fake_image_burner_client.h"
40 #include "chromeos/dbus/fake_introspectable_client.h"
41 #include "chromeos/dbus/fake_lorgnette_manager_client.h"
42 #include "chromeos/dbus/fake_modem_messaging_client.h"
43 #include "chromeos/dbus/fake_nfc_adapter_client.h"
44 #include "chromeos/dbus/fake_nfc_device_client.h"
45 #include "chromeos/dbus/fake_nfc_manager_client.h"
46 #include "chromeos/dbus/fake_nfc_record_client.h"
47 #include "chromeos/dbus/fake_nfc_tag_client.h"
48 #include "chromeos/dbus/fake_permission_broker_client.h"
49 #include "chromeos/dbus/fake_shill_device_client.h"
50 #include "chromeos/dbus/fake_shill_ipconfig_client.h"
51 #include "chromeos/dbus/fake_shill_manager_client.h"
52 #include "chromeos/dbus/fake_shill_profile_client.h"
53 #include "chromeos/dbus/fake_shill_service_client.h"
54 #include "chromeos/dbus/fake_sms_client.h"
55 #include "chromeos/dbus/fake_system_clock_client.h"
56 #include "chromeos/dbus/gsm_sms_client.h"
57 #include "chromeos/dbus/image_burner_client.h"
58 #include "chromeos/dbus/introspectable_client.h"
59 #include "chromeos/dbus/lorgnette_manager_client.h"
60 #include "chromeos/dbus/modem_messaging_client.h"
61 #include "chromeos/dbus/nfc_adapter_client.h"
62 #include "chromeos/dbus/nfc_device_client.h"
63 #include "chromeos/dbus/nfc_manager_client.h"
64 #include "chromeos/dbus/nfc_record_client.h"
65 #include "chromeos/dbus/nfc_tag_client.h"
66 #include "chromeos/dbus/permission_broker_client.h"
67 #include "chromeos/dbus/power_manager_client.h"
68 #include "chromeos/dbus/power_policy_controller.h"
69 #include "chromeos/dbus/session_manager_client.h"
70 #include "chromeos/dbus/shill_device_client.h"
71 #include "chromeos/dbus/shill_ipconfig_client.h"
72 #include "chromeos/dbus/shill_manager_client.h"
73 #include "chromeos/dbus/shill_profile_client.h"
74 #include "chromeos/dbus/shill_service_client.h"
75 #include "chromeos/dbus/sms_client.h"
76 #include "chromeos/dbus/system_clock_client.h"
77 #include "chromeos/dbus/update_engine_client.h"
78
79 namespace chromeos {
80
81 namespace {
82
83 // Command line switch mapping for --dbus-unstub-clients.
84 const struct {
85 const char* param_name;
86 DBusClientBundle::DBusClientType client_type;
87 } client_type_map[] = {
88 { "bluetooth", DBusClientBundle::BLUETOOTH },
89 { "cras", DBusClientBundle::CRAS },
90 { "cros_disks", DBusClientBundle::CROS_DISKS },
91 { "cryptohome", DBusClientBundle::CRYPTOHOME },
92 { "debug_daemon", DBusClientBundle::DEBUG_DAEMON },
93 { "easy_unlock", DBusClientBundle::EASY_UNLOCK },
94 { "lorgnette_manager", DBusClientBundle::LORGNETTE_MANAGER },
95 { "shill", DBusClientBundle::SHILL },
96 { "gsm_sms", DBusClientBundle::GSM_SMS },
97 { "image_burner", DBusClientBundle::IMAGE_BURNER },
98 { "introspectable", DBusClientBundle::INTROSPECTABLE },
99 { "modem_messaging", DBusClientBundle::MODEM_MESSAGING },
100 { "nfc", DBusClientBundle::NFC },
101 { "permission_broker", DBusClientBundle::PERMISSION_BROKER },
102 { "power_manager", DBusClientBundle::POWER_MANAGER },
103 { "session_manager", DBusClientBundle::SESSION_MANAGER },
104 { "sms", DBusClientBundle::SMS },
105 { "system_clock", DBusClientBundle::SYSTEM_CLOCK },
106 { "update_engine", DBusClientBundle::UPDATE_ENGINE },
107 };
108
109 // Parses single command line param value for dbus subsystem. If successful,
110 // returns its enum representation. Otherwise returns NO_CLIENT.
GetDBusClientType(const std::string & client_type_name)111 DBusClientBundle::DBusClientType GetDBusClientType(
112 const std::string& client_type_name) {
113 for (size_t i = 0; i < arraysize(client_type_map); i++) {
114 if (LowerCaseEqualsASCII(client_type_name, client_type_map[i].param_name))
115 return client_type_map[i].client_type;
116 }
117 return DBusClientBundle::NO_CLIENT;
118 }
119
120 } // namespace
121
DBusClientBundle(DBusClientTypeMask unstub_client_mask)122 DBusClientBundle::DBusClientBundle(DBusClientTypeMask unstub_client_mask)
123 : unstub_client_mask_(unstub_client_mask) {
124 if (!IsUsingStub(BLUETOOTH)) {
125 bluetooth_adapter_client_.reset(BluetoothAdapterClient::Create());
126 bluetooth_agent_manager_client_.reset(
127 BluetoothAgentManagerClient::Create());
128 bluetooth_device_client_.reset(BluetoothDeviceClient::Create());
129 bluetooth_input_client_.reset(BluetoothInputClient::Create());
130 bluetooth_profile_manager_client_.reset(
131 BluetoothProfileManagerClient::Create());
132 bluetooth_gatt_characteristic_client_.reset(
133 BluetoothGattCharacteristicClient::Create());
134 bluetooth_gatt_descriptor_client_.reset(
135 BluetoothGattDescriptorClient::Create());
136 bluetooth_gatt_manager_client_.reset(
137 BluetoothGattManagerClient::Create());
138 bluetooth_gatt_service_client_.reset(
139 BluetoothGattServiceClient::Create());
140 } else {
141 bluetooth_adapter_client_.reset(new FakeBluetoothAdapterClient);
142 bluetooth_agent_manager_client_.reset(new FakeBluetoothAgentManagerClient);
143 bluetooth_device_client_.reset(new FakeBluetoothDeviceClient);
144 bluetooth_input_client_.reset(new FakeBluetoothInputClient);
145 bluetooth_profile_manager_client_.reset(
146 new FakeBluetoothProfileManagerClient);
147 bluetooth_gatt_characteristic_client_.reset(
148 new FakeBluetoothGattCharacteristicClient);
149 bluetooth_gatt_descriptor_client_.reset(
150 new FakeBluetoothGattDescriptorClient);
151 bluetooth_gatt_manager_client_.reset(new FakeBluetoothGattManagerClient);
152 bluetooth_gatt_service_client_.reset(new FakeBluetoothGattServiceClient);
153 }
154
155 if (!IsUsingStub(CRAS))
156 cras_audio_client_.reset(CrasAudioClient::Create());
157 else
158 cras_audio_client_.reset(new CrasAudioClientStubImpl);
159
160 cros_disks_client_.reset(CrosDisksClient::Create(
161 IsUsingStub(CROS_DISKS) ? STUB_DBUS_CLIENT_IMPLEMENTATION
162 : REAL_DBUS_CLIENT_IMPLEMENTATION));
163
164 if (!IsUsingStub(CRYPTOHOME))
165 cryptohome_client_.reset(CryptohomeClient::Create());
166 else
167 cryptohome_client_.reset(new FakeCryptohomeClient);
168
169 if (!IsUsingStub(DEBUG_DAEMON))
170 debug_daemon_client_.reset(DebugDaemonClient::Create());
171 else
172 debug_daemon_client_.reset(new FakeDebugDaemonClient);
173
174 if (!IsUsingStub(EASY_UNLOCK))
175 easy_unlock_client_.reset(EasyUnlockClient::Create());
176 else
177 easy_unlock_client_.reset(new FakeEasyUnlockClient);
178
179 if (!IsUsingStub(LORGNETTE_MANAGER))
180 lorgnette_manager_client_.reset(LorgnetteManagerClient::Create());
181 else
182 lorgnette_manager_client_.reset(new FakeLorgnetteManagerClient);
183
184 if (!IsUsingStub(SHILL)) {
185 shill_manager_client_.reset(ShillManagerClient::Create());
186 shill_device_client_.reset(ShillDeviceClient::Create());
187 shill_ipconfig_client_.reset(ShillIPConfigClient::Create());
188 shill_service_client_.reset(ShillServiceClient::Create());
189 shill_profile_client_.reset(ShillProfileClient::Create());
190 } else {
191 shill_manager_client_.reset(new FakeShillManagerClient);
192 shill_device_client_.reset(new FakeShillDeviceClient);
193 shill_ipconfig_client_.reset(new FakeShillIPConfigClient);
194 shill_service_client_.reset(new FakeShillServiceClient);
195 shill_profile_client_.reset(new FakeShillProfileClient);
196 }
197
198 if (!IsUsingStub(GSM_SMS)) {
199 gsm_sms_client_.reset(GsmSMSClient::Create());
200 } else {
201 FakeGsmSMSClient* gsm_sms_client = new FakeGsmSMSClient();
202 gsm_sms_client->set_sms_test_message_switch_present(
203 CommandLine::ForCurrentProcess()->HasSwitch(
204 chromeos::switches::kSmsTestMessages));
205 gsm_sms_client_.reset(gsm_sms_client);
206 }
207
208 if (!IsUsingStub(IMAGE_BURNER))
209 image_burner_client_.reset(ImageBurnerClient::Create());
210 else
211 image_burner_client_.reset(new FakeImageBurnerClient);
212
213 if (!IsUsingStub(INTROSPECTABLE))
214 introspectable_client_.reset(IntrospectableClient::Create());
215 else
216 introspectable_client_.reset(new FakeIntrospectableClient);
217
218 if (!IsUsingStub(MODEM_MESSAGING))
219 modem_messaging_client_.reset(ModemMessagingClient::Create());
220 else
221 modem_messaging_client_.reset(new FakeModemMessagingClient);
222
223 // Create the NFC clients in the correct order based on their dependencies.
224 if (!IsUsingStub(NFC)) {
225 nfc_manager_client_.reset(NfcManagerClient::Create());
226 nfc_adapter_client_.reset(
227 NfcAdapterClient::Create(nfc_manager_client_.get()));
228 nfc_device_client_.reset(
229 NfcDeviceClient::Create(nfc_adapter_client_.get()));
230 nfc_tag_client_.reset(NfcTagClient::Create(nfc_adapter_client_.get()));
231 nfc_record_client_.reset(NfcRecordClient::Create(nfc_device_client_.get(),
232 nfc_tag_client_.get()));
233 } else {
234 nfc_manager_client_.reset(new FakeNfcManagerClient);
235 nfc_adapter_client_.reset(new FakeNfcAdapterClient);
236 nfc_device_client_.reset(new FakeNfcDeviceClient);
237 nfc_tag_client_.reset(new FakeNfcTagClient);
238 nfc_record_client_.reset(new FakeNfcRecordClient);
239 }
240
241 if (!IsUsingStub(PERMISSION_BROKER))
242 permission_broker_client_.reset(PermissionBrokerClient::Create());
243 else
244 permission_broker_client_.reset(new FakePermissionBrokerClient);
245
246 power_manager_client_.reset(PowerManagerClient::Create(
247 IsUsingStub(POWER_MANAGER) ? STUB_DBUS_CLIENT_IMPLEMENTATION
248 : REAL_DBUS_CLIENT_IMPLEMENTATION));
249
250 session_manager_client_.reset(SessionManagerClient::Create(
251 IsUsingStub(SESSION_MANAGER) ? STUB_DBUS_CLIENT_IMPLEMENTATION
252 : REAL_DBUS_CLIENT_IMPLEMENTATION));
253
254 if (!IsUsingStub(SMS))
255 sms_client_.reset(SMSClient::Create());
256 else
257 sms_client_.reset(new FakeSMSClient);
258
259 if (!IsUsingStub(SYSTEM_CLOCK))
260 system_clock_client_.reset(SystemClockClient::Create());
261 else
262 system_clock_client_.reset(new FakeSystemClockClient);
263
264 update_engine_client_.reset(UpdateEngineClient::Create(
265 IsUsingStub(UPDATE_ENGINE) ? STUB_DBUS_CLIENT_IMPLEMENTATION
266 : REAL_DBUS_CLIENT_IMPLEMENTATION));
267 }
268
~DBusClientBundle()269 DBusClientBundle::~DBusClientBundle() {
270 }
271
IsUsingStub(DBusClientType client)272 bool DBusClientBundle::IsUsingStub(DBusClientType client) {
273 return !(unstub_client_mask_ & client);
274 }
275
IsUsingAnyRealClient()276 bool DBusClientBundle::IsUsingAnyRealClient() {
277 // 'Using any real client' is equivalent to 'Unstubbed any client'.
278 return unstub_client_mask_ != 0;
279 }
280
SetupDefaultEnvironment()281 void DBusClientBundle::SetupDefaultEnvironment() {
282 ShillManagerClient::TestInterface* manager =
283 shill_manager_client_->GetTestInterface();
284 if (manager)
285 manager->SetupDefaultEnvironment();
286 }
287
288 // static
ParseUnstubList(const std::string & unstub_list)289 DBusClientBundle::DBusClientTypeMask DBusClientBundle::ParseUnstubList(
290 const std::string& unstub_list) {
291 DBusClientTypeMask unstub_mask = 0;
292 std::vector<std::string> unstub_components;
293 base::SplitString(unstub_list, ',', &unstub_components);
294 for (std::vector<std::string>::const_iterator iter =
295 unstub_components.begin();
296 iter != unstub_components.end(); ++iter) {
297 DBusClientBundle::DBusClientType client = GetDBusClientType(*iter);
298 if (client != NO_CLIENT) {
299 LOG(WARNING) << "Unstubbing dbus client for " << *iter;
300 unstub_mask |= client;
301 } else {
302 LOG(ERROR) << "Unknown dbus client: " << *iter;
303 }
304 }
305
306 return unstub_mask;
307 }
308
309 } // namespace chromeos
310