Linux内核之RCU锁

RCU全程read-only-update

RCU机制:RCU记录了所有指向共享数据的指针的使用者,当要修改共享数据的时候,首先创建一个副本,在副本中修改。所有读线程都离开读临界区之后,指针指向新的修改后副本的指针,并且删除旧数据。

例子:RCU相关API的使用

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/init.h>
#include <linux/slab.h>
#include <linux/spinlock.h>
#include <linux/rcupdate.h>
#include <linux/kthread.h>
#include <linux/delay.h>

struct foo
{
    int a;
    struct rcu_head rcu;
};

static struct foo* g_ptr;

//读者线程
static void myrcu_reader_thread(void* data)
{
    struct foo* p =NULL;

    while(1)
    {
        msleep(200);
        rcu_read_lock();
        /*
           用于获取被RCU保护的指针,读者线程要访问RCU保护的共享数据,
           需要使用该函数创建一个新的指针,并且指向被RCU保护的指针
        */
        p = rcu_dereference(g_ptr);
        if(p)
            printk("%s: read a=%d\n",__func__,p->a);
        rcu_read_unlock();
    }
}

static void myrcu_del(struct rcu_head* rh)
{
    struct foo* p = container_of(rh, struct foo, rcu);
    printk("%s: a=%d\n", __func__ ,p->a);
    kfree(p);
}

//写者线程
static void myrcu_writer_thread(void* p)
{
    struct foo* new;
    struct foo* old;

    int value = (unsigned long) p;

    while(1)
    {
        msleep(400);
        struct foo* new_ptr = kmalloc(sizeof(struct foo), GFP_KERNEL);
        old = g_ptr;
        printk("%s: write to new %d\n", __func__ ,value);
        *new_ptr = *old;
        new_ptr -> a = value;
        /*
           写者线程完成新数据的修改之后,调用该函数可以让被RCU保护的指针指向新创建的数据
           (publish 了更新后的数据)
        */
        rcu_assign_pointer(g_ptr,new_ptr);
        /*
            注册一个回调函数,当所有现存的读访问完成之后,调用这个回调函数销毁旧数据
        */
        call_rcu(&old->rcu,myrcu_del);
        value++;
    }
}

static int __init my_test_init(void)
{
    struct task_struct* reader_thread;
    struct task_struct* writer_thread;

    int value = 5;

    printk("figo: my module init\n");
    g_ptr = kzalloc(sizeof(struct foo), GFP_KERNEL);

    reader_thread = kthread_run(myrcu_header_thread,NULL,"rcu_reader");
    writer_thread = kthread_run(myrcu_writer_thread,(void*)(unsigned long) value, "rcu writer");

    return 0;
}

static void __exit my_test_exit(void)
{
    printk("goodbye\n");
    if(g_ptr)
        kfree(g_ptr);
}

MODULE_LICENSE("GPL");
module_init(my_test_init)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值