Add dummy name resolver

Co-authored-by: Jonatan Rhodin <jonatan.rhodin@mullvad.net>

3 files changed, 60 insertions, 1 deletions

diff --git a/android/lib/daemon-grpc/src/main/kotlin/net/mullvad/mullvadvpn/lib/daemon/grpc/ManagementService.kt b/android/lib/daemon-grpc/src/main/kotlin/net/mullvad/mullvadvpn/lib/daemon/grpc/ManagementService.kt
index d05df426f8..7f1ae90b45 100644
--- a/android/lib/daemon-grpc/src/main/kotlin/net/mullvad/mullvadvpn/lib/daemon/grpc/ManagementService.kt
+++ b/android/lib/daemon-grpc/src/main/kotlin/net/mullvad/mullvadvpn/lib/daemon/grpc/ManagementService.kt
@@ -41,6 +41,7 @@ import mullvad_daemon.management_interface.ManagementInterface
 import mullvad_daemon.management_interface.ManagementServiceGrpcKt
 import net.mullvad.mullvadvpn.lib.daemon.grpc.mapper.fromDomain
 import net.mullvad.mullvadvpn.lib.daemon.grpc.mapper.toDomain
+import net.mullvad.mullvadvpn.lib.daemon.grpc.resolver.DummyNameResolverFactory
 import net.mullvad.mullvadvpn.lib.daemon.grpc.util.LogInterceptor
 import net.mullvad.mullvadvpn.lib.daemon.grpc.util.connectivityFlow
 import net.mullvad.mullvadvpn.lib.model.AccountData
@@ -142,6 +143,10 @@ class ManagementService(
                 rpcSocketFile.absolutePath,
                 LocalSocketAddress.Namespace.FILESYSTEM
             )
+            // We need to provide a DummyNameResolver to avoid default NameResolver making incorrect
+            // InetSocketAddress look ups. For more info see:
+            // https://github.com/grpc/grpc-java/issues/11442
+            .nameResolverFactory(DummyNameResolverFactory())
             .build()
 
     val connectionState: StateFlow<GrpcConnectivityState> =
@@ -194,7 +199,6 @@ class ManagementService(
             error("ManagementService already started")
         }
 
-        channel.resetConnectBackoff()
         job = scope.launch { subscribeEvents() }
     }
 
@@ -204,6 +208,8 @@ class ManagementService(
         job = null
     }
 
+    fun enterIdle() = channel.enterIdle()
+
     private suspend fun subscribeEvents() =
         withContext(Dispatchers.IO) {
             launch {
diff --git a/android/lib/daemon-grpc/src/main/kotlin/net/mullvad/mullvadvpn/lib/daemon/grpc/resolver/DummyNameResolver.kt b/android/lib/daemon-grpc/src/main/kotlin/net/mullvad/mullvadvpn/lib/daemon/grpc/resolver/DummyNameResolver.kt
new file mode 100644
index 0000000000..37a3c9cdc0
--- /dev/null
+++ b/android/lib/daemon-grpc/src/main/kotlin/net/mullvad/mullvadvpn/lib/daemon/grpc/resolver/DummyNameResolver.kt
@@ -0,0 +1,35 @@
+package net.mullvad.mullvadvpn.lib.daemon.grpc.resolver
+
+import io.grpc.EquivalentAddressGroup
+import io.grpc.NameResolver
+import java.net.InetSocketAddress
+
+class DummyNameResolver : NameResolver() {
+
+    override fun getServiceAuthority(): String = SERVICE_AUTHORITY
+
+    override fun start(listener: Listener2) {
+        val resolutionResult =
+            ResolutionResult.newBuilder()
+                .setAddresses(
+                    listOf(
+                        EquivalentAddressGroup(
+                            InetSocketAddress.createUnresolved(DUMMY_HOST, DUMMY_PORT)
+                        )
+                    )
+                )
+                .build()
+
+        listener.onResult(resolutionResult)
+    }
+
+    override fun shutdown() {
+        // Do nothing
+    }
+
+    companion object {
+        const val SERVICE_AUTHORITY = "localhost"
+        private const val DUMMY_HOST = ""
+        private const val DUMMY_PORT = 80
+    }
+}
diff --git a/android/lib/daemon-grpc/src/main/kotlin/net/mullvad/mullvadvpn/lib/daemon/grpc/resolver/DummyNameResolverFactory.kt b/android/lib/daemon-grpc/src/main/kotlin/net/mullvad/mullvadvpn/lib/daemon/grpc/resolver/DummyNameResolverFactory.kt
new file mode 100644
index 0000000000..1a8141985d
--- /dev/null
+++ b/android/lib/daemon-grpc/src/main/kotlin/net/mullvad/mullvadvpn/lib/daemon/grpc/resolver/DummyNameResolverFactory.kt
@@ -0,0 +1,18 @@
+package net.mullvad.mullvadvpn.lib.daemon.grpc.resolver
+
+import io.grpc.NameResolver
+import java.net.URI
+
+class DummyNameResolverFactory : NameResolver.Factory() {
+    override fun newNameResolver(targetUri: URI, args: NameResolver.Args): NameResolver {
+        return DummyNameResolver()
+    }
+
+    override fun getDefaultScheme(): String {
+        return DNS_SCHEME
+    }
+
+    companion object {
+        private const val DNS_SCHEME = "dns"
+    }
+}