Add DeviceDisablingManager to manage device disabling

chrome/browser/chromeos/system/device_disabling_manager.cc

Index: chrome/browser/chromeos/system/device_disabling_manager.cc
diff --git a/chrome/browser/chromeos/system/device_disabling_manager.cc b/chrome/browser/chromeos/system/device_disabling_manager.cc
new file mode 100644
index 0000000000000000000000000000000000000000..045222afd18f6d86a09bc469ebeaf25e81d9a000
--- /dev/null
+++ b/chrome/browser/chromeos/system/device_disabling_manager.cc
@@ -0,0 +1,77 @@
+// 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/system/device_disabling_manager.h"
+
+#include "base/bind.h"
+#include "base/command_line.h"
+#include "base/prefs/pref_service.h"
+#include "base/values.h"
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/chromeos/policy/browser_policy_connector_chromeos.h"
+#include "chrome/browser/chromeos/policy/enterprise_install_attributes.h"
+#include "chrome/browser/chromeos/policy/server_backed_device_state.h"
+#include "chrome/common/pref_names.h"
+#include "chromeos/chromeos_switches.h"
+#include "components/policy/core/common/cloud/cloud_policy_constants.h"
+
+namespace chromeos {
+namespace system {
+
+DeviceDisablingManager::DeviceDisablingManager(
+    policy::BrowserPolicyConnectorChromeOS* browser_policy_connector)
+    : browser_policy_connector_(browser_policy_connector),
+      weak_factory_(this) {
+}
+
+const std::string& DeviceDisablingManager::GetDisabledMessage() const {
+  return disabled_message_;
+}
+
+void DeviceDisablingManager::CheckWhetherDeviceDisabledDuringOOBE(
+    const DeviceDisabledCheckCallback& callback) {
+  if (policy::GetRestoreMode() != policy::RESTORE_MODE_DISABLED ||
+      CommandLine::ForCurrentProcess()->HasSwitch(
+          switches::kDisableDeviceDisabling)) {
+    // Indicate that the device is not disabled if it is not marked as such in
+    // local state or device disabling has been turned off by flag.
+    callback.Run(false);
+    return;
+  }
+
+  if (browser_policy_connector_->GetDeviceMode() ==
+          policy::DEVICE_MODE_PENDING) {
+    // If the device mode is not known yet, request to be called back once it
+    // becomes known.
+    browser_policy_connector_->GetInstallAttributes()->ReadImmutableAttributes(
+        base::Bind(
+            &DeviceDisablingManager::CheckWhetherDeviceDisabledDuringOOBE,
+            weak_factory_.GetWeakPtr(),
+            callback));
+    return;
+  }
+
+  if (browser_policy_connector_->GetDeviceMode() !=
+          policy::DEVICE_MODE_NOT_SET) {
+    // If the device is owned already, this method must have been called after
+    // OOBE, which is an error. Indicate that the device is not disabled to
+    // prevent spurious disabling. Actual device disabling after OOBE will be
+    // handled elsewhere, by checking for disabled state in cros settings.
+    callback.Run(false);

[achuithb, 2014/11/05 19:59:54]

Should there be a NOTREACHED here? Does this indicate a bug in the code?

[bartfab (slow), 2014/11/06 16:14:38]

Yes, it does indicate a bug in the code. However, I am exercising this code path
in a test (to make sure that the error condition is handled correctly), so I
cannot do NOTREACHED() as it would crash the test. I added a LOG(ERROR).

+    return;
+  }
+
+  // Cache the disabled message.
+  disabled_message_.clear();
+  g_browser_process->local_state()->GetDictionary(
+      prefs::kServerBackedDeviceState)->GetString(
+          policy::kDeviceStateDisabledMessage,
+          &disabled_message_);
+
+  // Indicate that the device is disabled.
+  callback.Run(true);
+}
+
+}  // namespace system
+}  // namespace chromeos