widget/android/nsAndroidProtocolHandler.cpp
author Doug Thayer <dothayer@mozilla.com>
Fri, 01 Mar 2019 18:29:09 +0000
changeset 520116 8b3fe0426ffc1b3a2ad044ef6cdde6c4f736f8e2
parent 517975 ec2056010064217db0dc48b9a70a7174eb3d254a
child 530873 e1993a1f09ac53cd1a04fdf6a87f8cad8e44f73e
permissions -rw-r--r--
Bug 1442694 - Fix failures due to removing selected tab r=Gijs This adds test which reproduce the failure as well as the fix. Essentially, if we hit the edited case in SessionStore with `tab` equal to `tabbrowser.tabs[t]`, we remove the tab and then try to pin it, which obviously blows up. Note: the additional method in SessionStore.jsm was largely to get around complexity requirements inside restoreWindow. Cleaner solutions welcome. Differential Revision: https://phabricator.services.mozilla.com/D21383

/* -*- 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 "nsIIOService.h"
#include "nsIStandardURL.h"
#include "nsIURL.h"
#include "nsIURIMutator.h"
#include "android/log.h"
#include "nsBaseChannel.h"
#include "AndroidBridge.h"
#include "GeneratedJNIWrappers.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::NewURI(const nsACString &aSpec, const char *aCharset,
                                 nsIURI *aBaseURI, nsIURI **result) {
  nsCOMPtr<nsIURI> base(aBaseURI);
  return NS_MutateURI(NS_STANDARDURLMUTATOR_CONTRACTID)
      .Apply(NS_MutatorMethod(&nsIStandardURLMutator::Init,
                              nsIStandardURL::URLTYPE_STANDARD, -1,
                              nsCString(aSpec), aCharset, base, nullptr))
      .Finalize(result);
}

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