From 0be22bf4b6df49a814d7aba9831320df9afe4bb2 Mon Sep 17 00:00:00 2001
From: Jean-Baptiste Perrier <jeanbapt.perrier@gmail.com>
Date: Thu, 18 Feb 2021 18:07:24 +0100
Subject: [PATCH] Tweaks.

---
 Source/Engine/Platform/Win32/Win32Network.cpp | 6 ++++--
 Source/Engine/Platform/Win32/Win32Network.h   | 3 ---
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/Source/Engine/Platform/Win32/Win32Network.cpp b/Source/Engine/Platform/Win32/Win32Network.cpp
index c467405a8..3c24110fd 100644
--- a/Source/Engine/Platform/Win32/Win32Network.cpp
+++ b/Source/Engine/Platform/Win32/Win32Network.cpp
@@ -18,6 +18,8 @@ static const IN6_ADDR v4MappedPrefix = { { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0
 // @formatter:on
 
 /*
+ * Todo :
+ *  Return precise errors so user can understand what's happening ( disconnected, ect ... )
  * Known issues :
  *  Even if dualstacking is enabled it's not possible to bind an Ipv4mappedIPv6 endpoint. windows limitation
  */
@@ -418,7 +420,7 @@ int32 Win32Network::WriteSocket(NetworkSocket socket, byte* data, uint32 length,
 int32 Win32Network::ReadSocket(NetworkSocket socket, byte* buffer, uint32 bufferSize, NetworkEndPoint* endPoint)
 {
     uint32 size;
-    if (endPoint == nullptr) // TCP
+    if (endPoint == nullptr)
     {
         if ((size = recv(*(SOCKET*)socket.Data, (char*)buffer, bufferSize, 0)) == SOCKET_ERROR)
         {
@@ -429,7 +431,7 @@ int32 Win32Network::ReadSocket(NetworkSocket socket, byte* buffer, uint32 buffer
             return -1;
         }
     }
-    else // UDP
+    else
     {
         int32 addrsize = sizeof sockaddr_in6;
         sockaddr_in6 addr;
diff --git a/Source/Engine/Platform/Win32/Win32Network.h b/Source/Engine/Platform/Win32/Win32Network.h
index c23056026..578c7cee2 100644
--- a/Source/Engine/Platform/Win32/Win32Network.h
+++ b/Source/Engine/Platform/Win32/Win32Network.h
@@ -8,9 +8,6 @@
 
 class FLAXENGINE_API Win32Network : public NetworkBase
 {
-    friend NetworkEndPoint;
-    friend NetworkSocket;
-
 public:
     // [NetworkBase]
     static bool CreateSocket(NetworkSocket& socket, NetworkProtocol proto, NetworkIPVersion ipv);