mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-31 08:21:25 +01:00
c5bd0ac5cf
Drop loop device/mounts completely, use userspace utilities to extract and lay over module trees in the tmpfs. Discover kernel version automatically instead of hardcoding it to be current one (required for Image Service). Signed-off-by: Andrey Smirnov <andrey.smirnov@siderolabs.com>
111 lines
3.0 KiB
Go
111 lines
3.0 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
// Package extensions provides facilities for building initramfs.xz with extensions.
|
|
package extensions
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/siderolabs/talos/internal/pkg/extensions"
|
|
"github.com/siderolabs/talos/pkg/machinery/constants"
|
|
extinterface "github.com/siderolabs/talos/pkg/machinery/extensions"
|
|
)
|
|
|
|
// Builder rebuilds initramfs.xz with extensions.
|
|
type Builder struct {
|
|
// The initramfs will be rebuilt in-place.
|
|
InitramfsPath string
|
|
// Architecture of the initramfs.
|
|
Arch string
|
|
// ExtensionTreePath is a path to the extracted exension tree.
|
|
ExtensionTreePath string
|
|
// Printf is used for logging.
|
|
Printf func(format string, v ...any)
|
|
}
|
|
|
|
// Build rebuilds the initramfs.xz with extensions.
|
|
func (builder *Builder) Build() error {
|
|
extensionsList, err := extensions.List(builder.ExtensionTreePath)
|
|
if err != nil {
|
|
return fmt.Errorf("error listing extensions: %w", err)
|
|
}
|
|
|
|
if len(extensionsList) == 0 {
|
|
return nil
|
|
}
|
|
|
|
if err = builder.printExtensions(extensionsList); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err = builder.validateExtensions(extensionsList); err != nil {
|
|
return err
|
|
}
|
|
|
|
extensionsPathWithKernelModules := findExtensionsWithKernelModules(extensionsList)
|
|
|
|
if len(extensionsPathWithKernelModules) > 0 {
|
|
kernelModuleDepExtension, genErr := extensions.GenerateKernelModuleDependencyTreeExtension(extensionsPathWithKernelModules, builder.InitramfsPath, builder.Printf)
|
|
if genErr != nil {
|
|
return genErr
|
|
}
|
|
|
|
extensionsList = append(extensionsList, kernelModuleDepExtension)
|
|
}
|
|
|
|
tempDir, err := os.MkdirTemp("", "ext")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer os.RemoveAll(tempDir) //nolint:errcheck
|
|
|
|
var cfg *extinterface.Config
|
|
|
|
if cfg, err = builder.compressExtensions(extensionsList, tempDir); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err = cfg.Write(filepath.Join(tempDir, constants.ExtensionsConfigFile)); err != nil {
|
|
return err
|
|
}
|
|
|
|
return builder.rebuildInitramfs(tempDir)
|
|
}
|
|
|
|
func (builder *Builder) validateExtensions(extensions []*extensions.Extension) error {
|
|
builder.Printf("validating system extensions")
|
|
|
|
for _, ext := range extensions {
|
|
if err := ext.Validate(); err != nil {
|
|
return fmt.Errorf("error validating extension %q: %w", ext.Manifest.Metadata.Name, err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (builder *Builder) compressExtensions(extensions []*extensions.Extension, tempDir string) (*extinterface.Config, error) {
|
|
cfg := &extinterface.Config{}
|
|
|
|
builder.Printf("compressing system extensions")
|
|
|
|
for _, ext := range extensions {
|
|
path, err := ext.Compress(tempDir, tempDir)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error compressing extension %q: %w", ext.Manifest.Metadata.Name, err)
|
|
}
|
|
|
|
cfg.Layers = append(cfg.Layers, &extinterface.Layer{
|
|
Image: filepath.Base(path),
|
|
Metadata: ext.Manifest.Metadata,
|
|
})
|
|
}
|
|
|
|
return cfg, nil
|
|
}
|