| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011 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 "net/ftp/ftp_directory_listing_parser.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/callback.h" | |
| 9 #include "base/i18n/icu_encoding_detection.h" | |
| 10 #include "base/i18n/icu_string_conversions.h" | |
| 11 #include "base/stl_util.h" | |
| 12 #include "base/strings/string_util.h" | |
| 13 #include "base/strings/string_split.h" | |
| 14 #include "base/strings/utf_string_conversions.h" | |
| 15 #include "net/base/net_errors.h" | |
| 16 #include "net/ftp/ftp_directory_listing_parser_ls.h" | |
| 17 #include "net/ftp/ftp_directory_listing_parser_netware.h" | |
| 18 #include "net/ftp/ftp_directory_listing_parser_os2.h" | |
| 19 #include "net/ftp/ftp_directory_listing_parser_vms.h" | |
| 20 #include "net/ftp/ftp_directory_listing_parser_windows.h" | |
| 21 #include "net/ftp/ftp_server_type_histograms.h" | |
| 22 | |
| 23 namespace net { | |
| 24 | |
| 25 namespace { | |
| 26 | |
| 27 // Fills in |raw_name| for all |entries| using |encoding|. Returns network | |
| 28 // error code. | |
| 29 int FillInRawName(const std::string& encoding, | |
| 30 std::vector<FtpDirectoryListingEntry>* entries) { | |
| 31 for (size_t i = 0; i < entries->size(); i++) { | |
| 32 if (!base::UTF16ToCodepage(entries->at(i).name, encoding.c_str(), | |
| 33 base::OnStringConversionError::FAIL, | |
| 34 &entries->at(i).raw_name)) { | |
| 35 return ERR_ENCODING_CONVERSION_FAILED; | |
| 36 } | |
| 37 } | |
| 38 | |
| 39 return OK; | |
| 40 } | |
| 41 | |
| 42 // Parses |text| as an FTP directory listing. Fills in |entries| | |
| 43 // and |server_type| and returns network error code. | |
| 44 int ParseListing(const base::string16& text, | |
| 45 const base::string16& newline_separator, | |
| 46 const std::string& encoding, | |
| 47 const base::Time& current_time, | |
| 48 std::vector<FtpDirectoryListingEntry>* entries, | |
| 49 FtpServerType* server_type) { | |
| 50 std::vector<base::string16> lines; | |
| 51 base::SplitStringUsingSubstr(text, newline_separator, &lines); | |
| 52 | |
| 53 struct { | |
| 54 base::Callback<bool(void)> callback; | |
| 55 FtpServerType server_type; | |
| 56 } parsers[] = { | |
| 57 { | |
| 58 base::Bind(&ParseFtpDirectoryListingLs, lines, current_time, entries), | |
| 59 SERVER_LS | |
| 60 }, | |
| 61 { | |
| 62 base::Bind(&ParseFtpDirectoryListingWindows, lines, entries), | |
| 63 SERVER_WINDOWS | |
| 64 }, | |
| 65 { | |
| 66 base::Bind(&ParseFtpDirectoryListingVms, lines, entries), | |
| 67 SERVER_VMS | |
| 68 }, | |
| 69 { | |
| 70 base::Bind(&ParseFtpDirectoryListingNetware, | |
| 71 lines, current_time, entries), | |
| 72 SERVER_NETWARE | |
| 73 }, | |
| 74 { | |
| 75 base::Bind(&ParseFtpDirectoryListingOS2, lines, entries), | |
| 76 SERVER_OS2 | |
| 77 } | |
| 78 }; | |
| 79 | |
| 80 for (size_t i = 0; i < arraysize(parsers); i++) { | |
| 81 entries->clear(); | |
| 82 if (parsers[i].callback.Run()) { | |
| 83 *server_type = parsers[i].server_type; | |
| 84 return FillInRawName(encoding, entries); | |
| 85 } | |
| 86 } | |
| 87 | |
| 88 entries->clear(); | |
| 89 return ERR_UNRECOGNIZED_FTP_DIRECTORY_LISTING_FORMAT; | |
| 90 } | |
| 91 | |
| 92 // Detects encoding of |text| and parses it as an FTP directory listing. | |
| 93 // Fills in |entries| and |server_type| and returns network error code. | |
| 94 int DecodeAndParse(const std::string& text, | |
| 95 const base::Time& current_time, | |
| 96 std::vector<FtpDirectoryListingEntry>* entries, | |
| 97 FtpServerType* server_type) { | |
| 98 const char* const kNewlineSeparators[] = { "\n", "\r\n" }; | |
| 99 | |
| 100 std::vector<std::string> encodings; | |
| 101 if (!base::DetectAllEncodings(text, &encodings)) | |
| 102 return ERR_ENCODING_DETECTION_FAILED; | |
| 103 | |
| 104 // Use first encoding that can be used to decode the text. | |
| 105 for (size_t i = 0; i < encodings.size(); i++) { | |
| 106 base::string16 converted_text; | |
| 107 if (base::CodepageToUTF16(text, | |
| 108 encodings[i].c_str(), | |
| 109 base::OnStringConversionError::FAIL, | |
| 110 &converted_text)) { | |
| 111 for (size_t j = 0; j < arraysize(kNewlineSeparators); j++) { | |
| 112 int rv = ParseListing(converted_text, | |
| 113 base::ASCIIToUTF16(kNewlineSeparators[j]), | |
| 114 encodings[i], | |
| 115 current_time, | |
| 116 entries, | |
| 117 server_type); | |
| 118 if (rv == OK) | |
| 119 return rv; | |
| 120 } | |
| 121 } | |
| 122 } | |
| 123 | |
| 124 entries->clear(); | |
| 125 *server_type = SERVER_UNKNOWN; | |
| 126 return ERR_UNRECOGNIZED_FTP_DIRECTORY_LISTING_FORMAT; | |
| 127 } | |
| 128 | |
| 129 } // namespace | |
| 130 | |
| 131 FtpDirectoryListingEntry::FtpDirectoryListingEntry() | |
| 132 : type(UNKNOWN), | |
| 133 size(-1) { | |
| 134 } | |
| 135 | |
| 136 int ParseFtpDirectoryListing(const std::string& text, | |
| 137 const base::Time& current_time, | |
| 138 std::vector<FtpDirectoryListingEntry>* entries) { | |
| 139 FtpServerType server_type = SERVER_UNKNOWN; | |
| 140 int rv = DecodeAndParse(text, current_time, entries, &server_type); | |
| 141 UpdateFtpServerTypeHistograms(server_type); | |
| 142 return rv; | |
| 143 } | |
| 144 | |
| 145 } // namespace net | |
| OLD | NEW |