Simplify unpackString (#1012)

msg_helpers.go

@@ -269,33 +269,32 @@ func unpackString(msg []byte, off int) (string, int, error) {
 	if off+l > len(msg) {
 		return "", off, &Error{err: "overflow unpacking txt"}
 	}
-	escapedLen := l
-	for _, b := range msg[off : off+l] {
-		switch {
-		case b == '"' || b == '\\':
-			escapedLen++
-		case b < ' ' || b > '~': // unprintable
-			escapedLen += 3 // escapeByte always returns four characters
-		}
-	}
-	if escapedLen == l { // no escaping needed
-		return string(msg[off : off+l]), off + l, nil
-	}
 	var s strings.Builder
-	s.Grow(escapedLen)
+	consumed := 0
-	for _, b := range msg[off : off+l] {
+	for i, b := range msg[off : off+l] {
 		switch {
 		case b == '"' || b == '\\':
+			if consumed == 0 {
+				s.Grow(l * 2)
+			}
+			s.Write(msg[off+consumed : off+i])
 			s.WriteByte('\\')
 			s.WriteByte(b)
+			consumed = i + 1
 		case b < ' ' || b > '~': // unprintable
+			if consumed == 0 {
+				s.Grow(l * 2)
+			}
+			s.Write(msg[off+consumed : off+i])
 			s.WriteString(escapeByte(b))
-		default:
+			consumed = i + 1
-			s.WriteByte(b)
 		}
 	}
-	off += l
+	if consumed == 0 { // no escaping needed
-	return s.String(), off, nil
+		return string(msg[off : off+l]), off + l, nil
+	}
+	s.Write(msg[off+consumed : off+l])
+	return s.String(), off + l, nil
 }
 
 func packString(s string, msg []byte, off int) (int, error) {