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( |
Peter Qiu | fd02b6f | 2015-02-27 09:55:11 -0800 | [diff] [blame^] | 57 | scoped_ptr<DBusMethodResponse<dbus::ObjectPath>> response, |
| 58 | dbus::Message* message) { |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 59 | LOG(INFO) << "Manager::CreateService"; |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 60 | scoped_refptr<AsyncEventSequencer> sequencer(new AsyncEventSequencer()); |
Peter Qiu | fd02b6f | 2015-02-27 09:55:11 -0800 | [diff] [blame^] | 61 | scoped_ptr<Service> service(new Service(this, service_identifier_)); |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 62 | |
Peter Qiu | c9ce1f1 | 2014-12-05 11:14:29 -0800 | [diff] [blame] | 63 | service->RegisterAsync( |
Alex Vakulenko | e7a8bc5 | 2014-12-10 12:52:31 -0800 | [diff] [blame] | 64 | dbus_object_->GetObjectManager().get(), bus_, sequencer.get()); |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 65 | sequencer->OnAllTasksCompletedCall({ |
Peter Qiu | 1ff67a7 | 2014-11-22 07:06:10 -0800 | [diff] [blame] | 66 | base::Bind(&Manager::OnServiceRegistered, |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 67 | base::Unretained(this), |
| 68 | base::Passed(&response), |
| 69 | base::Passed(&service)) |
| 70 | }); |
Peter Qiu | fd02b6f | 2015-02-27 09:55:11 -0800 | [diff] [blame^] | 71 | |
| 72 | base::Closure on_connection_vanish = base::Bind( |
| 73 | &Manager::OnAPServiceOwnerDisappeared, |
| 74 | base::Unretained(this), |
| 75 | service_identifier_); |
| 76 | service_watchers_[service_identifier_].reset( |
| 77 | new DBusServiceWatcher{bus_, message->GetSender(), on_connection_vanish}); |
| 78 | service_identifier_++; |
Peter Qiu | 5dd242d | 2014-10-14 12:23:21 -0700 | [diff] [blame] | 79 | } |
| 80 | |
Peter Qiu | f073173 | 2014-11-11 09:46:41 -0800 | [diff] [blame] | 81 | bool Manager::RemoveService(chromeos::ErrorPtr* error, |
Peter Qiu | fd02b6f | 2015-02-27 09:55:11 -0800 | [diff] [blame^] | 82 | dbus::Message* message, |
Peter Qiu | f073173 | 2014-11-11 09:46:41 -0800 | [diff] [blame] | 83 | const dbus::ObjectPath& in_service) { |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 84 | for (auto it = services_.begin(); it != services_.end(); ++it) { |
| 85 | if ((*it)->dbus_path() == in_service) { |
Peter Qiu | fd02b6f | 2015-02-27 09:55:11 -0800 | [diff] [blame^] | 86 | // Verify the owner. |
| 87 | auto watcher = service_watchers_.find((*it)->identifier()); |
| 88 | CHECK(watcher != service_watchers_.end()) |
| 89 | << "DBus watcher not created for service: " << (*it)->identifier(); |
| 90 | if (watcher->second->connection_name() != message->GetSender()) { |
| 91 | chromeos::Error::AddToPrintf( |
| 92 | error, FROM_HERE, chromeos::errors::dbus::kDomain, kManagerError, |
| 93 | "Service %d is owned by another local process.", |
| 94 | (*it)->identifier()); |
| 95 | return false; |
| 96 | } |
| 97 | service_watchers_.erase(watcher); |
| 98 | |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 99 | services_.erase(it); |
| 100 | return true; |
| 101 | } |
| 102 | } |
| 103 | |
| 104 | chromeos::Error::AddTo( |
| 105 | error, FROM_HERE, chromeos::errors::dbus::kDomain, kManagerError, |
| 106 | "Service does not exist"); |
Peter Qiu | f073173 | 2014-11-11 09:46:41 -0800 | [diff] [blame] | 107 | return false; |
Peter Qiu | 5dd242d | 2014-10-14 12:23:21 -0700 | [diff] [blame] | 108 | } |
| 109 | |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 110 | scoped_refptr<Device> Manager::GetAvailableDevice() { |
| 111 | for (const auto& device : devices_) { |
Peter Qiu | 8e785b9 | 2014-11-24 10:01:08 -0800 | [diff] [blame] | 112 | // Look for an unused device with AP interface mode support. |
| 113 | if (!device->GetInUsed() && !device->GetPreferredApInterface().empty()) { |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 114 | return device; |
| 115 | } |
| 116 | } |
| 117 | return nullptr; |
| 118 | } |
| 119 | |
| 120 | scoped_refptr<Device> Manager::GetDeviceFromInterfaceName( |
| 121 | const string& interface_name) { |
| 122 | for (const auto& device : devices_) { |
| 123 | if (device->InterfaceExists(interface_name)) { |
| 124 | return device; |
| 125 | } |
| 126 | } |
| 127 | return nullptr; |
| 128 | } |
| 129 | |
| 130 | void Manager::RegisterDevice(scoped_refptr<Device> device) { |
| 131 | LOG(INFO) << "Manager::RegisterDevice: registering device " |
| 132 | << device->GetDeviceName(); |
| 133 | // Register device DBbus interfaces. |
| 134 | scoped_refptr<AsyncEventSequencer> sequencer(new AsyncEventSequencer()); |
| 135 | device->RegisterAsync(dbus_object_->GetObjectManager().get(), |
Peter Qiu | c9ce1f1 | 2014-12-05 11:14:29 -0800 | [diff] [blame] | 136 | bus_, |
Alex Vakulenko | e7a8bc5 | 2014-12-10 12:52:31 -0800 | [diff] [blame] | 137 | sequencer.get(), |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 138 | device_identifier_++); |
| 139 | sequencer->OnAllTasksCompletedCall({ |
Peter Qiu | 1ff67a7 | 2014-11-22 07:06:10 -0800 | [diff] [blame] | 140 | base::Bind(&Manager::OnDeviceRegistered, |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 141 | base::Unretained(this), |
| 142 | device) |
| 143 | }); |
| 144 | } |
| 145 | |
Peter Qiu | 7e0ffcf | 2014-12-02 12:53:27 -0800 | [diff] [blame] | 146 | void Manager::ClaimInterface(const string& interface_name) { |
Peter Qiu | 1810c01 | 2015-02-05 14:35:41 -0800 | [diff] [blame] | 147 | shill_proxy_.ClaimInterface(interface_name); |
Peter Qiu | 7e0ffcf | 2014-12-02 12:53:27 -0800 | [diff] [blame] | 148 | } |
| 149 | |
| 150 | void Manager::ReleaseInterface(const string& interface_name) { |
Peter Qiu | 1810c01 | 2015-02-05 14:35:41 -0800 | [diff] [blame] | 151 | shill_proxy_.ReleaseInterface(interface_name); |
Peter Qiu | 7e0ffcf | 2014-12-02 12:53:27 -0800 | [diff] [blame] | 152 | } |
| 153 | |
Peter Qiu | 943cf3a | 2015-02-24 10:59:17 -0800 | [diff] [blame] | 154 | void Manager::RequestDHCPPortAccess(const string& interface) { |
| 155 | firewall_manager_.RequestDHCPPortAccess(interface); |
| 156 | } |
| 157 | |
| 158 | void Manager::ReleaseDHCPPortAccess(const string& interface) { |
| 159 | firewall_manager_.ReleaseDHCPPortAccess(interface); |
| 160 | } |
| 161 | |
Peter Qiu | 1ff67a7 | 2014-11-22 07:06:10 -0800 | [diff] [blame] | 162 | void Manager::OnServiceRegistered( |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 163 | scoped_ptr<DBusMethodResponse<dbus::ObjectPath>> response, |
| 164 | scoped_ptr<Service> service, |
| 165 | bool success) { |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 166 | LOG(INFO) << "ServiceRegistered"; |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 167 | // Success should always be true since we've said that failures are fatal. |
| 168 | CHECK(success) << "Init of one or more objects has failed."; |
| 169 | |
Peter Qiu | fd02b6f | 2015-02-27 09:55:11 -0800 | [diff] [blame^] | 170 | // Remove this service if the owner doesn't exist anymore. It is theoretically |
| 171 | // possible to have the owner disappear before the AP service complete its |
| 172 | // registration with DBus. |
| 173 | if (service_watchers_.find(service->identifier()) == |
| 174 | service_watchers_.end()) { |
| 175 | LOG(INFO) << "Service " << service->identifier() |
| 176 | << ": owner doesn't exist anymore"; |
| 177 | service.reset(); |
| 178 | return; |
| 179 | } |
| 180 | |
Peter Qiu | 376e404 | 2014-11-13 09:40:28 -0800 | [diff] [blame] | 181 | // Add service to the service list and return the service dbus path for the |
| 182 | // CreateService call. |
| 183 | dbus::ObjectPath service_path = service->dbus_path(); |
| 184 | services_.push_back(std::unique_ptr<Service>(service.release())); |
| 185 | response->Return(service_path); |
| 186 | } |
| 187 | |
Peter Qiu | 1ff67a7 | 2014-11-22 07:06:10 -0800 | [diff] [blame] | 188 | void Manager::OnDeviceRegistered(scoped_refptr<Device> device, bool success) { |
Peter Qiu | fb39ba4 | 2014-11-21 09:09:59 -0800 | [diff] [blame] | 189 | // Success should always be true since we've said that failures are fatal. |
| 190 | CHECK(success) << "Init of one or more objects has failed."; |
| 191 | |
| 192 | devices_.push_back(device); |
| 193 | // TODO(zqiu): Property update for available devices. |
| 194 | } |
| 195 | |
Peter Qiu | fd02b6f | 2015-02-27 09:55:11 -0800 | [diff] [blame^] | 196 | void Manager::OnAPServiceOwnerDisappeared(int service_identifier) { |
| 197 | LOG(INFO) << "Owner for service " << service_identifier << " disappeared"; |
| 198 | // Remove service watcher. |
| 199 | auto watcher = service_watchers_.find(service_identifier); |
| 200 | CHECK(watcher != service_watchers_.end()) |
| 201 | << "Owner disappeared without watcher setup"; |
| 202 | service_watchers_.erase(watcher); |
| 203 | |
| 204 | // Remove the service. |
| 205 | for (auto it = services_.begin(); it != services_.end(); ++it) { |
| 206 | if ((*it)->identifier() == service_identifier) { |
| 207 | services_.erase(it); |
| 208 | return; |
| 209 | } |
| 210 | } |
| 211 | LOG(INFO) << "Owner for service " << service_identifier |
| 212 | << " disappeared before it is registered"; |
| 213 | } |
| 214 | |
Peter Qiu | 5dd242d | 2014-10-14 12:23:21 -0700 | [diff] [blame] | 215 | } // namespace apmanager |