Initial publish

1 files changed, 93 insertions, 0 deletions

diff --git a/wpd/WpdBasicHardwareDriver/Queue.h b/wpd/WpdBasicHardwareDriver/Queue.h
new file mode 100644
index 00000000..e5cea69d
--- /dev/null
+++ b/wpd/WpdBasicHardwareDriver/Queue.h
@@ -0,0 +1,93 @@
+// Queue.h : Declaration of the CQueue
+
+#pragma once
+#include "resource.h"       // main symbols
+#include "WpdBasicHardwareDriver.h"
+
+class ATL_NO_VTABLE CQueue :
+    public CComObjectRootEx<CComMultiThreadModel>,
+    public IQueueCallbackDeviceIoControl,
+    public IQueueCallbackCreate,
+    public IObjectCleanup
+{
+public:
+    CQueue()
+    {
+
+    }
+
+    DECLARE_NOT_AGGREGATABLE(CQueue)
+
+    BEGIN_COM_MAP(CQueue)
+        COM_INTERFACE_ENTRY(IQueueCallbackDeviceIoControl)
+        COM_INTERFACE_ENTRY(IQueueCallbackCreate)
+    END_COM_MAP()
+
+public:
+    static
+    HRESULT CreateInstance(
+        _COM_Outptr_ IUnknown   **ppUkwn)
+    {
+        *ppUkwn = NULL;
+        CComObject< CQueue> *pMyQueue = NULL;
+        HRESULT hr = CComObject<CQueue>::CreateInstance( &pMyQueue );
+        if( SUCCEEDED (hr) )
+        {
+            pMyQueue->AddRef();
+            hr = pMyQueue->QueryInterface( __uuidof(IUnknown), (void **) ppUkwn );
+            pMyQueue->Release();
+            pMyQueue = NULL;
+        }
+
+        return hr;
+    }
+
+    //
+    // Wdf Callbacks
+    //
+
+    // IQueueCallbackCreateClose
+    //
+    STDMETHOD_ (void, OnCreateFile)(
+        _In_ IWDFIoQueue *pQueue,
+        _In_ IWDFIoRequest *pRequest,
+        _In_ IWDFFile *pFileObject
+        );
+
+    //
+    // IQueueCallbackDeviceIoControl
+    //
+    STDMETHOD_ (void, OnDeviceIoControl)(
+        _In_ IWDFIoQueue*    pQueue,
+        _In_ IWDFIoRequest*  pRequest,
+             ULONG           ControlCode,
+             SIZE_T          InputBufferSizeInBytes,
+             SIZE_T          OutputBufferSizeInBytes
+        );
+
+    //
+    // IObjectCleanup
+    //
+    STDMETHOD_ (void, OnCleanup)(
+        _In_ IWDFObject* pWdfObject
+        );
+
+private:
+    HRESULT ProcessWpdMessage(
+                 ULONG       ControlCode,
+        _In_     ContextMap* pClientContextMap,
+        _In_     IWDFDevice* pDevice,
+        _In_reads_bytes_(ulInputBufferLength) PVOID pInBuffer,
+                 ULONG       ulInputBufferLength,
+        _Out_writes_bytes_to_(ulOutputBufferLength, *pdwBytesWritten) PVOID pOutBuffer,
+                 ULONG       ulOutputBufferLength,
+        _Out_    DWORD*      pdwBytesWritten);
+
+    HRESULT GetWpdBaseDriver(
+        _In_                          IWDFDevice*     pDevice,
+        _Outptr_result_nullonfailure_ WpdBaseDriver** ppWpdBaseDriver);
+
+    CComPtr<IWpdSerializer>         m_pWpdSerializer;
+    CComAutoCriticalSection         m_CriticalSection;
+};
+