feat: add audio device changes detect for windows. (#41)

* feat: add audio device changes detect for windows.

* Update audio_device_core_win.cc

Author: cloudwebrtc

modules/audio_device/audio_device_generic.h

@@ -135,6 +135,8 @@ class AudioDeviceGeneric {
   virtual int GetRecordAudioParameters(AudioParameters* params) const;
 #endif  // WEBRTC_IOS
 
+  virtual int32_t SetAudioDeviceSink(AudioDeviceSink* sink) = 0;
+
   virtual void AttachAudioBuffer(AudioDeviceBuffer* audioBuffer) = 0;
 
   virtual ~AudioDeviceGeneric() {}

modules/audio_device/dummy/audio_device_dummy.cc

@@ -222,5 +222,9 @@ int32_t AudioDeviceDummy::PlayoutDelay(uint16_t& delayMS) const {
   return -1;
 }
 
+int32_t AudioDeviceDummy::SetAudioDeviceSink(AudioDeviceSink* sink) {
+  return -1;
+}
+
 void AudioDeviceDummy::AttachAudioBuffer(AudioDeviceBuffer* audioBuffer) {}
 }  // namespace webrtc

modules/audio_device/dummy/audio_device_dummy.h

@@ -109,7 +109,10 @@ class AudioDeviceDummy : public AudioDeviceGeneric {
   // Delay information and control
   int32_t PlayoutDelay(uint16_t& delayMS) const override;
 
+  int32_t SetAudioDeviceSink(AudioDeviceSink* sink) override;
+
   void AttachAudioBuffer(AudioDeviceBuffer* audioBuffer) override;
+
 };
 
 }  // namespace webrtc

modules/audio_device/dummy/file_audio_device.cc

@@ -427,6 +427,10 @@ int32_t FileAudioDevice::PlayoutDelay(uint16_t& delayMS) const {
   return 0;
 }
 
