Add scheme to user-provided host if necessary

Ajay Kannan · Ajay Kannan · commit e1a50bc0063c · 2015-10-08T09:43:05.000-07:00
diff --git a/gcloud-java-datastore/src/main/java/com/google/gcloud/spi/DefaultDatastoreRpc.java b/gcloud-java-datastore/src/main/java/com/google/gcloud/spi/DefaultDatastoreRpc.java
@@ -40,6 +40,10 @@
 import org.json.JSONObject;
 import org.json.JSONTokener;
 
+import java.net.InetAddress;
+import java.net.MalformedURLException;
+import java.net.URL;
+import java.net.UnknownHostException;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -62,14 +66,48 @@ public class DefaultDatastoreRpc implements DatastoreRpc {
   }
 
   public DefaultDatastoreRpc(DatastoreOptions options) {
+    String normalizedHost = normalizeHost(options.host());
     client = DatastoreFactory.get().create(
         new Builder()
             .dataset(options.projectId())
-            .host(options.host())
+            .host(normalizedHost)
             .initializer(options.httpRequestInitializer())
             .build());
   }
 
+  private static String normalizeHost(String host) {
+    host = host.toLowerCase();
+    if (includesScheme(host)) {
+      if (host.startsWith("https://") && isLocalHost(host)) {
+        throw new IllegalArgumentException(
+            "\"https\" is not supported for localhost.  Use \"http\" instead.");
+      } else {
+        return host;
+      }
+    }
+    return "http://" + host;
+  }
+
+  private static boolean isLocalHost(String host) {
+    if (host != null) {
+      try {
+        String normalizedHost = host;
+        if (!includesScheme(normalizedHost)) {
+          normalizedHost = "http://" + normalizedHost;
+        }
+        InetAddress hostAddr = InetAddress.getByName(new URL(normalizedHost).getHost());
+        return hostAddr.isAnyLocalAddress() || hostAddr.isLoopbackAddress();
+      } catch (UnknownHostException | MalformedURLException e) {
+        // ignore
+      }
+    }
+    return false;
+  }
+
+  private static boolean includesScheme(String url) {
+    return url.startsWith("http://") || url.startsWith("https://");
+  }
+
   private static DatastoreRpcException translate(DatastoreException exception) {
     String message = exception.getMessage();
     String reasonStr = "";
