Fix source and buffer out-of-memory conditions
[openal-soft.git] / Alc / winmm.c
blob7018f0986078ec00c911780ae8c75de4c105c74f
1 /**
2 * OpenAL cross platform audio library
3 * Copyright (C) 1999-2007 by authors.
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Library General Public
6 * License as published by the Free Software Foundation; either
7 * version 2 of the License, or (at your option) any later version.
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Library General Public License for more details.
14 * You should have received a copy of the GNU Library General Public
15 * License along with this library; if not, write to the
16 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
17 * Boston, MA 02111-1307, USA.
18 * Or go to http://www.gnu.org/copyleft/lgpl.html
21 #include <stdlib.h>
22 #include <stdio.h>
23 #include <memory.h>
25 #include <windows.h>
26 #include <mmsystem.h>
28 #include "alMain.h"
29 #include "AL/al.h"
30 #include "AL/alc.h"
33 typedef struct {
34 // MMSYSTEM Capture Device
35 ALboolean bWaveInShutdown;
36 HANDLE hWaveInHdrEvent;
37 HANDLE hWaveInThreadEvent;
38 HANDLE hWaveInThread;
39 DWORD ulWaveInThreadID;
40 ALint lWaveInBuffersCommitted;
41 HWAVEIN hWaveInHandle;
42 WAVEHDR WaveInBuffer[4];
43 ALCchar *pCapturedSampleData;
44 ALuint ulCapturedDataSize;
45 ALuint ulReadCapturedDataPos;
46 ALuint ulWriteCapturedDataPos;
47 } WinMMData;
50 static ALCchar *CaptureDeviceList[16];
53 WaveInProc
55 Posts a message to 'CaptureThreadProc' everytime a WaveIn Buffer is completed and
56 returns to the application (with more data)
58 static void CALLBACK WaveInProc(HWAVEIN hDevice,UINT uMsg,DWORD_PTR dwInstance,DWORD_PTR dwParam1,DWORD_PTR dwParam2)
60 ALCdevice *pDevice = (ALCdevice *)dwInstance;
61 WinMMData *pData = pDevice->ExtraData;
63 (void)hDevice;
64 (void)dwParam2;
66 if ((uMsg==WIM_DATA) && (pDevice))
68 // Decrement number of buffers in use
69 pData->lWaveInBuffersCommitted--;
71 if (pData->bWaveInShutdown == AL_FALSE)
73 // Notify Wave Processor Thread that a Wave Header has returned
74 PostThreadMessage(pData->ulWaveInThreadID,uMsg,0,dwParam1);
76 else
78 if (pData->lWaveInBuffersCommitted == 0)
80 // Signal Wave Buffers Returned event
81 if (pData->hWaveInHdrEvent)
82 SetEvent(pData->hWaveInHdrEvent);
84 // Post 'Quit' Message to WaveIn Processor Thread
85 PostThreadMessage(pData->ulWaveInThreadID,WM_QUIT,0,0);
92 CaptureThreadProc
94 Used by "MMSYSTEM" Device. Called when a WaveIn buffer had been filled with new
95 audio data.
97 DWORD WINAPI CaptureThreadProc(LPVOID lpParameter)
99 ALCdevice *pDevice = (ALCdevice*)lpParameter;
100 WinMMData *pData = pDevice->ExtraData;
101 ALuint ulOffset, ulMaxSize, ulSection;
102 LPWAVEHDR pWaveHdr;
103 MSG msg;
105 while (GetMessage(&msg, NULL, 0, 0))
107 if ((msg.message==WIM_DATA)&&(!pData->bWaveInShutdown))
109 SuspendContext(NULL);
111 pWaveHdr = ((LPWAVEHDR)msg.lParam);
113 // Calculate offset in local buffer to write data to
114 ulOffset = pData->ulWriteCapturedDataPos % pData->ulCapturedDataSize;
116 if ((ulOffset + pWaveHdr->dwBytesRecorded) > pData->ulCapturedDataSize)
118 ulSection = pData->ulCapturedDataSize - ulOffset;
119 memcpy(pData->pCapturedSampleData + ulOffset, pWaveHdr->lpData, ulSection);
120 memcpy(pData->pCapturedSampleData, pWaveHdr->lpData + ulSection, pWaveHdr->dwBytesRecorded - ulSection);
122 else
124 memcpy(pData->pCapturedSampleData + ulOffset, pWaveHdr->lpData, pWaveHdr->dwBytesRecorded);
127 pData->ulWriteCapturedDataPos += pWaveHdr->dwBytesRecorded;
129 if (pData->ulWriteCapturedDataPos > (pData->ulReadCapturedDataPos + pData->ulCapturedDataSize))
131 // Application has not read enough audio data from the capture buffer so data has been
132 // overwritten. Reset ReadPosition.
133 pData->ulReadCapturedDataPos = pData->ulWriteCapturedDataPos - pData->ulCapturedDataSize;
136 // To prevent an over-flow prevent the offset values from getting too large
137 ulMaxSize = pData->ulCapturedDataSize << 4;
138 if ((pData->ulReadCapturedDataPos > ulMaxSize) && (pData->ulWriteCapturedDataPos > ulMaxSize))
140 pData->ulReadCapturedDataPos -= ulMaxSize;
141 pData->ulWriteCapturedDataPos -= ulMaxSize;
144 // Send buffer back to capture more data
145 waveInAddBuffer(pData->hWaveInHandle,pWaveHdr,sizeof(WAVEHDR));
146 pData->lWaveInBuffersCommitted++;
148 ProcessContext(NULL);
152 // Signal Wave Thread completed event
153 if (pData->hWaveInThreadEvent)
154 SetEvent(pData->hWaveInThreadEvent);
156 ExitThread(0);
158 return 0;
162 static ALCboolean WinMMOpenPlayback(ALCdevice *device, const ALCchar *deviceName)
164 (void)device;
165 (void)deviceName;
166 return ALC_FALSE;
169 static void WinMMClosePlayback(ALCdevice *device)
171 (void)device;
175 static ALCboolean WinMMOpenCapture(ALCdevice *pDevice, const ALCchar *deviceName, ALCuint frequency, ALCenum format, ALCsizei SampleSize)
177 WAVEFORMATEX wfexCaptureFormat;
178 WinMMData *pData = NULL;
179 ALint lDeviceID = 0;
180 ALint lBufferSize;
181 ALint i;
183 (void)format;
185 // Find the Device ID matching the deviceName if valid
186 if (deviceName)
188 for(i = 0;CaptureDeviceList[i];i++)
190 if (!strcmp(deviceName, CaptureDeviceList[i]))
192 lDeviceID = i;
193 break;
196 if(!CaptureDeviceList[i])
197 return ALC_FALSE;
199 pDevice->szDeviceName = CaptureDeviceList[lDeviceID];
201 pData = calloc(1, sizeof(*pData));
202 if(!pData)
204 SetALCError(ALC_OUT_OF_MEMORY);
205 return ALC_FALSE;
208 memset(&wfexCaptureFormat, 0, sizeof(WAVEFORMATEX));
209 wfexCaptureFormat.wFormatTag = WAVE_FORMAT_PCM;
210 wfexCaptureFormat.nChannels = pDevice->Channels;
211 wfexCaptureFormat.wBitsPerSample = pDevice->FrameSize / pDevice->Channels * 8;
212 wfexCaptureFormat.nBlockAlign = pDevice->FrameSize;
213 wfexCaptureFormat.nSamplesPerSec = frequency;
214 wfexCaptureFormat.nAvgBytesPerSec = wfexCaptureFormat.nSamplesPerSec *
215 pDevice->FrameSize;
216 wfexCaptureFormat.cbSize = 0;
218 if (waveInOpen(&pData->hWaveInHandle, lDeviceID, &wfexCaptureFormat, (DWORD_PTR)&WaveInProc, (DWORD_PTR)pDevice, CALLBACK_FUNCTION) != MMSYSERR_NOERROR)
219 goto failure;
221 pData->hWaveInHdrEvent = CreateEvent(NULL, AL_TRUE, AL_FALSE, "WaveInAllHeadersReturned");
222 if (pData->hWaveInHdrEvent == NULL)
223 goto failure;
225 pData->hWaveInThreadEvent = CreateEvent(NULL, AL_TRUE, AL_FALSE, "WaveInThreadDestroyed");
226 if (pData->hWaveInThreadEvent == NULL)
227 goto failure;
229 pData->hWaveInThread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)CaptureThreadProc, (LPVOID)pDevice, 0, &pData->ulWaveInThreadID);
230 if (pData->hWaveInThread == NULL)
231 goto failure;
233 // Allocate circular memory buffer for the captured audio
234 pData->ulCapturedDataSize = SampleSize * wfexCaptureFormat.nBlockAlign;
236 // Make sure circular buffer is at least 100ms in size (and an exact multiple of
237 // the block alignment
238 if (pData->ulCapturedDataSize < (wfexCaptureFormat.nAvgBytesPerSec / 10))
240 pData->ulCapturedDataSize = wfexCaptureFormat.nAvgBytesPerSec / 10;
241 pData->ulCapturedDataSize -= (pData->ulCapturedDataSize % wfexCaptureFormat.nBlockAlign);
244 pData->pCapturedSampleData = (ALCchar*)malloc(pData->ulCapturedDataSize);
245 pData->lWaveInBuffersCommitted=0;
247 // Create 4 Buffers of 50ms each
248 lBufferSize = wfexCaptureFormat.nAvgBytesPerSec / 20;
249 lBufferSize -= (lBufferSize % wfexCaptureFormat.nBlockAlign);
251 for (i=0;i<4;i++)
253 memset(&pData->WaveInBuffer[i], 0, sizeof(WAVEHDR));
254 pData->WaveInBuffer[i].dwBufferLength = lBufferSize;
255 pData->WaveInBuffer[i].lpData = calloc(1,pData->WaveInBuffer[i].dwBufferLength);
256 pData->WaveInBuffer[i].dwFlags = 0;
257 pData->WaveInBuffer[i].dwLoops = 0;
258 waveInPrepareHeader(pData->hWaveInHandle, &pData->WaveInBuffer[i], sizeof(WAVEHDR));
259 waveInAddBuffer(pData->hWaveInHandle, &pData->WaveInBuffer[i], sizeof(WAVEHDR));
260 pData->lWaveInBuffersCommitted++;
263 pData->ulReadCapturedDataPos = 0;
264 pData->ulWriteCapturedDataPos = 0;
266 pDevice->ExtraData = pData;
267 return ALC_TRUE;
269 failure:
270 if (pData->hWaveInThreadEvent)
271 CloseHandle(pData->hWaveInThreadEvent);
272 if (pData->hWaveInHdrEvent)
273 CloseHandle(pData->hWaveInHdrEvent);
274 if (pData->hWaveInHandle)
275 waveInClose(pData->hWaveInHandle);
277 free(pData);
278 return ALC_FALSE;
281 static void WinMMCloseCapture(ALCdevice *pDevice)
283 WinMMData *pData = (WinMMData*)pDevice->ExtraData;
284 int i;
286 // Call waveOutReset to shutdown wave device
287 pData->bWaveInShutdown = AL_TRUE;
288 waveInReset(pData->hWaveInHandle);
290 // Wait for signal that all Wave Buffers have returned
291 WaitForSingleObjectEx(pData->hWaveInHdrEvent, 5000, FALSE);
293 // Wait for signal that Wave Thread has been destroyed
294 WaitForSingleObjectEx(pData->hWaveInThreadEvent, 5000, FALSE);
296 // Release the wave buffers
297 for (i=0;i<4;i++)
299 waveInUnprepareHeader(pData->hWaveInHandle, &pData->WaveInBuffer[i], sizeof(WAVEHDR));
300 free(pData->WaveInBuffer[i].lpData);
303 // Free Audio Buffer data
304 free(pData->pCapturedSampleData);
305 pData->pCapturedSampleData = NULL;
307 // Close the Wave device
308 waveInClose(pData->hWaveInHandle);
309 pData->hWaveInHandle = 0;
311 CloseHandle(pData->hWaveInThread);
312 pData->hWaveInThread = 0;
314 if (pData->hWaveInHdrEvent)
316 CloseHandle(pData->hWaveInHdrEvent);
317 pData->hWaveInHdrEvent = 0;
320 if (pData->hWaveInThreadEvent)
322 CloseHandle(pData->hWaveInThreadEvent);
323 pData->hWaveInThreadEvent = 0;
326 free(pData);
327 pDevice->ExtraData = NULL;
330 static void WinMMStartCapture(ALCdevice *pDevice)
332 WinMMData *pData = (WinMMData*)pDevice->ExtraData;
333 waveInStart(pData->hWaveInHandle);
336 static void WinMMStopCapture(ALCdevice *pDevice)
338 WinMMData *pData = (WinMMData*)pDevice->ExtraData;
339 waveInStop(pData->hWaveInHandle);
342 static void WinMMCaptureSamples(ALCdevice *pDevice, ALCvoid *pBuffer, ALCuint lSamples)
344 WinMMData *pData = (WinMMData*)pDevice->ExtraData;
345 ALuint ulSamples = (unsigned long)lSamples;
346 ALuint ulBytes, ulBytesToCopy;
347 ALuint ulCapturedSamples;
348 ALuint ulReadOffset;
350 // Check that we have the requested numbers of Samples
351 ulCapturedSamples = (pData->ulWriteCapturedDataPos - pData->ulReadCapturedDataPos) / pDevice->FrameSize;
352 if(ulSamples > ulCapturedSamples)
354 SetALCError(ALC_INVALID_VALUE);
355 return;
358 ulBytes = ulSamples * pDevice->FrameSize;
360 // Get Read Offset
361 ulReadOffset = (pData->ulReadCapturedDataPos % pData->ulCapturedDataSize);
363 // Check for wrap-around condition
364 if ((ulReadOffset + ulBytes) > pData->ulCapturedDataSize)
366 // Copy data from last Read position to end of data
367 ulBytesToCopy = pData->ulCapturedDataSize - ulReadOffset;
368 memcpy(pBuffer, pData->pCapturedSampleData + ulReadOffset, ulBytesToCopy);
370 // Copy rest of the data from the start of the captured data
371 memcpy(((char *)pBuffer) + ulBytesToCopy, pData->pCapturedSampleData, ulBytes - ulBytesToCopy);
373 else
375 // Copy data from the read position in the captured data
376 memcpy(pBuffer, pData->pCapturedSampleData + ulReadOffset, ulBytes);
379 // Update Read Position
380 pData->ulReadCapturedDataPos += ulBytes;
383 static ALCuint WinMMAvailableSamples(ALCdevice *pDevice)
385 WinMMData *pData = (WinMMData*)pDevice->ExtraData;
386 ALCuint lCapturedBytes = (pData->ulWriteCapturedDataPos - pData->ulReadCapturedDataPos);
387 return lCapturedBytes / pDevice->FrameSize;
391 BackendFuncs WinMMFuncs = {
392 WinMMOpenPlayback,
393 WinMMClosePlayback,
394 WinMMOpenCapture,
395 WinMMCloseCapture,
396 WinMMStartCapture,
397 WinMMStopCapture,
398 WinMMCaptureSamples,
399 WinMMAvailableSamples
402 void alcWinMMInit(BackendFuncs *FuncList)
404 ALint lNumDevs;
405 ALint lLoop;
407 *FuncList = WinMMFuncs;
409 lNumDevs = waveInGetNumDevs();
410 for (lLoop = 0; lLoop < lNumDevs; lLoop++)
412 WAVEINCAPS WaveInCaps;
413 if(waveInGetDevCaps(lLoop, &WaveInCaps, sizeof(WAVEINCAPS)) == MMSYSERR_NOERROR)
415 char name[128];
416 snprintf(name, sizeof(name), "WaveIn on %s", WaveInCaps.szPname);
417 CaptureDeviceList[lLoop] = AppendCaptureDeviceList(name);