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

Commit 36875556 authored by Tim Yu's avatar Tim Yu
Browse files

[RESTRICT AUTOMERGE] Check permission of Autofill icon URIs 

* SaveUI's template
* Inline Suggestions slices

Fixes: b/286235483
Fixes: b/292104015

Test: atest CtsAutoFillServiceTestCases

Change-Id: I48879174664b70ced90492bb0991dc91cbf89b79
Merged-In: I48879174664b70ced90492bb0991dc91cbf89b79
parent 54f3c01d

services/autofill/java/com/android/server/autofill/Helper.java

+46 −2
Original line number Diff line number Diff line
@@ -23,7 +23,10 @@ import android.app.ActivityManager; 
import android.app.assist.AssistStructure;

import android.app.assist.AssistStructure.ViewNode;

import android.app.assist.AssistStructure.WindowNode;

import android.app.slice.Slice;

import android.app.slice.SliceItem;

import android.content.ComponentName;

import android.graphics.drawable.Icon;

import android.metrics.LogMaker;

import android.service.autofill.Dataset;

import android.service.autofill.InternalSanitizer;
@@ -47,7 +50,6 @@ import java.util.ArrayList; 
import java.util.Arrays;

import java.util.concurrent.atomic.AtomicBoolean;





public final class Helper {



    private static final String TAG = "AutofillHelper";
@@ -85,7 +87,7 @@ public final class Helper { 
        final AtomicBoolean permissionsOk = new AtomicBoolean(true);



        rView.visitUris(uri -> {

            int uriOwnerId = android.content.ContentProvider.getUserIdFromUri(uri);

            int uriOwnerId = android.content.ContentProvider.getUserIdFromUri(uri, userId);

            boolean allowed = uriOwnerId == userId;

            permissionsOk.set(allowed && permissionsOk.get());

        });
@@ -117,6 +119,48 @@ public final class Helper { 
        return (ok ? rView : null);

    }



    /**

     * Checks the URI permissions of the icon in the slice, to see if the current userId is able to

     * access it.

     *

     * <p>Returns null if slice contains user inaccessible icons

     *

     * <p>TODO: instead of returning a null Slice when the current userId cannot access an icon,

     * return a reconstructed Slice without the icons. This is currently non-trivial since there are

     * no public methods to generically add SliceItems to Slices

     */

    public static @Nullable Slice sanitizeSlice(Slice slice) {

        if (slice == null) {

            return null;

        }



        int userId = ActivityManager.getCurrentUser();



        // Recontruct the Slice, filtering out bad icons

        for (SliceItem sliceItem : slice.getItems()) {

            if (!sliceItem.getFormat().equals(SliceItem.FORMAT_IMAGE)) {

                // Not an image slice

                continue;

            }



            Icon icon = sliceItem.getIcon();

            if (icon.getType() !=  Icon.TYPE_URI

                    && icon.getType() != Icon.TYPE_URI_ADAPTIVE_BITMAP) {

                // No URIs to sanitize

                continue;

            }



            int iconUriId = android.content.ContentProvider.getUserIdFromUri(icon.getUri(), userId);



            if (iconUriId != userId) {

                Slog.w(TAG, "sanitizeSlice() user: " + userId + " cannot access icons in Slice");

                return null;

            }

        }



        return slice;

    }





    @Nullable

    static AutofillId[] toArray(@Nullable ArraySet<AutofillId> set) {

services/autofill/java/com/android/server/autofill/ui/RemoteInlineSuggestionViewConnector.java

+10 −8
Original line number Diff line number Diff line
@@ -27,6 +27,7 @@ import android.service.autofill.InlinePresentation; 
import android.util.Slog;



import com.android.server.LocalServices;

import com.android.server.autofill.Helper;

import com.android.server.autofill.RemoteInlineSuggestionRenderService;

import com.android.server.inputmethod.InputMethodManagerInternal;
@@ -39,12 +40,9 @@ import java.util.function.Consumer; 
final class RemoteInlineSuggestionViewConnector {

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



    @Nullable

    private final RemoteInlineSuggestionRenderService mRemoteRenderService;

    @NonNull

    private final InlinePresentation mInlinePresentation;

    @Nullable

    private final IBinder mHostInputToken;

    @Nullable private final RemoteInlineSuggestionRenderService mRemoteRenderService;

    @NonNull private final InlinePresentation mInlinePresentation;

    @Nullable private final IBinder mHostInputToken;

    private final int mDisplayId;

    private final int mUserId;

    private final int mSessionId;
@@ -78,8 +76,12 @@ final class RemoteInlineSuggestionViewConnector { 
     *

     * @return true if the call is made to the remote renderer service, false otherwise.

     */

    public boolean renderSuggestion(int width, int height,

            @NonNull IInlineSuggestionUiCallback callback) {

    public boolean renderSuggestion(

            int width, int height, @NonNull IInlineSuggestionUiCallback callback) {

        if (Helper.sanitizeSlice(mInlinePresentation.getSlice()) == null) {

            if (sDebug) Slog.d(TAG, "Skipped rendering inline suggestion.");

            return false;

        }

        if (mRemoteRenderService != null) {

            if (sDebug) Slog.d(TAG, "Request to recreate the UI");

            mRemoteRenderService.renderSuggestion(callback, mInlinePresentation, width, height,

services/autofill/java/com/android/server/autofill/ui/SaveUi.java

+2 −1
Original line number Diff line number Diff line
@@ -427,7 +427,8 @@ final class SaveUi { 
                    }

                    final BatchUpdates batchUpdates = pair.second;

                    // First apply the updates...

                    final RemoteViews templateUpdates = batchUpdates.getUpdates();

                    final RemoteViews templateUpdates =

                            Helper.sanitizeRemoteView(batchUpdates.getUpdates());

                    if (templateUpdates != null) {

                        if (sDebug) Slog.d(TAG, "Applying template updates for batch update #" + i);

                        templateUpdates.reapply(context, customSubtitleView);