From 6bddd878fb2adf577088f6dd5c33e07013ab7123 Mon Sep 17 00:00:00 2001
From: Adam Wick <awick@uhsure.com>
Date: Mon, 26 Jul 2021 20:39:52 -0700
Subject: [PATCH] Use `GenericStream::from` once, just to make sure it works.

---
 src/network/standard.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/network/standard.rs b/src/network/standard.rs
index abe06a9..5b5d40a 100644
--- a/src/network/standard.rs
+++ b/src/network/standard.rs
@@ -98,7 +98,7 @@ impl Networklike for Builtin {
             SOCKSv5Address::Name(n) => TcpStream::connect((n.as_str(), port)).await?,
         };
 
-        Ok(GenericStream::new(base_stream))
+        Ok(GenericStream::from(base_stream))
     }
 
     async fn listen<A: Send + Into<SOCKSv5Address>>(
@@ -179,6 +179,8 @@ fn check_sanity() {
         let (sender_address, sender_port) = sender.local_addr();
         let send_buffer = [0xa, 0xff, 0xab, 0x1e];
         sender.write_all(&send_buffer).await.expect("Couldn't send the write buffer");
+        sender.flush().await.expect("Couldn't flush the write buffer");
+        sender.close().await.expect("Couldn't close the write buffer");
         (sender_address, sender_port)
     });