login proxy works
This commit is contained in:
@ -6,7 +6,7 @@
|
|||||||
/* By: tomoron <tomoron@student.42angouleme.fr> +#+ +:+ +#+ */
|
/* By: tomoron <tomoron@student.42angouleme.fr> +#+ +:+ +#+ */
|
||||||
/* +#+#+#+#+#+ +#+ */
|
/* +#+#+#+#+#+ +#+ */
|
||||||
/* Created: 2026/05/07 17:23:09 by tomoron #+# #+# */
|
/* Created: 2026/05/07 17:23:09 by tomoron #+# #+# */
|
||||||
/* Updated: 2026/05/19 21:19:53 by tomoron ### ########.fr */
|
/* Updated: 2026/05/20 17:54:32 by tomoron ### ########.fr */
|
||||||
/* */
|
/* */
|
||||||
/* ************************************************************************** */
|
/* ************************************************************************** */
|
||||||
|
|
||||||
@ -22,6 +22,7 @@ use std::{fmt, io};
|
|||||||
pub struct Client {
|
pub struct Client {
|
||||||
pub in_stream: TcpStream,
|
pub in_stream: TcpStream,
|
||||||
|
|
||||||
|
read_buf: Vec<u8>,
|
||||||
buffer: Vec<u8>,
|
buffer: Vec<u8>,
|
||||||
out_stream: Option<TcpStream>,
|
out_stream: Option<TcpStream>,
|
||||||
handshake: Option<Handshake>
|
handshake: Option<Handshake>
|
||||||
@ -32,6 +33,7 @@ impl Client {
|
|||||||
Self {
|
Self {
|
||||||
in_stream: stream,
|
in_stream: stream,
|
||||||
|
|
||||||
|
read_buf: vec![0u8; 1024 * 8],
|
||||||
buffer: vec![],
|
buffer: vec![],
|
||||||
out_stream: None,
|
out_stream: None,
|
||||||
handshake: None
|
handshake: None
|
||||||
@ -39,15 +41,16 @@ impl Client {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub async fn in_read(&mut self) -> Result<u8, String> {
|
pub async fn in_read(&mut self) -> Result<u8, String> {
|
||||||
let mut buf = vec![0 as u8; 1024];
|
|
||||||
|
|
||||||
loop
|
loop
|
||||||
{
|
{
|
||||||
match self.in_stream.try_read(&mut buf) {
|
match self.in_stream.try_read(&mut self.read_buf) {
|
||||||
Ok(n) => {
|
Ok(n) => {
|
||||||
println!("[{}] read {} bytes",self, n);
|
if let Some(out_stream) = &self.out_stream {
|
||||||
println!("[{}] {:?}", self, &buf[..n]);
|
let _ = out_stream.writable().await;
|
||||||
self.buffer_append((&buf[..n]).to_vec()).await?;
|
let _ = out_stream.try_write(&self.read_buf[..n]);
|
||||||
|
} else {
|
||||||
|
self.buffer_append((&self.read_buf[..n]).to_vec()).await?;
|
||||||
|
}
|
||||||
if n == 0 {
|
if n == 0 {
|
||||||
return Ok(0)
|
return Ok(0)
|
||||||
}
|
}
|
||||||
@ -56,7 +59,6 @@ impl Client {
|
|||||||
return Ok(1);
|
return Ok(1);
|
||||||
}
|
}
|
||||||
Err(_e) => {
|
Err(_e) => {
|
||||||
println!("[{}] error", self);
|
|
||||||
return Err("read failed".to_string())
|
return Err("read failed".to_string())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -72,17 +74,15 @@ impl Client {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub async fn out_read(&mut self) -> Result<i32, String> {
|
pub async fn out_read(&mut self) -> Result<i32, String> {
|
||||||
let mut buf = vec![0 as u8; 1024];
|
|
||||||
|
|
||||||
if let None = self.out_stream {
|
if let None = self.out_stream {
|
||||||
return Ok(1);
|
return Ok(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
match self.out_stream.as_mut().unwrap().try_read(&mut buf) {
|
match self.out_stream.as_mut().unwrap().try_read(&mut self.read_buf) {
|
||||||
Ok(n) => {
|
Ok(n) => {
|
||||||
let _ = self.in_stream.writable().await;
|
let _ = self.in_stream.writable().await;
|
||||||
let _ = self.in_stream.try_write(&buf[..n]);
|
let _ = self.in_stream.try_write(&self.read_buf[..n]);
|
||||||
if n == 0 { Ok(0) } else { Ok(1) }
|
if n == 0 { Ok(0) } else { Ok(1) }
|
||||||
}
|
}
|
||||||
Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => {
|
Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => {
|
||||||
@ -90,7 +90,6 @@ impl Client {
|
|||||||
}
|
}
|
||||||
|
|
||||||
Err(_e) => {
|
Err(_e) => {
|
||||||
println!("try_read returned an error");
|
|
||||||
Err("Failed to read".to_string())
|
Err("Failed to read".to_string())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -101,14 +100,6 @@ impl Client {
|
|||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
|
||||||
println!("[{}] buffer handling", self);
|
|
||||||
|
|
||||||
if let Some(out_stream) = &self.out_stream {
|
|
||||||
println!("[{}] out stream present passing buffer", self);
|
|
||||||
let _ = out_stream.writable().await;
|
|
||||||
let _ = out_stream.try_write(&data);
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
|
|
||||||
if self.buffer.len() + data.len() > 65536 {
|
if self.buffer.len() + data.len() > 65536 {
|
||||||
return Err("buffer full".to_string());
|
return Err("buffer full".to_string());
|
||||||
@ -120,10 +111,19 @@ impl Client {
|
|||||||
if self.buffer[0] == 0 {
|
if self.buffer[0] == 0 {
|
||||||
return Err("invalid packet".to_string());
|
return Err("invalid packet".to_string());
|
||||||
}
|
}
|
||||||
println!("[{}] valid packet received", self);
|
|
||||||
|
|
||||||
let len = varint_read(&mut self.buffer.clone().into())? as usize;
|
let len = varint_read(&mut self.buffer.clone().into())? as usize;
|
||||||
self.handle_packet(self.buffer[1..=len].to_vec().into()).await?;
|
self.handle_packet(self.buffer[1..=len].to_vec().into()).await?;
|
||||||
self.buffer.drain(..len + 1);
|
self.buffer.drain(..len + 1);
|
||||||
|
|
||||||
|
if let Some(out_stream) = &self.out_stream {
|
||||||
|
if self.buffer.len() != 0 {
|
||||||
|
let _ = out_stream.writable().await;
|
||||||
|
let _ = out_stream.try_write(self.buffer.as_slice());
|
||||||
|
self.buffer.drain(..);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
@ -133,7 +133,6 @@ impl Client {
|
|||||||
|
|
||||||
sent_data.extend(data);
|
sent_data.extend(data);
|
||||||
let _ = self.in_stream.writable().await;
|
let _ = self.in_stream.writable().await;
|
||||||
println!("[{}] sending {:?}",self, sent_data);
|
|
||||||
match self.in_stream.try_write(sent_data.as_slice()) {
|
match self.in_stream.try_write(sent_data.as_slice()) {
|
||||||
Err(e) => { eprintln!("error while sending response {:?}", e); },
|
Err(e) => { eprintln!("error while sending response {:?}", e); },
|
||||||
_ => { }
|
_ => { }
|
||||||
@ -149,10 +148,11 @@ impl Client {
|
|||||||
return Err("packet 0 expected. invalid packet received".to_string());
|
return Err("packet 0 expected. invalid packet received".to_string());
|
||||||
}
|
}
|
||||||
self.handshake = Some(Handshake::from_packet(&mut packet)?);
|
self.handshake = Some(Handshake::from_packet(&mut packet)?);
|
||||||
use std::os::unix::io::AsRawFd;
|
|
||||||
let fd = self.in_stream.as_raw_fd();
|
if self.handshake.as_ref().unwrap().intent == 2 {
|
||||||
println!("{} => {}", fd, self);
|
self.login_intent_handle(&mut packet, packet_id).await?;
|
||||||
println!("[{}] got {}",self, self.handshake.as_ref().unwrap());
|
}
|
||||||
|
|
||||||
return Ok(());
|
return Ok(());
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -160,8 +160,6 @@ impl Client {
|
|||||||
|
|
||||||
if intent == 1 {
|
if intent == 1 {
|
||||||
self.status_intent_handle(&mut packet, packet_id).await?;
|
self.status_intent_handle(&mut packet, packet_id).await?;
|
||||||
} else if intent == 2 {
|
|
||||||
self.login_intent_handle(&mut packet, packet_id).await?;
|
|
||||||
}
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
@ -201,12 +199,11 @@ impl Client {
|
|||||||
}
|
}
|
||||||
|
|
||||||
async fn login_intent_handle(&mut self, mut _packet: &mut VecDeque<u8>, _packet_id: i32) -> Result<(),String> {
|
async fn login_intent_handle(&mut self, mut _packet: &mut VecDeque<u8>, _packet_id: i32) -> Result<(),String> {
|
||||||
let stream = TcpStream::connect("localhost:25566").await.map_err(|_| "failed to connect to remote host".to_string())?;
|
let stream = TcpStream::connect("play.hypixel.net:25565").await.map_err(|_| "failed to connect to remote host".to_string())?;
|
||||||
let mut new_handshake = self.handshake.as_ref().unwrap().clone();
|
let mut new_handshake = self.handshake.as_ref().unwrap().clone();
|
||||||
new_handshake.server_address = "localhost".to_string();
|
new_handshake.server_address = "play.hypixel.net".to_string();
|
||||||
new_handshake.server_port = 25566;
|
new_handshake.server_port = 25565;
|
||||||
|
|
||||||
println!("{}", new_handshake);
|
|
||||||
let handshake_packet = new_handshake.to_packet();
|
let handshake_packet = new_handshake.to_packet();
|
||||||
let mut packet = Vec::new();
|
let mut packet = Vec::new();
|
||||||
packet.extend(varint_write(handshake_packet.len() as i32));
|
packet.extend(varint_write(handshake_packet.len() as i32));
|
||||||
|
|||||||
@ -9,15 +9,12 @@ pub async fn process_mc_socket(stream: TcpStream) -> Result<(), String> {
|
|||||||
loop {
|
loop {
|
||||||
tokio::select! {
|
tokio::select! {
|
||||||
Ok(()) = client.in_stream.readable() => {
|
Ok(()) = client.in_stream.readable() => {
|
||||||
println!("[{}] in read", client);
|
|
||||||
let res = client.in_read().await?;
|
let res = client.in_read().await?;
|
||||||
if res == 0 {
|
if res == 0 {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
println!("[{}] continue listening", client);
|
|
||||||
}
|
}
|
||||||
_ = client.out_readable() => {
|
_ = client.out_readable() => {
|
||||||
println!("[{}] out read", client);
|
|
||||||
let res = client.out_read().await?;
|
let res = client.out_read().await?;
|
||||||
if res == 0 {
|
if res == 0 {
|
||||||
break;
|
break;
|
||||||
|
|||||||
Reference in New Issue
Block a user