Index: net/url_request/url_request_ftp_fuzzer.cc |
diff --git a/net/url_request/url_request_ftp_fuzzer.cc b/net/url_request/url_request_ftp_fuzzer.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..4f64b5ae8d917b2d9a9e3f2e7d90a97e780d89ad |
--- /dev/null |
+++ b/net/url_request/url_request_ftp_fuzzer.cc |
@@ -0,0 +1,89 @@ |
+// Copyright 2016 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "net/url_request/url_request.h" |
+ |
+#include <stddef.h> |
+#include <stdint.h> |
+ |
+#include <memory> |
+ |
+#include "base/macros.h" |
+#include "base/memory/ptr_util.h" |
+#include "base/run_loop.h" |
+#include "base/test/fuzzed_data_provider.h" |
+#include "net/base/request_priority.h" |
+#include "net/dns/fuzzed_host_resolver.h" |
+#include "net/ftp/ftp_network_transaction.h" |
+#include "net/ftp/ftp_transaction_factory.h" |
+#include "net/socket/client_socket_factory.h" |
+#include "net/socket/fuzzed_socket_factory.h" |
+#include "net/url_request/ftp_protocol_handler.h" |
+#include "net/url_request/url_request.h" |
+#include "net/url_request/url_request_context.h" |
+#include "net/url_request/url_request_job_factory_impl.h" |
+#include "net/url_request/url_request_test_util.h" |
+#include "url/gurl.h" |
+ |
+namespace { |
+ |
+// Returns FtpNetworkTransactions using the specified HostResolver |
+// and ClientSocketFactory. |
+class FuzzedFtpTransactionFactory : public net::FtpTransactionFactory { |
+ public: |
+ FuzzedFtpTransactionFactory(net::HostResolver* host_resolver, |
+ net::ClientSocketFactory* client_socket_factory) |
+ : host_resolver_(host_resolver), |
+ client_socket_factory_(client_socket_factory) {} |
+ |
+ // FtpTransactionFactory: |
+ std::unique_ptr<net::FtpTransaction> CreateTransaction() override { |
+ return base::MakeUnique<net::FtpNetworkTransaction>(host_resolver_, |
+ client_socket_factory_); |
+ } |
+ |
+ void Suspend(bool suspend) override { NOTREACHED(); } |
+ |
+ private: |
+ net::HostResolver* host_resolver_; |
+ net::ClientSocketFactory* client_socket_factory_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(FuzzedFtpTransactionFactory); |
+}; |
+ |
+} // namespace |
+ |
+// Integration fuzzer for URLRequestFtpJob. |
+extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) { |
+ base::FuzzedDataProvider data_provider(data, size); |
+ net::TestURLRequestContext url_request_context(true); |
+ net::FuzzedSocketFactory fuzzed_socket_factory(&data_provider); |
+ url_request_context.set_client_socket_factory(&fuzzed_socket_factory); |
+ |
+ // Need to fuzz the HostResolver to select between IPv4 and IPv6. |
+ net::FuzzedHostResolver host_resolver(net::HostResolver::Options(), nullptr, |
+ &data_provider); |
+ url_request_context.set_host_resolver(&host_resolver); |
+ |
+ net::URLRequestJobFactoryImpl job_factory; |
+ job_factory.SetProtocolHandler( |
+ "ftp", net::FtpProtocolHandler::CreateForTesting( |
+ base::MakeUnique<FuzzedFtpTransactionFactory>( |
+ &host_resolver, &fuzzed_socket_factory))); |
+ url_request_context.set_job_factory(&job_factory); |
+ |
+ url_request_context.Init(); |
+ |
+ net::TestDelegate delegate; |
+ |
+ std::unique_ptr<net::URLRequest> url_request( |
+ url_request_context.CreateRequest( |
+ GURL("ftp://foo/" + data_provider.ConsumeRandomLengthString(1000)), |
+ net::DEFAULT_PRIORITY, &delegate)); |
+ url_request->Start(); |
+ // TestDelegate quits the message loop on completion. |
+ base::RunLoop().Run(); |
+ |
+ return 0; |
+} |