diff --git a/.gitignore b/.gitignore index ad67955..0728338 100644 --- a/.gitignore +++ b/.gitignore @@ -19,3 +19,8 @@ target # and can be added to the global gitignore or merged into this file. For a more nuclear # option (not recommended) you can uncomment the following to ignore the entire idea folder. #.idea/ + + +# Added by cargo + +/target diff --git a/Cargo.lock b/Cargo.lock new file mode 100644 index 0000000..12bbb88 --- /dev/null +++ b/Cargo.lock @@ -0,0 +1,7 @@ +# This file is automatically @generated by Cargo. +# It is not intended for manual editing. +version = 4 + +[[package]] +name = "ghostnet-rs" +version = "0.1.0" diff --git a/Cargo.toml b/Cargo.toml new file mode 100644 index 0000000..8a3f4e5 --- /dev/null +++ b/Cargo.toml @@ -0,0 +1,9 @@ +[package] +name = "ghostnet-rs" +version = "0.1.0" +edition = "2024" + +[[bin]] +name = "ghostnet-rs" + +[dependencies] diff --git a/src/gateway.rs b/src/gateway.rs new file mode 100644 index 0000000..375bc39 --- /dev/null +++ b/src/gateway.rs @@ -0,0 +1,73 @@ +use std::{ + io::{Read, Write}, + net::{TcpListener, TcpStream}, +}; + +use crate::route_tcp; + +pub fn router() { + let listener = TcpListener::bind("0.0.0.0:7879").unwrap(); + println!("Router started on port 7879"); + + for stream in listener.incoming() { + match stream { + Ok(mut client_recv) => { + let mut address = [0; 6]; + let bytes_read = client_recv.read(&mut address).unwrap(); + if bytes_read == 0 { + continue; + } + + let mut client_send = client_recv.try_clone().unwrap(); + let mut server_recv = TcpStream::connect("127.0.0.1:7878").unwrap(); + let mut server_send = server_recv.try_clone().unwrap(); + + std::thread::spawn(move || while !route_tcp(&mut client_recv, &mut server_send) {}); + + while !route_tcp(&mut client_send, &mut server_recv) {} + } + Err(e) => { + eprintln!("Connection failed: {}", e); + } + } + } +} + +pub fn run() -> std::io::Result<()> { + let listener = TcpListener::bind("0.0.0.0:7878")?; + let mut connections: Vec<[u8; 6]> = Vec::new(); + println!("Gateway started on port 7878"); + + std::thread::spawn(router); + + for stream in listener.incoming() { + match stream { + Ok(mut stream) => { + let mut address = [0; 6]; + let bytes_read = stream.read(&mut address)?; + if bytes_read == 0 { + break; + } + + println!("Received: {:?}", &address); + + if let Some(con) = connections.last() { + stream.write_all(con)?; + } else { + stream.write_all(&[127, 0, 0, 1, 0x1E, 0xBF])?; + } + + connections.push(address); + + while stream.read(&mut []).unwrap() != 0 {} + + stream.shutdown(std::net::Shutdown::Both).unwrap(); + } + Err(e) => { + eprintln!("Connection failed: {}", e); + } + } + } + + Ok(()) +} diff --git a/src/lib.rs b/src/lib.rs new file mode 100644 index 0000000..bb42c3d --- /dev/null +++ b/src/lib.rs @@ -0,0 +1,19 @@ +use std::{ + io::{Read, Write}, + net::TcpStream, +}; + +pub mod gateway; + +pub fn route_tcp(from: &mut TcpStream, to: &mut TcpStream) -> bool { + let mut data = [0; 2048]; + + let bytes_read = from.read(&mut data).unwrap(); + if bytes_read == 0 { + return true; + } + + to.write_all(&data[..bytes_read]).unwrap(); + + false +} diff --git a/src/main.rs b/src/main.rs new file mode 100644 index 0000000..e8f4e42 --- /dev/null +++ b/src/main.rs @@ -0,0 +1,3 @@ +fn main() { + // ghostnet_rs::gateway::run().unwrap(); +}