chromeos/dbus/fake_shill_device_client.cc

   1 // Copyright 2013 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/fake_shill_device_client.h"
   6
   7 #include "base/bind.h"
   8 #include "base/message_loop/message_loop.h"
   9 #include "base/stl_util.h"
  10 #include "base/values.h"
  11 #include "chromeos/dbus/dbus_thread_manager.h"
  12 #include "chromeos/dbus/shill_manager_client.h"
  13 #include "chromeos/dbus/shill_property_changed_observer.h"
  14 #include "dbus/bus.h"
  15 #include "dbus/message.h"
  16 #include "dbus/object_path.h"
  17 #include "dbus/object_proxy.h"
  18 #include "dbus/values_util.h"
  19 #include "third_party/cros_system_api/dbus/service_constants.h"
  20
  21 namespace chromeos {
  22
  23 namespace {
  24
  25 std::string kSimPin = "1111";
  26
  27 void ErrorFunction(const std::string& device_path,
  28                    const std::string& error_name,
  29                    const std::string& error_message) {
  30   LOG(ERROR) << "Shill Error for: " << device_path
  31              << ": " << error_name << " : " << error_message;
  32 }
  33
  34 void PostDeviceNotFoundError(
  35     const ShillDeviceClient::ErrorCallback& error_callback) {
  36   std::string error_message("Failed");
  37   base::MessageLoop::current()->PostTask(
  38       FROM_HERE,
  39       base::Bind(error_callback, shill::kErrorResultNotFound, error_message));
  40 }
  41
  42 }  // namespace
  43
  44 FakeShillDeviceClient::FakeShillDeviceClient()
  45     : tdls_busy_count_(0),
  46       weak_ptr_factory_(this) {
  47 }
  48
  49 FakeShillDeviceClient::~FakeShillDeviceClient() {
  50   STLDeleteContainerPairSecondPointers(
  51       observer_list_.begin(), observer_list_.end());
  52 }
  53
  54 // ShillDeviceClient overrides.
  55
  56 void FakeShillDeviceClient::Init(dbus::Bus* bus) {}
  57
  58 void FakeShillDeviceClient::AddPropertyChangedObserver(
  59     const dbus::ObjectPath& device_path,
  60     ShillPropertyChangedObserver* observer) {
  61   GetObserverList(device_path).AddObserver(observer);
  62 }
  63
  64 void FakeShillDeviceClient::RemovePropertyChangedObserver(
  65     const dbus::ObjectPath& device_path,
  66     ShillPropertyChangedObserver* observer) {
  67   GetObserverList(device_path).RemoveObserver(observer);
  68 }
  69
  70 void FakeShillDeviceClient::GetProperties(
  71     const dbus::ObjectPath& device_path,
  72     const DictionaryValueCallback& callback) {
  73   base::MessageLoop::current()->PostTask(
  74       FROM_HERE,
  75       base::Bind(&FakeShillDeviceClient::PassStubDeviceProperties,
  76                  weak_ptr_factory_.GetWeakPtr(),
  77                  device_path, callback));
  78 }
  79
  80 void FakeShillDeviceClient::ProposeScan(
  81     const dbus::ObjectPath& device_path,
  82     const VoidDBusMethodCallback& callback) {
  83   PostVoidCallback(callback, DBUS_METHOD_CALL_SUCCESS);
  84 }
  85
  86 void FakeShillDeviceClient::SetProperty(const dbus::ObjectPath& device_path,
  87                                         const std::string& name,
  88                                         const base::Value& value,
  89                                         const base::Closure& callback,
  90                                         const ErrorCallback& error_callback) {
  91   base::DictionaryValue* device_properties = NULL;
  92   if (!stub_devices_.GetDictionaryWithoutPathExpansion(device_path.value(),
  93                                                        &device_properties)) {
  94     PostDeviceNotFoundError(error_callback);
  95     return;
  96   }
  97   device_properties->SetWithoutPathExpansion(name, value.DeepCopy());
  98   base::MessageLoop::current()->PostTask(
  99       FROM_HERE,
 100       base::Bind(&FakeShillDeviceClient::NotifyObserversPropertyChanged,
 101                  weak_ptr_factory_.GetWeakPtr(), device_path, name));
 102   base::MessageLoop::current()->PostTask(FROM_HERE, callback);
 103 }
 104
 105 void FakeShillDeviceClient::ClearProperty(
 106     const dbus::ObjectPath& device_path,
 107     const std::string& name,
 108     const VoidDBusMethodCallback& callback) {
 109   base::DictionaryValue* device_properties = NULL;
 110   if (!stub_devices_.GetDictionaryWithoutPathExpansion(device_path.value(),
 111                                                        &device_properties)) {
 112     PostVoidCallback(callback, DBUS_METHOD_CALL_FAILURE);
 113     return;
 114   }
 115   device_properties->RemoveWithoutPathExpansion(name, NULL);
 116   PostVoidCallback(callback, DBUS_METHOD_CALL_SUCCESS);
 117 }
 118
 119 void FakeShillDeviceClient::AddIPConfig(
 120     const dbus::ObjectPath& device_path,
 121     const std::string& method,
 122     const ObjectPathDBusMethodCallback& callback) {
 123   base::MessageLoop::current()->PostTask(FROM_HERE,
 124                                          base::Bind(callback,
 125                                                     DBUS_METHOD_CALL_SUCCESS,
 126                                                     dbus::ObjectPath()));
 127 }
 128
 129 void FakeShillDeviceClient::RequirePin(const dbus::ObjectPath& device_path,
 130                                        const std::string& pin,
 131                                        bool require,
 132                                        const base::Closure& callback,
 133                                        const ErrorCallback& error_callback) {
 134   VLOG(1) << "RequirePin: " << device_path.value();
 135   if (pin != kSimPin) {
 136     base::MessageLoop::current()->PostTask(
 137         FROM_HERE,
 138         base::Bind(error_callback, shill::kErrorResultIncorrectPin, ""));
 139     return;
 140   }
 141   base::DictionaryValue* device_properties = NULL;
 142   if (!stub_devices_.GetDictionaryWithoutPathExpansion(device_path.value(),
 143                                                        &device_properties)) {
 144     PostDeviceNotFoundError(error_callback);
 145     return;
 146   }
 147   base::DictionaryValue* simlock_dict = NULL;
 148   if (!device_properties->GetDictionaryWithoutPathExpansion(
 149           shill::kSIMLockStatusProperty, &simlock_dict)) {
 150     simlock_dict = new base::DictionaryValue;
 151     device_properties->SetWithoutPathExpansion(
 152         shill::kSIMLockStatusProperty, simlock_dict);
 153   }
 154   simlock_dict->Clear();
 155   simlock_dict->SetBoolean(shill::kSIMLockEnabledProperty, require);
 156   // TODO(stevenjb): Investigate why non-empty value breaks UI.
 157   std::string lock_type = "";  // shill::kSIMLockPin
 158   simlock_dict->SetString(shill::kSIMLockTypeProperty, lock_type);
 159   simlock_dict->SetInteger(shill::kSIMLockRetriesLeftProperty, 5);
 160
 161   NotifyObserversPropertyChanged(device_path, shill::kSIMLockStatusProperty);
 162   base::MessageLoop::current()->PostTask(FROM_HERE, callback);
 163 }
 164
 165 void FakeShillDeviceClient::EnterPin(const dbus::ObjectPath& device_path,
 166                                      const std::string& pin,
 167                                      const base::Closure& callback,
 168                                      const ErrorCallback& error_callback) {
 169   VLOG(1) << "EnterPin: " << device_path.value();
 170   if (pin != kSimPin) {
 171     base::MessageLoop::current()->PostTask(
 172         FROM_HERE,
 173         base::Bind(error_callback, shill::kErrorResultIncorrectPin, ""));
 174     return;
 175   }
 176   if (!stub_devices_.HasKey(device_path.value())) {
 177     PostDeviceNotFoundError(error_callback);
 178     return;
 179   }
 180   base::MessageLoop::current()->PostTask(FROM_HERE, callback);
 181 }
 182
 183 void FakeShillDeviceClient::UnblockPin(const dbus::ObjectPath& device_path,
 184                                        const std::string& puk,
 185                                        const std::string& pin,
 186                                        const base::Closure& callback,
 187                                        const ErrorCallback& error_callback) {
 188   VLOG(1) << "UnblockPin: " << device_path.value();
 189   if (!stub_devices_.HasKey(device_path.value())) {
 190     PostDeviceNotFoundError(error_callback);
 191     return;
 192   }
 193   base::MessageLoop::current()->PostTask(FROM_HERE, callback);
 194 }
 195
 196 void FakeShillDeviceClient::ChangePin(const dbus::ObjectPath& device_path,
 197                                       const std::string& old_pin,
 198                                       const std::string& new_pin,
 199                                       const base::Closure& callback,
 200                                       const ErrorCallback& error_callback) {
 201   VLOG(1) << "ChangePin: " << device_path.value();
 202   if (!stub_devices_.HasKey(device_path.value())) {
 203     PostDeviceNotFoundError(error_callback);
 204     return;
 205   }
 206   base::MessageLoop::current()->PostTask(FROM_HERE, callback);
 207 }
 208
 209 void FakeShillDeviceClient::Register(const dbus::ObjectPath& device_path,
 210                                      const std::string& network_id,
 211                                      const base::Closure& callback,
 212                                      const ErrorCallback& error_callback) {
 213   if (!stub_devices_.HasKey(device_path.value())) {
 214     PostDeviceNotFoundError(error_callback);
 215     return;
 216   }
 217   base::MessageLoop::current()->PostTask(FROM_HERE, callback);
 218 }
 219
 220 void FakeShillDeviceClient::SetCarrier(const dbus::ObjectPath& device_path,
 221                                        const std::string& carrier,
 222                                        const base::Closure& callback,
 223                                        const ErrorCallback& error_callback) {
 224   if (!stub_devices_.HasKey(device_path.value())) {
 225     PostDeviceNotFoundError(error_callback);
 226     return;
 227   }
 228   base::MessageLoop::current()->PostTask(FROM_HERE, callback);
 229 }
 230
 231 void FakeShillDeviceClient::Reset(const dbus::ObjectPath& device_path,
 232                                   const base::Closure& callback,
 233                                   const ErrorCallback& error_callback) {
 234   if (!stub_devices_.HasKey(device_path.value())) {
 235     PostDeviceNotFoundError(error_callback);
 236     return;
 237   }
 238   base::MessageLoop::current()->PostTask(FROM_HERE, callback);
 239 }
 240
 241 void FakeShillDeviceClient::PerformTDLSOperation(
 242     const dbus::ObjectPath& device_path,
 243     const std::string& operation,
 244     const std::string& peer,
 245     const StringCallback& callback,
 246     const ErrorCallback& error_callback) {
 247   if (!stub_devices_.HasKey(device_path.value())) {
 248     PostDeviceNotFoundError(error_callback);
 249     return;
 250   }
 251   if (tdls_busy_count_) {
 252     --tdls_busy_count_;
 253     std::string error_message("In-Progress");
 254     base::MessageLoop::current()->PostTask(
 255         FROM_HERE,
 256         base::Bind(error_callback,
 257                    shill::kErrorResultInProgress, error_message));
 258     return;
 259   }
 260   std::string result;
 261   if (operation == shill::kTDLSStatusOperation)
 262     result = shill::kTDLSConnectedState;
 263   base::MessageLoop::current()->PostTask(FROM_HERE,
 264                                          base::Bind(callback, result));
 265 }
 266
 267 ShillDeviceClient::TestInterface* FakeShillDeviceClient::GetTestInterface() {
 268   return this;
 269 }
 270
 271 // ShillDeviceClient::TestInterface overrides.
 272
 273 void FakeShillDeviceClient::AddDevice(const std::string& device_path,
 274                                       const std::string& type,
 275                                       const std::string& name) {
 276   DBusThreadManager::Get()->GetShillManagerClient()->GetTestInterface()->
 277       AddDevice(device_path);
 278
 279   base::DictionaryValue* properties = GetDeviceProperties(device_path);
 280   properties->SetStringWithoutPathExpansion(shill::kTypeProperty, type);
 281   properties->SetStringWithoutPathExpansion(shill::kNameProperty, name);
 282   properties->SetStringWithoutPathExpansion(shill::kDBusObjectProperty,
 283                                             device_path);
 284   properties->SetStringWithoutPathExpansion(
 285       shill::kDBusServiceProperty, modemmanager::kModemManager1ServiceName);
 286   if (type == shill::kTypeCellular) {
 287     properties->SetBooleanWithoutPathExpansion(
 288         shill::kCellularAllowRoamingProperty, false);
 289   }
 290 }
 291
 292 void FakeShillDeviceClient::RemoveDevice(const std::string& device_path) {
 293   DBusThreadManager::Get()->GetShillManagerClient()->GetTestInterface()->
 294       RemoveDevice(device_path);
 295
 296   stub_devices_.RemoveWithoutPathExpansion(device_path, NULL);
 297 }
 298
 299 void FakeShillDeviceClient::ClearDevices() {
 300   DBusThreadManager::Get()->GetShillManagerClient()->GetTestInterface()->
 301       ClearDevices();
 302
 303   stub_devices_.Clear();
 304 }
 305
 306 void FakeShillDeviceClient::SetDeviceProperty(const std::string& device_path,
 307                                               const std::string& name,
 308                                               const base::Value& value) {
 309   VLOG(1) << "SetDeviceProperty: " << device_path
 310           << ": " << name << " = " << value;
 311   SetProperty(dbus::ObjectPath(device_path), name, value,
 312               base::Bind(&base::DoNothing),
 313               base::Bind(&ErrorFunction, device_path));
 314 }
 315
 316 std::string FakeShillDeviceClient::GetDevicePathForType(
 317     const std::string& type) {
 318   for (base::DictionaryValue::Iterator iter(stub_devices_);
 319        !iter.IsAtEnd(); iter.Advance()) {
 320     const base::DictionaryValue* properties = NULL;
 321     if (!iter.value().GetAsDictionary(&properties))
 322       continue;
 323     std::string prop_type;
 324     if (!properties->GetStringWithoutPathExpansion(
 325             shill::kTypeProperty, &prop_type) ||
 326         prop_type != type)
 327       continue;
 328     return iter.key();
 329   }
 330   return std::string();
 331 }
 332
 333 void FakeShillDeviceClient::PassStubDeviceProperties(
 334     const dbus::ObjectPath& device_path,
 335     const DictionaryValueCallback& callback) const {
 336   const base::DictionaryValue* device_properties = NULL;
 337   if (!stub_devices_.GetDictionaryWithoutPathExpansion(
 338       device_path.value(), &device_properties)) {
 339     base::DictionaryValue empty_dictionary;
 340     callback.Run(DBUS_METHOD_CALL_FAILURE, empty_dictionary);
 341     return;
 342   }
 343   callback.Run(DBUS_METHOD_CALL_SUCCESS, *device_properties);
 344 }
 345
 346 // Posts a task to run a void callback with status code |status|.
 347 void FakeShillDeviceClient::PostVoidCallback(
 348     const VoidDBusMethodCallback& callback,
 349     DBusMethodCallStatus status) {
 350   base::MessageLoop::current()->PostTask(FROM_HERE,
 351                                          base::Bind(callback, status));
 352 }
 353
 354 void FakeShillDeviceClient::NotifyObserversPropertyChanged(
 355     const dbus::ObjectPath& device_path,
 356     const std::string& property) {
 357   base::DictionaryValue* dict = NULL;
 358   std::string path = device_path.value();
 359   if (!stub_devices_.GetDictionaryWithoutPathExpansion(path, &dict)) {
 360     LOG(ERROR) << "Notify for unknown service: " << path;
 361     return;
 362   }
 363   base::Value* value = NULL;
 364   if (!dict->GetWithoutPathExpansion(property, &value)) {
 365     LOG(ERROR) << "Notify for unknown property: "
 366         << path << " : " << property;
 367     return;
 368   }
 369   FOR_EACH_OBSERVER(ShillPropertyChangedObserver,
 370                     GetObserverList(device_path),
 371                     OnPropertyChanged(property, *value));
 372 }
 373
 374 base::DictionaryValue* FakeShillDeviceClient::GetDeviceProperties(
 375     const std::string& device_path) {
 376   base::DictionaryValue* properties = NULL;
 377   if (!stub_devices_.GetDictionaryWithoutPathExpansion(
 378       device_path, &properties)) {
 379     properties = new base::DictionaryValue;
 380     stub_devices_.SetWithoutPathExpansion(device_path, properties);
 381   }
 382   return properties;
 383 }
 384
 385 FakeShillDeviceClient::PropertyObserverList&
 386 FakeShillDeviceClient::GetObserverList(const dbus::ObjectPath& device_path) {
 387   std::map<dbus::ObjectPath, PropertyObserverList*>::iterator iter =
 388       observer_list_.find(device_path);
 389   if (iter != observer_list_.end())
 390     return *(iter->second);
 391   PropertyObserverList* observer_list = new PropertyObserverList();
 392   observer_list_[device_path] = observer_list;
 393   return *observer_list;
 394 }
 395
 396 }  // namespace chromeos