nsAndroidProtocolHandler.cpp

/* -*- 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;
}