手把手教你写一个DRM GEM CMA驱动：从dumb buffer到mmap映射的完整流程

从零构建DRM GEM CMA驱动：深入解析dumb buffer与mmap映射实战

在Linux图形驱动开发领域，DRM（Direct Rendering Manager）子系统扮演着核心角色，而GEM（Graphics Execution Manager）作为其内存管理框架，更是开发者必须掌握的关键技术。本文将带您从零开始，完整实现一个支持CMA Helper功能的DRM驱动模块，重点剖析drm_gem_cma_dumb_create和drm_gem_cma_mmap两大核心函数的实现细节，让您不仅理解原理，更能亲手构建可运行的驱动代码。

1. 环境准备与基础概念

在开始编码之前，我们需要确保开发环境配置正确，并明确几个关键概念：

开发环境要求：

• Linux内核版本 ≥ 4.19（推荐5.10+ LTS版本）
• 已安装内核头文件及开发工具链
• 启用DRM子系统及CMA相关配置

核心概念解析：

• DRM框架：Linux内核中管理图形设备的统一接口层
• GEM对象：表示图形内存的基本单元，包含元数据和实际内存资源
• dumb buffer：最简单的GEM对象类型，仅提供连续内存分配功能
• CMA Helper：DRM提供的通用API集合，简化连续内存管理

提示：虽然名为"CMA Helper"，但其实现并不依赖CONFIG_CMA配置，实际使用的是dma_alloc_wc()接口

2. 驱动框架搭建

首先构建最基本的DRM驱动骨架，这是所有DRM驱动的起点：

#include <drm/drmP.h> #include <drm/drm_gem.h> static struct drm_device drm; static const struct file_operations mygem_fops = { .owner = THIS_MODULE, .open = drm_open, .release = drm_release, .unlocked_ioctl = drm_ioctl, .mmap = drm_gem_cma_mmap, // 后续实现 }; static struct drm_driver mygem_driver = { .driver_features = DRIVER_GEM, .fops = &mygem_fops, .dumb_create = drm_gem_cma_dumb_create, // 后续实现 .name = "my-gem-cma", .desc = "My GEM CMA Driver", .major = 1, .minor = 0, }; static int __init mygem_init(void) { drm_dev_init(&drm, &mygem_driver, NULL); return drm_dev_register(&drm, 0); } module_init(mygem_init);

这个基础框架已经包含了DRM驱动的主要结构，接下来我们需要实现两个关键函数：drm_gem_cma_dumb_create和drm_gem_cma_mmap。

3. 实现dumb buffer创建

drm_gem_cma_dumb_create函数负责创建dumb buffer，其核心任务可分为三个部分：

1. 计算buffer参数：

   args->pitch = args->width * args->bpp / 8; args->size = args->pitch * args->height;

2. 创建GEM对象： 首先定义CMA对象结构体：

   struct drm_gem_cma_object { struct drm_gem_object base; dma_addr_t paddr; void *vaddr; };

   然后实现对象创建逻辑：

   cma_obj = kzalloc(sizeof(*cma_obj), GFP_KERNEL); drm_gem_object_init(drm, &cma_obj->base, args->size);

3. 分配连续内存：

   cma_obj->vaddr = dma_alloc_wc(drm->dev, args->size, &cma_obj->paddr, GFP_KERNEL | __GFP_NOWARN);

完整函数实现如下：

