Skip to content

Commit

Permalink
♻ Clean
Browse files Browse the repository at this point in the history
  • Loading branch information
clabby committed Sep 13, 2023
1 parent cd3ce82 commit cf3f630
Show file tree
Hide file tree
Showing 4 changed files with 37 additions and 13 deletions.
27 changes: 25 additions & 2 deletions .github/workflows/ci.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ jobs:
- name: Install Rust toolchain
uses: actions-rs/toolchain@v1
with:
toolchain: stable
toolchain: nightly
profile: minimal
override: true
- uses: Swatinem/rust-cache@v1
Expand Down Expand Up @@ -49,7 +49,7 @@ jobs:
- name: Install Rust toolchain
uses: actions-rs/toolchain@v1
with:
toolchain: stable
toolchain: nightly
profile: minimal
override: true
- uses: Swatinem/rust-cache@v1
Expand All @@ -59,3 +59,26 @@ jobs:
id: build
continue-on-error: true
run: cargo build --all
cargo-doc:
runs-on: ubuntu-latest
timeout-minutes: 20
continue-on-error: true
steps:
- name: Checkout sources
uses: actions/checkout@v3
- name: Install Rust toolchain
uses: actions-rs/toolchain@v1
with:
toolchain: nightly
profile: minimal
override: true
- uses: Swatinem/rust-cache@v1
with:
cache-on-failure: true
- name: doclint
id: build
continue-on-error: true
run: RUSTDOCFLAGS="-D warnings" cargo doc --all --no-deps --all-features --document-private-items
- name: doctest
run: cargo test --doc --all --all-features

3 changes: 2 additions & 1 deletion crates/mipsevm/src/lib.rs
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
// #![doc = include_str!("../README.md")]
#![feature(generic_const_exprs)]
#![allow(incomplete_features, dead_code)]

mod memory;
mod page;
mod state;
mod traits;
mod utils;
18 changes: 9 additions & 9 deletions crates/mipsevm/src/memory.rs → crates/mipsevm/src/page.rs
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
//! This module contains the data structure for a [Page] within the
//! MIPS emulator's memory.

use crate::utils::concat_arrays;
use crate::utils::concat_fixed;
use alloy_primitives::{keccak256, B256};
use once_cell::sync::Lazy;

Expand All @@ -19,7 +19,7 @@ type Page = [u8; PAGE_SIZE];
static ZERO_HASHES: Lazy<[B256; 256]> = Lazy::new(|| {
let mut out = [B256::ZERO; 256];
for i in 1..256 {
out[i] = keccak256(concat_arrays(out[i - 1].into(), out[i - 1].into()))
out[i] = keccak256(concat_fixed(out[i - 1].into(), out[i - 1].into()))
}
out
});
Expand Down Expand Up @@ -71,7 +71,7 @@ impl CachedPage {
pub fn merkle_root(&mut self) -> B256 {
// First, hash the bottom layer.
for i in (0..PAGE_SIZE).step_by(64) {
let j = PAGE_SIZE >> 5 + i >> 5;
let j = (PAGE_SIZE >> 6) + (i >> 6);
if self.valid[j] {
continue;
}
Expand All @@ -81,12 +81,12 @@ impl CachedPage {
}

// Then, hash the cache layers.
for i in (1..=PAGE_SIZE >> 5 - 2).rev().step_by(2) {
for i in (1..=(PAGE_SIZE >> 5) - 2).rev().step_by(2) {
let j = i >> 1;
if self.valid[j] {
continue;
}
self.cache[j] = *keccak256(concat_arrays(self.cache[i], self.cache[i + 1]));
self.cache[j] = *keccak256(concat_fixed(self.cache[i], self.cache[i + 1]));
self.valid[j] = true;
}

Expand All @@ -98,12 +98,12 @@ impl CachedPage {
let _ = self.merkle_root();

if g_index >= PAGE_SIZE >> 5 {
if g_index >= PAGE_SIZE >> 5 * 2 {
if g_index >= (PAGE_SIZE >> 5) * 2 {
panic!("Gindex too deep");
}

let node_index = g_index & (PAGE_ADDRESS_MASK >> 5);
return B256::from_slice(&self.data[(node_index << 5)..(node_index << 5 + 32)]);
return B256::from_slice(&self.data[node_index << 5..(node_index << 5) + 32]);
}

self.cache[g_index].into()
Expand All @@ -126,12 +126,12 @@ mod test {
assert_eq!(node, expected_leaf, "Leaf nodes should not be hashed");

let node = page.merklize_subtree(g_index >> 1);
let expected_parent = keccak256(concat_arrays(ZERO_HASHES[0].into(), expected_leaf.into()));
let expected_parent = keccak256(concat_fixed(ZERO_HASHES[0].into(), expected_leaf.into()));
assert_eq!(node, expected_parent, "Parent should be correct");

let node = page.merklize_subtree(g_index >> 2);
let expected_grandparent =
keccak256(concat_arrays(expected_parent.into(), ZERO_HASHES[1].into()));
keccak256(concat_fixed(expected_parent.into(), ZERO_HASHES[1].into()));
assert_eq!(node, expected_grandparent, "Grandparent should be correct");

let pre = page.merkle_root();
Expand Down
2 changes: 1 addition & 1 deletion crates/mipsevm/src/utils.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

/// Concatenate two fixed sized arrays together into a new array with minimal reallocation.
#[inline(always)]
pub(crate) fn concat_arrays<T, const N: usize, const M: usize>(a: [T; N], b: [T; M]) -> [T; N + M]
pub(crate) fn concat_fixed<T, const N: usize, const M: usize>(a: [T; N], b: [T; M]) -> [T; N + M]
where
T: Copy + Default,
{
Expand Down

0 comments on commit cf3f630

Please sign in to comment.