From f91cfb12e89c595fa4c5ff4e168d304ed73bb990 Mon Sep 17 00:00:00 2001
From: olly <olly@google.com>
Date: Fri, 15 May 2020 00:27:33 +0100
Subject: [PATCH] Attach ExoMediaCryptoType for progressive streams

PiperOrigin-RevId: 311628160
---
 .../android/exoplayer2/source/ProgressiveMediaPeriod.java    | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/library/core/src/main/java/com/google/android/exoplayer2/source/ProgressiveMediaPeriod.java b/library/core/src/main/java/com/google/android/exoplayer2/source/ProgressiveMediaPeriod.java
index 32c96e14f7..2630132044 100644
--- a/library/core/src/main/java/com/google/android/exoplayer2/source/ProgressiveMediaPeriod.java
+++ b/library/core/src/main/java/com/google/android/exoplayer2/source/ProgressiveMediaPeriod.java
@@ -781,6 +781,11 @@ import org.checkerframework.checker.nullness.qual.MonotonicNonNull;
           trackFormat = trackFormat.buildUpon().setAverageBitrate(icyHeaders.bitrate).build();
         }
       }
+      if (trackFormat.drmInitData != null) {
+        trackFormat =
+            trackFormat.copyWithExoMediaCryptoType(
+                drmSessionManager.getExoMediaCryptoType(trackFormat.drmInitData));
+      }
       trackArray[i] = new TrackGroup(trackFormat);
     }
     trackState = new TrackState(new TrackGroupArray(trackArray), trackIsAudioVideoFlags);