Milestone 5: deliver embedded RDP sessions and lifecycle hardening

third_party/FreeRDP/server/Windows/wf_directsound.c

@@ -0,0 +1,219 @@
+#include "wf_directsound.h"
+#include "wf_interface.h"
+#include "wf_info.h"
+#include "wf_rdpsnd.h"
+
+#include <winpr/windows.h>
+
+#define INITGUID
+#include <initguid.h>
+#include <objbase.h>
+
+#define CINTERFACE 1
+#include <mmsystem.h>
+#include <dsound.h>
+
+#include <freerdp/log.h>
+#define TAG SERVER_TAG("windows")
+
+IDirectSoundCapture8* cap;
+IDirectSoundCaptureBuffer8* capBuf;
+DSCBUFFERDESC dscbd;
+DWORD lastPos;
+wfPeerContext* latestPeer;
+
+int wf_rdpsnd_set_latest_peer(wfPeerContext* peer)
+{
+	latestPeer = peer;
+	return 0;
+}
+
+int wf_directsound_activate(RdpsndServerContext* context)
+{
+	HRESULT hr;
+	wfInfo* wfi;
+	HANDLE hThread;
+
+	LPDIRECTSOUNDCAPTUREBUFFER pDSCB;
+
+	wfi = wf_info_get_instance();
+	if (!wfi)
+	{
+		WLog_ERR(TAG, "Failed to wfi instance");
+		return 1;
+	}
+	WLog_DBG(TAG, "RDPSND (direct sound) Activated");
+	hr = DirectSoundCaptureCreate8(nullptr, &cap, nullptr);
+
+	if (FAILED(hr))
+	{
+		WLog_ERR(TAG, "Failed to create sound capture device");
+		return 1;
+	}
+
+	WLog_INFO(TAG, "Created sound capture device");
+	dscbd.dwSize = sizeof(DSCBUFFERDESC);
+	dscbd.dwFlags = 0;
+	dscbd.dwBufferBytes = wfi->agreed_format->nAvgBytesPerSec;
+	dscbd.dwReserved = 0;
+	dscbd.lpwfxFormat = wfi->agreed_format;
+	dscbd.dwFXCount = 0;
+	dscbd.lpDSCFXDesc = nullptr;
+
+	hr = cap->lpVtbl->CreateCaptureBuffer(cap, &dscbd, &pDSCB, nullptr);
+
+	if (FAILED(hr))
+	{
+		WLog_ERR(TAG, "Failed to create capture buffer");
+	}
+
+	WLog_INFO(TAG, "Created capture buffer");
+	hr = pDSCB->lpVtbl->QueryInterface(pDSCB, &IID_IDirectSoundCaptureBuffer8, (LPVOID*)&capBuf);
+	if (FAILED(hr))
+	{
+		WLog_ERR(TAG, "Failed to QI capture buffer");
+	}
+	WLog_INFO(TAG, "Created IDirectSoundCaptureBuffer8");
+	pDSCB->lpVtbl->Release(pDSCB);
+	lastPos = 0;
+
+	if (!(hThread = CreateThread(nullptr, 0, wf_rdpsnd_directsound_thread, latestPeer, 0, nullptr)))
+	{
+		WLog_ERR(TAG, "Failed to create direct sound thread");
+		return 1;
+	}
+	(void)CloseHandle(hThread);
+
+	return 0;
+}
+
+static DWORD WINAPI wf_rdpsnd_directsound_thread(LPVOID lpParam)
+{
+	HRESULT hr;
+	DWORD beg = 0;
+	DWORD end = 0;
+	DWORD diff, rate;
+	wfPeerContext* context;
+	wfInfo* wfi;
+
+	VOID* pbCaptureData = nullptr;
+	DWORD dwCaptureLength = 0;
+	VOID* pbCaptureData2 = nullptr;
+	DWORD dwCaptureLength2 = 0;
+	VOID* pbPlayData = nullptr;
+	DWORD dwReadPos = 0;
+	LONG lLockSize = 0;
+
+	wfi = wf_info_get_instance();
+	if (!wfi)
+	{
+		WLog_ERR(TAG, "Failed get instance");
+		return 1;
+	}
+
+	context = (wfPeerContext*)lpParam;
+	rate = 1000 / 24;
+	WLog_INFO(TAG, "Trying to start capture");
+	hr = capBuf->lpVtbl->Start(capBuf, DSCBSTART_LOOPING);
+	if (FAILED(hr))
+	{
+		WLog_ERR(TAG, "Failed to start capture");
+	}
+	WLog_INFO(TAG, "Capture started");
+
+	while (1)
+	{
+
+		end = GetTickCount();
+		diff = end - beg;
+
+		if (diff < rate)
+		{
+			Sleep(rate - diff);
+		}
+
+		beg = GetTickCount();
+
+		if (wf_rdpsnd_lock() > 0)
+		{
+			// check for main exit condition
+			if (wfi->snd_stop == TRUE)
+			{
+				wf_rdpsnd_unlock();
+				break;
+			}
+
+			hr = capBuf->lpVtbl->GetCurrentPosition(capBuf, nullptr, &dwReadPos);
+			if (FAILED(hr))
+			{
+				WLog_ERR(TAG, "Failed to get read pos");
+				wf_rdpsnd_unlock();
+				break;
+			}
+
+			lLockSize = dwReadPos - lastPos; // dscbd.dwBufferBytes;
+			if (lLockSize < 0)
+				lLockSize += dscbd.dwBufferBytes;
+
+			// WLog_DBG(TAG, "Last, read, lock = [%"PRIu32", %"PRIu32", %"PRId32"]\n", lastPos,
+			// dwReadPos, lLockSize);
+
+			if (lLockSize == 0)
+			{
+				wf_rdpsnd_unlock();
+				continue;
+			}
+
+			hr = capBuf->lpVtbl->Lock(capBuf, lastPos, lLockSize, &pbCaptureData, &dwCaptureLength,
+			                          &pbCaptureData2, &dwCaptureLength2, 0L);
+			if (FAILED(hr))
+			{
+				WLog_ERR(TAG, "Failed to lock sound capture buffer");
+				wf_rdpsnd_unlock();
+				break;
+			}
+
+			// fwrite(pbCaptureData, 1, dwCaptureLength, pFile);
+			// fwrite(pbCaptureData2, 1, dwCaptureLength2, pFile);
+
+			// FIXME: frames = bytes/(bytespersample * channels)
+
+			context->rdpsnd->SendSamples(context->rdpsnd, pbCaptureData, dwCaptureLength / 4,
+			                             (UINT16)(beg & 0xffff));
+			context->rdpsnd->SendSamples(context->rdpsnd, pbCaptureData2, dwCaptureLength2 / 4,
+			                             (UINT16)(beg & 0xffff));
+
+			hr = capBuf->lpVtbl->Unlock(capBuf, pbCaptureData, dwCaptureLength, pbCaptureData2,
+			                            dwCaptureLength2);
+			if (FAILED(hr))
+			{
+				WLog_ERR(TAG, "Failed to unlock sound capture buffer");
+				wf_rdpsnd_unlock();
+				return 0;
+			}
+
+			// TODO keep track of location in buffer
+			lastPos += dwCaptureLength;
+			lastPos %= dscbd.dwBufferBytes;
+			lastPos += dwCaptureLength2;
+			lastPos %= dscbd.dwBufferBytes;
+
+			wf_rdpsnd_unlock();
+		}
+	}
+
+	WLog_INFO(TAG, "Trying to stop sound capture");
+	hr = capBuf->lpVtbl->Stop(capBuf);
+	if (FAILED(hr))
+	{
+		WLog_ERR(TAG, "Failed to stop capture");
+	}
+
+	WLog_INFO(TAG, "Capture stopped");
+	capBuf->lpVtbl->Release(capBuf);
+	cap->lpVtbl->Release(cap);
+
+	lastPos = 0;
+
+	return 0;
+}