From 196bd6741ea8e1e952cd0fb3c26f844f365f6fbf Mon Sep 17 00:00:00 2001
From: Brad Fitzpatrick <bradfitz@golang.org>
Date: Thu, 4 Sep 2014 13:46:24 -0700
Subject: [PATCH] dashboard/builder: fallback to finding build keys from GCE
 metadata

LGTM=cmang
R=cmang
CC=adg, golang-codereviews
https://golang.org/cl/133610043
---
 dashboard/builder/main.go | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/dashboard/builder/main.go b/dashboard/builder/main.go
index b9498b5b..be3172e9 100644
--- a/dashboard/builder/main.go
+++ b/dashboard/builder/main.go
@@ -11,6 +11,8 @@ import (
 	"io"
 	"io/ioutil"
 	"log"
+	"net"
+	"net/http"
 	"os"
 	"os/exec"
 	"path/filepath"
@@ -280,12 +282,45 @@ func NewBuilder(goroot *Repo, name string) (*Builder, error) {
 	}
 	c, err := ioutil.ReadFile(fn)
 	if err != nil {
+		// If the on-disk file doesn't exist, also try the
+		// Google Compute Engine metadata.
+		if v := gceProjectMetadata("buildkey-" + name); v != "" {
+			b.key = v
+			return b, nil
+		}
 		return nil, fmt.Errorf("readKeys %s (%s): %s", b.name, fn, err)
 	}
 	b.key = string(bytes.TrimSpace(bytes.SplitN(c, []byte("\n"), 2)[0]))
 	return b, nil
 }
 
+func gceProjectMetadata(attr string) string {
+	client := &http.Client{
+		Transport: &http.Transport{
+			Dial: (&net.Dialer{
+				Timeout:   750 * time.Millisecond,
+				KeepAlive: 30 * time.Second,
+			}).Dial,
+			ResponseHeaderTimeout: 750 * time.Millisecond,
+		},
+	}
+	req, _ := http.NewRequest("GET", "http://metadata.google.internal/computeMetadata/v1/project/attributes/"+attr, nil)
+	req.Header.Set("Metadata-Flavor", "Google")
+	res, err := client.Do(req)
+	if err != nil {
+		return ""
+	}
+	defer res.Body.Close()
+	if res.StatusCode != 200 {
+		return ""
+	}
+	slurp, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		return ""
+	}
+	return string(bytes.TrimSpace(slurp))
+}
+
 // builderEnv returns the builderEnv for this buildTool.
 func (b *Builder) builderEnv(name string) (builderEnv, error) {
 	// get goos/goarch from builder string