Files
warewulf/internal/pkg/image/shimgrub.go
Jonathon Anderson 45a690ca4e Rename "container" to "image"
- Updated `wwctl upgrade` to handle updates
- Maintained `.Container` and `.ContainerName` in tstruct
- Added `ContainerName()` methods to node and profile objects
- Added `--container`, `-C` aliases to wwctl commands (`<node|profile> <add|set>`)
- Added `wwctl container` alias
- Added support for `container_exit.sh` if `image_exit.sh` is not found

Signed-off-by: Jonathon Anderson <janderson@ciq.com>
2025-01-19 05:54:19 -07:00

98 lines
2.0 KiB
Go

package image
import (
"os"
"path"
"path/filepath"
"github.com/warewulf/warewulf/internal/pkg/wwlog"
)
func shimDirs() []string {
return []string{
`/usr/share/efi/*/`,
`/usr/lib64/efi/`,
`/boot/efi/EFI/*/`,
}
}
func shimNames() []string {
return []string{
`shim.efi`,
`shim-sles.efi`,
`shimx64.efi`,
`shim-susesigned.efi`,
}
}
func grubDirs() []string {
return []string{
`/usr/lib64/efi/`,
`/usr/share/grub2/*-efi/`,
`/usr/share/efi/*/`,
`/boot/efi/EFI/*/`,
}
}
func grubNames() []string {
return []string{
`grub-tpm.efi`,
`grub.efi`,
`grubx64.efi`,
`grubia32.efi`,
`grubaa64.efi`,
`grubarm.efi`,
}
}
/*
find the path of the shim binary in image
*/
func ShimFind(image string) string {
var image_path string
if image != "" {
image_path = RootFsDir(image)
} else {
image_path = "/"
}
wwlog.Debug("Finding shim under paths: %s", image_path)
return BootLoaderFindPath(image_path, shimNames, shimDirs)
}
/*
find a grub.efi in the used image
*/
func GrubFind(image string) string {
var image_path string
if image != "" {
image_path = RootFsDir(image)
} else {
image_path = "/"
}
wwlog.Debug("Finding grub under paths: %s", image_path)
return BootLoaderFindPath(image_path, grubNames, grubDirs)
}
/*
find the path of the shim binary in image
*/
func BootLoaderFindPath(cpath string, names func() []string, paths func() []string) string {
for _, bdir := range paths() {
wwlog.Debug("Checking directory: %s", bdir)
for _, bname := range names() {
wwlog.Debug("Checking for bootloader name: %s", path.Join(cpath, bdir, bname))
shimPaths, err := filepath.Glob(path.Join(cpath, bdir, bname))
if err != nil {
wwlog.Debug("Got error when globing %s: %s", path.Join(cpath, bdir, bname), err)
}
for _, shimPath := range shimPaths {
wwlog.Debug("Checking for bootloader path: %s", shimPath)
// Only succeeds if shimPath exists and, if a
// symlink, links to a path that also exists
if _, err := os.Stat(shimPath); err == nil {
return shimPath
}
}
}
}
return ""
}