waffles-solana-bpf-loader-program 1.15.0

#![allow(clippy::integer_arithmetic)]

use {
    solana_program_runtime::invoke_context::{Alloc, AllocErr},
    solana_rbpf::{aligned_memory::AlignedMemory, ebpf::HOST_ALIGN},
    std::alloc::Layout,
};

#[derive(Debug)]
pub struct BpfAllocator {
    #[allow(dead_code)]
    heap: AlignedMemory<HOST_ALIGN>,
    start: u64,
    len: u64,
    pos: u64,
}

impl BpfAllocator {
    pub fn new(heap: AlignedMemory<HOST_ALIGN>, virtual_address: u64) -> Self {
        let len = heap.len() as u64;
        Self {
            heap,
            start: virtual_address,
            len,
            pos: 0,
        }
    }

    pub fn get_heap(&mut self) -> &mut [u8] {
        self.heap.as_slice_mut()
    }
}

impl Alloc for BpfAllocator {
    fn alloc(&mut self, layout: Layout) -> Result<u64, AllocErr> {
        let bytes_to_align = (self.pos as *const u8).align_offset(layout.align()) as u64;
        if self
            .pos
            .saturating_add(layout.size() as u64)
            .saturating_add(bytes_to_align)
            <= self.len
        {
            self.pos += bytes_to_align;
            let addr = self.start + self.pos;
            self.pos += layout.size() as u64;
            Ok(addr)
        } else {
            Err(AllocErr)
        }
    }

    fn dealloc(&mut self, _addr: u64, _layout: Layout) {
        // It's a bump allocator, free not supported
    }
}