absolutebi/serai
1
0
Fork 0
mirror of https://github.com/serai-dex/serai.git synced 2025-12-09 12:49:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Replace custom read/write impls in router with borsh

Browse Source
This commit is contained in:
Luke Parker
2025-01-21 03:49:29 -05:00
parent f690bf831f
commit c8f3a32fdf
9 changed files with 81 additions and 217 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
processor/ethereum/router/src/tests/mod.rs
View File

@@ -17,13 +17,12 @@ use alloy_provider::RootProvider;
use alloy_node_bindings::{Anvil, AnvilInstance};
use ethereum_primitives::LogIndex;
use ethereum_schnorr::{PublicKey, Signature};
use ethereum_deployer::Deployer;
use crate::{Coin, OutInstructions, Router};
mod read_write;
#[test]
fn execute_reentrancy_guard() {
let hash = alloy_core::primitives::keccak256(b"ReentrancyGuard Router.execute");
@@ -217,7 +216,10 @@ async fn test_eth_in_instruction() {
assert_eq!(parsed_in_instructions.len(), 1);
assert_eq!(
parsed_in_instructions[0].id,
(<[u8; 32]>::from(receipt.block_hash.unwrap()), receipt.inner.logs()[0].log_index.unwrap())
LogIndex {
block_hash: *receipt.block_hash.unwrap(),
index_within_block: receipt.inner.logs()[0].log_index.unwrap(),
},
);
assert_eq!(parsed_in_instructions[0].from, signer);
assert_eq!(parsed_in_instructions[0].coin, Coin::Ether);

85
processor/ethereum/router/src/tests/read_write.rs
View File

@@ -1,85 +0,0 @@
use rand_core::{RngCore, OsRng};
use alloy_core::primitives::U256;
use crate::{Coin, InInstruction, Executed};
fn coins() -> [Coin; 2] {
[Coin::Ether, {
let mut erc20 = [0; 20];
OsRng.fill_bytes(&mut erc20);
Coin::Erc20(erc20.into())
}]
}
#[test]
fn test_coin_read_write() {
for coin in coins() {
let mut res = vec![];
coin.write(&mut res).unwrap();
assert_eq!(coin, Coin::read(&mut res.as_slice()).unwrap());
}
}
#[test]
fn test_in_instruction_read_write() {
for coin in coins() {
let instruction = InInstruction {
id: (
{
let mut tx_id = [0; 32];
OsRng.fill_bytes(&mut tx_id);
tx_id
},
OsRng.next_u64(),
),
from: {
let mut from = [0; 20];
OsRng.fill_bytes(&mut from);
from
},
coin,
amount: U256::from_le_bytes({
let mut amount = [0; 32];
OsRng.fill_bytes(&mut amount);
amount
}),
data: {
let len = usize::try_from(OsRng.next_u64() % 65536).unwrap();
let mut data = vec![0; len];
OsRng.fill_bytes(&mut data);
data
},
};
let mut buf = vec![];
instruction.write(&mut buf).unwrap();
assert_eq!(InInstruction::read(&mut buf.as_slice()).unwrap(), instruction);
}
}
#[test]
fn test_executed_read_write() {
for executed in [
Executed::SetKey {
nonce: OsRng.next_u64(),
key: {
let mut key = [0; 32];
OsRng.fill_bytes(&mut key);
key
},
},
Executed::Batch {
nonce: OsRng.next_u64(),
message_hash: {
let mut message_hash = [0; 32];
OsRng.fill_bytes(&mut message_hash);
message_hash
},
},
] {
let mut res = vec![];
executed.write(&mut res).unwrap();
assert_eq!(executed, Executed::read(&mut res.as_slice()).unwrap());
}
}