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
79namespace chromeos {
80
81namespace {
82
83// Command line switch mapping for --dbus-unstub-clients.
84const 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.
111DBusClientBundle::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
122DBusClientBundle::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
269DBusClientBundle::~DBusClientBundle() {
270}
271
272bool DBusClientBundle::IsUsingStub(DBusClientType client) {
273  return !(unstub_client_mask_ & client);
274}
275
276bool DBusClientBundle::IsUsingAnyRealClient() {
277  // 'Using any real client' is equivalent to 'Unstubbed any client'.
278  return unstub_client_mask_ != 0;
279}
280
281void DBusClientBundle::SetupDefaultEnvironment() {
282  ShillManagerClient::TestInterface* manager =
283      shill_manager_client_->GetTestInterface();
284  if (manager)
285    manager->SetupDefaultEnvironment();
286}
287
288// static
289DBusClientBundle::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