原文链接：https://mp.weixin.qq.com/s/qqxDObNs8vjTFLQueX1J-A

哈喽，我是老吴。

非常怀念写文章的感觉。

昨晚复习了一些 Linux 驱动的基础知识，给大家分享一下吧。

先说结论：

多年来，我接触到的 Linux 驱动教程大多都是从 0 编写，这样对初学者而言最大的好处，就是可以接触到比较多的底层原理。

但是在真正的工作场景里，其实是应该尽量避免从 0 构建自己的设备驱动的。

更好的做法是在高度模块化的驱动框架里添加自己的设备驱动。

这样做的好处是最大程度地复用内核现有的代码，同时获得极大的弹性和可维护性，并且为应用程序提供出统一的访问接口。

下面详细地说说。

什么是设备驱动？

设备驱动程序 (device driver) 是对硬件的抽象:

提供基础框架来编写和运行设备驱动程序是操作系统内核责任的一部分。

尽管可以在用户空间中运行设备驱动程序（通过一些内核接口，如 UIO 或 I2CDEV），更常见的情况是让它们在内核空间中运行。

以字符设备驱动为例：

字符设备 (char device) 是一种最常见的硬件抽象。

/dev 目录下的设备节点文件就是内核导出给用户空间的访问设备驱动的接口。

设备节点文件中有三个基本信息：

• Type，用于标识是 block 还是 char device;

• Major number，用于标志是那一类 char deivce;

• Minor number，用于标志是哪一个 char device;

编写字符设备驱动的流程：

1、分配设备号，这通过 register_chrdev_region() 或 alloc_chrdev_region() 来完成;

2、实现文件操作（open、read、write、ioctl）等。

3、使用 cdev_init() 和 cdev_add() 向内核中注册字符设备。

以 LED 字符设备驱动为例

如果按照从 0 构建的思路编写驱动的话，伪代码如下：

硬件访问相关：

1.   static struct {
2.    dev_t devnum;
3.    struct cdev cdev;
4.    unsigned int led_status;
5.    void __iomem *regbase;
6.   } drvled_data;
7.    
8.   static void drvled_setled(unsigned int status)
9.   {
10.    u32 val;
11.    
12.    /* set value */
13.    val = readl(drvled_data.regbase + GPIO1_REG_DATA);
14.    if (status == LED_ON)
15.     val |= GPIO_BIT;
16.    else if (status == LED_OFF)
17.     val &= ~GPIO_BIT;
18.    writel(val, drvled_data.regbase + GPIO1_REG_DATA);
19.    
20.    /* update status */
21.    drvled_data.led_status = status;
22.   }
23.    
24.   static void drvled_setdirection(void)
25.   {
26.    ...
27.   }

文件操作相关：

1.   static ssize_t drvled_read(struct file *file, char __user *buf,
2.         size_t count, loff_t *ppos)
3.   {
4.    ...
5.   }
6.    
7.   static ssize_t drvled_write(struct file *file, const char __user *buf,
8.          size_t count, loff_t *ppos)
9.   {
10.    char kbuf = 0;
11.    
12.    if (copy_from_user(&kbuf, buf, 1))
13.     return -EFAULT;
14.    
15.    if (kbuf == '1') {
16.     drvled_setled(LED_ON);
17.     pr_info("LED ON!\n");
18.    } else if (kbuf == '0') {
19.     drvled_setled(LED_OFF);
20.     pr_info("LED OFF!\n");
21.    }
22.    
23.    return count;
24.   }
25.    
26.   static const struct file_operations drvled_fops = {
27.    .owner = THIS_MODULE,
28.    .write = drvled_write,
29.    .read = drvled_read,
30.   };

注册和卸载字符设备相关：

