sh: Allow selecting disk 5

fox32-arch · May 5, 2024 · 979617e · 979617e
1 parent 2c2f017
commit 979617e
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/applications/sh/commands/disk.asm b/applications/sh/commands/disk.asm
@@ -10,23 +10,29 @@ shell_disk_command:
     ; r0: disk ID
 
     ; check if it's in range
-    cmp r0, 4
+    cmp r0, 5
     ifgt jmp shell_disk_command_out_of_range
-    ifnz jmp shell_disk_command_is_disk
 
-    call is_romdisk_available
-    ifnz ret
-    jmp shell_disk_command_is_romdisk
+    cmp r0, 4
+    ifz jmp shell_disk_command_is_romdisk
+
+    cmp r0, 5
+    ifz jmp shell_disk_command_is_ramdisk
 shell_disk_command_is_disk:
     ; OR it with the IO port to get the current insert state of a disk
     or r0, 0x80001000
     in r1, r0
     cmp r1, 0
     ifz ret
-shell_disk_command_is_romdisk:
-    ; set the current disk ID
     call set_current_disk_id
-
+    ret
+shell_disk_command_is_romdisk:
+    call is_romdisk_available
+    ifz call set_current_disk_id
+    ret
+shell_disk_command_is_ramdisk:
+    call is_ramdisk_formatted
+    ifz call set_current_disk_id
     ret
 
 shell_disk_command_out_of_range: