pcie dma_alloc_coherent编程示例-DPDK中文网

在Linux内核中，dma_alloc_coherent()函数用于分配一块适合DMA操作的内存区域。这个函数会返回一个虚拟地址和相应的物理地址，这样设备就可以通过DMA访问这块内存。

以下是一个使用dma_alloc_coherent()的简单示例，该示例演示了如何在驱动程序中分配和释放DMA一致性内存：

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/dma-mapping.h>

#define DEVICE_NAME "my_device"
#define BUFFER_SIZE 1024

struct my_device {
    void *buffer;          // 用于DMA的缓冲区
    dma_addr_t dma_handle; // DMA地址
};

static struct my_device *dev;

static int __init my_driver_init(void) {
    dev = kmalloc(sizeof(struct my_device), GFP_KERNEL);
    if (!dev) {
        printk(KERN_ERR "Failed to allocate memory for device\n");
        return -ENOMEM;
    }

    // 分配DMA一致性内存
    dev->buffer = dma_alloc_coherent(NULL, BUFFER_SIZE, &dev->dma_handle, GFP_KERNEL);
    if (!dev->buffer) {
        printk(KERN_ERR "Failed to allocate DMA coherent memory\n");
        kfree(dev);
        return -ENOMEM;
    }

    printk(KERN_INFO "Allocated DMA buffer at virtual address: %p, physical address: %pad\n",
           dev->buffer, &dev->dma_handle);

    return 0; // 驱动加载成功
}

static void __exit my_driver_exit(void) {
    // 释放DMA一致性内存
    dma_free_coherent(NULL, BUFFER_SIZE, dev->buffer, dev->dma_handle);
    
    kfree(dev); // 释放设备结构体
}

module_init(my_driver_init);
module_exit(my_driver_exit);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Your Name");
MODULE_DESCRIPTION("A simple example of using dma_alloc_coherent");

代码说明：

结构体定义：定义了一个my_device结构体，包含指向DMA缓冲区的指针和其对应的DMA地址。
初始化函数 (my_driver_init)：
- 使用kmalloc分配设备结构体。
- 调用dma_alloc_coherent()来分配一致性的内存，并检查返回值。
- 如果成功，打印出虚拟地址和物理地址。
退出函数 (my_driver_exit)：
- 使用dma_free_coherent()释放之前分配的DMA缓冲区。
- 最后释放设备结构体。