| Index: chrome/browser/chromeos/settings/device_settings_service.cc
|
| diff --git a/chrome/browser/chromeos/settings/device_settings_service.cc b/chrome/browser/chromeos/settings/device_settings_service.cc
|
| index 9ba93243c923ee032a906fe81efc1329754b4f78..36de11dff7393d1c2d910d7d55c302b4d5521cd7 100644
|
| --- a/chrome/browser/chromeos/settings/device_settings_service.cc
|
| +++ b/chrome/browser/chromeos/settings/device_settings_service.cc
|
| @@ -35,6 +35,62 @@ int kLoadRetryDelayMs = 1000 * 5;
|
| // of retry time.
|
| int kMaxLoadRetries = (1000 * 60 * 10) / kLoadRetryDelayMs;
|
|
|
| +// Assembles PolicyData based on |settings|, |policy_data| and
|
| +// |user_id|.
|
| +scoped_ptr<em::PolicyData> AssemblePolicy(
|
| + const std::string& user_id,
|
| + const em::PolicyData* policy_data,
|
| + const em::ChromeDeviceSettingsProto* settings) {
|
| + scoped_ptr<em::PolicyData> policy(new em::PolicyData());
|
| + if (policy_data) {
|
| + // Preserve management settings.
|
| + if (policy_data->has_management_mode())
|
| + policy->set_management_mode(policy_data->management_mode());
|
| + if (policy_data->has_request_token())
|
| + policy->set_request_token(policy_data->request_token());
|
| + if (policy_data->has_device_id())
|
| + policy->set_device_id(policy_data->device_id());
|
| + } else {
|
| + // If there's no previous policy data, this is the first time the device
|
| + // setting is set. We set the management mode to NOT_MANAGED initially.
|
| + policy->set_management_mode(em::PolicyData::NOT_MANAGED);
|
| + }
|
| + policy->set_policy_type(policy::dm_protocol::kChromeDevicePolicyType);
|
| + policy->set_timestamp(
|
| + (base::Time::Now() - base::Time::UnixEpoch()).InMilliseconds());
|
| + policy->set_username(user_id);
|
| + if (!settings->SerializeToString(policy->mutable_policy_value()))
|
| + return scoped_ptr<em::PolicyData>();
|
| +
|
| + return policy.Pass();
|
| +}
|
| +
|
| +// Returns true if it is okay to transfer from the current mode to the new
|
| +// mode. This function should be called in SetManagementMode().
|
| +bool CheckManagementModeTransition(em::PolicyData::ManagementMode current_mode,
|
| + em::PolicyData::ManagementMode new_mode) {
|
| + // Mode is not changed.
|
| + if (current_mode == new_mode)
|
| + return true;
|
| +
|
| + switch (current_mode) {
|
| + case em::PolicyData::NOT_MANAGED:
|
| + // For consumer management enrollment.
|
| + return new_mode == em::PolicyData::CONSUMER_MANAGED;
|
| +
|
| + case em::PolicyData::ENTERPRISE_MANAGED:
|
| + // Management mode cannot be set when it is currently ENTERPRISE_MANAGED.
|
| + return false;
|
| +
|
| + case em::PolicyData::CONSUMER_MANAGED:
|
| + // For consumer management unenrollment.
|
| + return new_mode == em::PolicyData::NOT_MANAGED;
|
| + }
|
| +
|
| + NOTREACHED();
|
| + return false;
|
| +}
|
| +
|
| } // namespace
|
|
|
| namespace chromeos {
|
| @@ -116,10 +172,18 @@ void DeviceSettingsService::Load() {
|
| void DeviceSettingsService::SignAndStore(
|
| scoped_ptr<em::ChromeDeviceSettingsProto> new_settings,
|
| const base::Closure& callback) {
|
| - if (!delegate_)
|
| + if (!owner_settings_service_) {
|
| HandleError(STORE_KEY_UNAVAILABLE, callback);
|
| - else
|
| - delegate_->SignAndStoreAsync(new_settings.Pass(), callback);
|
| + return;
|
| + }
|
| + scoped_ptr<em::PolicyData> policy =
|
| + AssemblePolicy(GetUsername(), policy_data(), new_settings.get());
|
| + if (!policy) {
|
| + HandleError(STORE_POLICY_ERROR, callback);
|
| + return;
|
| + }
|
| +
|
| + owner_settings_service_->SignAndStorePolicyAsync(policy.Pass(), callback);
|
| }
|
|
|
| void DeviceSettingsService::SetManagementSettings(
|
| @@ -127,12 +191,34 @@ void DeviceSettingsService::SetManagementSettings(
|
| const std::string& request_token,
|
| const std::string& device_id,
|
| const base::Closure& callback) {
|
| - if (!delegate_) {
|
| + if (!owner_settings_service_) {
|
| HandleError(STORE_KEY_UNAVAILABLE, callback);
|
| - } else {
|
| - delegate_->SetManagementSettingsAsync(
|
| - management_mode, request_token, device_id, callback);
|
| + return;
|
| + }
|
| +
|
| + em::PolicyData::ManagementMode current_mode = em::PolicyData::NOT_MANAGED;
|
| + if (policy_data() && policy_data()->has_management_mode())
|
| + current_mode = policy_data()->management_mode();
|
| +
|
| + if (!CheckManagementModeTransition(current_mode, management_mode)) {
|
| + LOG(ERROR) << "Invalid management mode transition: current mode = "
|
| + << current_mode << ", new mode = " << management_mode;
|
| + HandleError(DeviceSettingsService::STORE_POLICY_ERROR, callback);
|
| + return;
|
| + }
|
| +
|
| + scoped_ptr<em::PolicyData> policy =
|
| + AssemblePolicy(GetUsername(), policy_data(), device_settings());
|
| + if (!policy) {
|
| + HandleError(DeviceSettingsService::STORE_POLICY_ERROR, callback);
|
| + return;
|
| }
|
| +
|
| + policy->set_management_mode(management_mode);
|
| + policy->set_request_token(request_token);
|
| + policy->set_device_id(device_id);
|
| +
|
| + owner_settings_service_->SignAndStorePolicyAsync(policy.Pass(), callback);
|
| }
|
|
|
| void DeviceSettingsService::Store(scoped_ptr<em::PolicyFetchResponse> policy,
|
| @@ -169,18 +255,19 @@ void DeviceSettingsService::GetOwnershipStatusAsync(
|
| }
|
|
|
| bool DeviceSettingsService::HasPrivateOwnerKey() {
|
| - return delegate_ && delegate_->IsOwner();
|
| + return owner_settings_service_ && owner_settings_service_->IsOwner();
|
| }
|
|
|
| void DeviceSettingsService::InitOwner(
|
| const std::string& username,
|
| - const base::WeakPtr<PrivateKeyDelegate>& delegate) {
|
| + const base::WeakPtr<ownership::OwnerSettingsService>&
|
| + owner_settings_service) {
|
| // When InitOwner() is called twice with the same |username| it's
|
| // worth to reload settings since owner key may become available.
|
| if (!username_.empty() && username_ != username)
|
| return;
|
| username_ = username;
|
| - delegate_ = delegate;
|
| + owner_settings_service_ = owner_settings_service;
|
|
|
| EnsureReload(true);
|
| }
|
| @@ -230,14 +317,15 @@ void DeviceSettingsService::EnqueueLoad(bool force_key_load) {
|
| base::Closure()));
|
| operation->set_force_key_load(force_key_load);
|
| operation->set_username(username_);
|
| - operation->set_delegate(delegate_);
|
| + operation->set_owner_settings_service(owner_settings_service_);
|
| Enqueue(operation);
|
| }
|
|
|
| void DeviceSettingsService::EnsureReload(bool force_key_load) {
|
| if (!pending_operations_.empty()) {
|
| pending_operations_.front()->set_username(username_);
|
| - pending_operations_.front()->set_delegate(delegate_);
|
| + pending_operations_.front()->set_owner_settings_service(
|
| + owner_settings_service_);
|
| pending_operations_.front()->RestartLoad(force_key_load);
|
| } else {
|
| EnqueueLoad(force_key_load);
|
|
|