Feat/advertise local hostports (#392)

* respond to authenticate with local host ports * wip
drachtio · Nov 22, 2024 · 5dc23a4 · 5dc23a4
1 parent 3d3e6d9
commit 5dc23a4
Show file tree

Hide file tree

Showing 5 changed files with 29 additions and 2 deletions.
diff --git a/src/client.cpp b/src/client.cpp
@@ -163,10 +163,12 @@ namespace drachtio {
                 return false ;       
             } 
             else {
-                vector<string> hps ;
+                vector<string> hps, local_hps ;
                 theOneAndOnlyController->getMyHostports( hps ) ;
+                theOneAndOnlyController->getMyLocalHostports( local_hps ) ;
                 string hostports = boost::algorithm::join(hps, ",") ;
-                string response = hostports + "|" + DRACHTIO_VERSION;
+                string localHostports = boost::algorithm::join(local_hps, ",") ;
+                string response = hostports + "|" + DRACHTIO_VERSION + "|" + localHostports ;
                 createResponseMsg( tokens[0], msgResponse, true, response.c_str()) ;
                 DR_LOG(log_debug) << "Client::processAuthentication - secret validated successfully: " << secret ;
                 return true ;

diff --git a/src/controller.cpp b/src/controller.cpp
@@ -1791,6 +1791,9 @@ namespace drachtio {
     void DrachtioController::getMyHostports( vector<string>& vec ) {
       return SipTransport::getAllHostports( vec ) ;
     }
+    void DrachtioController::getMyLocalHostports( vector<string>& vec ) {
+      return SipTransport::getAllLocalHostports( vec ) ;
+    }
     bool DrachtioController::getMySipAddress( const char* proto, string& host, string& port, bool ipv6 ) {
         string desc, p ;
         const tport_t* tp = getTportForProtocol(host, proto) ;

diff --git a/src/controller.hpp b/src/controller.hpp
@@ -149,6 +149,7 @@ namespace drachtio {
     su_home_t* getHome(void) { return m_home; }
 
     void getMyHostports( vector<string>& vec ) ;
+    void getMyLocalHostports( vector<string>& vec ) ;
 
     bool getMySipAddress( const char* proto, string& host, string& port, bool ipv6 = false ) ;
 

diff --git a/src/sip-transports.cpp b/src/sip-transports.cpp
@@ -173,6 +173,17 @@ namespace drachtio {
     s += getPort() ;
   }
 
+  void SipTransport::getLocalHostport(string& s) {
+    if( hasTport() ) {
+      s += getProtocol() ;
+      s += "/" ;
+      s += getHost() ;
+      s += ":" ;
+      s += getPort() ;
+    }
+  }
+
+
   void SipTransport::getBindableContactUri(string& contact) {
     contact = m_contactScheme ;
     contact.append(":");
@@ -538,6 +549,14 @@ namespace drachtio {
       vec.push_back(desc) ;
     }
   }
+  void SipTransport::getAllLocalHostports( vector<string>& vec ) {
+    for (mapTport2SipTransport::const_iterator it = m_mapTport2SipTransport.begin(); m_mapTport2SipTransport.end() != it; ++it ) {
+      std::shared_ptr<SipTransport> p = it->second ;
+      string desc ;
+      p->getLocalHostport(desc);
+      vec.push_back(desc) ;
+    }
+  }
 
   bool SipTransport::isLocalAddress(const char* szHost, tport_t* tp) {
     if( tp ) {

diff --git a/src/sip-transports.hpp b/src/sip-transports.hpp
@@ -78,6 +78,7 @@ namespace drachtio {
 
     void getDescription(string& s, bool shortVersion = true) ;
     void getHostport(string& s) ;
+    void getLocalHostport(string& s) ;
 
     uint32_t getOctetMatchCount(const string& address);
 
@@ -86,6 +87,7 @@ namespace drachtio {
     static std::shared_ptr<SipTransport> findAppropriateTransport(const char* remoteHost, const char* proto = "udp") ;
     static void logTransports() ;
     static void getAllHostports( vector<string>& vec ) ;
+    static void getAllLocalHostports( vector<string>& vec ) ;
     static void getAllExternalIps( vector<string>& vec ) ;
     static void getAllExternalContacts( vector< pair<string, string> >& vec ) ;
     static bool isLocalAddress(const char* szHost, tport_t* tp = NULL) ;