1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

// Copyright (c) 2018-2019 Parity Technologies (UK) Ltd.
//
// Licensed under the Apache License, Version 2.0 or MIT license, at your option.
//
// A copy of the Apache License, Version 2.0 is included in the software as
// LICENSE-APACHE and a copy of the MIT license is included in the software
// as LICENSE-MIT. You may also obtain a copy of the Apache License, Version 2.0
// at https://www.apache.org/licenses/LICENSE-2.0 and a copy of the MIT license
// at https://opensource.org/licenses/MIT.

use crate::frame::FrameDecodeError;
use thiserror::Error;

/// The various error cases a connection may encounter.
#[non_exhaustive]
#[derive(Debug, Error)]
pub enum ConnectionError {
    /// An underlying I/O error occured.
    #[error("i/o error: {0}")]
    Io(#[from] std::io::Error),

    /// Decoding a Yamux message frame failed.
    #[error("decode error: {0}")]
    Decode(#[from] FrameDecodeError),

    /// The whole range of stream IDs has been used up.
    #[error("number of stream ids has been exhausted")]
    NoMoreStreamIds,

    /// An operation fails because the connection is closed.
    #[error("connection is closed")]
    Closed,

    /// Too many streams are open, so no further ones can be opened at this time.
    #[error("maximum number of streams reached")]
    TooManyStreams
}

impl ConnectionError {
    /// Return the `ErrorKind` of this `ConnectionError` if it holds an I/O error.
    pub(crate) fn io_kind(&self) -> Option<std::io::ErrorKind> {
        match self {
            ConnectionError::Io(e) => Some(e.kind()),
            ConnectionError::Decode(FrameDecodeError::Io(e)) => Some(e.kind()),
            _ => None
        }
    }
}

impl From<futures::channel::mpsc::SendError> for ConnectionError {
    fn from(_: futures::channel::mpsc::SendError) -> Self {
        ConnectionError::Closed
    }
}

impl From<futures::channel::oneshot::Canceled> for ConnectionError {
    fn from(_: futures::channel::oneshot::Canceled) -> Self {
        ConnectionError::Closed
    }
}