dkanus/Avarice
2
1
Fork 1
Code Issues 9 Pull Requests 2 Releases Wiki Activity

Refactor push_byte's code

Browse Source 
Move logic below error checking with early exit.
This commit is contained in:
Anton Tarasenko 2021-07-23 01:31:40 +07:00
parent 9ca00c5b8c
commit e3f554218a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/link/mod.rs
View File

@ -100,7 +100,6 @@ impl MessageReader {
self.read_bytes += 1; self.read_bytes += 1;
if self.read_bytes >= UE_LENGTH_FIELD_SIZE { if self.read_bytes >= UE_LENGTH_FIELD_SIZE {
self.current_message_length = array_of_u8_to_u32(self.length_buffer) as usize; self.current_message_length = array_of_u8_to_u32(self.length_buffer) as usize;
self.change_state(ReadingState::Payload);
if self.current_message_length > MAX_UE_MESSAGE_LENGTH { if self.current_message_length > MAX_UE_MESSAGE_LENGTH {
self.is_broken = true; self.is_broken = true;
return Err(ReadingStreamError::MessageTooLong { return Err(ReadingStreamError::MessageTooLong {
@ -108,6 +107,7 @@ impl MessageReader {
}); });
} }
self.current_message = Vec::with_capacity(self.current_message_length); self.current_message = Vec::with_capacity(self.current_message_length);
self.change_state(ReadingState::Payload);
} }
} }
ReadingState::Payload => { ReadingState::Payload => {