1) Add initialization error page; 2) Add connection error page.

chrome/browser/chromeos/login/screens/host_pairing_screen.cc

 // Copyright 2014 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/login/screens/host_pairing_screen.h"
 
 #include "base/command_line.h"
+#include "base/prefs/pref_service.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chromeos/login/startup_utils.h"
 #include "chrome/browser/chromeos/login/wizard_controller.h"
 #include "chrome/browser/chromeos/policy/browser_policy_connector_chromeos.h"
 #include "chrome/browser/chromeos/policy/enrollment_status_chromeos.h"
+#include "chrome/common/pref_names.h"
 #include "components/pairing/host_pairing_controller.h"
 
 namespace chromeos {
 
 using namespace host_pairing;
 using namespace pairing_chromeos;
 
+namespace {
+
+// Clear Enrollment related local state after a successful enrollment or when
+// an error occurs.
+void ClearEnrollmentLocalState() {
+  g_browser_process->local_state()->ClearPref(
+      prefs::kDeviceEnrollmentAutoStart);
+  g_browser_process->local_state()->ClearPref(prefs::kDeviceEnrollmentCanExit);
+}
+
+}  // namespace
+
 HostPairingScreen::HostPairingScreen(
     BaseScreenDelegate* base_screen_delegate,
     Delegate* delegate,
     HostPairingScreenActor* actor,
     pairing_chromeos::HostPairingController* remora_controller)
     : BaseScreen(base_screen_delegate),
       delegate_(delegate),
       actor_(actor),
       remora_controller_(remora_controller),
       current_stage_(HostPairingController::STAGE_NONE),
@@ skipping 31 matching lines @@
     actor_->Hide();
 }
 
 std::string HostPairingScreen::GetName() const {
   return WizardController::kHostPairingScreenName;
 }
 
 void HostPairingScreen::PairingStageChanged(Stage new_stage) {
   std::string desired_page;
   switch (new_stage) {
+    case HostPairingController::STAGE_INITIALIZATION_ERROR: {
+      desired_page = kPageIntializationError;
+      ClearEnrollmentLocalState();
+      break;
+    }
     case HostPairingController::STAGE_WAITING_FOR_CONTROLLER:
     case HostPairingController::STAGE_WAITING_FOR_CONTROLLER_AFTER_UPDATE: {
       desired_page = kPageWelcome;
       break;
     }
     case HostPairingController::STAGE_WAITING_FOR_CODE_CONFIRMATION: {
       desired_page = kPageCodeConfirmation;
       context_.SetString(kContextKeyConfirmationCode,
                          remora_controller_->GetConfirmationCode());
       break;
     }
+    case HostPairingController::STAGE_CONTROLLER_CONNECTION_ERROR: {
+      desired_page = kPageConnectionError;
+      ClearEnrollmentLocalState();
+      break;
+    }
     case HostPairingController::STAGE_SETUP_BASIC_CONFIGURATION: {
       desired_page = kPageSetupBasicConfiguration;
       break;
     }
     case HostPairingController::STAGE_SETUP_NETWORK_ERROR: {
       desired_page = kPageSetupNetworkError;
+      ClearEnrollmentLocalState();
       break;
     }
     case HostPairingController::STAGE_WAITING_FOR_CREDENTIALS: {
       desired_page = kPageEnrollmentIntroduction;
       break;
     }
     case HostPairingController::STAGE_ENROLLING: {
       desired_page = kPageEnrollment;
       context_.SetString(kContextKeyEnrollmentDomain,
                          remora_controller_->GetEnrollmentDomain());
       break;
     }
     case HostPairingController::STAGE_ENROLLMENT_SUCCESS: {
       remora_controller_->RemoveObserver(this);
       Finish(WizardController::ENTERPRISE_ENROLLMENT_COMPLETED);
+      ClearEnrollmentLocalState();
       break;
     }
     case HostPairingController::STAGE_ENROLLMENT_ERROR: {
       // TODO(xdai): Maybe return to the Network Setup page?
       remora_controller_->RemoveObserver(this);
       desired_page = kPageEnrollmentError;
+      ClearEnrollmentLocalState();
       break;
     }
     default:
       break;
   }
   current_stage_ = new_stage;
   context_.SetString(kContextKeyDeviceName,
                      remora_controller_->GetDeviceName());
   context_.SetString(kContextKeyPage, desired_page);
   CommitContextChanges();
@@ skipping 74 matching lines @@
 }
 
 void HostPairingScreen::OnAnyEnrollmentError() {
   enrollment_helper_->ClearAuth(base::Bind(&HostPairingScreen::OnAuthCleared,
                                            weak_ptr_factory_.GetWeakPtr()));
   remora_controller_->OnEnrollmentStatusChanged(
       HostPairingController::ENROLLMENT_STATUS_FAILURE);
 }
 
 }  // namespace chromeos