12、Rust 内存分配器与单元测试详解

Rust 内存分配器与单元测试详解

1. 自定义内存分配器

当需要编写自定义分配器时，以下内容可为你提供起点。我们将探索一个更高级的自定义内存分配器示例，此分配器可逐块应用于单个数据结构，而非整个程序，便于进行性能微调。

在某些场景中，现代操作系统为编写安全或关键系统的开发者提供了多种内存保护功能。为在 Rust 程序中利用这些功能，需编写自己的内存分配代码。例如，dryoc crate 使用了 UNIX 类系统的 mprotect() 和 mlock() 系统调用，以及 Windows 系统的 VirtualProtect() 和 VirtualLock() 系统调用，这些调用可锁定和控制进程内特定内存区域的访问，对管理敏感数据（如密钥）的代码尤为重要。

实现内存锁定和保护功能时，内存必须通过特定的平台相关内存函数分配（UNIX 上使用 posix_memalign() ，Windows 上使用 VirtualAlloc() ），以确保与平台特定的内存页对齐。此外，在目标内存区域前后会额外分配两个内存块并锁定，可将这些区域视为汽车的缓冲器，提供额外的内存攻击防护。

以下是一个部分代码示例：

/// Custom page-aligned allocator implementation. Creates blocks of page-aligned
/// heap-allocated memory re