From 76e952830302271e07a4be9df6dfaa1c11e3e675 Mon Sep 17 00:00:00 2001
From: hozan23 <hozan23@proton.me>
Date: Wed, 13 Mar 2024 03:35:09 +0100
Subject: net: pass `Endpoint` to dial and listen functions

---
 p2p/src/listener.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'p2p/src/listener.rs')

diff --git a/p2p/src/listener.rs b/p2p/src/listener.rs
index ab6f7c1..254e4e6 100644
--- a/p2p/src/listener.rs
+++ b/p2p/src/listener.rs
@@ -155,7 +155,9 @@ impl Listener {
     async fn listend(&self, endpoint: &Endpoint) -> Result<Box<dyn ConnListener>> {
         if self.enable_tls {
             let tls_config = tls_server_config(&self.key_pair)?;
-            tls::listen(endpoint, tls_config).await
+            tls::listen_tls(endpoint, tls_config)
+                .await
+                .map(|l| Box::new(l) as Box<dyn ConnListener>)
         } else {
             listen(endpoint).await
         }
-- 
cgit v1.2.3