1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "net/socket/fuzzed_server_socket.h"
#include <utility>
#include "base/bind.h"
#include "base/location.h"
#include "base/threading/thread_task_runner_handle.h"
#include "net/socket/fuzzed_socket.h"
namespace net {
FuzzedServerSocket::FuzzedServerSocket(FuzzedDataProvider* data_provider,
net::NetLog* net_log)
: data_provider_(data_provider), net_log_(net_log) {}
FuzzedServerSocket::~FuzzedServerSocket() = default;
int FuzzedServerSocket::Listen(const IPEndPoint& address, int backlog) {
DCHECK(!listen_called_);
listening_on_ = address;
listen_called_ = true;
return OK;
}
int FuzzedServerSocket::GetLocalAddress(IPEndPoint* address) const {
*address = listening_on_;
return OK;
}
int FuzzedServerSocket::Accept(std::unique_ptr<StreamSocket>* socket,
CompletionOnceCallback callback) {
if (first_accept_) {
base::ThreadTaskRunnerHandle::Get()->PostTask(
FROM_HERE, base::BindOnce(&FuzzedServerSocket::DispatchAccept,
weak_factory_.GetWeakPtr(), socket,
std::move(callback)));
}
first_accept_ = false;
return ERR_IO_PENDING;
}
void FuzzedServerSocket::DispatchAccept(std::unique_ptr<StreamSocket>* socket,
CompletionOnceCallback callback) {
std::unique_ptr<FuzzedSocket> connected_socket(
std::make_unique<FuzzedSocket>(data_provider_, net_log_));
// The Connect call should always succeed synchronously, without using the
// callback, since connected_socket->set_fuzz_connect_result(true) has not
// been called.
CHECK_EQ(net::OK, connected_socket->Connect(CompletionOnceCallback()));
*socket = std::move(connected_socket);
std::move(callback).Run(OK);
}
} // namespace net
|
