cmdline support

rv32imac_args_tests.s

@@ -0,0 +1,91 @@
+.section .text
+.option norvc
+.global _start
+
+# Helper macro to exit with code if condition fails
+.macro ASSERT_EQ reg, val, error_code
+  li t4, \val
+  beq \reg, t4, .L_ok_\@
+  li a0, \error_code
+  li a7, 93
+  ecall
+.L_ok_\@:
+.endm
+
+# Compare string at register \str_reg with data label \label, exit with \error_code if not equal
+.macro ASSERT_STR_EQ str_reg, label, error_code
+  la t0, \label
+  mv t1, \str_reg
+.L_loop_\@:
+  lbu t2, 0(t0)
+  lbu t3, 0(t1)
+  bne t2, t3, .L_fail_\@
+  beqz t2, .L_ok_\@
+  addi t0, t0, 1
+  addi t1, t1, 1
+  j .L_loop_\@
+.L_fail_\@:
+  li a0, \error_code
+  li a7, 93
+  ecall
+.L_ok_\@:
+.endm
+
+_start:
+  # sp + 0: argc
+  # sp + 4: argv[0]
+  # sp + 8: argv[1]
+  # sp + 12: argv[2]
+
+  # 1. Print verification start message
+  li a7, 64       # sys_write
+  li a0, 1        # stdout
+  la a1, msg_start
+  li a2, 32       # message length
+  ecall
+
+  # 2. Check argc == 3
+  lw t0, 0(sp)
+  ASSERT_EQ t0, 3, 10
+
+  # 3. Check argv[0] is not null and not empty
+  lw t0, 4(sp)
+  beqz t0, .L_argv0_bad
+  lbu t1, 0(t0)
+  beqz t1, .L_argv0_bad
+  j .L_argv0_ok
+.L_argv0_bad:
+  li a0, 11
+  li a7, 93
+  ecall
+.L_argv0_ok:
+
+  # 4. Check argv[1] == "arg1"
+  lw t0, 8(sp)
+  ASSERT_STR_EQ t0, str_arg1, 12
+
+  # 5. Check argv[2] == "arg2"
+  lw t0, 12(sp)
+  ASSERT_STR_EQ t0, str_arg2, 13
+
+  # 6. Print success message
+  li a7, 64       # sys_write
+  li a0, 1        # stdout
+  la a1, msg_success
+  li a2, 32       # message length
+  ecall
+
+  # 7. Exit with code 0 on success
+  li a0, 0
+  li a7, 93
+  ecall
+
+.section .rodata
+msg_start:
+  .ascii "Running Command Line Args Tests\n"
+msg_success:
+  .ascii "Command Line Args Tests PASSED!\n"
+str_arg1:
+  .asciz "arg1"
+str_arg2:
+  .asciz "arg2"