mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-31 00:11:36 +01:00
22f375300c
Fix all discovered issues. Detected couple bugs, fixed them as well. Signed-off-by: Artem Chernyshev <artem.0xD2@gmail.com>
57 lines
1.3 KiB
Go
57 lines
1.3 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 keys
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/talos-systems/go-smbios/smbios"
|
|
)
|
|
|
|
// NodeIDKeyHandler generates the key based on current node information
|
|
// and provided template string.
|
|
type NodeIDKeyHandler struct{}
|
|
|
|
// NewNodeIDKeyHandler creates new NodeIDKeyHandler.
|
|
func NewNodeIDKeyHandler() (*NodeIDKeyHandler, error) {
|
|
return &NodeIDKeyHandler{}, nil
|
|
}
|
|
|
|
// GetKey implements KeyHandler interface.
|
|
func (h *NodeIDKeyHandler) GetKey(options ...KeyOption) ([]byte, error) {
|
|
opts, err := NewDefaultOptions(options)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
s, err := smbios.New()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
machineUUID, err := s.SystemInformation().UUID()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if machineUUID == uuid.Nil {
|
|
return nil, fmt.Errorf("machine UUID is not populated %s", machineUUID)
|
|
}
|
|
|
|
id := machineUUID.String()
|
|
|
|
// primitive entropy check
|
|
counts := map[rune]int{}
|
|
for _, s := range id {
|
|
counts[s]++
|
|
if counts[s] > len(id)/2 {
|
|
return nil, fmt.Errorf("machine UUID %s entropy check failed", machineUUID)
|
|
}
|
|
}
|
|
|
|
return []byte(id + opts.PartitionLabel), nil
|
|
}
|