From b965a2c0e35ced54367b0336dfd77e04e1f65c05 Mon Sep 17 00:00:00 2001
From: Josh Curl <hello@joshcurl.com>
Date: Mon, 28 Mar 2016 11:27:27 -0700
Subject: [PATCH] Only pull upgrade image if not found locally

---
 cmd/control/os.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/cmd/control/os.go b/cmd/control/os.go
index 54b2f8e2..37085f9b 100644
--- a/cmd/control/os.go
+++ b/cmd/control/os.go
@@ -199,10 +199,18 @@ func startUpgradeContainer(image string, stage, force, reboot, kexec bool) error
 		return err
 	}
 
-	if err := container.Pull(); err != nil {
+	client, err := docker.NewSystemClient()
+	if err != nil {
 		return err
 	}
 
+	// Only pull image if not found locally
+	if _, err := client.InspectImage(image); err != nil {
+		if err := container.Pull(); err != nil {
+			return err
+		}
+	}
+
 	if !stage {
 		fmt.Printf("Upgrading to %s\n", image)