set correct port on forwarded headers

2 files changed, 2 insertions, 1 deletions

diff --git a/src/modules/mod.rs b/src/modules/mod.rs
index d3e0dd3..61f4b74 100644
--- a/src/modules/mod.rs
+++ b/src/modules/mod.rs
@@ -56,6 +56,7 @@ pub struct NodeContext {
     pub state: Arc<State>,
     pub addr: SocketAddr,
     pub secure: bool,
+    pub listen_addr: SocketAddr,
 }
 
 pub trait NodeKind: Send + Sync + 'static {
diff --git a/src/modules/proxy.rs b/src/modules/proxy.rs
index e92b8fc..2fa3538 100644
--- a/src/modules/proxy.rs
+++ b/src/modules/proxy.rs
@@ -54,7 +54,7 @@ impl Node for Proxy {
                 );
                 request.headers_mut().insert(
                     "x-forwarded-port",
-                    HeaderValue::from_str(&context.addr.port().to_string()).unwrap(),
+                    HeaderValue::from_str(&context.listen_addr.port().to_string()).unwrap(),
                 );
                 let scheme =
                     HeaderValue::from_str(if context.secure { "https" } else { "http" }).unwrap();