Implemented dumping raw WinHttpReadData results.

Author: Atvaark

Eldorado.HttpProxy/include/Logger.h

@@ -1,6 +1,6 @@
 #pragma once
-#include <utility>
 #include <fstream>
+#include <mutex>
 
 class Logger
 {
@@ -11,8 +11,9 @@ class Logger
 	void Log(std::wstring message);
 	void LogFormatA(const char* format, ...);
 	void LogFormatW(const wchar_t* format, ...);
-
+	void LogRawData(std::wstring filename, const char* data, const int dataLength);
 private:
 	std::wofstream log_stream;
+	std::mutex log_mutex;
 };
 

Eldorado.HttpProxy/module/winhttp.def

@@ -2,7 +2,7 @@ LIBRARY
 
 EXPORTS
 	WinHttpCloseHandle       @8
-    WinHttpConnect           @9
+	WinHttpConnect           @9
 	WinHttpOpen              @16
 	WinHttpOpenRequest       @17
 	WinHttpQueryHeaders      @20

Eldorado.HttpProxy/resource/Resource.rc

@@ -1,7 +1,9 @@
 #include "Logger.h"
+#include <fstream>
+#include <iomanip>
 #include <stdarg.h>
 #include <time.h>
-#include <iomanip>
+#include <Windows.h>
 
 Logger::Logger(const wchar_t* log_file)
 {
@@ -45,4 +47,25 @@ void Logger::LogFormatW(const wchar_t* format, ...)
 	vswprintf_s(message, messageSize, format, args);
 	Log(message);
 	va_end(args);
+}
+
+void Logger::LogRawData(std::wstring filePrefix, const char* data, const int dataLength)
+{
+	std::wstring filePath = L"winhttp";
+	CreateDirectory(filePath.c_str(), nullptr);	
+	wchar_t tempFilePath[MAX_PATH];
+	GetTempFileName(filePath.c_str(), filePrefix.c_str(), 0, tempFilePath);
+
+	LogFormatW(L"Logging raw data to %s", tempFilePath);
+
+	std::ofstream outfile;
+	outfile.open(tempFilePath, std::ios::binary | std::ios::out);
+	if (!outfile.is_open())
+	{
+		LogFormatW(L"Logging raw data to %s failed. Could not create temporary file.", tempFilePath);
+		return;
+	}
+	outfile.write(data, dataLength);
+	LogFormatW(L"Logged %d bytes raw data to %s", dataLength, tempFilePath);
+	outfile.close();
 }

Eldorado.HttpProxy/source/Logger.cpp

@@ -158,6 +158,7 @@ BOOL WINAPI WinHttpReadData(
 	BOOL result = oWinHttpReadData(hRequest, lpBuffer, dwNumberOfBytesToRead, lpdwNumberOfBytesRead);
 	std::string readData = std::string(static_cast<char*>(lpBuffer), dwNumberOfBytesToRead);
 	theLogger().LogFormatA("WinHttpReadData(%x, ...) returend: %s", hRequest, readData.c_str());
+	theLogger().LogRawData(L"rd_", reinterpret_cast<const char*>(lpBuffer), dwNumberOfBytesToRead);
 	return result;
 }