Andrey Smirnov 519a48302e
fix: wipe system partitions correctly via kernel args
Use `DiscoveredVolumes` instead of `VolumeStatus`, force reboot to avoid
confusion in the volume controller.

Fixes #9448

Signed-off-by: Andrey Smirnov <andrey.smirnov@siderolabs.com>
2024-10-07 23:16:01 +04:00

90 lines
2.4 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 partition
import (
"context"
"fmt"
"time"
"github.com/siderolabs/go-blockdevice/v2/block"
blockres "github.com/siderolabs/talos/pkg/machinery/resources/block"
)
// VolumeWipeTarget is a target for wiping a volume.
type VolumeWipeTarget struct {
label string
parentDevName, devName string
}
// VolumeWipeTargetFromVolumeStatus creates a new VolumeWipeTarget from a VolumeStatus.
func VolumeWipeTargetFromVolumeStatus(vs *blockres.VolumeStatus) *VolumeWipeTarget {
parentDevName := vs.TypedSpec().ParentLocation
if parentDevName == "" {
parentDevName = vs.TypedSpec().Location
}
return &VolumeWipeTarget{
label: vs.Metadata().ID(),
parentDevName: parentDevName,
devName: vs.TypedSpec().Location,
}
}
// VolumeWipeTargetFromDiscoveredVolume creates a new VolumeWipeTarget from a DiscoveredVolume.
func VolumeWipeTargetFromDiscoveredVolume(dv *blockres.DiscoveredVolume) *VolumeWipeTarget {
parentDevName := dv.TypedSpec().ParentDevPath
if parentDevName == "" {
parentDevName = dv.TypedSpec().DevPath
}
return &VolumeWipeTarget{
label: dv.TypedSpec().PartitionLabel,
parentDevName: parentDevName,
devName: dv.TypedSpec().DevPath,
}
}
// GetLabel implements runtime.PartitionTarget.
func (v *VolumeWipeTarget) GetLabel() string {
return v.label
}
// String implements runtime.PartitionTarget.
func (v *VolumeWipeTarget) String() string {
return fmt.Sprintf("%s:%s", v.label, v.devName)
}
// Wipe implements runtime.PartitionTarget.
func (v *VolumeWipeTarget) Wipe(ctx context.Context, log func(string, ...any)) error {
parentBd, err := block.NewFromPath(v.parentDevName)
if err != nil {
return fmt.Errorf("error opening block device %q: %w", v.parentDevName, err)
}
defer parentBd.Close() //nolint:errcheck
if err = parentBd.RetryLockWithTimeout(ctx, true, time.Minute); err != nil {
return fmt.Errorf("error locking block device %q: %w", v.parentDevName, err)
}
defer parentBd.Unlock() //nolint:errcheck
bd, err := block.NewFromPath(v.devName, block.OpenForWrite())
if err != nil {
return fmt.Errorf("error opening block device %q: %w", v.devName, err)
}
defer bd.Close() //nolint:errcheck
log("wiping the volume %q (%s)", v.GetLabel(), v.devName)
return bd.FastWipe()
}