|
|
@@ -119,6 +119,7 @@ func (v *ClientSession) EncodeRequestHeader(header *protocol.RequestHeader, writ
|
|
|
|
|
|
func (v *ClientSession) EncodeRequestBody(request *protocol.RequestHeader, writer io.Writer) buf.Writer {
|
|
|
var authWriter io.Writer
|
|
|
+ sizeMask := serial.BytesToUint16(v.requestBodyKey[:2])
|
|
|
if request.Security.Is(protocol.SecurityType_NONE) {
|
|
|
if request.Option.Has(protocol.RequestOptionChunkStream) {
|
|
|
auth := &crypto.AEADAuthenticator{
|
|
|
@@ -126,7 +127,7 @@ func (v *ClientSession) EncodeRequestBody(request *protocol.RequestHeader, write
|
|
|
NonceGenerator: crypto.NoOpBytesGenerator{},
|
|
|
AdditionalDataGenerator: crypto.NoOpBytesGenerator{},
|
|
|
}
|
|
|
- authWriter = crypto.NewAuthenticationWriter(auth, writer)
|
|
|
+ authWriter = crypto.NewAuthenticationWriter(auth, writer, sizeMask)
|
|
|
} else {
|
|
|
authWriter = writer
|
|
|
}
|
|
|
@@ -139,7 +140,7 @@ func (v *ClientSession) EncodeRequestBody(request *protocol.RequestHeader, write
|
|
|
NonceGenerator: crypto.NoOpBytesGenerator{},
|
|
|
AdditionalDataGenerator: crypto.NoOpBytesGenerator{},
|
|
|
}
|
|
|
- authWriter = crypto.NewAuthenticationWriter(auth, cryptionWriter)
|
|
|
+ authWriter = crypto.NewAuthenticationWriter(auth, cryptionWriter, sizeMask)
|
|
|
} else {
|
|
|
authWriter = cryptionWriter
|
|
|
}
|
|
|
@@ -155,7 +156,7 @@ func (v *ClientSession) EncodeRequestBody(request *protocol.RequestHeader, write
|
|
|
},
|
|
|
AdditionalDataGenerator: crypto.NoOpBytesGenerator{},
|
|
|
}
|
|
|
- authWriter = crypto.NewAuthenticationWriter(auth, writer)
|
|
|
+ authWriter = crypto.NewAuthenticationWriter(auth, writer, sizeMask)
|
|
|
} else if request.Security.Is(protocol.SecurityType_CHACHA20_POLY1305) {
|
|
|
aead, _ := chacha20poly1305.New(GenerateChacha20Poly1305Key(v.requestBodyKey))
|
|
|
|
|
|
@@ -167,7 +168,7 @@ func (v *ClientSession) EncodeRequestBody(request *protocol.RequestHeader, write
|
|
|
},
|
|
|
AdditionalDataGenerator: crypto.NoOpBytesGenerator{},
|
|
|
}
|
|
|
- authWriter = crypto.NewAuthenticationWriter(auth, writer)
|
|
|
+ authWriter = crypto.NewAuthenticationWriter(auth, writer, sizeMask)
|
|
|
}
|
|
|
|
|
|
return buf.NewWriter(authWriter)
|
|
|
@@ -214,6 +215,7 @@ func (v *ClientSession) DecodeResponseHeader(reader io.Reader) (*protocol.Respon
|
|
|
|
|
|
func (v *ClientSession) DecodeResponseBody(request *protocol.RequestHeader, reader io.Reader) buf.Reader {
|
|
|
var authReader io.Reader
|
|
|
+ sizeMask := serial.BytesToUint16(v.responseBodyKey[:2])
|
|
|
if request.Security.Is(protocol.SecurityType_NONE) {
|
|
|
if request.Option.Has(protocol.RequestOptionChunkStream) {
|
|
|
auth := &crypto.AEADAuthenticator{
|
|
|
@@ -221,7 +223,7 @@ func (v *ClientSession) DecodeResponseBody(request *protocol.RequestHeader, read
|
|
|
NonceGenerator: crypto.NoOpBytesGenerator{},
|
|
|
AdditionalDataGenerator: crypto.NoOpBytesGenerator{},
|
|
|
}
|
|
|
- authReader = crypto.NewAuthenticationReader(auth, reader)
|
|
|
+ authReader = crypto.NewAuthenticationReader(auth, reader, sizeMask)
|
|
|
} else {
|
|
|
authReader = reader
|
|
|
}
|
|
|
@@ -232,7 +234,7 @@ func (v *ClientSession) DecodeResponseBody(request *protocol.RequestHeader, read
|
|
|
NonceGenerator: crypto.NoOpBytesGenerator{},
|
|
|
AdditionalDataGenerator: crypto.NoOpBytesGenerator{},
|
|
|
}
|
|
|
- authReader = crypto.NewAuthenticationReader(auth, v.responseReader)
|
|
|
+ authReader = crypto.NewAuthenticationReader(auth, v.responseReader, sizeMask)
|
|
|
} else {
|
|
|
authReader = v.responseReader
|
|
|
}
|
|
|
@@ -248,7 +250,7 @@ func (v *ClientSession) DecodeResponseBody(request *protocol.RequestHeader, read
|
|
|
},
|
|
|
AdditionalDataGenerator: crypto.NoOpBytesGenerator{},
|
|
|
}
|
|
|
- authReader = crypto.NewAuthenticationReader(auth, reader)
|
|
|
+ authReader = crypto.NewAuthenticationReader(auth, reader, sizeMask)
|
|
|
} else if request.Security.Is(protocol.SecurityType_CHACHA20_POLY1305) {
|
|
|
aead, _ := chacha20poly1305.New(GenerateChacha20Poly1305Key(v.responseBodyKey))
|
|
|
|
|
|
@@ -260,7 +262,7 @@ func (v *ClientSession) DecodeResponseBody(request *protocol.RequestHeader, read
|
|
|
},
|
|
|
AdditionalDataGenerator: crypto.NoOpBytesGenerator{},
|
|
|
}
|
|
|
- authReader = crypto.NewAuthenticationReader(auth, reader)
|
|
|
+ authReader = crypto.NewAuthenticationReader(auth, reader, sizeMask)
|
|
|
}
|
|
|
|
|
|
return buf.NewReader(authReader)
|