Initial publish

1 files changed, 112 insertions, 0 deletions

diff --git a/wpd/WpdServiceSampleDriver/WpdObjectResources.h b/wpd/WpdServiceSampleDriver/WpdObjectResources.h
new file mode 100644
index 00000000..1f8c4a0a
--- /dev/null
+++ b/wpd/WpdServiceSampleDriver/WpdObjectResources.h
@@ -0,0 +1,112 @@
+#pragma once
+
+#define FILE_OPTIMAL_READ_BUFFER_SIZE_VALUE  (2 * 1024 * 1024)
+#define FILE_OPTIMAL_WRITE_BUFFER_SIZE_VALUE (2 * 1024 * 1024)
+
+// This class is used to store the context for a specific resource operation.
+class WpdObjectResourceContext : public IUnknown
+{
+public:
+    WpdObjectResourceContext() :
+        m_cRef(1),
+        m_Resource(WPD_RESOURCE_DEFAULT),
+        m_BytesTransferred(0),
+        m_BytesTotal(0),
+        m_Scope(FULL_DEVICE_ACCESS)
+    {
+
+    }
+
+    ~WpdObjectResourceContext()
+    {
+
+    }
+
+public: // IUnknown
+    ULONG __stdcall AddRef()
+    {
+        InterlockedIncrement((long*) &m_cRef);
+        return m_cRef;
+    }
+
+    _At_(this, __drv_freesMem(Mem)) 
+    ULONG __stdcall Release()
+    {
+        ULONG ulRefCount = m_cRef - 1;
+
+        if (InterlockedDecrement((long*) &m_cRef) == 0)
+        {
+            delete this;
+            return 0;
+        }
+        return ulRefCount;
+    }
+
+    HRESULT __stdcall QueryInterface(
+        REFIID riid,
+        void** ppv)
+    {
+        HRESULT hr = S_OK;
+
+        if(riid == IID_IUnknown)
+        {
+            *ppv = static_cast<IUnknown*>(this);
+            AddRef();
+        }
+        else
+        {
+            *ppv = NULL;
+            hr = E_NOINTERFACE;
+        }
+
+        return hr;
+    }
+
+private:
+    DWORD m_cRef;
+
+// WpdObjectResourceContext specific data
+public:
+    CAtlStringW  m_strObjectID;      // object identifier of the object whose resource is being transferred
+    PROPERTYKEY  m_Resource;         // the specific resource being transferred
+    ULONGLONG    m_BytesTransferred; // number of bytes transferred from the resource to the caller
+    ULONGLONG    m_BytesTotal;       // total number of bytes of the resource data
+    ACCESS_SCOPE m_Scope;            // client access scope
+};
+
+class WpdObjectResources
+{
+public:
+    WpdObjectResources();
+    virtual ~WpdObjectResources();
+    
+    HRESULT Initialize(_In_ FakeDevice* pDevice);
+    
+    HRESULT DispatchWpdMessage(
+        _In_     REFPROPERTYKEY         Command,
+        _In_     IPortableDeviceValues* pParams,
+        _In_     IPortableDeviceValues* pResults);
+
+    HRESULT OnGetSupportedResources(
+        _In_    IPortableDeviceValues*  pParams,
+        _In_    IPortableDeviceValues*  pResults);
+
+    HRESULT OnGetResourceAttributes(
+        _In_    IPortableDeviceValues*  pParams,
+        _In_    IPortableDeviceValues*  pResults);
+
+    HRESULT OnOpenResource(
+        _In_    IPortableDeviceValues*  pParams,
+        _In_    IPortableDeviceValues*  pResults);
+
+    HRESULT OnReadResource(
+        _In_    IPortableDeviceValues*  pParams,
+        _In_    IPortableDeviceValues*  pResults);
+
+    HRESULT OnCloseResource(
+        _In_    IPortableDeviceValues*  pParams,
+        _In_    IPortableDeviceValues*  pResults);
+
+private:
+    FakeDevice* m_pDevice;
+};