Appropriate handling for Cellular First devices.

chrome/browser/chromeos/login/wizard_controller.cc

 // 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/login/wizard_controller.h"
 
 #include <signal.h>
 #include <stddef.h>
 #include <stdlib.h>
 #include <sys/types.h>
@@ skipping 166 matching lines @@
 }
 
 bool IsSharkRequisition() {
   policy::DeviceCloudPolicyManagerChromeOS* policy_manager =
       g_browser_process->platform_part()
           ->browser_policy_connector_chromeos()
           ->GetDeviceCloudPolicyManager();
   return policy_manager && policy_manager->IsSharkRequisition();
 }
 
+bool IsCellularFirstDevice() {

[xiyuan, 2017/02/28 20:26:41]

Maybe we should move this into chromeos_swithces.h/cc to share with other places
that check the switch directly.

[kumarniranjan, 2017/02/28 21:49:33]

Done.

+  return base::CommandLine::ForCurrentProcess()->HasSwitch(
+      chromeos::switches::kCellularFirst);
+}
+
 // Checks if a controller device ("Master") is detected during the bootstrapping
 // or shark/remora setup process.
 bool IsControllerDetected() {
   return g_browser_process->local_state()->GetBoolean(
       prefs::kOobeControllerDetected);
 }
 
 void SetControllerDetectedPref(bool value) {
   PrefService* prefs = g_browser_process->local_state();
   prefs->SetBoolean(prefs::kOobeControllerDetected, value);
@@ skipping 686 matching lines @@
   }
 }
 
 void WizardController::InitiateOOBEUpdate() {
   if (IsRemoraRequisition()) {
     VLOG(1) << "Skip OOBE Update for remora.";
     OnUpdateCompleted();
     return;
   }
 
+  // If this is a Cellular First device, instruct UpdateEngine to allow
+  // updates over cellular data connections.
+  if (IsCellularFirstDevice()) {
+    DBusThreadManager::Get()
+        ->GetUpdateEngineClient()
+        ->SetUpdateOverCellularPermission(
+            true, base::Bind(&WizardController::InitiateOOBEUpdateHelper,
+                             weak_factory_.GetWeakPtr()));
+  } else {
+    InitiateOOBEUpdateHelper();
+  }
+}
+
+void WizardController::InitiateOOBEUpdateHelper() {
   VLOG(1) << "InitiateOOBEUpdate";
   SetCurrentScreenSmooth(GetScreen(OobeScreen::SCREEN_OOBE_UPDATE), true);
   UpdateScreen::Get(this)->StartNetworkCheck();
 }
 
 void WizardController::StartTimezoneResolve() {
   geolocation_provider_.reset(new SimpleGeolocationProvider(
       g_browser_process->system_request_context(),
       SimpleGeolocationProvider::DefaultGeolocationProviderURL()));
   geolocation_provider_->RequestGeolocation(
@@ skipping 605 matching lines @@
             : policy::EnrollmentConfig::MODE_MANUAL_REENROLLMENT;
   }
 
   EnrollmentScreen* screen = EnrollmentScreen::Get(this);
   screen->SetParameters(effective_config, shark_controller_.get());
   SetStatusAreaVisible(true);
   SetCurrentScreen(screen);
 }
 
 }  // namespace chromeos