new files

2026-07-26 03:45:22 +00:00 · 2026-06-09 00:02:37 +02:00
parent efe15967b8
commit d2d9ab6c69
5 changed files with 336 additions and 23 deletions
@@ -0,0 +1,117 @@
+package com.bitchat.android.mesh
+
+import android.os.Build
+import com.bitchat.android.model.IdentityAnnouncement
+import com.bitchat.android.model.RoutedPacket
+import com.bitchat.android.protocol.BitchatPacket
+import com.bitchat.android.protocol.MessageType
+import com.bitchat.android.protocol.SpecialRecipients
+import com.bitchat.android.services.meshgraph.MeshGraphService
+import com.bitchat.android.util.AppConstants
+import kotlinx.coroutines.runBlocking
+import org.junit.After
+import org.junit.Assert.assertFalse
+import org.junit.Assert.assertTrue
+import org.junit.Before
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.mockito.kotlin.any
+import org.mockito.kotlin.eq
+import org.mockito.kotlin.isNull
+import org.mockito.kotlin.mock
+import org.mockito.kotlin.never
+import org.mockito.kotlin.verify
+import org.mockito.kotlin.whenever
+import org.robolectric.RobolectricTestRunner
+import org.robolectric.RuntimeEnvironment
+import org.robolectric.annotation.Config
+
+@RunWith(RobolectricTestRunner::class)
+@Config(sdk = [Build.VERSION_CODES.P], manifest = Config.NONE)
+class MessageHandlerTest {
+    private lateinit var handler: MessageHandler
+    private lateinit var delegate: MessageHandlerDelegate
+
+    private val myPeerID = "1111222233334444"
+    private val peerID = "aaaabbbbccccdddd"
+    private val nickname = "peer"
+    private val noiseKey = ByteArray(32) { 0x0B }
+    private val signingKey = ByteArray(32) { 0x0A }
+    private val signature = ByteArray(64) { 1 }
+    private val announceClockSkewToleranceMs = 10 * 60 * 1000L
+
+    @Before
+    fun setup() {
+        MeshGraphService.resetForTesting()
+        handler = MessageHandler(myPeerID, RuntimeEnvironment.getApplication())
+        delegate = mock()
+        handler.delegate = delegate
+
+        whenever(delegate.getPeerInfo(peerID)).thenReturn(null)
+        whenever(delegate.verifyEd25519Signature(any(), any(), any())).thenReturn(true)
+        whenever(delegate.updatePeerInfo(any(), any(), any(), any(), any())).thenReturn(true)
+    }
+
+    @After
+    fun tearDown() {
+        MeshGraphService.resetForTesting()
+    }
+
+    @Test
+    fun `handleAnnounce accepts announce within clock skew tolerance for identity binding`() = runBlocking {
+        val packet = announcePacket(ageMs = AppConstants.Mesh.STALE_PEER_TIMEOUT_MS + 1_000)
+
+        val result = handler.handleAnnounce(RoutedPacket(packet, peerID, "direct-link"))
+
+        assertTrue("Announce within clock skew tolerance should still store peer identity", result)
+        verify(delegate).updatePeerInfo(eq(peerID), eq(nickname), any(), any(), eq(true))
+        verify(delegate).updatePeerIDBinding(eq(peerID), eq(nickname), any(), isNull())
+    }
+
+    @Test
+    fun `handleAnnounce accepts future announce within clock skew tolerance`() = runBlocking {
+        val packet = announcePacket(ageMs = -(AppConstants.Mesh.STALE_PEER_TIMEOUT_MS + 1_000))
+
+        val result = handler.handleAnnounce(RoutedPacket(packet, peerID, "direct-link"))
+
+        assertTrue("Future announce within clock skew tolerance should still store peer identity", result)
+        verify(delegate).updatePeerInfo(eq(peerID), eq(nickname), any(), any(), eq(true))
+        Unit
+    }
+
+    @Test
+    fun `handleAnnounce rejects announce older than clock skew tolerance`() = runBlocking {
+        val packet = announcePacket(ageMs = announceClockSkewToleranceMs + 1_000)
+
+        val result = handler.handleAnnounce(RoutedPacket(packet, peerID, "relay-link"))
+
+        assertFalse("Announce older than clock skew tolerance should not store peer identity", result)
+        verify(delegate, never()).updatePeerInfo(any(), any(), any(), any(), any())
+        verify(delegate, never()).updatePeerIDBinding(any(), any(), any(), any())
+    }
+
+    private fun announcePacket(
+        ageMs: Long,
+        ttl: UByte = (AppConstants.MESSAGE_TTL_HOPS.toInt() - 1).toUByte()
+    ): BitchatPacket {
+        val announcement = IdentityAnnouncement(
+            nickname = nickname,
+            noisePublicKey = noiseKey,
+            signingPublicKey = signingKey
+        )
+        return BitchatPacket(
+            version = 1u,
+            type = MessageType.ANNOUNCE.value,
+            senderID = peerID.hexToBytes(),
+            recipientID = SpecialRecipients.BROADCAST,
+            timestamp = (System.currentTimeMillis() - ageMs).toULong(),
+            payload = announcement.encode()!!,
+            signature = signature,
+            ttl = ttl
+        )
+    }
+
+    private fun String.hexToBytes(): ByteArray {
+        return chunked(2).map { it.toInt(16).toByte() }.toByteArray()
+    }
+}