mysql/replication: bound the intervals preallocation in ParseMysql56GTIDSet

go/mysql/replication/mysql56_gtid_set_test.go

@@ -17,8 +17,10 @@ limitations under the License.
 package replication
 
 import (
+	"fmt"
 	"maps"
 	"reflect"
+	"runtime"
 	"strings"
 	"testing"
 
@@ -902,3 +904,40 @@ func TestSIDs(t *testing.T) {
 	require.Len(t, sids, 1)
 	assert.Equal(t, "8bc65cca-3fe4-11ed-bbfb-091034d48b3e", sids[0].String())
 }
+
+// TestParseMysql56GTIDSetIntervalsCapHint checks that the preallocation hint for
+// the intervals slice does not follow a count taken from the input beyond what that
+// input could hold, while a genuinely long interval list still parses unchanged.
+// The hostile case asserts on allocation volume, because a colon run parses without
+// error either way once the intervals are all discarded.
+func TestParseMysql56GTIDSetIntervalsCapHint(t *testing.T) {
+	const sid = "00010203-0405-0607-0809-0a0b0c0d0e0f"
+
+	// Results must be identical either side of the bound.
+	for _, n := range []int{1, 10, 100, 1023, 1024, 1025, 2051, 5000} {
+		var sb strings.Builder
+		sb.WriteString(sid)
+		for i := 0; i < n; i++ {
+			// non-overlapping ascending intervals, so none are merged or discarded
+			fmt.Fprintf(&sb, ":%d-%d", 2*i+1, 2*i+1)
+		}
+		got, err := ParseMysql56GTIDSet(sb.String())
+		require.NoError(t, err, "n=%d", n)
+		sidVal, err := ParseSID(sid)
+		require.NoError(t, err)
+		assert.Len(t, got[sidVal], n, "n=%d", n)
+	}
+
+	// A long run of colons carries no intervals at all, so reserving one per colon
+	// is pure waste. 1MiB of colons is 16MiB of interval structs unbounded.
+	hostile := sid + ":" + strings.Repeat(":", 1<<20)
+	var before, after runtime.MemStats
+	runtime.GC()
+	runtime.ReadMemStats(&before)
+	_, _ = ParseMysql56GTIDSet(hostile)
+	runtime.ReadMemStats(&after)
+	allocated := after.TotalAlloc - before.TotalAlloc
+	assert.Less(t, allocated, uint64(8<<20),
+		"parsing %d colons allocated %d bytes for intervals that are all discarded",
+		1<<20, allocated)
+}