commit 45b9dc1790eaf068bbaec7c62b0dbcfde115ca77
Author: bunnie <bunnie@kosagi.com>
Date:   Thu Aug 6 03:21:51 2020 +0800

    initial commit

diff --git a/Cargo.lock b/Cargo.lock
new file mode 100644
index 0000000..1e9d6f2
--- /dev/null
+++ b/Cargo.lock
@@ -0,0 +1,5 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+[[package]]
+name = "engine25519-as"
+version = "0.1.0"
diff --git a/Cargo.toml b/Cargo.toml
new file mode 100644
index 0000000..83d39b1
--- /dev/null
+++ b/Cargo.toml
@@ -0,0 +1,11 @@
+[package]
+name = "engine25519-as"
+version = "0.1.0"
+authors = ["bunnie <bunnie@kosagi.com>"]
+edition = "2018"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+
+[workspace]
\ No newline at end of file
diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..5b196bd
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,317 @@
+#![feature(trace_macros)]
+#![feature(log_syntax)]
+trace_macros!{true}
+
+/*
+   opcodes
+
+   PSA  Wd, Ra
+   PSB  Wd, Rb
+   MSK  Wd, Ra, Rb
+   XOR  Wd, Ra, Rb
+   NOT  Wd, Ra
+   ADD  Wd, Ra, Rb
+   SUB  Wd, Ra, Rb
+   MUL  Wd, Ra, Rb
+   TRD  Wd, Ra, Rb
+   BRZ  offset, Ra
+
+   all Rx can be expressed as #Rx which summons the constant at #Rx
+
+   examples of syntax
+   ADD  r0, r1, r2   // adds r1 and r2, stores into r0
+   ADD  r31, #4, #8  // takes constants in table positions 4 and 8, adds them, stores into r31
+   PSA  r1, #0       // takes constant in table position 0, stores in r1
+   BRZ  -0x3, r4     // if r4 is 0, mpc = mpc - 0x3
+   BRZ  0x2, r1      // if r1 is 0, mpc = mpc + 0x2
+ loop:
+   BRZ  loop, r2     // if r2 is 0, go to label "loop"
+*/
+
+/// A compile-time map from identifiers to arbitrary (heterogeneous) expressions
+#[macro_export]
+#[doc(hidden)]
+macro_rules! ident_map {
+    ( $name:ident = { $($key:ident => $e:expr),* $(,)* } ) => {
+        macro_rules! $name {
+            $(
+                ( $key ) => { $e };
+            )*
+            // Empty invocation expands to nothing. Needed when the map is empty.
+            () => {};
+        }
+    };
+}
+
+
+/// Returns the number of comma-separated expressions passed to it
+#[macro_export]
+#[doc(hidden)]
+macro_rules! codelen {
+    () => { 0 };
+    ( $one:expr ) => { 1 };
+    ( $first:expr, $($rest:expr),+ ) => { 1 + codelen!($($rest),+) };
+}
+
+
+
+
+/// Replace elements of "arrays" of expressions with sorted replacements by
+/// seeking along a "positional array" containing n expressions, then replacing
+/// elements in the source array.
+///
+/// Expands to the first array with replacements applied. The length doesn't
+/// change.
+#[macro_export]
+#[doc(hidden)]
+macro_rules! lockstep_replace {
+    ( [ $($result:expr),* ], [], ) => {
+        // `src` is empty, no relocations. We're done!
+        [ $($result),* ]
+    };
+    ( [ $($result:expr),* ], [ $($src:expr,)+ ], ) => {
+        // Empty list of replacements, but still `src` to go
+        [ $($result,)* $($src),+ ]
+    };
+    ( [ $($result:expr),* ], [ $($src:expr,)* ], [], [], $( [ $($pos:expr,)* ], [ $($rep:expr,)* ], )* ) => {
+        // All replacements applied. Pop the current replacement and continue.
+        lockstep_replace!(
+            [ $($result),* ],
+            [ $($src,)* ],
+            $(
+                [ $($pos,)* ],
+                [ $($rep,)* ],
+            )*
+        )
+    };
+    ( [ $($result:expr),* ], [ $src1_replaced:expr, $($src:expr,)* ], [], [ $rep1:expr, $($rep_rest:expr,)* ], $( [ $pos1:expr, $($pos:expr,)* ], [ $($rep:expr,)* ], )* ) => {
+        // Position of a replacement reached (or: inside a replacement)
+        // Coupled with a seek step
+        lockstep_replace!(
+            [ $($result,)* $rep1 ],
+            [ $($src,)* ],
+            [],
+            [ $($rep_rest,)* ],
+            $(
+                [ $($pos,)* ],
+                [ $($rep,)* ],
+            )*
+        )
+    };
+    ( [ $($result:expr),* ], [ $src1:expr, $($src:expr,)* ], $( [ $pos1:expr, $($pos:expr,)* ], [ $($rep:expr,)* ], )+ ) => {
+        // Seek to next replacement position (simultaneously for all
+        // replacements)
+        lockstep_replace!(
+            [ $($result,)* $src1 ],
+            [ $($src,)* ],
+            $(
+                [ $($pos,)* ],
+                [ $($rep,)* ],
+            )+
+        )
+    };
+}
+
+/// Performs relocation of machine code based on given labels and relocations.
+/// Looks up label names in an `ident_map`. Expands to the relocated machine
+/// code.
+///
+/// Relocation formats:
+/// { $label as ABS16 @ [$lockstepmcpos] }
+
+// $pos is
+#[macro_export]
+#[doc(hidden)]
+macro_rules! reloc {
+    ( { $($attr:tt)* }  [ $( [ $($pos:expr),* ], [ $($rep:expr),* ] ),* ], $lblmap:ident, [ $($mcode:expr),* ], [/* empty relocation list */] ) => {
+        lockstep_replace!([], [ $($mcode,)* ], $( [ $($pos,)* ], [ $($rep,)* ], )*)
+    };
+    ( { start: $start:expr }  [ $( [ $($pos:expr),* ], [ $($rep:expr),* ] ),* ], $lblmap:ident, [ $($mcode:expr),* ], [ { $lbl:ident as ABS16 @ [$($lockstepmcpos:expr),*] } $(,$reloc:tt)* ] ) => {
+        // Replace 2 Bytes with the absolute address
+        // Relocation position is given as "lock-step MC pos", an expression
+        // list that's as long as all mcode before the relocation should happen.
+        reloc!(
+            { start: $start }
+            [ $( [ $($pos),* ], [ $($rep),* ] ,)*
+            [ $($lockstepmcpos),* ], [ ($lblmap!($lbl) + $start) as u8, (($lblmap!($lbl) + $start) >> 8) as u8 ] ],
+            $lblmap, [ $($mcode),* ], [ $($reloc),* ])
+    };
+    ( { $($attr:tt)* }  [ $( [ $($pos:expr),* ], [ $($rep:expr),* ] ),* ], $lblmap:ident, [ $($mcode:expr),* ], [ { $lbl:ident as PCREL @ [$($lockstepmcpos:expr),*] } $(,$reloc:tt)* ] ) => {
+        // Replace 1 Byte with the PC relative address
+        // PC is the program counter *after* the relocated offset (the length of the
+        // `$lockstepmcpos` array + 1), so we need to subtract 1 additional byte.
+        reloc!(
+            { $($attr)* }
+            [ $( [ $($pos),* ], [ $($rep),* ] ,)*
+            [ $($lockstepmcpos),* ], [ ((( $lblmap!($lbl) as i32 - codelen!($($lockstepmcpos),*) as i32 - 1 ) & 0x3FF ) << 23) ] ],
+            $lblmap, [ $($mcode),* ], [ $($reloc),* ])
+    };
+}
+
+
+// $attr is a label
+// $mcode is a list that contains the machine code opcodes
+// $ident is the identifier list
+// $reloc is the relocation list
+
+
+#[macro_export]
+#[doc(hidden)]
+macro_rules! asm_ {
+    ( { $($attr:tt)* } [ $($mcode:expr),* ], [ $($lbl:ident => $lblval:expr),* ], [ $($reloc:tt),* ],
+        // EOF
+    ) => {{
+        ident_map!(labelmap = {
+            $($lbl => $lblval),*
+        });
+        reloc!({ $($attr)* } [], labelmap, [ $($mcode),* ], [ $($reloc),* ])
+    }};
+
+    // ==================================================================================
+    // ==================================================================================
+    // ==================================================================================
+
+    // Opcode assembly table.
+    // Note that the weird order is required because macros try to match each arm in order, but
+    // don't backtrack when a NT is parsed
+
+    // ADD
+    ( { $($attr:tt)* } [ $($mcode:expr),* ], [ $($lbl:ident => $lblval:expr),* ], [ $($reloc:tt),* ],
+        add % $wd:tt, % $ra:tt , % $rb:tt
+    $($rest:tt)* ) => {
+        asm_!({ $($attr)* } [ $($mcode,)* $wd << 16 | 0 << 11 | $ra << 6 | 0 << 17 | $rb << 12 | 0x5 ], [ $($lbl => $lblval),* ], [ $($reloc),* ], $($rest)*)
+    };
+    ( { $($attr:tt)* } [ $($mcode:expr),* ], [ $($lbl:ident => $lblval:expr),* ], [ $($reloc:tt),* ],
+        add % $wd:tt, % $ra:tt , # $rb:tt
+    $($rest:tt)* ) => {
+        asm_!({ $($attr)* } [ $($mcode,)* $wd << 16 | 0 << 11 | $ra << 6 | 1 << 17 | $rb << 12 | 0x5 ], [ $($lbl => $lblval),* ], [ $($reloc),* ], $($rest)*)
+    };
+    ( { $($attr:tt)* } [ $($mcode:expr),* ], [ $($lbl:ident => $lblval:expr),* ], [ $($reloc:tt),* ],
+        add % $wd:tt, # $ra:tt , % $rb:tt
+    $($rest:tt)* ) => {
+        asm_!({ $($attr)* } [ $($mcode,)* $wd << 16 | 1 << 11 | $ra << 6 | 0 << 17 | $rb << 12 | 0x5 ], [ $($lbl => $lblval),* ], [ $($reloc),* ], $($rest)*)
+    };
+    ( { $($attr:tt)* } [ $($mcode:expr),* ], [ $($lbl:ident => $lblval:expr),* ], [ $($reloc:tt),* ],
+        add % $wd:tt, # $ra:tt , # $rb:tt
+    $($rest:tt)* ) => {
+        asm_!({ $($attr)* } [ $($mcode,)* $wd << 16 | 1 << 11 | $ra << 6 | 1 << 17 | $rb << 12 | 0x5 ], [ $($lbl => $lblval),* ], [ $($reloc),* ], $($rest)*)
+    };
+
+    // BRZ
+    ( { $($attr:tt)* } [ $($mcode:expr),* ], [ $($lbl:ident => $lblval:expr),* ], [ $($reloc:tt),* ],
+        brz $label:ident , % $ra:tt
+    $($rest:tt)* ) => {
+        asm_!({ $($attr)* } [ $($mcode,)* 0x0 ],
+            [ $($lbl => $lblval),* ], [ $($reloc,)* { $label as PCREL @ [ $($mcode,)* ] } ], $($rest)*)
+    };
+    ( { $($attr:tt)* } [ $($mcode:expr),* ], [ $($lbl:ident => $lblval:expr),* ], [ $($reloc:tt),* ],
+        brz $label:ident , # $ra:tt
+    $($rest:tt)* ) => {
+        asm_!({ $($attr)* } [ $($mcode,)* 0 << 23 | 1 << 11 | $ra << 6 | 0x9 ],
+            [ $($lbl => $lblval),* ], [ $($reloc,)* { $label as PCREL @ [$($mcode,)* 0 << 23 | 1 << 11 | $ra << 6 | 0x9] } ], $($rest)*)
+    };
+
+    // ==================================================================================
+    // ==================================================================================
+    // ==================================================================================
+
+    // Check for labels
+    ( { $($attr:tt)* } [ $($mcode:expr),* ], [ $($lbl:ident => $lblval:expr),* ], [ $($reloc:tt),* ],
+        $label:ident :
+    $($rest:tt)* ) => {
+        asm_!(
+            { $($attr)* }
+            [ $($mcode),* ],
+            [ $($lbl => $lblval,)* $label => codelen!($($mcode),*) ],
+            [ $($reloc),* ],
+            $($rest)*
+        )
+    };
+}
+
+#[macro_export]
+macro_rules! assemble_engine25519 {
+    ( {
+        start: $start:expr,
+        code: {
+            $($tokens:tt)*
+        }
+    } ) => {
+        asm_!({ start: $start } [], [], [], $($tokens)*)
+    };
+    ( $($tokens:tt)* ) => {
+        assemble_engine25519!({
+            start: 0,
+            code: {
+                $($tokens)*
+            }
+        })
+    };
+}
+/*
+/// Does anything work
+#[test]
+fn basic_syntax() {
+    let mcode = assemble_engine25519!(
+        start:
+            add %0, %1, %2
+            add %2, %3, #4
+            add %5, #6, %7
+            add %8, #9, #10
+            brz start, %11
+    );
+    assert_eq!(mcode, [
+        0 << 16 | 0 << 11 | 1  << 6 | 0 << 17 | 2  << 12 | 0x5,
+        2 << 16 | 0 << 11 | 3  << 6 | 1 << 17 | 4  << 12 | 0x5,
+        5 << 16 | 1 << 11 | 6  << 6 | 0 << 17 | 7  << 12 | 0x5,
+        8 << 16 | 1 << 11 | 9  << 6 | 1 << 17 | 10 << 12 | 0x5,
+        0x3FC << 23 | 0 << 11 | 11 << 6 | 0x9,
+     ]);
+}
+*/
+
+/// Simple jump
+#[test]
+fn simple_jmp() {
+    let mcode = assemble_engine25519!(
+        start:
+            add %0, %1, %2
+            brz start, %11
+    );
+    print!("{}", mcode);
+    assert_eq!(mcode, [
+        0 << 16 | 0 << 11 | 1  << 6 | 0 << 17 | 2  << 12 | 0x5,
+        0x3FE << 23 | 0 << 11 | 11 << 6 | 0x9,
+     ]);
+}
+
+/*
+/// Test simple label relocation
+#[test]
+fn simple_jmp() {
+    let mcode = assemble_engine25519!(
+        start: brz start, r0
+    );
+    assert_eq!(mcode, [ 0x4C, 0x00, 0x00 ]);
+}
+
+/// Has to work without any relocations (label references)
+#[test]
+fn no_reloc() {
+    let mcode = assemble6502!(
+        start:
+            lda #0xfb
+    );
+    assert_eq!(mcode, [ 0xA9, 0xFB ]);
+}
+
+/// Has to work without any labels
+#[test]
+fn no_label() {
+    let mcode = assemble6502!(
+        lda #0xfb
+        lda #0xab
+    );
+    assert_eq!(mcode, [ 0xA9, 0xFB, 0xA9, 0xAB ]);
+}
+*/
\ No newline at end of file