added some more things to the backend

This commit is contained in:
Goren Barak 2023-11-16 19:01:52 -05:00
parent 043d7fcb0d
commit 4ed9e7064a
6 changed files with 102 additions and 34 deletions

1
.gitignore vendored
View file

@ -1 +1,2 @@
/target /target
tags

1
README.md Normal file
View file

@ -0,0 +1 @@
# SkyLang

View file

@ -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"),
} }
}, },
_ => 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"),
}
let asm = format!("{}{}", asm_text, asm_data);
println!("{}", asm); println!("{}", asm);
asm asm
} }

View file

@ -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,
}));
} }

View file

@ -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,
}