linux led驱动 arm

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/fs.h>
#include <linux/cdev.h>
#include <linux/device.h>
#include <linux/uaccess.h>
#include <linux/gpio.h>

#define DEVICE_NAME "myled"
#define CLASS_NAME "led_class"

static int major_number;
static struct class *led_class;
static struct device *led_device;
static int led_gpio = 21; // 假设使用GPIO21控制LED

static int led_open(struct inode *inode, struct file *file) {
    gpio_request(led_gpio, "myled");
    gpio_direction_output(led_gpio, 0);
    return 0;
}

static ssize_t led_write(struct file *file, const char __user *ubuf, size_t count, loff_t *ppos) {
    char buf[1];
    copy_from_user(buf, ubuf, count);
    if (buf[0] == '1')
        gpio_set_value(led_gpio, 1);
    else if (buf[0] == '0')
        gpio_set_value(led_gpio, 0);
    return count;
}

static int led_release(struct inode *inode, struct file *file) {
    gpio_free(led_gpio);
    return 0;
}

static struct file_operations fops = {
    .open = led_open,
    .write = led_write,
    .release = led_release,
};

static int __init led_init(void) {
    major_number = register_chrdev(0, DEVICE_NAME, &fops);
    if (major_number < 0) {
        printk(KERN_ALERT "Failed to register a major number\n");
        return major_number;
    }
    led_class = class_create(THIS_MODULE, CLASS_NAME);
    if (IS_ERR(led_class)) {
        unregister_chrdev(major_number, DEVICE_NAME);
        printk(KERN_ALERT "Failed to register device class\n");
        return PTR_ERR(led_class);
    }
    led_device = device_create(led_class, NULL, MKDEV(major_number, 0), NULL, DEVICE_NAME);
    if (IS_ERR(led_device)) {
        class_destroy(led_class);
        unregister_chrdev(major_number, DEVICE_NAME);
        printk(KERN_ALERT "Failed to create the device\n");
        return PTR_ERR(led_device);
    }
    printk(KERN_INFO "LED driver initialized successfully\n");
    return 0;
}

static void __exit led_exit(void) {
    device_destroy(led_class, MKDEV(major_number, 0));
    class_unregister(led_class);
    class_destroy(led_class);
    unregister_chrdev(major_number, DEVICE_NAME);
    printk(KERN_INFO "LED driver removed successfully\n");
}

module_init(led_init);
module_exit(led_exit);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Your Name");
MODULE_DESCRIPTION("A simple LED driver for ARM");