X-Git-Url: http://v3vee.org/palacios/gitweb/gitweb.cgi?a=blobdiff_plain;f=linux_module%2Fvm.c;h=abb798fce74abc7dfbc695b204bcef710e766f89;hb=b1128b2a9d269fefc786c60c26878b372d5e39c1;hp=967ecf6d78301bdc4562cc39f6248c9a0a3e5dad;hpb=9a9c14099c09e9ed86d8ac2f73f7406d0ac8a90b;p=palacios.git

diff --git a/linux_module/vm.c b/linux_module/vm.c
index 967ecf6..abb798f 100644
--- a/linux_module/vm.c
+++ b/linux_module/vm.c
@@ -12,7 +12,7 @@
 #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>
@@ -127,12 +127,6 @@ static long v3_vm_ioctl(struct file * filp,
     printk("V3 IOCTL %d\n", ioctl);
 
     switch (ioctl) {
-
-	case V3_VM_STOP: {
-	    printk("Stopping VM (%s)\n", guest->name);
-	    stop_palacios_vm(guest);
-	    break;
-	}
 	case V3_VM_PAUSE: {
 	    printk("Pausing VM (%s)\n", guest->name);
 	    v3_pause_vm(guest->v3_ctx);
@@ -143,7 +137,65 @@ static long v3_vm_ioctl(struct file * filp,
 	    v3_continue_vm(guest->v3_ctx);
 	    break;
 	}
+#ifdef V3_CONFIG_CHECKPOINT
+	case V3_VM_SAVE: {
+	    struct v3_chkpt_info chkpt;
+	    void __user * argp = (void __user *)arg;
+
+	    memset(&chkpt, 0, sizeof(struct v3_chkpt_info));
+
+	    if (copy_from_user(&chkpt, argp, sizeof(struct v3_chkpt_info))) {
+		printk("Copy from user error getting checkpoint info\n");
+		return -EFAULT;
+	    }
+	    
+	    printk("Saving Guest to %s:%s\n", chkpt.store, chkpt.url);
+
+	    if (v3_save_vm(guest->v3_ctx, chkpt.store, chkpt.url) == -1) {
+		printk("Error checkpointing VM state\n");
+		return -EFAULT;
+	    }
+	    
+	    break;
+	}
+	case V3_VM_LOAD: {
+	    struct v3_chkpt_info chkpt;
+	    void __user * argp = (void __user *)arg;
+
+	    memset(&chkpt, 0, sizeof(struct v3_chkpt_info));
+
+	    if (copy_from_user(&chkpt, argp, sizeof(struct v3_chkpt_info))) {
+		printk("Copy from user error getting checkpoint info\n");
+		return -EFAULT;
+	    }
+	    
+	    printk("Loading Guest to %s:%s\n", chkpt.store, chkpt.url);
+
+	    if (v3_load_vm(guest->v3_ctx, chkpt.store, chkpt.url) == -1) {
+		printk("Error Loading VM state\n");
+		return -EFAULT;
+	    }
+	    
+	    break;
+	}
+#endif
+	case V3_VM_MOVE_CORE: {
+	    struct v3_core_move_cmd cmd;
+	    void __user * argp = (void __user *)arg;
 
+	    memset(&cmd, 0, sizeof(struct v3_core_move_cmd));
+	    
+	    if (copy_from_user(&cmd, argp, sizeof(struct v3_core_move_cmd))) {
+		printk("copy from user error getting migrate command...\n");
+		return -EFAULT;
+	    }
+	
+	    printk("moving guest %s vcore %d to CPU %d\n", guest->name, cmd.vcore_id, cmd.pcore_id);
+
+	    v3_move_vm_core(guest->v3_ctx, cmd.vcore_id, cmd.pcore_id);
+
+	    break;
+	}
 	default: {
 	    struct vm_ctrl * ctrl = get_ctrl(guest, ioctl);
 
@@ -194,22 +246,15 @@ extern u32 pg_frees;
 extern u32 mallocs;
 extern u32 frees;
 
-int start_palacios_vm(void * arg)  {
-    struct v3_guest * guest = (struct v3_guest *)arg;
+int start_palacios_vm(struct v3_guest * guest)  {
     int err;
 
-
-    daemonize(guest->name);
-    // allow_signal(SIGKILL);
-
-    
     init_vm_extensions(guest);
 
     guest->v3_ctx = v3_create_vm(guest->img, (void *)guest, guest->name);
 
     if (guest->v3_ctx == NULL) { 
 	printk("palacios: failed to create vm\n");
-	complete(&(guest->start_done));
 	return -1;
     }
 
@@ -228,7 +273,6 @@ int start_palacios_vm(void * arg)  {
     if (err) {
 	printk("Fails to add cdev\n");
 	v3_free_vm(guest->v3_ctx);
-	complete(&(guest->start_done));
 	return -1;
     }
 
@@ -236,12 +280,9 @@ int start_palacios_vm(void * arg)  {
 	printk("Fails to create device\n");
 	cdev_del(&(guest->cdev));
 	v3_free_vm(guest->v3_ctx);
-	complete(&(guest->start_done));
 	return -1;
     }
 
-    complete(&(guest->start_done));
-
     printk("palacios: launching vm\n");
 
     if (v3_start_vm(guest->v3_ctx, 0xffffffff) < 0) { 
@@ -252,8 +293,6 @@ int start_palacios_vm(void * arg)  {
 	return -1;
     }
     
-    complete(&(guest->thread_done));
-
     printk("palacios: vm completed.  returning.\n");
 
     return 0;
@@ -264,17 +303,17 @@ int start_palacios_vm(void * arg)  {
 
 int stop_palacios_vm(struct v3_guest * guest) {
 
+
     v3_stop_vm(guest->v3_ctx);
 
-    wait_for_completion(&(guest->thread_done));
 
     v3_free_vm(guest->v3_ctx);
-    
+
     device_destroy(v3_class, guest->vm_dev);
 
     cdev_del(&(guest->cdev));
 
-    kfree(guest->img);
+    vfree(guest->img);
     kfree(guest);
 
     return 0;