enable grub boot for all nodes

Signed-off-by: Christian Goll <cgoll@suse.com>

internal/pkg/warewulfd/provision.go

@@ -3,20 +3,28 @@ package warewulfd
 import (
 	"bytes"
 	"errors"
+	"fmt"
 	"net/http"
 	"path"
 	"strconv"
 	"strings"
+	"sync"
 	"text/template"
 
 	warewulfconf "github.com/hpcng/warewulf/internal/pkg/config"
 	"github.com/hpcng/warewulf/internal/pkg/container"
 	"github.com/hpcng/warewulf/internal/pkg/kernel"
+	"github.com/hpcng/warewulf/internal/pkg/node"
 	"github.com/hpcng/warewulf/internal/pkg/overlay"
 	"github.com/hpcng/warewulf/internal/pkg/util"
 	"github.com/hpcng/warewulf/internal/pkg/wwlog"
 )
 
+var sentMap = make(map[string]string)
+
+// define a mutex lock
+var mu sync.Mutex
+
 type iPxeTemplate struct {
 	Message        string
 	WaitTime       string
@@ -262,3 +270,109 @@ func ProvisionSend(w http.ResponseWriter, req *http.Request) {
 	}
 
 }
+
+/*
+simple handler for sending only out grub to the nodes, parses the
+GET request, so that for "/grub/shim.efi" the "shim.efi" from the
+container of the default profile will be sent and for "/grub/shim.efi?$container" the
+shim of $container will be sent
+*/
+func GrubSend(w http.ResponseWriter, req *http.Request) {
+	req.Close = true
+	wwlog.Debug("Grub send called with url: %s host: %s", req.URL, req.Host)
+	url := strings.Split(req.URL.Path, "?")[0]
+	path_parts := strings.Split(path.Join(url), "/")
+	remoteIP := strings.Split(req.RemoteAddr, ":")[0]
+	mu.Lock()
+	if last_sent, ok := sentMap[remoteIP]; ok {
+		wwlog.Debug("last sent is: %s", last_sent)
+	}
+	mu.Unlock()
+	if len(path_parts) != 3 {
+		w.WriteHeader(http.StatusBadRequest)
+		wwlog.ErrorExc(fmt.Errorf("unknown path components in GET: %s", req.URL.Path), remoteIP)
+		return
+	}
+	requested_file := path_parts[2]
+	nodeDB, err := node.New()
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+	profiles, _ := nodeDB.MapAllProfiles()
+	containerName := ""
+	if profile, ok := profiles["default"]; ok {
+		containerName = profile.ContainerName.Get()
+	}
+	remoteNode, err := nodeDB.FindByIpaddr(remoteIP)
+	if err == nil {
+		containerName = remoteNode.ContainerName.Get()
+	} else {
+		if mac := ArpFind(remoteIP); mac != "" {
+			remoteNode, err := nodeDB.FindByHwaddr(mac)
+			if err != nil {
+				containerName = remoteNode.ContainerName.Get()
+			} else if remoteNode, err = GetNodeOrSetDiscoverable(mac); err == nil {
+				containerName = remoteNode.ContainerName.Get()
+			}
+		}
+	}
+
+	if len(url) == 2 {
+		containerName = strings.Split(req.URL.Path, "?")[1]
+	}
+	mu.Lock()
+	if last_sent, ok := sentMap[remoteIP]; ok && last_sent == "shim.efi" {
+		requested_file = "grub.efi"
+	}
+	mu.Unlock()
+	wwlog.Recv("remoteIP: %s, container: %s, filename: %s", remoteIP, containerName, requested_file)
+	var stage_file string
+	switch requested_file {
+	case "shim.efi":
+		stage_file = container.ShimFind(containerName)
+		if stage_file == "" {
+			wwlog.ErrorExc(fmt.Errorf("could't find shim.efi"), containerName)
+			w.WriteHeader(http.StatusNotFound)
+			return
+		}
+		mu.Lock()
+		sentMap[remoteIP] = "shim.efi"
+		mu.Unlock()
+	case "grub.efi", "grubx86.efi":
+		stage_file = container.GrubFind(containerName)
+		if stage_file == "" {
+			wwlog.ErrorExc(fmt.Errorf("could't find grub.efi"), containerName)
+			w.WriteHeader(http.StatusNotFound)
+			return
+		}
+		mu.Lock()
+		sentMap[remoteIP] = "grub.efi"
+		mu.Unlock()
+	}
+	wwlog.Serv("stage_file '%s'", stage_file)
+
+	if util.IsFile(stage_file) {
+		/*
+			fd, err := os.Open(stage_file)
+			if err != nil {
+				w.WriteHeader(http.StatusInternalServerError)
+				return
+			}
+			defer fd.Close()
+		*/
+		/*
+			stat, err := fd.Stat()
+			if err != nil {
+				w.WriteHeader(http.StatusInternalServerError)
+				return
+			}
+
+			http.ServeContent(w, req, stage_file, stat.ModTime(), fd)
+			io.Copy(w, fd)
+			req.Close = true
+		*/
+		http.ServeFile(w, req, stage_file)
+		wwlog.Send("%15s: %s", remoteIP, stage_file)
+	}
+}