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

pub mod trace_entry {
    use serde::{Deserialize, Serialize};

    use crate::{
        stdlib::prelude::*,
        types::relocatable::Relocatable,
        vm::errors::{memory_errors::MemoryError, trace_errors::TraceError},
    };

    ///A trace entry for every instruction that was executed.
    ///Holds the register values before the instruction was executed.
    ///Register values for ap & fp are represented as their offsets, as their indexes will always be 1
    #[derive(Debug, PartialEq, Eq, Deserialize, Serialize, Clone)]
    pub struct TraceEntry {
        pub pc: Relocatable,
        pub ap: usize,
        pub fp: usize,
    }

    /// A trace entry for every instruction that was executed.
    /// Holds the register values before the instruction was executed, after going through the relocation process
    #[derive(Debug, PartialEq, Eq, Deserialize, Serialize, Clone)]
    pub struct RelocatedTraceEntry {
        pub pc: usize,
        pub ap: usize,
        pub fp: usize,
    }

    pub fn relocate_trace_register(
        value: Relocatable,
        relocation_table: &[usize],
    ) -> Result<usize, TraceError> {
        let segment_index: usize = value.segment_index.try_into().map_err(|_| {
            TraceError::MemoryError(MemoryError::AddressInTemporarySegment(value.segment_index))
        })?;

        if relocation_table.len() <= segment_index {
            return Err(TraceError::NoRelocationFound);
        }
        Ok(relocation_table[segment_index] + value.offset)
    }
}