1 files changed, 99 insertions, 0 deletions
diff --git a/pkg/sentry/fsimpl/cgroupfs/bitmap_test.go b/pkg/sentry/fsimpl/cgroupfs/bitmap_test.go
new file mode 100644
index 000000000..5cc56de3b
--- /dev/null
+++ b/pkg/sentry/fsimpl/cgroupfs/bitmap_test.go
@@ -0,0 +1,99 @@
+// Copyright 2021 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 cgroupfs
+
+import (
+	"fmt"
+	"reflect"
+	"testing"
+
+	"gvisor.dev/gvisor/pkg/bitmap"
+)
+
+func TestFormat(t *testing.T) {
+	tests := []struct {
+		input  []uint32
+		output string
+	}{
+		{[]uint32{1, 2, 3, 4, 7}, "1-4,7"},
+		{[]uint32{2}, "2"},
+		{[]uint32{0, 1, 2}, "0-2"},
+		{[]uint32{}, ""},
+		{[]uint32{1, 3, 4, 5, 6, 9, 11, 13, 14, 15, 16, 17}, "1,3-6,9,11,13-17"},
+		{[]uint32{2, 3, 10, 12, 13, 14, 15, 16, 20, 21, 33, 34, 47}, "2-3,10,12-16,20-21,33-34,47"},
+	}
+	for i, tt := range tests {
+		t.Run(fmt.Sprintf("case-%d", i), func(t *testing.T) {
+			b := bitmap.New(64)
+			for _, v := range tt.input {
+				b.Add(v)
+			}
+			s := formatBitmap(&b)
+			if s != tt.output {
+				t.Errorf("Expected %q, got %q", tt.output, s)
+			}
+			b1, err := parseBitmap(s, 64)
+			if err != nil {
+				t.Fatalf("Failed to parse formatted bitmap: %v", err)
+			}
+			if got, want := b1.ToSlice(), b.ToSlice(); !reflect.DeepEqual(got, want) {
+				t.Errorf("Parsing formatted output doesn't result in the original bitmap. Got %v, want %v", got, want)
+			}
+		})
+	}
+}
+
+func TestParse(t *testing.T) {
+	tests := []struct {
+		input      string
+		output     []uint32
+		shouldFail bool
+	}{
+		{"1", []uint32{1}, false},
+		{"", []uint32{}, false},
+		{"1,2,3,4", []uint32{1, 2, 3, 4}, false},
+		{"1-4", []uint32{1, 2, 3, 4}, false},
+		{"1,2-4", []uint32{1, 2, 3, 4}, false},
+		{"1,2-3,4", []uint32{1, 2, 3, 4}, false},
+		{"1-2,3,4,10,11", []uint32{1, 2, 3, 4, 10, 11}, false},
+		{"1,2-4,5,16", []uint32{1, 2, 3, 4, 5, 16}, false},
+		{"abc", []uint32{}, true},
+		{"1,3-2,4", []uint32{}, true},
+		{"1,3-3,4", []uint32{}, true},
+		{"1,2,3\000,4", []uint32{1, 2, 3}, false},
+		{"1,2,3\n,4", []uint32{1, 2, 3}, false},
+	}
+	for i, tt := range tests {
+		t.Run(fmt.Sprintf("case-%d", i), func(t *testing.T) {
+			b, err := parseBitmap(tt.input, 64)
+			if tt.shouldFail {
+				if err == nil {
+					t.Fatalf("Expected parsing of %q to fail, but it didn't", tt.input)
+				}
+				return
+			}
+			if err != nil {
+				t.Fatalf("Failed to parse bitmap: %v", err)
+				return
+			}
+
+			got := b.ToSlice()
+			if !reflect.DeepEqual(got, tt.output) {
+				t.Errorf("Parsed bitmap doesn't match what we expected. Got %v, want %v", got, tt.output)
+			}
+
+		})
+	}
+}