/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-

 *

 * 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 "nsContentUtils.h"

#include "nsWyciwyg.h"

#include "nsWyciwygChannel.h"

#include "nsWyciwygProtocolHandler.h"

#include "nsNetCID.h"

#include "nsServiceManagerUtils.h"

#include "plstr.h"

#include "nsIObserverService.h"

#include "nsIURI.h"

#include "nsIURIMutator.h"

#include "mozilla/net/NeckoChild.h"

using namespace mozilla::net;

#include "mozilla/net/WyciwygChannelChild.h"

////////////////////////////////////////////////////////////////////////////////

nsWyciwygProtocolHandler::nsWyciwygProtocolHandler()

{

  LOG(("Creating nsWyciwygProtocolHandler [this=%p].\n", this));

}

nsWyciwygProtocolHandler::~nsWyciwygProtocolHandler()

{

  LOG(("Deleting nsWyciwygProtocolHandler [this=%p]\n", this));

}

NS_IMPL_ISUPPORTS(nsWyciwygProtocolHandler,

                  nsIProtocolHandler)

////////////////////////////////////////////////////////////////////////////////

// nsIProtocolHandler methods:

////////////////////////////////////////////////////////////////////////////////

NS_IMETHODIMP

nsWyciwygProtocolHandler::GetScheme(nsACString &result)

{

  result = "wyciwyg";

  return NS_OK;

}

NS_IMETHODIMP

nsWyciwygProtocolHandler::GetDefaultPort(int32_t *result)

{

  return NS_ERROR_NOT_AVAILABLE;

}

NS_IMETHODIMP

nsWyciwygProtocolHandler::AllowPort(int32_t port, const char *scheme, bool *_retval)

{

  // don't override anything.

  *_retval = false;

  return NS_OK;

}

NS_IMETHODIMP

nsWyciwygProtocolHandler::NewURI(const nsACString &aSpec,

                                 const char *aCharset, // ignored

                                 nsIURI *aBaseURI,

                                 nsIURI **result)

{

  return NS_MutateURI(NS_SIMPLEURIMUTATOR_CONTRACTID)

           .SetSpec(aSpec)

           .Finalize(result);

}

NS_IMETHODIMP

nsWyciwygProtocolHandler::NewChannel2(nsIURI* url,

                                      nsILoadInfo* aLoadInfo,

                                      nsIChannel** result)

{

  if (mozilla::net::IsNeckoChild())

    mozilla::net::NeckoChild::InitNeckoChild();

  NS_ENSURE_ARG_POINTER(url);

  nsresult rv;

  nsCOMPtr<nsIWyciwygChannel> channel;

  if (IsNeckoChild()) {

    NS_ENSURE_TRUE(gNeckoChild != nullptr, NS_ERROR_FAILURE);

    ContentChild* cc = static_cast<ContentChild*>(gNeckoChild->Manager());

    if (cc->IsShuttingDown()) {

      return NS_ERROR_FAILURE;

    }

    nsCOMPtr<nsIEventTarget> target =

      nsContentUtils::GetEventTargetByLoadInfo(aLoadInfo,

                                               mozilla::TaskCategory::Other);

    WyciwygChannelChild *wcc = new WyciwygChannelChild(target);

    if (!wcc)

      return NS_ERROR_OUT_OF_MEMORY;

    channel = wcc;

    rv = wcc->Init(url);

    if (NS_FAILED(rv))

      PWyciwygChannelChild::Send__delete__(wcc);

  } else

  {

    // If original channel used https, make sure PSM is initialized

    // (this may be first channel to load during a session restore)

    nsAutoCString path;

    rv = url->GetPathQueryRef(path);

    NS_ENSURE_SUCCESS(rv, rv);

    int32_t slashIndex = path.FindChar('/', 2);

    if (slashIndex == kNotFound)

      return NS_ERROR_FAILURE;

    if (path.Length() < (uint32_t)slashIndex + 1 + 5)

      return NS_ERROR_FAILURE;

    if (!PL_strncasecmp(path.get() + slashIndex + 1, "https", 5))

      net_EnsurePSMInit();

    nsWyciwygChannel *wc = new nsWyciwygChannel();

    channel = wc;

    rv = wc->Init(url);

  }

  if (NS_FAILED(rv))

    return rv;

  // set the loadInfo on the new channel

  rv = channel->SetLoadInfo(aLoadInfo);

  if (NS_FAILED(rv)) {

      return rv;

  }

  channel.forget(result);

  return NS_OK;

}

NS_IMETHODIMP

nsWyciwygProtocolHandler::NewChannel(nsIURI* url, nsIChannel* *result)

{

  return NewChannel2(url, nullptr, result);

}

NS_IMETHODIMP

nsWyciwygProtocolHandler::GetProtocolFlags(uint32_t *result)

{

  // Should this be an an nsINestedURI?  We don't really want random webpages

  // loading these URIs...

  // Note that using URI_INHERITS_SECURITY_CONTEXT here is OK -- untrusted code

  // is not allowed to link to wyciwyg URIs and users shouldn't be able to get

  // at them, and nsDocShell::InternalLoad forbids non-history loads of these

  // URIs.  And when loading from history we end up using the principal from

  // the history entry, which we put there ourselves, so all is ok.

  *result = URI_NORELATIVE | URI_NOAUTH | URI_DANGEROUS_TO_LOAD |

    URI_INHERITS_SECURITY_CONTEXT;

  return NS_OK;

}