Validate digest length on parsing

Signed-off-by: Tonis Tiigi <tonistiigi@gmail.com>
opencontainers · May 10, 2020 · 6ac142d · 6ac142d
1 parent 45599b9
commit 6ac142d
Showing 1 changed file with 33 additions and 32 deletions.
diff --git a/digestset/set_test.go b/digestset/set_test.go
@@ -15,14 +15,14 @@ func assertEqualDigests(t *testing.T, d1, d2 Digest) {
 
 func TestLookup(t *testing.T) {
 	digests := []Digest{
-		"sha256:12345",
-		"sha256:1234",
-		"sha256:12346",
-		"sha256:54321",
-		"sha256:65431",
-		"sha256:64321",
-		"sha256:65421",
-		"sha256:65321",
+		"sha256:1234511111111111111111111111111111111111111111111111111111111111",
+		"sha256:1234111111111111111111111111111111111111111111111111111111111111",
+		"sha256:1234611111111111111111111111111111111111111111111111111111111111",
+		"sha256:5432111111111111111111111111111111111111111111111111111111111111",
+		"sha256:6543111111111111111111111111111111111111111111111111111111111111",
+		"sha256:6432111111111111111111111111111111111111111111111111111111111111",
+		"sha256:6542111111111111111111111111111111111111111111111111111111111111",
+		"sha256:6532111111111111111111111111111111111111111111111111111111111111",
 	}
 
 	dset := NewSet()
@@ -55,10 +55,12 @@ func TestLookup(t *testing.T) {
 	}
 
 	dgst, err = dset.Lookup("sha256:1234")
-	if err != nil {
+	if err == nil {
+		t.Fatal("Expected ambiguous error looking up: sha256:1234")
+	}
+	if err != ErrDigestAmbiguous {
 		t.Fatal(err)
 	}
-	assertEqualDigests(t, dgst, digests[1])
 
 	dgst, err = dset.Lookup("sha256:12345")
 	if err != nil {
@@ -87,14 +89,14 @@ func TestLookup(t *testing.T) {
 
 func TestAddDuplication(t *testing.T) {
 	digests := []Digest{
-		"sha256:1234",
-		"sha256:12345",
-		"sha256:12346",
-		"sha256:54321",
-		"sha256:65431",
-		"sha512:65431",
-		"sha512:65421",
-		"sha512:65321",
+		"sha256:1234111111111111111111111111111111111111111111111111111111111111",
+		"sha256:1234511111111111111111111111111111111111111111111111111111111111",
+		"sha256:1234611111111111111111111111111111111111111111111111111111111111",
+		"sha256:5432111111111111111111111111111111111111111111111111111111111111",
+		"sha256:6543111111111111111111111111111111111111111111111111111111111111",
+		"sha512:65431111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111",
+		"sha512:65421111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111",
+		"sha512:65321111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111",
 	}
 
 	dset := NewSet()
@@ -108,15 +110,15 @@ func TestAddDuplication(t *testing.T) {
 		t.Fatal("Invalid dset size")
 	}
 
-	if err := dset.Add(Digest("sha256:12345")); err != nil {
+	if err := dset.Add(Digest("sha256:1234511111111111111111111111111111111111111111111111111111111111")); err != nil {
 		t.Fatal(err)
 	}
 
 	if len(dset.entries) != 8 {
 		t.Fatal("Duplicate digest insert allowed")
 	}
 
-	if err := dset.Add(Digest("sha384:12345")); err != nil {
+	if err := dset.Add(Digest("sha384:123451111111111111111111111111111111111111111111111111111111111111111111111111111111111111111111")); err != nil {
 		t.Fatal(err)
 	}
 
@@ -193,14 +195,14 @@ func assertEqualShort(t *testing.T, actual, expected string) {
 
 func TestShortCodeTable(t *testing.T) {
 	digests := []Digest{
-		"sha256:1234",
-		"sha256:12345",
-		"sha256:12346",
-		"sha256:54321",
-		"sha256:65431",
-		"sha256:64321",
-		"sha256:65421",
-		"sha256:65321",
+		"sha256:1234111111111111111111111111111111111111111111111111111111111111",
+		"sha256:1234511111111111111111111111111111111111111111111111111111111111",
+		"sha256:1234611111111111111111111111111111111111111111111111111111111111",
+		"sha256:5432111111111111111111111111111111111111111111111111111111111111",
+		"sha256:6543111111111111111111111111111111111111111111111111111111111111",
+		"sha256:6432111111111111111111111111111111111111111111111111111111111111",
+		"sha256:6542111111111111111111111111111111111111111111111111111111111111",
+		"sha256:6532111111111111111111111111111111111111111111111111111111111111",
 	}
 
 	dset := NewSet()
@@ -215,10 +217,9 @@ func TestShortCodeTable(t *testing.T) {
 	if len(dump) < len(digests) {
 		t.Fatalf("Error unexpected size: %d, expecting %d", len(dump), len(digests))
 	}
-
-	assertEqualShort(t, dump[digests[0]], "sha256:1234")
-	assertEqualShort(t, dump[digests[1]], "sha256:12345")
-	assertEqualShort(t, dump[digests[2]], "sha256:12346")
+	assertEqualShort(t, dump[digests[0]], "12341")
+	assertEqualShort(t, dump[digests[1]], "12345")
+	assertEqualShort(t, dump[digests[2]], "12346")
 	assertEqualShort(t, dump[digests[3]], "54")
 	assertEqualShort(t, dump[digests[4]], "6543")
 	assertEqualShort(t, dump[digests[5]], "64")