[Cronet] Implement methods to get response headers in HttpURLConnection.

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

 // Copyright 2014 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 android.util.Pair;
+
 import org.chromium.net.ExtendedResponseInfo;
 import org.chromium.net.ResponseInfo;
 import org.chromium.net.UrlRequest;
 import org.chromium.net.UrlRequestContext;
 import org.chromium.net.UrlRequestException;
 import org.chromium.net.UrlRequestListener;
 
 import java.io.FileNotFoundException;
 import java.io.IOException;
 import java.io.InputStream;
 import java.net.HttpURLConnection;
 import java.net.MalformedURLException;
 import java.net.URL;
 import java.nio.ByteBuffer;
+import java.util.List;
+import java.util.Map;
 
 /**
  * An implementation of HttpURLConnection that uses Cronet to send requests and
  * receive response. This class inherits a {@code connected} field from the
  * superclass. That field indicates whether a connection has ever been
  * attempted.
  */
 public class CronetHttpURLConnection extends HttpURLConnection {
     private final UrlRequestContext mUrlRequestContext;
     private final MessageLoop mMessageLoop;
@@ skipping 64 matching lines @@
     /**
      * Returns the response code returned by the remote HTTP server.
      */
     @Override
     public int getResponseCode() throws IOException {
         connect();
         return mResponseInfo.getHttpStatusCode();
     }
 
     /**
+     * Returns an unmodifiable map of the response-header fields and values.
+     */
+    @Override
+    public Map<String, List<String>> getHeaderFields() {
+        try {
+            connect();
+        } catch (IOException e) {
+            // Default implementation returns an empty map on K, but returns
+            // null on L.
+            return null;
+        }
+        return mResponseInfo.getAllHeaders();
+    }
+
+    /**
+     * Returns the value of the named header field. If called on a connection
+     * that sets the same header multiple times with possibly different values,
+     * only the last value is returned.
+     */
+    @Override
+    public final String getHeaderField(String fieldName) {
+        try {
+            connect();
+        } catch (IOException e) {
+            return null;
+        }
+        Map<String, List<String>> map = mResponseInfo.getAllHeaders();
+        if (!map.containsKey(fieldName)) {
+            return null;
+        }
+        List<String> values = map.get(fieldName);
+        return values.get(values.size() - 1);
+    }
+
+    /**
+     * Returns the name of the header field at the given position pos or null

[mmenke, 2015/01/16 19:43:13]

nit:  Suggest a comma after pos.

[xunjieli, 2015/01/16 20:01:12]

Done.

+     * if there are fewer than pos fields.
+     */
+    @Override
+    public final String getHeaderFieldKey(int pos) {
+        try {
+            connect();
+        } catch (IOException e) {
+            return null;
+        }
+        List<Pair<String, String>> headers =
+                mResponseInfo.getAllHeadersAsList();
+        if (pos >= headers.size()) {
+            return null;
+        }
+        return headers.get(pos).first;
+    }
+
+    /**
+     * Returns the header value at the field position pos or null if the header
+     * has fewer than pos fields.
+     */
+    @Override
+    public final String getHeaderField(int pos) {
+        try {
+            connect();
+        } catch (IOException e) {
+            return null;
+        }
+        List<Pair<String, String>> headers =
+                mResponseInfo.getAllHeadersAsList();
+        if (pos >= headers.size()) {
+            return null;
+        }
+        return headers.get(pos).second;
+    }
+
+    /**
      * Returns an InputStream for reading data from the resource pointed by this
      * URLConnection.
      * @throws FileNotFoundException if http response code is equal or greater
      *             than {@link HTTP_BAD_REQUEST}.
      * @throws IOException If the request gets a network error or HTTP error
      *             status code, or if the caller tried to read the response body
      *             of a redirect when redirects are disabled.
      */
     @Override
     public InputStream getInputStream() throws IOException {
@@ skipping 143 matching lines @@
      */
     private void checkHasResponse() throws IOException {
         if (mException != null) {
             throw mException;
         } else if (mResponseInfo == null) {
             throw new NullPointerException(
                     "Response info is null when there is no exception.");
         }
     }
 }