【C语言】Infiniband驱动mlx4_init_one

2024-03-27 19:04
文章标签 语言 init 驱动 one infiniband mlx4

本文主要是介绍【C语言】Infiniband驱动mlx4_init_one，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

一、注释

这是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_Hstruct devlink *devlink;
#endifstruct 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);
#elsepriv = kzalloc(sizeof(*priv), GFP_KERNEL);  // 分配内存if (!priv)return -ENOMEM;  // 如果内存分配失败，则返回错误
#endifdev       = &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的设置和注册...
#endifret =  __mlx4_init_one(pdev, id->driver_data, priv);  // 调用内部初始化函数
#ifdef HAVE_DEVLINK_H// 处理可能的devlink初始化错误，并注册devlink参数...
#endifif (ret) {// 如果有错误发生，进行清理工作...} else {pci_save_state(pdev);  // 保存PCI设备状态}#ifdef HAVE_DEVLINK_H// 如果支持devlink，处理错误时的清理工作...
#endifreturn 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_Hstruct devlink *devlink;
#endifstruct 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_PARAMSdevlink = devlink_alloc(&mlx4_devlink_ops, sizeof(*priv), &pdev->dev);
#elsedevlink = devlink_alloc(&mlx4_devlink_ops, sizeof(*priv));
#endifif (!devlink)return -ENOMEM;   // 内存分配失败时返回错误priv = devlink_priv(devlink); // 从devlink得到私有结构体的引用
#elsepriv = kzalloc(sizeof(*priv), GFP_KERNEL); // 在没有devlink的情况下分配内存if (!priv)return -ENOMEM;   // 内存分配失败时返回错误
#endifdev = &priv->dev;dev->persist = kzalloc(sizeof(*dev->persist), GFP_KERNEL); // 分配持久化设备结构if (!dev->persist) {
#ifdef HAVE_DEVLINK_Hret = -ENOMEM;goto err_devlink_free;   // 在devlink设置中，如果内存分配失败则跳转到相应清理环节
#elsekfree(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;    // 设置设备特定的数据// 初始化多个mutexmutex_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_PARAMSdevlink_register(devlink);
#elseret = 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
#endifret =  __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_PUBLISHEDdevlink_params_publish(devlink);
#endif
#ifdef HAVE_DEVLINK_RELOAD_ENABLE// 如果支持devlink重载，使能devlink的重载操作devlink_reload_enable(devlink);
#endif
#endif
#endifpci_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:
#endifkfree(dev->persist);  // 释放持久化设备结构内存// 如果devlink分配失败，则执行以下代码来释放内存
err_devlink_free:devlink_free(devlink);  // 释放devlink结构体所占用的内存
#endifreturn 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设备驱动初始化流程的一部分，包括资源的分配和初始化，以及设备在系统内的注册。如果在初始化过程中遇到任何错误，它会回滚已经执行的步骤，释放资源，确保系统稳定。

这篇关于【C语言】Infiniband驱动mlx4_init_one的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！


http://www.chinasem.cn/article/853079。

相关文章

科研绘图系列：R语言扩展物种堆积图（Extended Stacked Barplot）

科研绘图系列：R语言扩展物种堆积图（Extended Stacked Barplot）

