Add device policy to disallow shutdown - ash UI modifications

chrome/browser/chromeos/power/session_state_controller_delegate_chromeos.cc

Index: chrome/browser/chromeos/power/session_state_controller_delegate_chromeos.cc
diff --git a/chrome/browser/chromeos/power/session_state_controller_delegate_chromeos.cc b/chrome/browser/chromeos/power/session_state_controller_delegate_chromeos.cc
index a4cc4bde7812cbe2939eded1be37488b5f73dc17..299db3fb7643217031da5d75258cf6f373b3b580 100644
--- a/chrome/browser/chromeos/power/session_state_controller_delegate_chromeos.cc
+++ b/chrome/browser/chromeos/power/session_state_controller_delegate_chromeos.cc
@@ -5,24 +5,46 @@
 #include "chrome/browser/chromeos/power/session_state_controller_delegate_chromeos.h"
 
 #include "base/logging.h"
+#include "chrome/browser/chromeos/settings/cros_settings.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "chromeos/dbus/power_manager_client.h"
 #include "chromeos/dbus/session_manager_client.h"
+#include "chromeos/settings/cros_settings_names.h"
+#include "chromeos/settings/cros_settings_provider.h"
 
 namespace chromeos {
 
+SessionStateControllerDelegateChromeos::SessionStateControllerDelegateChromeos()
+    : weak_factory_(this) {
+}
+
+SessionStateControllerDelegateChromeos::
+    ~SessionStateControllerDelegateChromeos() {
+}
+
 void SessionStateControllerDelegateChromeos::RequestLockScreen() {
   // TODO(antrim) : additional logging for crbug/173178
   LOG(WARNING) << "Requesting screen lock from SessionStateControllerDelegate";
   DBusThreadManager::Get()->GetSessionManagerClient()->RequestLockScreen();
 }
 
-void SessionStateControllerDelegateChromeos::RequestRestart() {
-  DBusThreadManager::Get()->GetPowerManagerClient()->RequestRestart();
-}
-
 void SessionStateControllerDelegateChromeos::RequestShutdown() {
-  DBusThreadManager::Get()->GetPowerManagerClient()->RequestShutdown();
+  CrosSettings* cros_settings = CrosSettings::Get();
+  CrosSettingsProvider::TrustedStatus status =
+      cros_settings->PrepareTrustedValues(base::Bind(
+          &SessionStateControllerDelegateChromeos::RequestShutdown,
+          weak_factory_.GetWeakPtr()));
+  if (status != CrosSettingsProvider::TRUSTED)
+    return;
+
+  // Get the updated policy.
+  bool reboot_on_shutdown = false;
+  cros_settings->GetBoolean(kRebootOnShutdown, &reboot_on_shutdown);
+
+  if (reboot_on_shutdown)
+    DBusThreadManager::Get()->GetPowerManagerClient()->RequestRestart();
+  else
+    DBusThreadManager::Get()->GetPowerManagerClient()->RequestShutdown();
 }
 
 }  // namespace chromeos