pyc_editor 0.4.6

A Rust library for reading, modifying, and writing Python .pyc files.
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
43
44
45
46

use crate::v310::ext_instructions::ExtInstructions;

#[derive(Debug, Clone, PartialEq, Eq)]
pub enum BlockIndex {
    Index(usize),
    /// For jumps with invalid jump targets
    InvalidIndex(usize),
    /// For blocks without a target
    NoIndex,
}

/// Represents a block in the control flow graph
pub struct Block {
    instructions: ExtInstructions,
    /// Index to block for conditional jump
    branch_block: BlockIndex,
    /// Index to default block (unconditional)
    default_block: BlockIndex,
}

impl Block {
    pub fn is_terminating(&self) -> bool {
        matches!(self.default_block, BlockIndex::NoIndex)
    }

    /// Whether the block has a conditional jump or not
    pub fn is_conditional(&self) -> bool {
        matches!(self.branch_block, BlockIndex::Index(_))
    }
}

pub struct ControlFlowGraph {
    blocks: Vec<Block>,
    start_index: BlockIndex,
}

impl From<ExtInstructions> for ControlFlowGraph {
    fn from(value: ExtInstructions) -> Self {
        let blocks = vec![];

        ControlFlowGraph {
            blocks,
            start_index: BlockIndex::NoIndex,
        }
    }
}