session_manager_operation.cc revision 1e9bf3e0803691d0a228da41fc608347b6db4340
1// Copyright (c) 2012 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 "chrome/browser/chromeos/settings/session_manager_operation.h" 6 7#include "base/bind.h" 8#include "base/bind_helpers.h" 9#include "base/files/file_path.h" 10#include "base/message_loop/message_loop.h" 11#include "base/stl_util.h" 12#include "base/task_runner_util.h" 13#include "base/threading/sequenced_worker_pool.h" 14#include "base/time/time.h" 15#include "chrome/browser/chromeos/settings/owner_key_util.h" 16#include "chrome/browser/policy/cloud/cloud_policy_constants.h" 17#include "chrome/browser/policy/proto/chromeos/chrome_device_policy.pb.h" 18#include "chrome/browser/policy/proto/cloud/device_management_backend.pb.h" 19#include "content/public/browser/browser_thread.h" 20#include "crypto/rsa_private_key.h" 21#include "crypto/signature_creator.h" 22 23namespace em = enterprise_management; 24 25namespace chromeos { 26 27SessionManagerOperation::SessionManagerOperation(const Callback& callback) 28 : session_manager_client_(NULL), 29 weak_factory_(this), 30 callback_(callback), 31 force_key_load_(false), 32 is_loading_(false) {} 33 34SessionManagerOperation::~SessionManagerOperation() {} 35 36void SessionManagerOperation::Start( 37 SessionManagerClient* session_manager_client, 38 scoped_refptr<OwnerKeyUtil> owner_key_util, 39 scoped_refptr<OwnerKey> owner_key) { 40 session_manager_client_ = session_manager_client; 41 owner_key_util_ = owner_key_util; 42 owner_key_ = owner_key; 43 Run(); 44} 45 46void SessionManagerOperation::RestartLoad(bool key_changed) { 47 if (key_changed) 48 owner_key_ = NULL; 49 50 if (!is_loading_) 51 return; 52 53 // Abort previous load operations. 54 weak_factory_.InvalidateWeakPtrs(); 55 // Mark as not loading to start loading again. 56 is_loading_ = false; 57 StartLoading(); 58} 59 60void SessionManagerOperation::StartLoading() { 61 if (is_loading_) 62 return; 63 is_loading_ = true; 64 EnsureOwnerKey(base::Bind(&SessionManagerOperation::RetrieveDeviceSettings, 65 weak_factory_.GetWeakPtr())); 66} 67 68void SessionManagerOperation::ReportResult( 69 DeviceSettingsService::Status status) { 70 callback_.Run(this, status); 71} 72 73void SessionManagerOperation::EnsureOwnerKey(const base::Closure& callback) { 74 if (force_key_load_ || !owner_key_.get() || !owner_key_->public_key()) { 75 scoped_refptr<base::TaskRunner> task_runner = 76 content::BrowserThread::GetBlockingPool()-> 77 GetTaskRunnerWithShutdownBehavior( 78 base::SequencedWorkerPool::SKIP_ON_SHUTDOWN); 79 base::PostTaskAndReplyWithResult( 80 task_runner.get(), 81 FROM_HERE, 82 base::Bind(&SessionManagerOperation::LoadOwnerKey, 83 owner_key_util_, owner_key_), 84 base::Bind(&SessionManagerOperation::StoreOwnerKey, 85 weak_factory_.GetWeakPtr(), callback)); 86 } else { 87 callback.Run(); 88 } 89} 90 91// static 92scoped_refptr<OwnerKey> SessionManagerOperation::LoadOwnerKey( 93 scoped_refptr<OwnerKeyUtil> util, 94 scoped_refptr<OwnerKey> current_key) { 95 scoped_ptr<std::vector<uint8> > public_key; 96 scoped_ptr<crypto::RSAPrivateKey> private_key; 97 98 // Keep any already-existing keys. 99 if (current_key.get()) { 100 if (current_key->public_key()) 101 public_key.reset(new std::vector<uint8>(*current_key->public_key())); 102 if (current_key->private_key()) 103 private_key.reset(current_key->private_key()->Copy()); 104 } 105 106 if (!public_key.get() && util->IsPublicKeyPresent()) { 107 public_key.reset(new std::vector<uint8>()); 108 if (!util->ImportPublicKey(public_key.get())) 109 LOG(ERROR) << "Failed to load public owner key."; 110 } 111 112 if (public_key.get() && !private_key.get()) { 113 private_key.reset(util->FindPrivateKey(*public_key)); 114 if (!private_key.get()) 115 VLOG(1) << "Failed to load private owner key."; 116 } 117 118 return new OwnerKey(public_key.Pass(), private_key.Pass()); 119} 120 121void SessionManagerOperation::StoreOwnerKey(const base::Closure& callback, 122 scoped_refptr<OwnerKey> new_key) { 123 force_key_load_ = false; 124 owner_key_ = new_key; 125 126 if (!owner_key_.get() || !owner_key_->public_key()) { 127 ReportResult(DeviceSettingsService::STORE_KEY_UNAVAILABLE); 128 return; 129 } 130 131 callback.Run(); 132} 133 134void SessionManagerOperation::RetrieveDeviceSettings() { 135 session_manager_client()->RetrieveDevicePolicy( 136 base::Bind(&SessionManagerOperation::ValidateDeviceSettings, 137 weak_factory_.GetWeakPtr())); 138} 139 140void SessionManagerOperation::ValidateDeviceSettings( 141 const std::string& policy_blob) { 142 scoped_ptr<em::PolicyFetchResponse> policy(new em::PolicyFetchResponse()); 143 if (policy_blob.empty()) { 144 ReportResult(DeviceSettingsService::STORE_NO_POLICY); 145 return; 146 } 147 148 if (!policy->ParseFromString(policy_blob) || 149 !policy->IsInitialized()) { 150 ReportResult(DeviceSettingsService::STORE_INVALID_POLICY); 151 return; 152 } 153 154 base::SequencedWorkerPool* pool = 155 content::BrowserThread::GetBlockingPool(); 156 scoped_refptr<base::SequencedTaskRunner> background_task_runner = 157 pool->GetSequencedTaskRunnerWithShutdownBehavior( 158 pool->GetSequenceToken(), 159 base::SequencedWorkerPool::SKIP_ON_SHUTDOWN); 160 161 policy::DeviceCloudPolicyValidator* validator = 162 policy::DeviceCloudPolicyValidator::Create(policy.Pass(), 163 background_task_runner); 164 165 // Policy auto-generated by session manager doesn't include a timestamp, so we 166 // need to allow missing timestamps. 167 const bool require_timestamp = 168 policy_data_.get() && policy_data_->has_request_token(); 169 validator->ValidateAgainstCurrentPolicy( 170 policy_data_.get(), 171 require_timestamp ? 172 policy::CloudPolicyValidatorBase::TIMESTAMP_REQUIRED : 173 policy::CloudPolicyValidatorBase::TIMESTAMP_NOT_REQUIRED, 174 policy::CloudPolicyValidatorBase::DM_TOKEN_NOT_REQUIRED); 175 validator->ValidatePolicyType(policy::dm_protocol::kChromeDevicePolicyType); 176 validator->ValidatePayload(); 177 validator->ValidateSignature(*owner_key_->public_key(), false); 178 validator->StartValidation( 179 base::Bind(&SessionManagerOperation::ReportValidatorStatus, 180 weak_factory_.GetWeakPtr())); 181} 182 183void SessionManagerOperation::ReportValidatorStatus( 184 policy::DeviceCloudPolicyValidator* validator) { 185 DeviceSettingsService::Status status = 186 DeviceSettingsService::STORE_VALIDATION_ERROR; 187 if (validator->success()) { 188 status = DeviceSettingsService::STORE_SUCCESS; 189 policy_data_ = validator->policy_data().Pass(); 190 device_settings_ = validator->payload().Pass(); 191 } else { 192 LOG(ERROR) << "Policy validation failed: " << validator->status(); 193 194 // Those are mostly caused by RTC loss and are recoverable. 195 if (validator->status() == 196 policy::DeviceCloudPolicyValidator::VALIDATION_BAD_TIMESTAMP) { 197 status = DeviceSettingsService::STORE_TEMP_VALIDATION_ERROR; 198 } 199 } 200 201 ReportResult(status); 202} 203 204LoadSettingsOperation::LoadSettingsOperation(const Callback& callback) 205 : SessionManagerOperation(callback) {} 206 207LoadSettingsOperation::~LoadSettingsOperation() {} 208 209void LoadSettingsOperation::Run() { 210 StartLoading(); 211} 212 213StoreSettingsOperation::StoreSettingsOperation( 214 const Callback& callback, 215 scoped_ptr<em::PolicyFetchResponse> policy) 216 : SessionManagerOperation(callback), 217 policy_(policy.Pass()), 218 weak_factory_(this) {} 219 220StoreSettingsOperation::~StoreSettingsOperation() {} 221 222void StoreSettingsOperation::Run() { 223 session_manager_client()->StoreDevicePolicy( 224 policy_->SerializeAsString(), 225 base::Bind(&StoreSettingsOperation::HandleStoreResult, 226 weak_factory_.GetWeakPtr())); 227} 228 229void StoreSettingsOperation::HandleStoreResult(bool success) { 230 if (!success) 231 ReportResult(DeviceSettingsService::STORE_OPERATION_FAILED); 232 else 233 StartLoading(); 234} 235 236SignAndStoreSettingsOperation::SignAndStoreSettingsOperation( 237 const Callback& callback, 238 scoped_ptr<em::ChromeDeviceSettingsProto> new_settings, 239 const std::string& username) 240 : SessionManagerOperation(callback), 241 new_settings_(new_settings.Pass()), 242 username_(username), 243 weak_factory_(this) { 244 DCHECK(new_settings_.get()); 245} 246 247SignAndStoreSettingsOperation::~SignAndStoreSettingsOperation() {} 248 249void SignAndStoreSettingsOperation::Run() { 250 EnsureOwnerKey(base::Bind(&SignAndStoreSettingsOperation::StartSigning, 251 weak_factory_.GetWeakPtr())); 252} 253 254void SignAndStoreSettingsOperation::StartSigning() { 255 if (!owner_key().get() || !owner_key()->private_key() || username_.empty()) { 256 ReportResult(DeviceSettingsService::STORE_KEY_UNAVAILABLE); 257 return; 258 } 259 260 base::PostTaskAndReplyWithResult( 261 content::BrowserThread::GetBlockingPool(), 262 FROM_HERE, 263 base::Bind(&SignAndStoreSettingsOperation::AssembleAndSignPolicy, 264 base::Passed(&new_settings_), username_, owner_key()), 265 base::Bind(&SignAndStoreSettingsOperation::StoreDeviceSettingsBlob, 266 weak_factory_.GetWeakPtr())); 267} 268 269// static 270std::string SignAndStoreSettingsOperation::AssembleAndSignPolicy( 271 scoped_ptr<em::ChromeDeviceSettingsProto> device_settings, 272 const std::string& username, 273 scoped_refptr<OwnerKey> owner_key) { 274 // Assemble the policy. 275 em::PolicyFetchResponse policy_response; 276 em::PolicyData policy; 277 policy.set_policy_type(policy::dm_protocol::kChromeDevicePolicyType); 278 policy.set_timestamp((base::Time::NowFromSystemTime() - 279 base::Time::UnixEpoch()).InMilliseconds()); 280 policy.set_username(username); 281 if (!device_settings->SerializeToString(policy.mutable_policy_value()) || 282 !policy.SerializeToString(policy_response.mutable_policy_data())) { 283 LOG(ERROR) << "Failed to encode policy payload."; 284 return std::string(); 285 } 286 287 // Generate the signature. 288 scoped_ptr<crypto::SignatureCreator> signature_creator( 289 crypto::SignatureCreator::Create(owner_key->private_key())); 290 signature_creator->Update( 291 reinterpret_cast<const uint8*>(policy_response.policy_data().c_str()), 292 policy_response.policy_data().size()); 293 std::vector<uint8> signature_bytes; 294 std::string policy_blob; 295 if (!signature_creator->Final(&signature_bytes)) { 296 LOG(ERROR) << "Failed to create policy signature."; 297 return std::string(); 298 } 299 300 policy_response.mutable_policy_data_signature()->assign( 301 reinterpret_cast<const char*>(vector_as_array(&signature_bytes)), 302 signature_bytes.size()); 303 return policy_response.SerializeAsString(); 304} 305 306void SignAndStoreSettingsOperation::StoreDeviceSettingsBlob( 307 std::string device_settings_blob) { 308 if (device_settings_blob.empty()) { 309 ReportResult(DeviceSettingsService::STORE_POLICY_ERROR); 310 return; 311 } 312 313 session_manager_client()->StoreDevicePolicy( 314 device_settings_blob, 315 base::Bind(&SignAndStoreSettingsOperation::HandleStoreResult, 316 weak_factory_.GetWeakPtr())); 317} 318 319void SignAndStoreSettingsOperation::HandleStoreResult(bool success) { 320 if (!success) 321 ReportResult(DeviceSettingsService::STORE_OPERATION_FAILED); 322 else 323 StartLoading(); 324} 325 326} // namespace chromeos 327