| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "device/power_save_blocker/power_save_blocker.h" | 5 #include "device/power_save_blocker/power_save_blocker.h" |
| 6 | 6 |
| 7 #include <windows.h> | 7 #include <windows.h> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/logging.h" | 10 #include "base/logging.h" |
| 11 #include "base/macros.h" | 11 #include "base/macros.h" |
| 12 #include "base/sequenced_task_runner.h" | 12 #include "base/sequenced_task_runner.h" |
| 13 #include "base/strings/utf_string_conversions.h" | 13 #include "base/strings/utf_string_conversions.h" |
| 14 #include "base/win/scoped_handle.h" | 14 #include "base/win/scoped_handle.h" |
| 15 #include "base/win/windows_version.h" | 15 #include "base/win/windows_version.h" |
| 16 | 16 |
| 17 namespace device { | 17 namespace device { |
| 18 namespace { | 18 namespace { |
| 19 | 19 |
| 20 int g_blocker_count[2]; | |
| 21 | |
| 22 HANDLE CreatePowerRequest(POWER_REQUEST_TYPE type, | 20 HANDLE CreatePowerRequest(POWER_REQUEST_TYPE type, |
| 23 const std::string& description) { | 21 const std::string& description) { |
| 24 if (type == PowerRequestExecutionRequired && | 22 if (type == PowerRequestExecutionRequired && |
| 25 base::win::GetVersion() < base::win::VERSION_WIN8) { | 23 base::win::GetVersion() == base::win::VERSION_WIN7) { |
| 26 return INVALID_HANDLE_VALUE; | 24 return INVALID_HANDLE_VALUE; |
| 27 } | 25 } |
| 28 | 26 |
| 29 base::string16 wide_description = base::ASCIIToUTF16(description); | 27 base::string16 wide_description = base::ASCIIToUTF16(description); |
| 30 REASON_CONTEXT context = {0}; | 28 REASON_CONTEXT context = {0}; |
| 31 context.Version = POWER_REQUEST_CONTEXT_VERSION; | 29 context.Version = POWER_REQUEST_CONTEXT_VERSION; |
| 32 context.Flags = POWER_REQUEST_CONTEXT_SIMPLE_STRING; | 30 context.Flags = POWER_REQUEST_CONTEXT_SIMPLE_STRING; |
| 33 context.Reason.SimpleReasonString = | 31 context.Reason.SimpleReasonString = |
| 34 const_cast<wchar_t*>(wide_description.c_str()); | 32 const_cast<wchar_t*>(wide_description.c_str()); |
| 35 | 33 |
| 36 base::win::ScopedHandle handle(::PowerCreateRequest(&context)); | 34 base::win::ScopedHandle handle(::PowerCreateRequest(&context)); |
| 37 if (!handle.IsValid()) | 35 if (!handle.IsValid()) |
| 38 return INVALID_HANDLE_VALUE; | 36 return INVALID_HANDLE_VALUE; |
| 39 | 37 |
| 40 if (::PowerSetRequest(handle.Get(), type)) | 38 if (::PowerSetRequest(handle.Get(), type)) |
| 41 return handle.Take(); | 39 return handle.Take(); |
| 42 | 40 |
| 43 // Something went wrong. | 41 // Something went wrong. |
| 44 return INVALID_HANDLE_VALUE; | 42 return INVALID_HANDLE_VALUE; |
| 45 } | 43 } |
| 46 | 44 |
| 47 // Takes ownership of the |handle|. | 45 // Takes ownership of the |handle|. |
| 48 void DeletePowerRequest(POWER_REQUEST_TYPE type, HANDLE handle) { | 46 void DeletePowerRequest(POWER_REQUEST_TYPE type, HANDLE handle) { |
| 49 base::win::ScopedHandle request_handle(handle); | 47 base::win::ScopedHandle request_handle(handle); |
| 50 if (!request_handle.IsValid()) | 48 if (!request_handle.IsValid()) |
| 51 return; | 49 return; |
| 52 | 50 |
| 53 if (type == PowerRequestExecutionRequired && | 51 if (type == PowerRequestExecutionRequired && |
| 54 base::win::GetVersion() < base::win::VERSION_WIN8) { | 52 base::win::GetVersion() == base::win::VERSION_WIN7) { |
| 55 return; | 53 return; |
| 56 } | 54 } |
| 57 | 55 |
| 58 BOOL success = ::PowerClearRequest(request_handle.Get(), type); | 56 BOOL success = ::PowerClearRequest(request_handle.Get(), type); |
| 59 DCHECK(success); | 57 DCHECK(success); |
| 60 } | 58 } |
| 61 | 59 |
| 62 void ApplySimpleBlock(PowerSaveBlocker::PowerSaveBlockerType type, int delta) { | |
| 63 g_blocker_count[type] += delta; | |
| 64 DCHECK_GE(g_blocker_count[type], 0); | |
| 65 | |
| 66 if (g_blocker_count[type] > 1) | |
| 67 return; | |
| 68 | |
| 69 DWORD this_flag = 0; | |
| 70 if (type == PowerSaveBlocker::kPowerSaveBlockPreventAppSuspension) | |
| 71 this_flag |= ES_SYSTEM_REQUIRED; | |
| 72 else | |
| 73 this_flag |= ES_DISPLAY_REQUIRED; | |
| 74 | |
| 75 DCHECK(this_flag); | |
| 76 | |
| 77 static DWORD flags = ES_CONTINUOUS; | |
| 78 if (!g_blocker_count[type]) | |
| 79 flags &= ~this_flag; | |
| 80 else | |
| 81 flags |= this_flag; | |
| 82 | |
| 83 SetThreadExecutionState(flags); | |
| 84 } | |
| 85 | |
| 86 } // namespace | 60 } // namespace |
| 87 | 61 |
| 88 class PowerSaveBlocker::Delegate | 62 class PowerSaveBlocker::Delegate |
| 89 : public base::RefCountedThreadSafe<PowerSaveBlocker::Delegate> { | 63 : public base::RefCountedThreadSafe<PowerSaveBlocker::Delegate> { |
| 90 public: | 64 public: |
| 91 Delegate(PowerSaveBlockerType type, | 65 Delegate(PowerSaveBlockerType type, |
| 92 const std::string& description, | 66 const std::string& description, |
| 93 scoped_refptr<base::SequencedTaskRunner> ui_task_runner) | 67 scoped_refptr<base::SequencedTaskRunner> ui_task_runner) |
| 94 : type_(type), | 68 : type_(type), |
| 95 description_(description), | 69 description_(description), |
| (...skipping 13 matching lines...) Expand all Loading... |
| 109 PowerSaveBlockerType type_; | 83 PowerSaveBlockerType type_; |
| 110 const std::string description_; | 84 const std::string description_; |
| 111 base::win::ScopedHandle handle_; | 85 base::win::ScopedHandle handle_; |
| 112 scoped_refptr<base::SequencedTaskRunner> ui_task_runner_; | 86 scoped_refptr<base::SequencedTaskRunner> ui_task_runner_; |
| 113 | 87 |
| 114 DISALLOW_COPY_AND_ASSIGN(Delegate); | 88 DISALLOW_COPY_AND_ASSIGN(Delegate); |
| 115 }; | 89 }; |
| 116 | 90 |
| 117 void PowerSaveBlocker::Delegate::ApplyBlock() { | 91 void PowerSaveBlocker::Delegate::ApplyBlock() { |
| 118 DCHECK(ui_task_runner_->RunsTasksInCurrentSequence()); | 92 DCHECK(ui_task_runner_->RunsTasksInCurrentSequence()); |
| 119 if (base::win::GetVersion() < base::win::VERSION_WIN7) | |
| 120 return ApplySimpleBlock(type_, 1); | |
| 121 | |
| 122 handle_.Set(CreatePowerRequest(RequestType(), description_)); | 93 handle_.Set(CreatePowerRequest(RequestType(), description_)); |
| 123 } | 94 } |
| 124 | 95 |
| 125 void PowerSaveBlocker::Delegate::RemoveBlock() { | 96 void PowerSaveBlocker::Delegate::RemoveBlock() { |
| 126 DCHECK(ui_task_runner_->RunsTasksInCurrentSequence()); | 97 DCHECK(ui_task_runner_->RunsTasksInCurrentSequence()); |
| 127 if (base::win::GetVersion() < base::win::VERSION_WIN7) | |
| 128 return ApplySimpleBlock(type_, -1); | |
| 129 | |
| 130 DeletePowerRequest(RequestType(), handle_.Take()); | 98 DeletePowerRequest(RequestType(), handle_.Take()); |
| 131 } | 99 } |
| 132 | 100 |
| 133 POWER_REQUEST_TYPE PowerSaveBlocker::Delegate::RequestType() { | 101 POWER_REQUEST_TYPE PowerSaveBlocker::Delegate::RequestType() { |
| 134 if (type_ == kPowerSaveBlockPreventDisplaySleep) | 102 if (type_ == kPowerSaveBlockPreventDisplaySleep) |
| 135 return PowerRequestDisplayRequired; | 103 return PowerRequestDisplayRequired; |
| 136 | 104 |
| 137 if (base::win::GetVersion() < base::win::VERSION_WIN8) | 105 if (base::win::GetVersion() == base::win::VERSION_WIN7) |
| 138 return PowerRequestSystemRequired; | 106 return PowerRequestSystemRequired; |
| 139 | 107 |
| 140 return PowerRequestExecutionRequired; | 108 return PowerRequestExecutionRequired; |
| 141 } | 109 } |
| 142 | 110 |
| 143 PowerSaveBlocker::PowerSaveBlocker( | 111 PowerSaveBlocker::PowerSaveBlocker( |
| 144 PowerSaveBlockerType type, | 112 PowerSaveBlockerType type, |
| 145 Reason reason, | 113 Reason reason, |
| 146 const std::string& description, | 114 const std::string& description, |
| 147 scoped_refptr<base::SequencedTaskRunner> ui_task_runner, | 115 scoped_refptr<base::SequencedTaskRunner> ui_task_runner, |
| 148 scoped_refptr<base::SingleThreadTaskRunner> blocking_task_runner) | 116 scoped_refptr<base::SingleThreadTaskRunner> blocking_task_runner) |
| 149 : delegate_(new Delegate(type, description, ui_task_runner)), | 117 : delegate_(new Delegate(type, description, ui_task_runner)), |
| 150 ui_task_runner_(ui_task_runner), | 118 ui_task_runner_(ui_task_runner), |
| 151 blocking_task_runner_(blocking_task_runner) { | 119 blocking_task_runner_(blocking_task_runner) { |
| 152 ui_task_runner_->PostTask(FROM_HERE, | 120 ui_task_runner_->PostTask(FROM_HERE, |
| 153 base::Bind(&Delegate::ApplyBlock, delegate_)); | 121 base::Bind(&Delegate::ApplyBlock, delegate_)); |
| 154 } | 122 } |
| 155 | 123 |
| 156 PowerSaveBlocker::~PowerSaveBlocker() { | 124 PowerSaveBlocker::~PowerSaveBlocker() { |
| 157 ui_task_runner_->PostTask(FROM_HERE, | 125 ui_task_runner_->PostTask(FROM_HERE, |
| 158 base::Bind(&Delegate::RemoveBlock, delegate_)); | 126 base::Bind(&Delegate::RemoveBlock, delegate_)); |
| 159 } | 127 } |
| 160 | 128 |
| 161 } // namespace device | 129 } // namespace device |
| OLD | NEW |