mirror of
https://github.com/hanzoai/visor.git
synced 2026-09-24 17:04:14 +00:00
This moves the initialization of `cpuid.hostFeatureSet` out of package-level `init` and instead moves it to an explicit `cpuid.Initialize()` function. On AMD64, this saves about 512KiB of heap memory that would otherwise always be live. PiperOrigin-RevId: 514896323
76 lines
2.1 KiB
Go
76 lines
2.1 KiB
Go
// Copyright 2018 The gVisor Authors.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package cpuid
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
"testing"
|
|
|
|
"gvisor.dev/gvisor/pkg/hostos"
|
|
)
|
|
|
|
// TestHostFeatureFlags tests that all features detected by HostFeatureSet are
|
|
// on the host.
|
|
//
|
|
// It does *not* verify that all features reported by the host are detected by
|
|
// HostFeatureSet. Linux has synthetic Linux-specific features that have no
|
|
// analog in the actual CPUID feature set.
|
|
func TestHostFeatureFlags(t *testing.T) {
|
|
// Extract the kernel version.
|
|
version, err := hostos.KernelVersion()
|
|
if err != nil {
|
|
t.Fatalf("Unable to parse kernel version: %v", err)
|
|
}
|
|
|
|
// Extract all cpuinfo flags.
|
|
cpuinfoBytes, _ := ioutil.ReadFile("/proc/cpuinfo")
|
|
cpuinfo := string(cpuinfoBytes)
|
|
re := regexp.MustCompile(`(?m)^flags\s+: (.*)$`)
|
|
m := re.FindStringSubmatch(cpuinfo)
|
|
if len(m) != 2 {
|
|
t.Fatalf("Unable to extract flags from %q", cpuinfo)
|
|
}
|
|
cpuinfoFlags := make(map[string]struct{})
|
|
for _, f := range strings.Split(m[1], " ") {
|
|
cpuinfoFlags[f] = struct{}{}
|
|
}
|
|
|
|
// Check against host flags.
|
|
fs := HostFeatureSet()
|
|
for feature, info := range allFeatures {
|
|
// Special cases not consistently visible. We don't mind if
|
|
// they are exposed in earlier versions.
|
|
if archSkipFeature(feature, version) {
|
|
continue
|
|
}
|
|
|
|
// Check against the flags.
|
|
_, ok := cpuinfoFlags[feature.String()]
|
|
if !info.shouldAppear && ok {
|
|
t.Errorf("Unexpected flag: %v", feature)
|
|
} else if info.shouldAppear && fs.HasFeature(feature) && !ok {
|
|
t.Errorf("Missing flag: %v", feature)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestMain(m *testing.M) {
|
|
Initialize()
|
|
os.Exit(m.Run())
|
|
}
|