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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
use openvpn_plugin;
use std::collections::HashMap;
extern crate futures;
use jsonrpc_client_core::{Future, Result as ClientResult, Transport};
use jsonrpc_client_ipc::IpcTransport;
use tokio::reactor::Handle;
use tokio::runtime::Runtime;
use super::Arguments;
use std::thread;
error_chain! {
errors {
IpcSendingError {
description("Failed while sending an event over the IPC channel")
}
Shutdown {
description("Connection is shut down")
}
}
}
/// Struct processing OpenVPN events and notifies listeners over IPC
pub struct EventProcessor {
ipc_client: EventProxy,
client_stop: ::std::sync::mpsc::Receiver<ClientResult<()>>,
}
impl EventProcessor {
pub fn new(arguments: Arguments) -> Result<EventProcessor> {
trace!("Creating EventProcessor");
let (start_tx, start_rx) = futures::sync::oneshot::channel();
thread::spawn(move || {
let mut rt = Runtime::new().expect("failed to spawn runtime");
let (client, client_handle) =
IpcTransport::new(&arguments.ipc_socket_path, &Handle::current())
.expect("Unable to create IPC transport")
.into_client();
let (tx, client_stop) = ::std::sync::mpsc::channel();
let client_future = client.then(move |result| tx.send(result)).map_err(|_| ());
start_tx
.send((client_stop, client_handle))
.expect("failed to send client handles");
rt.block_on(client_future)
.expect("RPC client should not fail");
});
let (client_stop, client_handle) = start_rx.wait().chain_err(|| ErrorKind::Shutdown)?;
let ipc_client = EventProxy::new(client_handle);
Ok(EventProcessor {
ipc_client,
client_stop,
})
}
pub fn process_event(
&mut self,
event: openvpn_plugin::types::OpenVpnPluginEvent,
env: HashMap<String, String>,
) -> Result<()> {
trace!("Processing \"{:?}\" event", event);
let call_future = self
.ipc_client
.openvpn_event(event, env)
.map_err(|e| Error::with_chain(e, ErrorKind::IpcSendingError));
call_future.wait()?;
self.check_client_status()
}
fn check_client_status(&mut self) -> Result<()> {
use std::sync::mpsc::TryRecvError::*;
match self.client_stop.try_recv() {
Err(Empty) => Ok(()),
Err(Disconnected) => Err(ErrorKind::Shutdown.into()),
Ok(Ok(_)) => Err(ErrorKind::Shutdown.into()),
Ok(Err(e)) => Err(Error::with_chain(e, ErrorKind::IpcSendingError)),
}
}
}
jsonrpc_client!(pub struct EventProxy {
pub fn openvpn_event(&mut self, event: openvpn_plugin::types::OpenVpnPluginEvent, env: HashMap<String, String>) -> Future<()>;
});
|
