Linux 驱动开发中的内存管理：从 mem_cgroup_sk_alloc 到替代方案

引言

在 Linux 内核开发中，内存管理是核心任务之一。对于驱动开发者来说，理解内核内存控制组（memcg）和动态内存分配的机制至关重要。本文将从 mem_cgroup_sk_alloc 函数的功能解析入手，探讨其在驱动开发中的局限性，并提供适用于驱动场景的替代方案与最佳实践。

mem_cgroup_sk_alloc 的功能与限制

1. 功能解析

mem_cgroup_sk_alloc 是 Linux 内核中用于将 socket 内存使用关联到内存控制组的函数，其主要作用包括：

• 归属控制：将 socket 的内存消耗（如 TCP 缓冲区）绑定到当前进程的 memcg。

• 资源统计：支持 cgroup 对网络内存的统计和限制。

• 层级兼容：区分传统 cgroup（v1）和统一层级（v2）的行为。

其核心逻辑如下：

void mem_cgroup_sk_alloc(struct sock *sk) {
    // 1. 全局开关检查
    if (!mem_cgroup_sockets_enabled) return;
    // 2. 中断上下文保护
    if (in_interrupt(