implement Escape handler

3 files changed, 99 insertions, 14 deletions

diff --git a/qxldod/QxlDod.cpp b/qxldod/QxlDod.cpp
index f2d9fd9..8206e63 100755
--- a/qxldod/QxlDod.cpp
+++ b/qxldod/QxlDod.cpp
@@ -1,5 +1,6 @@
 #include "driver.h"

 #include "qxldod.h"

+#include "qxl_windows.h"

 

 #pragma code_seg(push)

 #pragma code_seg()

@@ -217,7 +218,7 @@ NTSTATUS QxlDod::DispatchIoRequest(_In_  ULONG VidPnSourceId,
     UNREFERENCED_PARAMETER(VidPnSourceId);

     UNREFERENCED_PARAMETER(pVideoRequestPacket);

 

-    DbgPrint(TRACE_LEVEL_INFORMATION, ("---> %s\n", __FUNCTION__));

+    DbgPrint(TRACE_LEVEL_FATAL, ("---> %s\n", __FUNCTION__));

     return STATUS_SUCCESS;

 }

 

@@ -458,9 +459,9 @@ NTSTATUS QxlDod::Escape(_In_ CONST DXGKARG_ESCAPE* pEscape)
     PAGED_CODE();

     QXL_ASSERT(pEscape != NULL);

 

-    DbgPrint(TRACE_LEVEL_VERBOSE, ("<---> %s Flags = %d\n", __FUNCTION__, pEscape->Flags));

+    DbgPrint(TRACE_LEVEL_FATAL, ("<---> %s Flags = %d\n", __FUNCTION__, pEscape->Flags));

 

-    return STATUS_NOT_IMPLEMENTED;

+    return m_pHWDevice->Escape(pEscape);

 }

 

 

@@ -2863,6 +2864,13 @@ NTSTATUS VgaDevice::SetPointerPosition(_In_ CONST DXGKARG_SETPOINTERPOSITION* pS
     return STATUS_SUCCESS;

 }

 

+NTSTATUS VgaDevice::Escape(_In_ CONST DXGKARG_ESCAPE* pEscap)

+{

+    DbgPrint(TRACE_LEVEL_VERBOSE, ("---> %s\n", __FUNCTION__));

+    DbgPrint(TRACE_LEVEL_VERBOSE, ("<--- %s\n", __FUNCTION__));

+    return STATUS_NOT_IMPLEMENTED;

+}

+

 QxlDevice::QxlDevice(_In_ QxlDod* pQxlDod)

 {

     m_pQxlDod = pQxlDod;

@@ -2870,6 +2878,7 @@ QxlDevice::QxlDevice(_In_ QxlDod* pQxlDod)
     m_ModeCount = 0;

     m_ModeNumbers = NULL;

     m_CurrentMode = 0;

+    m_CustomMode = 0;

     m_FreeOutputs = 0;

     m_Pending = 0;

 }

@@ -2917,13 +2926,38 @@ BOOL QxlDevice::SetVideoModeInfo(UINT Idx, QXLMode* pModeInfo)
     return TRUE;

 }

 

+BOOL QxlDevice::UpdateVideoModeInfo(UINT Idx, UINT xres, UINT yres, UINT bpp)

+{

+    PVIDEO_MODE_INFORMATION pMode = NULL;

+    UINT bytes_pp = (bpp + 7) / 8;
+    ULONG color_bits;

+    PAGED_CODE();

+
+    if (xres < MIN_WIDTH_SIZE || yres < MIN_HEIGHT_SIZE || bpp != 32)
+        return FALSE;
+    pMode = &m_ModeInfo[Idx];

+    pMode->VisScreenWidth = xres;

+    pMode->VisScreenHeight = yres;

+    pMode->ScreenStride = (xres * bytes_pp + 3) & ~0x3;

+    pMode->BitsPerPlane = bpp;

+    color_bits = (bpp == 16) ? 5 : 8;

+    pMode->NumberRedBits = color_bits;

+    pMode->NumberGreenBits = color_bits;

+    pMode->NumberBlueBits = color_bits;

+

+    pMode->BlueMask = (1 << color_bits) - 1;

+    pMode->GreenMask = pMode->BlueMask << color_bits;

+    pMode->RedMask = pMode->GreenMask << color_bits;

+    return TRUE;

+}

