summaryrefslogtreecommitdiff
path: root/src/node
diff options
context:
space:
mode:
Diffstat (limited to 'src/node')
-rw-r--r--src/node/encode.rs56
-rw-r--r--src/node/parse.rs90
2 files changed, 146 insertions, 0 deletions
diff --git a/src/node/encode.rs b/src/node/encode.rs
new file mode 100644
index 0000000..7ae038b
--- /dev/null
+++ b/src/node/encode.rs
@@ -0,0 +1,56 @@
+/*
+ Copyright 2023-2024 Gabriel Bjørnager Jensen.
+
+ This file is part of eAS.
+
+ eAS is free software: you can redistribute it
+ and/or modify it under the terms of the GNU
+ General Public License as published by the Free
+ Software Foundation, either version 3 of the
+ License, or (at your option) any later version.
+
+ eAS is distributed in the hope that it will
+ be useful, but WITHOUT ANY WARRANTY; without
+ even the implied warranty of MERCHANTABILITY or
+ FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ General Public License for more details.
+
+ You should have received a copy of the GNU
+ General Public License along with eAS. If not,
+ see <https://www.gnu.org/licenses/>.
+*/
+
+use crate::encode_state::EncodeState;
+use crate::node::Node;
+
+impl Node {
+ #[must_use]
+ pub fn encode(&self, _state: &EncodeState) -> Vec<u8> {
+ use Node::*;
+
+ let mut binary = Vec::new();
+
+ match *self {
+ // These don't leak into the binary:
+ | Code16
+ | Code32
+ | End
+ => {},
+
+ Fill(repeat, value, size) => {
+ let mut bytes = Vec::from(value.to_le_bytes());
+ bytes.resize(size as usize, 0x00);
+
+ for _ in 0x0..repeat { binary.extend_from_slice(&bytes) }
+ }
+
+ Space(length) => for _ in 0x0..length { binary.push(0x00) },
+
+ // In case we missed something.
+ #[allow(unreachable_patterns)]
+ _ => todo!(),
+ };
+
+ return binary;
+ }
+}
diff --git a/src/node/parse.rs b/src/node/parse.rs
new file mode 100644
index 0000000..3137e97
--- /dev/null
+++ b/src/node/parse.rs
@@ -0,0 +1,90 @@
+/*
+ Copyright 2023-2024 Gabriel Bjørnager Jensen.
+
+ This file is part of eAS.
+
+ eAS is free software: you can redistribute it
+ and/or modify it under the terms of the GNU
+ General Public License as published by the Free
+ Software Foundation, either version 3 of the
+ License, or (at your option) any later version.
+
+ eAS is distributed in the hope that it will
+ be useful, but WITHOUT ANY WARRANTY; without
+ even the implied warranty of MERCHANTABILITY or
+ FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ General Public License for more details.
+
+ You should have received a copy of the GNU
+ General Public License along with eAS. If not,
+ see <https://www.gnu.org/licenses/>.
+*/
+
+use crate::error::Error;
+use crate::node::Node;
+use crate::source_location::SourceLocation;
+use crate::token::Token;
+
+macro_rules! complete_node {
+ ($nodes: expr, $new_node: expr, $flag: ident) => {{
+ $nodes.push($new_node);
+ $flag = true;
+ continue;
+ }};
+}
+
+impl Node {
+ #[must_use]
+ pub fn parse(tokens: &[(SourceLocation, Token)]) -> Result<Vec<Self>, Error> {
+ assert!(tokens.len() > 0x0);
+
+ let mut file: Option<&str> = None;
+
+ let mut nodes: Vec<Self> = Vec::new();
+
+ let mut got_end = false;
+ let mut node_complete = true;
+
+ for (location, token) in tokens {
+ use Token::*;
+
+ file = Some(location.file());
+
+ match token {
+ Word(word) => match word.as_str() {
+ | "CODE16"
+ | "code16"
+ | "THUMB"
+ | "thumb"
+ => complete_node!(nodes, Node::Code16, node_complete),
+
+ | "ARM"
+ | "arm"
+ | "CODE32"
+ | "code32"
+ => complete_node!(nodes, Node::Code32, node_complete),
+
+ | "END"
+ | "end"
+ => {
+ got_end = true;
+ complete_node!(nodes, Node::End, node_complete);
+ },
+
+ _ => return Err(Error::UnknownMnemonic(word.clone(), location.clone())),
+ },
+
+ Return => {
+ if !node_complete { return Err(Error::IncompleteNode(location.clone())) };
+ continue;
+ },
+
+ _ => {},
+ };
+ }
+
+ if !got_end { return Err(Error::EndOfFile(file.unwrap().to_string())) };
+
+ return Ok(nodes);
+ }
+}