From f525e884edce9bcd2dc6f7707ec24e15f64b1cc7 Mon Sep 17 00:00:00 2001
From: Gareth Rees <grees@azuron.co.uk>
Date: Sun, 10 Jan 2021 18:29:37 +0000
Subject: [PATCH] Timeout on metadata server

---
 builder/googlecompute/driver_gce.go | 37 ++++++++++++++++++++++-------
 1 file changed, 29 insertions(+), 8 deletions(-)

diff --git a/builder/googlecompute/driver_gce.go b/builder/googlecompute/driver_gce.go
index ef2e5601b..c587f702a 100644
--- a/builder/googlecompute/driver_gce.go
+++ b/builder/googlecompute/driver_gce.go
@@ -149,14 +149,7 @@ func NewDriverGCE(config GCEDriverConfig) (Driver, error) {
 		return nil, err
 	}
 
-	if metadata.OnGCE() {
-		log.Printf("[INFO] On GCE, capture service account for OSLogin...")
-		thisGCEUser, err = metadata.NewClient(&http.Client{}).Email("")
-
-		if err != nil {
-			return nil, err
-		}
-	}
+	thisGCEUser = getGCEUser()
 
 	log.Printf("[INFO] Instantiating OS Login client...")
 	osLoginService, err := oslogin.NewService(context.TODO(), opts)
@@ -799,3 +792,31 @@ func (d *driverGCE) AddToInstanceMetadata(zone string, name string, metadata map
 
 	return nil
 }
+
+// getGCEUser determines if we're running packer on a GCE, and if we are, gets the associated service account email for subsequent use with OSLogin.
+// There are cases where we are running on a GCE, but the GCP metadata server isn't accessible. GitLab docker-engine runners are an edge case example of this.
+// It makes little sense to run packer on GCP in this way, however, we defensively timeout in those cases, rather than abort.
+func getGCEUser() string {
+
+	var thisGCEUser string
+
+	metadataCheckTimeout := 5 * time.Second
+	metadataCheckChl := make(chan string, 1)
+
+	go func() {
+		if metadata.OnGCE() {
+			log.Printf("[INFO] Attempt to capture the GCE service account for use with OSLogin...")
+			GCEUser, _ := metadata.NewClient(&http.Client{}).Email("")
+			metadataCheckChl <- GCEUser
+		}
+	}()
+
+	select {
+	case thisGCEUser := <-metadataCheckChl:
+		fmt.Printf("[INFO] GCE service account %s will be used for OSLogin", thisGCEUser)
+	case <-time.After(metadataCheckTimeout):
+		fmt.Printf("[INFO] Timeout after %s whilst waiting for google metadata server.", metadataCheckTimeout)
+	}
+
+	return thisGCEUser
+}