mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-28 15:01:13 +01:00
e912c0dfcf
Use the `go-blockdevice` library to zero partitions. Also added a test that writes `ones` to the partition and verifies its zeroes after zeroing it. Signed-off-by: Noel Georgi <git@frezbo.dev>
116 lines
3.1 KiB
Go
116 lines
3.1 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 provides common utils for system partition format.
|
|
package partition
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/siderolabs/go-blockdevice/blockdevice"
|
|
|
|
"github.com/siderolabs/talos/pkg/machinery/constants"
|
|
"github.com/siderolabs/talos/pkg/makefs"
|
|
)
|
|
|
|
// FormatOptions contains format parameters.
|
|
type FormatOptions struct {
|
|
Label string
|
|
PartitionType Type
|
|
FileSystemType FileSystemType
|
|
Size uint64
|
|
Force bool
|
|
}
|
|
|
|
// NewFormatOptions creates a new format options.
|
|
func NewFormatOptions(label string) *FormatOptions {
|
|
opts, ok := systemPartitions[label]
|
|
if ok {
|
|
return &opts
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Format zeroes the device and formats it using filesystem type provided.
|
|
func Format(devname string, t *FormatOptions) error {
|
|
if t.FileSystemType == FilesystemTypeNone {
|
|
return zeroPartition(devname)
|
|
}
|
|
|
|
opts := []makefs.Option{makefs.WithForce(t.Force), makefs.WithLabel(t.Label)}
|
|
log.Printf("formatting the partition %q as %q with label %q\n", devname, t.FileSystemType, t.Label)
|
|
|
|
switch t.FileSystemType {
|
|
case FilesystemTypeVFAT:
|
|
return makefs.VFAT(devname, opts...)
|
|
case FilesystemTypeXFS:
|
|
return makefs.XFS(devname, opts...)
|
|
default:
|
|
return fmt.Errorf("unsupported filesystem type: %q", t.FileSystemType)
|
|
}
|
|
}
|
|
|
|
// zeroPartition fills the partition with zeroes.
|
|
func zeroPartition(devname string) (err error) {
|
|
log.Printf("zeroing out %q", devname)
|
|
|
|
part, err := blockdevice.Open(devname, blockdevice.WithExclusiveLock(true))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer part.Close() //nolint:errcheck
|
|
|
|
_, err = part.Wipe()
|
|
|
|
return err
|
|
}
|
|
|
|
var systemPartitions = map[string]FormatOptions{
|
|
constants.EFIPartitionLabel: {
|
|
Label: constants.EFIPartitionLabel,
|
|
PartitionType: EFISystemPartition,
|
|
FileSystemType: FilesystemTypeVFAT,
|
|
Size: EFISize,
|
|
Force: true,
|
|
},
|
|
constants.BIOSGrubPartitionLabel: {
|
|
Label: constants.BIOSGrubPartitionLabel,
|
|
PartitionType: BIOSBootPartition,
|
|
FileSystemType: FilesystemTypeNone,
|
|
Size: BIOSGrubSize,
|
|
Force: true,
|
|
},
|
|
constants.BootPartitionLabel: {
|
|
Label: constants.BootPartitionLabel,
|
|
PartitionType: LinuxFilesystemData,
|
|
FileSystemType: FilesystemTypeXFS,
|
|
Size: BootSize,
|
|
Force: true,
|
|
},
|
|
constants.MetaPartitionLabel: {
|
|
Label: constants.MetaPartitionLabel,
|
|
PartitionType: LinuxFilesystemData,
|
|
FileSystemType: FilesystemTypeNone,
|
|
Size: MetaSize,
|
|
Force: true,
|
|
},
|
|
constants.StatePartitionLabel: {
|
|
Label: constants.StatePartitionLabel,
|
|
PartitionType: LinuxFilesystemData,
|
|
FileSystemType: FilesystemTypeXFS,
|
|
Size: StateSize,
|
|
Force: true,
|
|
},
|
|
constants.EphemeralPartitionLabel: {
|
|
Label: constants.EphemeralPartitionLabel,
|
|
PartitionType: LinuxFilesystemData,
|
|
FileSystemType: FilesystemTypeXFS,
|
|
Size: 0,
|
|
Force: true,
|
|
},
|
|
}
|