filesync: fix backward compatibility with encoding + and %

Signed-off-by: Tonis Tiigi <[email protected]>

Author: tonistiigi

session/filesync/filesync.go

@@ -27,7 +27,6 @@ const (
 	keyFollowPaths        = "followpaths"
 	keyDirName            = "dir-name"
 	keyExporterMetaPrefix = "exporter-md-"
-	keyOptsEncoded        = "opts-encoded"
 )
 
 type fsSyncProvider struct {
@@ -86,17 +85,7 @@ func (sp *fsSyncProvider) handle(method string, stream grpc.ServerStream) (retEr
 	}
 
 	opts, _ := metadata.FromIncomingContext(stream.Context()) // if no metadata continue with empty object
-
-	isDecoded := false
-	if v, ok := opts[keyOptsEncoded]; ok && len(v) > 0 {
-		if b, _ := strconv.ParseBool(v[0]); b {
-			isDecoded = true
-		}
-	}
-
-	if isDecoded {
-		opts = decodeOpts(opts)
-	}
+	opts = decodeOpts(opts)
 
 	dirName := ""
 	name, ok := opts[keyDirName]
@@ -224,9 +213,6 @@ func FSSync(ctx context.Context, c session.Caller, opt FSSendRequestOpt) error {
 
 	var stream grpc.ClientStream
 
-	// mark that we have encoded options so older versions with raw values can be detected on client side
-	opts[keyOptsEncoded] = []string{"1"}
-
 	opts = encodeOpts(opts)
 
 	ctx = metadata.NewOutgoingContext(ctx, opts)
@@ -361,11 +347,11 @@ func (e InvalidSessionError) Unwrap() error {
 func encodeOpts(opts map[string][]string) map[string][]string {
 	md := make(map[string][]string, len(opts))
 	for k, v := range opts {
-		out := make([]string, len(v))
-		for i, s := range v {
-			out[i] = encodeStringForHeader(s)
-		}
+		out, encoded := encodeStringForHeader(v)
 		md[k] = out
+		if encoded {
+			md[k+"-encoded"] = []string{"1"}
+		}
 	}
 	return md
 }
@@ -374,8 +360,18 @@ func decodeOpts(opts map[string][]string) map[string][]string {
 	md := make(map[string][]string, len(opts))
 	for k, v := range opts {
 		out := make([]string, len(v))
-		for i, s := range v {
-			out[i], _ = url.QueryUnescape(s)
+		var isDecoded bool
+		if v, ok := opts[k+"-encoded"]; ok && len(v) > 0 {
+			if b, _ := strconv.ParseBool(v[0]); b {
+				isDecoded = true
+			}
+		}
+		if isDecoded {
+			for i, s := range v {
+				out[i], _ = url.QueryUnescape(s)
+			}
+		} else {
+			copy(out, v)
 		}
 		md[k] = out
 	}
@@ -384,18 +380,23 @@ func decodeOpts(opts map[string][]string) map[string][]string {
 
 // encodeStringForHeader encodes a string value so it can be used in grpc header. This encoding
 // is backwards compatible and avoids encoding ASCII characters.
-func encodeStringForHeader(input string) string {
-	var output strings.Builder
-	for _, runeVal := range input {
-		// Only encode non-ASCII characters, and characters that have special
-		// meaning during decoding.
-		if runeVal > unicode.MaxASCII || runeVal == '%' || runeVal == '+' {
-			// Encode each non-ASCII character individually.
-			output.WriteString(url.QueryEscape(string(runeVal)))
-		} else {
-			// Directly append ASCII characters and '*' to the output.
-			output.WriteRune(runeVal)
+func encodeStringForHeader(inputs []string) ([]string, bool) {
+	var encode bool
+	for _, input := range inputs {
+		for _, runeVal := range input {
+			// Only encode non-ASCII characters, and characters that have special
+			// meaning during decoding.
+			if runeVal > unicode.MaxASCII {
+				encode = true
+				break
+			}
 		}
 	}
-	return output.String()
+	if !encode {
+		return inputs, false
+	}
+	for i, input := range inputs {
+		inputs[i] = url.QueryEscape(input)
+	}
+	return inputs, true
 }