revme/cmd/
bytecode.rs

Help
1use clap::Parser;
2use revm::{
3    bytecode::eof::{self, validate_raw_eof_inner, CodeType, EofError},
4    primitives::{hex, Bytes},
5};
6use std::io;
7
8/// `bytecode` subcommand.
9#[derive(Parser, Debug)]
10pub struct Cmd {
11    /// Is EOF code in INITCODE mode.
12    #[arg(long)]
13    eof_initcode: bool,
14    /// Is EOF code in RUNTIME mode.
15    #[arg(long)]
16    eof_runtime: bool,
17    /// Bytecode in hex format string.
18    ///
19    /// - If bytes start with 0xFE it will be interpreted as a EOF.
20    /// - Otherwise, it will be interpreted as a EOF bytecode.
21    /// - If not provided, it will operate in interactive EOF validation mode.
22    #[arg()]
23    bytes: Option<String>,
24}
25
26#[inline]
27fn trim_decode(input: &str) -> Option<Bytes> {
28    let trimmed = input.trim().trim_start_matches("0x");
29    let decoded = hex::decode(trimmed).ok().map(Into::into);
30    if decoded.is_none() {
31        eprintln!("Invalid hex string");
32        return None;
33    }
34    decoded
35}
36
37impl Cmd {
38    /// Runs statetest command.
39    pub fn run(&self) {
40        let container_kind = if self.eof_initcode {
41            Some(CodeType::Initcode)
42        } else if self.eof_runtime {
43            Some(CodeType::Runtime)
44        } else {
45            None
46        };
47
48        if let Some(input_bytes) = &self.bytes {
49            let Some(bytes) = trim_decode(input_bytes) else {
50                return;
51            };
52
53            if bytes[0] == 0xEF {
54                match validate_raw_eof_inner(bytes, container_kind) {
55                    Ok(eof) => {
56                        println!("Decoding: {:#?}", eof);
57                        println!("Validation: OK");
58                    }
59                    Err(eof_error) => {
60                        eprintln!("err: {}", eof_error);
61                    }
62                }
63            } else {
64                eof::printer::print(&bytes)
65            }
66            return;
67        }
68
69        // Else run command in loop.
70        loop {
71            let mut input = String::new();
72            io::stdin().read_line(&mut input).expect("Input Error");
73            if input.len() == 1 {
74                // Just a newline, so exit
75                return;
76            }
77            let Some(bytes) = trim_decode(&input) else {
78                return;
79            };
80            match validate_raw_eof_inner(bytes, container_kind) {
81                Ok(eof) => {
82                    println!(
83                        "OK {}/{}/{}",
84                        eof.body.code_section.len(),
85                        eof.body.container_section.len(),
86                        eof.body.data_section.len()
87                    );
88                }
89                Err(eof_error) => {
90                    if matches!(
91                        eof_error,
92                        EofError::Decode(eof::EofDecodeError::InvalidEOFSize)
93                    ) {
94                        continue;
95                    }
96                    println!("err: {}", eof_error);
97                }
98            }
99        }
100    }
101}
revme/cmd/bytecode.rs

revme/cmd/
bytecode.rs
