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
|
#![deny(warnings)]
extern crate ssh2;
extern crate tempfile;
use std::env;
use std::net::TcpStream;
mod agent;
mod channel;
mod knownhosts;
mod session;
mod sftp;
pub fn test_addr() -> String {
let port = env::var("RUST_SSH2_FIXTURE_PORT")
.map(|s| s.parse().unwrap())
.unwrap_or(22);
let addr = format!("127.0.0.1:{}", port);
addr
}
pub fn socket() -> TcpStream {
TcpStream::connect(&test_addr()).unwrap()
}
pub fn authed_session() -> ssh2::Session {
let user = env::var("USER").unwrap();
let socket = socket();
let mut sess = ssh2::Session::new().unwrap();
sess.set_tcp_stream(socket);
sess.handshake().unwrap();
assert!(!sess.authenticated());
{
let mut agent = sess.agent().unwrap();
agent.connect().unwrap();
agent.list_identities().unwrap();
let identities = agent.identities().unwrap();
let identity = &identities[0];
agent.userauth(&user, &identity).unwrap();
}
assert!(sess.authenticated());
sess
}
|
