linux驅動——原子操作
原子操作 API
Linux 內核定義了叫做 atomic_t 的結構體來完成整型數據的原操作，在使用是使用原子變量來代替整型變量。此結構體定義在 include/linux/types.h 文件中，定義如下：
typedef struct {int counter;
}atomic_t;
如果要使用原子操作 API 函數，需要首先定義一個 atomic_t 的變量，如下所示：
atomic_t value; /* 定義一個原子變量 value */
也可以在定義原子變量的時候給原子變量賦初值，如下所示
atomic_t value = ATOMIC_INIT(0) /* 定義原子變量 value 并賦初值為 0*/
Linux 內核提供的原子變量操作的 API 如下：
| 函數 | 描述 |
|---|---|
| ATOMIC_INIT(int i) | 定義原子變量的時候初始化 |
| int atomic_read(atomic_t *v) | 讀取原子變量 v 的值，并且返回 |
| void atomic_set(atomic_t *v, int i) | 向 v 寫入 i 值 |
| void atomic_add(int i, atomic_t *v) | 給 v 加上 i 值 |
| void atomic_sub(int i, atomic_t *v) | 給 v 減去 i 值 |
| void atomic_inc(atomic_t *v) | 自增 |
| void atomic_dec(atomic_t *v) | 自減 |
| int atomic_dec_return(atomic_t *v) | 自減，并返回 v 的值 |
| int atomic_inc_return(atomic_t *v) | 自增，并返回 v 的值 |
| int atomic_sub_and_test(int i, atomic_t *v) | 從 v 減 i，如果結果為 0 就返回真，否則返回假 |
| int atomic_dec_and_test(atomic_t *v) | 從 v 減 1，如果結果為 0 就返回真，否則返回假 |
| int atomic_inc_and_test(atomic_t *v) | 給 v 加 1，如果結果為 0 就返回真，否則返回假 |
| int atomic_inc_and_test(atomic_t *v) | 給 v 加 1，如果結果為 0 就返回真，否則返回假 |
| int atomic_add_negative(int i, atomic_t *v) | 給 v 加 i，如果結果為負就返回真，否則返回假 |
如果使用 64 位的 SOC 的話，就要用到 64 位的原子變量，Linux 內核也定義了 64 位原子結構體，如下所示：
#ifdef CONFIG_64BIT
typedef struct {s64 counter;
} atomic64_t;
#endif
typedef __s64 s64;
__extension__ typedef __signed__ long long __s64;
相應的也提供了 64 位原子變量的操作 API 函數，和上表用法一樣，只是將 atomic_ 前綴換為 atomic64_ 將 int 換為 long long。如果使用的是 64 位的 SOC，那么就要使用 64 位的原子操作函數。
原子變量使用示例如下：
atomic_t v = ATOMIC_INIT(0); /* 定義并初始化原子變零 v=0 */
atomic_set(&v, 10); /* 設置 v=10 */
atomic_read(&v); /* 讀取 v 的值，肯定是 10 */
atomic_inc(&v); /* v 的值加 1，v=11 */
原子位操作 API
位操作也是很常用的操作，Linux 內核也提供了一系列的原子位操作 API 函數，只不過原子位操作不像原子整形變量那樣有個 atomic_t 的數據結構，原子位操作是直接對內存進行操作，API 函數下所示：
| 函數 | 描述 |
|---|---|
| void set_bit(int nr, void *p) | 將 p 地址的第 nr 位置 1 |
| void clear_bit(int nr,void *p) | 將 p 地址的第 nr 位清零 |
| void change_bit(int nr, void *p) | 將 p 地址的第 nr 位進行翻轉 |
| int test_bit(int nr, void *p) | 獲取 p 地址的第 nr 位的值 |
| int test_and_set_bit(int nr, void *p) | 將 p 地址的第 nr 位置 1，并且返回 nr 位原來的值 |
| int test_and_clear_bit(int nr, void *p) | 將 p 地址的第 nr 位清零，并且返回 nr 位原來的值 |
| int test_and_change_bit(int nr, void *p) | 將 p 地址的第 nr 位翻轉，并且返回 nr 位原來的值 |
原子操作驅動
#include "linux/device/class.h"
#include "linux/export.h"
#include "linux/uaccess.h"
#include <linux/types.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/module.h>
#include <linux/fs.h>
#include <linux/device.h>
#include <linux/cdev.h>#define CHRDEVBASE_NAME "chrdev_atom" /* 設備名 */
#define CHRDEVBASE_NUM 1 /* 設備數目 */static char write_buf[100];
static char read_buf[100];static char *string_test = "kernel data this tyustli test";typedef struct {dev_t dev_id; /* 設備號 */struct cdev c_dev; /* cdev */struct class *class; /* 類 */struct device *device; /* 設備 */int major; /* 主設備號 */int minor; /* 次設備號 */atomic_t lock; /* 原子鎖 */
} new_chrdev_t;new_chrdev_t new_chrdev;static int chrdevbase_open(struct inode *inode, struct file *file)
{/* 通過判斷原子變量的值來檢查當前驅動有沒有被別的應用使用 */if (!atomic_dec_and_test(&new_chrdev.lock)) {atomic_inc(&new_chrdev.lock); /* 小于 0 的話就加 1,使其原子變量等于 0 */return -EBUSY; /* 驅動被使用，返回忙 */}printk("k: chrdevbase open\r\n");return 0;
}static ssize_t chrdevbase_read(struct file *file, char __user *buf,size_t count, loff_t *ppos)
{unsigned long ret = 0;printk("k: chrdevbase read\r\n");memcpy(read_buf, string_test, strlen(string_test));ret = copy_to_user(buf, read_buf, count);if (ret == 0) {printk("k: read data success\r\n");} else {printk("k: read data failed ret = %ld\r\n", ret);}return ret;
}static ssize_t chrdevbase_write(struct file *file, const char __user *buf,size_t count, loff_t *ppos)
{unsigned long ret = 0;printk("k: chrdevbase write\r\n");ret = copy_from_user(write_buf, buf, count);if (ret == 0) {printk("k: write data success write data is: %s\r\n", write_buf);} else {printk("k: write data failed ret = %ld\r\n", ret);}return count;
}static int chrdevbase_release(struct inode *inode, struct file *file)
{/* 關閉驅動文件的時候釋放原子變量 */atomic_inc(&new_chrdev.lock);printk("k: chrdevbase release\r\n");return 0;
}static struct file_operations chrdevbase_fops = {.owner = THIS_MODULE,.open = chrdevbase_open,.read = chrdevbase_read,.write = chrdevbase_write,.release = chrdevbase_release,
};static int __init chrdevbase_init(void)
{int err = 0;atomic_set(&new_chrdev.lock, 1); /* 原子變量初始值為1 */err = alloc_chrdev_region(&new_chrdev.dev_id, 0, CHRDEVBASE_NUM,CHRDEVBASE_NAME);if (err < 0) {printk("k: alloc chrdev region failed err = %d\r\n", err);return -1;}/* get major and minor */new_chrdev.major = MAJOR(new_chrdev.dev_id);new_chrdev.minor = MINOR(new_chrdev.dev_id);printk("k: newcheled major=%d,minor=%d\r\n", new_chrdev.major,new_chrdev.minor);new_chrdev.c_dev.owner = THIS_MODULE;cdev_init(&new_chrdev.c_dev, &chrdevbase_fops);err = cdev_add(&new_chrdev.c_dev, new_chrdev.dev_id, CHRDEVBASE_NUM);if (err < 0) {printk("k: cdev add failed err = %d\r\n", err);goto out;}new_chrdev.class = class_create(CHRDEVBASE_NAME);if (IS_ERR(new_chrdev.class)) {printk("k: class create failed\r\n");goto out_cdev;}new_chrdev.device = device_create(new_chrdev.class, NULL, new_chrdev.dev_id,NULL, CHRDEVBASE_NAME);if (IS_ERR(new_chrdev.device)) {printk("k: device create failed\r\n");goto out_class;}printk("k: base module init\r\n");return 0;out_class:class_destroy(new_chrdev.class);
out_cdev:cdev_del(&new_chrdev.c_dev);
out:unregister_chrdev_region(new_chrdev.dev_id, CHRDEVBASE_NUM);return err;
}static void __exit chrdevbase_exit(void)
{device_destroy(new_chrdev.class, new_chrdev.dev_id);class_destroy(new_chrdev.class);cdev_del(&new_chrdev.c_dev);unregister_chrdev_region(new_chrdev.dev_id, CHRDEVBASE_NUM);printk("k: base module exit!\r\n");
}module_init(chrdevbase_init);
module_exit(chrdevbase_exit);MODULE_LICENSE("GPL");
MODULE_AUTHOR("tyustli");
MODULE_INFO(intree, "Y"); /* loading out-of-tree module taints kernel */
原子操作 APP
#include "stdio.h"
#include "unistd.h"
#include "sys/types.h"
#include "sys/stat.h"
#include "fcntl.h"
#include "stdlib.h"
#include "string.h"static char usrdata[] = { "user data!" };int main(int argc, char *argv[])
{int fd, retvalue;char *filename;unsigned char run_cnt = 0;char readbuf[100], writebuf[100];if (argc != 3) {printf("u: error Usage!\r\n");return -1;}filename = argv[1];/* 打開驅動文件 */fd = open(filename, O_RDWR);if (fd < 0) {printf("u: can't open file %s\r\n", filename);return -1;}/* 從驅動文件讀取數據 */if (atoi(argv[2]) == 1) {retvalue = read(fd, readbuf, 50);if (retvalue < 0) {printf("u: read file %s failed!\r\n", filename);} else {/* 讀取成功，打印出讀取成功的數據 */printf("u: read data:%s\r\n", readbuf);}}/* 模擬占用驅動 25s，此時另一個線程去打開驅動 */while (1) {sleep(5);run_cnt++;if (run_cnt >= 5)break;}/* 向設備驅動寫數據 */if (atoi(argv[2]) == 2) {memcpy(writebuf, usrdata, sizeof(usrdata));retvalue = write(fd, writebuf, 50);if (retvalue < 0) {printf("u: write file %s failed!\r\n", filename);}}/* 關閉設備 */retvalue = close(fd);if (retvalue < 0) {printf("u: can't close file %s\r\n", filename);return -1;}return 0;
}
