veyron/products/lib/gce: Add a library to detect GCE
Add a library to detect if the current process is running on a GCE
instance, and if it is, determine the external IP address to use.
This is uses the GCE metadata API documented at
https://developers.google.com/compute/docs/metadata
Change-Id: I5de47506e8b244d9f42224aa29ba6bf91c011213
diff --git a/products/lib/gce/gce_test.go b/products/lib/gce/gce_test.go
new file mode 100644
index 0000000..ba2591b
--- /dev/null
+++ b/products/lib/gce/gce_test.go
@@ -0,0 +1,62 @@
+package gce
+
+import (
+ "fmt"
+ "net"
+ "net/http"
+ "testing"
+)
+
+func startServer(t *testing.T) (net.Addr, func()) {
+ l, err := net.Listen("tcp", "127.0.0.1:0")
+ if err != nil {
+ t.Fatal(err)
+ }
+ http.HandleFunc("/404", func(w http.ResponseWriter, r *http.Request) {
+ w.WriteHeader(http.StatusNotFound)
+ })
+ http.HandleFunc("/200_not_gce", func(w http.ResponseWriter, r *http.Request) {
+ fmt.Fprintf(w, "Hello")
+ })
+ http.HandleFunc("/gce_no_ip", func(w http.ResponseWriter, r *http.Request) {
+ // When a GCE instance doesn't have an external IP address, the
+ // request returns a 200 with an empty body.
+ w.Header().Add("Metadata-Flavor", "Google")
+ if m := r.Header["Metadata-Flavor"]; len(m) != 1 || m[0] != "Google" {
+ w.WriteHeader(http.StatusForbidden)
+ return
+ }
+ })
+ http.HandleFunc("/gce_with_ip", func(w http.ResponseWriter, r *http.Request) {
+ // When a GCE instance has an external IP address, the request
+ // returns the IP address as body.
+ w.Header().Add("Metadata-Flavor", "Google")
+ if m := r.Header["Metadata-Flavor"]; len(m) != 1 || m[0] != "Google" {
+ w.WriteHeader(http.StatusForbidden)
+ return
+ }
+ fmt.Fprintf(w, "1.2.3.4")
+ })
+
+ go http.Serve(l, nil)
+ return l.Addr(), func() { l.Close() }
+}
+
+func TestGCE(t *testing.T) {
+ addr, stop := startServer(t)
+ defer stop()
+ baseURL := "http://" + addr.String()
+
+ if isGCE, ip := googleComputeEngineTest(baseURL + "/404"); isGCE != false || ip != nil {
+ t.Errorf("Unexpected result. Got %v:%v, want false:nil", isGCE, ip)
+ }
+ if isGCE, ip := googleComputeEngineTest(baseURL + "/200_not_gce"); isGCE != false || ip != nil {
+ t.Errorf("Unexpected result. Got %v:%v, want false:nil", isGCE, ip)
+ }
+ if isGCE, ip := googleComputeEngineTest(baseURL + "/gce_no_ip"); isGCE != true || ip != nil {
+ t.Errorf("Unexpected result. Got %v:%v, want true:nil", isGCE, ip)
+ }
+ if isGCE, ip := googleComputeEngineTest(baseURL + "/gce_with_ip"); isGCE != true || ip.String() != "1.2.3.4" {
+ t.Errorf("Unexpected result. Got %v:%v, want true:1.2.3.4", isGCE, ip)
+ }
+}
