Back to home page

OSCL-LXR

 
 

    


0001 .. include:: ../disclaimer-zh_CN.rst
0002 
0003 :Original: Documentation/core-api/kref.rst
0004 
0005 翻译:
0006 
0007 司延腾 Yanteng Si <siyanteng@loongson.cn>
0008 
0009 校译:
0010 
0011  <此处请校译员签名(自愿),我将在下一个版本添加>
0012 
0013 .. _cn_core_api_kref.rst:
0014 
0015 =================================
0016 为内核对象添加引用计数器(krefs)
0017 =================================
0018 
0019 :作者: Corey Minyard <minyard@acm.org>
0020 :作者: Thomas Hellstrom <thellstrom@vmware.com>
0021 
0022 其中很多内容都是从Greg Kroah-Hartman2004年关于krefs的OLS论文和演讲中摘
0023 录的,可以在以下网址找到:
0024 
0025   - http://www.kroah.com/linux/talks/ols_2004_kref_paper/Reprint-Kroah-Hartman-OLS2004.pdf
0026   - http://www.kroah.com/linux/talks/ols_2004_kref_talk/
0027 
0028 简介
0029 ====
0030 
0031 krefs允许你为你的对象添加引用计数器。如果你有在多个地方使用和传递的对象,
0032 而你没有refcounts,你的代码几乎肯定是坏的。如果你想要引用计数,krefs是个
0033 好办法。
0034 
0035 要使用kref,请在你的数据结构中添加一个,如::
0036 
0037     struct my_data
0038     {
0039         .
0040         .
0041         struct kref refcount;
0042         .
0043         .
0044     };
0045 
0046 kref可以出现在数据结构体中的任何地方。
0047 
0048 初始化
0049 ======
0050 
0051 你必须在分配kref之后初始化它。 要做到这一点,可以这样调用kref_init::
0052 
0053      struct my_data *data;
0054 
0055      data = kmalloc(sizeof(*data), GFP_KERNEL);
0056      if (!data)
0057             return -ENOMEM;
0058      kref_init(&data->refcount);
0059 
0060 这将kref中的refcount设置为1。
0061 
0062 Kref规则
0063 ========
0064 
0065 一旦你有一个初始化的kref,你必须遵循以下规则:
0066 
0067 1) 如果你对一个指针做了一个非临时性的拷贝,特别是如果它可以被传递给另一个执
0068    行线程,你必须在传递之前用kref_get()增加refcount::
0069 
0070        kref_get(&data->refcount);
0071 
0072         如果你已经有了一个指向kref-ed结构体的有效指针(refcount不能为零),你
0073         可以在没有锁的情况下这样做。
0074 
0075 2) 当你完成对一个指针的处理时,你必须调用kref_put()::
0076 
0077        kref_put(&data->refcount, data_release);
0078 
0079    如果这是对该指针的最后一次引用,释放程序将被调用。如果代码从来没有尝试过
0080    在没有已经持有有效指针的情况下获得一个kref-ed结构体的有效指针,那么在没
0081    有锁的情况下这样做是安全的。
0082 
0083 3) 如果代码试图获得对一个kref-ed结构体的引用,而不持有一个有效的指针,它必
0084    须按顺序访问,在kref_put()期间不能发生kref_get(),并且该结构体在kref_get()
0085    期间必须保持有效。
0086 
0087 例如,如果你分配了一些数据,然后将其传递给另一个线程来处理::
0088 
0089     void data_release(struct kref *ref)
0090     {
0091         struct my_data *data = container_of(ref, struct my_data, refcount);
0092         kfree(data);
0093     }
0094 
0095     void more_data_handling(void *cb_data)
0096     {
0097         struct my_data *data = cb_data;
0098         .
0099         . do stuff with data here
0100         .
0101         kref_put(&data->refcount, data_release);
0102     }
0103 
0104     int my_data_handler(void)
0105     {
0106         int rv = 0;
0107         struct my_data *data;
0108         struct task_struct *task;
0109         data = kmalloc(sizeof(*data), GFP_KERNEL);
0110         if (!data)
0111                 return -ENOMEM;
0112         kref_init(&data->refcount);
0113 
0114         kref_get(&data->refcount);
0115         task = kthread_run(more_data_handling, data, "more_data_handling");
0116         if (task == ERR_PTR(-ENOMEM)) {
0117                 rv = -ENOMEM;
0118                 kref_put(&data->refcount, data_release);
0119                 goto out;
0120         }
0121 
0122         .
0123         . do stuff with data here
0124         .
0125     out:
0126         kref_put(&data->refcount, data_release);
0127         return rv;
0128     }
0129 
0130 这样,两个线程处理数据的顺序并不重要,kref_put()处理知道数据不再被引用并释
0131 放它。kref_get()不需要锁,因为我们已经有了一个有效的指针,我们拥有一个
0132 refcount。put不需要锁,因为没有任何东西试图在没有持有指针的情况下获取数据。
0133 
0134 在上面的例子中,kref_put()在成功和错误路径中都会被调用2次。这是必要的,因
0135 为引用计数被kref_init()和kref_get()递增了2次。
0136 
0137 请注意,规则1中的 "before "是非常重要的。你不应该做类似于::
0138 
0139         task = kthread_run(more_data_handling, data, "more_data_handling");
0140         if (task == ERR_PTR(-ENOMEM)) {
0141                 rv = -ENOMEM;
0142                 goto out;
0143         } else
0144                 /* BAD BAD BAD - 在交接后得到 */
0145                 kref_get(&data->refcount);
0146 
0147 不要以为你知道自己在做什么而使用上述构造。首先,你可能不知道自己在做什么。
0148 其次,你可能知道自己在做什么(有些情况下涉及到锁,上述做法可能是合法的),
0149 但其他不知道自己在做什么的人可能会改变代码或复制代码。这是很危险的作风。请
0150 不要这样做。
0151 
0152 在有些情况下,你可以优化get和put。例如,如果你已经完成了一个对象,并且给其
0153 他对象排队,或者把它传递给其他对象,那么就没有理由先做一个get,然后再做一个
0154 put::
0155 
0156         /* 糟糕的额外获取(get)和输出(put) */
0157         kref_get(&obj->ref);
0158         enqueue(obj);
0159         kref_put(&obj->ref, obj_cleanup);
0160 
0161 只要做enqueue就可以了。 我们随时欢迎对这个问题的评论::
0162 
0163         enqueue(obj);
0164         /* 我们已经完成了对obj的处理,所以我们把我们的refcount传给了队列。
0165          在这之后不要再碰obj了! */
0166 
0167 最后一条规则(规则3)是最难处理的一条。例如,你有一个每个项目都被krefed的列表,
0168 而你希望得到第一个项目。你不能只是从列表中抽出第一个项目,然后kref_get()它。
0169 这违反了规则3,因为你还没有持有一个有效的指针。你必须添加一个mutex(或其他锁)。
0170 比如说::
0171 
0172         static DEFINE_MUTEX(mutex);
0173         static LIST_HEAD(q);
0174         struct my_data
0175         {
0176                 struct kref      refcount;
0177                 struct list_head link;
0178         };
0179 
0180         static struct my_data *get_entry()
0181         {
0182                 struct my_data *entry = NULL;
0183                 mutex_lock(&mutex);
0184                 if (!list_empty(&q)) {
0185                         entry = container_of(q.next, struct my_data, link);
0186                         kref_get(&entry->refcount);
0187                 }
0188                 mutex_unlock(&mutex);
0189                 return entry;
0190         }
0191 
0192         static void release_entry(struct kref *ref)
0193         {
0194                 struct my_data *entry = container_of(ref, struct my_data, refcount);
0195 
0196                 list_del(&entry->link);
0197                 kfree(entry);
0198         }
0199 
0200         static void put_entry(struct my_data *entry)
0201         {
0202                 mutex_lock(&mutex);
0203                 kref_put(&entry->refcount, release_entry);
0204                 mutex_unlock(&mutex);
0205         }
0206 
0207 如果你不想在整个释放操作过程中持有锁,kref_put()的返回值是有用的。假设你不想在
0208 上面的例子中在持有锁的情况下调用kfree()(因为这样做有点无意义)。你可以使用kref_put(),
0209 如下所示::
0210 
0211         static void release_entry(struct kref *ref)
0212         {
0213                 /* 所有的工作都是在从kref_put()返回后完成的。*/
0214         }
0215 
0216         static void put_entry(struct my_data *entry)
0217         {
0218                 mutex_lock(&mutex);
0219                 if (kref_put(&entry->refcount, release_entry)) {
0220                         list_del(&entry->link);
0221                         mutex_unlock(&mutex);
0222                         kfree(entry);
0223                 } else
0224                         mutex_unlock(&mutex);
0225         }
0226 
0227 如果你必须调用其他程序作为释放操作的一部分,而这些程序可能需要很长的时间,或者可
0228 能要求相同的锁,那么这真的更有用。请注意,在释放例程中做所有的事情还是比较好的,
0229 因为它比较整洁。
0230 
0231 上面的例子也可以用kref_get_unless_zero()来优化,方法如下::
0232 
0233         static struct my_data *get_entry()
0234         {
0235                 struct my_data *entry = NULL;
0236                 mutex_lock(&mutex);
0237                 if (!list_empty(&q)) {
0238                         entry = container_of(q.next, struct my_data, link);
0239                         if (!kref_get_unless_zero(&entry->refcount))
0240                                 entry = NULL;
0241                 }
0242                 mutex_unlock(&mutex);
0243                 return entry;
0244         }
0245 
0246         static void release_entry(struct kref *ref)
0247         {
0248                 struct my_data *entry = container_of(ref, struct my_data, refcount);
0249 
0250                 mutex_lock(&mutex);
0251                 list_del(&entry->link);
0252                 mutex_unlock(&mutex);
0253                 kfree(entry);
0254         }
0255 
0256         static void put_entry(struct my_data *entry)
0257         {
0258                 kref_put(&entry->refcount, release_entry);
0259         }
0260 
0261 这对于在put_entry()中移除kref_put()周围的mutex锁是很有用的,但是重要的是
0262 kref_get_unless_zero被封装在查找表中的同一关键部分,否则kref_get_unless_zero
0263 可能引用已经释放的内存。注意,在不检查其返回值的情况下使用kref_get_unless_zero
0264 是非法的。如果你确信(已经有了一个有效的指针)kref_get_unless_zero()会返回true,
0265 那么就用kref_get()代替。
0266 
0267 Krefs和RCU
0268 ==========
0269 
0270 函数kref_get_unless_zero也使得在上述例子中使用rcu锁进行查找成为可能::
0271 
0272         struct my_data
0273         {
0274                 struct rcu_head rhead;
0275                 .
0276                 struct kref refcount;
0277                 .
0278                 .
0279         };
0280 
0281         static struct my_data *get_entry_rcu()
0282         {
0283                 struct my_data *entry = NULL;
0284                 rcu_read_lock();
0285                 if (!list_empty(&q)) {
0286                         entry = container_of(q.next, struct my_data, link);
0287                         if (!kref_get_unless_zero(&entry->refcount))
0288                                 entry = NULL;
0289                 }
0290                 rcu_read_unlock();
0291                 return entry;
0292         }
0293 
0294         static void release_entry_rcu(struct kref *ref)
0295         {
0296                 struct my_data *entry = container_of(ref, struct my_data, refcount);
0297 
0298                 mutex_lock(&mutex);
0299                 list_del_rcu(&entry->link);
0300                 mutex_unlock(&mutex);
0301                 kfree_rcu(entry, rhead);
0302         }
0303 
0304         static void put_entry(struct my_data *entry)
0305         {
0306                 kref_put(&entry->refcount, release_entry_rcu);
0307         }
0308 
0309 但要注意的是,在调用release_entry_rcu后,结构kref成员需要在有效内存中保留一个rcu
0310 宽限期。这可以通过使用上面的kfree_rcu(entry, rhead)来实现,或者在使用kfree之前
0311 调用synchronize_rcu(),但注意synchronize_rcu()可能会睡眠相当长的时间。