treewide: introduce osbase package and move things around
All except localregistry moved from metropolis/pkg to osbase,
localregistry moved to metropolis/test as its only used there anyway.
Change-Id: If1a4bf377364bef0ac23169e1b90379c71b06d72
Reviewed-on: https://review.monogon.dev/c/monogon/+/3079
Tested-by: Jenkins CI
Reviewed-by: Serge Bazanski <serge@monogon.tech>
diff --git a/osbase/gpt/gpt_test.go b/osbase/gpt/gpt_test.go
new file mode 100644
index 0000000..e662e69
--- /dev/null
+++ b/osbase/gpt/gpt_test.go
@@ -0,0 +1,136 @@
+package gpt
+
+import (
+ "bytes"
+ "crypto/sha256"
+ "io"
+ "os"
+ "testing"
+
+ "github.com/google/uuid"
+
+ "source.monogon.dev/osbase/blockdev"
+)
+
+func TestFreeSpaces(t *testing.T) {
+ cases := []struct {
+ name string
+ parts []*Partition
+ expected [][2]int64
+ expectedOverlap bool
+ }{
+ {"Empty", []*Partition{}, [][2]int64{{34, 2015}}, false},
+ {"OnePart", []*Partition{
+ {Type: PartitionTypeEFISystem, FirstBlock: 200, LastBlock: 1499},
+ }, [][2]int64{
+ {34, 200},
+ {1500, 2015},
+ }, false},
+ {"TwoOverlappingParts", []*Partition{
+ {Type: PartitionTypeEFISystem, FirstBlock: 200, LastBlock: 1499},
+ {Type: PartitionTypeEFISystem, FirstBlock: 1000, LastBlock: 1999},
+ }, [][2]int64{
+ {34, 200},
+ {2000, 2015},
+ }, true},
+ {"Full", []*Partition{
+ {Type: PartitionTypeEFISystem, FirstBlock: 34, LastBlock: 999},
+ {Type: PartitionTypeEFISystem, FirstBlock: 1000, LastBlock: 2014},
+ }, [][2]int64{}, false},
+ {"TwoSpacedParts", []*Partition{
+ {Type: PartitionTypeEFISystem, FirstBlock: 500, LastBlock: 899},
+ {Type: PartitionTypeEFISystem, FirstBlock: 1200, LastBlock: 1799},
+ }, [][2]int64{
+ {34, 500},
+ {900, 1200},
+ {1800, 2015},
+ }, false},
+ }
+
+ // Partitions are created manually as AddPartition calls FreeSpaces itself,
+ // which makes the test unreliable as well as making failures very hard to
+ // debug.
+ for _, c := range cases {
+ t.Run(c.name, func(t *testing.T) {
+ d := blockdev.MustNewMemory(512, 2048) // 1MiB
+ g, err := New(d)
+ if err != nil {
+ panic(err)
+ }
+ g.Partitions = c.parts
+ fs, overlap, err := g.GetFreeSpaces()
+ if err != nil {
+ t.Fatal(err)
+ }
+ if overlap != c.expectedOverlap {
+ t.Errorf("expected overlap %v, got %v", c.expectedOverlap, overlap)
+ }
+ if len(fs) != len(c.expected) {
+ t.Fatalf("expected %v, got %v", c.expected, fs)
+ }
+ for i := range fs {
+ if fs[i] != c.expected[i] {
+ t.Errorf("free space mismatch at pos %d: got [%d, %d), expected [%d, %d)", i, fs[i][0], fs[i][1], c.expected[i][0], c.expected[i][1])
+ }
+ }
+ })
+ }
+}
+
+func TestRoundTrip(t *testing.T) {
+ if os.Getenv("IN_KTEST") == "true" {
+ t.Skip("In ktest")
+ }
+ d := blockdev.MustNewMemory(512, 2048) // 1 MiB
+
+ g := Table{
+ ID: uuid.NewSHA1(uuid.Nil, []byte("test")),
+ BootCode: []byte("just some test code"),
+ Partitions: []*Partition{
+ nil,
+ // This emoji is very complex and exercises UTF16 surrogate encoding
+ // and composing.
+ {Name: "Test 🏃♂️", FirstBlock: 10, LastBlock: 19, Type: PartitionTypeEFISystem, ID: uuid.NewSHA1(uuid.Nil, []byte("test1")), Attributes: AttrNoBlockIOProto},
+ nil,
+ {Name: "Test2", FirstBlock: 20, LastBlock: 49, Type: PartitionTypeEFISystem, ID: uuid.NewSHA1(uuid.Nil, []byte("test2")), Attributes: 0},
+ },
+ b: d,
+ }
+ if err := g.Write(); err != nil {
+ t.Fatalf("Error while writing Table: %v", err)
+ }
+
+ originalHash := sha256.New()
+ sr1 := io.NewSectionReader(d, 0, d.BlockSize()*d.BlockCount())
+ if _, err := io.CopyBuffer(originalHash, sr1, make([]byte, d.OptimalBlockSize())); err != nil {
+ panic(err)
+ }
+
+ g2, err := Read(d)
+ if err != nil {
+ t.Fatalf("Failed to read back GPT: %v", err)
+ }
+ if g2.ID != g.ID {
+ t.Errorf("Disk UUID changed when reading back: %v", err)
+ }
+ // Destroy primary GPT
+ d.Zero(1*d.BlockSize(), 5*d.BlockSize())
+ g3, err := Read(d)
+ if err != nil {
+ t.Fatalf("Failed to read back GPT with primary GPT destroyed: %v", err)
+ }
+ if g3.ID != g.ID {
+ t.Errorf("Disk UUID changed when reading back: %v", err)
+ }
+ if err := g3.Write(); err != nil {
+ t.Fatalf("Error while writing back GPT: %v", err)
+ }
+ rewrittenHash := sha256.New()
+ sr2 := io.NewSectionReader(d, 0, d.BlockSize()*d.BlockCount())
+ if _, err := io.CopyBuffer(rewrittenHash, sr2, make([]byte, d.OptimalBlockSize())); err != nil {
+ panic(err)
+ }
+ if !bytes.Equal(originalHash.Sum(nil), rewrittenHash.Sum(nil)) {
+ t.Errorf("Write/Read/Write test was not reproducible: %x != %x", originalHash.Sum(nil), rewrittenHash.Sum(nil))
+ }
+}