1.   static int __init drvled_init(void)
2.   {
3.    int result = 0;
4.    
5.    if (!request_mem_region(GPIO1_BASE, GPIO1_SIZE, DRIVER_NAME)) {
6.     pr_err("%s: Error requesting I/O!\n", DRIVER_NAME);
7.     result = -EBUSY;
8.     goto ret_err_request_mem_region;
9.    }
10.    
11.    drvled_data.regbase = ioremap(GPIO1_BASE, GPIO1_SIZE);
12.    if (!drvled_data.regbase) {
13.     pr_err("%s: Error mapping I/O!\n", DRIVER_NAME);
14.     result = -ENOMEM;
15.     goto err_ioremap;
16.    }
17.    
18.    result = alloc_chrdev_region(&drvled_data.devnum, 0, 1, DRIVER_NAME);
19.    if (result) {
20.     pr_err("%s: Failed to allocate device number!\n", DRIVER_NAME);
21.     goto ret_err_alloc_chrdev_region;
22.    }
23.    
24.    cdev_init(&drvled_data.cdev, &drvled_fops);
25.    
26.    result = cdev_add(&drvled_data.cdev, drvled_data.devnum, 1);
27.    if (result) {
28.     pr_err("%s: Char device registration failed!\n", DRIVER_NAME);
29.     goto ret_err_cdev_add;
30.    }
31.    
32.    drvled_setdirection();
33.    
34.    drvled_setled(LED_OFF);
35.    
36.    pr_info("%s: initialized.\n", DRIVER_NAME);
37.    goto ret_ok;
38.    
39.   ret_err_cdev_add:
40.    unregister_chrdev_region(drvled_data.devnum, 1);
41.   ret_err_alloc_chrdev_region:
42.    iounmap(drvled_data.regbase);
43.   err_ioremap:
44.    release_mem_region(GPIO1_BASE, GPIO1_SIZE);
45.   ret_err_request_mem_region:
46.   ret_ok:
47.    return result;
48.   }
49.    
50.   static void __exit drvled_exit(void)
51.   {
52.       ...
53.   }
54.    
55.   module_init(drvled_init);
56.   module_exit(drvled_exit);

运行效果：

1.   $ install ledrv.ko
2.   $ ls /dev/led
3.   # 灯亮
4.   $ echo 1 >/dev/led
5.   # 灯灭
6.   $ echo 1 >/dev/led

三个问题

单从功能的角度看，上面的程序完全满足控制一个 LED 的需求。

但是，它不是一个好的驱动，这里有 3 个问题。

问题 1：

它创建的接口是 /dev/led，这不是一个通用接口，会增加上层开发人员的学习成本。

解决这个问题需要在 LED char driver 上再添加一层 LED framework，LED framework 负责给用户空间提供标准化的访问接口，同时用于添加可复用的逻辑功能。

基本上各种设备驱动都有自己的 framework，例如 input, IIO, ALSA, V2L2, RTC, watchdog 等。

使用这些 framework 驱动工程师不用考虑提供给用户空间的接口，应用开发人员也只需要学习一次标准的硬件访问接口接口。

问题 2：

它只是控制 1 个 gpio，但是却申请使用了 2 个寄存器，这 2 个寄存器负责控制芯片的 8 个 gpio。这意味着其他 7 个 gpio 再也没法被其他驱动申请使用。

解决这个问题需要引入一个 gpio 的管理者：gpiolib。gpiolib 负责统一管理和分配 gpio 资源。

问题 3：

它包含了硬件信息。如果我们想控制另外一个 gpio或者多个 gpio，就得改动源码，代码维护的工作量极大。

解决这个问题我们需要将硬件信息从代码中抽取出来，具体的就是引入总线、设备、驱动模型。

更好的 LED 驱动

我们用上面的思路，写一个更合理的 LED 驱动。

引入 LED framework：

1、初始化 led_classdev 结构体。

2、提供一个回调函数来改变状态 LED 的。

3、使用 led_classdev_register() 在想 LED framework 注册驱动程序。

引入 gpiolib：

内核管理 gpio 的思路是典型的 producer/consumer 模型。

GPIO controller driver 是 producer，LED driver 是 consumer。

下面是几个常用的 gpiolib api，它们的作用一目了然:

1.   #include <linux/gpio.h>
2.   #include <linux/gpio/consumer.h>
3.   struct gpio_desc *gpiod_get(struct device *dev, const char *con_id,enum gpiod_flags flags);
4.    
5.   void gpiod_put(struct gpio_desc *desc);
6.    
7.   int gpiod_direction_input(struct gpio_desc *desc);
8.    
9.   int gpiod_direction_output(struct gpio_desc *desc, int value);
10.    
11.   void gpiod_set_value(struct gpio_desc *desc, int value);
12.    
13.   int gpiod_get_value(const struct gpio_desc *desc);

引入总线、设备、驱动模型 ：

该模型包含 4 部分。

