soruh/centralex
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

reject with nc on end

Browse Source
This commit is contained in:
soruh 2023-03-18 16:36:13 +01:00
parent 9a3bac503c
commit cd2aed1c3d
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/main.rs
View File

@ -417,13 +417,20 @@ async fn connection_handler(
}; };
let (mut client, mut packet) = match result { let (mut client, mut packet) = match result {
Result::Packet { packet } => { Result::Packet { mut packet } => {
if matches!( if matches!(
packet.kind(), packet.kind(),
packets::PacketKind::End | packets::PacketKind::Reject packets::PacketKind::End | packets::PacketKind::Reject
) { ) {
println!("got disconnect packet: {packet:?}"); println!("got disconnect packet: {packet:?}");
if packet.kind() == packets::PacketKind::End {
packet.header.kind = packets::PacketKind::Reject.raw();
packet.data.clear();
packet.data.extend_from_slice(b"nc\0");
packet.header.length = packet.data.len() as u8;
}
port_handler.lock().await.start_rejector( port_handler.lock().await.start_rejector(
port, port,
handler_metadata handler_metadata