Compare commits
2 Commits
2383a86a60
...
cb89bc0e9a
Author | SHA1 | Date |
---|---|---|
Goren Barak | cb89bc0e9a | |
Goren Barak | 0b4e7deda0 |
|
@ -0,0 +1,6 @@
|
|||
run:
|
||||
cargo run -r
|
||||
test:
|
||||
cargo test
|
||||
build:
|
||||
cargo build -r
|
|
@ -0,0 +1,66 @@
|
|||
#![allow(warnings)]
|
||||
|
||||
pub mod lex;
|
||||
pub mod codegen;
|
||||
use crate::codegen::fasm::*;
|
||||
use crate::lex::tok::*;
|
||||
use crate::parse::ast::*;
|
||||
use crate::parse::parse::*;
|
||||
use logos::Logos;
|
||||
|
||||
pub mod parse;
|
||||
|
||||
macro_rules! arrow {
|
||||
($spaces:expr) => {
|
||||
println!("{}↓", $spaces);
|
||||
}
|
||||
}
|
||||
|
||||
fn main() {
|
||||
// let fc = fasm_codegen!(
|
||||
// vec![
|
||||
// Expr::VarDefinition(VarDefinition {name: "goren", value: Value::Number(10)}),
|
||||
// Expr::MathExpr(Math {
|
||||
// left: &Value::Var(VarReference { name: "goren"}),
|
||||
// right: &Value::Number(17),
|
||||
// operator: MathOperator::OP_MULT
|
||||
// }
|
||||
// ),
|
||||
// Expr::FunDefinition(FunDefinition {
|
||||
// name: "adder", contents: vec![
|
||||
// Expr::MathExpr(
|
||||
// Math {
|
||||
// left: &Value::Param(ParamReference {param_number: 0}),
|
||||
// right: &Value::Param(ParamReference {param_number: 1}),
|
||||
// operator: MathOperator::OP_ADD
|
||||
// }
|
||||
// )
|
||||
// ]
|
||||
// }),
|
||||
|
||||
// Expr::FunCall(
|
||||
// FunCall {
|
||||
// name: "adder",
|
||||
// params: vec![Value::Var(VarReference {name: "goren"}), Value::Number(6)]
|
||||
// }
|
||||
// ),
|
||||
|
||||
// Expr::Breakpoint
|
||||
// ]
|
||||
// );
|
||||
|
||||
|
||||
// println!("{}", fc);
|
||||
let parsed = "30 * 60";
|
||||
|
||||
let mut lexer = Token::lexer(parsed);
|
||||
|
||||
println!("\"{}\"", parsed);
|
||||
arrow!(" ");
|
||||
println!("{:?}", lex_str(parsed));
|
||||
arrow!(" ");
|
||||
let parsed = parse_math(lexer);
|
||||
println!("{:?}", parsed);
|
||||
arrow!(" ");
|
||||
println!("{}", fasm_codegen!(&vec![parsed.unwrap()]));
|
||||
}
|
|
@ -1,10 +1,11 @@
|
|||
use crate::parse::ast::*;
|
||||
use std::rc::Rc;
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! fasm_codegen {
|
||||
// Macro to make calling fasm_codegen function easier.
|
||||
($exprs:expr) => {
|
||||
fasm_codegen(&$exprs, true)
|
||||
fasm_codegen($exprs, true)
|
||||
};
|
||||
|
||||
(fun: $exprs:expr) => {
|
||||
|
@ -27,6 +28,13 @@ pub fn fasm_codegen(exprs: &Vec<Expr>, not_a_function: bool) -> String {
|
|||
// Define asm_start, used for the entry point.
|
||||
let mut asm_start = String::new();
|
||||
|
||||
macro_rules! unwrap {
|
||||
($item:expr) => {
|
||||
asm_start.push_str(fasm_codegen!(fun: &vec![*$item.as_ref()]).as_str());-
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// If not_a_function, push necessary headers to the asm_start variable.
|
||||
if not_a_function {
|
||||
asm_start.push_str("format ELF64 executable 3\n");
|
||||
|
@ -42,8 +50,10 @@ pub fn fasm_codegen(exprs: &Vec<Expr>, not_a_function: bool) -> String {
|
|||
match expr {
|
||||
// If the expression is a math expression.
|
||||
Expr::MathExpr(e) => {
|
||||
asm_start.push_str(format!("\tmov r10, {}\n", e.left.unwrap()).as_str());
|
||||
asm_start.push_str(format!("\tmov r11, {}\n", e.right.unwrap()).as_str());
|
||||
unwrap!(e.left);
|
||||
asm_start.push_str(format!("\tmov r10, rax\n").as_str());
|
||||
unwrap!(e.right);
|
||||
asm_start.push_str(format!("\tmov r11, rax").as_str());
|
||||
match e.operator {
|
||||
// If the operator is addition.
|
||||
MathOperator::OP_ADD => {
|
||||
|
@ -91,43 +101,51 @@ pub fn fasm_codegen(exprs: &Vec<Expr>, not_a_function: bool) -> String {
|
|||
for (i, p) in e.params.iter().enumerate() {
|
||||
match i {
|
||||
0 => {
|
||||
// First parameter. Put in %rdi.← asm_start.push_str(format!("\tmov rdi, {}\n", p.unwrap()).as_str());
|
||||
// First parameter. Put in %rdi.
|
||||
unwrap!(p);
|
||||
asm_start.push_str(format!("\tmov rdi, rax\n").as_str());
|
||||
// rdi ← e.params[0];
|
||||
},
|
||||
|
||||
1 => {
|
||||
// Second parameter. Put in %rsi.
|
||||
asm_start.push_str(format!("\tmov rsi, {}\n", p.unwrap()).as_str());
|
||||
unwrap!(p);
|
||||
asm_start.push_str(format!("\tmov rsi, rax\n").as_str());
|
||||
// rsi ← e.params[1];
|
||||
},
|
||||
|
||||
2 => {
|
||||
2 => {
|
||||
// Third parameter. Put in %rdx.
|
||||
asm_start.push_str(format!("\tmov rdx, {}\n", p.unwrap()).as_str());
|
||||
unwrap!(p);
|
||||
asm_start.push_str(format!("\tmov rdx, rax\n").as_str());
|
||||
// rdx ← e.params[2];
|
||||
},
|
||||
|
||||
3 => {
|
||||
// Fourth parameter. Put in %rcx.
|
||||
asm_start.push_str(format!("\tmov rcx, {}\n", p.unwrap()).as_str());
|
||||
unwrap!(p);
|
||||
asm_start.push_str(format!("\tmov rcx, rax\n").as_str());
|
||||
// rcx ← e.params[3];
|
||||
},
|
||||
|
||||
4 => {
|
||||
// Fifth parameter. Put in %r8.
|
||||
asm_start.push_str(format!("\tmov r8, {}\n", p.unwrap()).as_str());
|
||||
unwrap!(p);
|
||||
asm_start.push_str(format!("\tmov r8, rax").as_str());
|
||||
// r8 ← e.params[4];
|
||||
},
|
||||
|
||||
5 => {
|
||||
// Sixth parameter. Put in %r9.
|
||||
asm_start.push_str(format!("\tmov r9, {}\n", p.unwrap()).as_str());
|
||||
unwrap!(p);
|
||||
asm_start.push_str(format!("\tmov r9, rax\n").as_str());
|
||||
// r9 ← e.params[5];
|
||||
},
|
||||
|
||||
_ => {
|
||||
// Parameters after the sixth parameter are pushed to the stack.
|
||||
asm_start.push_str(format!("\tpush {}\n", p.unwrap()).as_str());
|
||||
unwrap!(p);
|
||||
asm_start.push_str(format!("\tpush rax\n").as_str());
|
||||
// STACK_TOP ← e.params[(6+)];
|
||||
}
|
||||
}
|
||||
|
@ -138,9 +156,10 @@ pub fn fasm_codegen(exprs: &Vec<Expr>, not_a_function: bool) -> String {
|
|||
},
|
||||
|
||||
// Define a global variable.
|
||||
Expr::VarDefinition(e) => {
|
||||
// Define a 64-bit variable.
|
||||
asm_data.push_str(format!("\t{} dq {}", e.name, e.value.unwrap()).as_str());
|
||||
Expr::GlobalDefinition(e) => {
|
||||
// Define a 64-bit global variable.
|
||||
|
||||
asm_data.push_str(format!("\t{} dq {}", e.name, e.value).as_str());
|
||||
},
|
||||
|
||||
// Breakpoint.
|
||||
|
@ -173,7 +192,8 @@ pub fn fasm_codegen(exprs: &Vec<Expr>, not_a_function: bool) -> String {
|
|||
// Increment the temporary variable/function counter.
|
||||
tmp_counter += 1;
|
||||
// Compare the left and right value.
|
||||
asm_start.push_str(format!("\tcmp {}, {}\n", e.left.unwrap(), e.right.unwrap()).as_str());
|
||||
|
||||
asm_start.push_str(format!("\tcmp rax, {}\n", e.left, e.right.unwrap()).as_str());
|
||||
// Check what the condition is.
|
||||
match e.cond {
|
||||
COND_OP::EQ => {
|
||||
|
|
|
@ -85,7 +85,7 @@ pub enum Token {
|
|||
String, // A string literal.
|
||||
#[regex("[0-9]+", |lex| lex.slice().parse().ok())]
|
||||
Number(u64), // An integer.
|
||||
#[regex(r#"[^[0-9]^"^-^[ \t\n\f]^\.^=^(^)^{^}.^,^;]+[^"^-^=^\..^[ \t\n\f]^(^)^{^}^,^;]*"#)]
|
||||
#[regex(r#"[^[0-9]^"^-^[ \t\n\f]^\.^=^(^)^{^}.^,^;^[+-/*%]]+[^"^-^=^\..^[ \t\n\f]^(^)^{^}^,^;^[+-/*%]]*"#)]
|
||||
Identifier, // An identifier.
|
||||
#[token("true")]
|
||||
True, // true
|
||||
|
@ -96,7 +96,6 @@ pub enum Token {
|
|||
}
|
||||
|
||||
pub fn lex_str(this: &str) -> Vec<(Token, &str)> {
|
||||
println!("\"{}\"", this);
|
||||
let mut buf = Vec::new();
|
||||
let mut lexer = Token::lexer(this);
|
||||
while let Some(Ok(token)) = lexer.next() {
|
||||
|
|
18
src/main.rs
18
src/main.rs
|
@ -10,6 +10,12 @@ use logos::Logos;
|
|||
|
||||
pub mod parse;
|
||||
|
||||
macro_rules! arrow {
|
||||
($spaces:expr) => {
|
||||
println!("{}↓", $spaces);
|
||||
}
|
||||
}
|
||||
|
||||
fn main() {
|
||||
// let fc = fasm_codegen!(
|
||||
// vec![
|
||||
|
@ -45,9 +51,17 @@ fn main() {
|
|||
|
||||
|
||||
// println!("{}", fc);
|
||||
let parsed = "hello(hi)";
|
||||
let parsed = "30 * 60";
|
||||
|
||||
let mut lexer = Token::lexer(parsed);
|
||||
|
||||
println!("\"{}\"", parsed);
|
||||
arrow!(" ");
|
||||
println!("{:?}", lex_str(parsed));
|
||||
println!("{:?}", parse_fun_call(lexer));
|
||||
arrow!(" ");
|
||||
let parsed = parse_math(lexer);
|
||||
println!("{:?}", parsed);
|
||||
arrow!(" ");
|
||||
println!("{}", fasm_codegen!(&vec![parsed.unwrap()]));
|
||||
}
|
||||
|
||||
|
|
|
@ -4,19 +4,23 @@ use std::rc::Rc;
|
|||
pub enum Expr<'a> {
|
||||
MathExpr(Math<'a>),
|
||||
FunCall(FunCall<'a>),
|
||||
FunDefinition(FunDefinition<'a>),
|
||||
VarDefinition(VarDefinition<'a>),
|
||||
FunDefinition(Rc<FunDefinition<'a>>),
|
||||
GlobalDefinition(Rc<VarDefinition<'a>>),
|
||||
Return(Vec<Expr<'a>>),
|
||||
If(IfCondition<'a>),
|
||||
If(Rc<IfCondition<'a>>),
|
||||
Var(VarReference<'a>),
|
||||
Param(ParamReference),
|
||||
Number(u64),
|
||||
|
||||
Breakpoint
|
||||
}
|
||||
|
||||
// MATH EXPRESSION
|
||||
|
||||
#[derive(Debug, Copy, Clone)]
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct Math<'a> {
|
||||
pub left: &'a Value<'a>,
|
||||
pub right: &'a Value<'a>,
|
||||
pub left: Rc<Expr<'a>>,
|
||||
pub right: Rc<Expr<'a>>,
|
||||
pub operator: MathOperator
|
||||
}
|
||||
|
||||
|
@ -34,7 +38,7 @@ pub enum MathOperator {
|
|||
#[derive(Debug)]
|
||||
pub struct FunCall<'a> {
|
||||
pub name: &'a str,
|
||||
pub params: Vec<Value<'a>>,
|
||||
pub params: Vec<Rc<Expr<'a>>>,
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
|
@ -51,7 +55,7 @@ pub struct FunParamDef<'a> {
|
|||
|
||||
#[derive(Debug)]
|
||||
pub struct FunParamCall<'a> {
|
||||
pub value: Value<'a>,
|
||||
pub value: Expr<'a>,
|
||||
}
|
||||
|
||||
// VARIABLES
|
||||
|
@ -59,7 +63,7 @@ pub struct FunParamCall<'a> {
|
|||
#[derive(Debug)]
|
||||
pub struct VarDefinition<'a> {
|
||||
pub name: &'a str,
|
||||
pub value: Value<'a>,
|
||||
pub value: u64,
|
||||
}
|
||||
|
||||
|
||||
|
@ -73,17 +77,12 @@ pub struct ParamReference {
|
|||
pub param_number: u64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Copy, Clone)]
|
||||
pub enum Value<'a> {
|
||||
Var(VarReference<'a>),
|
||||
Param(ParamReference),
|
||||
Number(u64),
|
||||
}
|
||||
// CONDITIONS
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct IfCondition<'a> {
|
||||
pub left: Value<'a>,
|
||||
pub right: Value<'a>,
|
||||
pub left: Expr<'a>,
|
||||
pub right: Expr<'a>,
|
||||
pub cond: COND_OP,
|
||||
pub action: Vec<Expr<'a>>
|
||||
}
|
||||
|
@ -94,28 +93,29 @@ pub enum COND_OP {
|
|||
NE,
|
||||
}
|
||||
|
||||
impl<'a> Value<'a> {
|
||||
pub fn unwrap(&self) -> String {
|
||||
match self {
|
||||
Value::Param(e) => {
|
||||
match e.param_number {
|
||||
0 => { return "rdi".to_string(); },
|
||||
1 => { return "rsi".to_string(); },
|
||||
2 => { return "rdx".to_string(); },
|
||||
3 => { return "rcx".to_string(); },
|
||||
4 => { return "r8".to_string(); },
|
||||
5 => { return "r9".to_string(); },
|
||||
_ => { unimplemented!() }
|
||||
}
|
||||
},
|
||||
// VALUE
|
||||
// impl<'a> Value<'a> {
|
||||
// pub fn unwrap(&self) -> String {
|
||||
// match self {
|
||||
// Value::Param(e) => {
|
||||
// match e.param_number {
|
||||
// 0 => { return "rdi".to_string(); },
|
||||
// 1 => { return "rsi".to_string(); },
|
||||
// 2 => { return "rdx".to_string(); },
|
||||
// 3 => { return "rcx".to_string(); },
|
||||
// 4 => { return "r8".to_string(); },
|
||||
// 5 => { return "r9".to_string(); },
|
||||
// _ => { unimplemented!() }
|
||||
// }
|
||||
// },
|
||||
|
||||
Value::Number(e) => {
|
||||
return e.to_string();
|
||||
},
|
||||
// Value::Number(e) => {
|
||||
// return e.to_string();
|
||||
// },
|
||||
|
||||
Value::Var(e) => {
|
||||
return format!("[{}]", e.name.to_string());
|
||||
},
|
||||
}
|
||||
}
|
||||
}
|
||||
// Value::Var(e) => {
|
||||
// return format!("[{}]", e.name.to_string());
|
||||
// },
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
|
|
|
@ -1,13 +1,39 @@
|
|||
use super::ast::*;
|
||||
use crate::lex::tok::*;
|
||||
use logos::Lexer;
|
||||
use std::rc::Rc;
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! unwrap {
|
||||
($var:expr) => {
|
||||
$var.next().unwrap().unwrap()
|
||||
}
|
||||
}
|
||||
|
||||
#[macro_export]
|
||||
macro_rules! parse_value {
|
||||
($parse:expr) => {
|
||||
parse_value(&($parse.next(), $parse.slice()))
|
||||
}
|
||||
}
|
||||
|
||||
pub fn parse_math(mut tokens: Lexer<Token>) -> Option<Expr> {
|
||||
// Is it a Value? → Is it an operator? → Is it a value?
|
||||
|
||||
if let Some(left) = parse_value!(tokens) {
|
||||
if let Some(operator) = match_operator(&mut tokens) {
|
||||
if let Some(right) = parse_value!(tokens) {
|
||||
let left = Rc::new(left);
|
||||
let right = Rc::new(right);
|
||||
|
||||
return Some(Expr::MathExpr(Math {left: left, right: right, operator}))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
None
|
||||
}
|
||||
|
||||
pub fn parse_var_declaration(mut tokens: Lexer<Token>) -> Option<Expr> {
|
||||
let mut tok = None;
|
||||
if unwrap!(tokens) == Let {
|
||||
|
@ -31,13 +57,13 @@ pub fn parse_var_declaration(mut tokens: Lexer<Token>) -> Option<Expr> {
|
|||
tok
|
||||
}
|
||||
|
||||
pub fn parse_value<'a>(token: (Option<Result<Token, ()>>, &'a str)) -> Option<Value<'a>> {
|
||||
if let Some(Ok(tt)) = token.0 {
|
||||
pub fn parse_value<'a>(token: &(Option<Result<Token, ()>>, &'a str)) -> Option<Expr<'a>> {
|
||||
if let Some(Ok(tt)) = &token.0 {
|
||||
let mut value = None;
|
||||
|
||||
if let Number(n) = tt {
|
||||
value = Some(Value::Number(n));
|
||||
} else if tt == Identifier {
|
||||
value = Some(Value::Number(*n));
|
||||
} else if *tt == Identifier {
|
||||
value = Some(Value::Var(VarReference { name: token.1 }));
|
||||
}
|
||||
|
||||
|
@ -58,7 +84,7 @@ pub fn parse_fun_call(mut tokens: Lexer<Token>) -> Option<Expr> {
|
|||
let name = tokens.slice();
|
||||
if unwrap!(tokens) == LeftParen {
|
||||
let mut params = Vec::new();
|
||||
while let Some(value) = parse_value((tokens.next(), tokens.slice())) {
|
||||
while let Some(value) = parse_value!(tokens) {
|
||||
params.push(value);
|
||||
}
|
||||
tok = Some(Expr::FunCall(FunCall {name, params: params.clone()}));
|
||||
|
@ -67,3 +93,14 @@ pub fn parse_fun_call(mut tokens: Lexer<Token>) -> Option<Expr> {
|
|||
|
||||
tok
|
||||
}
|
||||
|
||||
pub fn match_operator(tokens: &mut Lexer<Token>) -> Option<MathOperator> {
|
||||
match unwrap!(tokens) {
|
||||
Plus => Some(MathOperator::OP_ADD),
|
||||
Minus => Some(MathOperator::OP_SUB),
|
||||
Slash => Some(MathOperator::OP_DIV),
|
||||
Star => Some(MathOperator::OP_MULT),
|
||||
Percent => Some(MathOperator::OP_MOD),
|
||||
_ => None
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue