hostsidetests/securitybulletin/src/android/security/cts/CVE_2021_39804.java

/*
 * 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 android.security.cts;

import android.platform.test.annotations.AsbSecurityTest;

import com.android.compatibility.common.util.CrashUtils;
import com.android.compatibility.common.util.CrashUtils.Config.BacktraceFilterPattern;
import com.android.sts.common.tradefed.testtype.NonRootSecurityTestCase;
import com.android.tradefed.testtype.DeviceJUnit4ClassRunner;

import org.junit.Test;
import org.junit.runner.RunWith;

import java.util.Arrays;

@RunWith(DeviceJUnit4ClassRunner.class)
public class CVE_2021_39804 extends NonRootSecurityTestCase {

    /**
     * b/215002587
     * Vulnerability Behaviour: SIGSEGV in self
     * Vulnerable Library: libheif (As per AOSP code)
     * Vulnerable Function: reinit (As per AOSP code)
     */
    @AsbSecurityTest(cveBugId = 215002587)
    @Test
    public void testPocCVE_2021_39804() throws Exception {
        String inputFiles[] = {"cve_2021_39804.heif"};
        String binaryName = "CVE-2021-39804";
        String signals[] = {CrashUtils.SIGSEGV};
        AdbUtils.pocConfig testConfig = new AdbUtils.pocConfig(binaryName, getDevice());
        testConfig.config =
                new CrashUtils.Config().setProcessPatterns(binaryName).setBacktraceIncludes(
                        new BacktraceFilterPattern("libheif", "android::HeifDecoderImpl::reinit"));
        testConfig.config.checkMinAddress(false);
        testConfig.config.setSignals(signals);
        testConfig.arguments = AdbUtils.TMP_PATH + inputFiles[0];
        testConfig.inputFiles = Arrays.asList(inputFiles);
        testConfig.inputFilesDestination = AdbUtils.TMP_PATH;
        AdbUtils.runPocAssertNoCrashesNotVulnerable(testConfig);
    }
}