【C语言】Infiniband驱动mlx4_init_one

173 篇文章 1 订阅
75 篇文章 1 订阅

一、注释

这是Linux内核中Mellanox Ethernet网卡驱动程序mlx4模块的一部分代码,主要用于初始化一个PCI设备。以下是其注释:

// 驱动的主结构体,包含了供PCI核心使用的勾子(hooks)
static struct pci_driver mlx4_driver = {
    .name        = DRV_NAME,                 // 驱动名称
    .id_table    = mlx4_pci_table,          // 支持的PCI设备ID表
    .probe        = mlx4_init_one,           // 初始化一个PCI设备时调用的函数
    .shutdown    = mlx4_shutdown,           // 关闭设备时调用的函数
    .remove        = mlx4_remove_one,         // 移除设备时调用的函数
    .suspend    = mlx4_suspend,            // 暂停设备时调用的函数
    .resume        = mlx4_resume,             // 恢复设备时调用的函数
    .err_handler    = &mlx4_err_handler,      // 错误处理程序(一些PCI错误处理函数)
};

// 初始化PCI设备的函数
static int mlx4_init_one(struct pci_dev *pdev, const struct pci_device_id *id)
{
    // 下面的宏定义是用来处理不同内核版本之间的兼容性问题

#ifdef HAVE_DEVLINK_H
    struct devlink *devlink;
#endif
    struct mlx4_priv *priv;
    struct mlx4_dev *dev;
    int ret;

    printk_once(KERN_INFO "%s", mlx4_version);  // 打印驱动版本信息

#ifdef HAVE_DEVLINK_H
    // devlink相关的初始化。如果支持devlink,则使用devlink相应的API分配内存以及初始化
    ...
    priv = devlink_priv(devlink);
#else
    priv = kzalloc(sizeof(*priv), GFP_KERNEL);  // 分配内存
    if (!priv)
        return -ENOMEM;  // 如果内存分配失败,则返回错误
#endif
    dev       = &priv->dev;
    dev->persist = kzalloc(sizeof(*dev->persist), GFP_KERNEL);  // 分配持久化的设备结构
    if (!dev->persist) {
        // 如果分配失败,执行相应的清理工作然后返回错误
        ...
    }
    dev->persist->pdev = pdev;
    dev->persist->dev = dev;
    pci_set_drvdata(pdev, dev->persist);  // 设置PCI设备的驱动程序数据

    priv->pci_dev_data = id->driver_data;  // 设置设备私有数据
    // 初始化互斥锁
    mutex_init(&dev->persist->device_state_mutex);
    mutex_init(&dev->persist->interface_state_mutex);
    mutex_init(&dev->persist->pci_status_mutex);

#ifdef HAVE_DEVLINK_H
    // 如果支持devlink,则继续进行devlink的设置和注册
    ...
#endif

    ret =  __mlx4_init_one(pdev, id->driver_data, priv);  // 调用内部初始化函数
#ifdef HAVE_DEVLINK_H
    // 处理可能的devlink初始化错误,并注册devlink参数
    ...
#endif

    if (ret) {
        // 如果有错误发生,进行清理工作
        ...
    } else {
        pci_save_state(pdev);  // 保存PCI设备状态
    }

#ifdef HAVE_DEVLINK_H
    // 如果支持devlink,处理错误时的清理工作
    ...
#endif
    return ret;  // 返回初始化结果,成功或失败的错误码
}

以上代码是在内核代码中的一个PCI驱动程序的初始化函数。在该函数中,驱动程序初始化PCI设备并为其分配必要的资源(比如内存)。如果启用了`devlink`配置选项(用于一些设备的配置和监控功能),它会额外处理`devlink`相关的注册和初始化。该函数最后将返回一个错误码,表示初始化操作的成功或失败状态。

完整代码的中文注释:

