added some more things to the backend
This commit is contained in:
parent
043d7fcb0d
commit
4ed9e7064a
1
.gitignore
vendored
1
.gitignore
vendored
|
@ -1 +1,2 @@
|
||||||
/target
|
/target
|
||||||
|
tags
|
||||||
|
|
|
@ -1,53 +1,102 @@
|
||||||
use crate::parse::ast::*;
|
use crate::parse::ast::*;
|
||||||
|
|
||||||
pub fn fasm_codegen(expr: Expr) -> String {
|
pub fn fasm_codegen(expr: Expr) -> String {
|
||||||
let mut asm = String::new();
|
let mut asm_text = String::new();
|
||||||
|
let mut asm_data = String::new();
|
||||||
|
|
||||||
asm.push_str("format ELF64 executable 3\n");
|
asm_text.push_str("format ELF64 executable 3\n");
|
||||||
asm.push_str("segment readable executable\n");
|
asm_text.push_str("segment readable executable\n");
|
||||||
asm.push_str("entry _start\n");
|
asm_text.push_str("entry _start\n");
|
||||||
asm.push_str("_start:\n");
|
asm_text.push_str("_start:\n");
|
||||||
|
asm_data.push_str("segment readable writeable\n");
|
||||||
match expr {
|
match expr {
|
||||||
Expr::MathExpr(e) => {
|
Expr::MathExpr(e) => {
|
||||||
asm.push_str(format!("\tmov r10, {:?}\n", e.left).as_str());
|
asm_text.push_str(format!("\tmov r10, {:?}\n", e.left).as_str());
|
||||||
asm.push_str(format!("\tmov r11, {:?}\n", e.right).as_str());
|
asm_text.push_str(format!("\tmov r11, {:?}\n", e.right).as_str());
|
||||||
match e.operator {
|
match e.operator {
|
||||||
// If the operator is addition.
|
// If the operator is addition.
|
||||||
MathOperator::OP_ADD => {
|
MathOperator::OP_ADD => {
|
||||||
asm.push_str("\tadd r10, r11\n");
|
asm_text.push_str("\tadd r10, r11\n");
|
||||||
asm.push_str("\tmov rax, r10\n");
|
asm_text.push_str("\tmov rax, r10\n");
|
||||||
},
|
},
|
||||||
// If the operator is multiplication.
|
// If the operator is multiplication.
|
||||||
MathOperator::OP_MULT => {
|
MathOperator::OP_MULT => {
|
||||||
asm.push_str("\timul r10, r11\n");
|
asm_text.push_str("\timul r10, r11\n");
|
||||||
asm.push_str("\tmov rax, r10\n");
|
asm_text.push_str("\tmov rax, r10\n");
|
||||||
},
|
},
|
||||||
// If the operator is division.
|
// If the operator is division.
|
||||||
MathOperator::OP_DIV => {
|
MathOperator::OP_DIV => {
|
||||||
asm.push_str("\tmov rax, r10\n");
|
asm_text.push_str("\tmov rax, r10\n");
|
||||||
asm.push_str("\tmov rdx, r11\n");
|
asm_text.push_str("\tmov rdx, r11\n");
|
||||||
asm.push_str("\tidiv r10, r11\n");
|
asm_text.push_str("\tidiv r10, r11\n");
|
||||||
asm.push_str("\tmov rax, r10\n");
|
asm_text.push_str("\tmov rax, r10\n");
|
||||||
},
|
},
|
||||||
// If the operators is subtraction.
|
// If the operators is subtraction.
|
||||||
MathOperator::OP_SUB => {
|
MathOperator::OP_SUB => {
|
||||||
asm.push_str("\tsub r10, r11\n");
|
asm_text.push_str("\tsub r10, r11\n");
|
||||||
asm.push_str("\tmov rax, r10\n");
|
asm_text.push_str("\tmov rax, r10\n");
|
||||||
},
|
},
|
||||||
// If the operator is modulo.
|
// If the operator is modulo.
|
||||||
MathOperator::OP_MOD => {
|
MathOperator::OP_MOD => {
|
||||||
asm.push_str("\tmov rax, r10\n");
|
asm_text.push_str("\tmov rax, r10\n");
|
||||||
asm.push_str("\tmov rdx, r11\n");
|
asm_text.push_str("\tmov rdx, r11\n");
|
||||||
asm.push_str("\tidiv r10, r11\n");
|
asm_text.push_str("\tidiv r10, r11\n");
|
||||||
asm.push_str("\tmov rax, rdx\n");
|
asm_text.push_str("\tmov rax, rdx\n");
|
||||||
|
|
||||||
},
|
}
|
||||||
_ => unimplemented!("sorry unimplemented"),
|
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|
||||||
|
Expr::FunCall(e) => {
|
||||||
|
for (i, p) in e.params.iter().enumerate() {
|
||||||
|
match i {
|
||||||
|
0 => {
|
||||||
|
// First parameter. Put in %rdi.
|
||||||
|
asm_text.push_str(format!("\tmov rdi, {:?}\n", p.value).as_str());
|
||||||
|
},
|
||||||
|
|
||||||
|
1 => {
|
||||||
|
// Second parameter. Put in %rsi.
|
||||||
|
asm_text.push_str(format!("\tmov rsi, {:?}\n", p.value).as_str());
|
||||||
|
},
|
||||||
|
|
||||||
|
2 => {
|
||||||
|
// Third parameter. Put in %rdx.
|
||||||
|
asm_text.push_str(format!("\tmov rdx, {:?}\n", p.value).as_str());
|
||||||
|
},
|
||||||
|
|
||||||
|
3 => {
|
||||||
|
// Fourth parameter. Put in %rcx.
|
||||||
|
asm_text.push_str(format!("\tmov rcx, {:?}\n", p.value).as_str());
|
||||||
|
},
|
||||||
|
|
||||||
|
4 => {
|
||||||
|
// Fifth parameter. Put in %r8.
|
||||||
|
asm_text.push_str(format!("\tmov r8, {:?}\n", p.value).as_str());
|
||||||
|
},
|
||||||
|
|
||||||
|
5 => {
|
||||||
|
// Sixth parameter. Put in %r9.
|
||||||
|
asm_text.push_str(format!("\tmov r9, {:?}\n", p.value).as_str());
|
||||||
|
},
|
||||||
|
|
||||||
|
_ => {
|
||||||
|
// Parameters after the sixth parameter are pushed to the stack.
|
||||||
|
asm_text.push_str(format!("\tpush {:?}\n", p.value).as_str());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
asm_text.push_str(format!("call {:?}", e.name).as_str());
|
||||||
|
},
|
||||||
|
|
||||||
|
Expr::VarDefenition(e) => {
|
||||||
|
asm_data.push_str(format!("{} dq {:?}", e.name, e.value).as_str());
|
||||||
|
},
|
||||||
|
|
||||||
_ => unimplemented!("sorry unimplemented"),
|
_ => unimplemented!("sorry unimplemented"),
|
||||||
}
|
}
|
||||||
|
let asm = format!("{}{}", asm_text, asm_data);
|
||||||
println!("{}", asm);
|
println!("{}", asm);
|
||||||
asm
|
asm
|
||||||
}
|
}
|
||||||
|
|
|
@ -5,5 +5,8 @@ use crate::parse::ast::*;
|
||||||
pub mod parse;
|
pub mod parse;
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
fasm_codegen(Expr::MathExpr(Math {left: 1, right: 2, operator: MathOperator::OP_DIV}));
|
fasm_codegen(Expr::VarDefenition(VarDefenition {
|
||||||
|
name: "hi",
|
||||||
|
value: 100,
|
||||||
|
}));
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,7 +1,9 @@
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub enum Expr<'a> {
|
pub enum Expr<'a> {
|
||||||
MathExpr(Math),
|
MathExpr(Math),
|
||||||
FunCallExpr(FunCall<'a>),
|
FunCall(FunCall<'a>),
|
||||||
|
FunDefenition(FunDefenition<'a>),
|
||||||
|
VarDefenition(VarDefenition<'a>),
|
||||||
}
|
}
|
||||||
|
|
||||||
// MATH EXPRESSION
|
// MATH EXPRESSION
|
||||||
|
@ -26,19 +28,31 @@ pub enum MathOperator {
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct FunCall<'a> {
|
pub struct FunCall<'a> {
|
||||||
name: &'a str,
|
pub name: &'a str,
|
||||||
params: Vec<FunParam<'a>>,
|
pub params: Vec<FunParamCall>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct FunDefention<'a> {
|
pub struct FunDefenition<'a> {
|
||||||
name: &'a str,
|
name: &'a str,
|
||||||
params: Vec<FunParam<'a>>,
|
params: Vec<FunParamDef<'a>>,
|
||||||
contents: Vec<Expr<'a>>,
|
contents: Vec<Expr<'a>>,
|
||||||
return_value: Expr<'a>,
|
return_value: &'a Expr<'a>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct FunParam<'a> {
|
pub struct FunParamDef<'a> {
|
||||||
name: &'a str,
|
name: &'a str,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct FunParamCall {
|
||||||
|
// Everything is a u64 for now.
|
||||||
|
pub value: u64,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub struct VarDefenition<'a> {
|
||||||
|
pub name: &'a str,
|
||||||
|
pub value: u64,
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue