rwasm 0.4.3

/// A registry where local variables of a function are registered and resolved.
///
/// # Note
///
/// Note that in WebAssembly function parameters are also local variables.
///
/// The local registry efficiently registers and resolves local variables.
/// The problem is that the Wasm specification allows encoding up to `u32::MAX`
/// local variables in a small and constant space via the binary encoding.
/// Therefore, we need a way to efficiently cope with this worst-case scenario
///  to protect the `rwasm` interpreter against exploitation.
///
/// This implementation allows accessing local variables in this worst-case
/// scenario with the worst time complexity of O(log n) and space requirement
/// of O(m + n) where n is the number of registered groups of local variables
/// and m is the number of actually used local variables.
///
/// Besides that, local variable usages are cached to further minimize potential
/// exploitation impact.
#[derive(Debug, Default)]
pub struct LocalsRegistry {
    /// The number of registered local variables.
    len_registered: u32,
}

impl LocalsRegistry {
    /// Returns the number of registered local variables.
    ///
    /// # Note
    ///
    /// Since in WebAssembly function parameters are also local variables,
    /// this function actually returns the number of function parameters
    /// and explicitly defined local variables.
    pub(crate) fn len_registered(&self) -> u32 {
        self.len_registered
    }

    /// Registers an `amount` of local variables.
    ///
    /// # Panics
    ///
    /// If too many local variables have been registered.
    pub fn register_locals(&mut self, amount: u32) {
        if amount == 0 {
            return;
        }
        self.len_registered = self.len_registered.checked_add(amount).unwrap_or_else(|| {
            panic!(
                "tried to register too many local variables for the function: got {}, additional {amount}",
                self.len_registered
            )
        });
    }
}