auto merge of #9089 : anasazi/rust/fix-acceptor-iterator, r=cmr
The iterator over incoming connections has no natural end, so it should always return Some(_). Currently, if an incoming connection fails, the iterator returns None. Trying to accept another connection afterwards enters the realm of undefined behavior (due to the iterator protocol being silent on the issue). This PR changes wraps the underlying accept call in Some, so the iterator never finishes.
This commit is contained in:
commit
d41b558792
1 changed files with 9 additions and 4 deletions
|
|
@ -493,7 +493,7 @@ pub trait Acceptor<T> {
|
|||
/// then `accept` returns `None`.
|
||||
fn accept(&mut self) -> Option<T>;
|
||||
|
||||
/// Create an iterator over incoming connections
|
||||
/// Create an iterator over incoming connection attempts
|
||||
fn incoming<'r>(&'r mut self) -> IncomingIterator<'r, Self> {
|
||||
IncomingIterator { inc: self }
|
||||
}
|
||||
|
|
@ -501,13 +501,18 @@ pub trait Acceptor<T> {
|
|||
|
||||
/// An infinite iterator over incoming connection attempts.
|
||||
/// Calling `next` will block the task until a connection is attempted.
|
||||
///
|
||||
/// Since connection attempts can continue forever, this iterator always returns Some.
|
||||
/// The Some contains another Option representing whether the connection attempt was succesful.
|
||||
/// A successful connection will be wrapped in Some.
|
||||
/// A failed connection is represented as a None and raises a condition.
|
||||
struct IncomingIterator<'self, A> {
|
||||
priv inc: &'self mut A,
|
||||
}
|
||||
|
||||
impl<'self, T, A: Acceptor<T>> Iterator<T> for IncomingIterator<'self, A> {
|
||||
fn next(&mut self) -> Option<T> {
|
||||
self.inc.accept()
|
||||
impl<'self, T, A: Acceptor<T>> Iterator<Option<T>> for IncomingIterator<'self, A> {
|
||||
fn next(&mut self) -> Option<Option<T>> {
|
||||
Some(self.inc.accept())
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue