quantum/winscap
1
0
Fork 0
mirror of https://github.com/quantum5/winscap.git synced 2025-04-24 05:31:58 -04:00
Code Issues Packages Projects Releases Wiki Activity

Wait for a suitable device to appear

Browse source
This commit is contained in:
Quantum 2020-03-07 15:05:40 -08:00
parent 837d713edf
commit 6de12d3677
1 changed files with 36 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
winscap.cpp
View file

@ -5,6 +5,7 @@
#include <comdef.h> #include <comdef.h>
#include <comip.h> #include <comip.h>
#include <fcntl.h> #include <fcntl.h>
#include <functiondiscoverykeys_devpkey.h>
#include <io.h> #include <io.h>
#include <mmdeviceapi.h> #include <mmdeviceapi.h>
#include <stdio.h> #include <stdio.h>
@ -31,13 +32,16 @@ _COM_SMARTPTR_TYPEDEF(IMMDeviceEnumerator, __uuidof(IMMDeviceEnumerator));
_COM_SMARTPTR_TYPEDEF(IMMDevice, __uuidof(IMMDevice)); _COM_SMARTPTR_TYPEDEF(IMMDevice, __uuidof(IMMDevice));
_COM_SMARTPTR_TYPEDEF(IAudioClient, __uuidof(IAudioClient)); _COM_SMARTPTR_TYPEDEF(IAudioClient, __uuidof(IAudioClient));
_COM_SMARTPTR_TYPEDEF(IAudioCaptureClient, __uuidof(IAudioCaptureClient)); _COM_SMARTPTR_TYPEDEF(IAudioCaptureClient, __uuidof(IAudioCaptureClient));
_COM_SMARTPTR_TYPEDEF(IPropertyStore, __uuidof(IPropertyStore));
class DeviceChangeNotification : public IMMNotificationClient { class DeviceChangeNotification : public IMMNotificationClient {
volatile ULONG ref; volatile ULONG ref;
volatile bool &changed; volatile bool &changed;
HANDLE hEvent;
public: public:
DeviceChangeNotification(volatile bool &changed) : changed(changed) {} DeviceChangeNotification(volatile bool &changed, HANDLE hEvent)
: changed(changed), hEvent(hEvent) {}
// This is meant to be allocated on stack, so we don't actually free. // This is meant to be allocated on stack, so we don't actually free.
STDMETHODIMP_(ULONG) AddRef() { return InterlockedIncrement(&ref); } STDMETHODIMP_(ULONG) AddRef() { return InterlockedIncrement(&ref); }
@ -60,6 +64,7 @@ class DeviceChangeNotification : public IMMNotificationClient {
STDMETHODIMP OnDefaultDeviceChanged(EDataFlow flow, ERole role, LPCWSTR) { STDMETHODIMP OnDefaultDeviceChanged(EDataFlow flow, ERole role, LPCWSTR) {
if (flow == eRender && role == eConsole) { if (flow == eRender && role == eConsole) {
changed = true; changed = true;
SetEvent(hEvent);
} }
return S_OK; return S_OK;
} }
@ -93,19 +98,20 @@ struct {
{AUDCLNT_E_UNSUPPORTED_FORMAT, L"Requested sound format unsupported"}, {AUDCLNT_E_UNSUPPORTED_FORMAT, L"Requested sound format unsupported"},
}; };
LPCWSTR error_message(const _com_error &err) {
for (int i = 0; i < sizeof error_table / sizeof error_table[0]; ++i) {
if (error_table[i].hr == err.Error()) {
return error_table[i].error;
}
}
return err.ErrorMessage();
}
#define ensure(hr) ensure_(__FILE__, __LINE__, hr) #define ensure(hr) ensure_(__FILE__, __LINE__, hr)
void ensure_(const char *file, int line, HRESULT hr) { void ensure_(const char *file, int line, HRESULT hr) {
if (FAILED(hr)) { if (FAILED(hr)) {
_com_error err(hr); _com_error err(hr);
LPCWSTR msg = err.ErrorMessage(); fwprintf(stderr, L"Error at %S:%d (0x%08x): %s\n", file, line, hr, error_message(err));
for (int i = 0; i < sizeof error_table / sizeof error_table[0]; ++i) {
if (error_table[i].hr == hr) {
msg = error_table[i].error;
}
}
fwprintf(stderr, L"Error at %S:%d (0x%08x): %s\n", file, line, hr, msg);
exit(1); exit(1);
} }
} }
@ -149,18 +155,35 @@ int main(int argc, char *argv[]) {
ensure(pEnumerator.CreateInstance(__uuidof(MMDeviceEnumerator))); ensure(pEnumerator.CreateInstance(__uuidof(MMDeviceEnumerator)));
volatile bool deviceChanged = false; volatile bool deviceChanged = false;
DeviceChangeNotification deviceChangeNotification(deviceChanged); HANDLE hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
DeviceChangeNotification deviceChangeNotification(deviceChanged, hEvent);
pEnumerator->RegisterEndpointNotificationCallback(&deviceChangeNotification); pEnumerator->RegisterEndpointNotificationCallback(&deviceChangeNotification);
for (;;) { for (;;) {
ResetEvent(hEvent);
IMMDevicePtr pDevice; IMMDevicePtr pDevice;
ensure(pEnumerator->GetDefaultAudioEndpoint(eRender, eConsole, &pDevice)); ensure(pEnumerator->GetDefaultAudioEndpoint(eRender, eConsole, &pDevice));
IAudioClientPtr pClient; IAudioClientPtr pClient;
ensure(pDevice->Activate(__uuidof(IAudioClient), CLSCTX_ALL, nullptr, (void **)&pClient)); ensure(pDevice->Activate(__uuidof(IAudioClient), CLSCTX_ALL, nullptr, (void **)&pClient));
ensure(pClient->Initialize(AUDCLNT_SHAREMODE_SHARED, AUDCLNT_STREAMFLAGS_LOOPBACK, HRESULT hrInit = pClient->Initialize(AUDCLNT_SHAREMODE_SHARED, AUDCLNT_STREAMFLAGS_LOOPBACK,
16 * REFTIMES_PER_MILLISEC, 0, &wfx, nullptr)); 16 * REFTIMES_PER_MILLISEC, 0, &wfx, nullptr);
if (FAILED(hrInit)) {
_com_error err(hrInit);
IPropertyStorePtr pProps;
ensure(pDevice->OpenPropertyStore(STGM_READ, &pProps));
PROPVARIANT varName;
PropVariantInit(&varName);
ensure(pProps->GetValue(PKEY_Device_FriendlyName, &varName));
fwprintf(stderr, L"Failed to open: %s: %s\n", varName.pwszVal, error_message(err));
PropVariantClear(&varName);
WaitForSingleObject(hEvent, INFINITE);
continue;
}
UINT32 bufferFrameCount; UINT32 bufferFrameCount;
ensure(pClient->GetBufferSize(&bufferFrameCount)); ensure(pClient->GetBufferSize(&bufferFrameCount));