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

/* vim:set ts=4 sts=4 sw=4 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 "nsFTPChannel.h"

#include "nsFtpConnectionThread.h"  // defines nsFtpState

#include "nsThreadUtils.h"

#include "mozilla/Attributes.h"

using namespace mozilla;

using namespace mozilla::net;

extern LazyLogModule gFTPLog;

// There are two transport connections established for an

// ftp connection. One is used for the command channel , and

// the other for the data channel. The command channel is the first

// connection made and is used to negotiate the second, data, channel.

// The data channel is driven by the command channel and is either

// initiated by the server (PORT command) or by the client (PASV command).

// Client initiation is the most common case and is attempted first.

//-----------------------------------------------------------------------------

NS_IMPL_ISUPPORTS_INHERITED(nsFtpChannel, nsBaseChannel, nsIUploadChannel,

                            nsIResumableChannel, nsIFTPChannel,

                            nsIProxiedChannel, nsIForcePendingChannel,

                            nsIChannelWithDivertableParentListener)

//-----------------------------------------------------------------------------

NS_IMETHODIMP

nsFtpChannel::SetUploadStream(nsIInputStream *stream,

                              const nsACString &contentType,

                              int64_t contentLength) {

  NS_ENSURE_TRUE(!Pending(), NS_ERROR_IN_PROGRESS);

  mUploadStream = stream;

  // NOTE: contentLength is intentionally ignored here.

  return NS_OK;

}

NS_IMETHODIMP

nsFtpChannel::GetUploadStream(nsIInputStream **stream) {

  NS_ENSURE_ARG_POINTER(stream);

  *stream = mUploadStream;

  NS_IF_ADDREF(*stream);

  return NS_OK;

}

//-----------------------------------------------------------------------------

NS_IMETHODIMP

nsFtpChannel::ResumeAt(uint64_t aStartPos, const nsACString &aEntityID) {

  NS_ENSURE_TRUE(!Pending(), NS_ERROR_IN_PROGRESS);

  mEntityID = aEntityID;

  mStartPos = aStartPos;

  mResumeRequested = (mStartPos || !mEntityID.IsEmpty());

  return NS_OK;

}

NS_IMETHODIMP

nsFtpChannel::GetEntityID(nsACString &entityID) {

  if (mEntityID.IsEmpty()) return NS_ERROR_NOT_RESUMABLE;

  entityID = mEntityID;

  return NS_OK;

}

//-----------------------------------------------------------------------------

NS_IMETHODIMP

nsFtpChannel::GetProxyInfo(nsIProxyInfo **aProxyInfo) {

  *aProxyInfo = ProxyInfo();

  NS_IF_ADDREF(*aProxyInfo);

  return NS_OK;

}

//-----------------------------------------------------------------------------

nsresult nsFtpChannel::OpenContentStream(bool async, nsIInputStream **result,

                                         nsIChannel **channel) {

  if (!async) return NS_ERROR_NOT_IMPLEMENTED;

  nsFtpState *state = new nsFtpState();

  if (!state) return NS_ERROR_OUT_OF_MEMORY;

  NS_ADDREF(state);

  nsresult rv = state->Init(this);

  if (NS_FAILED(rv)) {

    NS_RELEASE(state);

    return rv;

  }

  *result = state;

  return NS_OK;

}

bool nsFtpChannel::GetStatusArg(nsresult status, nsString &statusArg) {

  nsAutoCString host;

  URI()->GetHost(host);

  CopyUTF8toUTF16(host, statusArg);

  return true;

}

void nsFtpChannel::OnCallbacksChanged() { mFTPEventSink = nullptr; }

//-----------------------------------------------------------------------------

namespace {

class FTPEventSinkProxy final : public nsIFTPEventSink {

  ~FTPEventSinkProxy() {}

 public:

  explicit FTPEventSinkProxy(nsIFTPEventSink *aTarget)

      : mTarget(aTarget), mEventTarget(GetCurrentThreadEventTarget()) {}

  NS_DECL_THREADSAFE_ISUPPORTS

  NS_DECL_NSIFTPEVENTSINK

  class OnFTPControlLogRunnable : public Runnable {

   public:

    OnFTPControlLogRunnable(nsIFTPEventSink *aTarget, bool aServer,

                            const char *aMessage)

        : mozilla::Runnable("FTPEventSinkProxy::OnFTPControlLogRunnable"),

          mTarget(aTarget),

          mServer(aServer),

          mMessage(aMessage) {}

    NS_DECL_NSIRUNNABLE

   private:

    nsCOMPtr<nsIFTPEventSink> mTarget;

    bool mServer;

    nsCString mMessage;

  };

 private:

  nsCOMPtr<nsIFTPEventSink> mTarget;

  nsCOMPtr<nsIEventTarget> mEventTarget;

};

NS_IMPL_ISUPPORTS(FTPEventSinkProxy, nsIFTPEventSink)

NS_IMETHODIMP

FTPEventSinkProxy::OnFTPControlLog(bool aServer, const char *aMsg) {

  RefPtr<OnFTPControlLogRunnable> r =

      new OnFTPControlLogRunnable(mTarget, aServer, aMsg);

  return mEventTarget->Dispatch(r, NS_DISPATCH_NORMAL);

}

NS_IMETHODIMP

FTPEventSinkProxy::OnFTPControlLogRunnable::Run() {

  mTarget->OnFTPControlLog(mServer, mMessage.get());

  return NS_OK;

}

}  // namespace

void nsFtpChannel::GetFTPEventSink(nsCOMPtr<nsIFTPEventSink> &aResult) {

  if (!mFTPEventSink) {

    nsCOMPtr<nsIFTPEventSink> ftpSink;

    GetCallback(ftpSink);

    if (ftpSink) {

      mFTPEventSink = new FTPEventSinkProxy(ftpSink);

    }

  }

  aResult = mFTPEventSink;

}

NS_IMETHODIMP

nsFtpChannel::ForcePending(bool aForcePending) {

  // Set true here so IsPending will return true.

  // Required for callback diversion from child back to parent. In such cases

  // OnStopRequest can be called in the parent before callbacks are diverted

  // back from the child to the listener in the parent.

  mForcePending = aForcePending;

  return NS_OK;

}

NS_IMETHODIMP

nsFtpChannel::IsPending(bool *result) {

  *result = Pending();

  return NS_OK;

}

bool nsFtpChannel::Pending() const {

  return nsBaseChannel::Pending() || mForcePending;

}

NS_IMETHODIMP

nsFtpChannel::Suspend() {

  LOG(("nsFtpChannel::Suspend [this=%p]\n", this));

  nsresult rv = nsBaseChannel::Suspend();

  nsresult rvParentChannel = NS_OK;

  if (mParentChannel) {

    rvParentChannel = mParentChannel->SuspendMessageDiversion();

  }

  return NS_FAILED(rv) ? rv : rvParentChannel;

}

NS_IMETHODIMP

nsFtpChannel::Resume() {

  LOG(("nsFtpChannel::Resume [this=%p]\n", this));

  nsresult rv = nsBaseChannel::Resume();

  nsresult rvParentChannel = NS_OK;

  if (mParentChannel) {

    rvParentChannel = mParentChannel->ResumeMessageDiversion();

  }

  return NS_FAILED(rv) ? rv : rvParentChannel;

}

//-----------------------------------------------------------------------------

// AChannelHasDivertableParentChannelAsListener internal functions

//-----------------------------------------------------------------------------

NS_IMETHODIMP

nsFtpChannel::MessageDiversionStarted(

    ADivertableParentChannel *aParentChannel) {

  MOZ_ASSERT(!mParentChannel);

  mParentChannel = aParentChannel;

  return NS_OK;

}

NS_IMETHODIMP

nsFtpChannel::MessageDiversionStop() {

  LOG(("nsFtpChannel::MessageDiversionStop [this=%p]", this));

  MOZ_ASSERT(mParentChannel);

  mParentChannel = nullptr;

  return NS_OK;

}

NS_IMETHODIMP

nsFtpChannel::SuspendInternal() {

  LOG(("nsFtpChannel::SuspendInternal [this=%p]\n", this));

  return nsBaseChannel::Suspend();

}

NS_IMETHODIMP

nsFtpChannel::ResumeInternal() {

  LOG(("nsFtpChannel::ResumeInternal [this=%p]\n", this));

  return nsBaseChannel::Resume();

}