add support for range requests

Change-Id: I56b546d55167c61f315b3cfa3ad9eee54d85defd

Author: koush

AndroidAsync/src/com/koushikdutta/async/AsyncNetworkSocket.java

@@ -81,7 +81,7 @@ public void run() {
             handleRemaining(list.remaining());
         }
         catch (IOException e) {
-            close();
+            closeInternal();
             reportEndPending(e);
             reportClose(e);
         }
@@ -122,7 +122,7 @@ public void run() {
             handleRemaining(b.remaining());
         }
         catch (IOException ex) {
-            close();
+            closeInternal();
             reportEndPending(ex);
             reportClose(ex);
         }

AndroidAsync/src/com/koushikdutta/async/DataEmitterReader.java

@@ -11,7 +11,7 @@ public void read(int count, DataCallback callback) {
         assert mPendingRead == null;
         mPendingReadLength = count;
         mPendingRead = callback;
-        mPendingData = new ByteBufferList();
+        mPendingData.recycle();
     }
 
     private boolean handlePendingData(DataEmitter emitter) {

AndroidAsync/src/com/koushikdutta/async/Util.java

@@ -40,6 +40,10 @@ public static void emitAllData(DataEmitter emitter, ByteBufferList list) {
     }
 
     public static void pump(final InputStream is, final DataSink ds, final CompletedCallback callback) {
+        pump(is, Integer.MAX_VALUE, ds, callback);
+    }
+
+    public static void pump(final InputStream is, final int max, final DataSink ds, final CompletedCallback callback) {
         final CompletedCallback wrapper = new CompletedCallback() {
             boolean reported;
             @Override
@@ -52,6 +56,7 @@ public void onCompleted(Exception ex) {
         };
 
         final WritableCallback cb = new WritableCallback() {
+            int totalRead = 0;
             private void close() {
                 try {
                     is.close();
@@ -70,15 +75,16 @@ private void close() {
             public void onWriteable() {
                 try {
                     int remaining;
-//                    long start = System.currentTimeMillis();
                     do {
                         if (pending.remaining() == 0) {
-                            int read = is.read(buffer);
-                            if (read == -1) {
+                            int toRead = Math.min(max - totalRead, buffer.length);
+                            int read = is.read(buffer, 0, toRead);
+                            if (read == -1 || totalRead == max) {
                                 close();
                                 wrapper.onCompleted(null);
                                 return;
                             }
+                            totalRead += read;
                             pending.position(0);
                             pending.limit(read);
                         }

AndroidAsync/src/com/koushikdutta/async/http/server/AsyncHttpServer.java

@@ -393,6 +393,7 @@ public void onCompleted(Exception ex) {
     private static Hashtable<Integer, String> mCodes = new Hashtable<Integer, String>();
     static {
         mCodes.put(200, "OK");
+        mCodes.put(206, "Partial Content");
         mCodes.put(101, "Switching Protocols");
         mCodes.put(404, "Not Found");
     }

AndroidAsync/src/com/koushikdutta/async/http/server/AsyncHttpServerResponseImpl.java

@@ -1,5 +1,7 @@
 package com.koushikdutta.async.http.server;
 
+import android.text.TextUtils;
+
 import com.koushikdutta.async.*;
 import com.koushikdutta.async.callback.CompletedCallback;
 import com.koushikdutta.async.callback.WritableCallback;
@@ -11,6 +13,7 @@
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileNotFoundException;
+import java.io.IOException;
 import java.io.UnsupportedEncodingException;
 import java.nio.ByteBuffer;
 
@@ -159,18 +162,56 @@ public void send(JSONObject json) {
     }
 
     public void sendFile(File file) {
+        int start = 0;
+        int end = (int)file.length();
+
+        String range = mRequest.getHeaders().getHeaders().get("Range");
+        if (range != null) {
+            String[] parts = range.split("=");
+            if (parts.length != 2 || !"bytes".equals(parts[0])) {
+                // Requested range not satisfiable
+                responseCode(416);
+                end();
+                return;
+            }
+
+            parts = parts[1].split("-");
+            try {
+                if (parts.length > 2)
+                    throw new Exception();
+                if (!TextUtils.isEmpty(parts[0]))
+                    start = Integer.parseInt(parts[0]);
+                if (parts.length == 2 && !TextUtils.isEmpty(parts[1]))
+                    end = Integer.parseInt(parts[1]);
+                else if (start != 0)
+                    end = (int)file.length();
+                else
+                    end = Math.min((int)file.length(), start + 50000);
+
+                responseCode(206);
+                getHeaders().getHeaders().set("Content-Range", String.format("bytes %d-%d/%d", start, end - 1, file.length()));
+            }
+            catch (Exception e) {
+                responseCode(416);
+                end();
+                return;
+            }
+        }
         try {
             FileInputStream fin = new FileInputStream(file);
+            if (start != fin.skip(start))
+                throw new Exception();
             mRawHeaders.set("Content-Type", AsyncHttpServer.getContentType(file.getAbsolutePath()));
-            responseCode(200);
-            Util.pump(fin, this, new CompletedCallback() {
+            if (getHeaders().getHeaders().getStatusLine() == null)
+                responseCode(200);
+            Util.pump(fin, end - start, this, new CompletedCallback() {
                 @Override
                 public void onCompleted(Exception ex) {
                     end();
                 }
             });
         }
-        catch (FileNotFoundException e) {
+        catch (Exception e) {
             responseCode(404);
             end();
         }