renet
/
renet
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Actually close TCP sockets with 0 references during pruning.

v0.7.x
whitequark 2017-01-25 02:45:17 +00:00
parent 5cb4e6f760
commit f6d542b739
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/socket/set.rs
View File

@ -137,13 +137,17 @@ impl<'a, 'b: 'a, 'c: 'a + 'b> Set<'a, 'b, 'c> {
pub fn prune(&mut self) {
for (index, item) in self.sockets.iter_mut().enumerate() {
let mut may_remove = false;
if let &mut Some(Item { refs: 0, ref socket }) = item {
if let &mut Some(Item { refs: 0, ref mut socket }) = item {
match socket {
&Socket::Udp(_) =>
&mut Socket::Udp(_) =>
may_remove = true,
&Socket::Tcp(ref socket) =>
may_remove = socket.state() == TcpState::Closed,
&Socket::__Nonexhaustive => unreachable!()
&mut Socket::Tcp(ref mut socket) =>
if socket.state() == TcpState::Closed {
may_remove = true
} else {
socket.close()
},
&mut Socket::__Nonexhaustive => unreachable!()
}
}
if may_remove {