1313#include < boost/noncopyable.hpp>
1414
1515#include < libp2p/common/metrics/instance_count.hpp>
16+ #include < libp2p/connection/as_asio_read_write.hpp>
1617#include < libp2p/connection/secure_connection.hpp>
1718#include < libp2p/crypto/key_marshaller.hpp>
1819#include < libp2p/peer/identity_manager.hpp>
@@ -25,28 +26,23 @@ namespace libp2p::connection {
2526 public std::enable_shared_from_this<TlsConnection>,
2627 private boost::noncopyable {
2728 public:
28- // / lower level socket type is TCP
29- using tcp_socket_t = boost::asio::ip::tcp::socket;
30-
31- // / reference as a parameter here allows to upgrade established TCP
32- // / connection
33- using ssl_socket_t = boost::asio::ssl::stream<tcp_socket_t &>;
29+ using ssl_socket_t = boost::asio::ssl::stream<AsAsioReadWrite>;
3430
3531 // / Upgraded connection passed to this callback
3632 using HandshakeCallback = std::function<void (
3733 outcome::result<std::shared_ptr<connection::SecureConnection>>)>;
3834
3935 // / Ctor.
40- // / \param original_connection TCP connection, established at the moment
36+ // / \param original_connection connection, established at the moment
4137 // / \param ssl_context Wrapper around SSL_CTX
4238 // / \param idmgr Identity manager, contains this host's keys
43- // / \param tcp_socket Raw socket extracted from raw connection
39+ // / \param io_context Asio io context
4440 // / \param remote_peer Expected peer id of remote peer, has value for
4541 // / outbound connections
4642 TlsConnection (std::shared_ptr<LayerConnection> original_connection,
4743 std::shared_ptr<boost::asio::ssl::context> ssl_context,
4844 const peer::IdentityManager &idmgr,
49- tcp_socket_t &tcp_socket ,
45+ std::shared_ptr<boost::asio::io_context> io_context ,
5046 boost::optional<peer::PeerId> remote_peer);
5147
5248 // / Performs async handshake and passes its result into callback. This fn is
@@ -113,7 +109,7 @@ namespace libp2p::connection {
113109 // / Local peer id
114110 const peer::PeerId local_peer_;
115111
116- // / Raw TCP connection
112+ // / Raw connection
117113 std::shared_ptr<LayerConnection> original_connection_;
118114
119115 // / SSL context, shared among connections
0 commit comments