async-ssh2 0.1.2-beta

Async wrapper over ssh2.
Documentation 
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

use crate::{channel::Channel, Error};
use smol::{Async, Timer};
use ssh2::{self};
use std::{net::TcpStream, sync::Arc, time::Duration};

/// See [`Listener`](ssh2::Listener).
pub struct Listener {
    inner: ssh2::Listener,
    stream: Arc<Async<TcpStream>>,
}

impl Listener {
    pub(crate) fn new(listener: ssh2::Listener, stream: Arc<Async<TcpStream>>) -> Self {
        Self {
            inner: listener,
            stream,
        }
    }

    /// See [`accept`](ssh2::Listener::accept).
    pub async fn accept(&mut self) -> Result<Channel, Error> {
        // The I/O object for Listener::accept is on the remote SSH server. There is no way to poll
        // its state so the best we can do is loop and check whether we have a new connection every
        // 10ms.
        let channel = loop {
            match self.inner.accept() {
                Ok(channel) => break channel,
                Err(e)
                    if std::io::Error::from(ssh2::Error::from_errno(e.code())).kind()
                        == std::io::ErrorKind::WouldBlock => {}
                Err(e) => return Err(Error::SSH2(e)),
            };

            Timer::after(Duration::from_millis(10)).await;
        };

        Ok(Channel::new(channel, self.stream.clone()))
    }
}