groche97/minecraft_rust_reverse_proxy
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Minor fixes and error tolerance

Browse Source
This commit is contained in:
Guillermo Roche
2022-09-24 20:49:00 +02:00
parent 230d5d1c82
commit f754030997
4 changed files with 36 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
src/main.rs
View File

@@ -1,6 +1,9 @@
use std::net::{TcpListener, TcpStream};
use std::io::prelude::*;
use crate::client::guard;
use std::thread;
use std::time::Duration;
mod client;
mod conf;
@@ -10,27 +13,20 @@ fn main() {
let listener = TcpListener::bind("127.0.0.1:25567").unwrap();
let mut buf: [u8; 256] = [1; 256];
let servers = conf::Servers::new();
let mut guard: guard::Guard = guard::Guard::new();
let mut guard = guard::Guard::new();
for stream in listener.incoming() {
if guard.can_add(){
match stream {
Ok(mut stream) => {
let leng = stream.read(&mut buf).unwrap();
stream.set_read_timeout(Some(Duration::from_millis(5000)));
//stream.set_write_timeout(Some(Duration::from_millis(5000)));
let leng = match stream.read(&mut buf) {
Ok(l) => l,
Err(_e) => break,
};
let mut hs = protocol::HandShake::new(&mut buf[.. leng]);
if hs.get_raw()[0] < 200 { //Filtra los ping, solo controlamos los handshakes
match servers.get_server(&hs.getHostName()) {
Some(s) => {
hs.replace_port(s.1);
let mut sstream = TcpStream::connect(s.0 + ":" + &s.1.to_string()).unwrap();
let p_id = sstream.local_addr().unwrap().port();
println!("port4: {}",sstream.peer_addr().unwrap().port());
sstream.write(hs.get_raw());
let c1 = client::Client::new(stream,sstream, hs);
guard.add_thread(c1.start_proxy());
},
None => println!("No server found for{}", hs.getHostName())
}
conect_server(&servers, hs, stream, &mut guard);
}
},
@@ -40,3 +36,20 @@ fn main() {
}
}
fn conect_server(servers: &conf::Servers,
mut hs: protocol::HandShake,
stream: TcpStream,
guard: &mut guard::Guard){
match servers.get_server(&hs.get_host_name()) {
Some(s) => {
hs.replace_port(s.1);
let mut sstream = TcpStream::connect(s.0 + ":" + &s.1.to_string()).unwrap();
sstream.write(hs.get_raw());
let c1 = client::Client::new(stream,sstream, hs);
guard.add_thread(c1.start_proxy());
},
None => println!("No server found for {}", hs.get_host_name())
}
}