+

 NTSTATUS QxlDevice::GetModeList(DXGK_DISPLAY_INFORMATION* pDispInfo)

 {

     PAGED_CODE();

     NTSTATUS Status = STATUS_SUCCESS;

     QXLModes *modes;

     ULONG ModeCount;

-    ULONG SuitableModeCount;

+    USHORT SuitableModeCount;

     USHORT CurrentMode;

     DbgPrint(TRACE_LEVEL_VERBOSE, ("---> %s\n", __FUNCTION__));

 

@@ -2946,7 +2980,7 @@ NTSTATUS QxlDevice::GetModeList(DXGK_DISPLAY_INFORMATION* pDispInfo)
     UINT Width = pDispInfo->Width;

     UINT BitsPerPixel = BPPFromPixelFormat(pDispInfo->ColorFormat);

     for (CurrentMode = 0, SuitableModeCount = 0;

-         CurrentMode < ModeCount;

+         CurrentMode < modes->n_modes;

          CurrentMode++)

     {

 

@@ -2958,12 +2992,12 @@ NTSTATUS QxlDevice::GetModeList(DXGK_DISPLAY_INFORMATION* pDispInfo)
             tmpModeInfo->y_res >= Height &&

             tmpModeInfo->bits == 32)

         {

-            m_ModeNumbers[SuitableModeCount] = CurrentMode;

+            m_ModeNumbers[SuitableModeCount] = SuitableModeCount;//CurrentMode;

             SetVideoModeInfo(SuitableModeCount, tmpModeInfo);

             if (tmpModeInfo->x_res == MIN_WIDTH_SIZE &&

                 tmpModeInfo->y_res == MIN_HEIGHT_SIZE)

             {

-                m_CurrentMode = (USHORT)SuitableModeCount;

+                m_CurrentMode = SuitableModeCount;

             }

             SuitableModeCount++;

         }

@@ -2976,9 +3010,17 @@ NTSTATUS QxlDevice::GetModeList(DXGK_DISPLAY_INFORMATION* pDispInfo)
     }

 

 //    m_CurrentMode = m_ModeNumbers[0];

-    m_ModeCount = SuitableModeCount;

+    m_CustomMode = SuitableModeCount;

+    for (CurrentMode = SuitableModeCount;

+         CurrentMode < SuitableModeCount + 2;

+         CurrentMode++)

+    {

+        m_ModeNumbers[CurrentMode] = CurrentMode;

+        memcpy(&m_ModeInfo[CurrentMode], &m_ModeInfo[m_CurrentMode], sizeof(VIDEO_MODE_INFORMATION));

+    }

+    m_ModeCount = SuitableModeCount + 2;

     DbgPrint(TRACE_LEVEL_ERROR, ("ModeCount filtered %d\n", m_ModeCount));

-    for (ULONG idx = 0; idx < m_ModeCount; idx++)

