模块API之module_put/__module_get

最新推荐文章于 2024-03-24 06:35:39 发布

tiantao2012

最新推荐文章于 2024-03-24 06:35:39 发布

阅读量3.3k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Linux 源码分析 kernel常用API源码分析

本文链接：https://blog.youkuaiyun.com/tiantao2012/article/details/78414130

Linux 源码分析同时被 2 个专栏收录

769 篇文章

订阅专栏

kernel常用API源码分析

225 篇文章

订阅专栏

本文介绍了Linux内核中模块管理函数module_put和__module_get的使用方法及其源码实现。这些函数通过原子操作来调整模块的引用计数，并在特定条件下禁用内核抢占以确保操作的一致性和安全性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

module_put和__module_get 是一对函数，用于减少或者增加模块的引用计数.
具体使用的例子如下：
void nfs4_schedule_state_manager(struct nfs_client *clp)
{
	struct task_struct *task;
	char buf[INET6_ADDRSTRLEN + sizeof("-manager") + 1];

	if (test_and_set_bit(NFS4CLNT_MANAGER_RUNNING, &clp->cl_state) != 0)
		return;
	__module_get(THIS_MODULE);
	atomic_inc(&clp->cl_count);

	task = kthread_run(nfs4_run_state_manager, clp, "%s", buf);
	if (IS_ERR(task)) {
		printk(KERN_ERR "%s: kthread_run: %ld\n",
			__func__, PTR_ERR(task));
		nfs4_clear_state_manager_bit(clp);
		nfs_put_client(clp);
		module_put(THIS_MODULE);
	}
}
其源码如下：
void module_put(struct module *module)
{
	int ret;

	if (module) {
		preempt_disable();
		ret = atomic_dec_if_positive(&module->refcnt);
		WARN_ON(ret < 0);	/* Failed to put refcount */
		trace_module_put(module, _RET_IP_);
		preempt_enable();
	}
}
EXPORT_SYMBOL(module_put);
void __module_get(struct module *module)
{
	if (module) {
		preempt_disable();
		atomic_inc(&module->refcnt);
		trace_module_get(module, _RET_IP_);
		preempt_enable();
	}
}
EXPORT_SYMBOL(__module_get);
可见在使用module_put和__module_get的时候，要通过preempt_disable()/preempt_enable() 来禁止/使能内核的抢占
module->refcnt是以原子的方式增加或者减小的，因此不用加锁
在module_put 中通过atomic_dec_if_positive 来使module->refcnt 减一时，要先判断减一后是否小于零，也就是当前的module->refcnt == 0.这种情况下就
不能在减一了，从这里也可以知道module->refcnt的最小值是零.
static inline int atomic_dec_if_positive(atomic_t *v)
{
	int c, old, dec;
	c = atomic_read(v);
	for (;;) {
		dec = c - 1;
		if (unlikely(dec < 0))
			break;
		old = atomic_cmpxchg((v), c, dec);
		if (likely(old == c))
			break;
		c = old;
	}
	return dec;
}