Fix offload buffer full detection after setEndOfStream

This issue has been observed on a test app stress
testing setEndOfStream.

The issue has not been observed on ExoPlayer,
probably due to timing differences, but it is fixed
preventively.

#exo-offload

PiperOrigin-RevId: 333472136

library/core/src/main/java/com/google/android/exoplayer2/audio/DefaultAudioSink.java

@@ -336,6 +336,7 @@ public final class DefaultAudioSink implements AudioSink {
   private boolean tunneling;
   private boolean tunneling;
   private long lastFeedElapsedRealtimeMs;
   private long lastFeedElapsedRealtimeMs;
   private boolean offloadDisabledUntilNextConfiguration;
   private boolean offloadDisabledUntilNextConfiguration;
+  private boolean isWaitingForOffloadEndOfStreamHandled;
 
 
   /**
   /**
    * Creates a new default audio sink.
    * Creates a new default audio sink.
@@ -712,6 +713,7 @@ public final class DefaultAudioSink implements AudioSink {
           audioTrack.setOffloadEndOfStream();
           audioTrack.setOffloadEndOfStream();
           audioTrack.setOffloadDelayPadding(
           audioTrack.setOffloadDelayPadding(
               configuration.inputFormat.encoderDelay, configuration.inputFormat.encoderPadding);
               configuration.inputFormat.encoderDelay, configuration.inputFormat.encoderPadding);
+          isWaitingForOffloadEndOfStreamHandled = true;
         }
         }
       }
       }
       // Re-apply playback parameters.
       // Re-apply playback parameters.
@@ -932,14 +934,27 @@ public final class DefaultAudioSink implements AudioSink {
       throw new WriteException(bytesWritten);
       throw new WriteException(bytesWritten);
     }
     }
 
 
+    if (isOffloadedPlayback(audioTrack)) {
+      // After calling AudioTrack.setOffloadEndOfStream, the AudioTrack internally stops and
+      // restarts during which AudioTrack.write will return 0. This situation must be detected to
+      // prevent reporting the buffer as full even though it is not which could lead ExoPlayer to
+      // sleep forever waiting for a onDataRequest that will never come.
+      if (writtenEncodedFrames > 0) {
+        isWaitingForOffloadEndOfStreamHandled = false;
+      }
+
+      // Consider the offload buffer as full if the AudioTrack is playing and AudioTrack.write could
+      // not write all the data provided to it. This relies on the assumption that AudioTrack.write
+      // always writes as much as possible.
       if (playing
       if (playing
           && listener != null
           && listener != null
           && bytesWritten < bytesRemaining
           && bytesWritten < bytesRemaining
-        && isOffloadedPlayback(audioTrack)) {
+          && !isWaitingForOffloadEndOfStreamHandled) {
         long pendingDurationMs =
         long pendingDurationMs =
             audioTrackPositionTracker.getPendingBufferDurationMs(writtenEncodedFrames);
             audioTrackPositionTracker.getPendingBufferDurationMs(writtenEncodedFrames);
         listener.onOffloadBufferFull(pendingDurationMs);
         listener.onOffloadBufferFull(pendingDurationMs);
       }
       }
+    }
 
 
     if (configuration.outputMode == OUTPUT_MODE_PCM) {
     if (configuration.outputMode == OUTPUT_MODE_PCM) {
       writtenPcmBytes += bytesWritten;
       writtenPcmBytes += bytesWritten;
@@ -1221,6 +1236,7 @@ public final class DefaultAudioSink implements AudioSink {
     submittedEncodedFrames = 0;
     submittedEncodedFrames = 0;
     writtenPcmBytes = 0;
     writtenPcmBytes = 0;
     writtenEncodedFrames = 0;
     writtenEncodedFrames = 0;
+    isWaitingForOffloadEndOfStreamHandled = false;
     framesPerEncodedSample = 0;
     framesPerEncodedSample = 0;
     mediaPositionParameters =
     mediaPositionParameters =
         new MediaPositionParameters(
         new MediaPositionParameters(