Fixed overflow panic in handshake_panic

openethereum · Oct 6, 2016 · ce0b4fe · ce0b4fe
1 parent ecf098e
commit ce0b4fe
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/util/network/src/host.rs b/util/network/src/host.rs
@@ -591,7 +591,10 @@ impl Host {
 	}
 
 	fn handshake_count(&self) -> usize {
-		self.sessions.read().count() - self.session_count()
+		let total_count = self.sessions.read().count();
+		let session_count = self.session_count();
+		// session_count < total_count is possible because of the data race.
+		return if total_count >= session_count { total_count - session_count } else { 0 };
 	}
 
 	fn keep_alive(&self, io: &IoContext<NetworkIoMessage>) {