+    for (ULONG idx = 0; idx < GetModeCount(); idx++)

     {

         DbgPrint(TRACE_LEVEL_ERROR, ("type %x, XRes = %d, YRes = %d, BPP = %d\n",

                                     m_ModeNumbers[idx],

@@ -3002,7 +3044,7 @@ NTSTATUS QxlDevice::QueryCurrentMode(PVIDEO_MODE RequestedMode)
 NTSTATUS QxlDevice::SetCurrentMode(ULONG Mode)

 {

     DbgPrint(TRACE_LEVEL_INFORMATION, ("---> %s Mode = %x\n", __FUNCTION__, Mode));

-    for (ULONG idx = 0; idx < m_ModeCount; idx++)

+    for (ULONG idx = 0; idx < GetModeCount(); idx++)

     {

         if (Mode == m_ModeNumbers[idx])

         {

@@ -4325,7 +4367,7 @@ NTSTATUS QxlDevice::SetPointerPosition(_In_ CONST DXGKARG_SETPOINTERPOSITION* pS
         QXLCursorCmd *cursor_cmd = CursorCmd();

         if (pSetPointerPosition->X < 0) {

            cursor_cmd->type = QXL_CURSOR_HIDE;

-	} else {

+    } else {

            cursor_cmd->type = QXL_CURSOR_MOVE;

            cursor_cmd->u.position.x = (INT16)pSetPointerPosition->X;

            cursor_cmd->u.position.y = (INT16)pSetPointerPosition->Y;

@@ -4336,6 +4378,42 @@ NTSTATUS QxlDevice::SetPointerPosition(_In_ CONST DXGKARG_SETPOINTERPOSITION* pS
     return STATUS_SUCCESS;

 }

 

+NTSTATUS QxlDevice::Escape(_In_ CONST DXGKARG_ESCAPE* pEscap)

+{

+    DbgPrint(TRACE_LEVEL_VERBOSE, ("---> %s\n", __FUNCTION__));

+    QXLEscapeSetCustomDisplay *custom_display;

+    UINT xres;
+    UINT yres;
+    UINT bpp;
+

+    if (pEscap->PrivateDriverDataSize != sizeof(QXLEscapeSetCustomDisplay)) {

+        DbgPrint(TRACE_LEVEL_FATAL, ("<--> %s Incorrect buffer size %d instead of %d\n", __FUNCTION__, pEscap->PrivateDriverDataSize, sizeof(QXLEscapeSetCustomDisplay)));

+        return STATUS_INVALID_BUFFER_SIZE;

+    }

+    custom_display = (QXLEscapeSetCustomDisplay*)pEscap->pPrivateDriverData;

+    xres = custom_display->xres;
+    yres = custom_display->yres;
+    bpp = custom_display->bpp;
+
+    /* alternate custom mode index */
+    if (m_CustomMode == (m_ModeCount - 1))
+        m_CustomMode = (USHORT)(m_ModeCount - 2);
+    else
+        m_CustomMode = (USHORT)(m_ModeCount - 1);
+
+    if ((xres * yres * bpp / 8) > m_RomHdr->surface0_area_size) {
+        DbgPrint(TRACE_LEVEL_FATAL, ("%s: Mode (%dx%d#%d) doesn't fit in memory (%d)\n",
+                    __FUNCTION__, xres, yres, bpp, m_RomHdr->surface0_area_size));
+        return ERROR_NOT_ENOUGH_MEMORY;
+    }
+    if (!UpdateVideoModeInfo(m_CustomMode, xres, yres, bpp)) {

+        return ERROR_INVALID_DATA;

+    }

+    DbgPrint(TRACE_LEVEL_FATAL, ("<--> %s xres %d yres %d bpp %d\n", __FUNCTION__, xres, yres, bpp));

+    DbgPrint(TRACE_LEVEL_VERBOSE, ("<--- %s\n", __FUNCTION__));

+    return STATUS_SUCCESS;// SetCurrentMode(m_CustomMode);

+}

+

 VOID QxlDevice::WaitForCmdRing()

 {

     int wait;

diff --git a/qxldod/QxlDod.h b/qxldod/QxlDod.h
index fbbe542..3a6d009 100755
--- a/qxldod/QxlDod.h
+++ b/qxldod/QxlDod.h
@@ -223,7 +223,7 @@ public:
     virtual VOID DpcRoutine(PVOID) = 0;

     virtual VOID ResetDevice(void) = 0;

 

-    ULONG GetModeCount(void) {return m_ModeCount;}

+    virtual ULONG GetModeCount(void) = 0;

     PVIDEO_MODE_INFORMATION GetModeInfo(UINT idx) {return &m_ModeInfo[idx];}

     USHORT GetModeNumber(USHORT idx) {return m_ModeNumbers[idx];}

     USHORT GetCurrentModeIndex(void) {return m_CurrentMode;}

@@ -244,6 +244,7 @@ public:
     virtual VOID BlackOutScreen(CURRENT_BDD_MODE* pCurrentBddMod) = 0;

     virtual NTSTATUS SetPointerShape(_In_ CONST DXGKARG_SETPOINTERSHAPE* pSetPointerShape) = 0;

     virtual NTSTATUS SetPointerPosition(_In_ CONST DXGKARG_SETPOINTERPOSITION* pSetPointerPosition) = 0;

+    virtual NTSTATUS Escape(_In_ CONST DXGKARG_ESCAPE* pEscap) = 0;

     ULONG GetId(void) { return m_Id; }

 protected:

     virtual NTSTATUS GetModeList(DXGK_DISPLAY_INFORMATION* pDispInfo) = 0;

@@ -253,6 +254,7 @@ protected:
     ULONG m_ModeCount;

     PUSHORT m_ModeNumbers;

     USHORT m_CurrentMode;

+    USHORT m_CustomMode;

     ULONG  m_Id;

 };

 

@@ -265,6 +267,7 @@ public:
     NTSTATUS QueryCurrentMode(PVIDEO_MODE RequestedMode);

     NTSTATUS SetCurrentMode(ULONG Mode);

     NTSTATUS GetCurrentMode(ULONG* Mode);

+    ULONG GetModeCount(void) {return m_ModeCount;}

     NTSTATUS SetPowerState(DEVICE_POWER_STATE DevicePowerState, DXGK_DISPLAY_INFORMATION* pDispInfo);

     NTSTATUS HWInit(PCM_RESOURCE_LIST pResList, DXGK_DISPLAY_INFORMATION* pDispInfo);

     NTSTATUS HWClose(void);

@@ -286,6 +289,7 @@ public:
     VOID ResetDevice(VOID);

     NTSTATUS SetPointerShape(_In_ CONST DXGKARG_SETPOINTERSHAPE* pSetPointerShape);

     NTSTATUS SetPointerPosition(_In_ CONST DXGKARG_SETPOINTERPOSITION* pSetPointerPosition);

+    NTSTATUS Escape(_In_ CONST DXGKARG_ESCAPE* pEscap);

 protected:

     NTSTATUS GetModeList(DXGK_DISPLAY_INFORMATION* pDispInfo);

 private:

@@ -418,6 +422,7 @@ public:
     NTSTATUS QueryCurrentMode(PVIDEO_MODE RequestedMode);

     NTSTATUS SetCurrentMode(ULONG Mode);

     NTSTATUS GetCurrentMode(ULONG* Mode);

+    ULONG GetModeCount(void) {return m_ModeCount/* - 2*/;}

     NTSTATUS SetPowerState(DEVICE_POWER_STATE DevicePowerState, DXGK_DISPLAY_INFORMATION* pDispInfo);

     NTSTATUS HWInit(PCM_RESOURCE_LIST pResList, DXGK_DISPLAY_INFORMATION* pDispInfo);

     NTSTATUS HWClose(void);

@@ -439,6 +444,7 @@ public:
     VOID ResetDevice(VOID);

     NTSTATUS SetPointerShape(_In_ CONST DXGKARG_SETPOINTERSHAPE* pSetPointerShape);

     NTSTATUS SetPointerPosition(_In_ CONST DXGKARG_SETPOINTERPOSITION* pSetPointerPosition);

+    NTSTATUS Escape(_In_ CONST DXGKARG_ESCAPE* pEscap);

 protected:

     NTSTATUS GetModeList(DXGK_DISPLAY_INFORMATION* pDispInfo);

     VOID BltBits (BLT_INFO* pDst,

@@ -465,6 +471,7 @@ private:
     void QxlClose(void);

     void UnmapMemory(void);

     BOOL SetVideoModeInfo(UINT Idx, QXLMode* pModeInfo);

+    BOOL UpdateVideoModeInfo(UINT Idx, UINT xres, UINT yres, UINT bpp);

     BOOL InitMemSlots(void);

     BOOL CreateMemSlots(void);

     void DestroyMemSlots(void);

diff --git a/qxldod/driver.cpp b/qxldod/driver.cpp
index e8f48f2..3758735 100755
--- a/qxldod/driver.cpp
+++ b/qxldod/driver.cpp
@@ -28,7 +28,7 @@ DriverEntry(
     // Initialize DDI function pointers and dxgkrnl

     KMDDOD_INITIALIZATION_DATA InitialData = {0};

 

-	InitialData.Version = DXGKDDI_INTERFACE_VERSION;

+    InitialData.Version = DXGKDDI_INTERFACE_VERSION;

 

     InitialData.DxgkDdiAddDevice                    = DodAddDevice;

     InitialData.DxgkDdiStartDevice                  = DodStartDevice;

@@ -465,7 +465,7 @@ DodEnumVidPnCofuncModality(
 {

     PAGED_CODE();

     QXL_ASSERT_CHK(hAdapter != NULL);

-    DbgPrint(TRACE_LEVEL_VERBOSE, ("<---> %s\n", __FUNCTION__));

+    DbgPrint(TRACE_LEVEL_FATAL, ("<---> %s\n", __FUNCTION__));

 

     QxlDod* pQxl = reinterpret_cast<QxlDod*>(hAdapter);

     if (!pQxl->IsDriverActive())