Simple gateway (not tested)
This commit is contained in:
@@ -19,3 +19,8 @@ target
|
|||||||
# and can be added to the global gitignore or merged into this file. For a more nuclear
|
# 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.
|
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
||||||
#.idea/
|
#.idea/
|
||||||
|
|
||||||
|
|
||||||
|
# Added by cargo
|
||||||
|
|
||||||
|
/target
|
||||||
|
|||||||
Generated
+7
@@ -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"
|
||||||
@@ -0,0 +1,9 @@
|
|||||||
|
[package]
|
||||||
|
name = "ghostnet-rs"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2024"
|
||||||
|
|
||||||
|
[[bin]]
|
||||||
|
name = "ghostnet-rs"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
@@ -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(())
|
||||||
|
}
|
||||||
+19
@@ -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
|
||||||
|
}
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
fn main() {
|
||||||
|
// ghostnet_rs::gateway::run().unwrap();
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user