Peter Qiu | 5dd242d | 2014-10-14 12:23:21 -0700 | [diff] [blame] | 1 | // Copyright 2014 The Chromium OS 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 "apmanager/manager.h" |
| 6 | |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 7 | #include <base/bind.h> |
Peter Qiu | 5dd242d | 2014-10-14 12:23:21 -0700 | [diff] [blame] | 8 | #include <chromeos/dbus/service_constants.h> |
| 9 | |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 10 | using chromeos::dbus_utils::AsyncEventSequencer; |
| 11 | using chromeos::dbus_utils::ExportedObjectManager; |
| 12 | using chromeos::dbus_utils::DBusMethodResponse; |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 13 | using std::string; |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 14 | |
Peter Qiu | 5dd242d | 2014-10-14 12:23:21 -0700 | [diff] [blame] | 15 | namespace apmanager { |
| 16 | |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 17 | Manager::Manager() |
| 18 | : org::chromium::apmanager::ManagerAdaptor(this), |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 19 | service_identifier_(0), |
| 20 | device_identifier_(0), |
Peter Qiu | 1810c01 | 2015-02-05 14:35:41 -0800 | [diff] [blame] | 21 | device_info_(this) {} |
Peter Qiu | 5dd242d | 2014-10-14 12:23:21 -0700 | [diff] [blame] | 22 | |
Peter Qiu | bf8e36c | 2014-12-03 22:59:45 -0800 | [diff] [blame] | 23 | Manager::~Manager() { |
| 24 | // Terminate all services before cleanup other resources. |
| 25 | for (auto& service : services_) { |
| 26 | service.reset(); |
| 27 | } |
| 28 | } |
Peter Qiu | 5dd242d | 2014-10-14 12:23:21 -0700 | [diff] [blame] | 29 | |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 30 | void Manager::RegisterAsync(ExportedObjectManager* object_manager, |
Peter Qiu | c9ce1f1 | 2014-12-05 11:14:29 -0800 | [diff] [blame] | 31 | const scoped_refptr<dbus::Bus>& bus, |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 32 | AsyncEventSequencer* sequencer) { |
Peter Qiu | f073173 | 2014-11-11 09:46:41 -0800 | [diff] [blame] | 33 | CHECK(!dbus_object_) << "Already registered"; |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 34 | dbus_object_.reset( |
| 35 | new chromeos::dbus_utils::DBusObject( |
| 36 | object_manager, |
Peter Qiu | c9ce1f1 | 2014-12-05 11:14:29 -0800 | [diff] [blame] | 37 | bus, |
Vitaly Buka | a463092 | 2014-12-11 18:46:46 -0800 | [diff] [blame] | 38 | org::chromium::apmanager::ManagerAdaptor::GetObjectPath())); |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 39 | RegisterWithDBusObject(dbus_object_.get()); |
Peter Qiu | f073173 | 2014-11-11 09:46:41 -0800 | [diff] [blame] | 40 | dbus_object_->RegisterAsync( |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 41 | sequencer->GetHandler("Manager.RegisterAsync() failed.", true)); |
Peter Qiu | c9ce1f1 | 2014-12-05 11:14:29 -0800 | [diff] [blame] | 42 | bus_ = bus; |
| 43 | |
Peter Qiu | 1810c01 | 2015-02-05 14:35:41 -0800 | [diff] [blame] | 44 | shill_proxy_.Init(bus); |
Peter Qiu | 943cf3a | 2015-02-24 10:59:17 -0800 | [diff] [blame^] | 45 | firewall_manager_.Init(bus); |
Peter Qiu | 5dd242d | 2014-10-14 12:23:21 -0700 | [diff] [blame] | 46 | } |
| 47 | |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 48 | void Manager::Start() { |
| 49 | device_info_.Start(); |
| 50 | } |
| 51 | |
| 52 | void Manager::Stop() { |
| 53 | device_info_.Stop(); |
| 54 | } |
| 55 | |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 56 | void Manager::CreateService( |
| 57 | scoped_ptr<DBusMethodResponse<dbus::ObjectPath>> response) { |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 58 | LOG(INFO) << "Manager::CreateService"; |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 59 | scoped_refptr<AsyncEventSequencer> sequencer(new AsyncEventSequencer()); |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 60 | scoped_ptr<Service> service(new Service(this, service_identifier_++)); |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 61 | |
Peter Qiu | c9ce1f1 | 2014-12-05 11:14:29 -0800 | [diff] [blame] | 62 | service->RegisterAsync( |
Alex Vakulenko | e7a8bc5 | 2014-12-10 12:52:31 -0800 | [diff] [blame] | 63 | dbus_object_->GetObjectManager().get(), bus_, sequencer.get()); |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 64 | sequencer->OnAllTasksCompletedCall({ |
Peter Qiu | 1ff67a7 | 2014-11-22 07:06:10 -0800 | [diff] [blame] | 65 | base::Bind(&Manager::OnServiceRegistered, |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 66 | base::Unretained(this), |
| 67 | base::Passed(&response), |
| 68 | base::Passed(&service)) |
| 69 | }); |
Peter Qiu | 5dd242d | 2014-10-14 12:23:21 -0700 | [diff] [blame] | 70 | } |
| 71 | |
Peter Qiu | f073173 | 2014-11-11 09:46:41 -0800 | [diff] [blame] | 72 | bool Manager::RemoveService(chromeos::ErrorPtr* error, |
| 73 | const dbus::ObjectPath& in_service) { |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 74 | for (auto it = services_.begin(); it != services_.end(); ++it) { |
| 75 | if ((*it)->dbus_path() == in_service) { |
| 76 | services_.erase(it); |
| 77 | return true; |
| 78 | } |
| 79 | } |
| 80 | |
| 81 | chromeos::Error::AddTo( |
| 82 | error, FROM_HERE, chromeos::errors::dbus::kDomain, kManagerError, |
| 83 | "Service does not exist"); |
Peter Qiu | f073173 | 2014-11-11 09:46:41 -0800 | [diff] [blame] | 84 | return false; |
Peter Qiu | 5dd242d | 2014-10-14 12:23:21 -0700 | [diff] [blame] | 85 | } |
| 86 | |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 87 | scoped_refptr<Device> Manager::GetAvailableDevice() { |
| 88 | for (const auto& device : devices_) { |
Peter Qiu | 8e785b9 | 2014-11-24 10:01:08 -0800 | [diff] [blame] | 89 | // Look for an unused device with AP interface mode support. |
| 90 | if (!device->GetInUsed() && !device->GetPreferredApInterface().empty()) { |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 91 | return device; |
| 92 | } |
| 93 | } |
| 94 | return nullptr; |
| 95 | } |
| 96 | |
| 97 | scoped_refptr<Device> Manager::GetDeviceFromInterfaceName( |
| 98 | const string& interface_name) { |
| 99 | for (const auto& device : devices_) { |
| 100 | if (device->InterfaceExists(interface_name)) { |
| 101 | return device; |
| 102 | } |
| 103 | } |
| 104 | return nullptr; |
| 105 | } |
| 106 | |
| 107 | void Manager::RegisterDevice(scoped_refptr<Device> device) { |
| 108 | LOG(INFO) << "Manager::RegisterDevice: registering device " |
| 109 | << device->GetDeviceName(); |
| 110 | // Register device DBbus interfaces. |
| 111 | scoped_refptr<AsyncEventSequencer> sequencer(new AsyncEventSequencer()); |
| 112 | device->RegisterAsync(dbus_object_->GetObjectManager().get(), |
Peter Qiu | c9ce1f1 | 2014-12-05 11:14:29 -0800 | [diff] [blame] | 113 | bus_, |
Alex Vakulenko | e7a8bc5 | 2014-12-10 12:52:31 -0800 | [diff] [blame] | 114 | sequencer.get(), |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 115 | device_identifier_++); |
| 116 | sequencer->OnAllTasksCompletedCall({ |
Peter Qiu | 1ff67a7 | 2014-11-22 07:06:10 -0800 | [diff] [blame] | 117 | base::Bind(&Manager::OnDeviceRegistered, |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 118 | base::Unretained(this), |
| 119 | device) |
| 120 | }); |
| 121 | } |
| 122 | |
Peter Qiu | 7e0ffcf | 2014-12-02 12:53:27 -0800 | [diff] [blame] | 123 | void Manager::ClaimInterface(const string& interface_name) { |
Peter Qiu | 1810c01 | 2015-02-05 14:35:41 -0800 | [diff] [blame] | 124 | shill_proxy_.ClaimInterface(interface_name); |
Peter Qiu | 7e0ffcf | 2014-12-02 12:53:27 -0800 | [diff] [blame] | 125 | } |
| 126 | |
| 127 | void Manager::ReleaseInterface(const string& interface_name) { |
Peter Qiu | 1810c01 | 2015-02-05 14:35:41 -0800 | [diff] [blame] | 128 | shill_proxy_.ReleaseInterface(interface_name); |
Peter Qiu | 7e0ffcf | 2014-12-02 12:53:27 -0800 | [diff] [blame] | 129 | } |
| 130 | |
Peter Qiu | 943cf3a | 2015-02-24 10:59:17 -0800 | [diff] [blame^] | 131 | void Manager::RequestDHCPPortAccess(const string& interface) { |
| 132 | firewall_manager_.RequestDHCPPortAccess(interface); |
| 133 | } |
| 134 | |
| 135 | void Manager::ReleaseDHCPPortAccess(const string& interface) { |
| 136 | firewall_manager_.ReleaseDHCPPortAccess(interface); |
| 137 | } |
| 138 | |
Peter Qiu | 1ff67a7 | 2014-11-22 07:06:10 -0800 | [diff] [blame] | 139 | void Manager::OnServiceRegistered( |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 140 | scoped_ptr<DBusMethodResponse<dbus::ObjectPath>> response, |
| 141 | scoped_ptr<Service> service, |
| 142 | bool success) { |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 143 | LOG(INFO) << "ServiceRegistered"; |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 144 | // Success should always be true since we've said that failures are fatal. |
| 145 | CHECK(success) << "Init of one or more objects has failed."; |
| 146 | |
| 147 | // Add service to the service list and return the service dbus path for the |
| 148 | // CreateService call. |
| 149 | dbus::ObjectPath service_path = service->dbus_path(); |
| 150 | services_.push_back(std::unique_ptr<Service>(service.release())); |
| 151 | response->Return(service_path); |
| 152 | } |
| 153 | |
Peter Qiu | 1ff67a7 | 2014-11-22 07:06:10 -0800 | [diff] [blame] | 154 | void Manager::OnDeviceRegistered(scoped_refptr<Device> device, bool success) { |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 155 | // Success should always be true since we've said that failures are fatal. |
| 156 | CHECK(success) << "Init of one or more objects has failed."; |
| 157 | |
| 158 | devices_.push_back(device); |
| 159 | // TODO(zqiu): Property update for available devices. |
| 160 | } |
| 161 | |
Peter Qiu | 5dd242d | 2014-10-14 12:23:21 -0700 | [diff] [blame] | 162 | } // namespace apmanager |