add assert to check integer overflow for table size (#1402)

dgraph-io · Jul 10, 2020 · 7f4e4b5 · 7f4e4b5
1 parent 8e896a7
commit 7f4e4b5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/table/builder.go b/table/builder.go
@@ -327,6 +327,9 @@ func (b *Builder) shouldFinishBlock(key []byte, value y.ValueStruct) bool {
 		// So, size of IV is added to estimatedSize.
 		estimatedSize += aes.BlockSize
 	}
+	// Integer overflow check for table size.
+	y.AssertTrue(uint64(b.sz)+uint64(estimatedSize) < math.MaxUint32)
+
 	return estimatedSize > uint32(b.opt.BlockSize)
 }