package main import ( "archive/tar" "encoding/json" "flag" "fmt" "io" "log" "net/http" "os" yaml "gopkg.in/yaml.v2" "novit.tech/direktil/pkg/cpiocat" ) var initrdV2 = flag.String("initrd-v2", "2.1.0", "initrd V2 version (temporary flag)") // FIXME func renderBootstrapConfig(w http.ResponseWriter, r *http.Request, ctx *renderContext, asJson bool) (err error) { log.Printf("sending bootstrap config for %q", ctx.Host.Name) _, cfg, err := ctx.BootstrapConfig() if err != nil { return err } if asJson { err = json.NewEncoder(w).Encode(cfg) } else { err = yaml.NewEncoder(w).Encode(cfg) } return nil } func buildInitrd(out io.Writer, ctx *renderContext) (err error) { _, cfg, err := ctx.Config() if err != nil { return } cat := cpiocat.New(out) // initrd initrdPath, err := ctx.distFetch("initrd", *initrdV2) if err != nil { return } cat.AppendArchFile(initrdPath) // embedded layers (modules) for _, layer := range cfg.Layers { switch layer { case "modules": layerVersion := ctx.Host.Versions[layer] modulesPath, err := ctx.distFetch("layers", layer, layerVersion) if err != nil { return err } cat.AppendFile(modulesPath, "modules.sqfs") } } // config cfgBytes, _, err := ctx.BootstrapConfig() if err != nil { return } cat.AppendBytes(cfgBytes, "config.yaml", 0600) // ssh keys // FIXME we want a bootstrap-stage key instead of the real host key cat.AppendBytes(cfg.FileContent("/etc/ssh/ssh_host_rsa_key"), "id_rsa", 0600) return cat.Close() } func buildBootstrap(out io.Writer, ctx *renderContext) (err error) { arch := tar.NewWriter(out) defer arch.Close() // config cfgBytes, cfg, err := ctx.Config() if err != nil { return err } err = arch.WriteHeader(&tar.Header{Name: "config.yaml", Size: int64(len(cfgBytes))}) if err != nil { return } _, err = arch.Write(cfgBytes) if err != nil { return } // layers for _, layer := range cfg.Layers { if layer == "modules" { continue // modules are with the kernel in boot v2 } layerVersion := ctx.Host.Versions[layer] if layerVersion == "" { return fmt.Errorf("layer %q not mapped to a version", layer) } outPath, err := ctx.distFetch("layers", layer, layerVersion) if err != nil { return err } f, err := os.Open(outPath) if err != nil { return err } defer f.Close() stat, err := f.Stat() if err != nil { return err } if err = arch.WriteHeader(&tar.Header{ Name: layer + ".fs", Size: stat.Size(), }); err != nil { return err } _, err = io.Copy(arch, f) if err != nil { return err } } return nil }