Fixed overflow panic in handshake_panic (#2495)

openethereum · Oct 7, 2016 · 533af43 · 533af43
1 parent 5f0ed9d
commit 533af43
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/util/network/src/host.rs b/util/network/src/host.rs
@@ -591,7 +591,8 @@ impl Host {
 	}
 
 	fn handshake_count(&self) -> usize {
-		self.sessions.read().count() - self.session_count()
+		// session_count < total_count is possible because of the data race.
+		self.sessions.read().count().saturating_sub(self.session_count())
 	}
 
 	fn keep_alive(&self, io: &IoContext<NetworkIoMessage>) {