From 1e4141f737a2642ff80fef3e34ba0db1056962af Mon Sep 17 00:00:00 2001
From: Ludovic 'Archivist' Lagouardette <ludovic@nekoit.xyz>
Date: Wed, 25 Jan 2023 13:22:06 +0100
Subject: [PATCH] Added more instructions to get closer to the hello world

---
 include/molasses/parser_primitives.h |   5 +-
 src/main.cpp                         |  15 +--
 src/molasses/parser_primitives.cpp   | 194 ++++++++++++++++++++++++---
 3 files changed, 182 insertions(+), 32 deletions(-)

diff --git a/include/molasses/parser_primitives.h b/include/molasses/parser_primitives.h
index dd8bb7b..39eaa96 100644
--- a/include/molasses/parser_primitives.h
+++ b/include/molasses/parser_primitives.h
@@ -144,8 +144,9 @@ namespace molasses {
 	};
 	
 	parser_context parse(parser_context, const lexed_output&);
-	
-	parser_context register_integers(parser_context);
+
+    parser_context register_integers(parser_context);
+    parser_context register_i32_operations(parser_context);
 	
 	bool type_check(const parser_context&, const lexed_output&, const std::vector<symbol>&, std::vector<std::string> execution_input, const std::vector<std::string>& execution_output);
 }
diff --git a/src/main.cpp b/src/main.cpp
index 80c7929..181c8b9 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -44,20 +44,7 @@ int main() {
 	
 	molasses::parser_context ctx;
 	ctx = molasses::register_integers(ctx);
-	ctx.operations.emplace_back(
-        std::make_shared<molasses::primitive_operation>(
-            std::string{"+"},
-            std::vector<std::string>({"i32", "i32"}),
-            std::vector<std::string>({"i32"}),
-            std::vector<std::string>({
-                 "    popq %rax\n",
-                 "    popq %rbx\n",
-                 "    addl %ebx, %eax\n",
-                 "    andl $0xFFFFFFFF, %eax\n",
-                 "    pushq %rax\n"
-            })
-        )
-    );
+	ctx = molasses::register_i32_operations(ctx);
 	molasses::parse(ctx, lexed);
 	
 }
diff --git a/src/molasses/parser_primitives.cpp b/src/molasses/parser_primitives.cpp
index 3503ef8..06e9189 100644
--- a/src/molasses/parser_primitives.cpp
+++ b/src/molasses/parser_primitives.cpp
@@ -3,19 +3,178 @@
 #include <iostream>
 #include "molasses/parser_primitives.h"
 
