Initial commit

http.go

@@ -0,0 +1,97 @@
+package main
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+
+	"entanglement.garden/common/cloudinit"
+	"github.com/sirupsen/logrus"
+	"gopkg.in/yaml.v2"
+)
+
+var (
+	httpStaticContent = map[string][]byte{}
+)
+
+func runHTTPServer(bind string) {
+	http.ListenAndServe(bind, httphandler{})
+}
+
+type httphandler struct{}
+
+func (httphandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	log := logrus.WithFields(logrus.Fields{
+		"method":      r.Method,
+		"path":        r.URL,
+		"remote_addr": r.RemoteAddr,
+	})
+
+	log.Debug("handling HTTP request")
+
+	switch r.Method {
+	case http.MethodGet:
+		resp, ok := httpStaticContent[r.URL.Path]
+		if !ok {
+			w.WriteHeader(http.StatusNotFound)
+			return
+		}
+
+		if resp == nil {
+			w.WriteHeader(http.StatusNoContent)
+			return
+		}
+
+		_, err := w.Write(resp)
+		if err != nil {
+			log.WithError(err).Error("error serving HTTP response")
+		}
+
+	case http.MethodPost:
+		body, err := io.ReadAll(r.Body)
+		if err != nil {
+			log.WithError(err).Error("error reading body")
+		}
+		r.Body.Close()
+
+		log.Debug(string(body))
+
+		log.Info("VM booted")
+		w.WriteHeader(http.StatusNoContent)
+		wg.Done()
+
+	default:
+		w.WriteHeader(http.StatusNotFound)
+		w.Write([]byte("not found"))
+	}
+
+}
+
+func buildCloudConfig(i int, name string, phoneHomeURL string) error {
+	userdata, err := yaml.Marshal(cloudinit.UserData{
+		Packages:          argPackages,
+		SSHAuthorizedKeys: argSSHKeys,
+		PhoneHome: cloudinit.PhoneHome{
+			URL:  phoneHomeURL,
+			Post: []string{"pub_key_dsa"},
+		},
+	})
+	if err != nil {
+		return err
+	}
+
+	httpStaticContent[fmt.Sprintf("/%d/user-data", i)] = append([]byte("#cloud-config\n"), userdata...)
+
+	metadata, err := yaml.Marshal(cloudinit.MetaData{
+		InstanceID:    name,
+		LocalHostname: name,
+	})
+	if err != nil {
+		return err
+	}
+	httpStaticContent[fmt.Sprintf("/%d/meta-data", i)] = metadata
+
+	httpStaticContent[fmt.Sprintf("/%d/vendor-data", i)] = nil
+
+	return nil
+}