opensource/grpc-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use unpadded base64 encoding for binary metadata headers; handle padded or unpadded input (#1209)

Browse Source
This commit is contained in:
dfawley
2017-04-28 11:32:27 -07:00
committed by GitHub
parent b610ffd3f8
commit 4d1604cc04
2 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
transport/http_util.go
View File

@ -167,11 +167,15 @@ func (d *decodeState) status() *status.Status {
const binHdrSuffix = "-bin"
func encodeBinHeader(v []byte) string {
return base64.StdEncoding.EncodeToString(v)
return base64.RawStdEncoding.EncodeToString(v)
}
func decodeBinHeader(v string) ([]byte, error) {
return base64.StdEncoding.DecodeString(v)
if len(v)%4 == 0 {
// Input was padded, or padding was not necessary.
return base64.StdEncoding.DecodeString(v)
}
return base64.RawStdEncoding.DecodeString(v)
}
func encodeMetadataHeader(k, v string) string {

3
transport/http_util_test.go
View File

@ -158,7 +158,7 @@ func TestEncodeMetadataHeader(t *testing.T) {
{"key", "abc", "abc"},
{"KEY", "abc", "abc"},
{"key-bin", "abc", "YWJj"},
{"key-bin", binaryValue, "woA="},
{"key-bin", binaryValue, "woA"},
} {
v := encodeMetadataHeader(test.kin, test.vin)
if !reflect.DeepEqual(v, test.vout) {
@ -178,6 +178,7 @@ func TestDecodeMetadataHeader(t *testing.T) {
}{
{"a", "abc", "abc", nil},
{"key-bin", "Zm9vAGJhcg==", "foo\x00bar", nil},
{"key-bin", "Zm9vAGJhcg", "foo\x00bar", nil},
{"key-bin", "woA=", binaryValue, nil},
{"a", "abc,efg", "abc,efg", nil},
} {