From 2a8e8f65d6336622fabedca5587187f349c93cdb Mon Sep 17 00:00:00 2001
From: Vincent Richard <vincent@vincent-richard.net>
Date: Thu, 16 Jan 2014 19:17:50 +0100
Subject: [PATCH] Fixes for socket using 'byte_t' type.

---
 src/vmime/platforms/windows/windowsSocket.cpp | 15 ++++++++-------
 src/vmime/platforms/windows/windowsSocket.hpp | 11 ++++++-----
 2 files changed, 14 insertions(+), 12 deletions(-)

diff --git a/src/vmime/platforms/windows/windowsSocket.cpp b/src/vmime/platforms/windows/windowsSocket.cpp
index 40e69363..cb96481a 100644
--- a/src/vmime/platforms/windows/windowsSocket.cpp
+++ b/src/vmime/platforms/windows/windowsSocket.cpp
@@ -31,6 +31,7 @@
 
 #include "vmime/platforms/windows/windowsSocket.hpp"
 
+#include "vmime/utility/stringUtils.hpp"
 #include "vmime/exception.hpp"
 
 #include <ws2tcpip.h>
@@ -239,11 +240,11 @@ size_t windowsSocket::getBlockSize() const
 void windowsSocket::receive(vmime::string& buffer)
 {
 	const size_t size = receiveRaw(m_buffer, sizeof(m_buffer));
-	buffer = vmime::string(m_buffer, size);
+	buffer = utility::stringUtils::makeStringFromBytes(m_buffer, size);
 }
 
 
-size_t windowsSocket::receiveRaw(char* buffer, const size_t count)
+size_t windowsSocket::receiveRaw(byte_t* buffer, const size_t count)
 {
 	m_status &= ~STATUS_WOULDBLOCK;
 
@@ -275,7 +276,7 @@ size_t windowsSocket::receiveRaw(char* buffer, const size_t count)
 	}
 
 	// Read available data
-	int ret = ::recv(m_desc, buffer, count, 0);
+	int ret = ::recv(m_desc, reinterpret_cast <char*>(buffer), count, 0);
 
 	if (ret == SOCKET_ERROR)
 	{
@@ -317,7 +318,7 @@ void windowsSocket::send(const char* str)
 }
 
 
-void windowsSocket::sendRaw(const char* buffer, const size_t count)
+void windowsSocket::sendRaw(const byte_t* buffer, const size_t count)
 {
 	m_status &= ~STATUS_WOULDBLOCK;
 
@@ -325,7 +326,7 @@ void windowsSocket::sendRaw(const char* buffer, const size_t count)
 
 	while (size > 0)
 	{
-		const int ret = ::send(m_desc, buffer, size, 0);
+		const int ret = ::send(m_desc, reinterpret_cast <const char*>(buffer), size, 0);
 
 		if (ret == SOCKET_ERROR)
 		{
@@ -350,11 +351,11 @@ void windowsSocket::sendRaw(const char* buffer, const size_t count)
 }
 
 
-size_t windowsSocket::sendRawNonBlocking(const char* buffer, const size_t count)
+size_t windowsSocket::sendRawNonBlocking(const byte_t* buffer, const size_t count)
 {
 	m_status &= ~STATUS_WOULDBLOCK;
 
-	const int ret = ::send(m_desc, buffer, count, 0);
+	const int ret = ::send(m_desc, reinterpret_cast <const char*>(buffer), count, 0);
 
 	if (ret == SOCKET_ERROR)
 	{
diff --git a/src/vmime/platforms/windows/windowsSocket.hpp b/src/vmime/platforms/windows/windowsSocket.hpp
index e3f43b7c..cb8a6e67 100644
--- a/src/vmime/platforms/windows/windowsSocket.hpp
+++ b/src/vmime/platforms/windows/windowsSocket.hpp
@@ -55,11 +55,12 @@ public:
 	void disconnect();
 
 	void receive(vmime::string& buffer);
-	size_t receiveRaw(char* buffer, const size_t count);
-
+	size_t receiveRaw(byte_t* buffer, const size_t count);
+	
 	void send(const vmime::string& buffer);
-	void sendRaw(const char* buffer, const size_t count);
-	size_t sendRawNonBlocking(const char* buffer, const size_t count);
+	void send(const char* str);
+	void sendRaw(const byte_t* buffer, const size_t count);
+	size_t sendRawNonBlocking(const byte_t* buffer, const size_t count);
 
 	size_t getBlockSize() const;
 
@@ -85,7 +86,7 @@ private:
 
 	shared_ptr <vmime::net::timeoutHandler> m_timeoutHandler;
 
-	char m_buffer[65536];
+	byte_t m_buffer[65536];
 	SOCKET m_desc;
 
 	unsigned int m_status;