react-native-litert-lm 0.3.6 → 0.4.0

package/android/src/test/java/com/margelo/nitro/dev/litert/litertlm/HybridLiteRTLMTest.kt

@@ -0,0 +1,83 @@
+package com.margelo.nitro.dev.litert.litertlm
+
+import org.junit.Assert.*
+import org.junit.Before
+import org.junit.After
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.robolectric.RobolectricTestRunner
+import org.robolectric.RuntimeEnvironment
+import dev.litert.litertlm.LiteRTLMInitProvider
+import java.lang.IllegalArgumentException
+
+@RunWith(RobolectricTestRunner::class)
+class HybridLiteRTLMTest {
+    private lateinit var bridge: HybridLiteRTLM
+
+    @Before
+    fun setUp() {
+        // Initialize the static applicationContext inside LiteRTLMInitProvider via reflection
+        try {
+            val field = LiteRTLMInitProvider::class.java.getDeclaredField("applicationContext")
+            field.isAccessible = true
+            field.set(null, RuntimeEnvironment.getApplication())
+        } catch (e: Exception) {
+            e.printStackTrace()
+        }
+        
+        bridge = HybridLiteRTLM()
+    }
+
+    @After
+    fun tearDown() {
+        bridge.close()
+    }
+
+    @Test
+    fun testAndroidPathTraversalPrevention() {
+        val traversals = arrayOf("../secret", "/etc/hosts", "nested\\..\\file", "..", "../", "..\\")
+        for (traversal in traversals) {
+            val promise = bridge.deleteModel(traversal)
+            assertNotNull("Promise should not be null", promise)
+            assertTrue("Promise should be completed", promise.isCompleted)
+            assertNotNull("Promise should have rejected with an error for filename: $traversal", promise.error)
+            val error = promise.error!!
+            val errMsg = error.message ?: error.cause?.message ?: ""
+            assertTrue("Expected message to contain traversal warning, got: $errMsg",
+                errMsg.contains("path traversal or directory separators are not allowed"))
+        }
+    }
+
+    @Test
+    fun testAndroidHTTPSDownloadEnforcement() {
+        val promise = bridge.downloadModel("http://insecure.site/model.bin", "model.bin", null)
+        assertNotNull("Promise should not be null", promise)
+        assertTrue("Promise should be completed", promise.isCompleted)
+        assertNotNull("Promise should have rejected with an error", promise.error)
+        val error = promise.error!!
+        val errMsg = error.message ?: error.cause?.message ?: ""
+        assertTrue("Expected message to contain HTTPS warning, got: $errMsg",
+            errMsg.contains("HTTPS is required for security"))
+    }
+
+    @Test
+    fun testAndroidMemoryTelemetry() {
+        val mem = bridge.getMemoryUsage()
+        assertNotNull(mem)
+        assertTrue(mem.nativeHeapBytes >= 0.0)
+        assertTrue(mem.residentBytes >= 0.0)
+        assertTrue(mem.availableMemoryBytes >= 0.0)
+    }
+
+    @Test
+    fun testAndroidInitialStats() {
+        val stats = bridge.getStats()
+        assertNotNull(stats)
+        assertEquals(0.0, stats.promptTokens, 0.0)
+        assertEquals(0.0, stats.completionTokens, 0.0)
+        assertEquals(0.0, stats.totalTokens, 0.0)
+        assertEquals(0.0, stats.timeToFirstToken, 0.0)
+        assertEquals(0.0, stats.totalTime, 0.0)
+        assertEquals(0.0, stats.tokensPerSecond, 0.0)
+    }
+}

package/cpp/include/README.md

@@ -1,34 +1,32 @@
 # LiteRT-LM Headers Fallback
 
-This directory is a fallback location for LiteRT-LM C++ headers when Prefab doesn't expose them correctly.
+This directory contains the LiteRT-LM C API header (`litert_lm_engine.h`) used by the iOS C++ implementation.
 
 ## If Headers Are Missing
 
-If you get compilation errors like `litert/lm/engine.h: No such file or directory`, you need to manually copy LiteRT-LM headers here:
+If you get compilation errors like `litert_lm_engine.h: No such file or directory`, you need to manually copy the LiteRT-LM C API header here:
 
 1. Clone LiteRT-LM repository:
 
    ```bash
    git clone https://github.com/google-ai-edge/LiteRT-LM.git /tmp/LiteRT-LM
+   cd /tmp/LiteRT-LM && git checkout v0.10.2
    ```
 
-2. Copy the headers:
+2. Copy the header:
    ```bash
-   cp -r /tmp/LiteRT-LM/runtime/include/litert ./
+   cp /tmp/LiteRT-LM/c/litert_lm_engine.h ./
    ```
 
 The expected directory structure after copying:
 
 ```
 cpp/include/
-└── litert/
-    └── lm/
-        ├── engine.h
-        ├── conversation.h
-        ├── types.h
-        └── ...
+├── litert_lm_engine.h   # LiteRT-LM C API header
+├── stb_image.h          # Image loading for multimodal
+└── README.md
 ```
 
 ## Note
 
-The ideal scenario is that the Maven package `litertlm-android:0.9.0-alpha01` exposes headers via Prefab, making this directory unnecessary. This is just a fallback.
+On **Android**, headers are provided by the `litertlm-android` AAR via Prefab — this directory is only needed for the **iOS** build which uses the raw C API via the prebuilt XCFramework.