kernel: vm: Add vm_object refcount

Signed-off-by: Ian Moffett <ian@osmora.org>

1 files changed, 5 insertions, 0 deletions

diff --git a/sys/vm/vm_obj.c b/sys/vm/vm_obj.c
index 589ca47..7487b47 100644
--- a/sys/vm/vm_obj.c
+++ b/sys/vm/vm_obj.c
@@ -59,6 +59,7 @@ vm_obj_init(struct vm_object **res, struct vnode *vnode)
 
     memset(obj, 0, sizeof(struct vm_object));
     obj->vnode = vnode;
+    obj->ref = 1;
 
     vm_set_pgops(obj, vnode);
     *res = obj;
@@ -73,6 +74,10 @@ vm_obj_destroy(struct vm_object *obj)
     if (vp->vmobj != NULL)
         vp->vmobj = NULL;
 
+    /* Check the ref count */
+    if (obj->ref > 1)
+        return -EBUSY;
+
     dynfree(obj);
     return 0;
 }