1 files changed, 45 insertions, 0 deletions
diff --git a/wpd/WpdServiceSampleDriver/WpdBaseDriver.h b/wpd/WpdServiceSampleDriver/WpdBaseDriver.h
new file mode 100644
index 00000000..3e731f52
--- /dev/null
+++ b/wpd/WpdServiceSampleDriver/WpdBaseDriver.h
@@ -0,0 +1,45 @@
+#pragma once
+
+class WpdBaseDriver :
+    public IUnknown
+{
+public:
+    WpdBaseDriver();
+    virtual ~WpdBaseDriver();
+
+    HRESULT Initialize();
+    VOID    Uninitialize();
+
+    HRESULT DispatchWpdMessage(_In_     IPortableDeviceValues* pParams,
+                               _In_     IPortableDeviceValues* pResults);
+
+private:
+    HRESULT OnGetObjectIDsFromPersistentUniqueIDs(_In_  IPortableDeviceValues* pParams,
+                                                  _In_  IPortableDeviceValues* pResults);
+
+
+    HRESULT OnSaveClientInfo(_In_   IPortableDeviceValues* pParams,
+                             _In_   IPortableDeviceValues* pResults);
+
+public: // IUnknown
+    ULONG __stdcall AddRef();
+
+    _At_(this, __drv_freesMem(Mem)) 
+    ULONG __stdcall Release();
+
+    HRESULT __stdcall QueryInterface(REFIID riid, void** ppv);
+
+public:
+    WpdObjectEnumerator     m_ObjectEnum;
+    WpdObjectManagement     m_ObjectManagement;
+    WpdObjectProperties     m_ObjectProperties;
+    WpdObjectResources      m_ObjectResources;
+    WpdObjectPropertiesBulk m_ObjectPropertiesBulk;
+    WpdCapabilities         m_Capabilities;
+    WpdService              m_Service;
+
+private:
+    FakeDevice              m_Device;
+    ULONG                   m_cRef;
+};
+