介绍 R语言的扩展物种堆积图是一种数据可视化工具，它不仅展示了物种的堆积结果，还整合了不同样本分组之间的差异性分析结果。这种图形表示方法能够直观地比较不同物种在各个分组中的显著性差异，为研究者提供了一种有效的数据解读方式。 加载R包 knitr::opts_chunk$set(warning = F, message = F)library(tidyverse)library(phyl
阅读更多...
透彻！驯服大型语言模型（LLMs）的五种方法，及具体方法选择思路

透彻！驯服大型语言模型（LLMs）的五种方法，及具体方法选择思路

引言 随着时间的发展，大型语言模型不再停留在演示阶段而是逐步面向生产系统的应用，随着人们期望的不断增加，目标也发生了巨大的变化。在短短的几个月的时间里，人们对大模型的认识已经从对其zero-shot能力感到惊讶，转变为考虑改进模型质量、提高模型可用性。 「大语言模型(LLMs)其实就是利用高容量的模型架构（例如Transformer）对海量的、多种多样的数据分布进行建模得到，它包含了大量的先验
阅读更多...
Linux_kernel驱动开发11

Linux_kernel驱动开发11

一、改回nfs方式挂载根文件系统         在产品将要上线之前，需要制作不同类型格式的根文件系统         在产品研发阶段，我们还是需要使用nfs的方式挂载根文件系统         优点：可以直接在上位机中修改文件系统内容，延长EMMC的寿命         【1】重启上位机nfs服务         sudo service nfs-kernel-server resta
阅读更多...
C语言 | Leetcode C语言题解之第393题UTF-8编码验证

C语言 | Leetcode C语言题解之第393题UTF-8编码验证

题目： 题解： static const int MASK1 = 1 << 7;static const int MASK2 = (1 << 7) + (1 << 6);bool isValid(int num) {return (num & MASK2) == MASK1;}int getBytes(int num) {if ((num & MASK1) == 0) {return
阅读更多...
MiniGPT-3D, 首个高效的3D点云大语言模型，仅需一张RTX3090显卡，训练一天时间，已开源

MiniGPT-3D, 首个高效的3D点云大语言模型，仅需一张RTX3090显卡，训练一天时间，已开源

项目主页：https://tangyuan96.github.io/minigpt_3d_project_page/ 代码：https://github.com/TangYuan96/MiniGPT-3D 论文：https://arxiv.org/pdf/2405.01413 MiniGPT-3D在多个任务上取得了SoTA，被ACM MM2024接收，只拥有47.8M的可训练参数，在一张RTX
阅读更多...
如何确定 Go 语言中 HTTP 连接池的最佳参数？

如何确定 Go 语言中 HTTP 连接池的最佳参数？

确定 Go 语言中 HTTP 连接池的最佳参数可以通过以下几种方式： 一、分析应用场景和需求 并发请求量： 确定应用程序在特定时间段内可能同时发起的 HTTP 请求数量。如果并发请求量很高，需要设置较大的连接池参数以满足需求。例如，对于一个高并发的 Web 服务，可能同时有数百个请求在处理，此时需要较大的连接池大小。可以通过压力测试工具模拟高并发场景，观察系统在不同并发请求下的性能表现，从而
阅读更多...
C语言：柔性数组

C语言：柔性数组

数组定义 柔性数组 err int arr[0] = {0}; // ERROR 柔性数组 // 常见struct Test{int len;char arr[1024];} // 柔性数组struct Test{int len;char arr[0];}struct Test *t；t = malloc(sizeof(Test) + 11);strcpy(t->arr,
阅读更多...
C语言指针入门 《C语言非常道》

C语言指针入门 《C语言非常道》

C语言指针入门 《C语言非常道》 作为一个程序员，我接触 C 语言有十年了。有的朋友让我推荐 C 语言的参考书，我不敢乱推荐，尤其是国内作者写的书，往往七拼八凑，漏洞百出。 但是，李忠老师的《C语言非常道》值得一读。对了，李老师有个官网，网址是： 李忠老师官网 最棒的是，有配套的教学视频，可以试看。 试看点这里 接下来言归正传，讲解指针。以下内容很多都参考了李忠老师的《C语言非
阅读更多...
C 语言基础之数组

C 语言基础之数组

文章目录 什么是数组数组变量的声明多维数组 什么是数组 数组，顾名思义，就是一组数。 假如班上有 30 个同学，让你编程统计每个人的分数，求最高分、最低分、平均分等。如果不知道数组，你只能这样写代码： int ZhangSan_score = 95;int LiSi_score = 90;......int LiuDong_score = 100;int Zhou
阅读更多...
C 语言的基本数据类型

C 语言的基本数据类型

C 语言的基本数据类型 注：本文面向 C 语言初学者，如果你是熟手，那就不用看了。 有人问我，char、short、int、long、float、double 等这些关键字到底是什么意思，如果说他们是数据类型的话，那么为啥有这么多数据类型呢？ 如果写了一句： int a; 那么执行的时候在内存中会有什么变化呢？ 橡皮泥大家都玩过吧，一般你买橡皮泥的时候，店家会赠送一些模板。 上
阅读更多...

Copyright China编程 23002807@qq.com

沪ICP备2023033072号-2