evmdasm 0.1.1

EVM bytecode disassembler
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

use crate::dasm::instruction::disasm_one;

use super::Instruction;

#[derive(Default)]
struct State {
    offset: usize,
}

pub struct ByteCodeReader<'a> {
    bytecode: &'a [u8],
    state: State,
}

impl<'a> ByteCodeReader<'a> {
    pub fn new(bytecode: &'a [u8]) -> Self {
        ByteCodeReader {
            bytecode,
            state: Default::default(),
        }
    }

    fn advance(&mut self, step: usize) {
        self.state.offset += step;
    }

    pub fn next_instrunction(&mut self) -> Option<Instruction<'a>> {
        let bytecode = &self.bytecode[(self.state.offset)..];
        let instruction = disasm_one(self.state.offset, bytecode)?;

        self.advance(instruction.size);
        Some(instruction)
    }
}

impl<'a> Iterator for ByteCodeReader<'a> {
    type Item = Instruction<'a>;

    fn next(&mut self) -> Option<Self::Item> {
        self.next_instrunction()
    }
}