Home | History | Annotate | Download | only in child 
      1 // Copyright (c) 2012 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 "webkit/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 net::FtpDirectoryListingEntry;
     25 
     26 using WebKit::WebURLLoader;
     27 using WebKit::WebURLLoaderClient;
     28 using WebKit::WebURLResponse;
     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 (IsStringUTF8(path))
     36     return 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 WideToUTF16Hack(base::SysNativeMBToWide(path));
     52 }
     53 
     54 }  // namespace
     55 
     56 namespace webkit_glue {
     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::OnReceivedData(const char* data,
     74                                                          int data_len) {
     75   buffer_.append(data, data_len);
     76 }
     77 
     78 void FtpDirectoryListingResponseDelegate::OnCompletedRequest() {
     79   std::vector<FtpDirectoryListingEntry> entries;
     80   int rv = net::ParseFtpDirectoryListing(buffer_, base::Time::Now(), &entries);
     81   if (rv != net::OK) {
     82     SendDataToClient("<script>onListingParsingError();</script>\n");
     83     return;
     84   }
     85   for (size_t i = 0; i < entries.size(); i++) {
     86     FtpDirectoryListingEntry entry = entries[i];
     87 
     88     // Skip the current and parent directory entries in the listing. Our header
     89     // always includes them.
     90     if (EqualsASCII(entry.name, ".") || EqualsASCII(entry.name, ".."))
     91       continue;
     92 
     93     bool is_directory = (entry.type == FtpDirectoryListingEntry::DIRECTORY);
     94     int64 size = entry.size;
     95     if (entry.type != FtpDirectoryListingEntry::FILE)
     96       size = 0;
     97     SendDataToClient(net::GetDirectoryListingEntry(
     98         entry.name, entry.raw_name, is_directory, size, entry.last_modified));
     99   }
    100 }
    101 
    102 void FtpDirectoryListingResponseDelegate::Init(const GURL& response_url) {
    103   net::UnescapeRule::Type unescape_rules = net::UnescapeRule::SPACES |
    104                                            net::UnescapeRule::URL_SPECIAL_CHARS;
    105   std::string unescaped_path = net::UnescapeURLComponent(response_url.path(),
    106                                                          unescape_rules);
    107   SendDataToClient(net::GetDirectoryListingHeader(
    108       ConvertPathToUTF16(unescaped_path)));
    109 
    110   // If this isn't top level directory (i.e. the path isn't "/",)
    111   // add a link to the parent directory.
    112   if (response_url.path().length() > 1) {
    113     SendDataToClient(net::GetDirectoryListingEntry(
    114         ASCIIToUTF16(".."), std::string(), false, 0, base::Time()));
    115   }
    116 }
    117 
    118 void FtpDirectoryListingResponseDelegate::SendDataToClient(
    119     const std::string& data) {
    120   client_->didReceiveData(loader_, data.data(), data.length(), -1);
    121 }
    122 
    123 }  // namespace webkit_glue
    124