mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-12-22 14:15:16 -05:00
109 lines
2.2 KiB
Go
109 lines
2.2 KiB
Go
//go:build integration
|
|
|
|
package integration
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"testing"
|
|
|
|
"github.com/edgelesssys/constellation/internal/logger"
|
|
"github.com/edgelesssys/constellation/state/internal/mapper"
|
|
"github.com/martinjungblut/go-cryptsetup"
|
|
"go.uber.org/zap/zapcore"
|
|
)
|
|
|
|
func BenchmarkMapper(b *testing.B) {
|
|
cryptsetup.SetDebugLevel(cryptsetup.CRYPT_LOG_ERROR)
|
|
cryptsetup.SetLogCallback(func(_ int, message string) { fmt.Println(message) })
|
|
|
|
testPath := *diskPath
|
|
if testPath == "" {
|
|
// no disk specified, use 1GB loopback disk
|
|
testPath = devicePath
|
|
if err := setup(1); err != nil {
|
|
b.Fatal("Failed to setup test environment:", err)
|
|
}
|
|
|
|
defer func() {
|
|
if err := teardown(); err != nil {
|
|
b.Fatal("failed to delete test disk:", err)
|
|
}
|
|
}()
|
|
}
|
|
|
|
passphrase := "benchmark"
|
|
mapper, err := mapper.New(testPath, logger.New(logger.PlainLog, zapcore.InfoLevel))
|
|
if err != nil {
|
|
b.Fatal("Failed to create mapper:", err)
|
|
}
|
|
defer mapper.Close()
|
|
|
|
if err := mapper.FormatDisk(passphrase); err != nil {
|
|
b.Fatal("Failed to format disk:", err)
|
|
}
|
|
|
|
testCases := map[string]struct {
|
|
wipeBlockSize int
|
|
}{
|
|
"16KiB": {
|
|
wipeBlockSize: int(math.Pow(2, 14)),
|
|
},
|
|
"32KiB": {
|
|
wipeBlockSize: int(math.Pow(2, 15)),
|
|
},
|
|
"64KiB": {
|
|
wipeBlockSize: int(math.Pow(2, 16)),
|
|
},
|
|
"128KiB": {
|
|
wipeBlockSize: int(math.Pow(2, 17)),
|
|
},
|
|
"256KiB": {
|
|
wipeBlockSize: int(math.Pow(2, 18)),
|
|
},
|
|
"512KiB": {
|
|
wipeBlockSize: int(math.Pow(2, 19)),
|
|
},
|
|
"1MiB": {
|
|
wipeBlockSize: int(math.Pow(2, 20)),
|
|
},
|
|
"2MiB": {
|
|
wipeBlockSize: int(math.Pow(2, 21)),
|
|
},
|
|
"4MiB": {
|
|
wipeBlockSize: int(math.Pow(2, 22)),
|
|
},
|
|
"8MiB": {
|
|
wipeBlockSize: int(math.Pow(2, 23)),
|
|
},
|
|
"16MiB": {
|
|
wipeBlockSize: int(math.Pow(2, 24)),
|
|
},
|
|
"32MiB": {
|
|
wipeBlockSize: int(math.Pow(2, 25)),
|
|
},
|
|
"64MiB": {
|
|
wipeBlockSize: int(math.Pow(2, 26)),
|
|
},
|
|
"128MiB": {
|
|
wipeBlockSize: int(math.Pow(2, 27)),
|
|
},
|
|
"256MiB": {
|
|
wipeBlockSize: int(math.Pow(2, 28)),
|
|
},
|
|
"512MiB": {
|
|
wipeBlockSize: int(math.Pow(2, 29)),
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
b.Run(name, func(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
if err := mapper.Wipe(tc.wipeBlockSize); err != nil {
|
|
b.Fatal("Failed to wipe disk:", err)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|