Home | History | Annotate | Download | only in enrollment 
      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 #ifndef CHROME_BROWSER_CHROMEOS_LOGIN_ENROLLMENT_ENROLLMENT_SCREEN_H_
      6 #define CHROME_BROWSER_CHROMEOS_LOGIN_ENROLLMENT_ENROLLMENT_SCREEN_H_
      7 
      8 #include <string>
      9 
     10 #include "base/basictypes.h"
     11 #include "base/compiler_specific.h"
     12 #include "base/gtest_prod_util.h"
     13 #include "base/memory/scoped_ptr.h"
     14 #include "base/memory/weak_ptr.h"
     15 #include "chrome/browser/chromeos/login/enrollment/enrollment_screen_actor.h"
     16 #include "chrome/browser/chromeos/login/screens/wizard_screen.h"
     17 #include "components/pairing/host_pairing_controller.h"
     18 #include "components/policy/core/common/cloud/cloud_policy_constants.h"
     19 #include "components/policy/core/common/cloud/enterprise_metrics.h"
     20 
     21 namespace base {
     22 class ElapsedTimer;
     23 }
     24 
     25 namespace pairing_chromeos {
     26 class ControllerPairingController;
     27 }
     28 
     29 namespace chromeos {
     30 
     31 class ScreenManager;
     32 class ScreenObserver;
     33 
     34 // The screen implementation that links the enterprise enrollment UI into the
     35 // OOBE wizard.
     36 class EnrollmentScreen
     37     : public WizardScreen,
     38       public pairing_chromeos::HostPairingController::Observer,
     39       public EnrollmentScreenActor::Controller {
     40  public:
     41   typedef pairing_chromeos::HostPairingController::Stage Stage;
     42 
     43   EnrollmentScreen(ScreenObserver* observer,
     44                    EnrollmentScreenActor* actor);
     45   virtual ~EnrollmentScreen();
     46 
     47   static EnrollmentScreen* Get(ScreenManager* manager);
     48 
     49   // Setup how this screen will handle enrollment.
     50   //   |auth_token| is an optional OAuth token to attempt to enroll with.
     51   //   |shark_controller| is an interface that is used to communicate with a
     52   //     remora device for remote enrollment.
     53   //   |remora_controller| is an interface that is used to communicate with a
     54   //     shark device for remote enrollment.
     55   void SetParameters(
     56       EnrollmentScreenActor::EnrollmentMode enrollment_mode,
     57       const std::string& management_domain,
     58       const std::string& enrollment_user,
     59       const std::string& auth_token,
     60       pairing_chromeos::ControllerPairingController* shark_controller,
     61       pairing_chromeos::HostPairingController* remora_controller);
     62 
     63   // WizardScreen implementation:
     64   virtual void PrepareToShow() OVERRIDE;
     65   virtual void Show() OVERRIDE;
     66   virtual void Hide() OVERRIDE;
     67   virtual std::string GetName() const OVERRIDE;
     68 
     69   // pairing_chromeos::HostPairingController::Observer:
     70   virtual void PairingStageChanged(Stage new_stage) OVERRIDE;
     71   virtual void ConfigureHost(bool accepted_eula,
     72                              const std::string& lang,
     73                              const std::string& timezone,
     74                              bool send_reports,
     75                              const std::string& keyboard_layout) OVERRIDE;
     76   virtual void EnrollHost(const std::string& auth_token) OVERRIDE;
     77 
     78   // EnrollmentScreenActor::Controller implementation:
     79   virtual void OnLoginDone(const std::string& user) OVERRIDE;
     80   virtual void OnAuthError(const GoogleServiceAuthError& error) OVERRIDE;
     81   virtual void OnOAuthTokenAvailable(const std::string& oauth_token) OVERRIDE;
     82   virtual void OnRetry() OVERRIDE;
     83   virtual void OnCancel() OVERRIDE;
     84   virtual void OnConfirmationClosed() OVERRIDE;
     85 
     86   // Used for testing.
     87   EnrollmentScreenActor* GetActor() {
     88     return actor_;
     89   }
     90 
     91  private:
     92   FRIEND_TEST_ALL_PREFIXES(EnrollmentScreenTest, TestSuccess);
     93 
     94   // Starts the Lockbox storage process.
     95   void WriteInstallAttributesData();
     96 
     97   // Kicks off the policy infrastructure to register with the service.
     98   void RegisterForDevicePolicy(const std::string& token);
     99 
    100   // Sends an enrollment access token to a remote device.
    101   void SendEnrollmentAuthToken(const std::string& token);
    102 
    103   // Handles enrollment completion. Logs a UMA sample and requests the actor to
    104   // show the specified enrollment status.
    105   void ReportEnrollmentStatus(policy::EnrollmentStatus status);
    106 
    107   // Shows successful enrollment status after all enrollment related file
    108   // operations are completed.
    109   void ShowEnrollmentStatusOnSuccess(const policy::EnrollmentStatus& status);
    110 
    111   // Logs an UMA event in the kMetricEnrollment or the kMetricEnrollmentRecovery
    112   // histogram, depending on |enrollment_mode_|.
    113   void UMA(policy::MetricEnrollment sample);
    114 
    115   // Logs an UMA event in the kMetricEnrollment or the kMetricEnrollmentRecovery
    116   // histogram, depending on |enrollment_mode_|.  If auto-enrollment is on,
    117   // |sample| is ignored and a kMetricEnrollmentAutoFailed sample is logged
    118   // instead.
    119   void UMAFailure(policy::MetricEnrollment sample);
    120 
    121   // Shows the signin screen. Used as a callback to run after auth reset.
    122   void ShowSigninScreen();
    123 
    124   // Convenience helper to check for auto enrollment mode.
    125   bool is_auto_enrollment() const {
    126     return enrollment_mode_ == EnrollmentScreenActor::ENROLLMENT_MODE_AUTO;
    127   }
    128 
    129   pairing_chromeos::ControllerPairingController* shark_controller_;
    130   pairing_chromeos::HostPairingController* remora_controller_;
    131   EnrollmentScreenActor* actor_;
    132   EnrollmentScreenActor::EnrollmentMode enrollment_mode_;
    133   bool enrollment_failed_once_;
    134   bool remora_token_sent_;
    135   std::string user_;
    136   std::string auth_token_;
    137   int lockbox_init_duration_;
    138   scoped_ptr<base::ElapsedTimer> elapsed_timer_;
    139   base::WeakPtrFactory<EnrollmentScreen> weak_ptr_factory_;
    140 
    141   DISALLOW_COPY_AND_ASSIGN(EnrollmentScreen);
    142 };
    143 
    144 }  // namespace chromeos
    145 
    146 #endif  // CHROME_BROWSER_CHROMEOS_LOGIN_ENROLLMENT_ENROLLMENT_SCREEN_H_
    147