andrey/k210-sdk-stuff
1
0
Fork 0
mirror of https://github.com/laanwj/k210-sdk-stuff.git synced 2024-11-22 01:16:20 +04:00
Code Issues Projects Releases Wiki Activity

rust: Test sha256 accelerator too

Browse Source
This commit is contained in:
Wladimir J. van der Laan 2020-04-13 18:02:52 +00:00
parent b0bcfec157
commit 378f08da8e
1 changed files with 64 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
rust/cryptest/src/main.rs
View File

@ -4,6 +4,7 @@
#![no_std]
#![no_main]
use core::sync::atomic::{self, Ordering};
use k210_hal::{Peripherals, pac};
use k210_hal::prelude::*;
use k210_hal::stdout::Stdout;
@ -21,6 +22,9 @@ fn main() -> ! {
// Enable clocks for AES and reset the engine
sysctl::clock_enable(sysctl::clock::AES);
sysctl::reset(sysctl::reset::AES);
// Enable clocks for SHA256 and reset the engine
sysctl::clock_enable(sysctl::clock::SHA);
sysctl::reset(sysctl::reset::SHA);
// Configure UART
let serial = p
@ -69,14 +73,14 @@ fn main() -> ! {
// Can write up to 80 bytes (20 words) here at once before the queue is full.
for &v in pt.iter() {
while aes.data_in_flag.read().data_in_flag() != DATA_IN_FLAG_A::CAN_INPUT {
continue;
atomic::compiler_fence(Ordering::SeqCst)
}
aes.text_data.write(|w| w.bits(v));
}
for i in 0..4 {
while aes.data_out_flag.read().data_out_flag() != DATA_OUT_FLAG_A::CAN_OUTPUT {
continue;
atomic::compiler_fence(Ordering::SeqCst)
}
ct[i] = aes.out_data.read().bits();
//write!(stdout, "{:08x} ", val).unwrap();
@ -87,6 +91,64 @@ fn main() -> ! {
writeln!(stdout, "MISMATCH").unwrap();
}
}
/*
$ echo -n "abc" | sha256sum
ba7816bf8f01cfea414140de5dae2223b00361a396177a9cb410ff61f20015ad -
*/
let sha_in: [u8; 64] = [
b'a', b'b', b'c', 0x80, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x18, // Padding (total size in bits, big-endian)
];
let sha_out_ref: [u8; 32] = [
0xba, 0x78, 0x16, 0xbf, 0x8f, 0x01, 0xcf, 0xea,
0x41, 0x41, 0x40, 0xde, 0x5d, 0xae, 0x22, 0x23,
0xb0, 0x03, 0x61, 0xa3, 0x96, 0x17, 0x7a, 0x9c,
0xb4, 0x10, 0xff, 0x61, 0xf2, 0x00, 0x15, 0xad,
];
let mut sha_out = [0u8; 32];
write!(stdout, "SHA256: ").unwrap();
unsafe {
use pac::sha256::function_reg_0::{ENDIAN_A};
sha256.num_reg.write(|w|
w.data_cnt().bits(1));
sha256.function_reg_0.write(|w|
w.endian().variant(ENDIAN_A::BE)
.en().bit(true));
sha256.function_reg_1.write(|w|
w.dma_en().bit(false));
for i in 0..16 {
while sha256.function_reg_1.read().fifo_in_full().bit() {
atomic::compiler_fence(Ordering::SeqCst)
}
sha256.data_in.write(|w| w.bits(
((sha_in[i*4 + 0] as u32) << 0)
| ((sha_in[i*4 + 1] as u32) << 8)
| ((sha_in[i*4 + 2] as u32) << 16)
| ((sha_in[i*4 + 3] as u32) << 24)
));
}
while !sha256.function_reg_0.read().en().bit() {
atomic::compiler_fence(Ordering::SeqCst)
}
for i in 0..8 {
let val = sha256.result[7 - i].read().bits();
sha_out[i*4 + 0] = (val >> 0) as u8;
sha_out[i*4 + 1] = (val >> 8) as u8;
sha_out[i*4 + 2] = (val >> 16) as u8;
sha_out[i*4 + 3] = (val >> 24) as u8;
}
if sha_out == sha_out_ref {
writeln!(stdout, "MATCH").unwrap();
} else {
writeln!(stdout, "MISMATCH").unwrap();
}
}
loop {
unsafe { asm::wfi(); }
}