From 4d5edcaac098c21eefc61d294e8608533121eeff Mon Sep 17 00:00:00 2001
From: Mitchell Hashimoto <mitchell.hashimoto@gmail.com>
Date: Tue, 2 Jul 2013 09:15:52 -0700
Subject: [PATCH] builder/vmware: Improve check for VMware cleanup

---
 builder/vmware/step_compact_disk.go |  3 ++-
 builder/vmware/step_shutdown.go     | 31 +++++++++++++++++++++++++++--
 2 files changed, 31 insertions(+), 3 deletions(-)

diff --git a/builder/vmware/step_compact_disk.go b/builder/vmware/step_compact_disk.go
index d581e483d..d399322a7 100644
--- a/builder/vmware/step_compact_disk.go
+++ b/builder/vmware/step_compact_disk.go
@@ -10,6 +10,7 @@ import (
 //
 // Uses:
 //   driver Driver
+//   full_disk_path string
 //   ui     packer.Ui
 //
 // Produces:
@@ -23,7 +24,7 @@ func (stepCompactDisk) Run(state map[string]interface{}) multistep.StepAction {
 
 	ui.Say("Compacting the disk image")
 	if err := driver.CompactDisk(full_disk_path); err != nil {
-		state["error"] := fmt.Errorf("Error compacting disk: %s", err)
+		state["error"] = fmt.Errorf("Error compacting disk: %s", err)
 		return multistep.ActionHalt
 	}
 
diff --git a/builder/vmware/step_shutdown.go b/builder/vmware/step_shutdown.go
index b17436b8b..38574238f 100644
--- a/builder/vmware/step_shutdown.go
+++ b/builder/vmware/step_shutdown.go
@@ -6,6 +6,8 @@ import (
 	"github.com/mitchellh/multistep"
 	"github.com/mitchellh/packer/packer"
 	"log"
+	"path/filepath"
+	"strings"
 	"time"
 )
 
@@ -72,8 +74,33 @@ func (s *stepShutdown) Run(state map[string]interface{}) multistep.StepAction {
 		}
 	}
 
-	log.Println("Giving VMware a few extra seconds to clean up after itself...")
-	time.Sleep(5 * time.Second)
+	ui.Message("Waiting for VMware to clean up after itself...")
+	lockPattern := filepath.Join(config.OutputDir, "*.lck")
+	timer := time.After(15 * time.Second)
+LockWaitLoop:
+	for {
+		locks, err := filepath.Glob(lockPattern)
+		if err == nil {
+			if len(locks) == 0 {
+				log.Println("No more lock files found. VMware is clean.")
+				break
+			}
+
+			if len(locks) == 1 && strings.HasSuffix(locks[0], ".vmx.lck") {
+				log.Println("Only waiting on VMX lock. VMware is clean.")
+				break
+			}
+
+			log.Printf("Waiting on lock files: %#v", locks)
+		}
+
+		select {
+		case <-timer:
+			log.Println("Reached timeout on waiting for clean VMware. Assuming clean.")
+			break LockWaitLoop
+		case <-time.After(1 * time.Second):
+		}
+	}
 
 	log.Println("VM shut down.")
 	return multistep.ActionContinue