[Cronet] Check whether request is done before spinning up message loop

components/cronet/android/java/src/org/chromium/net/urlconnection/CronetFixedModeOutputStream.java

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package org.chromium.net.urlconnection;
 
 import org.chromium.base.VisibleForTesting;
 import org.chromium.net.UploadDataProvider;
 import org.chromium.net.UploadDataSink;
 
@@ skipping 55 matching lines @@
         mContentLength = contentLength;
         int bufferSize = (int) Math.min(mContentLength, sDefaultBufferLength);
         mBuffer = ByteBuffer.allocate(bufferSize);
         mConnection = connection;
         mMessageLoop = messageLoop;
         mBytesWritten = 0;
     }
 
     @Override
     public void write(int oneByte) throws IOException {
+        checkNotClosed();
         checkNotExceedContentLength(1);
         ensureBufferHasRemaining();
         mBuffer.put((byte) oneByte);
         mBytesWritten++;
         uploadIfComplete();
     }
 
     @Override
     public void write(byte[] buffer, int offset, int count) throws IOException {
+        checkNotClosed();
         if (buffer.length - offset < count || offset < 0 || count < 0) {
             throw new IndexOutOfBoundsException();
         }
         checkNotExceedContentLength(count);
         int toSend = count;
         while (toSend > 0) {
             ensureBufferHasRemaining();
             int sent = Math.min(toSend, mBuffer.remaining());
             mBuffer.put(buffer, offset + count - toSend, sent);
             toSend -= sent;
@@ skipping 24 matching lines @@
             uploadBufferInternal();
         }
     }
 
     /**
      * Helper function to upload {@code mBuffer} to the native stack. This
      * function blocks until {@code mBuffer} is consumed and there is space to
      * write more data.
      */
     private void uploadBufferInternal() throws IOException {
+        checkNotClosed();
         mBuffer.flip();
         mMessageLoop.loop();
     }
 
     /**
      * Throws {@link java.net.ProtocolException} if adding {@code numBytes} will
      * exceed content length.
      */
     private void checkNotExceedContentLength(int numBytes) throws ProtocolException {
         if (mBytesWritten + numBytes > mContentLength) {
             throw new ProtocolException("expected "
                     + (mContentLength - mBytesWritten) + " bytes but received "
                     + numBytes);
         }
     }
 
-    // TODO(xunjieli): implement close().
-
     // Below are CronetOutputStream implementations:
 
     @Override
     void setConnected() throws IOException {
         // Do nothing.
     }
 
     @Override
     void checkReceivedEnoughContent() throws IOException {
         if (mBytesWritten < mContentLength) {
@@ skipping 38 matching lines @@
     }
 
     /**
      * Sets the default buffer length for use in tests.
      */
     @VisibleForTesting
     static void setDefaultBufferLengthForTesting(int length) {
         sDefaultBufferLength = length;
     }
 }