Chinaunix首页 | 论坛 | 博客
  • 博客访问: 473768
  • 博文数量: 223
  • 博客积分: 0
  • 博客等级: 民兵
  • 技术积分: 2145
  • 用 户 组: 普通用户
  • 注册时间: 2014-03-01 10:23
个人简介

该坚持的时候坚持,该妥协的时候妥协,该放弃的时候放弃

文章分类

全部博文(223)

文章存档

2017年(56)

2016年(118)

2015年(3)

2014年(46)

我的朋友

分类: LINUX

2014-12-15 23:41:53

RCU(Read-Copy Update),顾名思义就是读-拷贝修改,它是基于其原理命名的。对于被RCU保护的共享数据结构,读者不需要获得任何锁就可以访问它,但写者在访问它 时首先拷贝一个副本,然后对副本进行修改,最后使用一个回调(callback)机制在适当的时机把指向原来数据的指针重新指向新的被修改的数据。这个时 机就是所有引用该数据的CPU都退出对共享数据的操作。
在定义需要使用rcu的资源时,定义中可以加上__rcu修饰,如:
  1. static struct our_data __rcu *pmy_data = &my_data;
好处:1、告诉读者这个变量是由rcu所保护的。
2、linux kernel有一个静态的检查工具,他会去检查用rcu申明的指针,是不是用rcu_read_lock()保护的。

读者

使用rcu_read_lock()来进入读侧

使用rcu_dereference()来获取指针对应的资源

使用rcu_read_unlock()来退出读侧

可以这样用比如

  1. struct our_data *data;
  2. rcu_read_lock();
  3. data = rcu_dereference(pmy_data);//pmy_data是我们需要读取的指针,返回到data中获取
  4. printk("read count1 %d count2 %d .\n", data->count1, data->count2);
  5. rcu_read_unlock();


写者

如果有多个写端,需要写端自己做同步

使用rcu_assign_pointer()来更新指针

使用call_rcu()/synchronize_rcu()来释放旧指针对应的资源

    使用call_rcu()和synchronize_rcu()的区别

    call_rcu()用来将释放的工作交给后台,可以快速返回,不会导致睡眠

    synchronisze_rcu()会等待所有的读者退出再返回,会导致睡眠


  1.     data = kmalloc(sizeof(*data), GFP_KERNEL);//先分配一块内存空间
  2.     if (!data)
  3.         return;
  4. /*memcpy
    函数原型:extern void *memmove(void *dest, const void *src, unsigned int count)
    参数说明:dest为目的字符串,src为源字符串,count为要拷贝的字节数。*/
  5.     memcpy(data, pmy_data, sizeof(struct our_data));//然后把需要改写的内容复制到data中
  6.     data->count1 ++;
  7.     data->count2 += 10;

  8.     rcu_assign_pointer(pmy_data, data);//改写内容后通过rcu_assign_pointer将指针更新

  9.     if (tmp != &my_data) {
  10.         synchronize_rcu();//会等待所有的读者退出再返回,会导致睡眠
  11.         kfree(tmp);
  12.     }
然后是call_rcu,原型如下:
  1. extern void call_rcu(struct rcu_head *head,
  2.              void (*func)(struct rcu_head *head));
call_rcu可以这样使用:
  1. struct our_data {
  2.     int count1;
  3.     int count2;
  4.     struct rcu_head rhead;//先设定一个rcu_hread
  5. };

  6. static void rcu_free(struct rcu_head *head)
  7. {
  8.     struct our_data *data;

  9.     data = container_of(head, struct our_data, rhead);//从结构体中的指针,获取结构体的指针
  10.     kfree(data);//用来释放data
  11. }

  12. ...
  13. call_rcu(&tmp->rhead, rcu_free);//传入rcu_hread指针,定义函数名字
  14. ...


