Adds tests on shell & shell commands

shell.c

@@ -85,7 +85,10 @@ int asmsh_exec(asmsh_t *sh, const char *cmd)
 			ret = _compile_step(sh, cmd);
 			break;
 	}
-	asmsh_env_update_regs(sh->env);
+	if(!ret)
+	{
+		asmsh_env_update_regs(sh->env);
+	}
 	return ret;
 }
 

tests/asmsh_check.h

@@ -10,6 +10,8 @@
 #include <sys/types.h>
 #include <sys/stat.h>
 
+#define CHILD_PATH "../child"
+
 /**@brief Start a asmsh test
  *@param const char* suite_str test suite string
  *@param const char* tc_str test case string

tests/tests_compile.c

@@ -1,4 +1,4 @@
-#define _GNU_SOURCE
+#include "config.h"
 #include <check.h>
 #include <errno.h>
 #include <stdio.h>

tests/tests_shell.c

@@ -0,0 +1,52 @@
+#include "config.h"
+#include <check.h>
+#include <errno.h>
+#include <stdio.h>
+#include <string.h>
+#include <unistd.h>
+
+#include "asmsh_check.h"
+#include "shell.h"
+
+START_TEST(test_init)
+{
+	asmsh_t sh;
+	ck_assert_int_eq(asmsh_init(&sh, CHILD_PATH), 0);
+	asmsh_cleanup(&sh);
+}
+END_TEST
+
+START_TEST(test_exit)
+{
+	asmsh_t sh;
+	ck_assert_int_eq(asmsh_init(&sh, CHILD_PATH), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, "mov $60, %rax"), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, "xor %rdi, %rdi"), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, "syscall"), 1);
+	asmsh_cleanup(&sh);
+
+	ck_assert_int_eq(asmsh_init(&sh, CHILD_PATH), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, "mov $60, %rax"), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, "mov $0x29, %rdi"), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, "syscall"), 42);
+	asmsh_cleanup(&sh);
+}
+END_TEST
+
+START_TEST(test_cmd)
+{
+	asmsh_t sh;
+	ck_assert_int_eq(asmsh_init(&sh, CHILD_PATH), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, ".reg"), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, ".q"), 1);
+	asmsh_cleanup(&sh);
+
+}
+END_TEST
+
+ASMSH_CHECK_START("Testing shell", "testing shell init/exec functions")
+	ASMSH_ADD_TEST(test_init);
+	ASMSH_ADD_TEST(test_exit);
+	ASMSH_ADD_TEST(test_cmd);
+ASMSH_CHECK_END
+

tests/tests_shell_cmds.c

@@ -0,0 +1,74 @@
+#include "config.h"
+
+#include <check.h>
+#include <errno.h>
+#include <stdio.h>
+#include <unistd.h>
+
+#include "asmsh_check.h"
+#include "shell_cmds.h"
+#include "logger.h"
+
+typedef struct {
+	const char *cmd;
+	const char *parsed;
+	const int argc;
+	const char **args;
+} parse_cmds_t;
+
+static const char *args0[]={NULL};
+static const char *args1[]={"hello", "world", NULL};
+static const char *args2[]={"mov", "$0x42,", "%rax", NULL};
+static const parse_cmds_t parse_cmds[] = {
+	{".foo",
+	".foo", 0, args0},
+	{".a.b hello world",
+	 ".a.b", 2, args1},
+	{".b mov $0x42, %rax",
+	 ".b", 3, args2},
+};
+START_TEST(parse)
+{
+	asmsh_logger_t *log =  asmsh_logger_new(ASMSH_TRACE);
+	asmsh_logger_setup(log);
+	const parse_cmds_t *cmd = &parse_cmds[_i];
+	asmsh_cmd_args_t *args = asmsh_cmd_parse(cmd->cmd);
+	asmsh_logger_dprint(2, _default_logger);
+	ck_assert_ptr_nonnull(args);
+	ck_assert_str_eq(args->cmd, cmd->parsed);
+	ck_assert_int_eq(args->argc, cmd->argc);
+	for(int i=0; i<args->argc; i++)
+	{
+		ck_assert_str_eq(args->args[i], cmd->args[i]);
+	}
+	ck_assert_ptr_null(args->args[args->argc]);
+	asmsh_cmd_args_free(args);
+}
+END_TEST
+
+
+START_TEST(test_cmds)
+{
+	asmsh_t sh;
+	ck_assert_int_eq(asmsh_init(&sh, CHILD_PATH), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, ".h"), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, ".help"), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, ".b"), -1);
+	ck_assert_int_eq(asmsh_exec(&sh, ".bytecode syscall"), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, ".bytecode mov $42, %rax"), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, "push %rax"), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, ".b"), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, ".flags"), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, ".reset"), 0);
+	ck_assert_int_eq(asmsh_exec(&sh, ".b"), -1);
+	ck_assert_int_eq(asmsh_exec(&sh, ".xxdsdfsdfsdfsdf"), -1);
+	ck_assert_int_eq(asmsh_exec(&sh, ".q"), 1);
+	asmsh_cleanup(&sh);
+
+}
+END_TEST
+
+ASMSH_CHECK_START("shell commands tests", "testing shell commands")
+	ASMSH_ADD_LOOP(parse, parse_cmds);
+	ASMSH_ADD_TEST(test_cmds);
+ASMSH_CHECK_END