1 files changed, 140 insertions, 0 deletions
diff --git a/widget/android/nsAndroidProtocolHandler.cpp b/widget/android/nsAndroidProtocolHandler.cpp
new file mode 100644
index 0000000000..6453db0df6
--- /dev/null
+++ b/widget/android/nsAndroidProtocolHandler.cpp
@@ -0,0 +1,140 @@
+/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim:set ts=4 sw=2 sts=2 et cin: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "nsAndroidProtocolHandler.h"
+#include "nsCOMPtr.h"
+#include "nsIChannel.h"
+#include "android/log.h"
+#include "nsBaseChannel.h"
+#include "AndroidBridge.h"
+#include "mozilla/java/GeckoAppShellWrappers.h"
+
+using namespace mozilla;
+
+class AndroidInputStream : public nsIInputStream {
+ public:
+  explicit AndroidInputStream(jni::Object::Param connection) {
+    mBridgeInputStream = java::GeckoAppShell::CreateInputStream(connection);
+    mBridgeChannel = AndroidBridge::ChannelCreate(mBridgeInputStream);
+  }
+
+ private:
+  virtual ~AndroidInputStream() {}
+
+ public:
+  NS_DECL_THREADSAFE_ISUPPORTS
+  NS_DECL_NSIINPUTSTREAM
+
+ private:
+  jni::Object::GlobalRef mBridgeInputStream;
+  jni::Object::GlobalRef mBridgeChannel;
+};
+
+NS_IMPL_ISUPPORTS(AndroidInputStream, nsIInputStream)
+
+NS_IMETHODIMP AndroidInputStream::Close(void) {
+  AndroidBridge::InputStreamClose(mBridgeInputStream);
+  return NS_OK;
+}
+
+NS_IMETHODIMP AndroidInputStream::Available(uint64_t* _retval) {
+  *_retval = AndroidBridge::InputStreamAvailable(mBridgeInputStream);
+  return NS_OK;
+}
+
+NS_IMETHODIMP AndroidInputStream::Read(char* aBuf, uint32_t aCount,
+                                       uint32_t* _retval) {
+  return AndroidBridge::InputStreamRead(mBridgeChannel, aBuf, aCount, _retval);
+}
+
+NS_IMETHODIMP AndroidInputStream::ReadSegments(nsWriteSegmentFun aWriter,
+                                               void* aClosure, uint32_t aCount,
+                                               uint32_t* _retval) {
+  return NS_ERROR_NOT_IMPLEMENTED;
+}
+
+NS_IMETHODIMP AndroidInputStream::IsNonBlocking(bool* _retval) {
+  *_retval = false;
+  return NS_OK;
+}
+
+class AndroidChannel : public nsBaseChannel {
+ private:
+  AndroidChannel(nsIURI* aURI, jni::Object::Param aConnection) {
+    mConnection = aConnection;
+    SetURI(aURI);
+
+    auto type = java::GeckoAppShell::ConnectionGetMimeType(mConnection);
+    if (type) {
+      SetContentType(type->ToCString());
+    }
+  }
+
+ public:
+  static AndroidChannel* CreateChannel(nsIURI* aURI) {
+    nsCString spec;
+    aURI->GetSpec(spec);
+
+    auto connection = java::GeckoAppShell::GetConnection(spec);
+    return connection ? new AndroidChannel(aURI, connection) : nullptr;
+  }
+
+  virtual ~AndroidChannel() {}
+
+  virtual nsresult OpenContentStream(bool async, nsIInputStream** result,
+                                     nsIChannel** channel) {
+    nsCOMPtr<nsIInputStream> stream = new AndroidInputStream(mConnection);
+    NS_ADDREF(*result = stream);
+    return NS_OK;
+  }
+
+ private:
+  jni::Object::GlobalRef mConnection;
+};
+
+NS_IMPL_ISUPPORTS(nsAndroidProtocolHandler, nsIProtocolHandler,
+                  nsISupportsWeakReference)
+
+NS_IMETHODIMP
+nsAndroidProtocolHandler::GetScheme(nsACString& result) {
+  result.AssignLiteral("android");
+  return NS_OK;
+}
+
+NS_IMETHODIMP
+nsAndroidProtocolHandler::GetDefaultPort(int32_t* result) {
+  *result = -1;  // no port for android: URLs
+  return NS_OK;
+}
+
+NS_IMETHODIMP
+nsAndroidProtocolHandler::AllowPort(int32_t port, const char* scheme,
+                                    bool* _retval) {
+  // don't override anything.
+  *_retval = false;
+  return NS_OK;
+}
+
+NS_IMETHODIMP
+nsAndroidProtocolHandler::GetProtocolFlags(uint32_t* result) {
+  *result = URI_STD | URI_IS_UI_RESOURCE | URI_IS_LOCAL_RESOURCE |
+            URI_NORELATIVE | URI_DANGEROUS_TO_LOAD;
+  return NS_OK;
+}
+
+NS_IMETHODIMP
+nsAndroidProtocolHandler::NewChannel(nsIURI* aURI, nsILoadInfo* aLoadInfo,
+                                     nsIChannel** aResult) {
+  nsCOMPtr<nsIChannel> channel = AndroidChannel::CreateChannel(aURI);
+  if (!channel) return NS_ERROR_FAILURE;
+
+  // set the loadInfo on the new channel
+  nsresult rv = channel->SetLoadInfo(aLoadInfo);
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  NS_ADDREF(*aResult = channel);
+  return NS_OK;
+}