runtime: make gostringnocopy update maxstring

Fixes #8706

LGTM=josharian
R=josharian
CC=golang-codereviews
https://golang.org/cl/143880043

3 files changed, 22 insertions, 1 deletions

diff --git a/src/runtime/export_test.go b/src/runtime/export_test.go
index 1f1b5fc794..be352557fb 100644
--- a/src/runtime/export_test.go
+++ b/src/runtime/export_test.go
@@ -160,3 +160,6 @@ func GostringW(w []uint16) (s string) {
 	})
 	return
 }
+
+var Gostringnocopy = gostringnocopy
+var Maxstring = &maxstring
diff --git a/src/runtime/string.c b/src/runtime/string.c
index 811a289060..ed5debc33e 100644
--- a/src/runtime/string.c
+++ b/src/runtime/string.c
@@ -42,10 +42,15 @@ String
 runtime·gostringnocopy(byte *str)
 {
 	String s;
+	uintptr ms;
 	
 	s.str = str;
 	s.len = runtime·findnull(str);
-	return s;
+	while(true) {
+		ms = runtime·maxstring;
+		if(s.len <= ms || runtime·casp((void**)&runtime·maxstring, (void*)ms, (void*)s.len))
+			return s;
+	}
 }
 
 // TODO: move this elsewhere
diff --git a/src/runtime/string_test.go b/src/runtime/string_test.go
index e7ac51a5f0..1551ecc82b 100644
--- a/src/runtime/string_test.go
+++ b/src/runtime/string_test.go
@@ -145,3 +145,16 @@ func main() {
 	panic(s)
 }
 `
+
+func TestGostringnocopy(t *testing.T) {
+	max := *runtime.Maxstring
+	b := make([]byte, max+10)
+	for i := uintptr(0); i < max+9; i++ {
+		b[i] = 'a'
+	}
+	_ = runtime.Gostringnocopy(&b[0])
+	newmax := *runtime.Maxstring
+	if newmax != max+9 {
+		t.Errorf("want %d, got %d", max+9, newmax)
+	}
+}