From 90b5ecbeef4d3aac22d885b71840cc942c0a1bef Mon Sep 17 00:00:00 2001
From: kali <kali@kali.com>
Date: Mon, 8 May 2023 03:49:13 -0400
Subject: [PATCH] Beacon smb

---
 core                                          |  2 +-
 .../src/SocketHandler.cpp                     | 62 +++++++++++--------
 2 files changed, 37 insertions(+), 27 deletions(-)

diff --git a/core b/core
index c459ccc..5df62c8 160000
--- a/core
+++ b/core
@@ -1 +1 @@
-Subproject commit c459ccc717dfb5cbc13a30d0517bd45d9e064b51
+Subproject commit 5df62c8baa8e9d8a3a113cff22c34f4d04647849
diff --git a/libs/libBoostSocketHandler/src/SocketHandler.cpp b/libs/libBoostSocketHandler/src/SocketHandler.cpp
index 78d74e3..bc6dabf 100644
--- a/libs/libBoostSocketHandler/src/SocketHandler.cpp
+++ b/libs/libBoostSocketHandler/src/SocketHandler.cpp
@@ -78,13 +78,16 @@ bool Server::sendData(std::string& data)
 		return false;
 	}
 
-	sendSocketTcp(m_socketTcp, (char*)&data[0], nbBytes, &m_error);
-
-	if(m_error)
+	if(nbBytes!=0)
 	{
-		// std::cerr << "send failed: " << m_error << std::endl;
-		reset();
-		return false;
+		sendSocketTcp(m_socketTcp, (char*)&data[0], nbBytes, &m_error);
+
+		if(m_error)
+		{
+			// std::cerr << "send failed: " << m_error << std::endl;
+			reset();
+			return false;
+		}
 	}
 
 	return true;
@@ -110,14 +113,16 @@ bool Server::receive(std::string& data)
 		return false;
 	}
 
-	readSocketTcp(m_socketTcp, &data[0], nbBytes, &m_error);
-
-
-	if(m_error)
+	if(nbBytes!=0)
 	{
-		// std::cerr << "receive failed: " << m_error.message() << std::endl;
-		reset();
-		return false;
+		readSocketTcp(m_socketTcp, &data[0], nbBytes, &m_error);
+
+		if(m_error)
+		{
+			// std::cerr << "receive failed: " << m_error.message() << std::endl;
+			reset();
+			return false;
+		}
 	}
 
 	return true;
@@ -174,13 +179,16 @@ bool Client::sendData(std::string& data)
 		return false;
 	}
 
-	sendSocketTcp(m_socketTcp, (char*)&data[0], nbBytes, &m_error);
-
-	if(m_error)
+	if(nbBytes!=0)
 	{
-		// std::cerr << "send failed: " << m_error << std::endl;
-		reset();
-		return false;
+		sendSocketTcp(m_socketTcp, (char*)&data[0], nbBytes, &m_error);
+
+		if(m_error)
+		{
+			// std::cerr << "send failed: " << m_error << std::endl;
+			reset();
+			return false;
+		}
 	}
 
 	return true;
@@ -200,14 +208,16 @@ bool Client::receive(std::string& data)
 		return false;
 	}
 
-	readSocketTcp(m_socketTcp, &data[0], nbBytes, &m_error);
-
-
-	if(m_error)
+	if(nbBytes!=0)
 	{
-		// std::cerr << "receive failed: " << m_error.message() << std::endl;
-		reset();
-		return false;
+		readSocketTcp(m_socketTcp, &data[0], nbBytes, &m_error);
+
+		if(m_error)
+		{
+			// std::cerr << "receive failed: " << m_error.message() << std::endl;
+			reset();
+			return false;
+		}
 	}
 
 	return true;