1
mirror of https://github.com/rapid7/metasploit-payloads synced 2024-11-26 17:41:08 +01:00

Fix error with recording on certain Windows versions.

git-svn-id: file:///home/svn/framework3/trunk@11311 4d416f70-5f16-0410-b530-b9f4589650da
This commit is contained in:
Matt Weeks 2010-12-12 20:12:03 +00:00
parent 19fddbee1d
commit 8918ab0469

View File

@ -3,6 +3,9 @@
#include <windows.h>
#include "audio.h"
//Handle used for synchronization. Main thread waits for event to be signalled to clean up
HANDLE recordMicEvent;
//All these default values should be overwritten
UINT buffersize = 0;
UINT riffsize = 0;
@ -15,6 +18,7 @@ void CALLBACK waveInProc(HWAVEIN hwi, UINT uMsg, DWORD_PTR dwInstance, DWORD_PTR
if(uMsg != WIM_DATA)
return;
memcpy(dataBuffer, recordBuffer, buffersize);
SetEvent(recordMicEvent);
}
/*
@ -91,9 +95,15 @@ DWORD request_ui_record_mic( Remote * remote, Packet * request ){
wh.dwFlags = 0;
waveInPrepareHeader(hWavIn, &wh, sizeof(wh));
waveInAddBuffer(hWavIn, &wh, sizeof(wh));
recordMicEvent = CreateEvent(
NULL, // default security attributes
FALSE, // auto-reset event
FALSE, // initial state is nonsignaled
NULL); // no object name
dwResult = waveInStart(hWavIn);
if(dwResult != MMSYSERR_NOERROR)
BREAK_WITH_ERROR("request_ui_record_mic: WaveInStart failed", dwResult)
WaitForSingleObject(recordMicEvent, seconds * 1000 + 1000);
dwResult = waveInStop(hWavIn);//seems to wait for buffer to complete
if(dwResult != MMSYSERR_NOERROR)
BREAK_WITH_ERROR("request_ui_record_mic: WaveInStop failed", dwResult)