From a23c92df9f6604dae1c8baa50021820bdb48b2da Mon Sep 17 00:00:00 2001
From: yeoleobun <yangli245@live.com>
Date: Wed, 15 Apr 2026 20:14:44 +0800
Subject: [PATCH] fix: bypass in-dialog re-INVITE

---
 src/proxy/auth.rs            | 13 +++++++++++++
 src/proxy/tests/test_auth.rs |  1 +
 2 files changed, 14 insertions(+)

diff --git a/src/proxy/auth.rs b/src/proxy/auth.rs
index ea34d134..fc6598d1 100644
--- a/src/proxy/auth.rs
+++ b/src/proxy/auth.rs
@@ -5,11 +5,13 @@ use crate::call::{CalleeDisplayName, TransactionCookie, TrunkContext};
 use crate::config::ProxyConfig;
 use anyhow::{Error, Result};
 use async_trait::async_trait;
+use rsipstack::dialog::DialogId;
 use rsipstack::sip::Header;
 use rsipstack::sip::headers::{ProxyAuthenticate, WwwAuthenticate};
 use rsipstack::sip::prelude::HeadersExt;
 use rsipstack::sip::typed::Authorization;
 use rsipstack::dialog::authenticate::verify_digest;
+use rsipstack::transaction::key::TransactionRole;
 use rsipstack::transaction::transaction::Transaction;
 use std::sync::Arc;
 use tokio_util::sync::CancellationToken;
@@ -215,6 +217,17 @@ impl ProxyModule for AuthModule {
             .map(|d| d.to_string())
             .unwrap_or_else(|| "unknown".to_string());
 
+        if tx.original.method == rsipstack::sip::Method::Invite {
+            if let Ok(dialog_id) = DialogId::try_from((&tx.original, TransactionRole::Server)) {
+                if !dialog_id.local_tag.is_empty()
+                    && self.server.dialog_layer.get_dialog(&dialog_id).is_some()
+                {
+                    cookie.set_user(tx_user.clone());
+                    return Ok(ProxyAction::Continue);
+                }
+            }
+        }
+
         for backend in self.server.auth_backend.iter() {
             match backend.authenticate(&tx.original, &cookie).await {
                 Ok(Some(mut user)) => {
diff --git a/src/proxy/tests/test_auth.rs b/src/proxy/tests/test_auth.rs
index 09951f20..3a641fa5 100644
--- a/src/proxy/tests/test_auth.rs
+++ b/src/proxy/tests/test_auth.rs
@@ -16,6 +16,7 @@ use crate::proxy::server::SipServerInner;
 use crate::proxy::user::MemoryUserBackend;
 use crate::proxy::{ProxyAction, ProxyModule};
 use rsipstack::sip::Header;
+use rsipstack::sip::Method;
 use rsipstack::sip::prelude::{HasHeaders, HeadersExt};
 use rsipstack::sip::services::DigestGenerator;
 use rsipstack::EndpointBuilder;