+int32_t FileAudioDevice::SetAudioDeviceSink(AudioDeviceSink* sink) {
+  return -1;
+}
+
 void FileAudioDevice::AttachAudioBuffer(AudioDeviceBuffer* audioBuffer) {
   MutexLock lock(&mutex_);
 

modules/audio_device/dummy/file_audio_device.h

@@ -123,6 +123,8 @@ class FileAudioDevice : public AudioDeviceGeneric {
   // Delay information and control
   int32_t PlayoutDelay(uint16_t& delayMS) const override;
 
+  int32_t SetAudioDeviceSink(AudioDeviceSink* sink) override;
+
   void AttachAudioBuffer(AudioDeviceBuffer* audioBuffer) override;
 
  private:

modules/audio_device/win/audio_device_core_win.cc

@@ -462,6 +462,8 @@ AudioDeviceWindowsCore::AudioDeviceWindowsCore()
   _playChannelsPrioList[0] = 2;  // stereo is prio 1
   _playChannelsPrioList[1] = 1;  // mono is prio 2
 
+  _deviceStateListener = new DeviceStateListener();
+
   HRESULT hr;
 
   // We know that this API will work since it has already been verified in
@@ -475,6 +477,8 @@ AudioDeviceWindowsCore::AudioDeviceWindowsCore()
                    reinterpret_cast<void**>(&_ptrEnumerator));
   RTC_DCHECK(_ptrEnumerator);
 
+  _ptrEnumerator->RegisterEndpointNotificationCallback(_deviceStateListener);
+
   // DMO initialization for built-in WASAPI AEC.
   {
     IMediaObject* ptrDMO = NULL;
@@ -500,6 +504,8 @@ AudioDeviceWindowsCore::~AudioDeviceWindowsCore() {
 
   Terminate();
 
+  _ptrEnumerator->UnregisterEndpointNotificationCallback(_deviceStateListener);
+
   // The IMMDeviceEnumerator is created during construction. Must release
   // it here and not in Terminate() since we don't recreate it in Init().
   SAFE_RELEASE(_ptrEnumerator);
@@ -536,6 +542,11 @@ AudioDeviceWindowsCore::~AudioDeviceWindowsCore() {
     _hShutdownCaptureEvent = NULL;
   }
 
+  if(NULL != _deviceStateListener) {
+    delete _deviceStateListener;
+    _deviceStateListener = NULL;
+  }
+
   if (_avrtLibrary) {
     BOOL freeOK = FreeLibrary(_avrtLibrary);
     if (!freeOK) {
@@ -3894,6 +3905,65 @@ int32_t AudioDeviceWindowsCore::_GetDeviceID(IMMDevice* pDevice,
   return 0;
 }
 
+int32_t AudioDeviceWindowsCore::SetAudioDeviceSink(AudioDeviceSink* sink) {
+  _deviceStateListener->SetAudioDeviceSink(sink);
+  return 0;
+}
+
+void AudioDeviceWindowsCore::DeviceStateListener::SetAudioDeviceSink(AudioDeviceSink *sink) {
+  callback_ = sink;
+}
+
+HRESULT AudioDeviceWindowsCore::DeviceStateListener::OnDeviceStateChanged(LPCWSTR pwstrDeviceId, DWORD dwNewState) {
+  RTC_DLOG(LS_INFO) << "AudioDeviceWindowsCore::OnDeviceStateChanged => " << pwstrDeviceId << ", NewState => " << dwNewState;
+  if(callback_) callback_->OnDevicesUpdated();
+  return S_OK;
+}
+
+HRESULT AudioDeviceWindowsCore::DeviceStateListener::OnDeviceAdded(LPCWSTR pwstrDeviceId) {
+  RTC_DLOG(LS_INFO) << "AudioDeviceWindowsCore::OnDeviceAdded => " << pwstrDeviceId;
+  return S_OK;
+}
+
+HRESULT AudioDeviceWindowsCore::DeviceStateListener::OnDeviceRemoved(LPCWSTR pwstrDeviceId) {
+  RTC_DLOG(LS_INFO) << "AudioDeviceWindowsCore::OnDeviceRemoved => " << pwstrDeviceId;
+  return S_OK;
+}
+
+HRESULT AudioDeviceWindowsCore::DeviceStateListener::OnDefaultDeviceChanged(EDataFlow flow, ERole role, LPCWSTR pwstrDefaultDeviceId) {
+  RTC_DLOG(LS_INFO) << "AudioDeviceWindowsCore::OnDefaultDeviceChanged => " << pwstrDefaultDeviceId;
+  return S_OK;
+}
+
+HRESULT AudioDeviceWindowsCore::DeviceStateListener::OnPropertyValueChanged(LPCWSTR pwstrDeviceId, const PROPERTYKEY key) {
+  //RTC_DLOG(LS_INFO) << "AudioDeviceWindowsCore::OnPropertyValueChanged => " << pwstrDeviceId;
+  return S_OK;
+}
+
+ULONG AudioDeviceWindowsCore::DeviceStateListener::AddRef() {
+  ULONG new_ref = InterlockedIncrement(&ref_count_);
+  // RTC_DLOG(LS_INFO) << "__AddRef => " << new_ref;
+  return new_ref;
+}
+
+ULONG AudioDeviceWindowsCore::DeviceStateListener::Release() {
+  ULONG new_ref = InterlockedDecrement(&ref_count_);
+  // RTC_DLOG(LS_INFO) << "__Release => " << new_ref;
+  return new_ref;
+}
+
+HRESULT AudioDeviceWindowsCore::DeviceStateListener::QueryInterface(REFIID iid, void** object) {
+  if (object == nullptr) {
+    return E_POINTER;
+  }
+  if (iid == IID_IUnknown || iid == __uuidof(IMMNotificationClient)) {
+    *object = static_cast<IMMNotificationClient*>(this);
+    return S_OK;
+  }
+  *object = nullptr;
+  return E_NOINTERFACE;
+}
+
 // ----------------------------------------------------------------------------
 //  _GetDefaultDevice
 // ----------------------------------------------------------------------------

modules/audio_device/win/audio_device_core_win.h

@@ -25,6 +25,8 @@
 #include <endpointvolume.h>
 #include <mediaobj.h>     // IMediaObject
 #include <mmdeviceapi.h>  // MMDevice
+#include <comdef.h>
+#include <objbase.h>
 
 #include "api/scoped_refptr.h"
 #include "rtc_base/synchronization/mutex.h"
@@ -51,6 +53,33 @@ class AudioDeviceWindowsCore : public AudioDeviceGeneric {
   AudioDeviceWindowsCore();
   ~AudioDeviceWindowsCore();
 
+  class DeviceStateListener : public IMMNotificationClient {
+   public:
+    virtual ~DeviceStateListener() = default;
+    HRESULT __stdcall OnDeviceStateChanged(LPCWSTR pwstrDeviceId,
+                                           DWORD dwNewState) override;
+    HRESULT __stdcall OnDeviceAdded(LPCWSTR pwstrDeviceId) override;
+
+    HRESULT __stdcall OnDeviceRemoved(LPCWSTR pwstrDeviceId) override;
+
+    HRESULT
+    __stdcall OnDefaultDeviceChanged(EDataFlow flow,
+                                     ERole role,
+                                     LPCWSTR pwstrDefaultDeviceId) override;
+
+    HRESULT __stdcall OnPropertyValueChanged(LPCWSTR pwstrDeviceId,
+                                             const PROPERTYKEY key) override;
+    // IUnknown (required by IMMNotificationClient).
+    ULONG __stdcall AddRef() override;
+    ULONG __stdcall Release() override;
+    HRESULT __stdcall QueryInterface(REFIID iid, void** object) override;
+    void SetAudioDeviceSink(AudioDeviceSink *sink);
+
+   private:
+    LONG ref_count_ = 1;
+    AudioDeviceSink *callback_ = nullptr;
+  };
+
   static bool CoreAudioIsSupported();
 
   // Retrieve the currently utilized audio layer
@@ -151,6 +180,8 @@ class AudioDeviceWindowsCore : public AudioDeviceGeneric {
 
   virtual int32_t EnableBuiltInAEC(bool enable);
 
+  virtual int32_t SetAudioDeviceSink(AudioDeviceSink* sink);
+
  public:
   virtual void AttachAudioBuffer(AudioDeviceBuffer* audioBuffer);
 
@@ -238,6 +269,7 @@ class AudioDeviceWindowsCore : public AudioDeviceGeneric {
   IAudioEndpointVolume* _ptrCaptureVolume;
   ISimpleAudioVolume* _ptrRenderSimpleVolume;
 
+  DeviceStateListener *_deviceStateListener = nullptr;
   // DirectX Media Object (DMO) for the built-in AEC.
   rtc::scoped_refptr<IMediaObject> _dmo;
   rtc::scoped_refptr<IMediaBuffer> _mediaBuffer;