第一次写,失败了。又写了一遍


  1. #include
  2. #include <linux/module.h>
  3. #include <linux/delay.h>
  4. #include <linux/err.h>
  5. #include <linux/slab.h>
  6. #include <linux/rcupdate.h>

  7. MODULE_LICENSE("GPL");
  8. MODULE_AUTHOR("chen");
  9. MODULE_DESCRIPTION("exam for opensource.");

  10. #define MAX_KTHREADS 10

  11. struct our_data {
  12.     int count1;
  13.     int count2;
  14.     struct rcu_head rhead;
  15. };

  16. static unsigned long reader_bitmap;
  17. static struct task_struct *kthreads[MAX_KTHREADS];
  18. static struct our_data my_data;
  19. static struct our_data __rcu *pmy_data = &my_data;

  20. static void show_my_data(void)
  21. {
  22.     printk("my_data.count1 is %d, count2 is %d.\n", my_data.count1, my_data.count2);
  23.     printk("pmy_data.count1 is %d, count2 is %d.\n", pmy_data->count1, pmy_data->count2);
  24.     
  25. }

  26. static void set_reader_number(int reader)
  27. {
  28.     WARN_ON((MAX_KTHREADS - reader) != 1);
  29.     reader_bitmap = 0;
  30.     while(reader)
  31.     {
  32.         reader_bitmap |= 1 << --reader;
  33.     }
  34. }

  35. static void reader_do(void)
  36. {
  37.     struct our_data *data;
  38.     rcu_read_lock();
  39.     data = rcu_dereference(pmy_data);
  40.     printk("read count1 %d count2 %d\n", data->count1, data->count2);
  41.     rcu_read_unlock();
  42. }

  43. static void rcu_free(struct rcu_head *head)
  44. {
  45.     struct our_data *data;
  46.     
  47.     data = container_of(head, struct our_data, rhead);
  48.     kfree(data);
  49. }

  50. static void writer_do(void)
  51. {
  52.     struct our_data *data, *tmp;
  53.     tmp = pmy_data;

  54.     data = kmalloc(sizeof(*data), GFP_KERNEL);
  55.     if (!data)
  56.         return;

  57.     memcpy(data, pmy_data, sizeof(struct our_data));
  58.     data->count1 ++;
  59.     data->count2 += 10;

  60.     rcu_assign_pointer(pmy_data, data);

  61.     if (tmp != &my_data)
  62.         call_rcu(&tmp->rhead, rcu_free);
  63. }

  64. static void kthread_cleanup(void)
  65. {
  66.     int i = 0;
  67.     for(i = 0; i < MAX_KTHREADS; i++)
  68.         if (kthreads[i])
  69.             kthread_stop(kthreads[i]);
  70. }

  71. static int kthread_do(void *data)
  72. {
  73.     long i = (long)data;
  74.     int reader = (reader_bitmap & (1 << i));
  75.     printk("run...%ld %s ...\n", i, reader ? "reader" : "writer");
  76.     while(!kthread_should_stop()) {
  77.         if (reader_bitmap & (1 << i))
  78.             reader_do();
  79.         else
  80.             writer_do();

  81.         msleep(10);
  82.     }
  83.     
  84.     return 0;
  85. }

  86. static int create_kthread(void)
  87. {
  88.     int i = 0, ret = 0;
  89.     for(i = 0; i < MAX_KTHREADS; i++) {
  90.         struct task_struct *tsk;
  91.         tsk = kthread_run(kthread_do, (void *)(long)i, "kthread-%d", i);
  92.         if (IS_ERR(tsk)) {
  93.             printk("create kthread err.\n");
  94.             ret = PTR_ERR(tsk);
  95.             goto err;
  96.         }
  97.         kthreads[i] = tsk;
  98.     }
  99.     return ret;

  100. err:
  101.     kthread_cleanup();
  102.     return ret;
  103. }

  104. static int __init main_init(void)
  105. {
  106.     printk("call %s.\n", __FUNCTION__);
  107.     set_reader_number(9);

  108.     if (create_kthread())
  109.         goto err;
  110.     return 0;

  111. err:
  112.     kthread_cleanup();
  113.     return -1;
  114. }

  115. static void __exit main_exit(void)
  116. {
  117.     kthread_cleanup();
  118.     show_my_data();
  119.     printk("call %s .\n", __FUNCTION__);
  120. }

  121. module_init(main_init);
  122. module_exit(main_exit);

为什么pmy_data->count1和pmy_data->count2能输出,而用my_data.count1和my_data.count2就显示是0
因为指针变了。。。

http://www.ibm.com/developerworks/cn/linux/l-rcu/
阅读(1798) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~