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

Commit fde770d5 authored by Nino Jagar's avatar Nino Jagar
Browse files

Add RemoteContentProtectionService 

Bug: 275732576
Test: Added new tests
Change-Id: Iad7fbc4727621233517a649bce8469519e243d6f
parent 59fc3204

core/java/android/service/contentcapture/ContentCaptureService.java

+12 −0
Original line number Diff line number Diff line
@@ -87,6 +87,18 @@ public abstract class ContentCaptureService extends Service { 
    public static final String SERVICE_INTERFACE =

            "android.service.contentcapture.ContentCaptureService";



    /**

     * The {@link Intent} that must be declared as handled by the protection service.

     *

     * <p>To be supported, the service must also require the {@link

     * android.Manifest.permission#BIND_CONTENT_CAPTURE_SERVICE} permission so that other

     * applications can not abuse it.

     *

     * @hide

     */

    public static final String PROTECTION_SERVICE_INTERFACE =

            "android.service.contentcapture.ContentProtectionService";



    /**

     * Name under which a ContentCaptureService component publishes information about itself.

     *

services/contentcapture/java/com/android/server/contentprotection/RemoteContentProtectionService.java

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

 * Copyright (C) 2023 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.contentprotection;



import static android.view.contentcapture.ContentCaptureSession.FLUSH_REASON_LOGIN_DETECTED;



import android.annotation.NonNull;

import android.content.ComponentName;

import android.content.Context;

import android.content.Intent;

import android.content.pm.ParceledListSlice;

import android.service.contentcapture.ContentCaptureService;

import android.util.Slog;

import android.view.contentcapture.ContentCaptureEvent;

import android.view.contentcapture.IContentCaptureDirectManager;



import com.android.internal.infra.ServiceConnector;



import java.time.Duration;



/**

 * Connector for the remote content protection service.

 *

 * @hide

 */

class RemoteContentProtectionService extends ServiceConnector.Impl<IContentCaptureDirectManager> {



    private static final String TAG = RemoteContentProtectionService.class.getSimpleName();



    private static final Duration AUTO_DISCONNECT_TIMEOUT = Duration.ofSeconds(3);



    @NonNull private final ComponentName mComponentName;



    protected RemoteContentProtectionService(

            @NonNull Context context,

            @NonNull ComponentName componentName,

            int userId,

            boolean bindAllowInstant) {

        super(

                context,

                new Intent(ContentCaptureService.PROTECTION_SERVICE_INTERFACE)

                        .setComponent(componentName),

                bindAllowInstant ? Context.BIND_ALLOW_INSTANT : 0,

                userId,

                IContentCaptureDirectManager.Stub::asInterface);

        mComponentName = componentName;

    }



    @Override // from ServiceConnector.Impl

    protected long getAutoDisconnectTimeoutMs() {

        return AUTO_DISCONNECT_TIMEOUT.toMillis();

    }



    @Override // from ServiceConnector.Impl

    protected void onServiceConnectionStatusChanged(

            @NonNull IContentCaptureDirectManager service, boolean isConnected) {

        Slog.i(

                TAG,

                "Connection status for: "

                        + mComponentName

                        + " changed to: "

                        + (isConnected ? "connected" : "disconnected"));

    }



    public void onLoginDetected(@NonNull ParceledListSlice<ContentCaptureEvent> events) {

        run(

                service ->

                        service.sendEvents(

                                events, FLUSH_REASON_LOGIN_DETECTED, /* options= */ null));

    }

}

services/tests/servicestests/src/com/android/server/contentprotection/RemoteContentProtectionServiceTest.java

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

 * Copyright (C) 2023 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.contentprotection;



import static android.view.contentcapture.ContentCaptureSession.FLUSH_REASON_LOGIN_DETECTED;



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



import static org.junit.Assert.fail;

import static org.mockito.Mockito.verify;

import static org.mockito.Mockito.verifyZeroInteractions;



import android.annotation.NonNull;

import android.content.ComponentName;

import android.content.Context;

import android.content.pm.ParceledListSlice;

import android.os.UserHandle;

import android.view.contentcapture.ContentCaptureEvent;

import android.view.contentcapture.IContentCaptureDirectManager;



import androidx.test.core.app.ApplicationProvider;

import androidx.test.ext.junit.runners.AndroidJUnit4;

import androidx.test.filters.SmallTest;



import com.android.internal.infra.AndroidFuture;

import com.android.internal.infra.ServiceConnector;



import com.google.common.collect.ImmutableList;



import org.junit.Before;

import org.junit.Rule;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.Mock;

import org.mockito.junit.MockitoJUnit;

import org.mockito.junit.MockitoRule;



/**

 * Test for {@link RemoteContentProtectionService}.

 *

 * <p>Run with: {@code atest

 * FrameworksServicesTests:com.android.server.contentprotection.RemoteContentProtectionServiceTest}

 */

@RunWith(AndroidJUnit4.class)

@SmallTest

public class RemoteContentProtectionServiceTest {



    private final Context mContext = ApplicationProvider.getApplicationContext();



    @Rule public final MockitoRule mMockitoRule = MockitoJUnit.rule();



    @Mock private IContentCaptureDirectManager mMockContentCaptureDirectManager;



    private RemoteContentProtectionService mRemoteContentProtectionService;



    private int mConnectCallCount = 0;



    @Before

    public void setup() {

        ComponentName componentName = new ComponentName(mContext.getPackageName(), "TestClass");

        mRemoteContentProtectionService =

                new TestRemoteContentProtectionService(mContext, componentName);

    }



    @Test

    public void doesNotAutoConnect() {

        assertThat(mConnectCallCount).isEqualTo(0);

        verifyZeroInteractions(mMockContentCaptureDirectManager);

    }



    @Test

    public void getAutoDisconnectTimeoutMs() {

        long actual = mRemoteContentProtectionService.getAutoDisconnectTimeoutMs();



        assertThat(actual).isEqualTo(3000L);

    }



    @Test

    public void onLoginDetected() throws Exception {

        ContentCaptureEvent event =

                new ContentCaptureEvent(/* sessionId= */ 1111, /* type= */ 2222);

        ParceledListSlice<ContentCaptureEvent> events =

                new ParceledListSlice<>(ImmutableList.of(event));



        mRemoteContentProtectionService.onLoginDetected(events);



        verify(mMockContentCaptureDirectManager)

                .sendEvents(events, FLUSH_REASON_LOGIN_DETECTED, /* options= */ null);

    }



    private final class TestRemoteContentProtectionService extends RemoteContentProtectionService {



        TestRemoteContentProtectionService(Context context, ComponentName componentName) {

            super(context, componentName, UserHandle.myUserId(), /* bindAllowInstant= */ false);

        }



        @Override // from ServiceConnector

        public synchronized AndroidFuture<IContentCaptureDirectManager> connect() {

            mConnectCallCount++;

            return AndroidFuture.completedFuture(mMockContentCaptureDirectManager);

        }



        @Override // from ServiceConnector

        public boolean run(@NonNull ServiceConnector.VoidJob<IContentCaptureDirectManager> job) {

            try {

                job.run(mMockContentCaptureDirectManager);

            } catch (Exception ex) {

                fail("Unexpected exception: " + ex);

            }

            return true;

        }

    }

}