content/child/ftp_directory_listing_response_delegate.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/child/ftp_directory_listing_response_delegate.h"
   6
   7 #include <vector>
   8
   9 #include "base/i18n/icu_encoding_detection.h"
  10 #include "base/i18n/icu_string_conversions.h"
  11 #include "base/logging.h"
  12 #include "base/strings/string_util.h"
  13 #include "base/strings/sys_string_conversions.h"
  14 #include "base/strings/utf_string_conversions.h"
  15 #include "base/time/time.h"
  16 #include "net/base/escape.h"
  17 #include "net/base/net_errors.h"
  18 #include "net/base/net_util.h"
  19 #include "net/ftp/ftp_directory_listing_parser.h"
  20 #include "third_party/WebKit/public/platform/WebURL.h"
  21 #include "third_party/WebKit/public/platform/WebURLLoaderClient.h"
  22 #include "webkit/child/weburlresponse_extradata_impl.h"
  23
  24 using blink::WebURLLoader;
  25 using blink::WebURLLoaderClient;
  26 using blink::WebURLResponse;
  27 using net::FtpDirectoryListingEntry;
  28 using webkit_glue::WebURLResponseExtraDataImpl;
  29
  30 namespace {
  31
  32 base::string16 ConvertPathToUTF16(const std::string& path) {
  33   // Per RFC 2640, FTP servers should use UTF-8 or its proper subset ASCII,
  34   // but many old FTP servers use legacy encodings. Try UTF-8 first.
  35   if (base::IsStringUTF8(path))
  36     return base::UTF8ToUTF16(path);
  37
  38   // Try detecting the encoding. The sample is rather small though, so it may
  39   // fail.
  40   std::string encoding;
  41   if (base::DetectEncoding(path, &encoding) && !encoding.empty()) {
  42     base::string16 path_utf16;
  43     if (base::CodepageToUTF16(path, encoding.c_str(),
  44                               base::OnStringConversionError::SUBSTITUTE,
  45                               &path_utf16)) {
  46       return path_utf16;
  47     }
  48   }
  49
  50   // Use system native encoding as the last resort.
  51   return base::WideToUTF16(base::SysNativeMBToWide(path));
  52 }
  53
  54 }  // namespace
  55
  56 namespace content {
  57
  58 FtpDirectoryListingResponseDelegate::FtpDirectoryListingResponseDelegate(
  59     WebURLLoaderClient* client,
  60     WebURLLoader* loader,
  61     const WebURLResponse& response)
  62     : client_(client),
  63       loader_(loader) {
  64   if (response.extraData()) {
  65     // extraData can be NULL during tests.
  66     WebURLResponseExtraDataImpl* extra_data =
  67         static_cast<WebURLResponseExtraDataImpl*>(response.extraData());
  68     extra_data->set_is_ftp_directory_listing(true);
  69   }
  70   Init(response.url());
  71 }
  72
  73 void FtpDirectoryListingResponseDelegate::Cancel() {
  74   client_ = NULL;
  75   loader_ = NULL;
  76 }
  77
  78 void FtpDirectoryListingResponseDelegate::OnReceivedData(const char* data,
  79                                                          int data_len) {
  80   buffer_.append(data, data_len);
  81 }
  82
  83 void FtpDirectoryListingResponseDelegate::OnCompletedRequest() {
  84   std::vector<FtpDirectoryListingEntry> entries;
  85   int rv = -1;
  86 #if !defined(DISABLE_FTP_SUPPORT)
  87   rv = net::ParseFtpDirectoryListing(buffer_, base::Time::Now(), &entries);
  88 #endif
  89   if (rv != net::OK) {
  90     SendDataToClient("<script>onListingParsingError();</script>\n");
  91     return;
  92   }
  93   for (size_t i = 0; i < entries.size(); i++) {
  94     const FtpDirectoryListingEntry& entry = entries[i];
  95
  96     // Skip the current and parent directory entries in the listing. Our header
  97     // always includes them.
  98     if (EqualsASCII(entry.name, ".") || EqualsASCII(entry.name, ".."))
  99       continue;
 100
 101     bool is_directory = (entry.type == FtpDirectoryListingEntry::DIRECTORY);
 102     int64 size = entry.size;
 103     if (entry.type != FtpDirectoryListingEntry::FILE)
 104       size = 0;
 105     SendDataToClient(net::GetDirectoryListingEntry(
 106         entry.name, entry.raw_name, is_directory, size, entry.last_modified));
 107   }
 108 }
 109
 110 void FtpDirectoryListingResponseDelegate::Init(const GURL& response_url) {
 111   net::UnescapeRule::Type unescape_rules = net::UnescapeRule::SPACES |
 112                                            net::UnescapeRule::URL_SPECIAL_CHARS;
 113   std::string unescaped_path = net::UnescapeURLComponent(response_url.path(),
 114                                                          unescape_rules);
 115   SendDataToClient(net::GetDirectoryListingHeader(
 116       ConvertPathToUTF16(unescaped_path)));
 117
 118   // If this isn't top level directory (i.e. the path isn't "/",)
 119   // add a link to the parent directory.
 120   if (response_url.path().length() > 1) {
 121     SendDataToClient(net::GetDirectoryListingEntry(
 122         base::ASCIIToUTF16(".."), std::string(), false, 0, base::Time()));
 123   }
 124 }
 125
 126 void FtpDirectoryListingResponseDelegate::SendDataToClient(
 127     const std::string& data) {
 128   if (client_)
 129     client_->didReceiveData(loader_, data.data(), data.length(), -1);
 130 }
 131
 132 }  // namespace content