Add device policy to disallow shutdown

ash/wm/lock_state_controller.cc

 // Copyright 2013 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 "ash/wm/lock_state_controller.h"
 
 #include <algorithm>
 #include <string>
 
 #include "ash/accessibility_delegate.h"
 #include "ash/ash_switches.h"
 #include "ash/cancel_mode.h"
 #include "ash/metrics/user_metrics_recorder.h"
 #include "ash/shell.h"
 #include "ash/shell_delegate.h"
 #include "ash/shell_window_ids.h"
 #include "ash/wm/session_state_animator.h"
 #include "ash/wm/session_state_animator_impl.h"
+#include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/command_line.h"
+#include "base/location.h"
+#include "base/logging.h"
 #include "base/strings/string_util.h"
 #include "base/timer/timer.h"
 #include "ui/aura/window_tree_host.h"
 #include "ui/views/controls/menu/menu_controller.h"
 #include "ui/wm/core/compound_event_filter.h"
 
 #if defined(OS_CHROMEOS)
 #include "base/sys_info.h"
 #include "media/audio/sounds/sounds_manager.h"
 #endif
@@ skipping 126 matching lines @@
 }
 
 void LockStateController::OnStartingLock() {
   if (shutting_down_ || system_is_locked_)
     return;
   if (animating_lock_)
     return;
   StartImmediatePreLockAnimation(false /* request_lock_on_completion */);
 }
 
-void LockStateController::RequestShutdown() {
+void LockStateController::RequestShutdown(ShutdownMode mode) {
   if (shutting_down_)
     return;
 
   shutting_down_ = true;
 
   Shell* shell = ash::Shell::GetInstance();
   shell->cursor_manager()->HideCursor();
   shell->cursor_manager()->LockCursor();
 
   animator_->StartAnimation(
       SessionStateAnimator::ROOT_CONTAINER,
       SessionStateAnimator::ANIMATION_GRAYSCALE_BRIGHTNESS,
       SessionStateAnimator::ANIMATION_SPEED_SHUTDOWN);
-  StartRealShutdownTimer(true);
+  StartRealShutdownTimer(true, mode);
 }
 
 void LockStateController::OnLockScreenHide(
   base::Callback<void(void)>& callback) {
   StartUnlockAnimationBeforeUIDestroyed(callback);
 }
 
 void LockStateController::SetLockScreenDisplayedCallback(
     const base::Closure& callback) {
   lock_screen_displayed_callback_ = callback;
@@ skipping 70 matching lines @@
       &LockStateController::OnPreShutdownAnimationTimeout);
 }
 
 void LockStateController::OnPreShutdownAnimationTimeout() {
   VLOG(1) << "OnPreShutdownAnimationTimeout";
   shutting_down_ = true;
 
   Shell* shell = ash::Shell::GetInstance();
   shell->cursor_manager()->HideCursor();
 
-  StartRealShutdownTimer(false);
+  StartRealShutdownTimer(false, POWER_OFF);
 }
 
-void LockStateController::StartRealShutdownTimer(bool with_animation_time) {
+void LockStateController::StartRealShutdownTimer(bool with_animation_time,
+                                                 ShutdownMode shutdown_mode) {
   base::TimeDelta duration =
       base::TimeDelta::FromMilliseconds(kShutdownRequestDelayMs);
   if (with_animation_time) {
     duration +=
         animator_->GetDuration(SessionStateAnimator::ANIMATION_SPEED_SHUTDOWN);
   }
 
 #if defined(OS_CHROMEOS)
   const AccessibilityDelegate* const delegate =
       Shell::GetInstance()->accessibility_delegate();
   base::TimeDelta sound_duration = delegate->PlayShutdownSound();
   sound_duration =
       std::min(sound_duration,
                base::TimeDelta::FromMilliseconds(kMaxShutdownSoundDurationMs));
   duration = std::max(duration, sound_duration);
 #endif
 
   real_shutdown_timer_.Start(
-      FROM_HERE, duration, this, &LockStateController::OnRealShutdownTimeout);
+      FROM_HERE, duration, base::Bind(&LockStateController::OnRealPowerTimeout,
+                                      base::Unretained(this), shutdown_mode));
 }
 
-void LockStateController::OnRealShutdownTimeout() {
-  VLOG(1) << "OnRealShutdownTimeout";

[bartfab (slow), 2014/12/08 16:49:44]

Why did you remove this?

[cschuet (SLOW), 2014/12/08 18:57:04]

Removed by accident. Restored.

+void LockStateController::OnRealPowerTimeout(ShutdownMode shutdown_mode) {
   DCHECK(shutting_down_);
 #if defined(OS_CHROMEOS)
   if (!base::SysInfo::IsRunningOnChromeOS()) {
     ShellDelegate* delegate = Shell::GetInstance()->delegate();
     if (delegate) {
       delegate->Exit();
       return;
     }
   }
 #endif
-  Shell::GetInstance()->metrics()->RecordUserMetricsAction(
-      UMA_ACCEL_SHUT_DOWN_POWER_BUTTON);
-  delegate_->RequestShutdown();
+  if (shutdown_mode == POWER_OFF) {
+    Shell::GetInstance()->metrics()->RecordUserMetricsAction(
+        UMA_ACCEL_SHUT_DOWN_POWER_BUTTON);
+    delegate_->RequestShutdown();

[bartfab (slow), 2014/12/08 16:49:44]

Nit: If you return here, you can save yourself the else {}.

[cschuet (SLOW), 2014/12/08 18:57:04]

Done.

+  } else {
+    Shell::GetInstance()->metrics()->RecordUserMetricsAction(
+        UMA_ACCEL_RESTART_POWER_BUTTON);
+    delegate_->RequestRestart();
+  }
 }
 
 void LockStateController::StartCancellableShutdownAnimation() {
   Shell* shell = ash::Shell::GetInstance();
   // Hide cursor, but let it reappear if the mouse moves.
   shell->cursor_manager()->HideCursor();
 
   animator_->StartAnimation(
       SessionStateAnimator::ROOT_CONTAINER,
       SessionStateAnimator::ANIMATION_GRAYSCALE_BRIGHTNESS,
@@ skipping 243 matching lines @@
   if (unlocked_properties_.get() &&
       unlocked_properties_->background_is_hidden) {
     animation_sequence->StartAnimation(
         SessionStateAnimator::DESKTOP_BACKGROUND,
         SessionStateAnimator::ANIMATION_FADE_OUT,
         speed);
   }
 }
 
 }  // namespace ash