From cd2aed1c3d0d90e948e24fa5d5e0c7f577c13b4e Mon Sep 17 00:00:00 2001
From: soruh <mail@soruh.de>
Date: Sat, 18 Mar 2023 16:36:13 +0100
Subject: [PATCH] reject with nc on end

---
 src/main.rs | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/main.rs b/src/main.rs
index 625a44d..d1300e0 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -417,13 +417,20 @@ async fn connection_handler(
     };
 
     let (mut client, mut packet) = match result {
-        Result::Packet { packet } => {
+        Result::Packet { mut packet } => {
             if matches!(
                 packet.kind(),
                 packets::PacketKind::End | packets::PacketKind::Reject
             ) {
                 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_metadata