tests/tests/security/src/android/security/cts/MediaRecorderInfoLeakTest.java

/**
 * Copyright (C) 2018 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 android.media.MediaRecorder;
import com.android.sts.common.util.StsExtraBusinessLogicTestCase;
import android.util.Log;

import java.io.File;

import static org.junit.Assert.*;

import androidx.test.runner.AndroidJUnit4;
import org.junit.runner.RunWith;
import org.junit.Test;

@RunWith(AndroidJUnit4.class)
public class MediaRecorderInfoLeakTest extends StsExtraBusinessLogicTestCase {

   /**
    *  b/27855172
    */
    @Test
    @AsbSecurityTest(cveBugId = 27855172)
    public void test_cve_2016_2499() throws Exception {
        MediaRecorder mediaRecorder = null;
        long end = System.currentTimeMillis() + 600_000; // 10 minutes from now
        String TAG = "MediaRecorderInfoLeakTest";
        boolean recorderPrepare = false;
        int retryCount = 2;

        try {
            while(System.currentTimeMillis() < end) {
                recorderPrepare = false;
                for (int i = 0; i < retryCount; i++) {
                    try {
                        mediaRecorder = new MediaRecorder();
                        mediaRecorder.setAudioSource(MediaRecorder.AudioSource.DEFAULT);
                        mediaRecorder.setVideoSource(MediaRecorder.VideoSource.SURFACE);
                        mediaRecorder.setOutputFormat(MediaRecorder.OutputFormat.THREE_GPP);
                        mediaRecorder.setAudioEncoder(MediaRecorder.AudioEncoder.AAC);
                        mediaRecorder.setVideoEncoder(MediaRecorder.VideoEncoder.H263);
                        mediaRecorder.setVideoFrameRate(30);
                        mediaRecorder.setVideoSize(352, 288);
                        mediaRecorder.setOutputFile(
                                new File(getContext().getFilesDir(), "record.output").getPath());
                        mediaRecorder.prepare();
                        recorderPrepare = true;
                        break;
                    } catch (Exception ex) {
                        Log.w(TAG, "Media Recorder Prepare Exception" + ex.getMessage());
                        Thread.sleep(200);
                    } finally {
                        if (recorderPrepare == false && mediaRecorder != null){
                            mediaRecorder.release();
                        }
                    }
                }
                if(recorderPrepare){
                    int test = mediaRecorder.getMaxAmplitude();
                    mediaRecorder.reset();
                    mediaRecorder.release();
                    if(test != 0){
                        fail("MediaRecorderInfoLeakTest failed");
                    }
                } else {
                    fail("Media Recorder prepare fail");
                }
            }
        } catch (Exception e) {
            fail("Media Recorder Exception" + e.getMessage());
        } finally {
            if (mediaRecorder != null){
                mediaRecorder.release();
            }
        }
    }
}