// 驱动的主结构体,包含了供PCI核心使用的勾子(hooks)
static struct pci_driver mlx4_driver = {
    .name        = DRV_NAME,                   // 驱动名称
    .id_table    = mlx4_pci_table,             // 支持的PCI设备ID表
    .probe        = mlx4_init_one,              // 初始化一个PCI设备时调用的函数
    .shutdown    = mlx4_shutdown,              // 关闭设备时调用的函数
    .remove        = mlx4_remove_one,            // 移除设备时调用的函数
    .suspend    = mlx4_suspend,               // 暂停设备时调用的函数
    .resume        = mlx4_resume,                // 恢复设备时调用的函数
    .err_handler    = &mlx4_err_handler,         // 错误处理程序(一些PCI错误处理函数)
};
// 初始化一个PCI设备的函数
static int mlx4_init_one(struct pci_dev *pdev, const struct pci_device_id *id)
{
#ifdef HAVE_DEVLINK_H
    struct devlink *devlink;
#endif
    struct mlx4_priv *priv;
    struct mlx4_dev *dev;
    int ret;
    // 仅打印一次驱动版本信息
    printk_once(KERN_INFO "%s", mlx4_version);
#ifdef HAVE_DEVLINK_H
    // 根据编译时的设置,以不同的方式分配并初始化devlink
#ifdef HAVE_DEVLINK_ALLOC_GET_3_PARAMS
    devlink = devlink_alloc(&mlx4_devlink_ops, sizeof(*priv), &pdev->dev);
#else
    devlink = devlink_alloc(&mlx4_devlink_ops, sizeof(*priv));
#endif
    if (!devlink)
        return -ENOMEM;   // 内存分配失败时返回错误
    priv = devlink_priv(devlink); // 从devlink得到私有结构体的引用
#else
    priv = kzalloc(sizeof(*priv), GFP_KERNEL); // 在没有devlink的情况下分配内存
    if (!priv)
        return -ENOMEM;   // 内存分配失败时返回错误
#endif
    dev = &priv->dev;
    dev->persist = kzalloc(sizeof(*dev->persist), GFP_KERNEL); // 分配持久化设备结构
    if (!dev->persist) {
#ifdef HAVE_DEVLINK_H
        ret = -ENOMEM;
        goto err_devlink_free;   // 在devlink设置中,如果内存分配失败则跳转到相应清理环节
#else
        kfree(priv);            // 在非devlink设置中清理分配的内存
        return -ENOMEM;
#endif
    }
    // 初始化pdev和dev之间的引用
    dev->persist->pdev = pdev;
    dev->persist->dev = dev;
    pci_set_drvdata(pdev, dev->persist);     // 将持久化的dev结构与pci设备关联起来
    priv->pci_dev_data = id->driver_data;    // 设置设备特定的数据
    // 初始化多个mutex
    mutex_init(&dev->persist->device_state_mutex);
    mutex_init(&dev->persist->interface_state_mutex);
    mutex_init(&dev->persist->pci_status_mutex);
#ifdef HAVE_DEVLINK_H
    // 如果支持devlink,根据编译时的配置注册devlink实例
#ifdef HAVE_DEVLINK_REGISTER_GET_1_PARAMS
    devlink_register(devlink);
#else
    ret = devlink_register(devlink, &pdev->dev);
    if (ret)
        goto err_persist_free;  // 注册失败时清理并返回
#endif
#if defined(HAVE_DEVLINK_PARAM) && defined(HAVE_DEVLINK_PARAMS_PUBLISHED)
    // 如果支持devlink参数,注册设备的参数
    ret = devlink_params_register(devlink, mlx4_devlink_params,
                      ARRAY_SIZE(mlx4_devlink_params));
    if (ret)
        goto err_devlink_unregister;  // 注册失败时清理并返回
    // 设置初始化时的参数值
    mlx4_devlink_set_params_init_values(devlink);
#endif
#endif
    ret =  __mlx4_init_one(pdev, id->driver_data, priv);  // 调用内部的初始化函数
    if (ret) {
        goto err_handling;  // 如果初始化失败,处理错误
    }
#ifdef HAVE_DEVLINK_H
#if defined(HAVE_DEVLINK_PARAM) && defined(HAVE_DEVLINK_PARAMS_PUBLISHED)
    // 如果支持devlink参数并且已经发布,发布参数设置
#ifdef HAVE_DEVLINK_PARAMS_PUBLISHED
    devlink_params_publish(devlink);
#endif
#ifdef HAVE_DEVLINK_RELOAD_ENABLE
    // 如果支持devlink重载,使能devlink的重载操作
    devlink_reload_enable(devlink);
#endif
#endif
#endif
    pci_save_state(pdev);   // 保存当前的PCI设备状态,这样在需要恢复设备时(比如从挂起状态恢复时)可以用到
    return 0;               // 返回0表示成功
#ifdef HAVE_DEVLINK_H
#if defined(HAVE_DEVLINK_PARAM) && defined(HAVE_DEVLINK_PARAMS_PUBLISHED)
    // 如果支持devlink参数和发布功能,以下是失败时的清理过程
err_params_unregister:
    // 取消注册之前注册的devlink参数
    devlink_params_unregister(devlink, mlx4_devlink_params,
                  ARRAY_SIZE(mlx4_devlink_params));
#endif
    // 错误处理部分:如果devlink注册失败,则取消注册
err_devlink_unregister:
    devlink_unregister(devlink);
#ifndef HAVE_DEVLINK_REGISTER_GET_1_PARAMS
    // 如果出现错误且版本不支持单参数devlink注册,则需要释放持久化设备结构内存
err_persist_free:
#endif
    kfree(dev->persist);  // 释放持久化设备结构内存
    // 如果devlink分配失败,则执行以下代码来释放内存
err_devlink_free:
    devlink_free(devlink);  // 释放devlink结构体所占用的内存
#endif
    return ret;  // 返回错误码。如果初始化过程中的某个步骤失败,则返回相应的错误码
}

