Chris Masone | 52cd19b | 2011-06-29 17:23:04 -0700 | [diff] [blame] | 1 | // Copyright (c) 2011 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 "shill/profile.h" |
| 6 | |
Chris Masone | 6515aab | 2011-10-12 16:19:09 -0700 | [diff] [blame] | 7 | #include <set> |
Chris Masone | 52cd19b | 2011-06-29 17:23:04 -0700 | [diff] [blame] | 8 | #include <string> |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 9 | #include <vector> |
Chris Masone | 52cd19b | 2011-06-29 17:23:04 -0700 | [diff] [blame] | 10 | |
Chris Masone | b9c0059 | 2011-10-06 13:10:39 -0700 | [diff] [blame] | 11 | #include <base/file_path.h> |
Paul Stewart | e73d05c | 2012-03-29 16:26:05 -0700 | [diff] [blame] | 12 | #include <base/file_util.h> |
Chris Masone | 52cd19b | 2011-06-29 17:23:04 -0700 | [diff] [blame] | 13 | #include <base/logging.h> |
Eric Shienbrood | 3e20a23 | 2012-02-16 11:35:56 -0500 | [diff] [blame] | 14 | #include <base/stl_util.h> |
Darin Petkov | a476682 | 2011-07-07 10:42:22 -0700 | [diff] [blame] | 15 | #include <base/string_util.h> |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 16 | #include <base/stringprintf.h> |
Chris Masone | 88cbd5f | 2011-07-03 14:30:04 -0700 | [diff] [blame] | 17 | #include <chromeos/dbus/service_constants.h> |
Chris Masone | 52cd19b | 2011-06-29 17:23:04 -0700 | [diff] [blame] | 18 | |
| 19 | #include "shill/adaptor_interfaces.h" |
| 20 | #include "shill/control_interface.h" |
Chris Masone | b9c0059 | 2011-10-06 13:10:39 -0700 | [diff] [blame] | 21 | #include "shill/key_file_store.h" |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 22 | #include "shill/manager.h" |
Chris Masone | 52cd19b | 2011-06-29 17:23:04 -0700 | [diff] [blame] | 23 | #include "shill/property_accessor.h" |
Chris Masone | 7aa5f90 | 2011-07-11 11:13:35 -0700 | [diff] [blame] | 24 | #include "shill/service.h" |
Chris Masone | 9d77993 | 2011-08-25 16:33:41 -0700 | [diff] [blame] | 25 | #include "shill/store_interface.h" |
Chris Masone | 52cd19b | 2011-06-29 17:23:04 -0700 | [diff] [blame] | 26 | |
Chris Masone | 6515aab | 2011-10-12 16:19:09 -0700 | [diff] [blame] | 27 | using std::set; |
Chris Masone | 52cd19b | 2011-06-29 17:23:04 -0700 | [diff] [blame] | 28 | using std::string; |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 29 | using std::vector; |
Chris Masone | 52cd19b | 2011-06-29 17:23:04 -0700 | [diff] [blame] | 30 | |
| 31 | namespace shill { |
Darin Petkov | a476682 | 2011-07-07 10:42:22 -0700 | [diff] [blame] | 32 | |
Darin Petkov | a476682 | 2011-07-07 10:42:22 -0700 | [diff] [blame] | 33 | Profile::Profile(ControlInterface *control_interface, |
Chris Masone | 7df0c67 | 2011-07-15 10:24:54 -0700 | [diff] [blame] | 34 | Manager *manager, |
| 35 | const Identifier &name, |
Chris Masone | 2ae797d | 2011-08-23 20:41:00 -0700 | [diff] [blame] | 36 | const string &user_storage_format, |
Chris Masone | 7df0c67 | 2011-07-15 10:24:54 -0700 | [diff] [blame] | 37 | bool connect_to_rpc) |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 38 | : manager_(manager), |
Chris Masone | 7df0c67 | 2011-07-15 10:24:54 -0700 | [diff] [blame] | 39 | name_(name), |
Chris Masone | 2ae797d | 2011-08-23 20:41:00 -0700 | [diff] [blame] | 40 | storage_format_(user_storage_format) { |
Chris Masone | 7df0c67 | 2011-07-15 10:24:54 -0700 | [diff] [blame] | 41 | if (connect_to_rpc) |
| 42 | adaptor_.reset(control_interface->CreateProfileAdaptor(this)); |
| 43 | |
Chris Masone | 88cbd5f | 2011-07-03 14:30:04 -0700 | [diff] [blame] | 44 | // flimflam::kCheckPortalListProperty: Registered in DefaultProfile |
| 45 | // flimflam::kCountryProperty: Registered in DefaultProfile |
Chris Masone | 7df0c67 | 2011-07-15 10:24:54 -0700 | [diff] [blame] | 46 | store_.RegisterConstString(flimflam::kNameProperty, &name_.identifier); |
Chris Masone | 88cbd5f | 2011-07-03 14:30:04 -0700 | [diff] [blame] | 47 | |
| 48 | // flimflam::kOfflineModeProperty: Registered in DefaultProfile |
| 49 | // flimflam::kPortalURLProperty: Registered in DefaultProfile |
| 50 | |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 51 | HelpRegisterDerivedStrings(flimflam::kServicesProperty, |
| 52 | &Profile::EnumerateAvailableServices, |
| 53 | NULL); |
Chris Masone | 7df0c67 | 2011-07-15 10:24:54 -0700 | [diff] [blame] | 54 | HelpRegisterDerivedStrings(flimflam::kEntriesProperty, |
| 55 | &Profile::EnumerateEntries, |
| 56 | NULL); |
Chris Masone | 52cd19b | 2011-06-29 17:23:04 -0700 | [diff] [blame] | 57 | } |
| 58 | |
Chris Masone | 6515aab | 2011-10-12 16:19:09 -0700 | [diff] [blame] | 59 | Profile::~Profile() {} |
Chris Masone | 52cd19b | 2011-06-29 17:23:04 -0700 | [diff] [blame] | 60 | |
Paul Stewart | 5dc40aa | 2011-10-28 19:43:43 -0700 | [diff] [blame] | 61 | bool Profile::InitStorage(GLib *glib, InitStorageOption storage_option, |
| 62 | Error *error) { |
Chris Masone | b9c0059 | 2011-10-06 13:10:39 -0700 | [diff] [blame] | 63 | FilePath final_path; |
| 64 | if (!GetStoragePath(&final_path)) { |
Paul Stewart | be00517 | 2011-11-02 18:10:29 -0700 | [diff] [blame] | 65 | Error::PopulateAndLog(error, Error::kInvalidArguments, |
Chris Masone | b9c0059 | 2011-10-06 13:10:39 -0700 | [diff] [blame] | 66 | base::StringPrintf("Could not set up profile storage for %s:%s", |
Paul Stewart | be00517 | 2011-11-02 18:10:29 -0700 | [diff] [blame] | 67 | name_.user.c_str(), name_.identifier.c_str())); |
Chris Masone | b9c0059 | 2011-10-06 13:10:39 -0700 | [diff] [blame] | 68 | return false; |
| 69 | } |
| 70 | scoped_ptr<KeyFileStore> storage(new KeyFileStore(glib)); |
| 71 | storage->set_path(final_path); |
Paul Stewart | 5dc40aa | 2011-10-28 19:43:43 -0700 | [diff] [blame] | 72 | bool already_exists = storage->IsNonEmpty(); |
| 73 | if (!already_exists && storage_option != kCreateNew && |
| 74 | storage_option != kCreateOrOpenExisting) { |
Paul Stewart | be00517 | 2011-11-02 18:10:29 -0700 | [diff] [blame] | 75 | Error::PopulateAndLog(error, Error::kNotFound, |
Paul Stewart | 5dc40aa | 2011-10-28 19:43:43 -0700 | [diff] [blame] | 76 | base::StringPrintf("Profile storage for %s:%s does not already exist", |
Paul Stewart | be00517 | 2011-11-02 18:10:29 -0700 | [diff] [blame] | 77 | name_.user.c_str(), name_.identifier.c_str())); |
Paul Stewart | 5dc40aa | 2011-10-28 19:43:43 -0700 | [diff] [blame] | 78 | return false; |
| 79 | } else if (already_exists && storage_option != kOpenExisting && |
| 80 | storage_option != kCreateOrOpenExisting) { |
Paul Stewart | be00517 | 2011-11-02 18:10:29 -0700 | [diff] [blame] | 81 | Error::PopulateAndLog(error, Error::kAlreadyExists, |
Paul Stewart | 5dc40aa | 2011-10-28 19:43:43 -0700 | [diff] [blame] | 82 | base::StringPrintf("Profile storage for %s:%s already exists", |
Paul Stewart | be00517 | 2011-11-02 18:10:29 -0700 | [diff] [blame] | 83 | name_.user.c_str(), name_.identifier.c_str())); |
Paul Stewart | 5dc40aa | 2011-10-28 19:43:43 -0700 | [diff] [blame] | 84 | return false; |
| 85 | } |
Chris Masone | b9c0059 | 2011-10-06 13:10:39 -0700 | [diff] [blame] | 86 | if (!storage->Open()) { |
Paul Stewart | be00517 | 2011-11-02 18:10:29 -0700 | [diff] [blame] | 87 | Error::PopulateAndLog(error, Error::kInternalError, |
Chris Masone | b9c0059 | 2011-10-06 13:10:39 -0700 | [diff] [blame] | 88 | base::StringPrintf("Could not open profile storage for %s:%s", |
Paul Stewart | be00517 | 2011-11-02 18:10:29 -0700 | [diff] [blame] | 89 | name_.user.c_str(), name_.identifier.c_str())); |
Chris Masone | b9c0059 | 2011-10-06 13:10:39 -0700 | [diff] [blame] | 90 | return false; |
| 91 | } |
Paul Stewart | 5dc40aa | 2011-10-28 19:43:43 -0700 | [diff] [blame] | 92 | if (!already_exists) { |
| 93 | // Add a descriptive header to the profile so even if nothing is stored |
| 94 | // to it, it still has some content. Completely empty keyfiles are not |
| 95 | // valid for reading. |
| 96 | storage->SetHeader( |
| 97 | base::StringPrintf("Profile %s:%s", name_.user.c_str(), |
| 98 | name_.identifier.c_str())); |
| 99 | } |
Chris Masone | b9c0059 | 2011-10-06 13:10:39 -0700 | [diff] [blame] | 100 | set_storage(storage.release()); |
| 101 | return true; |
| 102 | } |
| 103 | |
Paul Stewart | e73d05c | 2012-03-29 16:26:05 -0700 | [diff] [blame] | 104 | bool Profile::RemoveStorage(GLib *glib, Error *error) { |
| 105 | FilePath path; |
| 106 | |
| 107 | CHECK(!storage_.get()); |
| 108 | |
| 109 | if (!GetStoragePath(&path)) { |
| 110 | Error::PopulateAndLog( |
| 111 | error, Error::kInvalidArguments, |
| 112 | base::StringPrintf("Could get the storage path for %s:%s", |
| 113 | name_.user.c_str(), name_.identifier.c_str())); |
| 114 | return false; |
| 115 | } |
| 116 | |
| 117 | if (!file_util::Delete(path, false)) { |
| 118 | Error::PopulateAndLog( |
| 119 | error, Error::kOperationFailed, |
| 120 | base::StringPrintf("Could not remove path %s", path.value().c_str())); |
| 121 | return false; |
| 122 | } |
| 123 | |
| 124 | return true; |
| 125 | } |
| 126 | |
Chris Masone | 7df0c67 | 2011-07-15 10:24:54 -0700 | [diff] [blame] | 127 | string Profile::GetFriendlyName() { |
| 128 | return (name_.user.empty() ? "" : name_.user + "/") + name_.identifier; |
| 129 | } |
| 130 | |
| 131 | string Profile::GetRpcIdentifier() { |
Paul Stewart | 19c871d | 2011-12-15 16:10:13 -0800 | [diff] [blame] | 132 | if (!adaptor_.get()) { |
| 133 | // NB: This condition happens in unit tests. |
| 134 | return string(); |
| 135 | } |
Chris Masone | 7df0c67 | 2011-07-15 10:24:54 -0700 | [diff] [blame] | 136 | return adaptor_->GetRpcIdentifier(); |
| 137 | } |
| 138 | |
Chris Masone | b9c0059 | 2011-10-06 13:10:39 -0700 | [diff] [blame] | 139 | void Profile::set_storage(StoreInterface *storage) { |
| 140 | storage_.reset(storage); |
| 141 | } |
| 142 | |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 143 | bool Profile::AdoptService(const ServiceRefPtr &service) { |
Paul Stewart | 451aa7f | 2012-04-11 19:07:58 -0700 | [diff] [blame] | 144 | if (service->profile() == this) { |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 145 | return false; |
Paul Stewart | 451aa7f | 2012-04-11 19:07:58 -0700 | [diff] [blame] | 146 | } |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 147 | service->set_profile(this); |
Chris Masone | 6515aab | 2011-10-12 16:19:09 -0700 | [diff] [blame] | 148 | return service->Save(storage_.get()) && storage_->Flush(); |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 149 | } |
| 150 | |
Chris Masone | 6515aab | 2011-10-12 16:19:09 -0700 | [diff] [blame] | 151 | bool Profile::AbandonService(const ServiceRefPtr &service) { |
| 152 | if (service->profile() == this) |
| 153 | service->set_profile(NULL); |
| 154 | return storage_->DeleteGroup(service->GetStorageIdentifier()) && |
| 155 | storage_->Flush(); |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 156 | } |
| 157 | |
Chris Masone | 6515aab | 2011-10-12 16:19:09 -0700 | [diff] [blame] | 158 | bool Profile::UpdateService(const ServiceRefPtr &service) { |
| 159 | return service->Save(storage_.get()) && storage_->Flush(); |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 160 | } |
| 161 | |
Paul Stewart | bba6a5b | 2011-11-02 18:45:59 -0700 | [diff] [blame] | 162 | bool Profile::ConfigureService(const ServiceRefPtr &service) { |
| 163 | if (!ContainsService(service)) |
Chris Masone | 6515aab | 2011-10-12 16:19:09 -0700 | [diff] [blame] | 164 | return false; |
| 165 | service->set_profile(this); |
| 166 | return service->Load(storage_.get()); |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 167 | } |
| 168 | |
Paul Stewart | a41e38d | 2011-11-11 07:47:29 -0800 | [diff] [blame] | 169 | bool Profile::ConfigureDevice(const DeviceRefPtr &device) { |
| 170 | return device->Load(storage_.get()); |
| 171 | } |
| 172 | |
Chris Masone | 6515aab | 2011-10-12 16:19:09 -0700 | [diff] [blame] | 173 | bool Profile::ContainsService(const ServiceConstRefPtr &service) { |
Paul Stewart | bba6a5b | 2011-11-02 18:45:59 -0700 | [diff] [blame] | 174 | return service->IsLoadableFrom(storage_.get()); |
Chris Masone | 7aa5f90 | 2011-07-11 11:13:35 -0700 | [diff] [blame] | 175 | } |
| 176 | |
Paul Stewart | 7522551 | 2012-01-26 22:51:33 -0800 | [diff] [blame] | 177 | void Profile::DeleteEntry(const std::string &entry_name, Error *error) { |
| 178 | if (!storage_->ContainsGroup(entry_name)) { |
| 179 | Error::PopulateAndLog(error, Error::kNotFound, |
| 180 | base::StringPrintf("Entry %s does not exist in profile", |
| 181 | entry_name.c_str())); |
| 182 | return; |
| 183 | } |
| 184 | if (!manager_->HandleProfileEntryDeletion(this, entry_name)) { |
| 185 | // If HandleProfileEntryDeletion() returns succeeds, DeleteGroup() |
| 186 | // has already been called when AbandonService was called. |
| 187 | // Otherwise, we need to delete the group ourselves. |
| 188 | storage_->DeleteGroup(entry_name); |
| 189 | } |
Paul Stewart | 0756db9 | 2012-01-27 08:34:47 -0800 | [diff] [blame] | 190 | Save(); |
| 191 | } |
| 192 | |
| 193 | ServiceRefPtr Profile::GetServiceFromEntry(const std::string &entry_name, |
| 194 | Error *error) { |
| 195 | return manager_->GetServiceWithStorageIdentifier(this, entry_name, error); |
Paul Stewart | 7522551 | 2012-01-26 22:51:33 -0800 | [diff] [blame] | 196 | } |
| 197 | |
Paul Stewart | a41e38d | 2011-11-11 07:47:29 -0800 | [diff] [blame] | 198 | bool Profile::IsValidIdentifierToken(const string &token) { |
Darin Petkov | a476682 | 2011-07-07 10:42:22 -0700 | [diff] [blame] | 199 | if (token.empty()) { |
| 200 | return false; |
| 201 | } |
| 202 | for (string::const_iterator it = token.begin(); it != token.end(); ++it) { |
| 203 | if (!IsAsciiAlpha(*it) && !IsAsciiDigit(*it)) { |
| 204 | return false; |
| 205 | } |
| 206 | } |
| 207 | return true; |
| 208 | } |
| 209 | |
| 210 | bool Profile::ParseIdentifier(const string &raw, Identifier *parsed) { |
| 211 | if (raw.empty()) { |
| 212 | return false; |
| 213 | } |
| 214 | if (raw[0] == '~') { |
| 215 | // Format: "~user/identifier". |
| 216 | size_t slash = raw.find('/'); |
| 217 | if (slash == string::npos) { |
| 218 | return false; |
| 219 | } |
| 220 | string user(raw.begin() + 1, raw.begin() + slash); |
| 221 | string identifier(raw.begin() + slash + 1, raw.end()); |
| 222 | if (!IsValidIdentifierToken(user) || !IsValidIdentifierToken(identifier)) { |
| 223 | return false; |
| 224 | } |
| 225 | parsed->user = user; |
| 226 | parsed->identifier = identifier; |
| 227 | return true; |
| 228 | } |
| 229 | |
| 230 | // Format: "identifier". |
| 231 | if (!IsValidIdentifierToken(raw)) { |
| 232 | return false; |
| 233 | } |
| 234 | parsed->user = ""; |
| 235 | parsed->identifier = raw; |
| 236 | return true; |
| 237 | } |
| 238 | |
Paul Stewart | 5dc40aa | 2011-10-28 19:43:43 -0700 | [diff] [blame] | 239 | bool Profile::MatchesIdentifier(const Identifier &name) const { |
| 240 | return name.user == name_.user && name.identifier == name_.identifier; |
| 241 | } |
| 242 | |
Chris Masone | b9c0059 | 2011-10-06 13:10:39 -0700 | [diff] [blame] | 243 | bool Profile::Save() { |
Chris Masone | 6515aab | 2011-10-12 16:19:09 -0700 | [diff] [blame] | 244 | return storage_->Flush(); |
Chris Masone | 9d77993 | 2011-08-25 16:33:41 -0700 | [diff] [blame] | 245 | } |
| 246 | |
Chris Masone | 2ae797d | 2011-08-23 20:41:00 -0700 | [diff] [blame] | 247 | bool Profile::GetStoragePath(FilePath *path) { |
| 248 | if (name_.user.empty()) { |
| 249 | LOG(ERROR) << "Non-default profiles cannot be stored globally."; |
| 250 | return false; |
| 251 | } |
| 252 | FilePath dir(base::StringPrintf(storage_format_.c_str(), name_.user.c_str())); |
Darin Petkov | a476682 | 2011-07-07 10:42:22 -0700 | [diff] [blame] | 253 | // TODO(petkov): Validate the directory permissions, etc. |
| 254 | *path = dir.Append(base::StringPrintf("%s.profile", |
Chris Masone | 2ae797d | 2011-08-23 20:41:00 -0700 | [diff] [blame] | 255 | name_.identifier.c_str())); |
Darin Petkov | a476682 | 2011-07-07 10:42:22 -0700 | [diff] [blame] | 256 | return true; |
| 257 | } |
| 258 | |
Gaurav Shah | 1b7a616 | 2011-11-09 11:41:01 -0800 | [diff] [blame] | 259 | vector<string> Profile::EnumerateAvailableServices(Error *error) { |
Paul Stewart | 1b25314 | 2012-01-26 14:05:52 -0800 | [diff] [blame] | 260 | // We should return the Manager's service list if this is the active profile. |
| 261 | if (manager_->IsActiveProfile(this)) { |
| 262 | return manager_->EnumerateAvailableServices(error); |
| 263 | } else { |
| 264 | return vector<string>(); |
| 265 | } |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 266 | } |
| 267 | |
Gaurav Shah | 1b7a616 | 2011-11-09 11:41:01 -0800 | [diff] [blame] | 268 | vector<string> Profile::EnumerateEntries(Error */*error*/) { |
Chris Masone | 6515aab | 2011-10-12 16:19:09 -0700 | [diff] [blame] | 269 | set<string> groups(storage_->GetGroups()); |
Paul Stewart | 1b25314 | 2012-01-26 14:05:52 -0800 | [diff] [blame] | 270 | vector<string> service_groups; |
| 271 | |
| 272 | // Filter this list down to only entries that correspond |
| 273 | // to a technology. (wifi_*, etc) |
| 274 | for (set<string>::iterator it = groups.begin(); |
| 275 | it != groups.end(); ++it) { |
| 276 | if (Technology::IdentifierFromStorageGroup(*it) != Technology::kUnknown) |
| 277 | service_groups.push_back(*it); |
| 278 | } |
| 279 | |
| 280 | return service_groups; |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 281 | } |
| 282 | |
mukesh agrawal | ffa3d04 | 2011-10-06 15:26:10 -0700 | [diff] [blame] | 283 | void Profile::HelpRegisterDerivedStrings( |
| 284 | const string &name, |
Gaurav Shah | 1b7a616 | 2011-11-09 11:41:01 -0800 | [diff] [blame] | 285 | Strings(Profile::*get)(Error *), |
mukesh agrawal | ffa3d04 | 2011-10-06 15:26:10 -0700 | [diff] [blame] | 286 | void(Profile::*set)(const Strings&, Error *)) { |
Chris Masone | 6791a43 | 2011-07-12 13:23:19 -0700 | [diff] [blame] | 287 | store_.RegisterDerivedStrings( |
| 288 | name, |
| 289 | StringsAccessor(new CustomAccessor<Profile, Strings>(this, get, set))); |
| 290 | } |
| 291 | |
Chris Masone | 52cd19b | 2011-06-29 17:23:04 -0700 | [diff] [blame] | 292 | } // namespace shill |