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

Commit a08bf16b authored by Sharon Su's avatar Sharon Su
Browse files

Ensure service info is updated so that checking for remote service uid returns the correct value. 

Bug: 232421349
Test: atest WallpaperEffectsGenerationPerUserServiceTest,  and manual
test on device

Change-Id: I32638b25e8786ce915942f5b04c17836ded890cb
parent 6e086786

services/tests/mockingservicestests/Android.bp

+1 −0
Original line number Diff line number Diff line
@@ -60,6 +60,7 @@ android_test { 
        "services.devicepolicy",

        "services.net",

        "services.usage",

        "services.wallpapereffectsgeneration",

        "servicestests-core-utils",

        "servicestests-utils-mockito-extended",

        "testables",

services/tests/mockingservicestests/src/com/android/server/wallpapereffectsgeneration/WallpaperEffectsGenerationPerUserServiceTest.java

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

 * Copyright (C) 2022 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.wallpapereffectsgeneration;



import static com.android.dx.mockito.inline.extended.ExtendedMockito.spyOn;



import static org.junit.Assert.assertFalse;

import static org.junit.Assert.assertNotNull;

import static org.junit.Assert.assertNull;

import static org.junit.Assert.assertTrue;

import static org.mockito.ArgumentMatchers.any;

import static org.mockito.ArgumentMatchers.anyInt;

import static org.mockito.ArgumentMatchers.anyLong;

import static org.mockito.Mockito.spy;

import static org.mockito.Mockito.when;



import android.Manifest;

import android.annotation.SuppressLint;

import android.app.AppGlobals;

import android.app.wallpapereffectsgeneration.CinematicEffectRequest;

import android.app.wallpapereffectsgeneration.ICinematicEffectListener;

import android.content.Context;

import android.content.pm.ApplicationInfo;

import android.content.pm.IPackageManager;

import android.content.pm.PackageManager;

import android.content.pm.ServiceInfo;

import android.graphics.Bitmap;



import androidx.test.core.app.ApplicationProvider;

import androidx.test.filters.SmallTest;



import org.junit.Before;

import org.junit.Test;

import org.mockito.Mock;

import org.mockito.MockitoAnnotations;



@SmallTest

@SuppressLint("GuardedBy")

public class WallpaperEffectsGenerationPerUserServiceTest {

    private static final int BIND_SERVICE_UID = 123;

    @Mock

    PackageManager mMockPackageManager;

    @Mock

    ICinematicEffectListener mListener;



    private Context mContext;

    private IPackageManager mIPackageManager;

    private WallpaperEffectsGenerationPerUserService mService;

    private ServiceInfo mServiceInfo;



    @Before

    public void setUp() throws Exception {

        MockitoAnnotations.initMocks(this);

        mContext = spy(ApplicationProvider.getApplicationContext());

        mIPackageManager = AppGlobals.getPackageManager();

        spyOn(mIPackageManager);

        when(mContext.getPackageManager()).thenReturn(mMockPackageManager);

        when(mMockPackageManager.checkPermission(any(), any()))

                .thenReturn(PackageManager.PERMISSION_GRANTED);

        ApplicationInfo mockApplicationInfo = new ApplicationInfo();

        mockApplicationInfo.uid = BIND_SERVICE_UID;

        mServiceInfo = new ServiceInfo();

        mServiceInfo.permission = Manifest.permission.BIND_WALLPAPER_EFFECTS_GENERATION_SERVICE;

        mServiceInfo.applicationInfo = mockApplicationInfo;

        when(mIPackageManager.getServiceInfo(any(), anyLong(), anyInt()))

                .thenReturn(mServiceInfo);



        WallpaperEffectsGenerationManagerService managerService =

                new WallpaperEffectsGenerationManagerService(mContext);

        mService = new WallpaperEffectsGenerationPerUserService(

                managerService, new Object(), mContext.getUserId());

    }



    @Test

    public void testIsCallingUidAllowed_returnFalseWhenServiceNotBound() {

        assertNull(mService.getServiceInfo());

        assertFalse(mService.isCallingUidAllowed(BIND_SERVICE_UID));

    }



    @Test

    public void testIsCallingUidAllowed_uidSetCorrectlyAfterFirstCall() {

        CinematicEffectRequest request = new CinematicEffectRequest(

                "test-id", Bitmap.createBitmap(32, 48, Bitmap.Config.ARGB_8888));

        mService.onGenerateCinematicEffectLocked(request, mListener);

        assertNotNull(mService.getServiceInfo());

        assertTrue(mService.isCallingUidAllowed(BIND_SERVICE_UID));

    }

}

services/wallpapereffectsgeneration/java/com/android/server/wallpapereffectsgeneration/WallpaperEffectsGenerationPerUserService.java

+2 −3
Original line number Diff line number Diff line
@@ -233,14 +233,13 @@ public class WallpaperEffectsGenerationPerUserService extends 
    @Nullable

    private RemoteWallpaperEffectsGenerationService ensureRemoteServiceLocked() {

        if (mRemoteService == null) {

            final String serviceName = getComponentNameLocked();

            if (serviceName == null) {

            final ComponentName serviceComponent = updateServiceInfoLocked();

            if (serviceComponent == null) {

                if (mMaster.verbose) {

                    Slog.v(TAG, "ensureRemoteServiceLocked(): not set");

                }

                return null;

            }

            ComponentName serviceComponent = ComponentName.unflattenFromString(serviceName);



            mRemoteService = new RemoteWallpaperEffectsGenerationService(getContext(),

                    serviceComponent, mUserId, this,