上面的 pci_save_state(pdev) 函数调用负责保存当前PCI设备的核心配置寄存器。这在系统休眠或PCI设备电源管理事件中是必要的,因为在这样的事件中,设备的PCI配置可能会丢失或被重置,系统在恢复时需要这信息重新对设备进行配置。

return 0; 表明函数执行成功,返回值0符合Linux内核中处理成功操作的返回值约定。在C语言中,返回状态通常使用整形返回,而0通常表示成功,非0值表示某种形式的错误。

备注:此代码段涉及到了错误处理和内存清理部分。在初始化一个PCI设备时,如果过程中遇到错误,代码会跳转到相应的错误处理标签进行清理。这些清理工作可能包括注销注册的devlink参数、释放分配的内存、取消注册devlink设备等,以确保在发生错误时不会泄露资源。一旦完成清理,函数将返回一个表示错误状态的负值错误码。如果初始化成功,代码则将保存PCI设备的状态,并且返回0表示初始化成功。

二、讲解

这段代码主体是一个`mlx4_init_one`函数,它是Mellanox公司网络设备驱动的初始化函数,用于对PCI网络设备进行初始化。这段代码来自Linux内核的Mellanox网络驱动,其中的`mlx4_driver`结构体是PCI驱动的注册信息。`mlx4_init_one`函数是在PCI设备开始初始化时由内核调用的。函数主体涉及资源分配、设备注册和错误处理。
代码以及注释的大致意思如下:
- mlx4_driver:一个PCI驱动结构体,提供驱动的基本信息和回调函数。
- mlx4_init_one:驱动的初始化函数,它会针对每个被发现的PCI设备被调用。
  - 输出一次版本信息到内核日志。
  - 分配`devlink`资源,作为网络设备到devlink层的接口。
  - 分配`mlx4_priv`结构体以存储私有数据。
  - 分配`mlx4_dev`结构体中的`persist`字段,存储设备持续性状态。
  - 设置PCI设备与驱动数据的关联。
  - 初始化一些互斥锁,保护设备状态和PCI状态。
  - 在devlink层注册该设备,以及设备参数。
  - 调用`__mlx4_init_one`进行进一步的设备初始化工作。
  - 最后,保存PCI设备的状态以便恢复时使用。
- 如果初始化失败,函数会逐步释放已分配的资源,并且返回错误代码。
这段代码展示了复杂PCI设备驱动初始化流程的一部分,包括资源的分配和初始化,以及设备在系统内的注册。如果在初始化过程中遇到任何错误,它会回滚已经执行的步骤,释放资源,确保系统稳定。

  • 12
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

109702008

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值