1 files changed, 83 insertions, 0 deletions
diff --git a/netwerk/protocol/ftp/nsFtpControlConnection.h b/netwerk/protocol/ftp/nsFtpControlConnection.h
new file mode 100644
index 0000000000..c29e23eeed
--- /dev/null
+++ b/netwerk/protocol/ftp/nsFtpControlConnection.h
@@ -0,0 +1,83 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim:set et ts=4 sts=2 sw=2 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/. */
+
+#ifndef nsFtpControlConnection_h___
+#define nsFtpControlConnection_h___
+
+#include "nsCOMPtr.h"
+
+#include "nsISocketTransport.h"
+#include "nsIAsyncInputStream.h"
+#include "nsString.h"
+#include "mozilla/Attributes.h"
+
+class nsIOutputStream;
+class nsIProxyInfo;
+class nsITransportEventSink;
+
+class nsFtpControlConnectionListener : public nsISupports {
+ public:
+  /**
+   * Called when a chunk of data arrives on the control connection.
+   * @param data
+   *        The new data or null if an error occurred.
+   * @param dataLen
+   *        The data length in bytes.
+   */
+  virtual void OnControlDataAvailable(const char* data, uint32_t dataLen) = 0;
+
+  /**
+   * Called when an error occurs on the control connection.
+   * @param status
+   *        A failure code providing more info about the error.
+   */
+  virtual void OnControlError(nsresult status) = 0;
+};
+
+class nsFtpControlConnection final : public nsIInputStreamCallback {
+  ~nsFtpControlConnection();
+
+ public:
+  NS_DECL_ISUPPORTS
+  NS_DECL_NSIINPUTSTREAMCALLBACK
+
+  nsFtpControlConnection(const nsACString& host, uint32_t port);
+
+  nsresult Connect(nsIProxyInfo* proxyInfo, nsITransportEventSink* eventSink);
+  nsresult Disconnect(nsresult status);
+  nsresult Write(const nsACString& command);
+
+  bool IsAlive();
+
+  nsITransport* Transport() { return mSocket; }
+
+  /**
+   * Call this function to be notified asynchronously when there is data
+   * available for the socket.  The listener passed to this method replaces
+   * any existing listener, and the listener can be null to disconnect the
+   * previous listener.
+   */
+  nsresult WaitData(nsFtpControlConnectionListener* listener);
+
+  uint32_t mServerType;  // what kind of server is it.
+  nsString mPassword;
+  int32_t mSuspendedWrite;
+  nsCString mPwd;
+  uint32_t mSessionId;
+  bool mUseUTF8;
+
+ private:
+  nsCString mHost;
+  uint32_t mPort;
+
+  nsCOMPtr<nsISocketTransport> mSocket;
+  nsCOMPtr<nsIOutputStream> mSocketOutput;
+  nsCOMPtr<nsIAsyncInputStream> mSocketInput;
+
+  RefPtr<nsFtpControlConnectionListener> mListener;
+};
+
+#endif