Linux内核里的智能指针

struct kref {
	atomic_t refcount;
};

void kref_init(struct kref *kref);
void kref_get(struct kref *kref);
int kref_put(struct kref *kref, void (*release) (struct kref *kref));

struct my_obj
{
	int val;
	struct kref refcnt;
};

struct my_obj *obj;

void obj_release(struct kref *ref) 
{
	struct my_obj *obj = container_of(ref, struct my_obj, refcnt);
	kfree(obj);
}

device_probe() 
{
	obj = kmalloc(sizeof(*obj), GFP_KERNEL);
	kref_init(&obj->refcnt);
}

device_disconnect() 
{
	kref_put(&obj->refcnt, obj_release);
}

.open() 
{
	kref_get(&obj->refcnt);
}

.close() 
{
	kref_put(&obj->refcnt, obj_release);
}

#include <linux/kernel.h>
#include <linux/module.h>

struct my_obj {
        int val;
        struct kref refcnt;
};

struct my_obj *obj;

void obj_release(struct kref *ref)
{
        struct my_obj *obj = container_of(ref, struct my_obj, refcnt);
        printk(KERN_INFO "obj_release\n");
        kfree(obj);
}

static int __init kreftest_init(void)
{
        printk(KERN_INFO "kreftest_init\n");
        obj = kmalloc(sizeof(*obj), GFP_KERNEL);
        kref_init(&obj->refcnt);
        return 0;
}

static void __exit kreftest_exit(void)
{
        printk(KERN_INFO "kreftest_exit\n");
        kref_put(&obj->refcnt, obj_release);
        return;
}

module_init(kreftest_init);
module_exit(kreftest_exit);

MODULE_LICENSE("GPL");