Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(83)

Side by Side Diff: net/socket/socks_client_socket_fuzzer.cc

Issue 1946793002: net: Add fuzzer for HostResolverImpl. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: merge, fix build Created 4 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright 2016 The Chromium Authors. All rights reserved. 1 // Copyright 2016 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include <stddef.h> 5 #include <stddef.h>
6 #include <stdint.h> 6 #include <stdint.h>
7 7
8 #include <memory> 8 #include <memory>
9 9
10 #include "base/logging.h" 10 #include "base/logging.h"
(...skipping 17 matching lines...) Expand all
28 net::TestNetLog test_net_log; 28 net::TestNetLog test_net_log;
29 29
30 net::FuzzedDataProvider data_provider(data, size); 30 net::FuzzedDataProvider data_provider(data, size);
31 31
32 // Determine if the DNS lookup returns synchronously or asynchronously, 32 // Determine if the DNS lookup returns synchronously or asynchronously,
33 // succeeds or fails, and returns an IPv4 or IPv6 address. 33 // succeeds or fails, and returns an IPv4 or IPv6 address.
34 net::MockHostResolver mock_host_resolver; 34 net::MockHostResolver mock_host_resolver;
35 scoped_refptr<net::RuleBasedHostResolverProc> rules( 35 scoped_refptr<net::RuleBasedHostResolverProc> rules(
36 new net::RuleBasedHostResolverProc(nullptr)); 36 new net::RuleBasedHostResolverProc(nullptr));
37 mock_host_resolver.set_synchronous_mode(data_provider.ConsumeBool()); 37 mock_host_resolver.set_synchronous_mode(data_provider.ConsumeBool());
38 switch (data_provider.ConsumeValueInRange(0, 2)) { 38 switch (data_provider.ConsumeInt32InRange(0, 2)) {
39 case 0: 39 case 0:
40 rules->AddRule("*", "127.0.0.1"); 40 rules->AddRule("*", "127.0.0.1");
41 break; 41 break;
42 case 1: 42 case 1:
43 rules->AddRule("*", "::1"); 43 rules->AddRule("*", "::1");
44 break; 44 break;
45 case 2: 45 case 2:
46 rules->AddSimulatedFailure("*"); 46 rules->AddSimulatedFailure("*");
47 break; 47 break;
48 } 48 }
49 mock_host_resolver.set_rules(rules.get()); 49 mock_host_resolver.set_rules(rules.get());
50 50
51 net::TestCompletionCallback callback; 51 net::TestCompletionCallback callback;
52 std::unique_ptr<net::FuzzedSocket> fuzzed_socket( 52 std::unique_ptr<net::FuzzedSocket> fuzzed_socket(
53 new net::FuzzedSocket(&data_provider, &test_net_log)); 53 new net::FuzzedSocket(&data_provider, &test_net_log));
54 CHECK_EQ(net::OK, fuzzed_socket->Connect(callback.callback())); 54 CHECK_EQ(net::OK, fuzzed_socket->Connect(callback.callback()));
55 55
56 std::unique_ptr<net::ClientSocketHandle> socket_handle( 56 std::unique_ptr<net::ClientSocketHandle> socket_handle(
57 new net::ClientSocketHandle()); 57 new net::ClientSocketHandle());
58 socket_handle->SetSocket(std::move(fuzzed_socket)); 58 socket_handle->SetSocket(std::move(fuzzed_socket));
59 59
60 net::HostResolver::RequestInfo request_info(net::HostPortPair("foo", 80)); 60 net::HostResolver::RequestInfo request_info(net::HostPortPair("foo", 80));
61 net::SOCKSClientSocket socket(std::move(socket_handle), request_info, 61 net::SOCKSClientSocket socket(std::move(socket_handle), request_info,
62 net::DEFAULT_PRIORITY, &mock_host_resolver); 62 net::DEFAULT_PRIORITY, &mock_host_resolver);
63 int result = socket.Connect(callback.callback()); 63 int result = socket.Connect(callback.callback());
64 callback.GetResult(result); 64 callback.GetResult(result);
65 return 0; 65 return 0;
66 } 66 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698