From 5bfeb4a0c539410faceccd38bc7cff563b6e38fe Mon Sep 17 00:00:00 2001
From: sairson <zirs0n@qq.com>
Date: Tue, 11 Jan 2022 09:10:49 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81=E7=BB=93?=
 =?UTF-8?q?=E6=9E=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmd/eternalblue.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/cmd/eternalblue.go b/cmd/eternalblue.go
index 08ed3bd..fe8485a 100644
--- a/cmd/eternalblue.go
+++ b/cmd/eternalblue.go
@@ -91,7 +91,10 @@ func RequestMs17010(conn net.Conn, ip string) (bool, error) {
 	treeConnectRequest[32] = userID[0]
 	treeConnectRequest[33] = userID[1]
 	// TODO change the ip in tree path though it doesn't matter
-	conn.Write(treeConnectRequest)
+	_, err = conn.Write(treeConnectRequest)
+	if err != nil {
+		return false, err
+	}
 
 	if n, err := conn.Read(reply); err != nil || n < 36 {
 		return false, err