ownserver 0.7.1

Expose your local game server to the Internet
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
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

use std::io::{self, ErrorKind};
use std::sync::Arc;

use futures::channel::mpsc::{unbounded, UnboundedReceiver};
use futures::StreamExt;

use tokio::net::UdpSocket;

use crate::{LocalStream, Store, StreamMessage};
use log::*;
use ownserver_lib::{ControlPacketV2, EndpointId, RemoteInfo, StreamId};

/// Establish a new local stream and start processing messages to it
pub async fn setup_new_stream(
    store: Arc<Store>,
    stream_id: StreamId,
    endpoint_id: EndpointId,
    remote_info: RemoteInfo,
) -> io::Result<()>{
    info!("sid={} eid={} setting up local udp stream", stream_id, endpoint_id);
    let local_addr = store.get_local_addr_by_endpoint_id(endpoint_id).ok_or(io::Error::from(ErrorKind::Other))?;

    let local_udp = match UdpSocket::bind("127.0.0.1:0").await {
        Ok(s) => s,
        Err(e) => {
            warn!("sid={} eid={} failed to bind socket: {:?}", stream_id, endpoint_id, e);
            return Err(e);
        } 
    };
    debug!("local udp addr: {:?}", local_udp.local_addr());

    if let Err(e) = local_udp.connect(local_addr).await {
        warn!("sid={} failed to bind socket: {:?}", &stream_id, e);
        return Err(e);
    }

    let local_udp = Arc::new(local_udp);

    // Read local tcp bytes, send them tunnel
    let store_ = store.clone();
    let local_udp_ = local_udp.clone();
    tokio::spawn(async move {
        let _ = process_local_udp(store_.clone(), local_udp_, stream_id).await;
        store_.remove_stream(&stream_id);
        info!("sid={} remove stream to active_streams. len={}", &stream_id, store_.len_stream());
    });

    // Forward remote packets to local tcp
    let (tx, rx) = unbounded();
    let local_stream = LocalStream::new(tx, remote_info);
    store.add_stream(stream_id, local_stream);
    info!("sid={} insert stream to active_streams. len={}", &stream_id, store.len_stream());

    tokio::spawn(async move {
        forward_to_local_udp(stream_id, local_udp, rx).await;
        info!("sid={} end forward to local", &stream_id);
    });

    Ok(())
}

pub async fn process_local_udp(
    store: Arc<Store>,
    stream: Arc<UdpSocket>,
    stream_id: StreamId,
) {
    let mut buf = [0; 4 * 1024];

    loop {
        // let n = match stream.read(&mut buf).await {
        let n = match stream.recv(&mut buf).await {
            Ok(n) => n,
            Err(e) => {
                error!("sid={} failed to read data from socket: {:?}", &stream_id, e);
                return;
            }
        };

        if n == 0 {
            info!("sid={} done reading from client stream", &stream_id);
            return;
        }

        let data = buf[..n].to_vec();
        debug!(
            "sid={} read from local service: {}",
            &stream_id,
            data.len(),
        );

        let packet = ControlPacketV2::Data(stream_id, data.clone());
        if let Err(e) = store.send_to_server(packet).await {
            error!("sid={} failed to tunnel packet from local tcp to tunnel: {:?}", &stream_id, e);
            return;
        }
    }
}

pub async fn forward_to_local_udp(
    stream_id: StreamId,
    // mut sink: WriteHalf<TcpStream>,
    sink: Arc<UdpSocket>,
    mut queue: UnboundedReceiver<StreamMessage>,
) {
    loop {
        let data = match queue.next().await {
            Some(StreamMessage::Data(data)) => data,
            None | Some(StreamMessage::Close) => {
                warn!("sid={} closing stream", &stream_id);
                // let _ = sink.shutdown().await.map_err(|e| {
                //     error!("sid={} failed to shutdown: {:?}", &stream_id, e);
                // });
                return;
            }
        };

        // if let Err(e) = sink.write_all(&data).await {
        if let Err(e) = sink.send(&data).await {
            error!("sid={} failed to write packet data to local tcp socket: {:?}", &stream_id, e);
            return;
        }
        debug!("sid={} wrote to local service: {}", &stream_id, data.len());
    }
}