Stub implementation of IMMNotificationClient.

Author: eiz

SarAsio/mmwrapper.h

@@ -28,7 +28,7 @@ struct ATL_NO_VTABLE SarMMDeviceEnumerator:
 {
     BEGIN_COM_MAP(SarMMDeviceEnumerator)
         COM_INTERFACE_ENTRY(IMMDeviceEnumerator)
-    END_COM_MAP();
+    END_COM_MAP()
 
     DECLARE_REGISTRY_RESOURCEID(IDR_SARMMDEVICE)
 

SarAsio/sarclient.cpp

@@ -169,6 +169,12 @@ bool SarClient::start()
         return false;
     }
 
+    if (!openMmNotificationClient()) {
+        OutputDebugString(_T("Couldn't open MMDevice notification client"));
+        stop();
+        return false;
+    }
+
     if (!setBufferLayout()) {
         OutputDebugString(_T("Couldn't set layout"));
         stop();
@@ -190,6 +196,21 @@ bool SarClient::start()
 
 void SarClient::stop()
 {
+    if (_mmNotificationClientRegistered) {
+        _mmEnumerator->UnregisterEndpointNotificationCallback(
+            _mmNotificationClient);
+        _mmNotificationClientRegistered = false;
+    }
+
+    if (_mmNotificationClient) {
+        _mmNotificationClient->Release();
+        _mmNotificationClient = nullptr;
+    }
+
+    if (_mmEnumerator) {
+        _mmEnumerator = nullptr;
+    }
+
     if (_device != INVALID_HANDLE_VALUE) {
         CancelIoEx(_device, nullptr);
         CloseHandle(_device);
@@ -267,6 +288,33 @@ bool SarClient::openControlDevice()
     return true;
 }
 
+bool SarClient::openMmNotificationClient()
+{
+    if (FAILED(CoCreateInstance(
+        __uuidof(MMDeviceEnumerator), nullptr, CLSCTX_ALL,
+        __uuidof(IMMDeviceEnumerator), (LPVOID *)&_mmEnumerator))) {
+
+        return false;
+    }
+
+    if (FAILED(CComObject<NotificationClient>::CreateInstance(
+        &_mmNotificationClient))) {
+
+        return false;
+    }
+
+    _mmNotificationClient->AddRef();
+
+    if (FAILED(_mmEnumerator->RegisterEndpointNotificationCallback(
+        _mmNotificationClient))) {
+
+        return false;
+    }
+
+    _mmNotificationClientRegistered = true;
+    return true;
+}
+
 bool SarClient::setBufferLayout()
 {
     SarSetBufferLayoutRequest request = {};
@@ -470,4 +518,43 @@ void SarClient::mux(
     }
 }
 
+HRESULT STDMETHODCALLTYPE SarClient::NotificationClient::OnDeviceStateChanged(
+    _In_  LPCWSTR pwstrDeviceId,
+    _In_  DWORD dwNewState)
+{
+    OutputDebugString(_T("OnDeviceStateChanged"));
+    return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE SarClient::NotificationClient::OnDeviceAdded(
+    _In_  LPCWSTR pwstrDeviceId)
+{
+    OutputDebugString(_T("OnDeviceAdded"));
+    return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE SarClient::NotificationClient::OnDeviceRemoved(
+    _In_  LPCWSTR pwstrDeviceId)
+{
+    OutputDebugString(_T("OnDeviceRemoved"));
+    return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE SarClient::NotificationClient::OnDefaultDeviceChanged(
+    _In_  EDataFlow flow,
+    _In_  ERole role,
+    _In_  LPCWSTR pwstrDefaultDeviceId)
+{
+    OutputDebugString(_T("OnDefaultDeviceChanged"));
+    return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE SarClient::NotificationClient::OnPropertyValueChanged(
+    _In_  LPCWSTR pwstrDeviceId,
+    _In_  const PROPERTYKEY key)
+{
+    OutputDebugString(_T("OnPropertyValueChanged"));
+    return S_OK;
+}
+
 } // namespace Sar

SarAsio/sarclient.h

@@ -59,7 +59,34 @@ struct SarClient
         SarHandleQueueResponse responses[32];
     };
 
+    struct ATL_NO_VTABLE NotificationClient:
+        public CComObjectRootEx<CComMultiThreadModel>,
+        public IMMNotificationClient
+    {
+        BEGIN_COM_MAP(NotificationClient)
+            COM_INTERFACE_ENTRY(IMMNotificationClient)
+        END_COM_MAP()
+
+        DECLARE_NO_REGISTRY()
+
+        virtual HRESULT STDMETHODCALLTYPE OnDeviceStateChanged(
+            _In_  LPCWSTR pwstrDeviceId,
+            _In_  DWORD dwNewState) override;
+        virtual HRESULT STDMETHODCALLTYPE OnDeviceAdded(
+            _In_  LPCWSTR pwstrDeviceId) override;
+        virtual HRESULT STDMETHODCALLTYPE OnDeviceRemoved(
+            _In_  LPCWSTR pwstrDeviceId) override;
+        virtual HRESULT STDMETHODCALLTYPE OnDefaultDeviceChanged(
+            _In_  EDataFlow flow,
+            _In_  ERole role,
+            _In_  LPCWSTR pwstrDefaultDeviceId) override;
+        virtual HRESULT STDMETHODCALLTYPE OnPropertyValueChanged(
+            _In_  LPCWSTR pwstrDeviceId,
+            _In_  const PROPERTYKEY key) override;
+    };
+
     bool openControlDevice();
+    bool openMmNotificationClient();
     bool setBufferLayout();
     bool createEndpoints();
     bool enableRegistryFilter();
@@ -87,6 +114,9 @@ struct SarClient
     volatile SarEndpointRegisters *_registers;
     HandleQueueCompletion _handleQueueCompletion;
     bool _handleQueueStarted;
+    CComPtr<IMMDeviceEnumerator> _mmEnumerator;
+    CComObject<NotificationClient> *_mmNotificationClient = nullptr;
+    bool _mmNotificationClientRegistered = false;
 };
 
 } // namespace Sar