+enum class architecture_t {
+    x86_64
+};
+
+constexpr architecture_t architecture = architecture_t::x86_64;
+
 namespace molasses {
-	parser_context register_integers(parser_context ctx) {
-		ctx.types.push_back(std::make_shared<primitive_type>("i8",1));
-		ctx.types.push_back(std::make_shared<primitive_type>("i16",2));
-		ctx.types.push_back(std::make_shared<primitive_type>("i32",4));
-		ctx.types.push_back(std::make_shared<primitive_type>("i64",8));
-		ctx.types.push_back(std::make_shared<primitive_type>("u8",1));
-		ctx.types.push_back(std::make_shared<primitive_type>("u16",2));
-		ctx.types.push_back(std::make_shared<primitive_type>("u32",4));
-		ctx.types.push_back(std::make_shared<primitive_type>("u64",8));
-		
-		return ctx;
-	}
+    parser_context register_integers(parser_context ctx) {
+        ctx.types.push_back(std::make_shared<primitive_type>("i8",1));
+        ctx.types.push_back(std::make_shared<primitive_type>("i16",2));
+        ctx.types.push_back(std::make_shared<primitive_type>("i32",4));
+        ctx.types.push_back(std::make_shared<primitive_type>("i64",8));
+        ctx.types.push_back(std::make_shared<primitive_type>("u8",1));
+        ctx.types.push_back(std::make_shared<primitive_type>("u16",2));
+        ctx.types.push_back(std::make_shared<primitive_type>("u32",4));
+        ctx.types.push_back(std::make_shared<primitive_type>("u64",8));
+
+        return ctx;
+    }
+
+    parser_context register_i32_operations(parser_context ctx) {
+        ctx.operations.emplace_back(
+            std::make_shared<molasses::primitive_operation>(
+                std::string{"+"},
+                std::vector<std::string>({"i32", "i32"}),
+                std::vector<std::string>({"i32"}),
+                std::vector<std::string>({
+                    "    popq %rax\n",
+                    "    popq %rbx\n",
+                    "    addl %ebx, %eax\n",
+                    "    andl $0xFFFFFFFF, %eax\n",
+                    "    pushq %rax\n"
+                })
+            )
+        );
+        ctx.operations.emplace_back(
+            std::make_shared<molasses::primitive_operation>(
+                std::string{"*"},
+                std::vector<std::string>({"i32", "i32"}),
+                std::vector<std::string>({"i32"}),
+                std::vector<std::string>({
+                    "    popq %rax\n",
+                    "    popq %rbx\n",
+                    "    multl %ebx, %eax\n",
+                    "    andl $0xFFFFFFFF, %eax\n",
+                    "    pushq %rax\n"
+                })
+            )
+        );
+        ctx.operations.emplace_back(
+            std::make_shared<molasses::primitive_operation>(
+                std::string{"-"},
+                std::vector<std::string>({"i32", "i32"}),
+                std::vector<std::string>({"i32"}),
+                std::vector<std::string>({
+                    "    popq %rax\n",
+                    "    popq %rbx\n",
+                    "    subl %ebx, %eax\n",
+                    "    andl $0xFFFFFFFF, %eax\n",
+                    "    pushq %rax\n"
+                })
+            )
+        );
+        ctx.operations.emplace_back(
+            std::make_shared<molasses::primitive_operation>(
+                std::string{"i32-to-i64"},
+                std::vector<std::string>({"i32"}),
+                std::vector<std::string>({"i64"}),
+                std::vector<std::string>({
+                    "    popq %rax\n",
+                    "    movslq %rax\n",
+                    "    pushq %rax\n"
+                })
+            )
+        );
+
+        ctx.operations.emplace_back(
+            std::make_shared<molasses::primitive_operation>(
+                std::string{"syscall1"},
+                std::vector<std::string>({"i64", "i64"}),
+                std::vector<std::string>({"i64"}),
+                std::vector<std::string>({
+                    "    popq %rax\n",
+                    "    popq %rdi\n",
+                    "    syscall\n",
+                    "    pushq %rax\n"
+                })
+            )
+        );
+        ctx.operations.emplace_back(
+            std::make_shared<molasses::primitive_operation>(
+                std::string{"syscall2"},
+                std::vector<std::string>({"i64", "i64", "i64"}),
+                std::vector<std::string>({"i64"}),
+                std::vector<std::string>({
+                    "    popq %rax\n",
+                    "    popq %rdi\n",
+                    "    popq %rsi\n",
+                    "    syscall\n",
+                    "    pushq %rax\n"
+                })
+            )
+        );
+        ctx.operations.emplace_back(
+            std::make_shared<molasses::primitive_operation>(
+                std::string{"syscall3"},
+                std::vector<std::string>({"i64", "i64", "i64", "i64"}),
+                std::vector<std::string>({"i64"}),
+                std::vector<std::string>({
+                    "    popq %rax\n",
+                    "    popq %rdi\n",
+                    "    popq %rsi\n",
+                    "    popq %rdx\n",
+                    "    syscall\n",
+                    "    pushq %rax\n"
+                })
+            )
+        );
+        ctx.operations.emplace_back(
+            std::make_shared<molasses::primitive_operation>(
+                std::string{"syscall4"},
+                std::vector<std::string>({"i64", "i64", "i64", "i64", "i64"}),
+                std::vector<std::string>({"i64"}),
+                std::vector<std::string>({
+                    "    popq %rax\n",
+                    "    popq %rdi\n",
+                    "    popq %rsi\n",
+                    "    popq %rdx\n",
+                    "    popq %r10\n",
+                    "    syscall\n",
+                    "    pushq %rax\n"
+                })
+            )
+        );
+        ctx.operations.emplace_back(
+            std::make_shared<molasses::primitive_operation>(
+                std::string{"syscall5"},
+                std::vector<std::string>({"i64", "i64", "i64", "i64", "i64", "i64"}),
+                std::vector<std::string>({"i64"}),
+                std::vector<std::string>({
+                    "    popq %rax\n",
+                    "    popq %rdi\n",
+                    "    popq %rsi\n",
+                    "    popq %rdx\n",
+                    "    popq %r10\n",
+                    "    popq %r8\n",
+                    "    syscall\n",
+                    "    pushq %rax\n"
+                })
+            )
+        );
+        ctx.operations.emplace_back(
+            std::make_shared<molasses::primitive_operation>(
+                std::string{"syscall6"},
+                std::vector<std::string>({"i64", "i64", "i64", "i64", "i64", "i64", "i64"}),
+                std::vector<std::string>({"i64"}),
+                std::vector<std::string>({
+                    "    popq %rax\n",
+                    "    popq %rdi\n",
+                    "    popq %rsi\n",
+                    "    popq %rdx\n",
+                    "    popq %r10\n",
+                    "    popq %r8\n",
+                    "    popq %r9\n",
+                    "    syscall\n",
+                    "    pushq %rax\n"
+                })
+            )
+        );
+
+        return ctx;
+    }
 	
 	std::vector<std::string> operator>>(std::vector<std::string> current_stack, const operation& next_op) {
 		{
@@ -90,7 +249,8 @@ namespace molasses {
 		return type_stack == execution_output;
 	}
 
-    std::vector<std::string> initialize_stack() {
+    std::vector<std::string> initialize_stack()
+    requires (architecture == architecture_t::x86_64) {
         return {
                 ".bss\n",// TODO: make threadlocal
                 "stack_instruction:\n",
@@ -215,7 +375,8 @@ namespace molasses {
 		return ctx;
 	}
 
-    std::vector<std::string> generate_call(std::string target) {
+    std::vector<std::string> generate_call(std::string target)
+    requires (architecture == architecture_t::x86_64) {
         static uint64_t label_count= 0;
         return {
                 "    movq return_label_n"+std::to_string(label_count)+", (stack_instruction)\n",
@@ -225,13 +386,14 @@ namespace molasses {
         };
     }
 
-    std::vector<std::string> generate_push_int32(int32_t target) {
+    std::vector<std::string> generate_push_int32(int32_t target) requires (architecture == architecture_t::x86_64) {
         return {
                 "    pushq $" +std::to_string(target)+ "\n"
         };
     }
 
-    std::vector<std::string> procedure_operation::generate(const parser_context& ctx, const lexed_output& lexer_data) const {
+    std::vector<std::string> procedure_operation::generate(const parser_context& ctx, const lexed_output& lexer_data) const
+    requires (architecture == architecture_t::x86_64) {
         std::vector<std::string> ops;
         ops.emplace_back(name()+":\n");