1 /* -*- Mode: C++; tab-width: 40; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
2 /* This Source Code Form is subject to the terms of the Mozilla Public
3 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
4 * You can obtain one at http://mozilla.org/MPL/2.0/. */
6 #include "mozilla/Hal.h"
7 #include "mozilla/HalWakeLock.h"
8 #include "mozilla/Services.h"
9 #include "mozilla/StaticPtr.h"
10 #include "mozilla/dom/ContentParent.h"
11 #include "nsClassHashtable.h"
12 #include "nsDataHashtable.h"
13 #include "nsHashKeys.h"
14 #include "nsIPropertyBag2.h"
15 #include "nsIObserverService.h"
17 using namespace mozilla
;
18 using namespace mozilla::hal
;
29 nsTArray
<uint64_t> processes
;
32 typedef nsDataHashtable
<nsUint64HashKey
, LockCount
> ProcessLockTable
;
33 typedef nsClassHashtable
<nsStringHashKey
, ProcessLockTable
> LockTable
;
35 int sActiveListeners
= 0;
36 StaticAutoPtr
<LockTable
> sLockTable
;
37 bool sInitialized
= false;
38 bool sIsShuttingDown
= false;
41 WakeLockInfoFromLockCount(const nsAString
& aTopic
, const LockCount
& aLockCount
)
43 // TODO: Once we abandon b2g18, we can switch this to use the
44 // WakeLockInformation constructor, which is better because it doesn't let us
45 // forget to assign a param. For now we have to do it this way, because
46 // b2g18 doesn't have the nsTArray <--> InfallibleTArray conversion (bug
49 WakeLockInformation info
;
50 info
.topic() = aTopic
;
51 info
.numLocks() = aLockCount
.numLocks
;
52 info
.numHidden() = aLockCount
.numHidden
;
53 info
.lockingProcesses().AppendElements(aLockCount
.processes
);
58 CountWakeLocks(const uint64_t& aKey
, LockCount aCount
, void* aUserArg
)
62 LockCount
* totalCount
= static_cast<LockCount
*>(aUserArg
);
63 totalCount
->numLocks
+= aCount
.numLocks
;
64 totalCount
->numHidden
+= aCount
.numHidden
;
66 // This is linear in the number of processes, but that should be small.
67 if (!totalCount
->processes
.Contains(aKey
)) {
68 totalCount
->processes
.AppendElement(aKey
);
74 static PLDHashOperator
75 RemoveChildFromList(const nsAString
& aKey
, nsAutoPtr
<ProcessLockTable
>& aTable
,
80 PLDHashOperator op
= PL_DHASH_NEXT
;
81 uint64_t childID
= *static_cast<uint64_t*>(aUserArg
);
82 if (aTable
->Get(childID
, nullptr)) {
83 aTable
->Remove(childID
);
86 aTable
->EnumerateRead(CountWakeLocks
, &totalCount
);
87 if (!totalCount
.numLocks
) {
91 if (sActiveListeners
) {
92 NotifyWakeLockChange(WakeLockInfoFromLockCount(aKey
, totalCount
));
99 class ClearHashtableOnShutdown MOZ_FINAL
: public nsIObserver
{
105 NS_IMPL_ISUPPORTS(ClearHashtableOnShutdown
, nsIObserver
)
108 ClearHashtableOnShutdown::Observe(nsISupports
* aSubject
, const char* aTopic
, const char16_t
* data
)
110 MOZ_ASSERT(!strcmp(aTopic
, "xpcom-shutdown"));
112 sIsShuttingDown
= true;
113 sLockTable
= nullptr;
118 class CleanupOnContentShutdown MOZ_FINAL
: public nsIObserver
{
124 NS_IMPL_ISUPPORTS(CleanupOnContentShutdown
, nsIObserver
)
127 CleanupOnContentShutdown::Observe(nsISupports
* aSubject
, const char* aTopic
, const char16_t
* data
)
129 MOZ_ASSERT(!strcmp(aTopic
, "ipc:content-shutdown"));
131 if (sIsShuttingDown
) {
135 nsCOMPtr
<nsIPropertyBag2
> props
= do_QueryInterface(aSubject
);
137 NS_WARNING("ipc:content-shutdown message without property bag as subject");
141 uint64_t childID
= 0;
142 nsresult rv
= props
->GetPropertyAsUint64(NS_LITERAL_STRING("childID"),
144 if (NS_SUCCEEDED(rv
)) {
145 sLockTable
->Enumerate(RemoveChildFromList
, &childID
);
147 NS_WARNING("ipc:content-shutdown message without childID property");
155 sLockTable
= new LockTable();
158 nsCOMPtr
<nsIObserverService
> obs
= mozilla::services::GetObserverService();
160 obs
->AddObserver(new ClearHashtableOnShutdown(), "xpcom-shutdown", false);
161 obs
->AddObserver(new CleanupOnContentShutdown(), "ipc:content-shutdown", false);
165 } // anonymous namespace
172 ComputeWakeLockState(int aNumLocks
, int aNumHidden
)
174 if (aNumLocks
== 0) {
175 return WAKE_LOCK_STATE_UNLOCKED
;
176 } else if (aNumLocks
== aNumHidden
) {
177 return WAKE_LOCK_STATE_HIDDEN
;
179 return WAKE_LOCK_STATE_VISIBLE
;
188 EnableWakeLockNotifications()
194 DisableWakeLockNotifications()
200 ModifyWakeLock(const nsAString
& aTopic
,
201 hal::WakeLockControl aLockAdjust
,
202 hal::WakeLockControl aHiddenAdjust
,
205 MOZ_ASSERT(NS_IsMainThread());
206 MOZ_ASSERT(aProcessID
!= CONTENT_PROCESS_ID_UNKNOWN
);
208 if (sIsShuttingDown
) {
215 ProcessLockTable
* table
= sLockTable
->Get(aTopic
);
216 LockCount processCount
;
217 LockCount totalCount
;
219 table
= new ProcessLockTable();
220 sLockTable
->Put(aTopic
, table
);
222 table
->Get(aProcessID
, &processCount
);
223 table
->EnumerateRead(CountWakeLocks
, &totalCount
);
226 MOZ_ASSERT(processCount
.numLocks
>= processCount
.numHidden
);
227 MOZ_ASSERT(aLockAdjust
>= 0 || processCount
.numLocks
> 0);
228 MOZ_ASSERT(aHiddenAdjust
>= 0 || processCount
.numHidden
> 0);
229 MOZ_ASSERT(totalCount
.numLocks
>= totalCount
.numHidden
);
230 MOZ_ASSERT(aLockAdjust
>= 0 || totalCount
.numLocks
> 0);
231 MOZ_ASSERT(aHiddenAdjust
>= 0 || totalCount
.numHidden
> 0);
233 WakeLockState oldState
= ComputeWakeLockState(totalCount
.numLocks
, totalCount
.numHidden
);
234 bool processWasLocked
= processCount
.numLocks
> 0;
236 processCount
.numLocks
+= aLockAdjust
;
237 processCount
.numHidden
+= aHiddenAdjust
;
239 totalCount
.numLocks
+= aLockAdjust
;
240 totalCount
.numHidden
+= aHiddenAdjust
;
242 if (processCount
.numLocks
) {
243 table
->Put(aProcessID
, processCount
);
245 table
->Remove(aProcessID
);
247 if (!totalCount
.numLocks
) {
248 sLockTable
->Remove(aTopic
);
251 if (sActiveListeners
&&
252 (oldState
!= ComputeWakeLockState(totalCount
.numLocks
,
253 totalCount
.numHidden
) ||
254 processWasLocked
!= (processCount
.numLocks
> 0))) {
256 WakeLockInformation info
;
257 hal::GetWakeLockInfo(aTopic
, &info
);
258 NotifyWakeLockChange(info
);
263 GetWakeLockInfo(const nsAString
& aTopic
, WakeLockInformation
* aWakeLockInfo
)
265 if (sIsShuttingDown
) {
266 NS_WARNING("You don't want to get wake lock information during xpcom-shutdown!");
267 *aWakeLockInfo
= WakeLockInformation();
274 ProcessLockTable
* table
= sLockTable
->Get(aTopic
);
276 *aWakeLockInfo
= WakeLockInfoFromLockCount(aTopic
, LockCount());
279 LockCount totalCount
;
280 table
->EnumerateRead(CountWakeLocks
, &totalCount
);
281 *aWakeLockInfo
= WakeLockInfoFromLockCount(aTopic
, totalCount
);