brk系统调用与进程内存管理深度解析

1. 进程内存管理：brk系统调用深度解析

在x86架构的操作系统开发中，内存管理是最核心的子系统之一。brk系统调用作为进程内存管理的基础设施，负责动态调整进程的堆内存边界。理解其实现原理对于开发内存分配器（如malloc/free）和优化程序内存使用至关重要。

1.1 进程内存布局与brk定位

在典型的32位x86系统中，用户进程的内存空间通常被划分为几个关键区域：

• 代码段（Text Segment）：存放可执行指令，从固定地址（如8MB）开始加载
• 数据段（Data Segment）：存放已初始化的全局和静态变量
• BSS段：存放未初始化的全局和静态变量
• 堆区（Heap）：动态内存分配区域，向上增长
• 栈区（Stack）：向下增长，用于函数调用和局部变量

brk指针标记了堆区的当前上限。当程序需要更多堆内存时，通过brk系统调用请求操作系统调整这个边界值。我们的实现中，用户空间从8MB扩展到128MB，堆区位于ELF各段之后，栈区之前。

1.2 brk系统调用实现剖析

brk的核心实现需要考虑以下几个关键方面：

c复制int32 sys_brk(void *addr) {
    u32 brk = (u32)addr;
    ASSERT_PAGE(brk);  // 确保地址按页对齐
    task_t *task = running_task();
    
    // 安全性检查
    assert(task->uid != KERNEL_USER);
    assert(KERNEL_MEMORY_SIZE < brk < USER_STACK_BOTTOM);
    
    u32 old_brk = task->brk;
    
    // 缩小堆空间
    if (old_brk > brk) {
        for (; brk < old_brk; brk += PAGE_SIZE) {
            unlink_page(brk);  // 逐页释放物理内存
        }
    } 
    // 扩大堆空间（延迟分配）
    else if (IDX(brk - old_brk) > free_pages) {
        return -1;  // 内存不足
    }
    
    task->brk = brk;  // 更新堆边界
    return 0;
}

关键设计决策：

1. 延迟分配策略：仅修改虚拟地址边界，不立即分配物理页。当程序实际访问新区域时触发缺页异常，再由内核分配物理页。这种"懒加载"方式避免了不必要的内存占用。

2. 按页管理：内存以4KB页为单位管理，brk地址必须页对齐。缩小堆时立即回收物理页，而扩大堆时只记录新的边界。

3. 安全边界检查：确保请求的地址在用户空间范围内，防止进程越界访问内核空间。

1.3 缺页异常处理机制

当程序访问尚未映射的虚拟页时，CPU触发缺页异常（Page Fault），内核的page_fault_handler负责处理：

c复制void page_fault_handler(...) {
    u32 vaddr = get_cr2();  // 获取触发异常的地址
    page_error_code_t *code = (page_error_code_t *)&error;
    task_t *task = running_task();
    
    // 写时复制处理
    if (code->present && code->write) {
        handle_cow(vaddr);  // 处理写时复制
        return;
    }
    
    // 按需分配处理
    if (!code->present && 
        (vaddr < task->brk || vaddr >