Fix for inputsize calculation

Bug: 361349204
Flag: EXEMPT does not affect framework behaviour

Change-Id: Ic479198ace2954df1cecc269139968468ef4309c
diff --git a/media/tests/benchmark/MediaBenchmarkTest/src/main/java/com/android/media/benchmark/library/MultiAccessUnitBlockModelDecoder.java b/media/tests/benchmark/MediaBenchmarkTest/src/main/java/com/android/media/benchmark/library/MultiAccessUnitBlockModelDecoder.java
index ed2defe..d1a5d79 100644
--- a/media/tests/benchmark/MediaBenchmarkTest/src/main/java/com/android/media/benchmark/library/MultiAccessUnitBlockModelDecoder.java
+++ b/media/tests/benchmark/MediaBenchmarkTest/src/main/java/com/android/media/benchmark/library/MultiAccessUnitBlockModelDecoder.java
@@ -72,8 +72,11 @@
         final String mime = format.getString(MediaFormat.KEY_MIME);
         final int maxOutputSize = format.getNumber(
             MediaFormat.KEY_BUFFER_BATCH_MAX_OUTPUT_SIZE, 0).intValue();
-        final int maxInputSizeInBytes = format.getInteger(
-                MediaFormat.KEY_MAX_INPUT_SIZE);
+        int maxInputSizeInBytes = 0;
+        if (format.containsKey(MediaFormat.KEY_MAX_INPUT_SIZE)) {
+            maxInputSizeInBytes = format.getNumber(
+                    MediaFormat.KEY_MAX_INPUT_SIZE, 0).intValue();
+        }
         mMaxInputSize = Math.max(maxInputSizeInBytes,
                 (int) (maxOutputSize * CodecUtils.getCompressionRatio(mime)));
     }