From b9545da71c2f5e93355d82a1f9b5ead02f2bc617 Mon Sep 17 00:00:00 2001 From: Ethan Reesor Date: Thu, 5 Mar 2026 14:20:45 -0600 Subject: testing: escapes framing markers Uses `^[` to escape the framing marker `^V` used to delimit test output. A test that itself executes a go test binary, or otherwise emits that control character, previously would corrupt the test2json parse of the enclosing run. Updates #62728. Change-Id: I0e8790a05fd7af469cd7ee2e8ccc13786cc372dc Reviewed-on: https://go-review.googlesource.com/c/go/+/751940 LUCI-TryBot-Result: Go LUCI Reviewed-by: Michael Matloob Reviewed-by: Damien Neil --- src/testing/testing.go | 49 ++++++++++++++++++++++++++++++++++++++++++++----- 1 file changed, 44 insertions(+), 5 deletions(-) (limited to 'src/testing/testing.go') diff --git a/src/testing/testing.go b/src/testing/testing.go index bf95f1cfbb..6cce1a451f 100644 --- a/src/testing/testing.go +++ b/src/testing/testing.go @@ -562,11 +562,14 @@ func (f *chattyFlag) Get() any { return f.on } -const marker = byte(0x16) // ^V for framing +const ( + markFraming byte = 'V' &^ '@' // ^V: framing + markEscape byte = '[' &^ '@' // ^[: escape +) func (f *chattyFlag) prefix() string { if f.json { - return string(marker) + return string(markFraming) } return "" } @@ -588,7 +591,7 @@ func newChattyPrinter(w io.Writer) *chattyPrinter { // that as not in json mode (because it's not chatty at all). func (p *chattyPrinter) prefix() string { if p != nil && p.json { - return string(marker) + return string(markFraming) } return "" } @@ -869,8 +872,8 @@ func (w indenter) Write(b []byte) (n int, err error) { // An indent of 4 spaces will neatly align the dashes with the status // indicator of the parent. line := b[:end] - if line[0] == marker { - w.c.output = append(w.c.output, marker) + if line[0] == markFraming { + w.c.output = append(w.c.output, markFraming) line = line[1:] } w.c.output = append(w.c.output, indent...) @@ -1166,6 +1169,9 @@ func (o *outputWriter) Write(p []byte) (int, error) { // writeLine generates the output for a given line. func (o *outputWriter) writeLine(b []byte) { if !o.c.done && (o.c.chatty != nil) { + // Escape the framing marker. + b = escapeMarkers(b) + if o.c.bench { // Benchmarks don't print === CONT, so we should skip the test // printer and just print straight to stdout. @@ -1179,6 +1185,39 @@ func (o *outputWriter) writeLine(b []byte) { o.c.output = append(o.c.output, b...) } +func escapeMarkers(b []byte) []byte { + j := nextMark(b) + if j < 0 { + // Allocation-free fast path. + return b + } + + c := make([]byte, 0, len(b)+10) + i := 0 + for i < len(b) && j >= i { + if j > i { + c = append(c, b[i:j]...) + } + c = append(c, markEscape, b[j]) + i = j + 1 + j = i + nextMark(b[i:]) + } + if i < len(b) { + c = append(c, b[i:]...) + } + return c +} + +func nextMark(b []byte) int { + for i, b := range b { + switch b { + case markFraming, markEscape: + return i + } + } + return -1 +} + // Log formats its arguments using default formatting, analogous to [fmt.Println], // and records the text in the error log. For tests, the text will be printed only if // the test fails or the -test.v flag is set. For benchmarks, the text is always -- cgit v1.3