mirror of
https://github.com/kubernetes-sigs/node-feature-discovery.git
synced 2024-12-14 11:57:51 +00:00
102 lines
2.8 KiB
Go
102 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os/exec"
|
|
"path"
|
|
|
|
"github.com/klauspost/cpuid"
|
|
)
|
|
|
|
// FeatureSource represents a source of discovered node features.
|
|
type FeatureSource interface {
|
|
// Returns a friendly name for this source of node features.
|
|
Name() string
|
|
|
|
// Returns discovered features for this node.
|
|
Discover() ([]string, error)
|
|
}
|
|
|
|
const (
|
|
// RDTBin is the path to RDT detection helpers.
|
|
RDTBin = "/go/src/github.com/kubernetes-incubator/node-feature-discovery/rdt-discovery"
|
|
)
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// CPUID Source
|
|
|
|
// Implements main.FeatureSource.
|
|
type cpuidSource struct{}
|
|
|
|
func (s cpuidSource) Name() string { return "cpuid" }
|
|
|
|
// Returns feature names for all the supported CPU features.
|
|
func (s cpuidSource) Discover() ([]string, error) {
|
|
// Get the cpu features as strings
|
|
return cpuid.CPU.Features.Strings(), nil
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// RDT (Intel Resource Director Technology) Source
|
|
|
|
// Implements main.FeatureSource.
|
|
type rdtSource struct{}
|
|
|
|
func (s rdtSource) Name() string { return "rdt" }
|
|
|
|
// Returns feature names for CMT and CAT if suppported.
|
|
func (s rdtSource) Discover() ([]string, error) {
|
|
features := []string{}
|
|
|
|
cmd := exec.Command("bash", "-c", path.Join(RDTBin, "mon-discovery"))
|
|
if err := cmd.Run(); err != nil {
|
|
stderrLogger.Printf("support for RDT monitoring was not detected: %s", err.Error())
|
|
} else {
|
|
// RDT monitoring detected.
|
|
features = append(features, "RDTMON")
|
|
}
|
|
|
|
cmd = exec.Command("bash", "-c", path.Join(RDTBin, "l3-alloc-discovery"))
|
|
if err := cmd.Run(); err != nil {
|
|
stderrLogger.Printf("support for RDT L3 allocation was not detected: %s", err.Error())
|
|
} else {
|
|
// RDT L3 cache allocation detected.
|
|
features = append(features, "RDTL3CA")
|
|
}
|
|
|
|
cmd = exec.Command("bash", "-c", path.Join(RDTBin, "l2-alloc-discovery"))
|
|
if err := cmd.Run(); err != nil {
|
|
stderrLogger.Printf("support for RDT L2 allocation was not detected: %s", err.Error())
|
|
} else {
|
|
// RDT L2 cache allocation detected.
|
|
features = append(features, "RDTL2CA")
|
|
}
|
|
|
|
return features, nil
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// PState Source
|
|
|
|
// Implements main.FeatureSource.
|
|
type pstateSource struct{}
|
|
|
|
func (s pstateSource) Name() string { return "pstate" }
|
|
|
|
// Returns feature names for p-state related features such as turbo boost.
|
|
func (s pstateSource) Discover() ([]string, error) {
|
|
features := []string{}
|
|
|
|
// Only looking for turbo boost for now...
|
|
bytes, err := ioutil.ReadFile("/sys/devices/system/cpu/intel_pstate/no_turbo")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("can't detect whether turbo boost is enabled: %s", err.Error())
|
|
}
|
|
if bytes[0] == byte('0') {
|
|
// Turbo boost is enabled.
|
|
features = append(features, "turbo")
|
|
}
|
|
|
|
return features, nil
|
|
}
|