RUM-6454: fixing potential issues with subdomain host lookup

DataDog · Dec 5, 2024 · 8715a13 · 8715a13
1 parent 53e3a46
commit 8715a13
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/...n/kotlin/com/datadog/android/core/internal/net/DefaultFirstPartyHostHeaderTypeResolver.kt b/...n/kotlin/com/datadog/android/core/internal/net/DefaultFirstPartyHostHeaderTypeResolver.kt
@@ -48,10 +48,11 @@ class DefaultFirstPartyHostHeaderTypeResolver(
     /** @inheritdoc */
     override fun headerTypesForUrl(url: HttpUrl): Set<TracingHeaderType> {
         val host = url.host
-        val filteredHosts = knownHosts.filter {
-            it.key == "*" || it.key == host || host.endsWith(".${it.key}")
-        }
-        return filteredHosts.values.flatten().toSet()
+
+        return knownHosts[host]
+            ?: knownHosts.entries.firstOrNull { host.endsWith(".${it.key}") }?.value
+            ?: knownHosts["*"]
+            ?: emptySet()
     }
 
     /** @inheritdoc */

diff --git a/...tlin/com/datadog/android/core/internal/net/DefaultFirstPartyHostHeaderTypeResolverTest.kt b/...tlin/com/datadog/android/core/internal/net/DefaultFirstPartyHostHeaderTypeResolverTest.kt
@@ -350,6 +350,22 @@ internal class DefaultFirstPartyHostHeaderTypeResolverTest {
         }
     }
 
+    @Test
+    fun `M return correct header type W headerTypesForUrl(String) {domain and subdomain has different types}`() {
+        val resolver = DefaultFirstPartyHostHeaderTypeResolver(
+            mapOf(
+                "bar.com" to setOf(TracingHeaderType.DATADOG),
+                "foo.bar.com" to setOf(TracingHeaderType.TRACECONTEXT)
+            )
+        )
+
+        assertThat(resolver.headerTypesForUrl("http://bar.com"))
+            .isEqualTo(setOf(TracingHeaderType.DATADOG))
+
+        assertThat(resolver.headerTypesForUrl("http://foo.bar.com"))
+            .isEqualTo(setOf(TracingHeaderType.TRACECONTEXT))
+    }
+
     companion object {
         private const val HOST_REGEX = "([a-z][a-z0-9_~-]{3,9}\\.){1,4}[a-z][a-z0-9]{2,3}"
     }