openaws-vpn-client 0.1.0

Unofficial open-source AWS VPN client written in Rust
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

use crate::cmd::{kill_openvpn, ProcessInfo};
use crate::Log;
use std::sync::Arc;

pub struct OavcTask<T> {
    pub name: String,
    pub handle: tokio::task::JoinHandle<T>,
    pub log: Arc<Log>,
}

pub struct OavcProcessTask<T> {
    pub name: String,
    pub handle: tokio::task::JoinHandle<T>,
    pub log: Arc<Log>,
    pub info: Arc<ProcessInfo>,
}

impl<T> OavcTask<T> {
    pub fn abort(&self, log: bool) {
        self.handle.abort();
        if log {
            self.log
                .append(format!("Stopped '{}'!", self.name).as_str());
        }
    }
}

impl<T> OavcProcessTask<T> {
    pub fn new(
        name: String,
        handle: tokio::task::JoinHandle<T>,
        log: Arc<Log>,
        info: Arc<ProcessInfo>,
    ) -> Self {
        Self {
            name,
            handle,
            log,
            info,
        }
    }

    pub fn abort(&self, log: bool) {
        self.handle.abort();
        {
            let pid = self.info.pid.lock().unwrap();

            if let Some(ref pid) = *pid {
                kill_openvpn(*pid)
            }

            if log {
                self.log
                    .append(format!("Stopped '{}' pid '{:?}'!", self.name, pid).as_str());
            }
        }
    }
}

unsafe impl<T> Send for OavcTask<T> {}
unsafe impl<T> Sync for OavcTask<T> {}