Implement Chrome OS device enrollment on the new cloud policy stack.

chrome/browser/policy/device_cloud_policy_manager_chromeos.h

 // 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.
 
 #ifndef CHROME_BROWSER_POLICY_DEVICE_CLOUD_POLICY_MANAGER_CHROMEOS_H_
 #define CHROME_BROWSER_POLICY_DEVICE_CLOUD_POLICY_MANAGER_CHROMEOS_H_
 
+#include <bitset>
 #include <string>
 
 #include "base/basictypes.h"
+#include "base/callback.h"
 #include "base/compiler_specific.h"
 #include "base/memory/scoped_ptr.h"
 #include "chrome/browser/policy/cloud_policy_manager.h"
 #include "chrome/browser/policy/cloud_policy_store.h"
+#include "chrome/browser/policy/enrollment_status_chromeos.h"
 
 class PrefService;
 
 namespace policy {
 
 class CloudPolicyClient;
 class DeviceCloudPolicyStoreChromeOS;
 class DeviceManagementService;
+class EnrollmentHandlerChromeOS;
 class EnterpriseInstallAttributes;
 
-// CloudPolicyManager specialization for device policy on Chrome OS.
+// CloudPolicyManager specialization for device policy on Chrome OS. The most
+// significant addition is support for device enrollment.
 class DeviceCloudPolicyManagerChromeOS : public CloudPolicyManager {
  public:
+  typedef std::bitset<32> AllowedDeviceModes;
+  typedef base::Callback<void(EnrollmentStatus)> EnrollmentCallback;
+
   DeviceCloudPolicyManagerChromeOS(
       scoped_ptr<DeviceCloudPolicyStoreChromeOS> store,
       EnterpriseInstallAttributes* install_attributes);
   virtual ~DeviceCloudPolicyManagerChromeOS();
 
   // Establishes the connection to the cloud, updating policy as necessary.
   void Connect(PrefService* local_state,
                DeviceManagementService* device_management_service);
 
+  // Starts enrollment or re-enrollment. Once the enrollment process completes,
+  // |callback| is invoked and gets passed the status of the operation.
+  // |allowed_modes| specifies acceptable DEVICE_MODE_* constants for
+  // enrollment.
+  void StartEnrollment(const std::string& auth_token,
+                       const AllowedDeviceModes& allowed_modes,
+                       const EnrollmentCallback& callback);
+
+  // Cancels a pending enrollment operation, if any.
+  void CancelEnrollment();
+
   // CloudPolicyStore::Observer:
   virtual void OnStoreLoaded(CloudPolicyStore* store) OVERRIDE;
 
   // Returns the device serial number, or an empty string if not available.
   static std::string GetMachineID();
 
   // Returns the machine model, or an empty string if not available.
   static std::string GetMachineModel();
 
  private:
   // Creates a new CloudPolicyClient.
   scoped_ptr<CloudPolicyClient> CreateClient();
 
   // Starts policy refreshes if |store_| indicates a managed device and the
   // necessary dependencies have been provided via Initialize().
   void StartIfManaged();
 
+  // Handles completion signaled by |enrollment_handler_|.
+  void EnrollmentCompleted(const EnrollmentCallback& callback,
+                           EnrollmentStatus status);
+
   // Points to the same object as the base CloudPolicyManager::store(), but with
   // actual device policy specific type.
   DeviceCloudPolicyStoreChromeOS* device_store_;
   EnterpriseInstallAttributes* install_attributes_;
 
   DeviceManagementService* device_management_service_;
 
   // PrefService instance to read the policy refresh rate from.
   PrefService* local_state_;
 
+  // Non-null if there is an enrollment operation pending.
+  scoped_ptr<EnrollmentHandlerChromeOS> enrollment_handler_;
+
   DISALLOW_COPY_AND_ASSIGN(DeviceCloudPolicyManagerChromeOS);
 };
 
 }  // namespace policy
 
 #endif  // CHROME_BROWSER_POLICY_DEVICE_CLOUD_POLICY_MANAGER_CHROMEOS_H_