1 files changed, 254 insertions, 0 deletions
diff --git a/netwerk/protocol/ftp/nsFtpConnectionThread.h b/netwerk/protocol/ftp/nsFtpConnectionThread.h
new file mode 100644
index 0000000000..ec1cf999c7
--- /dev/null
+++ b/netwerk/protocol/ftp/nsFtpConnectionThread.h
@@ -0,0 +1,254 @@
+/* -*- 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/. */
+
+#ifndef __nsFtpConnectionThread__h_
+#define __nsFtpConnectionThread__h_
+
+#include "nsBaseContentStream.h"
+
+#include "nsString.h"
+#include "nsCOMPtr.h"
+#include "nsIAsyncInputStream.h"
+#include "nsITransport.h"
+#include "mozilla/net/DNS.h"
+#include "nsFtpControlConnection.h"
+#include "nsIProtocolProxyCallback.h"
+
+// ftp server types
+#define FTP_GENERIC_TYPE 0
+#define FTP_UNIX_TYPE 1
+#define FTP_VMS_TYPE 8
+#define FTP_NT_TYPE 9
+#define FTP_OS2_TYPE 11
+
+// ftp states
+typedef enum _FTP_STATE {
+  ///////////////////////
+  //// Internal states
+  FTP_INIT,
+  FTP_COMMAND_CONNECT,
+  FTP_READ_BUF,
+  FTP_ERROR,
+  FTP_COMPLETE,
+
+  ///////////////////////
+  //// Command channel connection setup states
+  FTP_S_USER,
+  FTP_R_USER,
+  FTP_S_PASS,
+  FTP_R_PASS,
+  FTP_S_SYST,
+  FTP_R_SYST,
+  FTP_S_ACCT,
+  FTP_R_ACCT,
+  FTP_S_TYPE,
+  FTP_R_TYPE,
+  FTP_S_CWD,
+  FTP_R_CWD,
+  FTP_S_SIZE,
+  FTP_R_SIZE,
+  FTP_S_MDTM,
+  FTP_R_MDTM,
+  FTP_S_REST,
+  FTP_R_REST,
+  FTP_S_RETR,
+  FTP_R_RETR,
+  FTP_S_STOR,
+  FTP_R_STOR,
+  FTP_S_LIST,
+  FTP_R_LIST,
+  FTP_S_PASV,
+  FTP_R_PASV,
+  FTP_S_PWD,
+  FTP_R_PWD,
+  FTP_S_FEAT,
+  FTP_R_FEAT,
+  FTP_S_OPTS,
+  FTP_R_OPTS
+} FTP_STATE;
+
+// higher level ftp actions
+typedef enum _FTP_ACTION { GET, PUT } FTP_ACTION;
+
+class nsFtpChannel;
+class nsICancelable;
+class nsIProxyInfo;
+class nsIStreamListener;
+
+// The nsFtpState object is the content stream for the channel.  It implements
+// nsIInputStreamCallback, so it can read data from the control connection.  It
+// implements nsITransportEventSink so it can mix status events from both the
+// control connection and the data connection.
+
+class nsFtpState final : public nsBaseContentStream,
+                         public nsIInputStreamCallback,
+                         public nsITransportEventSink,
+                         public nsIRequestObserver,
+                         public nsFtpControlConnectionListener,
+                         public nsIProtocolProxyCallback {
+ public:
+  NS_DECL_ISUPPORTS_INHERITED
+  NS_DECL_NSIINPUTSTREAMCALLBACK
+  NS_DECL_NSITRANSPORTEVENTSINK
+  NS_DECL_NSIREQUESTOBSERVER
+  NS_DECL_NSIPROTOCOLPROXYCALLBACK
+
+  // Override input stream methods:
+  NS_IMETHOD CloseWithStatus(nsresult status) override;
+  NS_IMETHOD Available(uint64_t* result) override;
+  NS_IMETHOD ReadSegments(nsWriteSegmentFun fun, void* closure, uint32_t count,
+                          uint32_t* result) override;
+
+  // nsFtpControlConnectionListener methods:
+  virtual void OnControlDataAvailable(const char* data,
+                                      uint32_t dataLen) override;
+  virtual void OnControlError(nsresult status) override;
+
+  nsFtpState();
+  nsresult Init(nsFtpChannel* channel);
+
+ protected:
+  // Notification from nsBaseContentStream::AsyncWait
+  virtual void OnCallbackPending() override;
+
+ private:
+  virtual ~nsFtpState();
+
+  ///////////////////////////////////
+  // BEGIN: STATE METHODS
+  nsresult S_user();
+  FTP_STATE R_user();
+  nsresult S_pass();
+  FTP_STATE R_pass();
+  nsresult S_syst();
+  FTP_STATE R_syst();
+  nsresult S_acct();
+  FTP_STATE R_acct();
+
+  nsresult S_type();
+  FTP_STATE R_type();
+  nsresult S_cwd();
+  FTP_STATE R_cwd();
+
+  nsresult S_size();
+  FTP_STATE R_size();
+  nsresult S_mdtm();
+  FTP_STATE R_mdtm();
+  nsresult S_list();
+  FTP_STATE R_list();
+
+  nsresult S_rest();
+  FTP_STATE R_rest();
+  nsresult S_retr();
+  FTP_STATE R_retr();
+  nsresult S_stor();
+  FTP_STATE R_stor();
+  nsresult S_pasv();
+  FTP_STATE R_pasv();
+  nsresult S_pwd();
+  FTP_STATE R_pwd();
+  nsresult S_feat();
+  FTP_STATE R_feat();
+  nsresult S_opts();
+  FTP_STATE R_opts();
+  // END: STATE METHODS
+  ///////////////////////////////////
+
+  // internal methods
+  void MoveToNextState(FTP_STATE nextState);
+  nsresult Process();
+
+  void KillControlConnection();
+  nsresult StopProcessing();
+  nsresult EstablishControlConnection();
+  nsresult SendFTPCommand(const nsACString& command);
+  void ConvertFilespecToVMS(nsCString& fileSpec);
+  void ConvertDirspecToVMS(nsCString& fileSpec);
+  void ConvertDirspecFromVMS(nsCString& fileSpec);
+  nsresult BuildStreamConverter(nsIStreamListener** convertStreamListener);
+  nsresult SetContentType();
+
+  /**
+   * This method is called to kick-off the FTP state machine.  mState is
+   * reset to FTP_COMMAND_CONNECT, and the FTP state machine progresses from
+   * there.  This method is initially called (indirectly) from the channel's
+   * AsyncOpen implementation.
+   */
+  void Connect();
+
+  ///////////////////////////////////
+  // Private members
+
+  nsCOMPtr<nsIProxiedProtocolHandler> mHandler;  // Ref to gFtpHandler
+
+  // ****** state machine vars
+  FTP_STATE mState;        // the current state
+  FTP_STATE mNextState;    // the next state
+  bool mKeepRunning;       // thread event loop boolean
+  int32_t mResponseCode;   // the last command response code
+  nsCString mResponseMsg;  // the last command response text
+
+  // ****** channel/transport/stream vars
+  RefPtr<nsFtpControlConnection>
+      mControlConnection;  // cacheable control connection (owns mCPipe)
+  bool mReceivedControlData;
+  bool mTryingCachedControl;  // retrying the password
+  bool mRETRFailed;           // Did we already try a RETR and it failed?
+  uint64_t mFileSize;
+  nsCString mModTime;
+
+  // ****** consumer vars
+  RefPtr<nsFtpChannel>
+      mChannel;  // our owning FTP channel we pass through our events
+  nsCOMPtr<nsIProxyInfo> mProxyInfo;
+
+  // ****** connection cache vars
+  int32_t mServerType;  // What kind of server are we talking to
+
+  // ****** protocol interpretation related state vars
+  nsString mUsername;       // username
+  nsString mPassword;       // password
+  FTP_ACTION mAction;       // the higher level action (GET/PUT)
+  bool mAnonymous;          // try connecting anonymous (default)
+  bool mRetryPass;          // retrying the password
+  bool mStorReplyReceived;  // FALSE if waiting for STOR
+                            // completion status from server
+  bool mRlist1xxReceived;   // TRUE if we have received a LIST 1xx
+                            // response from the server
+  bool mRretr1xxReceived;   // TRUE if we have received a RETR 1xx
+                            // response from the server
+  bool mRstor1xxReceived;   // TRUE if we have received a STOR 1xx
+                            // response from the server
+  nsresult mInternalError;  // represents internal state errors
+  bool mReconnectAndLoginAgain;
+  bool mCacheConnection;
+
+  // ****** URI vars
+  int32_t mPort;       // the port to connect to
+  nsString mFilename;  // url filename (if any)
+  nsCString mPath;     // the url's path
+  nsCString mPwd;      // login Path
+
+  // ****** other vars
+  nsCOMPtr<nsITransport> mDataTransport;
+  nsCOMPtr<nsIAsyncInputStream> mDataStream;
+  nsCOMPtr<nsIRequest> mUploadRequest;
+  bool mAddressChecked;
+  bool mServerIsIPv6;
+  bool mUseUTF8;
+
+  mozilla::net::NetAddr mServerAddress;
+
+  // ***** control read gvars
+  nsresult mControlStatus;
+  nsCString mControlReadCarryOverBuf;
+
+  nsCString mSuppliedEntityID;
+
+  nsCOMPtr<nsICancelable> mProxyRequest;
+  bool mDeferredCallbackPending;
+};
+
+#endif  //__nsFtpConnectionThread__h_