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