mirror of
https://github.com/hanzoai/visor.git
synced 2026-09-24 17:04:14 +00:00
Etienne PerotandgVisor bot
d53fe574e0
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
98 lines
2.6 KiB
Go
98 lines
2.6 KiB
Go
// Copyright 2020 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 (
|
|
"encoding/binary"
|
|
"math/rand"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestFeatureFromString(t *testing.T) {
|
|
// Check that known features do match.
|
|
for feature := range allFeatures {
|
|
f, ok := FeatureFromString(feature.String())
|
|
if f != feature || !ok {
|
|
t.Errorf("got %v, %v want %v, true", f, ok, feature)
|
|
}
|
|
}
|
|
|
|
// Check that "bad" doesn't match.
|
|
f, ok := FeatureFromString("bad")
|
|
if ok {
|
|
t.Errorf("got %v, %v want false", f, ok)
|
|
}
|
|
}
|
|
|
|
func TestReadHwCap(t *testing.T) {
|
|
// Make an auxv with fake entries
|
|
const (
|
|
auxvEntries = 16
|
|
uint64Size = 8
|
|
)
|
|
auxv := [auxvEntries * uint64Size * 2]byte{}
|
|
|
|
hwCap1Idx := rand.Intn(auxvEntries)
|
|
hwCap2Idx := rand.Intn(auxvEntries)
|
|
if hwCap1Idx == hwCap2Idx {
|
|
hwCap2Idx = (hwCap2Idx + 1 + rand.Intn(auxvEntries-2)) % auxvEntries
|
|
}
|
|
// Set the entries we are interested in to not 0.
|
|
hwCap1Val := 1 + uint64(rand.Int63())
|
|
hwCap2Val := 1 + uint64(rand.Int63())
|
|
|
|
binary.LittleEndian.PutUint64(auxv[hwCap1Idx*uint64Size*2:], _AT_HWCAP)
|
|
binary.LittleEndian.PutUint64(auxv[hwCap1Idx*uint64Size*2+8:], hwCap1Val)
|
|
binary.LittleEndian.PutUint64(auxv[hwCap2Idx*uint64Size*2:], _AT_HWCAP2)
|
|
binary.LittleEndian.PutUint64(auxv[hwCap2Idx*uint64Size*2+8:], hwCap2Val)
|
|
|
|
file, err := os.CreateTemp(t.TempDir(), "fake-self-auxv")
|
|
if err != nil {
|
|
t.Errorf("failed to create file: %v", err)
|
|
}
|
|
_, err = file.Write(auxv[:])
|
|
if err != nil {
|
|
t.Errorf("failed to write to file: %v", err)
|
|
}
|
|
err = file.Close()
|
|
if err != nil {
|
|
t.Errorf("failed to close file: %v", err)
|
|
}
|
|
|
|
c, err := readHWCap(file.Name())
|
|
if err != nil {
|
|
t.Errorf("readHwCap got err %v, want nil", err)
|
|
}
|
|
if c.hwCap1 != hwCap1Val {
|
|
t.Errorf("c.hwCap1 got %d, want %d", c.hwCap1, hwCap1Val)
|
|
}
|
|
if c.hwCap2 != hwCap2Val {
|
|
t.Errorf("c.hwCap2 got %d, want %d", c.hwCap1, hwCap1Val)
|
|
}
|
|
}
|
|
|
|
func TestReadingSelfProcAuxv(t *testing.T) {
|
|
_, err := readHWCap("/proc/self/auxv")
|
|
if err != nil {
|
|
t.Errorf("got %v, expected nil", err)
|
|
}
|
|
}
|
|
|
|
func TestMain(m *testing.M) {
|
|
Initialize()
|
|
os.Exit(m.Run())
|
|
}
|