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

Commit 7e6b0eeb authored by Evan Chen's avatar Evan Chen Committed by Automerger Merge Worker
Browse files

Merge "Notify/bind app after the phone is unlocked" into udc-dev am: 14537765 

Original change: https://googleplex-android-review.googlesource.com/c/platform/frameworks/base/+/22462089



Change-Id: Ic0b2b78237f34be1d2fb5babc5fb249e75e97eb3
Signed-off-by: default avatarAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
parents 6f8d98ce 14537765

services/companion/java/com/android/server/companion/CompanionDeviceManagerService.java

+17 −1
Original line number Diff line number Diff line
@@ -54,6 +54,7 @@ import android.app.ActivityManagerInternal; 
import android.app.AppOpsManager;

import android.app.NotificationManager;

import android.app.PendingIntent;

import android.bluetooth.BluetoothDevice;

import android.companion.AssociationInfo;

import android.companion.AssociationRequest;

import android.companion.DeviceNotAssociatedException;
@@ -228,7 +229,7 @@ public class CompanionDeviceManagerService extends SystemService { 
        loadAssociationsFromDisk();

        mAssociationStore.registerListener(mAssociationStoreChangeListener);



        mDevicePresenceMonitor = new CompanionDevicePresenceMonitor(

        mDevicePresenceMonitor = new CompanionDevicePresenceMonitor(mUserManager,

                mAssociationStore, mDevicePresenceCallback);



        mAssociationRequestsProcessor = new AssociationRequestsProcessor(
@@ -314,6 +315,21 @@ public class CompanionDeviceManagerService extends SystemService { 
                MINUTES.toMillis(10));

    }



    @Override

    public void onUserUnlocked(@NonNull TargetUser user) {

        // Notify and bind the app after the phone is unlocked.

        final int userId = user.getUserIdentifier();

        final Set<BluetoothDevice> blueToothDevices =

                mDevicePresenceMonitor.getPendingReportConnectedDevices().get(userId);

        for (BluetoothDevice bluetoothDevice : blueToothDevices) {

            for (AssociationInfo ai:

                    mAssociationStore.getAssociationsByAddress(bluetoothDevice.getAddress())) {

                Slog.i(TAG, "onUserUnlocked, device id( " + ai.getId() + " ) is connected");

                mDevicePresenceMonitor.onBluetoothCompanionDeviceConnected(ai.getId());

            }

        }

    }



    @NonNull

    AssociationInfo getAssociationWithCallerChecks(

            @UserIdInt int userId, @NonNull String packageName, @NonNull String macAddress) {

services/companion/java/com/android/server/companion/presence/BluetoothCompanionDeviceConnectionListener.java

+53 −4
Original line number Diff line number Diff line
@@ -27,17 +27,23 @@ import android.companion.AssociationInfo; 
import android.net.MacAddress;

import android.os.Handler;

import android.os.HandlerExecutor;

import android.os.UserHandle;

import android.os.UserManager;

import android.util.Log;

import android.util.SparseArray;



import com.android.internal.annotations.GuardedBy;

import com.android.server.companion.AssociationStore;



import java.util.Arrays;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import java.util.Set;



@SuppressLint("LongLogTag")

class BluetoothCompanionDeviceConnectionListener

public class BluetoothCompanionDeviceConnectionListener

        extends BluetoothAdapter.BluetoothConnectionCallback

        implements AssociationStore.OnChangeListener {

    private static final String TAG = "CDM_BluetoothCompanionDeviceConnectionListener";
@@ -48,15 +54,23 @@ class BluetoothCompanionDeviceConnectionListener 
        void onBluetoothCompanionDeviceDisconnected(int associationId);

    }



    private final UserManager mUserManager;

    private final @NonNull AssociationStore mAssociationStore;

    private final @NonNull Callback mCallback;

    /** A set of ALL connected BT device (not only companion.) */

    private final @NonNull Map<MacAddress, BluetoothDevice> mAllConnectedDevices = new HashMap<>();



    BluetoothCompanionDeviceConnectionListener(@NonNull AssociationStore associationStore,

            @NonNull Callback callback) {



    @GuardedBy("mPendingReportConnectedDevices")

    @NonNull

    final SparseArray<Set<BluetoothDevice>> mPendingReportConnectedDevices =

            new SparseArray<>();



    BluetoothCompanionDeviceConnectionListener(UserManager userManager,

            @NonNull AssociationStore associationStore, @NonNull Callback callback) {

        mAssociationStore = associationStore;

        mCallback = callback;

        mUserManager = userManager;

    }



    public void init(@NonNull BluetoothAdapter btAdapter) {
@@ -76,13 +90,33 @@ class BluetoothCompanionDeviceConnectionListener 
        if (DEBUG) Log.i(TAG, "onDevice_Connected() " + btDeviceToString(device));



        final MacAddress macAddress = MacAddress.fromString(device.getAddress());

        final int userId = UserHandle.myUserId();



        if (mAllConnectedDevices.put(macAddress, device) != null) {

            if (DEBUG) Log.w(TAG, "Device " + btDeviceToString(device) + " is already connected.");

            return;

        }

        // Try to bind and notify the app after the phone is unlocked.

        if (!mUserManager.isUserUnlockingOrUnlocked(UserHandle.myUserId())) {

            if (DEBUG) {

                Log.i(TAG, "Current user is not in unlocking or unlocked stage yet. Notify "

                        + "the application when the phone is unlocked");

            }

            synchronized (mPendingReportConnectedDevices) {

                Set<BluetoothDevice> bluetoothDevices = mPendingReportConnectedDevices.get(userId);



                if (bluetoothDevices == null) {

                    bluetoothDevices = new HashSet<>();

                    mPendingReportConnectedDevices.put(userId, bluetoothDevices);

                }



                bluetoothDevices.add(device);

            }



        } else {

            onDeviceConnectivityChanged(device, true);

        }

    }



    /**

     * Overrides
@@ -98,6 +132,8 @@ class BluetoothCompanionDeviceConnectionListener 
        }



        final MacAddress macAddress = MacAddress.fromString(device.getAddress());

        final int userId = UserHandle.myUserId();



        if (mAllConnectedDevices.remove(macAddress) == null) {

            if (DEBUG) {

                Log.w(TAG, "The device wasn't tracked as connected " + btDeviceToString(device));
@@ -105,6 +141,19 @@ class BluetoothCompanionDeviceConnectionListener 
            return;

        }



        // Do not need to report the connectivity since the user is not unlock the phone so

        // that cdm is not bind with the app yet.

        if (!mUserManager.isUserUnlockingOrUnlocked(userId)) {

            synchronized (mPendingReportConnectedDevices) {

                Set<BluetoothDevice> bluetoothDevices = mPendingReportConnectedDevices.get(userId);

                if (bluetoothDevices != null) {

                    bluetoothDevices.remove(device);

                }

            }



            return;

        }



        onDeviceConnectivityChanged(device, false);

    }

services/companion/java/com/android/server/companion/presence/CompanionDevicePresenceMonitor.java

+16 −5
Original line number Diff line number Diff line
@@ -23,13 +23,16 @@ import android.annotation.NonNull; 
import android.annotation.SuppressLint;

import android.annotation.TestApi;

import android.bluetooth.BluetoothAdapter;

import android.bluetooth.BluetoothDevice;

import android.companion.AssociationInfo;

import android.content.Context;

import android.os.Binder;

import android.os.Handler;

import android.os.Looper;

import android.os.Message;

import android.os.UserManager;

import android.util.Log;

import android.util.SparseArray;



import com.android.server.companion.AssociationStore;
@@ -86,13 +89,12 @@ public class CompanionDevicePresenceMonitor implements AssociationStore.OnChange 
    private final SimulatedDevicePresenceSchedulerHelper mSchedulerHelper =

            new SimulatedDevicePresenceSchedulerHelper();



    public CompanionDevicePresenceMonitor(@NonNull AssociationStore associationStore,

            @NonNull Callback callback) {

    public CompanionDevicePresenceMonitor(UserManager userManager,

            @NonNull AssociationStore associationStore, @NonNull Callback callback) {

        mAssociationStore = associationStore;

        mCallback = callback;



        mBtConnectionListener = new BluetoothCompanionDeviceConnectionListener(associationStore,

                /* BluetoothCompanionDeviceConnectionListener.Callback */ this);

        mBtConnectionListener = new BluetoothCompanionDeviceConnectionListener(userManager,

                associationStore, /* BluetoothCompanionDeviceConnectionListener.Callback */ this);

        mBleScanner = new BleCompanionDeviceScanner(associationStore,

                /* BleCompanionDeviceScanner.Callback */ this);

    }
@@ -298,6 +300,15 @@ public class CompanionDevicePresenceMonitor implements AssociationStore.OnChange 
        // what's needed.

    }



    /**

     * Return a set of devices that pending to report connectivity

     */

    public SparseArray<Set<BluetoothDevice>> getPendingReportConnectedDevices() {

        synchronized (mBtConnectionListener.mPendingReportConnectedDevices) {

            return mBtConnectionListener.mPendingReportConnectedDevices;

        }

    }



    private static void enforceCallerShellOrRoot() {

        final int callingUid = Binder.getCallingUid();

        if (callingUid == SHELL_UID || callingUid == ROOT_UID) return;