OLD | NEW |
| (Empty) |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #ifndef BLIMP_NET_TCP_CLIENT_TRANSPORT_H_ | |
6 #define BLIMP_NET_TCP_CLIENT_TRANSPORT_H_ | |
7 | |
8 #include <memory> | |
9 #include <string> | |
10 | |
11 #include "base/callback.h" | |
12 #include "base/macros.h" | |
13 #include "blimp/net/blimp_net_export.h" | |
14 #include "blimp/net/blimp_transport.h" | |
15 #include "net/base/address_list.h" | |
16 #include "net/base/net_errors.h" | |
17 | |
18 namespace net { | |
19 class ClientSocketFactory; | |
20 class NetLog; | |
21 class StreamSocket; | |
22 } // namespace net | |
23 | |
24 namespace blimp { | |
25 | |
26 class MessagePort; | |
27 | |
28 // BlimpTransport which creates a TCP connection to one of the specified | |
29 // |addresses| on each call to Connect(). | |
30 class BLIMP_NET_EXPORT TCPClientTransport : public BlimpTransport { | |
31 public: | |
32 TCPClientTransport(const net::IPEndPoint& ip_endpoint, | |
33 net::NetLog* net_log); | |
34 ~TCPClientTransport() override; | |
35 | |
36 void SetClientSocketFactoryForTest(net::ClientSocketFactory* factory); | |
37 | |
38 // BlimpTransport implementation. | |
39 void Connect(const net::CompletionCallback& callback) override; | |
40 std::unique_ptr<BlimpConnection> MakeConnection() override; | |
41 | |
42 std::unique_ptr<MessagePort> TakeMessagePort(); | |
43 | |
44 const char* GetName() const override; | |
45 | |
46 protected: | |
47 // Called when the TCP connection completed. | |
48 virtual void OnTCPConnectComplete(int result); | |
49 | |
50 // Called when the connection attempt completed or failed. | |
51 // Resets |socket_| if |result| indicates a failure (!= net::OK). | |
52 void OnConnectComplete(int result); | |
53 | |
54 // Methods for taking and setting |socket_|. Can be used by subclasses to | |
55 // swap out a socket for an upgraded one, e.g. adding SSL encryption. | |
56 std::unique_ptr<net::StreamSocket> TakeSocket(); | |
57 void SetSocket(std::unique_ptr<net::StreamSocket> socket); | |
58 | |
59 // Gets the socket factory instance. | |
60 net::ClientSocketFactory* socket_factory() const; | |
61 | |
62 private: | |
63 net::IPEndPoint ip_endpoint_; | |
64 net::NetLog* net_log_; | |
65 net::CompletionCallback connect_callback_; | |
66 net::ClientSocketFactory* socket_factory_ = nullptr; | |
67 std::unique_ptr<net::StreamSocket> socket_; | |
68 | |
69 DISALLOW_COPY_AND_ASSIGN(TCPClientTransport); | |
70 }; | |
71 | |
72 } // namespace blimp | |
73 | |
74 #endif // BLIMP_NET_TCP_CLIENT_TRANSPORT_H_ | |
OLD | NEW |