Bus core： 对硬件总线的抽象，不同总线有不同的 Bus core，例如 USB core, SPI core, I2C core, PCI core ，在内核中由 bus_type 结构表示。

Bus adapters： 总线控制器驱动程序，在内核中由 device_driver 结构体表示。

Bus drivers： 负责管理连接到总线的设备的驱动程序，在内核中由 device_driver 结构体表示。

Bus devices： 连接到总线的设备，在内核中由结构 device 表示。

内核虚拟了一条叫 Platform 的总线，用于适配 LED 这种不属于任何总线的设备。

看下改造后的代码：

设备信息：

1.   LED {
2.   <&gpio1 9>
3.   }

硬件控制：

1.   tatic struct drvled_data_st *drvled_data;
2.    
3.   static void drvled_setled(unsigned int status)
4.   {
5.    // 控制 gpio
6.    if (status == LED_ON)
7.     gpiod_set_value(drvled_data->desc, 1);
8.    else
9.     gpiod_set_value(drvled_data->desc, 0);
10.   }
11.    
12.   static void drvled_change_state(struct led_classdev *led_cdev,
13.       enum led_brightness brightness)
14.   {
15.    if (brightness)
16.     drvled_setled(LED_ON);
17.    else
18.     drvled_setled(LED_OFF);
19.   }

向 LED framework 注册：

1.   static int drvled_probe(struct platform_device *pdev)
2.   {
3.    struct device_node *np = pdev->dev.of_node;
4.    struct device_node *child = NULL;
5.    int result, gpio;
6.    
7.    child = of_get_next_child(np, NULL);
8.    
9.    drvled_data = devm_kzalloc(&pdev->dev, sizeof(*drvled_data),
10.          GFP_KERNEL);
11.    if (!drvled_data)
12.     return -ENOMEM;
13.    
14.    // 从设备数中获得硬件信息
15.    gpio = of_get_gpio(child, 0);
16.    
17.    result = devm_gpio_request(&pdev->dev, gpio, pdev->name);
18.    if (result) {
19.     dev_err(&pdev->dev, "Error requesting GPIO\n");
20.     return result;
21.    }
22.    
23.    drvled_data->desc = gpio_to_desc(gpio);
24.    
25.    drvled_data->led_cdev.name = of_get_property(child, "label", NULL);
26.    drvled_data->led_cdev.brightness_set = drvled_change_state;
27.    
28.    // 注册进 LED framework
29.    result = devm_led_classdev_register(&pdev->dev, &drvled_data->led_cdev);
30.    if (result) {
31.     dev_err(&pdev->dev, "Error registering led\n");
32.     return result;
33.    }
34.    
35.    gpiod_direction_output(drvled_data->desc, 0);
36.    
37.    dev_info(&pdev->dev, "initialized.\n");
38.    
39.    return 0;
40.   }
41.    
42.   static int drvled_remove(struct platform_device *pdev)
43.   {
44.    dev_info(&pdev->dev, "exiting.\n");
45.    
46.    return 0;
47.   }
48.    
49.   static const struct of_device_id of_drvled_match[] = {
50.    { .compatible = "labworks,drvled" },
51.    {},
52.   };
53.    
54.   static struct platform_driver drvled_driver = {
55.    .driver = {
56.     .name = "drvleds",
57.     .owner = THIS_MODULE,
58.     .of_match_table = of_drvled_match,
59.    },
60.    .probe  = drvled_probe,
61.    .remove  = drvled_remove,
62.   };
63.    
64.   module_platform_driver(drvled_driver);

改造后，应用总是通过下面这种标准的接口访问 LED：

1.   # 灯亮
2.   $ echo 1 > /sys/class/leds/<LED driver name>/brightness
3.    
4.   # 灯灭
5.   $ echo 0 > /sys/class/leds/<LED driver name>/brightness

并且有大量的 trigger 可供使用，例如让 LED 呈心跳状态的 heartbeat trigger：

$ echo heartbeat > /sys/class/leds/<LED driver name>/trigger

假设这时你想改用 gpio expander 芯片来控制 LED：

只需要添加这个 gpio expander 的驱动代码，并且修改设备树即可，其他部分完全不需要改动：

1.   gpioexp {
2.   I2C0, 0x10
3.   }
4.   LED {
5.   <&gpioexp 3>
6.   }

到此，你是否更清楚如何为 Linux 添加设备驱动了呢？

—— The End ——