AB#2260 Refactor disk-mapper recovery (#82)

* Refactor disk-mapper recovery

* Adapt constellation recover command to use new disk-mapper recovery API

* Fix Cilium connectivity on rebooting nodes (#89)

* Lower CoreDNS reschedule timeout to 10 seconds (#93)

Signed-off-by: Daniel Weiße <dw@edgeless.systems>

disk-mapper/internal/test/benchmark_test.go

@@ -0,0 +1,114 @@
+//go:build integration
+
+/*
+Copyright (c) Edgeless Systems GmbH
+
+SPDX-License-Identifier: AGPL-3.0-only
+*/
+
+package integration
+
+import (
+	"fmt"
+	"math"
+	"testing"
+
+	"github.com/edgelesssys/constellation/disk-mapper/internal/mapper"
+	"github.com/edgelesssys/constellation/internal/logger"
+	"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)
+				}
+			}
+		})
+	}
+}