tunneler-core 0.2.0

Provides the Core functionality for the Tunneler software
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
40
41
42
43
44
45
46
47
48
49
50

use crate::server::client::{Client, ClientManager};

use rand::Rng;
use tokio::net::TcpListener;

use log::{error, info};

mod validate_connection;

pub async fn accept_clients(
    listen: TcpListener,
    key: Vec<u8>,
    clients: std::sync::Arc<ClientManager>,
) {
    loop {
        // Get Client
        let mut client_socket = match listen.accept().await {
            Ok((socket, _)) => socket,
            Err(e) => {
                error!("Accepting client-connection: {}", e);
                continue;
            }
        };

        if !validate_connection::validate_connection(&mut client_socket, &key).await {
            error!("Rejected Client");
            continue;
        }

        let c_id: u32 = rand::thread_rng().gen();

        info!("Accepted client: {}", c_id);

        let (rx, tx) = client_socket.into_split();

        let (queue_tx, queue_rx) = tokio::sync::mpsc::channel(100);

        let client = Client::new(c_id, clients.clone(), queue_tx);

        tokio::task::spawn(Client::sender(c_id, tx, queue_rx, clients.clone()));
        tokio::task::spawn(Client::receiver(
            c_id,
            rx,
            client.get_user_cons(),
            clients.clone(),
        ));

        clients.add(client);
    }
}