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