FTP Transaction code for new Portable FTP(Patch SET-3)

net/ftp/ftp_network_transaction.cc

Index: net/ftp/ftp_network_transaction.cc
===================================================================
--- net/ftp/ftp_network_transaction.cc	(revision 13921)
+++ net/ftp/ftp_network_transaction.cc	(working copy)
@@ -5,30 +5,46 @@
 #include "net/ftp/ftp_network_transaction.h"
 
 #include "base/compiler_specific.h"
-#include "base/logging.h"
+#include "base/string_util.h"
 #include "net/base/client_socket.h"
+#include "net/base/client_socket_factory.h"
+#include "net/base/connection_type_histograms.h"
+#include "net/base/dns_resolution_observer.h"
 #include "net/base/net_errors.h"
 #include "net/ftp/ftp_network_session.h"
+#include "net/ftp/ftp_request_info.h"
 
+
+#define CRLF "\015\012"
+
+const int kCtrlBufLen = 1024;
+
 namespace net {
 
 FtpNetworkTransaction::FtpNetworkTransaction(
     FtpNetworkSession* session, ClientSocketFactory* socket_factory)
-  :  ALLOW_THIS_IN_INITIALIZER_LIST(io_callback_(this,
-                                         &FtpNetworkTransaction::OnIOComplete)),
+    : ALLOW_THIS_IN_INITIALIZER_LIST(
+      io_callback_(this, &FtpNetworkTransaction::OnIOComplete)),
       user_callback_(NULL),
       session_(session),
       socket_factory_(socket_factory),
-      next_state_(STATE_NONE) {
+      next_state_(STATE_NONE),
+      command_sent_(COMMAND_NONE),
+      read_data_buf_(NULL),
+      read_data_buf_len_(0),
+      file_data_len_(0),
+      data_connection_port_(0),
+      read_ctrl_buf_size_(0),
+      response_message_buf_len_(0),
+      last_error_(OK) {
+  read_ctrl_buf_size_ = kCtrlBufLen;
+  read_ctrl_buf_ = new IOBuffer(kCtrlBufLen);
+  response_message_buf_ = new IOBuffer(kCtrlBufLen);
 }
 
 FtpNetworkTransaction::~FtpNetworkTransaction() {
 }
 
-void FtpNetworkTransaction::Destroy() {
-  delete this;
-}
-
 int FtpNetworkTransaction::Start(
     const FtpRequestInfo* request_info, CompletionCallback* callback) {
   request_ = request_info;
@@ -39,6 +55,11 @@
     user_callback_ = callback;
   return rv;
 }
+int FtpNetworkTransaction::Stop(int error) {
+  next_state_ = STATE_CTRL_WRITE_QUIT;
+  last_error_ = error;
+  return OK;
+}
 
 int FtpNetworkTransaction::RestartWithAuth(
     const std::wstring& username, const std::wstring& password,
@@ -46,13 +67,38 @@
   return ERR_FAILED;
 }
 
-int FtpNetworkTransaction::Read(
-    char* buf, int buf_len, CompletionCallback* callback) {
+int FtpNetworkTransaction::RestartIgnoringLastError(
+    CompletionCallback* callback) {
   return ERR_FAILED;
 }
 
+int FtpNetworkTransaction::Read(
+    IOBuffer* buf, int buf_len, CompletionCallback* callback) {
+  DCHECK(buf);
+  DCHECK(buf_len > 0);
+
+  if (data_socket_ == NULL)
+    return 0;  // Data socket closed, no more data left.
+
+  if (!data_socket_->IsConnected())
+    return 0;  // Data socket disconnected, no more data left.
+
+  read_data_buf_ = buf;
+  read_data_buf_len_ = buf_len;
+
+  next_state_ = STATE_DATA_READ;
+
+  int rv = DoLoop(OK);
+  if (rv == ERR_IO_PENDING)
+    user_callback_ = callback;
+  if (rv == 0) {
+    data_socket_->Disconnect();
+  }
+  return rv;
+}
+
 const FtpResponseInfo* FtpNetworkTransaction::GetResponseInfo() const {
-  return NULL;
+  return &response_;
 }
 
 LoadState FtpNetworkTransaction::GetLoadState() const {
@@ -63,6 +109,71 @@
   return 0;
 }
 
+// Used to prepare and send FTP commad.
+int FtpNetworkTransaction::SendFtpCommand(std::string& command, Command cmd) {
+  response_message_buf_len_ = 0;
+  command_sent_ = cmd;
+  DLOG(INFO) << " >> " << command;
+  command.append(CRLF);
+  const char* buf = command.c_str();
+  int buf_len = command.size();
+  return ctrl_socket_->Write(buf, buf_len, &io_callback_);
+}
+
+int FtpNetworkTransaction::GetRespnseCode() {
+  std::string str(response_message_buf_->data(), 3);
+  return StringToInt(str);
+}
+
+int FtpNetworkTransaction::ProcessResponse(int response_code) {
+  int rv = OK;
+  switch (command_sent_) {
+    case COMMAND_NONE:
+      next_state_ = STATE_CTRL_WRITE_USER;
+      break;
+    case COMMAND_USER:
+      rv = ProcessResponseUSER(response_code);
+      break;
+    case COMMAND_PASS:
+      rv = ProcessResponsePASS(response_code);
+      break;
+    case COMMAND_ACCT:
+      rv = ProcessResponseACCT(response_code);
+      break;
+    case COMMAND_SYST:
+      rv = ProcessResponseSYST(response_code);
+      break;
+    case COMMAND_PWD:
+      rv = ProcessResponsePWD(response_code);
+      break;
+    case COMMAND_TYPE:
+      rv = ProcessResponseTYPE(response_code);
+      break;
+    case COMMAND_PASV:
+      rv = ProcessResponsePASV(response_code);
+      break;
+    case COMMAND_SIZE:
+      rv = ProcessResponseSIZE(response_code);
+      break;
+    case COMMAND_RETR:
+      rv = ProcessResponseRETR(response_code);
+      break;
+    case COMMAND_CWD:
+      rv = ProcessResponseCWD(response_code);
+      break;
+    case COMMAND_LIST:
+      rv = ProcessResponseLIST(response_code);
+      break;
+    case COMMAND_QUIT:
+      rv = ProcessResponseQUIT(response_code);
+      break;
+    default:
+      DLOG(INFO) << "Missing Command response handling!";
+      return ERR_FAILED;
+  }
+  return rv;
+}
+
 void FtpNetworkTransaction::DoCallback(int rv) {
   DCHECK(rv != ERR_IO_PENDING);
   DCHECK(user_callback_);
@@ -108,20 +219,58 @@
       case STATE_CTRL_CONNECT_COMPLETE:
         rv = DoCtrlConnectComplete(rv);
         break;
-      case STATE_CTRL_WRITE:
-        DCHECK(rv == OK);
-        rv = DoCtrlWrite();
-        break;
-      case STATE_CTRL_WRITE_COMPLETE:
-        rv = DoCtrlWriteComplete(rv);
-        break;
       case STATE_CTRL_READ:
-        DCHECK(rv == OK);
         rv = DoCtrlRead();
         break;
       case STATE_CTRL_READ_COMPLETE:
         rv = DoCtrlReadComplete(rv);
         break;
+      case STATE_CTRL_WRITE_USER:
+        DCHECK(rv == OK);
+        rv = DoCtrlWriteUSER();
+        break;
+      case STATE_CTRL_WRITE_PASS:
+        DCHECK(rv == OK);
+        rv = DoCtrlWritePASS();
+        break;
+      case STATE_CTRL_WRITE_SYST:
+        DCHECK(rv == OK);
+        rv = DoCtrlWriteSYST();
+        break;
+      case STATE_CTRL_WRITE_ACCT:
+        DCHECK(rv == OK);
+        rv = DoCtrlWriteACCT();
+        break;
+      case STATE_CTRL_WRITE_PWD:
+        DCHECK(rv == OK);
+        rv = DoCtrlWritePWD();
+        break;
+      case STATE_CTRL_WRITE_TYPE:
+        DCHECK(rv == OK);
+        rv = DoCtrlWriteTYPE();
+        break;
+      case STATE_CTRL_WRITE_PASV:
+        DCHECK(rv == OK);
+        rv = DoCtrlWritePASV();
+        break;
+      case STATE_CTRL_WRITE_RETR:
+        DCHECK(rv == OK);
+        rv = DoCtrlWriteRETR();
+        break;
+      case STATE_CTRL_WRITE_SIZE:
+        DCHECK(rv == OK);
+        rv = DoCtrlWriteSIZE();
+        break;
+      case STATE_CTRL_WRITE_CWD:
+        rv = DoCtrlWriteCWD();
+        break;
+      case STATE_CTRL_WRITE_LIST:
+        rv = DoCtrlWriteLIST();
+        break;
+      case STATE_CTRL_WRITE_QUIT:
+        rv = DoCtrlWriteQUIT();
+        break;
+
       case STATE_DATA_CONNECT:
         DCHECK(rv == OK);
         rv = DoDataConnect();
@@ -130,7 +279,6 @@
         rv = DoDataConnectComplete(rv);
         break;
       case STATE_DATA_READ:
-        DCHECK(rv == OK);
         rv = DoDataRead();
         break;
       case STATE_DATA_READ_COMPLETE:
@@ -142,64 +290,467 @@
         break;
     }
   } while (rv != ERR_IO_PENDING && next_state_ != STATE_NONE);
-
   return rv;
 }
 
+// TODO(ibrar): Yet to see if we need any intialization
 int FtpNetworkTransaction::DoCtrlInit() {
-  return ERR_FAILED;  // TODO(darin): implement me
+  next_state_ = STATE_CTRL_INIT_COMPLETE;
+  return OK;
 }
 
 int FtpNetworkTransaction::DoCtrlInitComplete(int result) {
-  return ERR_FAILED;  // TODO(darin): implement me
+  next_state_ = STATE_CTRL_RESOLVE_HOST;
+  return OK;
 }
 
 int FtpNetworkTransaction::DoCtrlResolveHost() {
-  return ERR_FAILED;  // TODO(darin): implement me
+  next_state_ = STATE_CTRL_RESOLVE_HOST_COMPLETE;
+
+  std::string host;
+  int port;
+
+  host = request_->url.host();
+  port = request_->url.EffectiveIntPort();
+
+  DidStartDnsResolution(host, this);
+  return resolver_.Resolve(host, port, &addresses_, &io_callback_);
 }
 
 int FtpNetworkTransaction::DoCtrlResolveHostComplete(int result) {
-  return ERR_FAILED;  // TODO(darin): implement me
+  bool ok = (result == OK);
+  DidFinishDnsResolutionWithStatus(ok, GURL(), this);
+  if (ok) {
+    next_state_ = STATE_CTRL_CONNECT;
+    return result;
+  }
+  return ERR_FAILED;
 }
 
 int FtpNetworkTransaction::DoCtrlConnect() {
-  return ERR_FAILED;  // TODO(darin): implement me
+  next_state_ = STATE_CTRL_CONNECT_COMPLETE;
+  ctrl_socket_.reset(socket_factory_->CreateTCPClientSocket(addresses_));
+  return ctrl_socket_->Connect(&io_callback_);
 }
 
 int FtpNetworkTransaction::DoCtrlConnectComplete(int result) {
-  return ERR_FAILED;  // TODO(darin): implement me
+  next_state_ = STATE_CTRL_READ;
+  return result;
 }
 
-int FtpNetworkTransaction::DoCtrlWrite() {
-  return ERR_FAILED;  // TODO(darin): implement me
+int FtpNetworkTransaction::DoCtrlRead() {
+  next_state_ = STATE_CTRL_READ_COMPLETE;
+  read_ctrl_buf_->data()[0] = 0;
+  return ctrl_socket_->Read(read_ctrl_buf_->data(),
+      read_ctrl_buf_size_ - 1, &io_callback_);
 }
 
-int FtpNetworkTransaction::DoCtrlWriteComplete(int result) {
-  return ERR_FAILED;  // TODO(darin): implement me
+int FtpNetworkTransaction::DoCtrlReadComplete(int result) {
+  if (result < 0)
+    return Stop(ERR_FAILED);
+  int response_code;
+  // Null termination added, now we can treat this as string.
+  read_ctrl_buf_->data()[result] = 0;
+  memcpy(response_message_buf_->data() + response_message_buf_len_,
+      read_ctrl_buf_->data(), result);
+
+  response_message_buf_len_ = response_message_buf_len_ + result;
+  for (int i = 0; i < response_message_buf_len_; i++) {
+    if (response_message_buf_->data()[i] == '\r' &&
+        response_message_buf_->data()[i + 1] == '\n') {
+      if (response_message_buf_len_ > 3 &&
+          response_message_buf_->data()[3] == ' ') {
+        response_message_buf_->data()[response_message_buf_len_ - 2] = 0;
+        response_code = GetRespnseCode();
+        return ProcessResponse(response_code);
+      } else {
+        response_message_buf_len_ -= (i + 2);
+        memcpy(response_message_buf_->data(),
+               response_message_buf_->data() + i + 2,
+               response_message_buf_len_);
+        i = 0;
+      }
+    }
+  }
+  next_state_ = STATE_CTRL_READ;
+  return OK;
 }
 
-int FtpNetworkTransaction::DoCtrlRead() {
-  return ERR_FAILED;  // TODO(darin): implement me
+// FTP Commands and responses
+
+// USER Command.
+int FtpNetworkTransaction::DoCtrlWriteUSER() {
+  std::string command = "USER ";
+  if (request_->url.has_username()) {
+    command.append(request_->url.username());
+  } else {
+    command.append("anonymous");
+  }
+  next_state_ = STATE_CTRL_READ;
+  return SendFtpCommand(command, COMMAND_USER);
 }
 
-int FtpNetworkTransaction::DoCtrlReadComplete(int result) {
-  return ERR_FAILED;  // TODO(darin): implement me
+int FtpNetworkTransaction::ProcessResponseUSER(int response_code) {
+  switch (GetErrorClass(response_code)) {
+    case ERROR_CLASS_INITIATED:
+    case ERROR_CLASS_OK:
+      next_state_ = STATE_CTRL_WRITE_USER;
+      break;
+    case ERROR_CLASS_PENDING:
+      next_state_ = STATE_CTRL_WRITE_PASS;
+      break;
+    case ERROR_CLASS_ERROR_RETRY:
+      if (response_code == 421)
+        return Stop(ERR_FAILED);
+      break;
+    case ERROR_CLASS_ERROR:
+      return Stop(ERR_FAILED);
+    default:
+      return Stop(ERR_FAILED);
+  }
+  return OK;
 }
 
+// PASS command.
+int FtpNetworkTransaction::DoCtrlWritePASS() {
+  std::string command = "PASS ";
+  if (request_->url.has_password()) {
+    command.append(request_->url.password());
+  } else {
+    command.append("IEUser@");
+  }
+  next_state_ = STATE_CTRL_READ;
+  return SendFtpCommand(command, COMMAND_PASS);
+}
+
+int FtpNetworkTransaction::ProcessResponsePASS(int response_code) {
+  switch (GetErrorClass(response_code)) {
+    case ERROR_CLASS_INITIATED:
+    case ERROR_CLASS_OK:
+      next_state_ = STATE_CTRL_WRITE_SYST;
+      break;
+    case ERROR_CLASS_PENDING:
+      next_state_ = STATE_CTRL_WRITE_ACCT;
+      break;
+    case ERROR_CLASS_ERROR_RETRY:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_ERROR:
+      if (response_code == 503) {
+        next_state_ = STATE_CTRL_WRITE_PASS;
+      } else {
+        return Stop(ERR_FAILED);
+      }
+      break;
+    default:
+      return Stop(ERR_FAILED);
+  }
+  return OK;
+}
+
+// ACCT command.
+int FtpNetworkTransaction::DoCtrlWriteACCT() {
+  std::string command = "ACCT noaccount";
+  next_state_ = STATE_CTRL_READ;
+  return SendFtpCommand(command, COMMAND_ACCT);
+}
+
+int FtpNetworkTransaction::ProcessResponseACCT(int response_code) {
+  switch (GetErrorClass(response_code)) {
+    case ERROR_CLASS_INITIATED:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_OK:
+      next_state_ = STATE_CTRL_WRITE_SYST;
+      break;
+    case ERROR_CLASS_PENDING:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_ERROR_RETRY:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_ERROR:
+      return Stop(ERR_FAILED);
+    default:
+      return Stop(ERR_FAILED);
+  }
+  return OK;
+}
+
+// SYST command.
+int FtpNetworkTransaction::DoCtrlWriteSYST() {
+  std::string command = "SYST";
+  next_state_ = STATE_CTRL_READ;
+  return SendFtpCommand(command, COMMAND_SYST);
+}
+
+int FtpNetworkTransaction::ProcessResponseSYST(int response_code) {
+  next_state_ = STATE_CTRL_WRITE_PWD;
+  return OK;
+}
+
+// PWD command.
+int FtpNetworkTransaction::DoCtrlWritePWD() {
+  std::string command = "PWD";
+  next_state_ = STATE_CTRL_READ;
+  return SendFtpCommand(command, COMMAND_PWD);
+}
+
+int FtpNetworkTransaction::ProcessResponsePWD(int response_code) {
+  switch (GetErrorClass(response_code)) {
+    case ERROR_CLASS_INITIATED:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_OK:
+      next_state_ = STATE_CTRL_WRITE_TYPE;
+      break;
+    case ERROR_CLASS_PENDING:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_ERROR_RETRY:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_ERROR:
+      return Stop(ERR_FAILED);
+    default:
+      return Stop(ERR_FAILED);
+  }
+  return OK;
+}
+
+// TYPE command.
+int FtpNetworkTransaction::DoCtrlWriteTYPE() {
+  std::string command = "TYPE I";
+  next_state_ = STATE_CTRL_READ;
+  return SendFtpCommand(command, COMMAND_TYPE);
+}
+
+int FtpNetworkTransaction::ProcessResponseTYPE(int response_code) {
+  switch (GetErrorClass(response_code)) {
+    case ERROR_CLASS_INITIATED:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_OK:
+      next_state_ = STATE_CTRL_WRITE_PASV;
+      break;
+    case ERROR_CLASS_PENDING:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_ERROR_RETRY:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_ERROR:
+      return Stop(ERR_FAILED);
+    default:
+      return Stop(ERR_FAILED);
+  }
+  return OK;
+}
+
+// PASV command
+int FtpNetworkTransaction::DoCtrlWritePASV() {
+  std::string command = "PASV ";
+  next_state_ = STATE_CTRL_READ;
+  return SendFtpCommand(command, COMMAND_PASV);
+}
+
+// There are two way we can receive IP address and port.
+// (127,0,0,1,23,21) IP address and port encapsulate in ().
+// 127,0,0,1,23,21  IP address and port without ().
+int FtpNetworkTransaction::ProcessResponsePASV(int response_code) {
+  switch (GetErrorClass(response_code)) {
+    case ERROR_CLASS_INITIATED:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_OK:
+      char* ptr;
+      int i0, i1, i2, i3, p0, p1;
+      ptr = read_ctrl_buf_->data();  // Try with bracket.
+      while (*ptr && *ptr != '(')
+        ++ptr;
+      if (*ptr) {
+        ++ptr;
+      } else {
+        ptr = read_ctrl_buf_->data();  // Try without bracket.
+        while (*ptr && *ptr != ',')
+          ++ptr;
+        while (*ptr && *ptr != ' ')
+        --ptr;
+      }
+      if (sscanf_s(ptr, "%ld,%ld,%ld,%ld,%ld,%ld", &i0, &i1, &i2, &i3, &p0,
+                   &p1) == 6) {
+        data_connection_ip_ = StringPrintf("%ld.%ld.%ld.%ld", i0, i1, i2, i3);
+        data_connection_port_ = (p0 << 8) + p1;
+        next_state_ = STATE_DATA_CONNECT;
+      } else {
+        return Stop(ERR_FAILED);
+      }
+      break;
+    case ERROR_CLASS_PENDING:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_ERROR_RETRY:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_ERROR:
+      return Stop(ERR_FAILED);
+    default:
+      return Stop(ERR_FAILED);
+  }
+  return OK;
+}
+
+// SIZE command
+int FtpNetworkTransaction::DoCtrlWriteSIZE() {
+  std::string command = "SIZE ";
+  if (request_->url.has_path())
+    command.append(request_->url.path());
+  next_state_ = STATE_CTRL_READ;
+  return SendFtpCommand(command, COMMAND_SIZE);
+}
+
+int FtpNetworkTransaction::ProcessResponseSIZE(int response_code) {
+  switch (GetErrorClass(response_code)) {
+    case ERROR_CLASS_INITIATED:
+      next_state_ = STATE_CTRL_WRITE_LIST;
+      break;
+    case ERROR_CLASS_OK:
+      next_state_ = STATE_CTRL_WRITE_RETR;
+      if (!StringToInt(read_ctrl_buf_->data() + 4, &file_data_len_))
+        return Stop(ERR_FAILED);
+      break;
+    case ERROR_CLASS_PENDING:
+      next_state_ = STATE_CTRL_WRITE_LIST;
+      break;
+    case ERROR_CLASS_ERROR_RETRY:
+      next_state_ = STATE_CTRL_WRITE_LIST;
+      break;
+    case ERROR_CLASS_ERROR:
+      next_state_ = STATE_CTRL_WRITE_LIST;
+      break;
+    default:
+      return Stop(ERR_FAILED);
+  }
+  return OK;
+}
+
+// RETR command
+int FtpNetworkTransaction::DoCtrlWriteRETR() {
+  std::string command = "RETR ";
+  if (request_->url.has_path())
+    command.append(request_->url.path());
+  next_state_ = STATE_CTRL_READ;
+  return SendFtpCommand(command, COMMAND_RETR);
+}
+
+int FtpNetworkTransaction::ProcessResponseRETR(int response_code) {
+  switch (GetErrorClass(response_code)) {
+    case ERROR_CLASS_INITIATED:
+      next_state_ = STATE_CTRL_WRITE_QUIT;
+      ctrl_socket_->Disconnect();
+      break;
+    case ERROR_CLASS_OK:
+      break;  // FTP Done
+    case ERROR_CLASS_PENDING:
+      next_state_ = STATE_CTRL_WRITE_PASV;
+      break;
+    case ERROR_CLASS_ERROR_RETRY:
+      if (response_code == 421 || response_code == 425 || response_code == 426)
+        return Stop(ERR_FAILED);
+      return ERR_FAILED;  // TODO(ibrar): Retry here.
+    case ERROR_CLASS_ERROR:
+      next_state_ = STATE_CTRL_WRITE_CWD;
+      break;
+    default:
+      return Stop(ERR_FAILED);
+  }
+  return OK;
+}
+
+// CWD command
+int FtpNetworkTransaction::DoCtrlWriteCWD() {
+  std::string command = "CWD";
+  next_state_ = STATE_CTRL_READ;
+  return SendFtpCommand(command, COMMAND_CWD);
+}
+
+int FtpNetworkTransaction::ProcessResponseCWD(int response_code) {
+  switch (GetErrorClass(response_code)) {
+    case ERROR_CLASS_INITIATED:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_OK:
+      next_state_ = STATE_CTRL_WRITE_LIST;
+    case ERROR_CLASS_PENDING:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_ERROR_RETRY:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_ERROR:
+      return Stop(ERR_FAILED);
+    default:
+      return Stop(ERR_FAILED);
+  }
+  return OK;
+}
+
+// LIST command
+int FtpNetworkTransaction::DoCtrlWriteLIST() {
+  std::string command = "LIST ";
+  if (request_->url.has_path())
+    command.append(request_->url.path());
+  next_state_ = STATE_CTRL_READ;
+  return SendFtpCommand(command, COMMAND_LIST);
+}
+
+int FtpNetworkTransaction::ProcessResponseLIST(int response_code) {
+  switch (GetErrorClass(response_code)) {
+    case ERROR_CLASS_INITIATED:
+      next_state_ = STATE_CTRL_WRITE_QUIT;
+      response_.is_directory_listing = true;
+      break;
+    case ERROR_CLASS_OK:
+      next_state_ = STATE_CTRL_WRITE_QUIT;
+      response_.is_directory_listing = true;
+      break;
+    case ERROR_CLASS_PENDING:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_ERROR_RETRY:
+      return Stop(ERR_FAILED);
+    case ERROR_CLASS_ERROR:
+      return Stop(ERR_FAILED);
+    default:
+      return Stop(ERR_FAILED);
+  }
+  return OK;
+}
+
+// Quit command
+int FtpNetworkTransaction::DoCtrlWriteQUIT() {
+  std::string command = "QUIT";
+  next_state_ = STATE_CTRL_READ;
+  return SendFtpCommand(command, COMMAND_QUIT);
+}
+
+int FtpNetworkTransaction::ProcessResponseQUIT(int response_code) {
+  ctrl_socket_->Disconnect();
+  return last_error_;
+}
+
+// Data Connection
 int FtpNetworkTransaction::DoDataConnect() {
-  return ERR_FAILED;  // TODO(darin): implement me
+  next_state_ = STATE_DATA_CONNECT_COMPLETE;
+  AddressList adr;
+  int err = resolver_.Resolve(data_connection_ip_,
+      data_connection_port_, &adr, NULL);
+  data_socket_.reset(socket_factory_->CreateTCPClientSocket(adr));
+  return data_socket_->Connect(&io_callback_);
 }
 
 int FtpNetworkTransaction::DoDataConnectComplete(int result) {
-  return ERR_FAILED;  // TODO(darin): implement me
+  next_state_ = STATE_CTRL_WRITE_SIZE;
+  return result;
 }
 
 int FtpNetworkTransaction::DoDataRead() {
-  return ERR_FAILED;  // TODO(darin): implement me
+  DCHECK(read_data_buf_);
+  DCHECK(read_data_buf_len_ > 0);
+
+  next_state_ = STATE_DATA_READ_COMPLETE;
+  read_data_buf_->data()[0] = 0;
+  return data_socket_->Read(read_data_buf_->data(),
+      read_data_buf_len_, &io_callback_);
 }
 
 int FtpNetworkTransaction::DoDataReadComplete(int result) {
-  return ERR_FAILED;  // TODO(darin): implement me
+  DLOG(INFO) << read_data_buf_->data();  // The read_data_buf_ is NULL
+                                         // terminated string.
+  return result;
 }
 
 }  // namespace net