OLD | NEW |
| (Empty) |
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/glue/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/glue/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 | |
OLD | NEW |