grassrootseconomics/openethereum
5
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update handshake.rs

Browse Source
This commit is contained in:
Gav Wood 2016-02-27 15:07:21 +01:00
parent 213bfc4417
commit faa6347896
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
util/src/network/handshake.rs
View File

@ -239,12 +239,12 @@ impl Handshake {
} }
Err(_) => { Err(_) => {
// Try to interpret as EIP-8 packet // Try to interpret as EIP-8 packet
let size = ((data[0] as u16) << 8 | (data[1] as u16)) as usize; let total = (((data[0] as u16) << 8 | (data[1] as u16)) as usize) + 2;
if size < V4_AUTH_PACKET_SIZE - 2 { if total < V4_AUTH_PACKET_SIZE {
debug!(target:"net", "Wrong EIP8 auth packet size"); debug!(target:"net", "Wrong EIP8 auth packet size");
return Err(From::from(NetworkError::BadProtocol)); return Err(From::from(NetworkError::BadProtocol));
} }
let rest = size - data.len() + 2; let rest = total - data.len();
self.state = HandshakeState::ReadingAuthEip8; self.state = HandshakeState::ReadingAuthEip8;
self.connection.expect(rest); self.connection.expect(rest);
} }
@ -284,7 +284,7 @@ impl Handshake {
// Try to interpret as EIP-8 packet // Try to interpret as EIP-8 packet
let total = (((data[0] as u16) << 8 | (data[1] as u16)) as usize) + 2; let total = (((data[0] as u16) << 8 | (data[1] as u16)) as usize) + 2;
if total < V4_ACK_PACKET_SIZE { if total < V4_ACK_PACKET_SIZE {
debug!(target: "net", "Wrong EIP8 ack packet size"); debug!(target:"net", "Wrong EIP8 ack packet size");
return Err(From::from(NetworkError::BadProtocol)); return Err(From::from(NetworkError::BadProtocol));
} }
let rest = total - data.len(); let rest = total - data.len();