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

Commit a3ee82a8 authored by Kevin Han's avatar Kevin Han
Browse files

Add shell commands for app hibernation service 

Add some shell commands to make it easier to test and tinker with app
hibernation service manually.

Bug: 175829330
Test: adb shell cmd app_hibernation set-state
com.google.android.deskclock true
adb shell cmd cmd app_hibernation get-state com.google.android.deskclock
Test: manual testing

Change-Id: I8fffd95288c65efedac55463c79b25f1252de260
parent 78fd9db1

services/core/java/com/android/server/apphibernation/AppHibernationService.java

+12 −0
Original line number Diff line number Diff line
@@ -25,6 +25,7 @@ import static android.content.pm.PackageManager.MATCH_ALL; 
import static android.provider.DeviceConfig.NAMESPACE_APP_HIBERNATION;



import android.annotation.NonNull;

import android.annotation.Nullable;

import android.app.ActivityManager;

import android.app.IActivityManager;

import android.apphibernation.IAppHibernationService;
@@ -37,7 +38,9 @@ import android.content.pm.PackageInfo; 
import android.content.pm.UserInfo;

import android.os.Binder;

import android.os.RemoteException;

import android.os.ResultReceiver;

import android.os.ServiceManager;

import android.os.ShellCallback;

import android.os.Trace;

import android.os.UserHandle;

import android.os.UserManager;
@@ -49,6 +52,7 @@ import com.android.internal.annotations.GuardedBy; 
import com.android.internal.annotations.VisibleForTesting;

import com.android.server.SystemService;



import java.io.FileDescriptor;

import java.util.List;

import java.util.Map;
@@ -281,6 +285,14 @@ public final class AppHibernationService extends SystemService { 
        public void setHibernating(String packageName, int userId, boolean isHibernating) {

            mService.setHibernating(packageName, userId, isHibernating);

        }



        @Override

        public void onShellCommand(@Nullable FileDescriptor in, @Nullable FileDescriptor out,

                @Nullable FileDescriptor err, @NonNull String[] args,

                @Nullable ShellCallback callback, @NonNull ResultReceiver resultReceiver) {

            new AppHibernationShellCommand(mService).exec(this, in, out, err, args, callback,

                    resultReceiver);

        }

    }



    // Broadcast receiver for user and package add/removal events

services/core/java/com/android/server/apphibernation/AppHibernationShellCommand.java

0 → 100644
+109 −0
Original line number Diff line number Diff line 
/*

 * Copyright (C) 2021 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */



package com.android.server.apphibernation;



import android.os.ShellCommand;

import android.os.UserHandle;

import android.text.TextUtils;



import java.io.PrintWriter;



/**

 * Shell command implementation for {@link AppHibernationService}.

 */

final class AppHibernationShellCommand extends ShellCommand {

    private static final String USER_OPT = "--user";

    private static final int SUCCESS = 0;

    private static final int ERROR = -1;

    private final AppHibernationService mService;



    AppHibernationShellCommand(AppHibernationService service) {

        mService = service;

    }



    @Override

    public int onCommand(String cmd) {

        if (cmd == null) {

            return handleDefaultCommands(cmd);

        }

        switch (cmd) {

            case "set-state":

                return runSetState();

            case "get-state":

                return runGetState();

            default:

                return handleDefaultCommands(cmd);

        }

    }



    private int runSetState() {

        int userId = parseUserOption();



        String pkg = getNextArgRequired();

        if (pkg == null) {

            getErrPrintWriter().println("Error: no package specified");

            return ERROR;

        }



        String newStateRaw = getNextArgRequired();

        if (newStateRaw == null) {

            getErrPrintWriter().println("Error: No state to set specified");

            return ERROR;

        }

        boolean newState = Boolean.parseBoolean(newStateRaw);



        mService.setHibernating(pkg, userId, newState);

        return SUCCESS;

    }



    private int runGetState() {

        int userId = parseUserOption();



        String pkg = getNextArgRequired();

        if (pkg == null) {

            getErrPrintWriter().println("Error: No package specified");

            return ERROR;

        }

        boolean isHibernating = mService.isHibernating(pkg, userId);

        final PrintWriter pw = getOutPrintWriter();

        pw.println(isHibernating);

        return SUCCESS;

    }



    private int parseUserOption() {

        String option = getNextOption();

        if (TextUtils.equals(option, USER_OPT)) {

            return UserHandle.parseUserArg(getNextArgRequired());

        }

        return UserHandle.USER_CURRENT;

    }



    @Override

    public void onHelp() {

        final PrintWriter pw = getOutPrintWriter();

        pw.println("App hibernation (app_hibernation) commands: ");

        pw.println("  help");

        pw.println("    Print this help text.");

        pw.println("");

        pw.println("  set-state [--user USER_ID] PACKAGE true|false");

        pw.println("    Sets the hibernation state of the package to value specified");

        pw.println("");

        pw.println("  get-state [--user USER_ID] PACKAGE");

        pw.println("    Gets the hibernation state of the package");

        pw.println("");

    }

}