blob: 1596d0af84ed6709f839f94aac13d113b4330849 [file] [log] [blame]
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/chromeos/policy/enrollment_handler_chromeos.h"
#include <utility>
#include "base/bind.h"
#include "base/command_line.h"
#include "base/location.h"
#include "base/logging.h"
#include "base/single_thread_task_runner.h"
#include "base/threading/thread_task_runner_handle.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/chromeos/login/enrollment/auto_enrollment_controller.h"
#include "chrome/browser/chromeos/ownership/owner_settings_service_chromeos.h"
#include "chrome/browser/chromeos/policy/device_cloud_policy_store_chromeos.h"
#include "chrome/browser/chromeos/policy/dm_token_storage.h"
#include "chrome/browser/chromeos/policy/enrollment_status_chromeos.h"
#include "chrome/browser/chromeos/policy/proto/chrome_device_policy.pb.h"
#include "chrome/browser/chromeos/policy/server_backed_state_keys_broker.h"
#include "chrome/browser/chromeos/profiles/profile_helper.h"
#include "chrome/browser/chromeos/settings/device_oauth2_token_service.h"
#include "chrome/browser/chromeos/settings/device_oauth2_token_service_factory.h"
#include "chrome/browser/profiles/profile.h"
#include "chromeos/attestation/attestation_flow.h"
#include "chromeos/chromeos_switches.h"
#include "chromeos/dbus/auth_policy_client.h"
#include "chromeos/dbus/dbus_thread_manager.h"
#include "google_apis/gaia/gaia_auth_util.h"
#include "google_apis/gaia/gaia_urls.h"
#include "net/http/http_status_code.h"
namespace em = enterprise_management;
namespace policy {
namespace {
// Retry for InstallAttrs initialization every 500ms.
const int kLockRetryIntervalMs = 500;
// Maximum time to retry InstallAttrs initialization before we give up.
const int kLockRetryTimeoutMs = 10 * 60 * 1000; // 10 minutes.
em::DeviceRegisterRequest::Flavor EnrollmentModeToRegistrationFlavor(
EnrollmentConfig::Mode mode) {
switch (mode) {
case EnrollmentConfig::MODE_NONE:
break;
case EnrollmentConfig::MODE_MANUAL:
return em::DeviceRegisterRequest::FLAVOR_ENROLLMENT_MANUAL;
case EnrollmentConfig::MODE_MANUAL_REENROLLMENT:
return em::DeviceRegisterRequest::FLAVOR_ENROLLMENT_MANUAL_RENEW;
case EnrollmentConfig::MODE_LOCAL_FORCED:
return em::DeviceRegisterRequest::FLAVOR_ENROLLMENT_LOCAL_FORCED;
case EnrollmentConfig::MODE_LOCAL_ADVERTISED:
return em::DeviceRegisterRequest::FLAVOR_ENROLLMENT_LOCAL_ADVERTISED;
case EnrollmentConfig::MODE_SERVER_FORCED:
return em::DeviceRegisterRequest::FLAVOR_ENROLLMENT_SERVER_FORCED;
case EnrollmentConfig::MODE_SERVER_ADVERTISED:
return em::DeviceRegisterRequest::FLAVOR_ENROLLMENT_SERVER_ADVERTISED;
case EnrollmentConfig::MODE_RECOVERY:
return em::DeviceRegisterRequest::FLAVOR_ENROLLMENT_RECOVERY;
case EnrollmentConfig::MODE_ATTESTATION:
return em::DeviceRegisterRequest::FLAVOR_ENROLLMENT_ATTESTATION;
case EnrollmentConfig::MODE_ATTESTATION_FORCED:
return em::DeviceRegisterRequest::FLAVOR_ENROLLMENT_ATTESTATION_FORCED;
}
NOTREACHED() << "Bad enrollment mode: " << mode;
return em::DeviceRegisterRequest::FLAVOR_ENROLLMENT_MANUAL;
}
} // namespace
EnrollmentHandlerChromeOS::EnrollmentHandlerChromeOS(
DeviceCloudPolicyStoreChromeOS* store,
chromeos::InstallAttributes* install_attributes,
ServerBackedStateKeysBroker* state_keys_broker,
chromeos::attestation::AttestationFlow* attestation_flow,
std::unique_ptr<CloudPolicyClient> client,
scoped_refptr<base::SequencedTaskRunner> background_task_runner,
const EnrollmentConfig& enrollment_config,
const std::string& auth_token,
const std::string& client_id,
const std::string& requisition,
const EnrollmentCallback& completion_callback)
: store_(store),
install_attributes_(install_attributes),
state_keys_broker_(state_keys_broker),
attestation_flow_(attestation_flow),
client_(std::move(client)),
background_task_runner_(background_task_runner),
enrollment_config_(enrollment_config),
auth_token_(auth_token),
client_id_(client_id),
requisition_(requisition),
completion_callback_(completion_callback),
device_mode_(DEVICE_MODE_NOT_SET),
skip_robot_auth_(false),
enrollment_step_(STEP_PENDING),
lockbox_init_duration_(0),
weak_ptr_factory_(this) {
CHECK(!client_->is_registered());
CHECK_EQ(DM_STATUS_SUCCESS, client_->status());
CHECK((enrollment_config_.mode == EnrollmentConfig::MODE_ATTESTATION ||
enrollment_config_.mode ==
EnrollmentConfig::MODE_ATTESTATION_FORCED) == auth_token_.empty());
CHECK(enrollment_config_.auth_mechanism !=
EnrollmentConfig::AUTH_MECHANISM_ATTESTATION ||
attestation_flow_);
store_->AddObserver(this);
client_->AddObserver(this);
client_->AddPolicyTypeToFetch(dm_protocol::kChromeDevicePolicyType,
std::string());
}
EnrollmentHandlerChromeOS::~EnrollmentHandlerChromeOS() {
Stop();
store_->RemoveObserver(this);
}
void EnrollmentHandlerChromeOS::StartEnrollment() {
CHECK_EQ(STEP_PENDING, enrollment_step_);
SetStep(STEP_STATE_KEYS);
if (client_->machine_id().empty()) {
LOG(ERROR) << "Machine id empty.";
ReportResult(EnrollmentStatus::ForStatus(
EnrollmentStatus::NO_MACHINE_IDENTIFICATION));
return;
}
if (client_->machine_model().empty()) {
LOG(ERROR) << "Machine model empty.";
ReportResult(EnrollmentStatus::ForStatus(
EnrollmentStatus::NO_MACHINE_IDENTIFICATION));
return;
}
state_keys_broker_->RequestStateKeys(
base::Bind(&EnrollmentHandlerChromeOS::HandleStateKeysResult,
weak_ptr_factory_.GetWeakPtr()));
}
std::unique_ptr<CloudPolicyClient> EnrollmentHandlerChromeOS::ReleaseClient() {
Stop();
return std::move(client_);
}
void EnrollmentHandlerChromeOS::OnPolicyFetched(CloudPolicyClient* client) {
DCHECK_EQ(client_.get(), client);
CHECK_EQ(STEP_POLICY_FETCH, enrollment_step_);
SetStep(STEP_VALIDATION);
// Validate the policy.
const em::PolicyFetchResponse* policy = client_->GetPolicyFor(
dm_protocol::kChromeDevicePolicyType, std::string());
if (!policy) {
ReportResult(EnrollmentStatus::ForFetchError(
DM_STATUS_RESPONSE_DECODING_ERROR));
return;
}
std::unique_ptr<DeviceCloudPolicyValidator> validator(
DeviceCloudPolicyValidator::Create(
std::unique_ptr<em::PolicyFetchResponse>(
new em::PolicyFetchResponse(*policy)),
background_task_runner_));
validator->ValidateTimestamp(
base::Time(), base::Time::NowFromSystemTime(),
CloudPolicyValidatorBase::TIMESTAMP_FULLY_VALIDATED);
// If this is re-enrollment, make sure that the new policy matches the
// previously-enrolled domain. (Currently only implemented for cloud
// management.)
std::string domain;
if (install_attributes_->IsCloudManaged()) {
domain = install_attributes_->GetDomain();
validator->ValidateDomain(domain);
}
validator->ValidateDMToken(client->dm_token(),
CloudPolicyValidatorBase::DM_TOKEN_REQUIRED);
validator->ValidatePolicyType(dm_protocol::kChromeDevicePolicyType);
validator->ValidatePayload();
// If |domain| is empty here, the policy validation code will just use the
// domain from the username field in the policy itself to do key validation.
// TODO(mnissler): Plumb the enrolling user's username into this object so we
// can validate the username on the resulting policy, and use the domain from
// that username to validate the key below (http://crbug.com/343074).
validator->ValidateInitialKey(domain);
validator.release()->StartValidation(
base::Bind(&EnrollmentHandlerChromeOS::HandlePolicyValidationResult,
weak_ptr_factory_.GetWeakPtr()));
}
void EnrollmentHandlerChromeOS::OnRegistrationStateChanged(
CloudPolicyClient* client) {
DCHECK_EQ(client_.get(), client);
if (enrollment_step_ == STEP_REGISTRATION && client_->is_registered()) {
SetStep(STEP_POLICY_FETCH);
device_mode_ = client_->device_mode();
if (!((device_mode_ == DEVICE_MODE_ENTERPRISE &&
enrollment_config_.management_realm.empty()) ||
(device_mode_ == DEVICE_MODE_ENTERPRISE_AD &&
!enrollment_config_.management_realm.empty()))) {
LOG(ERROR) << "Bad device mode " << device_mode_;
ReportResult(
EnrollmentStatus::ForStatus(EnrollmentStatus::REGISTRATION_BAD_MODE));
return;
}
client_->FetchPolicy();
} else {
LOG(FATAL) << "Registration state changed to " << client_->is_registered()
<< " in step " << enrollment_step_ << ".";
}
}
void EnrollmentHandlerChromeOS::OnClientError(CloudPolicyClient* client) {
DCHECK_EQ(client_.get(), client);
if (enrollment_step_ == STEP_ROBOT_AUTH_FETCH) {
LOG(ERROR) << "API authentication code fetch failed: "
<< client_->status();
ReportResult(EnrollmentStatus::ForRobotAuthFetchError(client_->status()));
} else if (enrollment_step_ < STEP_POLICY_FETCH) {
ReportResult(EnrollmentStatus::ForRegistrationError(client_->status()));
} else {
ReportResult(EnrollmentStatus::ForFetchError(client_->status()));
}
}
void EnrollmentHandlerChromeOS::OnStoreLoaded(CloudPolicyStore* store) {
DCHECK_EQ(store_, store);
if (enrollment_step_ == STEP_LOADING_STORE) {
// If the |store_| wasn't initialized when StartEnrollment() was called,
// then StartRegistration() bails silently. This gets registration rolling
// again after the store finishes loading.
StartRegistration();
} else if (enrollment_step_ == STEP_STORE_POLICY) {
ReportResult(EnrollmentStatus::ForStatus(EnrollmentStatus::SUCCESS));
}
}
void EnrollmentHandlerChromeOS::OnStoreError(CloudPolicyStore* store) {
DCHECK_EQ(store_, store);
LOG(ERROR) << "Error in device policy store.";
ReportResult(EnrollmentStatus::ForStoreError(store_->status(),
store_->validation_status()));
}
void EnrollmentHandlerChromeOS::HandleStateKeysResult(
const std::vector<std::string>& state_keys) {
DCHECK_EQ(STEP_STATE_KEYS, enrollment_step_);
// Make sure state keys are available if forced re-enrollment is on.
if (chromeos::AutoEnrollmentController::GetMode() ==
chromeos::AutoEnrollmentController::MODE_FORCED_RE_ENROLLMENT) {
client_->SetStateKeysToUpload(state_keys);
current_state_key_ = state_keys_broker_->current_state_key();
if (state_keys.empty() || current_state_key_.empty()) {
ReportResult(
EnrollmentStatus::ForStatus(EnrollmentStatus::NO_STATE_KEYS));
return;
}
}
SetStep(STEP_LOADING_STORE);
StartRegistration();
}
void EnrollmentHandlerChromeOS::StartRegistration() {
DCHECK_EQ(STEP_LOADING_STORE, enrollment_step_);
if (!store_->is_initialized()) {
// Do nothing. StartRegistration() will be called again from OnStoreLoaded()
// after the CloudPolicyStore has initialized.
return;
}
SetStep(STEP_REGISTRATION);
if (enrollment_config_.is_mode_attestation()) {
StartAttestationBasedEnrollmentFlow();
} else {
client_->Register(
em::DeviceRegisterRequest::DEVICE,
EnrollmentModeToRegistrationFlavor(enrollment_config_.mode),
auth_token_, client_id_, requisition_, current_state_key_);
}
}
void EnrollmentHandlerChromeOS::StartAttestationBasedEnrollmentFlow() {
const chromeos::attestation::AttestationFlow::CertificateCallback callback =
base::Bind(
&EnrollmentHandlerChromeOS::HandleRegistrationCertificateResult,
weak_ptr_factory_.GetWeakPtr());
attestation_flow_->GetCertificate(
chromeos::attestation::PROFILE_ENTERPRISE_ENROLLMENT_CERTIFICATE,
EmptyAccountId(), "" /* request_origin */, false /* force_new_key */,
callback);
}
void EnrollmentHandlerChromeOS::HandleRegistrationCertificateResult(
bool success,
const std::string& pem_certificate_chain) {
if (success)
client_->RegisterWithCertificate(
em::DeviceRegisterRequest::DEVICE,
EnrollmentModeToRegistrationFlavor(enrollment_config_.mode),
pem_certificate_chain, client_id_, requisition_, current_state_key_);
else
ReportResult(EnrollmentStatus::ForStatus(
EnrollmentStatus::REGISTRATION_CERT_FETCH_FAILED));
}
void EnrollmentHandlerChromeOS::HandlePolicyValidationResult(
DeviceCloudPolicyValidator* validator) {
DCHECK_EQ(STEP_VALIDATION, enrollment_step_);
if (validator->success()) {
std::string username = validator->policy_data()->username();
device_id_ = validator->policy_data()->device_id();
policy_ = std::move(validator->policy());
if (device_mode_ == DEVICE_MODE_ENTERPRISE_AD) {
// Don't use robot account for the Active Directory managed devices.
skip_robot_auth_ = true;
SetStep(STEP_LOCK_DEVICE);
StartLockDevice();
} else {
domain_ = gaia::ExtractDomainName(gaia::CanonicalizeEmail(username));
SetStep(STEP_ROBOT_AUTH_FETCH);
client_->FetchRobotAuthCodes(auth_token_);
}
} else {
ReportResult(EnrollmentStatus::ForValidationError(validator->status()));
}
}
void EnrollmentHandlerChromeOS::OnRobotAuthCodesFetched(
CloudPolicyClient* client) {
DCHECK_EQ(client_.get(), client);
CHECK_EQ(STEP_ROBOT_AUTH_FETCH, enrollment_step_);
if (client->robot_api_auth_code().empty()) {
// If the server doesn't provide an auth code, skip the robot auth setup.
// This allows clients running against the test server to transparently skip
// robot auth.
skip_robot_auth_ = true;
SetStep(STEP_LOCK_DEVICE);
StartLockDevice();
return;
}
SetStep(STEP_ROBOT_AUTH_REFRESH);
gaia::OAuthClientInfo client_info;
client_info.client_id = GaiaUrls::GetInstance()->oauth2_chrome_client_id();
client_info.client_secret =
GaiaUrls::GetInstance()->oauth2_chrome_client_secret();
client_info.redirect_uri = "oob";
// Use the system request context to avoid sending user cookies.
gaia_oauth_client_.reset(new gaia::GaiaOAuthClient(
g_browser_process->system_request_context()));
gaia_oauth_client_->GetTokensFromAuthCode(client_info,
client->robot_api_auth_code(),
0 /* max_retries */,
this);
}
// GaiaOAuthClient::Delegate callback for OAuth2 refresh token fetched.
void EnrollmentHandlerChromeOS::OnGetTokensResponse(
const std::string& refresh_token,
const std::string& access_token,
int expires_in_seconds) {
CHECK_EQ(STEP_ROBOT_AUTH_REFRESH, enrollment_step_);
robot_refresh_token_ = refresh_token;
SetStep(STEP_LOCK_DEVICE);
StartLockDevice();
}
// GaiaOAuthClient::Delegate
void EnrollmentHandlerChromeOS::OnRefreshTokenResponse(
const std::string& access_token,
int expires_in_seconds) {
// We never use the code that should trigger this callback.
LOG(FATAL) << "Unexpected callback invoked.";
}
// GaiaOAuthClient::Delegate OAuth2 error when fetching refresh token request.
void EnrollmentHandlerChromeOS::OnOAuthError() {
CHECK_EQ(STEP_ROBOT_AUTH_REFRESH, enrollment_step_);
// OnOAuthError is only called if the request is bad (malformed) or the
// response is bad (empty access token returned).
LOG(ERROR) << "OAuth protocol error while fetching API refresh token.";
ReportResult(
EnrollmentStatus::ForRobotRefreshFetchError(net::HTTP_BAD_REQUEST));
}
// GaiaOAuthClient::Delegate network error when fetching refresh token.
void EnrollmentHandlerChromeOS::OnNetworkError(int response_code) {
CHECK_EQ(STEP_ROBOT_AUTH_REFRESH, enrollment_step_);
LOG(ERROR) << "Network error while fetching API refresh token: "
<< response_code;
ReportResult(
EnrollmentStatus::ForRobotRefreshFetchError(response_code));
}
void EnrollmentHandlerChromeOS::StartLockDevice() {
DCHECK_EQ(STEP_LOCK_DEVICE, enrollment_step_);
// Since this method is also called directly.
weak_ptr_factory_.InvalidateWeakPtrs();
install_attributes_->LockDevice(
device_mode_, domain_, enrollment_config_.management_realm, device_id_,
base::Bind(&EnrollmentHandlerChromeOS::HandleLockDeviceResult,
weak_ptr_factory_.GetWeakPtr()));
}
void EnrollmentHandlerChromeOS::HandleDMTokenStoreResult(bool success) {
CHECK_EQ(STEP_STORE_TOKEN, enrollment_step_);
if (!success) {
ReportResult(
EnrollmentStatus::ForStatus(EnrollmentStatus::DM_TOKEN_STORE_FAILED));
return;
}
StartStoreRobotAuth();
}
void EnrollmentHandlerChromeOS::HandleLockDeviceResult(
chromeos::InstallAttributes::LockResult lock_result) {
DCHECK_EQ(STEP_LOCK_DEVICE, enrollment_step_);
switch (lock_result) {
case chromeos::InstallAttributes::LOCK_SUCCESS:
if (device_mode_ == DEVICE_MODE_ENTERPRISE_AD) {
StartStoreDMToken();
} else {
StartStoreRobotAuth();
}
break;
case chromeos::InstallAttributes::LOCK_NOT_READY:
// We wait up to |kLockRetryTimeoutMs| milliseconds and if it hasn't
// succeeded by then show an error to the user and stop the enrollment.
if (lockbox_init_duration_ < kLockRetryTimeoutMs) {
// InstallAttributes not ready yet, retry later.
LOG(WARNING) << "Install Attributes not ready yet will retry in "
<< kLockRetryIntervalMs << "ms.";
base::ThreadTaskRunnerHandle::Get()->PostDelayedTask(
FROM_HERE, base::Bind(&EnrollmentHandlerChromeOS::StartLockDevice,
weak_ptr_factory_.GetWeakPtr()),
base::TimeDelta::FromMilliseconds(kLockRetryIntervalMs));
lockbox_init_duration_ += kLockRetryIntervalMs;
} else {
HandleLockDeviceResult(chromeos::InstallAttributes::LOCK_TIMEOUT);
}
break;
case chromeos::InstallAttributes::LOCK_TIMEOUT:
case chromeos::InstallAttributes::LOCK_BACKEND_INVALID:
case chromeos::InstallAttributes::LOCK_ALREADY_LOCKED:
case chromeos::InstallAttributes::LOCK_SET_ERROR:
case chromeos::InstallAttributes::LOCK_FINALIZE_ERROR:
case chromeos::InstallAttributes::LOCK_READBACK_ERROR:
case chromeos::InstallAttributes::LOCK_WRONG_DOMAIN:
case chromeos::InstallAttributes::LOCK_WRONG_MODE:
ReportResult(EnrollmentStatus::ForLockError(lock_result));
break;
}
}
void EnrollmentHandlerChromeOS::StartStoreDMToken() {
DCHECK(device_mode_ == DEVICE_MODE_ENTERPRISE_AD);
SetStep(STEP_STORE_TOKEN);
dm_token_storage_ = base::MakeUnique<policy::DMTokenStorage>(
g_browser_process->local_state());
dm_token_storage_->StoreDMToken(
client_->dm_token(),
base::Bind(&EnrollmentHandlerChromeOS::HandleDMTokenStoreResult,
weak_ptr_factory_.GetWeakPtr()));
}
void EnrollmentHandlerChromeOS::StartStoreRobotAuth() {
SetStep(STEP_STORE_ROBOT_AUTH);
// Don't store the token if robot auth was skipped.
if (skip_robot_auth_) {
HandleStoreRobotAuthTokenResult(true);
return;
}
chromeos::DeviceOAuth2TokenServiceFactory::Get()->SetAndSaveRefreshToken(
robot_refresh_token_,
base::Bind(&EnrollmentHandlerChromeOS::HandleStoreRobotAuthTokenResult,
weak_ptr_factory_.GetWeakPtr()));
}
void EnrollmentHandlerChromeOS::HandleStoreRobotAuthTokenResult(bool result) {
DCHECK_EQ(STEP_STORE_ROBOT_AUTH, enrollment_step_);
if (!result) {
LOG(ERROR) << "Failed to store API refresh token.";
ReportResult(EnrollmentStatus::ForStatus(
EnrollmentStatus::ROBOT_REFRESH_STORE_FAILED));
return;
}
SetStep(STEP_STORE_POLICY);
if (device_mode_ == policy::DEVICE_MODE_ENTERPRISE_AD) {
CHECK(install_attributes_->IsActiveDirectoryManaged());
// Update device settings so that in case of Active Directory unsigned
// policy is accepted.
chromeos::DeviceSettingsService::Get()->SetDeviceMode(
install_attributes_->GetMode());
chromeos::DBusThreadManager::Get()
->GetAuthPolicyClient()
->RefreshDevicePolicy(base::Bind(
&EnrollmentHandlerChromeOS::HandleActiveDirectoryPolicyRefreshed,
weak_ptr_factory_.GetWeakPtr()));
} else {
store_->InstallInitialPolicy(*policy_);
}
}
void EnrollmentHandlerChromeOS::HandleActiveDirectoryPolicyRefreshed(
bool success) {
DCHECK_EQ(STEP_STORE_POLICY, enrollment_step_);
if (!success) {
LOG(ERROR) << "Failed to load Active Directory policy.";
ReportResult(EnrollmentStatus::ForStatus(
EnrollmentStatus::ACTIVE_DIRECTORY_POLICY_FETCH_FAILED));
return;
}
// After that, the enrollment flow continues in one of the OnStore* observers.
store_->Load();
}
void EnrollmentHandlerChromeOS::Stop() {
if (client_.get())
client_->RemoveObserver(this);
SetStep(STEP_FINISHED);
weak_ptr_factory_.InvalidateWeakPtrs();
completion_callback_.Reset();
}
void EnrollmentHandlerChromeOS::ReportResult(EnrollmentStatus status) {
EnrollmentCallback callback = completion_callback_;
Stop();
if (status.status() != EnrollmentStatus::SUCCESS) {
LOG(WARNING) << "Enrollment failed: " << status.status()
<< ", client: " << status.client_status()
<< ", validation: " << status.validation_status()
<< ", store: " << status.store_status()
<< ", lock: " << status.lock_status();
}
if (!callback.is_null())
callback.Run(status);
}
void EnrollmentHandlerChromeOS::SetStep(EnrollmentStep step) {
DCHECK_LE(enrollment_step_, step);
VLOG(1) << "Step: " << step;
enrollment_step_ = step;
}
} // namespace policy