Test

Author: itsmikethetech

Virtual Display Driver (HDR)/MttVDD/Common.h

@@ -0,0 +1,12 @@
+#pragma once
+
+#include <string>
+#include <chrono>
+#include <windows.h>
+
+// Forward declaration for the logging function used throughout the driver
+void vddlog(const char* type, const char* message);
+
+// Forward declarations for existing string conversion functions
+std::string WStringToString(const std::wstring& wstr);
+std::wstring StringToWString(const std::string& str);

Virtual Display Driver (HDR)/MttVDD/Driver.cpp

@@ -79,6 +79,10 @@ EVT_IDD_CX_MONITOR_QUERY_TARGET_MODES2 VirtualDisplayDriverEvtIddCxMonitorQueryT
 EVT_IDD_CX_ADAPTER_COMMIT_MODES2 VirtualDisplayDriverEvtIddCxAdapterCommitModes2;
 
 EVT_IDD_CX_MONITOR_SET_GAMMA_RAMP VirtualDisplayDriverEvtIddCxMonitorSetGammaRamp;
+EVT_IDD_CX_DEVICE_IO_CONTROL MttVddEvtIoDeviceControl;
+
+// Forward declaration for IOCTL queue setup function
+NTSTATUS SetupIoctlQueue(WDFDEVICE Device);
 
 struct
 {
@@ -152,16 +156,7 @@ const char* XorCursorSupportLevelToString(IDDCX_XOR_CURSOR_SUPPORT level) {
 
 vector<unsigned char> Microsoft::IndirectDisp::IndirectDeviceContext::s_KnownMonitorEdid; //Changed to support static vector
 
-struct IndirectDeviceContextWrapper
-{
-	IndirectDeviceContext* pContext;
 
-	void Cleanup()
-	{
-		delete pContext;
-		pContext = nullptr;
-	}
-};
 void LogQueries(const char* severity, const std::wstring& xmlName) {
 	if (xmlName.find(L"logging") == std::wstring::npos) { 
 		int size_needed = WideCharToMultiByte(CP_UTF8, 0, xmlName.c_str(), (int)xmlName.size(), NULL, 0, NULL, NULL);
@@ -634,10 +629,6 @@ void InitializeD3DDeviceAndLogGPU() {
 	vddlog("i", logtext.c_str());
 }
 
-
-// This macro creates the methods for accessing an IndirectDeviceContextWrapper as a context for a WDF object
-WDF_DECLARE_CONTEXT_TYPE(IndirectDeviceContextWrapper);
-
 extern "C" BOOL WINAPI DllMain(
 	_In_ HINSTANCE hInstance,
 	_In_ UINT dwReason,
@@ -1783,6 +1774,9 @@ NTSTATUS VirtualDisplayDriverDeviceAdd(WDFDRIVER Driver, PWDFDEVICE_INIT pDevice
 	IddConfig.EvtIddCxMonitorAssignSwapChain = VirtualDisplayDriverMonitorAssignSwapChain;
 	IddConfig.EvtIddCxMonitorUnassignSwapChain = VirtualDisplayDriverMonitorUnassignSwapChain;
 
+	// Enable IOCTL handling
+	IddConfig.EvtIddCxDeviceIoControl = MttVddEvtIoDeviceControl;
+
 	if (IDD_IS_FIELD_AVAILABLE(IDD_CX_CLIENT_CONFIG, EvtIddCxAdapterQueryTargetInfo))
 	{
 		IddConfig.EvtIddCxAdapterQueryTargetInfo = VirtualDisplayDriverEvtIddCxAdapterQueryTargetInfo;

Virtual Display Driver (HDR)/MttVDD/Driver.h

@@ -16,6 +16,28 @@
 #include <vector>
 
 #include "Trace.h"
+#include "mttvdd-ioctl.h"
+#include "Common.h"
+
+// Forward declaration for the IndirectDeviceContext class
+namespace Microsoft
+{
+    namespace IndirectDisp
+    {
+        class IndirectDeviceContext;
+    }
+}
+
+// Forward declarations for driver-defined context type
+struct IndirectDeviceContextWrapper
+{
+    Microsoft::IndirectDisp::IndirectDeviceContext* pContext;
+
+    void Cleanup();  // Just declare it, don't implement
+};
+
+// The WDF context access function is defined by WDF_DECLARE_CONTEXT_TYPE macro
+WDF_DECLARE_CONTEXT_TYPE(IndirectDeviceContextWrapper);
 
 namespace Microsoft
 {
@@ -85,18 +107,26 @@ namespace Microsoft
             void FinishInit();
 
             void CreateMonitor(unsigned int index);
+            NTSTATUS CreateMonitorWithParams(ULONG Width, ULONG Height, ULONG RefreshRate, GUID* MonitorId);
+            NTSTATUS DestroyMonitor(ULONG MonitorIndex);
+            NTSTATUS SetMonitorMode(ULONG MonitorIndex, ULONG Width, ULONG Height, ULONG RefreshRate);
+            NTSTATUS SetPreferredMode(ULONG MonitorIndex, ULONG Width, ULONG Height, ULONG RefreshRate);
+            NTSTATUS SetGpuPreference(LUID AdapterLuid);
+            ULONG GetMonitorCount();
 
             void AssignSwapChain(IDDCX_MONITOR& Monitor, IDDCX_SWAPCHAIN SwapChain, LUID RenderAdapter, HANDLE NewFrameEvent);
             void UnassignSwapChain();
 
         protected:
-
             WDFDEVICE m_WdfDevice;
             IDDCX_ADAPTER m_Adapter;
             IDDCX_MONITOR m_Monitor;
             IDDCX_MONITOR m_Monitor2;
 
             std::unique_ptr<SwapChainProcessor> m_ProcessingThread;
+            
+            // Store a list of connected monitors and their info
+            std::vector<std::pair<IDDCX_MONITOR, GUID>> m_Monitors;
 
         public:
             static const DISPLAYCONFIG_VIDEO_SIGNAL_INFO s_KnownMonitorModes[];

Virtual Display Driver (HDR)/MttVDD/MttVDD.inf

@@ -36,12 +36,16 @@
   </ItemGroup>
   <ItemGroup>
     <ClCompile Include="Driver.cpp" />
+    <ClCompile Include="mttioctl.cpp">
+      <ExcludedFromBuild Condition="'$(Configuration)|$(Platform)'=='Release|x64'">false</ExcludedFromBuild>
+    </ClCompile>
   </ItemGroup>
   <ItemGroup>
     <ClInclude Include="..\..\..\..\Downloads\wudfwdm.h">
       <DeploymentContent>true</DeploymentContent>
     </ClInclude>
     <ClInclude Include="Driver.h" />
+    <ClInclude Include="mttvdd-ioctl.h" />
     <ClInclude Include="Trace.h" />
   </ItemGroup>
   <ItemGroup>