Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit f548942f authored by Michael Wachenschwanz's avatar Michael Wachenschwanz
Browse files

Ref count isolated uid use for long partial wakelocks 

The isolated uids need to be kept in the map while it is holding a long
partial wakelock.

Bug: 218298973
Test: atest BatteryStatsNoteTest
Change-Id: I86456137e6d6ad47f9974da281defc04d20fb32f
(cherry picked from commit fd48aa6f)
Merged-In: I86456137e6d6ad47f9974da281defc04d20fb32f
parent 3f0d2e8c

core/java/com/android/internal/os/BatteryStatsImpl.java

+17 −8
Original line number Diff line number Diff line
@@ -4664,7 +4664,6 @@ public class BatteryStatsImpl extends BatteryStats { 














    public void noteLongPartialWakelockStart(String name, String historyName, int uid,















            long elapsedRealtimeMs, long uptimeMs) {













        uid = mapUid(uid);















        noteLongPartialWakeLockStartInternal(name, historyName, uid, elapsedRealtimeMs, uptimeMs);















    }
























@@ -4696,15 +4695,21 @@ public class BatteryStatsImpl extends BatteryStats {



























    private void noteLongPartialWakeLockStartInternal(String name, String historyName, int uid,















            long elapsedRealtimeMs, long uptimeMs) {













        final int mappedUid = mapUid(uid);















        if (historyName == null) {















            historyName = name;















        }













        if (!mActiveEvents.updateState(HistoryItem.EVENT_LONG_WAKE_LOCK_START, historyName, uid,













                0)) {













        if (!mActiveEvents.updateState(HistoryItem.EVENT_LONG_WAKE_LOCK_START, historyName,













                mappedUid, 0)) {















            return;















        }















        addHistoryEventLocked(elapsedRealtimeMs, uptimeMs, HistoryItem.EVENT_LONG_WAKE_LOCK_START,













                historyName, uid);













                historyName, mappedUid);













        if (mappedUid != uid) {













            // Prevent the isolated uid mapping from being removed while the wakelock is













            // being held.













            incrementIsolatedUidRefCount(uid);













        }















    }





























    public void noteLongPartialWakelockFinish(String name, String historyName, int uid) {









@@ -4714,7 +4719,6 @@ public class BatteryStatsImpl extends BatteryStats {



























    public void noteLongPartialWakelockFinish(String name, String historyName, int uid,















            long elapsedRealtimeMs, long uptimeMs) {













        uid = mapUid(uid);















        noteLongPartialWakeLockFinishInternal(name, historyName, uid, elapsedRealtimeMs, uptimeMs);















    }
























@@ -4746,15 +4750,20 @@ public class BatteryStatsImpl extends BatteryStats {



























    private void noteLongPartialWakeLockFinishInternal(String name, String historyName, int uid,















            long elapsedRealtimeMs, long uptimeMs) {













        final int mappedUid = mapUid(uid);















        if (historyName == null) {















            historyName = name;















        }













        if (!mActiveEvents.updateState(HistoryItem.EVENT_LONG_WAKE_LOCK_FINISH, historyName, uid,













                0)) {













        if (!mActiveEvents.updateState(HistoryItem.EVENT_LONG_WAKE_LOCK_FINISH, historyName,













                mappedUid, 0)) {















            return;















        }















        addHistoryEventLocked(elapsedRealtimeMs, uptimeMs, HistoryItem.EVENT_LONG_WAKE_LOCK_FINISH,













                historyName, uid);













                historyName, mappedUid);













        if (mappedUid != uid) {













            // Decrement the ref count for the isolated uid and delete the mapping if uneeded.













            maybeRemoveIsolatedUidLocked(uid, elapsedRealtimeMs, uptimeMs);













        }















    }





























    void aggregateLastWakeupUptimeLocked(long elapsedRealtimeMs, long uptimeMs) {

































core/tests/coretests/src/com/android/internal/os/BatteryStatsNoteTest.java



+112
−0




























Original line number
Diff line number
Diff line








@@ -23,6 +23,8 @@ import static android.os.BatteryStats.WAKE_TYPE_PARTIAL;













import static com.android.internal.os.BatteryStatsImpl.ExternalStatsSync.UPDATE_CPU;















import static com.android.internal.os.BatteryStatsImpl.ExternalStatsSync.UPDATE_DISPLAY;





























import static com.google.common.truth.Truth.assertThat;





























import android.app.ActivityManager;















import android.os.BatteryStats;















import android.os.BatteryStats.HistoryItem;










@@ -206,6 +208,116 @@ public class BatteryStatsNoteTest extends TestCase {













        assertEquals(120_000, bgTime);















    }





























    /**













     * Test BatteryStatsImpl.Uid.noteLongPartialWakelockStart for an isolated uid.













     */













    @SmallTest













    public void testNoteLongPartialWakelockStart_isolatedUid() throws Exception {













        final MockClocks clocks = new MockClocks(); // holds realtime and uptime in ms













        MockBatteryStatsImpl bi = new MockBatteryStatsImpl(clocks);









































        bi.setRecordAllHistoryLocked(true);













        bi.forceRecordAllHistory();



























        int pid = 10;













        String name = "name";













        String historyName = "historyName";



























        WorkSource.WorkChain isolatedWorkChain = new WorkSource.WorkChain();













        isolatedWorkChain.addNode(ISOLATED_UID, name);



























        // Map ISOLATED_UID to UID.













        bi.addIsolatedUidLocked(ISOLATED_UID, UID);



























        bi.updateTimeBasesLocked(true, Display.STATE_OFF, 0, 0);













        bi.noteUidProcessStateLocked(UID, ActivityManager.PROCESS_STATE_TOP);













        bi.noteLongPartialWakelockStart(name, historyName, ISOLATED_UID);



























        clocks.realtime = clocks.uptime = 100;













        bi.noteUidProcessStateLocked(UID, ActivityManager.PROCESS_STATE_IMPORTANT_BACKGROUND);



























        clocks.realtime = clocks.uptime = 220;













        bi.noteLongPartialWakelockFinish(name, historyName, ISOLATED_UID);



























        final BatteryStatsHistoryIterator iterator =













                bi.createBatteryStatsHistoryIterator();



























        BatteryStats.HistoryItem item = new BatteryStats.HistoryItem();



























        while (iterator.next(item)) {













            if (item.eventCode == HistoryItem.EVENT_LONG_WAKE_LOCK_START) break;













        }













        assertThat(item.eventCode).isEqualTo(HistoryItem.EVENT_LONG_WAKE_LOCK_START);













        assertThat(item.eventTag).isNotNull();













        assertThat(item.eventTag.string).isEqualTo(historyName);













        assertThat(item.eventTag.uid).isEqualTo(UID);



























        while (iterator.next(item)) {













            if (item.eventCode == HistoryItem.EVENT_LONG_WAKE_LOCK_FINISH) break;













        }













        assertThat(item.eventCode).isEqualTo(HistoryItem.EVENT_LONG_WAKE_LOCK_FINISH);













        assertThat(item.eventTag).isNotNull();













        assertThat(item.eventTag.string).isEqualTo(historyName);













        assertThat(item.eventTag.uid).isEqualTo(UID);













    }



























    /**













     * Test BatteryStatsImpl.Uid.noteLongPartialWakelockStart for an isolated uid.













     */













    @SmallTest













    public void testNoteLongPartialWakelockStart_isolatedUidRace() throws Exception {













        final MockClocks clocks = new MockClocks(); // holds realtime and uptime in ms













        MockBatteryStatsImpl bi = new MockBatteryStatsImpl(clocks);









































        bi.setRecordAllHistoryLocked(true);













        bi.forceRecordAllHistory();



























        int pid = 10;













        String name = "name";













        String historyName = "historyName";



























        WorkSource.WorkChain isolatedWorkChain = new WorkSource.WorkChain();













        isolatedWorkChain.addNode(ISOLATED_UID, name);



























        // Map ISOLATED_UID to UID.













        bi.addIsolatedUidLocked(ISOLATED_UID, UID);



























        bi.updateTimeBasesLocked(true, Display.STATE_OFF, 0, 0);













        bi.noteUidProcessStateLocked(UID, ActivityManager.PROCESS_STATE_TOP);













        bi.noteLongPartialWakelockStart(name, historyName, ISOLATED_UID);



























        clocks.realtime = clocks.uptime = 100;













        bi.noteUidProcessStateLocked(UID, ActivityManager.PROCESS_STATE_IMPORTANT_BACKGROUND);



























        clocks.realtime = clocks.uptime = 150;













        bi.maybeRemoveIsolatedUidLocked(ISOLATED_UID, clocks.realtime, clocks.uptime);



























        clocks.realtime = clocks.uptime = 220;













        bi.noteLongPartialWakelockFinish(name, historyName, ISOLATED_UID);



























        final BatteryStatsHistoryIterator iterator =













                bi.createBatteryStatsHistoryIterator();



























        BatteryStats.HistoryItem item = new BatteryStats.HistoryItem();



























        while (iterator.next(item)) {













            if (item.eventCode == HistoryItem.EVENT_LONG_WAKE_LOCK_START) break;













        }













        assertThat(item.eventCode).isEqualTo(HistoryItem.EVENT_LONG_WAKE_LOCK_START);













        assertThat(item.eventTag).isNotNull();













        assertThat(item.eventTag.string).isEqualTo(historyName);













        assertThat(item.eventTag.uid).isEqualTo(UID);



























        while (iterator.next(item)) {













            if (item.eventCode == HistoryItem.EVENT_LONG_WAKE_LOCK_FINISH) break;













        }













        assertThat(item.eventCode).isEqualTo(HistoryItem.EVENT_LONG_WAKE_LOCK_FINISH);













        assertThat(item.eventTag).isNotNull();













        assertThat(item.eventTag.string).isEqualTo(historyName);













        assertThat(item.eventTag.uid).isEqualTo(UID);













    }































    /**















     * Test BatteryStatsImpl.noteUidProcessStateLocked.