Linux Kernel 学习笔记8：同步与互斥之信号量

(本章基于：Linux-4.4.0-37)

内核中信号量的概念与应用层一致，本质是一个整数值，表示当前资源数量。可对这个值进行PV操作。P操作表示如果此信号量大于0，则将信号量减一，进程继续，反之如果信号量小于等于0则等待，等到信号量变为正值，即其他进程释放资源。V操作释放资源，即将信号量值加一，并在必要的时候唤醒正在等待的进程。

当信号量资源数量为1时，其实质就是一个互斥体，这与之前提到的自旋锁非常相似。区别是信号量在无法获取资源时会睡眠等待，释放CPU，而自旋锁不会。因此自旋锁只适用于多核处理器或抢占式内核系统，并且在临界区中不能有可能导致睡眠的操作。在其他情况下，信号量是最佳选择。

信号量数据结构  linux/semaphore.h

struct semaphore {

    raw_spinlock_t        lock;

    unsigned int        count;

    struct list_head    wait_list;

};

static inline void sema_init(struct semaphore *sem, int val);

初始化一个信号量，val表示初始资源个数

在linux内核中P操作被称为down，内核中有多种down操作

extern void down(struct semaphore *sem);

不可中断down，谨慎使用，如果获取不到资源进程将无法中断，无法被kill

extern int __must_check down_interruptible(struct semaphore *sem);


可中断down，最常用

extern int __must_check down_killable(struct semaphore *sem);

可被致命信号唤醒

extern int __must_check down_trylock(struct semaphore *sem);

不睡眠，返回0表示获取到信号量，返回1表示未获取

extern int __must_check down_timeout(struct semaphore *sem, long jiffies);
限制睡眠时间，返回0表示成功获取信号量

V被称之为up操作

extern void up(struct semaphore *sem);

例：

sema.c

#include <linux/init.h>
#include <linux/module.h>
#include <linux/stat.h>
#include <linux/kdev_t.h>
#include <linux/fs.h>
#include <linux/device.h>
#include <linux/cdev.h>
#include <asm/uaccess.h>
#include <linux/delay.h>
#include <linux/semaphore.h>

static dev_t devId;
static struct class *cls = NULL;
static struct cdev myDev;

static struct semaphore mySema;

static int
my_open(struct inode *inode, struct file *file)
{
int i;
while(down_interruptible(&mySema) != 0);
for(i=0; i<10; i++) {
printk(KERN_WARNING "%d\n", i);
ssleep(1);
}
up(&mySema);
printk(KERN_WARNING "open success!\n");
return 0;
}

static int
my_release(struct inode *inode, struct file *file)
{
printk(KERN_WARNING "close success!\n");
return 0;
}

//定义文件操作
static struct file_operations myFops = {
.owner = THIS_MODULE,
.open = my_open,
.release = my_release,
};

static void
hello_cleanup(void)
{
cdev_del(&myDev);
device_destroy(cls, devId);
class_destroy(cls);
unregister_chrdev_region(devId, 1);
}

static __init int hello_init(void)
{
int result;

sema_init(&mySema, 1);

//动态注册设备号
if(( result = alloc_chrdev_region(&devId, 0, 1, "stone-alloc-dev") ) != 0) {
printk(KERN_WARNING "register dev id error:%d\n", result);
goto err;
} else {
printk(KERN_WARNING "register dev id success!\n");
}
//动态创建设备节点
cls = class_create(THIS_MODULE, "stone-class");
if(IS_ERR(cls)) {
printk(KERN_WARNING "create class error!\n");
goto err;
}
if(device_create(cls, NULL, devId, "", "hello%d", 0) == NULL) {
printk(KERN_WARNING "create device error!\n");
goto err;
}
//字符设备注册
myDev.owner = THIS_MODULE;      //必要的成员初始化
myDev.ops = &myFops;
cdev_init(&myDev, &myFops);
//添加一个设备
result = cdev_add(&myDev, devId, 1);
if(result != 0) {
printk(KERN_WARNING "add cdev error!\n");
goto err;
}

printk(KERN_ALERT "hello init success!\n");
return 0;
err:
hello_cleanup();
return -1;
}

static __exit void hello_exit(void)
{
hello_cleanup();
printk(KERN_WARNING "helloworld exit!\n");
}

module_init(hello_init);
module_exit(hello_exit);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Stone");

应用层：

a.c

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>

#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

int
main(void)
{
int fd;
char buf[100];
int size;

fd = open("/dev/hello0", O_RDWR);
if(!fd) {
perror("open");
exit(-1);
}
printf("open success!\n");

close(fd);
return 0;
}