upgrade_detector.cc revision 5821806d5e7f356e8fa4b058a389a808ea183019
1// Copyright (c) 2012 The Chromium Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style license that can be 3// found in the LICENSE file. 4 5#include "chrome/browser/upgrade_detector.h" 6 7#include "base/bind.h" 8#include "base/command_line.h" 9#include "chrome/browser/prefs/pref_service.h" 10#include "chrome/browser/lifetime/application_lifetime.h" 11#include "chrome/browser/ui/browser_otr_state.h" 12#include "chrome/common/chrome_notification_types.h" 13#include "chrome/common/chrome_switches.h" 14#include "chrome/common/pref_names.h" 15#include "content/public/browser/notification_service.h" 16#include "grit/theme_resources.h" 17 18// How long to wait between checks for whether the user has been idle. 19static const int kIdleRepeatingTimerWait = 10; // Minutes (seconds if testing). 20 21// How much idle time (since last input even was detected) must have passed 22// until we notify that a critical update has occurred. 23static const int kIdleAmount = 2; // Hours (or seconds, if testing). 24 25bool UseTestingIntervals() { 26 // If a command line parameter specifying how long the upgrade check should 27 // be, we assume it is for testing and switch to using seconds instead of 28 // hours. 29 const CommandLine& cmd_line = *CommandLine::ForCurrentProcess(); 30 return !cmd_line.GetSwitchValueASCII( 31 switches::kCheckForUpdateIntervalSec).empty(); 32} 33 34// static 35void UpgradeDetector::RegisterPrefs(PrefService* prefs) { 36 prefs->RegisterBooleanPref(prefs::kRestartLastSessionOnShutdown, false); 37 prefs->RegisterBooleanPref(prefs::kWasRestarted, false); 38} 39 40int UpgradeDetector::GetIconResourceID(UpgradeNotificationIconType type) { 41 bool badge = type == UPGRADE_ICON_TYPE_BADGE; 42 switch (upgrade_notification_stage_) { 43 case UPGRADE_ANNOYANCE_CRITICAL: 44 // The critical annoyance state, somewhat ironically, re-purposes the 45 // icon for the second highest severity state, since that state has the 46 // icon most closely resembling the one requested of this feature and the 47 // critical annoyance is never part of the sliding scale of severity 48 // anyway (always shown on its own). 49 return badge ? IDR_UPDATE_BADGE3 : IDR_UPDATE_MENU3; 50 case UPGRADE_ANNOYANCE_SEVERE: 51 return badge ? IDR_UPDATE_BADGE4 : IDR_UPDATE_MENU4; 52 case UPGRADE_ANNOYANCE_HIGH: 53 return badge ? IDR_UPDATE_BADGE3 : IDR_UPDATE_MENU3; 54 case UPGRADE_ANNOYANCE_ELEVATED: 55 return badge ? IDR_UPDATE_BADGE2 : IDR_UPDATE_MENU2; 56 case UPGRADE_ANNOYANCE_LOW: 57 return badge ? IDR_UPDATE_BADGE : IDR_UPDATE_MENU; 58 default: 59 return 0; 60 } 61} 62 63UpgradeDetector::UpgradeDetector() 64 : is_critical_upgrade_(false), 65 critical_update_acknowledged_(false), 66 upgrade_notification_stage_(UPGRADE_ANNOYANCE_NONE), 67 notify_upgrade_(false) { 68} 69 70UpgradeDetector::~UpgradeDetector() { 71} 72 73void UpgradeDetector::NotifyUpgradeDetected() { 74 upgrade_detected_time_ = base::Time::Now(); 75 critical_update_acknowledged_ = false; 76} 77 78void UpgradeDetector::NotifyUpgradeRecommended() { 79 notify_upgrade_ = true; 80 81 content::NotificationService::current()->Notify( 82 chrome::NOTIFICATION_UPGRADE_RECOMMENDED, 83 content::Source<UpgradeDetector>(this), 84 content::NotificationService::NoDetails()); 85 86 if (is_critical_upgrade_) { 87 int idle_timer = UseTestingIntervals() ? 88 kIdleRepeatingTimerWait : 89 kIdleRepeatingTimerWait * 60; // To minutes. 90 idle_check_timer_.Start(FROM_HERE, 91 base::TimeDelta::FromSeconds(idle_timer), 92 this, &UpgradeDetector::CheckIdle); 93 } 94} 95 96void UpgradeDetector::CheckIdle() { 97 // CalculateIdleState expects an interval in seconds. 98 int idle_time_allowed = UseTestingIntervals() ? kIdleAmount : 99 kIdleAmount * 60 * 60; 100 101 CalculateIdleState( 102 idle_time_allowed, base::Bind(&UpgradeDetector::IdleCallback, 103 base::Unretained(this))); 104} 105 106void UpgradeDetector::IdleCallback(IdleState state) { 107 // Don't proceed while an incognito window is open. The timer will still 108 // keep firing, so this function will get a chance to re-evaluate this. 109 if (chrome::IsOffTheRecordSessionActive()) 110 return; 111 112 switch (state) { 113 case IDLE_STATE_LOCKED: 114 // Computer is locked, auto-restart. 115 idle_check_timer_.Stop(); 116 browser::AttemptRestart(); 117 break; 118 case IDLE_STATE_IDLE: 119 // Computer has been idle for long enough, show warning. 120 idle_check_timer_.Stop(); 121 content::NotificationService::current()->Notify( 122 chrome::NOTIFICATION_CRITICAL_UPGRADE_INSTALLED, 123 content::Source<UpgradeDetector>(this), 124 content::NotificationService::NoDetails()); 125 break; 126 case IDLE_STATE_ACTIVE: 127 case IDLE_STATE_UNKNOWN: 128 break; 129 default: 130 NOTREACHED(); // Need to add any new value above (either providing 131 // automatic restart or show notification to user). 132 break; 133 } 134} 135