static int drm_gem_cma_dumb_create(struct drm_file *file_priv, struct drm_device *drm, struct drm_mode_create_dumb *args) { struct drm_gem_cma_object *cma_obj; // 计算buffer参数 args->pitch = args->width * args->bpp / 8; args->size = args->pitch * args->height; // 创建GEM对象 cma_obj = kzalloc(sizeof(*cma_obj), GFP_KERNEL); if (!cma_obj) return -ENOMEM; drm_gem_object_init(drm, &cma_obj->base, args->size); // 分配连续内存 cma_obj->vaddr = dma_alloc_wc(drm->dev, args->size, &cma_obj->paddr, GFP_KERNEL | __GFP_NOWARN); if (!cma_obj->vaddr) { drm_gem_object_release(&cma_obj->base); kfree(cma_obj); return -ENOMEM; } // 创建handle return drm_gem_handle_create(file_priv, &cma_obj->base, &args->handle); }

4. 实现mmap映射功能

drm_gem_cma_mmap函数负责将dumb buffer映射到用户空间，其实现要点包括：

1. 调用drm_gem_mmap完成基础映射设置
2. 获取CMA对象从vma私有数据
3. 建立物理到虚拟的映射

现代内核推荐使用dma_mmap_wc替代直接调用remap_pfn_range：

static int drm_gem_cma_mmap(struct file *filp, struct vm_area_struct *vma) { struct drm_gem_cma_object *cma_obj; int ret; ret = drm_gem_mmap(filp, vma); if (ret) return ret; cma_obj = vma->vm_private_data; return dma_mmap_wc(vma->vm_file->private_data, vma, cma_obj->vaddr, cma_obj->paddr, vma->vm_end - vma->vm_start); }

注意：早期内核版本可能需要使用remap_pfn_range，但现代内核(≥4.6)建议使用dma_mmap_wc以获得更好的可移植性和缓存一致性

5. 内存释放与完整实现

完整的驱动必须包含内存释放功能，避免内存泄漏：

static void drm_gem_cma_free_object(struct drm_gem_object *gem_obj) { struct drm_gem_cma_object *cma_obj = to_drm_gem_cma_obj(gem_obj); if (cma_obj->vaddr) dma_free_wc(gem_obj->dev->dev, gem_obj->size, cma_obj->vaddr, cma_obj->paddr); drm_gem_object_release(gem_obj); kfree(cma_obj); }

将free函数添加到驱动结构中：

static struct drm_driver mygem_driver = { // ...其他成员不变... .gem_free_object_unlocked = drm_gem_cma_free_object, };

6. 测试与验证

编写用户空间测试程序验证驱动功能：

#include <fcntl.h> #include <stdio.h> #include <sys/mman.h> #include <xf86drm.h> #include <xf86drmMode.h> int main() { int fd = open("/dev/dri/card0", O_RDWR); struct drm_mode_create_dumb create = { .bpp = 32, .width = 240, .height = 320 }; // 创建dumb buffer drmIoctl(fd, DRM_IOCTL_MODE_CREATE_DUMB, &create); // 映射buffer struct drm_mode_map_dumb map = { .handle = create.handle }; drmIoctl(fd, DRM_IOCTL_MODE_MAP_DUMB, &map); char *vaddr = mmap(0, create.size, PROT_READ|PROT_WRITE, MAP_SHARED, fd, map.offset); // 测试读写 sprintf(vaddr, "DRM GEM CMA test"); printf("Read from buffer: %s\n", vaddr); // 清理 munmap(vaddr, create.size); struct drm_mode_destroy_dumb destroy = { .handle = create.handle }; drmIoctl(fd, DRM_IOCTL_MODE_DESTROY_DUMB, &destroy); close(fd); return 0; }

预期输出应显示成功写入和读取的测试字符串，证明驱动工作正常。

7. 高级主题与优化

在实际项目中，我们还需要考虑以下进阶问题：

性能优化技巧：

• 使用DEFINE_DRM_GEM_CMA_FOPS宏简化文件操作结构
• 实现prime接口支持buffer共享
• 添加drm_driver的ioctls回调处理自定义命令

调试与问题排查：

• 通过drm_print_registered_clients检查资源泄漏
• 使用drm_mm调试工具分析内存分配情况
• 添加DRM_DEBUG输出关键路径信息

跨版本兼容性处理：

#if LINUX_VERSION_CODE >= KERNEL_VERSION(5, 11, 0) // 新版内核API #else // 旧版兼容实现 #endif

通过本文的实践，我们不仅构建了一个完整的DRM GEM CMA驱动，更重要的是理解了DRM框架下内存管理的核心机制。在实际开发中，这种从底层实现入手的经验，能帮助开发者更高效地调试和优化图形驱动。