Fixed overflow panic in handshake_panic (#2495)

2016-10-07 10:34:06 +02:00 · 2016-10-07 10:34:06 +02:00 · 533af43313
commit 533af43313
parent 5f0ed9ddce
1 changed files with 2 additions and 1 deletions
--- 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>) {