Initial publish

1 files changed, 107 insertions, 0 deletions

diff --git a/wpd/WpdBasicHardwareDriver/WpdObjectEnum.h b/wpd/WpdBasicHardwareDriver/WpdObjectEnum.h
new file mode 100644
index 00000000..e6dc0542
--- /dev/null
+++ b/wpd/WpdBasicHardwareDriver/WpdObjectEnum.h
@@ -0,0 +1,107 @@
+#pragma once
+
+// This class is used to store the context for a specific enumeration.
+class WpdObjectEnumeratorContext : public IUnknown
+{
+public:
+    WpdObjectEnumeratorContext() :
+        m_cRef(1),
+        m_TotalChildren(0),
+        m_ChildrenEnumerated(0)
+    {
+
+    }
+
+    ~WpdObjectEnumeratorContext()
+    {
+
+    }
+
+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;
+
+public:
+    BOOL HasMoreChildrenToEnumerate()
+    {
+        return (m_TotalChildren > m_ChildrenEnumerated)?TRUE:FALSE;
+    }
+
+// WpdObjectEnumeratorContext specific data
+public:
+    CAtlStringW m_strParentObjectID;    // object identifier of the object whose children are being enumerated
+    DWORD       m_TotalChildren;        // number of bytes transferred from the resource to the caller
+    DWORD       m_ChildrenEnumerated;   // number of children returned during the enumeration operation
+};
+
+class WpdObjectEnumerator
+{
+public:
+    WpdObjectEnumerator();
+    virtual ~WpdObjectEnumerator();
+
+    HRESULT Initialize(_In_ WpdBaseDriver* pBaseDriver);
+
+    HRESULT DispatchWpdMessage(_In_ REFPROPERTYKEY         Command,
+                               _In_ IPortableDeviceValues* pParams,
+                               _In_ IPortableDeviceValues* pResults);
+
+    HRESULT OnStartFind(_In_ IPortableDeviceValues*  pParams,
+                        _In_ IPortableDeviceValues*  pResults);
+
+    HRESULT OnFindNext(_In_ IPortableDeviceValues*  pParams,
+                       _In_ IPortableDeviceValues*  pResults);
+
+    HRESULT OnEndFind(_In_ IPortableDeviceValues*  pParams,
+                      _In_ IPortableDeviceValues*  pResults);
+
+private:
+    WpdBaseDriver*           m_pBaseDriver;
+
+    VOID InitializeEnumerationContext(
+        _In_ WpdObjectEnumeratorContext* pEnumeratorContext,
+        _In_ LPCWSTR                     wszParentObjectID);
+
+    HRESULT AddStringValueToPropVariantCollection(
+        _In_ IPortableDevicePropVariantCollection* pCollection,
+        _In_ LPCWSTR                               wszValue);
+};