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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef SERVICES_NETWORK_TCP_BOUND_SOCKET_H_
#define SERVICES_NETWORK_TCP_BOUND_SOCKET_H_
#include <memory>
#include "base/component_export.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
#include "mojo/public/cpp/bindings/receiver_set.h"
#include "net/base/ip_endpoint.h"
#include "net/socket/tcp_socket.h"
#include "net/traffic_annotation/network_traffic_annotation.h"
#include "services/network/public/mojom/tcp_socket.mojom.h"
#include "services/network/tcp_server_socket.h"
namespace net {
class IPEndPoint;
class NetLog;
} // namespace net
namespace network {
class SocketFactory;
// A socket bound to an address. Can be converted into either a TCPServerSocket
// or a TCPConnectedSocket.
class COMPONENT_EXPORT(NETWORK_SERVICE) TCPBoundSocket
: public mojom::TCPBoundSocket {
public:
// Constructs a TCPBoundSocket. |socket_factory| must outlive |this|. When a
// new connection is accepted, |socket_factory| will be notified to take
// ownership of the connection.
TCPBoundSocket(SocketFactory* socket_factory,
net::NetLog* net_log,
const net::NetworkTrafficAnnotationTag& traffic_annotation);
TCPBoundSocket(const TCPBoundSocket&) = delete;
TCPBoundSocket& operator=(const TCPBoundSocket&) = delete;
~TCPBoundSocket() override;
// Attempts to bind a socket to the specified address. Returns net::OK on
// success, setting |local_addr_out| to the bound address. Returns a network
// error code on failure. Must be called before Listen() or Connect().
int Bind(const net::IPEndPoint& local_addr, net::IPEndPoint* local_addr_out);
// Sets the id used to remove the socket from the owning ReceiverSet. Must be
// called before Listen() or Connect().
void set_id(mojo::ReceiverId receiver_id) { receiver_id_ = receiver_id; }
// mojom::TCPBoundSocket implementation.
void Listen(uint32_t backlog,
mojo::PendingReceiver<mojom::TCPServerSocket> receiver,
ListenCallback callback) override;
void Connect(const net::AddressList& remote_addr,
mojom::TCPConnectedSocketOptionsPtr tcp_connected_socket_options,
mojo::PendingReceiver<mojom::TCPConnectedSocket> receiver,
mojo::PendingRemote<mojom::SocketObserver> observer,
ConnectCallback callback) override;
private:
void OnConnectComplete(int result,
const std::optional<net::IPEndPoint>& local_addr,
const std::optional<net::IPEndPoint>& peer_addr,
mojo::ScopedDataPipeConsumerHandle receive_stream,
mojo::ScopedDataPipeProducerHandle send_stream);
virtual int ListenInternal(int backlog);
net::IPEndPoint bind_address_;
mojo::ReceiverId receiver_id_ = -1;
const raw_ptr<SocketFactory> socket_factory_;
std::unique_ptr<net::TCPSocket> socket_;
const net::NetworkTrafficAnnotationTag traffic_annotation_;
mojo::PendingReceiver<mojom::TCPConnectedSocket> connected_socket_receiver_;
ConnectCallback connect_callback_;
// Takes ownership of |socket_| if Connect() is called.
std::unique_ptr<TCPConnectedSocket> connecting_socket_;
base::WeakPtrFactory<TCPBoundSocket> weak_factory_{this};
};
} // namespace network
#endif // SERVICES_NETWORK_TCP_BOUND_SOCKET_H_
|
