dma_alloc_coherent的驱动代码编写示例-DPDK中文网

dma_alloc_coherent 是 Linux 内核中用于分配可用于 DMA 的内存的一种函数。它确保分配的内存是连续的，并且适合于设备进行直接访问。

以下是一个简单的示例，展示如何在驱动程序中使用 dma_alloc_coherent 分配 DMA 可用的内存：

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

struct my_device {
    struct device *dev;
    void *dma_buf;  // Pointer to the DMA buffer
    dma_addr_t dma_handle;  // DMA address of the buffer
};

static int my_driver_probe(struct platform_device *pdev)
{
    struct my_device *my_dev;
    
    my_dev = devm_kzalloc(&pdev->dev, sizeof(*my_dev), GFP_KERNEL);
    if (!my_dev) {
        dev_err(&pdev->dev, "Failed to allocate memory for device structure\n");
        return -ENOMEM;
    }

    my_dev->dev = &pdev->dev;

    // Allocate coherent DMA memory
    my_dev->dma_buf = dma_alloc_coherent(my_dev->dev, PAGE_SIZE,
                                          &my_dev->dma_handle, GFP_KERNEL);
    if (!my_dev->dma_buf) {
        dev_err(&pdev->dev, "Failed to allocate DMA coherent memory\n");
        return -ENOMEM;
    }

    dev_info(&pdev->dev, "DMA buffer allocated at %p with handle %pad\n",
             my_dev->dma_buf, &my_dev->dma_handle);

    // Your device initialization code here...

    return 0;
}

static int my_driver_remove(struct platform_device *pdev)
{
    struct my_device *my_dev = platform_get_drvdata(pdev);

    // Free the coherent memory
    if (my_dev && my_dev->dma_buf) {
        dma_free_coherent(my_dev->dev, PAGE_SIZE,
                           my_dev->dma_buf, my_dev->dma_handle);
        dev_info(&pdev->dev, "DMA buffer freed\n");
    }

    return 0;
}

static const struct of_device_id my_driver_of_match[] = {
   { .compatible = "vendor,my-device", },
   {},
};
MODULE_DEVICE_TABLE(of, my_driver_of_match);

static struct platform_driver my_platform_driver = {
   .probe = my_driver_probe,
   .remove = my_driver_remove,
   .driver = {
       .name = "my_device",
       .of_match_table = of_match_ptr(my_driver_of_match),
   },
};

module_platform_driver(my_platform_driver);

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

代码说明：

结构体定义：首先定义了一个 my_device 结构体来保存设备信息，包括指向设备的指针和分配的 DMA 缓冲区及其句柄。
Probe 函数：在 my_driver_probe 中，调用 dma_alloc_coherent 分配可用于 DMA 的缓冲区。如果成功，会输出缓冲区地址和句柄。
Remove 函数：在 my_driver_remove 中，通过调用 dma_free_coherent 来释放之前分配的缓冲区。
驱动注册：使用 platform_driver 将探测和移除功能与平台设备关联起来。