Snapshot and migrate functionality.

cmd/minimega/command_socket.go

@@ -30,7 +30,7 @@ func commandSocketStart() {
 				log.Error("commandSocketStart: accept: %v", err)
 				continue
 			}
-			log.Infoln("client connected")
+			log.Debugln("client connected")
 
 			go commandSocketHandle(conn)
 		}
@@ -185,7 +185,7 @@ func commandSocketHandle(c net.Conn) {
 
 	// finally, log the error, if there was one
 	if err == nil || err == io.EOF {
-		log.Infoln("command client disconnected")
+		log.Debugln("command client disconnected")
 	} else if err != nil && strings.Contains(err.Error(), "write: broken pipe") {
 		log.Infoln("command client disconnected without waiting for responses")
 	} else if err != nil {

cmd/minimega/kvm.go

@@ -593,28 +593,33 @@ func (vm *KvmVM) Save(filename string) error {
 	json.Unmarshal(rString, &v)
 
 	// find the device name
-	var device string
+	var device []string
 	for _, dev := range v {
 		if dev.Inserted != nil {
 			if strings.HasPrefix(dev.Inserted.File, fp) {
-				device = dev.Device
-				break
+				device = append(device, dev.Device)
 			}
 		}
 	}
 
-	err = vm.q.SaveDisk(filename, device)
-	if err != nil {
-		return err
-	}
+	for d, devi := range device {
+		if d > 0 {
+			filename = fmt.Sprintf("%s.%s",filename,strconv.Itoa(d))
+			//Appending drive number if there are multiple drives.
+		}
+		log.Info("Saving disk %s to %s", devi, filename)
+
+		if err := vm.q.SaveDisk(filename, devi); err != nil {
+			return err
+		}
 
-	// wait for drive-backup to finish
-	for {
-		if r, _ := vm.q.QueryBlockJobs(); len(r) == 0 {
-			break
+		// wait for drive-backup to finish
+		for {
+			if r, _ := vm.q.QueryBlockJobs(); len(r) == 0 {
+				break
+			}
+			time.Sleep(time.Second * 1)
 		}
-		time.Sleep(time.Second * 1)
-	}
 
 	return err
 }

cmd/minimega/namespace.go

@@ -913,7 +913,7 @@ func SetNamespace(name string) error {
 		return errors.New("namespace name cannot be the empty string")
 	}
 
-	log.Info("setting active namespace: %v", name)
+	log.Debug("setting active namespace: %v", name)
 
 	if name == namespace {
 		return fmt.Errorf("already in namespace: %v", name)
@@ -929,7 +929,7 @@ func RevertNamespace(old, curr *Namespace) {
 	namespaceLock.Lock()
 	defer namespaceLock.Unlock()
 
-	log.Info("reverting to namespace: %v", old)
+	log.Debug("reverting to namespace: %v", old)
 
 	// This is very odd and should *never* happen unless something has gone
 	// horribly wrong.

cmd/minimega/vm_cli.go

@@ -363,27 +363,30 @@ You can also specify the maximum dimension:
 	{ // vm snapshot
 		HelpShort: "write VM state and disk to file",
 		HelpLong: `
-Write VM state (migrate) and disk to file, which can later be booted with 'vm config
-migrate ...' and 'vm config disk ...', respectively.
+Write VM disk to file, which can later be booted with
+'vm config disk ...'.
 
-Saved migrate and disk files are written to the files directory as specified with
--filepath. On success, a call to snapshot a VM will return immediately. You can
+Saved disk files are written to the files directory as specified with
+<filename>. Multiple disks will be handled automatically by appending a unique value.
+On success, a call to snapshot a VM will return immediately. You can
 check the status of in-flight snapshots by invoking vm snapshot with no arguments.`,
 		Patterns: []string{
 			"vm snapshot",
-			"vm snapshot <vm name> <state filename> <disk filename>",
+			"vm snapshot <vm name> <filename>",
 		},
 		Call:    wrapVMTargetCLI(cliVMSnapshot),
 		Suggest: wrapVMSuggest(VM_ANY_STATE, false),
 	},
 	{ // vm migrate
 		HelpShort: "write VM state to disk",
 		HelpLong: `
-Migrate runtime state of a VM to disk, which can later be booted with vm config
-migrate.
+Migrate runtime state of a VM to disk, which can later be booted with 'vm config
+migrate ...' and 'vm config disk ...', respectively. The migrate file may have a 
+dependency with the corresponding disk snapshot image.
 
-Migration files are written to the files directory as specified with -filepath.
-On success, a call to migrate a VM will return immediately. You can check the
+Migration and disk files are written to the files directory as specified with <filename>.
+Migrate file will have .migrate appended to filename, while drive(s) will hold filename
+provided. On success, a call to migrate a VM will return immediately. You can check the
 status of in-flight migrations by invoking vm migrate with no arguments.`,
 		Patterns: []string{
 			"vm migrate",
@@ -820,7 +823,7 @@ func cliVMSnapshot(ns *Namespace, c *minicli.Command, resp *minicli.Response) er
 	}
 
 	// save disk
-	filename := c.StringArgs["disk"]
+	filename := c.StringArgs["filename"]
 
 	if !filepath.IsAbs(filename) {
 		filename = filepath.Join(*f_iomBase, filename)
@@ -838,22 +841,7 @@ func cliVMSnapshot(ns *Namespace, c *minicli.Command, resp *minicli.Response) er
 		return err
 	}
 
-	// save state
-	filename = c.StringArgs["state"]
-
-	if !filepath.IsAbs(filename) {
-		filename = filepath.Join(*f_iomBase, filename)
-	}
-
-	if _, err := os.Stat(filepath.Dir(filename)); os.IsNotExist(err) {
-		if err := os.MkdirAll(filepath.Dir(filename), 0755); err != nil {
-			return err
-		}
-	} else if err != nil {
-		return err
-	}
-
-	return vm.Migrate(filename)
+	return nil
 }
 
 func cliVMMigrate(ns *Namespace, c *minicli.Command, resp *minicli.Response) error {
@@ -884,10 +872,10 @@ func cliVMMigrate(ns *Namespace, c *minicli.Command, resp *minicli.Response) err
 		return err
 	}
 
+	//save disk (a migrate file is often useless without the state of the drive)
 	fname := c.StringArgs["filename"]
 
 	if !filepath.IsAbs(fname) {
-		// TODO: should we write to the VM directory instead?
 		fname = filepath.Join(*f_iomBase, fname)
 	}
 
@@ -899,6 +887,16 @@ func cliVMMigrate(ns *Namespace, c *minicli.Command, resp *minicli.Response) err
 		return err
 	}
 
+	//Saving disk
+	if err := vm.Save(fname); err != nil {
+		return err
+	}
+
+	//Saving memory/migrate
+	fname = fmt.Sprintf("%s.migrate", fname)
+
+	log.Info("Migrating to file %s", fname)
+
 	return vm.Migrate(fname)
 }