X-Git-Url: http://v3vee.org/palacios/gitweb/gitweb.cgi?a=blobdiff_plain;f=linux_module%2Fvm.c;h=ac1b363c5d00a71f66dfdeb2e1ad7c7410209eaf;hb=fc04309148f320c3f656d56c936bc45b43ee74a6;hp=624cc39236a7de14c4661b7cbfcaed0a9f24f016;hpb=5ce19b21aac351221fe41e95e717127bfec0c467;p=palacios.releases.git

diff --git a/linux_module/vm.c b/linux_module/vm.c
index 624cc39..ac1b363 100644
--- a/linux_module/vm.c
+++ b/linux_module/vm.c
@@ -12,7 +12,6 @@
 #include <linux/poll.h>
 #include <linux/anon_inodes.h>
 #include <linux/sched.h>
-#include <linux/vmalloc.h>
 #include <linux/file.h>
 #include <linux/spinlock.h>
 #include <linux/rbtree.h>
@@ -124,7 +123,7 @@ int remove_guest_ctrl(struct v3_guest * guest, unsigned int cmd) {
 
     rb_erase(&(ctrl->tree_node), &(guest->vm_ctrls));
 
-    kfree(ctrl);
+    palacios_free(ctrl);
 
     return 0;
 }
@@ -132,16 +131,15 @@ int remove_guest_ctrl(struct v3_guest * guest, unsigned int cmd) {
 static void free_guest_ctrls(struct v3_guest * guest) {
     struct rb_node * node = rb_first(&(guest->vm_ctrls));
     struct vm_ctrl * ctrl = NULL;
-    struct rb_node * tmp_node = NULL;
 
     while (node) {
 	ctrl = rb_entry(node, struct vm_ctrl, tree_node);
-	tmp_node = node;
+
 	node = rb_next(node);
 	
 	WARNING("Cleaning up guest ctrl that was not removed explicitly (%d)\n", ctrl->cmd);
 
-	kfree(ctrl);
+	palacios_free(ctrl);
     }
 }
 
@@ -213,7 +211,7 @@ static long v3_vm_ioctl(struct file * filp,
 	    
 	    NOTICE("Saving Guest to %s:%s\n", chkpt.store, chkpt.url);
 
-	    if (v3_save_vm(guest->v3_ctx, chkpt.store, chkpt.url) == -1) {
+	    if (v3_save_vm(guest->v3_ctx, chkpt.store, chkpt.url, chkpt.opts) == -1) {
 		WARNING("Error checkpointing VM state\n");
 		return -EFAULT;
 	    }
@@ -231,9 +229,9 @@ static long v3_vm_ioctl(struct file * filp,
 		return -EFAULT;
 	    }
 	    
-	    NOTICE("Loading Guest to %s:%s\n", chkpt.store, chkpt.url);
+	    NOTICE("Loading Guest from %s:%s\n", chkpt.store, chkpt.url);
 
-	    if (v3_load_vm(guest->v3_ctx, chkpt.store, chkpt.url) == -1) {
+	    if (v3_load_vm(guest->v3_ctx, chkpt.store, chkpt.url, chkpt.opts) == -1) {
 		WARNING("Error Loading VM state\n");
 		return -EFAULT;
 	    }
@@ -257,7 +255,7 @@ static long v3_vm_ioctl(struct file * filp,
 	    
 	    NOTICE("Sending (live-migrating) Guest to %s:%s\n",chkpt_info.store, chkpt_info.url); 
 	    
-	    if (v3_send_vm(guest->v3_ctx, chkpt_info.store, chkpt_info.url) == -1) {
+	    if (v3_send_vm(guest->v3_ctx, chkpt_info.store, chkpt_info.url, chkpt_info.opts) == -1) {
 		WARNING("Error sending VM\n");
 		return -EFAULT;
 	    }
@@ -279,7 +277,7 @@ static long v3_vm_ioctl(struct file * filp,
 	    
 	    NOTICE("Receiving (live-migrating) Guest to %s:%s\n",chkpt_info.store, chkpt_info.url);
 	    
-	    if (v3_receive_vm(guest->v3_ctx, chkpt_info.store, chkpt_info.url) == -1) {
+	    if (v3_receive_vm(guest->v3_ctx, chkpt_info.store, chkpt_info.url, chkpt_info.opts) == -1) {
 		WARNING("Error receiving VM\n");
 		return -EFAULT;
 	    }
@@ -434,7 +432,9 @@ out_err:
 
 int free_palacios_vm(struct v3_guest * guest) {
 
-    v3_free_vm(guest->v3_ctx);
+    if (v3_free_vm(guest->v3_ctx)<0) { 
+	return -1;
+    }
 
     device_destroy(v3_class, guest->vm_dev);
 
@@ -443,7 +443,7 @@ int free_palacios_vm(struct v3_guest * guest) {
     free_guest_ctrls(guest);
 
 
-    vfree(guest->img);
+    palacios_vfree(guest->img);
     palacios_free(guest);
 
     return 0;