From 474783bc46e4ae58ce64129a645e01cbc27458a9 Mon Sep 17 00:00:00 2001
From: Jojo-1000 <33495614+Jojo-1000@users.noreply.github.com>
Date: Thu, 1 Jul 2021 17:31:52 +0200
Subject: [PATCH] Fix WinHttpHandler send function.

---
 src/BaseHttpHandler.cpp | 22 ++++++++++++++--------
 src/WinHttpHandler.cpp  | 23 +++++++++++------------
 2 files changed, 25 insertions(+), 20 deletions(-)

diff --git a/src/BaseHttpHandler.cpp b/src/BaseHttpHandler.cpp
index aeef09b..d0fc846 100644
--- a/src/BaseHttpHandler.cpp
+++ b/src/BaseHttpHandler.cpp
@@ -56,14 +56,20 @@ std::string BaseHttpHandler::sendHTTPRequest(const std::string& method, const st
     request.append("HTTP/1.0"); // HTTP-Version
     request.append("\r\n"); // Ending
                             // Entities
-    request.append("Content-Type:"); // entity-header
-    request.append(" "); // Separation
-    request.append(contentType); // media-type
-    request.append("\r\n"); // Entity ending
-    request.append("Content-Length:"); // entity-header
-    request.append(" "); // Separation
-    request.append(std::to_string(body.size())); // length
-    request.append("\r\n\r\n"); // Entity ending & Request-Line ending
+    if (!contentType.empty())
+    {
+        request.append("Content-Type:"); // entity-header
+        request.append(" "); // Separation
+        request.append(contentType); // media-type
+        request.append("\r\n"); // Entity ending
+    }
+    if (!body.empty())
+    {
+        request.append("Content-Length:"); // entity-header
+        request.append(" "); // Separation
+        request.append(std::to_string(body.size())); // length
+        request.append("\r\n\r\n"); // Entity ending & Request-Line ending
+    }
     request.append(body); // message-body
     request.append("\r\n\r\n"); // Ending
 
diff --git a/src/WinHttpHandler.cpp b/src/WinHttpHandler.cpp
index 6014e97..23f0b34 100644
--- a/src/WinHttpHandler.cpp
+++ b/src/WinHttpHandler.cpp
@@ -39,7 +39,7 @@ namespace
 class AddrInfoFreer
 {
 public:
-    explicit AddrInfoFreer(addrinfo* p) : p(p) {}
+    explicit AddrInfoFreer(addrinfo* p) : p(p) { }
     ~AddrInfoFreer() { freeaddrinfo(p); }
 
 private:
@@ -48,7 +48,7 @@ private:
 class SocketCloser
 {
 public:
-    explicit SocketCloser(SOCKET s) : s(s) {}
+    explicit SocketCloser(SOCKET s) : s(s) { }
     ~SocketCloser() { closesocket(s); }
 
 private:
@@ -135,15 +135,6 @@ std::string WinHttpHandler::send(const std::string& msg, const std::string& adr,
         throw(std::system_error(err, std::system_category(), "WinHttpHandler: send failed"));
     }
 
-    // shutdown the connection for sending since no more data will be sent
-    // the client can still use the ConnectSocket for receiving data
-    if (shutdown(connect_socket, SD_SEND) == SOCKET_ERROR)
-    {
-        int err = WSAGetLastError();
-        std::cerr << "WinHttpHandler: shutdown failed: " << err << std::endl;
-        throw(std::system_error(err, std::system_category(), "WinHttpHandler: shutdown failed"));
-    }
-
     const int recvbuflen = 128;
     char recvbuf[recvbuflen];
 
@@ -170,6 +161,14 @@ std::string WinHttpHandler::send(const std::string& msg, const std::string& adr,
         }
     } while (res > 0);
 
+    // shutdown the connection
+    if (shutdown(connect_socket, SD_BOTH) == SOCKET_ERROR)
+    {
+        int err = WSAGetLastError();
+        std::cerr << "WinHttpHandler: shutdown failed: " << err << std::endl;
+        throw(std::system_error(err, std::system_category(), "WinHttpHandler: shutdown failed"));
+    }
+
     return response;
 }
 
@@ -250,7 +249,7 @@ std::vector<std::string> WinHttpHandler::sendMulticast(
     }
 
     // shutdown the connection for sending since no more data will be sent
-    // the client can still use the ConnectSocket for receiving data
+    // the client can still use the ConnectSocket for receiving data (no issue here because this is a UDP socket)
     if (shutdown(connect_socket, SD_SEND) == SOCKET_ERROR)
     {
         int err = WSAGetLastError();
--
libgit2 0.21.4