initial commit
This commit is contained in:
bunnie
5
Cargo.lock
generated
Normal file
5
Cargo.lock
generated
Normal file
@@ -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"
|
||||
11
Cargo.toml
Normal file
11
Cargo.toml
Normal file
@@ -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]
|
||||
317
src/lib.rs
Normal file
317
src/lib.rs
Normal file
@@ -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 ]);
|
||||
}
|
||||
*/
|
||||
Reference in New Issue
Block a user