unix/main: Add coverage test for mounting ROMFS filesystem at startup.

dpgeorge · dpgeorge · commit ffcc0f6d2999 · 2025-03-04T23:12:08.000+11:00
Signed-off-by: Damien George &lt;damien@micropython.org&gt;
diff --git a/ports/unix/main.c b/ports/unix/main.c
@@ -549,7 +549,14 @@ MP_NOINLINE int main_(int argc, char **argv) {
             MP_OBJ_NEW_QSTR(MP_QSTR__slash_),
         };
         mp_vfs_mount(2, args, (mp_map_t *)&mp_const_empty_map);
+
+        // Make sure the root that was just mounted is the current VFS (it's always at
+        // the end of the linked list).  Can't use chdir('/') because that will change
+        // the current path within the VfsPosix object.
         MP_STATE_VM(vfs_cur) = MP_STATE_VM(vfs_mount_table);
+        while (MP_STATE_VM(vfs_cur)->next != NULL) {
+            MP_STATE_VM(vfs_cur) = MP_STATE_VM(vfs_cur)->next;
+        }
     }
     #endif
 
@@ -805,6 +812,21 @@ void nlr_jump_fail(void *val) {
     exit(1);
 }
 
+#if defined(MICROPY_UNIX_COVERAGE)
+static uint8_t romfs_buf[4] = { 0xd2, 0xcd, 0x31, 0x00 }; // empty ROMFS
+static const MP_DEFINE_MEMORYVIEW_OBJ(romfs_obj, 'B', 0, sizeof(romfs_buf), romfs_buf);
+#endif
+
 mp_obj_t mp_vfs_rom_ioctl(size_t n_args, const mp_obj_t *args) {
+    #if defined(MICROPY_UNIX_COVERAGE)
+    switch (mp_obj_get_int(args[0])) {
+        case MP_VFS_ROM_IOCTL_GET_NUMBER_OF_SEGMENTS:
+            return MP_OBJ_NEW_SMALL_INT(1);
+
+        case MP_VFS_ROM_IOCTL_GET_SEGMENT:
+            return MP_OBJ_FROM_PTR(&romfs_obj);
+    }
+    #endif
+
     return MP_OBJ_NEW_SMALL_INT(-MP_EINVAL);
 }
