128 lines
2.6 KiB
Go
128 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"novit.nc/direktil/pkg/localconfig"
|
|
)
|
|
|
|
var (
|
|
reHost = regexp.MustCompile("^/hosts/([^/]+)/([^/]+)$")
|
|
|
|
trustXFF = flag.Bool("trust-xff", true, "Trust the X-Forwarded-For header")
|
|
)
|
|
|
|
func serveHostByIP(w http.ResponseWriter, r *http.Request) {
|
|
host, cfg := hostByIP(w, r)
|
|
if host == nil {
|
|
return
|
|
}
|
|
|
|
what := strings.TrimLeft(r.URL.Path, "/")
|
|
|
|
renderHost(w, r, what, host, cfg)
|
|
}
|
|
|
|
func hostByIP(w http.ResponseWriter, r *http.Request) (*localconfig.Host, *localconfig.Config) {
|
|
remoteAddr := r.RemoteAddr
|
|
|
|
if *trustXFF {
|
|
if xff := r.Header.Get("X-Forwarded-For"); xff != "" {
|
|
remoteAddr = strings.Split(xff, ",")[0]
|
|
}
|
|
}
|
|
|
|
hostIP, _, err := net.SplitHostPort(remoteAddr)
|
|
|
|
if err != nil {
|
|
hostIP = remoteAddr
|
|
}
|
|
|
|
cfg, err := readConfig()
|
|
if err != nil {
|
|
http.Error(w, "", http.StatusServiceUnavailable)
|
|
return nil, nil
|
|
}
|
|
|
|
host := cfg.HostByIP(hostIP)
|
|
|
|
if host == nil {
|
|
log.Print("no host found for IP ", hostIP)
|
|
http.NotFound(w, r)
|
|
return nil, nil
|
|
}
|
|
|
|
return host, cfg
|
|
}
|
|
|
|
func renderHost(w http.ResponseWriter, r *http.Request, what string, host *localconfig.Host, cfg *localconfig.Config) {
|
|
ctx, err := newRenderContext(host, cfg)
|
|
if err != nil {
|
|
log.Printf("host %s: %s: failed to render: %v", what, host.Name, err)
|
|
http.Error(w, "", http.StatusServiceUnavailable)
|
|
return
|
|
}
|
|
|
|
switch what {
|
|
case "ipxe":
|
|
w.Header().Set("Content-Type", "text/x-ipxe")
|
|
case "config":
|
|
w.Header().Set("Content-Type", "text/vnd.yaml")
|
|
default:
|
|
w.Header().Set("Content-Type", "application/octet-stream")
|
|
}
|
|
|
|
switch what {
|
|
case "ipxe":
|
|
err = renderIPXE(w, ctx)
|
|
|
|
case "kernel":
|
|
err = renderKernel(w, r, ctx)
|
|
|
|
case "initrd":
|
|
err = renderCtx(w, r, ctx, what, buildInitrd)
|
|
|
|
case "boot.iso":
|
|
err = renderCtx(w, r, ctx, what, buildBootISO)
|
|
|
|
case "boot.tar":
|
|
err = renderCtx(w, r, ctx, what, buildBootTar)
|
|
|
|
case "boot.img":
|
|
err = renderCtx(w, r, ctx, what, buildBootImg)
|
|
|
|
case "boot.img.gz":
|
|
err = renderCtx(w, r, ctx, what, buildBootImgGZ)
|
|
|
|
case "boot.img.lz4":
|
|
err = renderCtx(w, r, ctx, what, buildBootImgLZ4)
|
|
|
|
case "config":
|
|
err = renderConfig(w, r, ctx)
|
|
|
|
default:
|
|
http.NotFound(w, r)
|
|
}
|
|
|
|
if err != nil {
|
|
if isNotFound(err) {
|
|
log.Printf("host %s: %s: %v", what, host.Name, err)
|
|
http.NotFound(w, r)
|
|
} else {
|
|
log.Printf("host %s: %s: failed to render: %v", what, host.Name, err)
|
|
http.Error(w, "", http.StatusServiceUnavailable)
|
|
}
|
|
}
|
|
}
|
|
|
|
func writeError(w http.ResponseWriter, err error) {
|
|
log.Print("request failed: ", err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write([]byte(http.StatusText(http.StatusInternalServerError)))
|
|
}
|