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

use crate::{util::run_ssh2_fn, Error};
use smol::Async;
use ssh2::{self, PublicKey};
use std::{convert::From, net::TcpStream, sync::Arc};

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

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

    /// See [`connect`](ssh2::Agent::connect).
    pub async fn connect(&mut self) -> Result<(), Error> {
        run_ssh2_fn(&self.stream.clone(), || self.inner.connect()).await
    }

    /// See [`disconnect`](ssh2::Agent::disconnect).
    pub async fn disconnect(&mut self) -> Result<(), Error> {
        run_ssh2_fn(&self.stream.clone(), || self.inner.disconnect()).await
    }

    /// See [`list_identities`](ssh2::Agent::list_identities).
    pub fn list_identities(&mut self) -> Result<(), Error> {
        self.inner.list_identities().map_err(From::from)
    }

    /// See [`identities`](ssh2::Agent::identities).
    pub fn identities(&self) -> Result<Vec<PublicKey>, Error> {
        self.inner.identities().map_err(From::from)
    }

    /// See [`userauth`](ssh2::Agent::userauth).
    pub async fn userauth(&self, username: &str, identity: &PublicKey) -> Result<(), Error> {
        run_ssh2_fn(&self.stream.clone(), || {
            self.inner.userauth(username, identity)
        })
        .await
    }
}