Add add

5 files changed, 125 insertions, 0 deletions

diff --git a/asm/assembler.h b/asm/assembler.h
index ddea4af..3d3e9a9 100644
--- a/asm/assembler.h
+++ b/asm/assembler.h
@@ -2,6 +2,8 @@
 
 #include "chunk.h"
 
+#include <boost/endian/conversion.hpp>
+
 #include <any>
 #include <functional>
 #include <iostream>
@@ -47,8 +49,44 @@ class Immediate8
 public:
  Immediate8(uint8_t value): m_value(value) {}
  uint8_t value() {return m_value;}
+ std::vector<uint8_t> getCode() {return {m_value};};
 
 private:
  uint8_t m_value;
 };
 
+class Immediate32
+{
+public:
+ Immediate32(uint32_t value): m_value(value) {}
+ uint32_t value() { return m_value; }
+ std::vector<uint8_t> getCode() {
+  std::vector<uint8_t> result(size_t(4));
+  *(reinterpret_cast<uint32_t*>(result.data())) = boost::endian::native_to_little(m_value);
+  return result;
+ };
+
+private:
+ uint32_t m_value;
+};
+
+class Register32
+{
+public:
+ Register32(const std::string& name): m_name(name) {}
+ std::string name() { return m_name; }
+
+private:
+ std::string m_name;
+};
+
+class Register64
+{
+public:
+ Register64(const std::string& name): m_name(name) {}
+ std::string name() { return m_name; }
+
+private:
+ std::string m_name;
+};
+
diff --git a/asm/intel64/add.cpp b/asm/intel64/add.cpp
new file mode 100644
index 0000000..dc5c704
--- /dev/null
+++ b/asm/intel64/add.cpp
@@ -0,0 +1,31 @@
+#include "add.h"
+
+#include "codes.h"
+
+#include <asm/assembler.h>
+#include <asm/operators.h>
+
+using namespace std::string_literals;
+
+Op_add::Op_add(AsmArgs& args)
+{
+ if (args[0].type() == typeid(Register32) && std::any_cast<Register32>(args[0]).name() == "eax" && args[1].type() == typeid(Immediate32)) { // add eax, imm32
+  machine_code = std::vector<uint8_t>{ 0x05 } + std::any_cast<Immediate32>(args[1]).getCode();
+ } else if (args[0].type() == typeid(Register64) && std::any_cast<Register64>(args[0]).name() == "rax"  && args[1].type() == typeid(Immediate32)) { // add rax, imm32
+  machine_code = REX("W") + std::vector<uint8_t>{ 0x05 } + std::any_cast<Immediate32>(args[1]).getCode();
+ } else {
+  throw std::runtime_error("Unimplemented: add "s + args[0].type().name() + " "s + args[1].type().name());
+ }
+}
+
+namespace {
+
+bool registered0 { registerOp(mangleName<Register32, Immediate32>("add"), [](AsmArgs& args) -> std::shared_ptr<Op>{
+                             return std::make_shared<Op_add>(args);
+                             }) };
+// TODO
+bool registered1 { registerOp(mangleName<Register64, Immediate32>("add"), [](AsmArgs& args) -> std::shared_ptr<Op>{
+                             return std::make_shared<Op_add>(args);
+                             }) };
+
+}
diff --git a/asm/intel64/add.h b/asm/intel64/add.h
new file mode 100644
index 0000000..4c6b589
--- /dev/null
+++ b/asm/intel64/add.h
@@ -0,0 +1,31 @@
+// Integer Addition
+
+#pragma once
+
+#include <asm/assembler.h>
+
+class Op_add: public Op
+{
+public:
+ Op_add(AsmArgs& args);
+
+public:
+ std::vector<uint8_t> getCode() override
+ {
+  return machine_code;
+ }
+
+ size_t size() override
+ {
+  return machine_code.size();
+ }
+
+ bool optimize() override ///< returns true if changed
+ {
+  return false;
+ }
+
+protected:
+ std::vector<uint8_t> machine_code;
+};
+
diff --git a/asm/intel64/codes.cpp b/asm/intel64/codes.cpp
new file mode 100644
index 0000000..a1d9e87
--- /dev/null
+++ b/asm/intel64/codes.cpp
@@ -0,0 +1,17 @@
+#include "codes.h"
+
+// REX prefix: 0b0100WRXB
+std::vector<uint8_t> REX(std::string s) {
+ uint8_t result{0b01000000};
+ if (s == "W")
+  result |= 0b00001000;
+ if (s == "R")
+  result |= 0b00000100;
+ if (s == "X")
+  result |= 0b00000010;
+ if (s == "B")
+  result |= 0b00000001;
+
+ return { result }; 
+}
+
diff --git a/asm/intel64/codes.h b/asm/intel64/codes.h
new file mode 100644
index 0000000..32eff1c
--- /dev/null
+++ b/asm/intel64/codes.h
@@ -0,0 +1,8 @@
+#pragma once
+
+#include <cstdint>
+#include <string>
+#include <vector>
+
+// REX prefix: 0b0100WRXB
+std::vector<uint8_t> REX(std::string s);