使用Rust构建.wasm二进制文件时如何访问__heap_base变量

你编写的代码错误点在于：链接器只是将__heap_base这个外部静态变量的内存地址设置为线性内存的堆起始位置，而非将堆起始值存在这个变量指向的内存空间里。你直接读取__heap_base的取值，本质是读取堆起始位置上存储的初始值，Wasm线性内存默认初始化为0，所以返回值自然是0。

1. 基础读取方法 #

你只需要取__heap_base的地址就能拿到正确的堆起始值，变量本身的类型不影响结果，仅作为地址标记使用：

extern "C" {
    static __heap_base: u8; // 类型可以任意设置，仅用于取地址
}

#[no_mangle]
pub unsafe extern "C" fn main() -> usize {
    // 取变量地址即为堆起始位置
    &__heap_base as *const u8 as usize
}

2. 自定义分配器最简示例 #

如果你要实现自定义 bump 分配器，可以参考以下框架：

use core::alloc::{GlobalAlloc, Layout};
use core::ptr;

extern "C" {
    static __heap_base: u8;
    // 可选：如果在链接参数中指定了堆上限，可以声明该变量获取堆结束地址
    static __heap_end: u8;
}

struct CustomAllocator {
    current: core::sync::atomic::AtomicUsize,
}

unsafe impl GlobalAlloc for CustomAllocator {
    unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
        let heap_start = &__heap_base as *const u8 as usize;
        let heap_end = &__heap_end as *const u8 as usize;
        // 按分配要求做内存对齐
        let align = layout.align();
        let size = layout.size();
        
        let mut current = self.current.load(core::sync::atomic::Ordering::SeqCst);
        let alloc_start = (current + align - 1) & !(align - 1);
        
        if alloc_start + size > heap_end {
            return ptr::null_mut();
        }
        
        self.current.store(alloc_start + size, core::sync::atomic::Ordering::SeqCst);
        alloc_start as *mut u8
    }

    unsafe fn dealloc(&self, _ptr: *mut u8, _layout: Layout) {
        // 简单 bump 分配器无需实现内存释放，可根据需求扩展逻辑
    }
}

#[global_allocator]
static ALLOCATOR: CustomAllocator = CustomAllocator {
    current: core::sync::atomic::AtomicUsize::new(unsafe { &__heap_base as *const u8 as usize }),
};

3. 编译注意事项 #

使用cargo build --target wasm32-unknown-unknown或者wasm-pack编译时，默认会自动设置__heap_base的地址，不需要额外添加链接参数。如果需要自定义线性内存上限，可以添加编译参数-C link-arg=--max-memory=自定义内存大小。

内容的提问来源于stack exchange，提问作者Federico Rodríguez