Sweden-Number/dlls/mfreadwrite/main.c

1697 lines
48 KiB
C
Raw Normal View History

2014-12-11 00:51:07 +01:00
/*
*
* Copyright 2014 Austin English
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
#include <stdarg.h>
#define COBJMACROS
#define NONAMELESSUNION
2014-12-11 00:51:07 +01:00
#include "windef.h"
#include "winbase.h"
#include "ole2.h"
#include "rpcproxy.h"
#undef INITGUID
#include <guiddef.h>
#include "mfapi.h"
#include "mferror.h"
#include "mfidl.h"
#include "mfreadwrite.h"
#include "wine/debug.h"
#include "wine/heap.h"
#include "wine/list.h"
WINE_DEFAULT_DEBUG_CHANNEL(mfplat);
2014-12-11 00:51:07 +01:00
static HINSTANCE mfinstance;
2014-12-11 00:51:07 +01:00
BOOL WINAPI DllMain(HINSTANCE instance, DWORD reason, LPVOID reserved)
{
switch (reason)
{
case DLL_WINE_PREATTACH:
return FALSE; /* prefer native version */
case DLL_PROCESS_ATTACH:
mfinstance = instance;
2014-12-11 00:51:07 +01:00
DisableThreadLibraryCalls(instance);
break;
}
return TRUE;
}
HRESULT WINAPI DllCanUnloadNow(void)
{
return S_FALSE;
}
HRESULT WINAPI DllRegisterServer(void)
{
return __wine_register_resources( mfinstance );
}
HRESULT WINAPI DllUnregisterServer(void)
{
return __wine_unregister_resources( mfinstance );
}
struct sample
{
struct list entry;
IMFSample *sample;
};
enum media_stream_state
{
STREAM_STATE_READY = 0,
STREAM_STATE_EOS,
};
struct media_stream
{
IMFMediaStream *stream;
IMFMediaType *current;
DWORD id;
CRITICAL_SECTION cs;
CONDITION_VARIABLE sample_event;
struct list samples;
enum media_stream_state state;
};
struct source_reader
{
IMFSourceReader IMFSourceReader_iface;
IMFAsyncCallback source_events_callback;
IMFAsyncCallback stream_events_callback;
LONG refcount;
IMFMediaSource *source;
IMFPresentationDescriptor *descriptor;
DWORD first_audio_stream_index;
DWORD first_video_stream_index;
IMFSourceReaderCallback *async_callback;
BOOL shutdown_on_release;
struct media_stream *streams;
DWORD stream_count;
CRITICAL_SECTION cs;
};
struct sink_writer
{
IMFSinkWriter IMFSinkWriter_iface;
LONG refcount;
};
static inline struct source_reader *impl_from_IMFSourceReader(IMFSourceReader *iface)
{
return CONTAINING_RECORD(iface, struct source_reader, IMFSourceReader_iface);
}
static struct source_reader *impl_from_source_callback_IMFAsyncCallback(IMFAsyncCallback *iface)
{
return CONTAINING_RECORD(iface, struct source_reader, source_events_callback);
}
static struct source_reader *impl_from_stream_callback_IMFAsyncCallback(IMFAsyncCallback *iface)
{
return CONTAINING_RECORD(iface, struct source_reader, stream_events_callback);
}
static inline struct sink_writer *impl_from_IMFSinkWriter(IMFSinkWriter *iface)
{
return CONTAINING_RECORD(iface, struct sink_writer, IMFSinkWriter_iface);
}
static HRESULT media_event_get_object(IMFMediaEvent *event, REFIID riid, void **obj)
{
PROPVARIANT value;
HRESULT hr;
PropVariantInit(&value);
if (FAILED(hr = IMFMediaEvent_GetValue(event, &value)))
{
WARN("Failed to get event value, hr %#x.\n", hr);
return hr;
}
if (value.vt != VT_UNKNOWN || !value.u.punkVal)
{
WARN("Unexpected value type %d.\n", value.vt);
PropVariantClear(&value);
return E_UNEXPECTED;
}
hr = IUnknown_QueryInterface(value.u.punkVal, riid, obj);
PropVariantClear(&value);
if (FAILED(hr))
{
WARN("Unexpected object type.\n");
return hr;
}
return hr;
}
static HRESULT media_stream_get_id(IMFMediaStream *stream, DWORD *id)
{
IMFStreamDescriptor *sd;
HRESULT hr;
if (SUCCEEDED(hr = IMFMediaStream_GetStreamDescriptor(stream, &sd)))
{
hr = IMFStreamDescriptor_GetStreamIdentifier(sd, id);
IMFStreamDescriptor_Release(sd);
}
return hr;
}
static HRESULT WINAPI source_reader_source_events_callback_QueryInterface(IMFAsyncCallback *iface,
REFIID riid, void **obj)
{
TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), obj);
if (IsEqualIID(riid, &IID_IMFAsyncCallback) ||
IsEqualIID(riid, &IID_IUnknown))
{
*obj = iface;
IMFAsyncCallback_AddRef(iface);
return S_OK;
}
WARN("Unsupported %s.\n", debugstr_guid(riid));
*obj = NULL;
return E_NOINTERFACE;
}
static ULONG WINAPI source_reader_source_events_callback_AddRef(IMFAsyncCallback *iface)
{
struct source_reader *reader = impl_from_source_callback_IMFAsyncCallback(iface);
return IMFSourceReader_AddRef(&reader->IMFSourceReader_iface);
}
static ULONG WINAPI source_reader_source_events_callback_Release(IMFAsyncCallback *iface)
{
struct source_reader *reader = impl_from_source_callback_IMFAsyncCallback(iface);
return IMFSourceReader_Release(&reader->IMFSourceReader_iface);
}
static HRESULT WINAPI source_reader_source_events_callback_GetParameters(IMFAsyncCallback *iface,
DWORD *flags, DWORD *queue)
{
return E_NOTIMPL;
}
static HRESULT source_reader_new_stream_handler(struct source_reader *reader, IMFMediaEvent *event)
{
IMFMediaStream *stream;
unsigned int i;
DWORD id = 0;
HRESULT hr;
if (FAILED(hr = media_event_get_object(event, &IID_IMFMediaStream, (void **)&stream)))
{
WARN("Failed to get stream object, hr %#x.\n", hr);
return hr;
}
TRACE("Got new stream %p.\n", stream);
if (FAILED(hr = media_stream_get_id(stream, &id)))
{
WARN("Unidentified stream %p, hr %#x.\n", stream, hr);
IMFMediaStream_Release(stream);
return hr;
}
for (i = 0; i < reader->stream_count; ++i)
{
if (id == reader->streams[i].id)
{
if (!InterlockedCompareExchangePointer((void **)&reader->streams[i].stream, stream, NULL))
{
IMFMediaStream_AddRef(reader->streams[i].stream);
if (FAILED(hr = IMFMediaStream_BeginGetEvent(stream, &reader->stream_events_callback,
(IUnknown *)stream)))
{
WARN("Failed to subscribe to stream events, hr %#x.\n", hr);
}
/* Wake so any waiting ReadSample() calls have a chance to make requests. */
WakeAllConditionVariable(&reader->streams[i].sample_event);
}
break;
}
}
if (i == reader->stream_count)
WARN("Stream with id %#x was not present in presentation descriptor.\n", id);
IMFMediaStream_Release(stream);
return hr;
}
static HRESULT WINAPI source_reader_source_events_callback_Invoke(IMFAsyncCallback *iface, IMFAsyncResult *result)
{
struct source_reader *reader = impl_from_source_callback_IMFAsyncCallback(iface);
MediaEventType event_type;
IMFMediaSource *source;
IMFMediaEvent *event;
HRESULT hr;
TRACE("%p, %p.\n", iface, result);
source = (IMFMediaSource *)IMFAsyncResult_GetStateNoAddRef(result);
if (FAILED(hr = IMFMediaSource_EndGetEvent(source, result, &event)))
return hr;
IMFMediaEvent_GetType(event, &event_type);
TRACE("Got event %u.\n", event_type);
switch (event_type)
{
case MENewStream:
hr = source_reader_new_stream_handler(reader, event);
break;
default:
;
}
if (FAILED(hr))
WARN("Failed while handling %d event, hr %#x.\n", event_type, hr);
IMFMediaEvent_Release(event);
IMFMediaSource_BeginGetEvent(source, iface, (IUnknown *)source);
return S_OK;
}
static const IMFAsyncCallbackVtbl source_events_callback_vtbl =
{
source_reader_source_events_callback_QueryInterface,
source_reader_source_events_callback_AddRef,
source_reader_source_events_callback_Release,
source_reader_source_events_callback_GetParameters,
source_reader_source_events_callback_Invoke,
};
static ULONG WINAPI source_reader_stream_events_callback_AddRef(IMFAsyncCallback *iface)
{
struct source_reader *reader = impl_from_stream_callback_IMFAsyncCallback(iface);
return IMFSourceReader_AddRef(&reader->IMFSourceReader_iface);
}
static ULONG WINAPI source_reader_stream_events_callback_Release(IMFAsyncCallback *iface)
{
struct source_reader *reader = impl_from_stream_callback_IMFAsyncCallback(iface);
return IMFSourceReader_Release(&reader->IMFSourceReader_iface);
}
static HRESULT source_reader_media_sample_handler(struct source_reader *reader, IMFMediaStream *stream,
IMFMediaEvent *event)
{
IMFSample *sample;
unsigned int i;
DWORD id = 0;
HRESULT hr;
TRACE("Got new sample for stream %p.\n", stream);
if (FAILED(hr = media_event_get_object(event, &IID_IMFSample, (void **)&sample)))
{
WARN("Failed to get sample object, hr %#x.\n", hr);
return hr;
}
if (FAILED(hr = media_stream_get_id(stream, &id)))
{
WARN("Unidentified stream %p, hr %#x.\n", stream, hr);
IMFSample_Release(sample);
return hr;
}
for (i = 0; i < reader->stream_count; ++i)
{
if (id == reader->streams[i].id)
{
struct sample *pending_sample;
if (!(pending_sample = heap_alloc(sizeof(*pending_sample))))
{
hr = E_OUTOFMEMORY;
goto failed;
}
pending_sample->sample = sample;
IMFSample_AddRef(pending_sample->sample);
EnterCriticalSection(&reader->streams[i].cs);
list_add_tail(&reader->streams[i].samples, &pending_sample->entry);
LeaveCriticalSection(&reader->streams[i].cs);
WakeAllConditionVariable(&reader->streams[i].sample_event);
break;
}
}
if (i == reader->stream_count)
WARN("Stream with id %#x was not present in presentation descriptor.\n", id);
failed:
IMFSample_Release(sample);
return hr;
}
static HRESULT source_reader_media_stream_state_handler(struct source_reader *reader, IMFMediaStream *stream,
MediaEventType event)
{
unsigned int i;
HRESULT hr;
DWORD id;
if (FAILED(hr = media_stream_get_id(stream, &id)))
{
WARN("Unidentified stream %p, hr %#x.\n", stream, hr);
return hr;
}
for (i = 0; i < reader->stream_count; ++i)
{
if (id == reader->streams[i].id)
{
EnterCriticalSection(&reader->streams[i].cs);
switch (event)
{
case MEEndOfStream:
reader->streams[i].state = STREAM_STATE_EOS;
break;
case MEStreamSeeked:
case MEStreamStarted:
reader->streams[i].state = STREAM_STATE_READY;
break;
default:
;
}
LeaveCriticalSection(&reader->streams[i].cs);
WakeAllConditionVariable(&reader->streams[i].sample_event);
break;
}
}
return S_OK;
}
static HRESULT WINAPI source_reader_stream_events_callback_Invoke(IMFAsyncCallback *iface, IMFAsyncResult *result)
{
struct source_reader *reader = impl_from_stream_callback_IMFAsyncCallback(iface);
MediaEventType event_type;
IMFMediaStream *stream;
IMFMediaEvent *event;
HRESULT hr;
TRACE("%p, %p.\n", iface, result);
stream = (IMFMediaStream *)IMFAsyncResult_GetStateNoAddRef(result);
if (FAILED(hr = IMFMediaStream_EndGetEvent(stream, result, &event)))
return hr;
IMFMediaEvent_GetType(event, &event_type);
TRACE("Got event %u.\n", event_type);
switch (event_type)
{
case MEMediaSample:
hr = source_reader_media_sample_handler(reader, stream, event);
break;
case MEStreamSeeked:
case MEStreamStarted:
case MEEndOfStream:
hr = source_reader_media_stream_state_handler(reader, stream, event_type);
break;
default:
;
}
if (FAILED(hr))
WARN("Failed while handling %d event, hr %#x.\n", event_type, hr);
IMFMediaEvent_Release(event);
IMFMediaStream_BeginGetEvent(stream, iface, (IUnknown *)stream);
return S_OK;
}
static const IMFAsyncCallbackVtbl stream_events_callback_vtbl =
{
source_reader_source_events_callback_QueryInterface,
source_reader_stream_events_callback_AddRef,
source_reader_stream_events_callback_Release,
source_reader_source_events_callback_GetParameters,
source_reader_stream_events_callback_Invoke,
};
static HRESULT WINAPI src_reader_QueryInterface(IMFSourceReader *iface, REFIID riid, void **out)
{
struct source_reader *reader = impl_from_IMFSourceReader(iface);
TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), out);
if(IsEqualGUID(riid, &IID_IUnknown) ||
IsEqualGUID(riid, &IID_IMFSourceReader))
{
*out = &reader->IMFSourceReader_iface;
}
else
{
FIXME("(%s, %p)\n", debugstr_guid(riid), out);
*out = NULL;
return E_NOINTERFACE;
}
IUnknown_AddRef((IUnknown*)*out);
return S_OK;
}
static ULONG WINAPI src_reader_AddRef(IMFSourceReader *iface)
{
struct source_reader *reader = impl_from_IMFSourceReader(iface);
ULONG refcount = InterlockedIncrement(&reader->refcount);
TRACE("%p, refcount %u.\n", iface, refcount);
return refcount;
}
static ULONG WINAPI src_reader_Release(IMFSourceReader *iface)
{
struct source_reader *reader = impl_from_IMFSourceReader(iface);
ULONG refcount = InterlockedDecrement(&reader->refcount);
unsigned int i;
TRACE("%p, refcount %u.\n", iface, refcount);
if (!refcount)
{
if (reader->async_callback)
IMFSourceReaderCallback_Release(reader->async_callback);
if (reader->shutdown_on_release)
IMFMediaSource_Shutdown(reader->source);
if (reader->descriptor)
IMFPresentationDescriptor_Release(reader->descriptor);
IMFMediaSource_Release(reader->source);
for (i = 0; i < reader->stream_count; ++i)
{
struct media_stream *stream = &reader->streams[i];
struct sample *ptr, *next;
if (stream->stream)
IMFMediaStream_Release(stream->stream);
if (stream->current)
IMFMediaType_Release(stream->current);
DeleteCriticalSection(&stream->cs);
LIST_FOR_EACH_ENTRY_SAFE(ptr, next, &stream->samples, struct sample, entry)
{
IMFSample_Release(ptr->sample);
list_remove(&ptr->entry);
heap_free(ptr);
}
}
heap_free(reader->streams);
DeleteCriticalSection(&reader->cs);
heap_free(reader);
}
return refcount;
}
static HRESULT WINAPI src_reader_GetStreamSelection(IMFSourceReader *iface, DWORD index, BOOL *selected)
{
struct source_reader *reader = impl_from_IMFSourceReader(iface);
IMFStreamDescriptor *sd;
TRACE("%p, %#x, %p.\n", iface, index, selected);
switch (index)
{
case MF_SOURCE_READER_FIRST_VIDEO_STREAM:
index = reader->first_video_stream_index;
break;
case MF_SOURCE_READER_FIRST_AUDIO_STREAM:
index = reader->first_audio_stream_index;
break;
default:
;
}
if (FAILED(IMFPresentationDescriptor_GetStreamDescriptorByIndex(reader->descriptor, index, selected, &sd)))
return MF_E_INVALIDSTREAMNUMBER;
IMFStreamDescriptor_Release(sd);
return S_OK;
}
static HRESULT WINAPI src_reader_SetStreamSelection(IMFSourceReader *iface, DWORD index, BOOL selected)
{
struct source_reader *reader = impl_from_IMFSourceReader(iface);
unsigned int count;
HRESULT hr;
TRACE("%p, %#x, %d.\n", iface, index, selected);
switch (index)
{
case MF_SOURCE_READER_FIRST_VIDEO_STREAM:
index = reader->first_video_stream_index;
break;
case MF_SOURCE_READER_FIRST_AUDIO_STREAM:
index = reader->first_audio_stream_index;
break;
case MF_SOURCE_READER_ALL_STREAMS:
if (FAILED(hr = IMFPresentationDescriptor_GetStreamDescriptorCount(reader->descriptor, &count)))
return hr;
for (index = 0; index < count; ++index)
{
if (selected)
IMFPresentationDescriptor_SelectStream(reader->descriptor, index);
else
IMFPresentationDescriptor_DeselectStream(reader->descriptor, index);
}
return S_OK;
default:
;
}
if (selected)
hr = IMFPresentationDescriptor_SelectStream(reader->descriptor, index);
else
hr = IMFPresentationDescriptor_DeselectStream(reader->descriptor, index);
if (FAILED(hr))
return MF_E_INVALIDSTREAMNUMBER;
return S_OK;
}
static HRESULT WINAPI src_reader_GetNativeMediaType(IMFSourceReader *iface, DWORD index, DWORD type_index,
IMFMediaType **type)
{
struct source_reader *reader = impl_from_IMFSourceReader(iface);
IMFMediaTypeHandler *handler;
IMFStreamDescriptor *sd;
IMFMediaType *src_type;
BOOL selected;
HRESULT hr;
TRACE("%p, %#x, %#x, %p.\n", iface, index, type_index, type);
switch (index)
{
case MF_SOURCE_READER_FIRST_VIDEO_STREAM:
index = reader->first_video_stream_index;
break;
case MF_SOURCE_READER_FIRST_AUDIO_STREAM:
index = reader->first_audio_stream_index;
break;
default:
;
}
if (FAILED(IMFPresentationDescriptor_GetStreamDescriptorByIndex(reader->descriptor, index, &selected, &sd)))
return MF_E_INVALIDSTREAMNUMBER;
hr = IMFStreamDescriptor_GetMediaTypeHandler(sd, &handler);
IMFStreamDescriptor_Release(sd);
if (FAILED(hr))
return hr;
if (type_index == MF_SOURCE_READER_CURRENT_TYPE_INDEX)
hr = IMFMediaTypeHandler_GetCurrentMediaType(handler, &src_type);
else
hr = IMFMediaTypeHandler_GetMediaTypeByIndex(handler, type_index, &src_type);
IMFMediaTypeHandler_Release(handler);
if (SUCCEEDED(hr))
{
if (SUCCEEDED(hr = MFCreateMediaType(type)))
hr = IMFMediaType_CopyAllItems(src_type, (IMFAttributes *)*type);
IMFMediaType_Release(src_type);
}
return hr;
}
static HRESULT WINAPI src_reader_GetCurrentMediaType(IMFSourceReader *iface, DWORD index, IMFMediaType **type)
{
struct source_reader *reader = impl_from_IMFSourceReader(iface);
HRESULT hr;
TRACE("%p, %#x, %p.\n", iface, index, type);
switch (index)
{
case MF_SOURCE_READER_FIRST_VIDEO_STREAM:
index = reader->first_video_stream_index;
break;
case MF_SOURCE_READER_FIRST_AUDIO_STREAM:
index = reader->first_audio_stream_index;
break;
default:
;
}
if (index >= reader->stream_count)
return MF_E_INVALIDSTREAMNUMBER;
if (FAILED(hr = MFCreateMediaType(type)))
return hr;
EnterCriticalSection(&reader->cs);
hr = IMFMediaType_CopyAllItems(reader->streams[index].current, (IMFAttributes *)*type);
LeaveCriticalSection(&reader->cs);
return hr;
}
static HRESULT WINAPI src_reader_SetCurrentMediaType(IMFSourceReader *iface, DWORD index, DWORD *reserved,
IMFMediaType *type)
{
struct source_reader *reader = impl_from_IMFSourceReader(iface);
HRESULT hr;
TRACE("%p, %#x, %p, %p.\n", iface, index, reserved, type);
switch (index)
{
case MF_SOURCE_READER_FIRST_VIDEO_STREAM:
index = reader->first_video_stream_index;
break;
case MF_SOURCE_READER_FIRST_AUDIO_STREAM:
index = reader->first_audio_stream_index;
break;
default:
;
}
if (index >= reader->stream_count)
return MF_E_INVALIDSTREAMNUMBER;
/* FIXME: validate passed type and current presentation state. */
EnterCriticalSection(&reader->cs);
hr = IMFMediaType_CopyAllItems(type, (IMFAttributes *)reader->streams[index].current);
LeaveCriticalSection(&reader->cs);
return hr;
}
static HRESULT WINAPI src_reader_SetCurrentPosition(IMFSourceReader *iface, REFGUID format, REFPROPVARIANT position)
{
struct source_reader *reader = impl_from_IMFSourceReader(iface);
DWORD flags;
HRESULT hr;
TRACE("%p, %s, %p.\n", iface, debugstr_guid(format), position);
/* FIXME: fail if we got pending samples. */
if (FAILED(hr = IMFMediaSource_GetCharacteristics(reader->source, &flags)))
return hr;
if (!(flags & MFMEDIASOURCE_CAN_SEEK))
return MF_E_INVALIDREQUEST;
return IMFMediaSource_Start(reader->source, reader->descriptor, format, position);
}
static IMFSample *media_stream_pop_sample(struct media_stream *stream, DWORD *stream_flags)
{
IMFSample *ret = NULL;
struct list *head;
if ((head = list_head(&stream->samples)))
{
struct sample *pending_sample = LIST_ENTRY(head, struct sample, entry);
ret = pending_sample->sample;
list_remove(&pending_sample->entry);
heap_free(pending_sample);
}
*stream_flags = stream->state == STREAM_STATE_EOS ? MF_SOURCE_READERF_ENDOFSTREAM : 0;
return ret;
}
static HRESULT WINAPI src_reader_ReadSample(IMFSourceReader *iface, DWORD index, DWORD flags, DWORD *actual_index,
DWORD *stream_flags, LONGLONG *timestamp, IMFSample **sample)
{
struct source_reader *reader = impl_from_IMFSourceReader(iface);
DWORD stream_index;
HRESULT hr;
TRACE("%p, %#x, %#x, %p, %p, %p, %p\n", iface, index, flags, actual_index, stream_flags, timestamp, sample);
switch (index)
{
case MF_SOURCE_READER_FIRST_VIDEO_STREAM:
stream_index = reader->first_video_stream_index;
break;
case MF_SOURCE_READER_FIRST_AUDIO_STREAM:
stream_index = reader->first_audio_stream_index;
break;
case MF_SOURCE_READER_ANY_STREAM:
FIXME("Non-specific requests are not supported.\n");
return E_NOTIMPL;
default:
stream_index = index;
}
/* FIXME: probably should happen once */
IMFMediaSource_Start(reader->source, reader->descriptor, NULL, NULL);
if (reader->async_callback)
{
FIXME("Async mode is not implemented.\n");
return E_NOTIMPL;
}
else
{
struct media_stream *stream;
if (!stream_flags || !sample)
return E_POINTER;
*sample = NULL;
if (stream_index >= reader->stream_count)
{
*stream_flags = MF_SOURCE_READERF_ERROR;
if (actual_index)
*actual_index = index;
return MF_E_INVALIDSTREAMNUMBER;
}
if (actual_index)
*actual_index = stream_index;
stream = &reader->streams[stream_index];
EnterCriticalSection(&stream->cs);
if (!(flags & MF_SOURCE_READER_CONTROLF_DRAIN))
{
while (list_empty(&stream->samples) && stream->state != STREAM_STATE_EOS)
{
if (stream->stream)
{
if (FAILED(hr = IMFMediaStream_RequestSample(stream->stream, NULL)))
WARN("Sample request failed, hr %#x.\n", hr);
}
SleepConditionVariableCS(&stream->sample_event, &stream->cs, INFINITE);
}
}
*sample = media_stream_pop_sample(stream, stream_flags);
LeaveCriticalSection(&stream->cs);
TRACE("Got sample %p.\n", *sample);
if (timestamp)
{
/* TODO: it's possible timestamp has to be set for some events.
For MEEndOfStream it's correct to return 0. */
*timestamp = 0;
if (*sample)
IMFSample_GetSampleTime(*sample, timestamp);
}
}
return S_OK;
}
static HRESULT WINAPI src_reader_Flush(IMFSourceReader *iface, DWORD index)
{
FIXME("%p, %#x.\n", iface, index);
return E_NOTIMPL;
}
static HRESULT WINAPI src_reader_GetServiceForStream(IMFSourceReader *iface, DWORD index, REFGUID service,
REFIID riid, void **object)
{
struct source_reader *reader = impl_from_IMFSourceReader(iface);
IUnknown *obj = NULL;
HRESULT hr;
TRACE("%p, %#x, %s, %s, %p\n", iface, index, debugstr_guid(service), debugstr_guid(riid), object);
switch (index)
{
case MF_SOURCE_READER_MEDIASOURCE:
obj = (IUnknown *)reader->source;
break;
default:
FIXME("Unsupported index %#x.\n", index);
return E_NOTIMPL;
}
if (IsEqualGUID(service, &GUID_NULL))
{
hr = IUnknown_QueryInterface(obj, riid, object);
}
else
{
IMFGetService *gs;
hr = IUnknown_QueryInterface(obj, &IID_IMFGetService, (void **)&gs);
if (SUCCEEDED(hr))
{
hr = IMFGetService_GetService(gs, service, riid, object);
IMFGetService_Release(gs);
}
}
return hr;
}
static HRESULT WINAPI src_reader_GetPresentationAttribute(IMFSourceReader *iface, DWORD index,
REFGUID guid, PROPVARIANT *value)
{
struct source_reader *reader = impl_from_IMFSourceReader(iface);
IMFStreamDescriptor *sd;
BOOL selected;
HRESULT hr;
TRACE("%p, %#x, %s, %p.\n", iface, index, debugstr_guid(guid), value);
switch (index)
{
case MF_SOURCE_READER_MEDIASOURCE:
if (IsEqualGUID(guid, &MF_SOURCE_READER_MEDIASOURCE_CHARACTERISTICS))
{
DWORD flags;
if (FAILED(hr = IMFMediaSource_GetCharacteristics(reader->source, &flags)))
return hr;
value->vt = VT_UI4;
value->u.ulVal = flags;
return S_OK;
}
else
{
return IMFPresentationDescriptor_GetItem(reader->descriptor, guid, value);
}
break;
case MF_SOURCE_READER_FIRST_VIDEO_STREAM:
index = reader->first_video_stream_index;
break;
case MF_SOURCE_READER_FIRST_AUDIO_STREAM:
index = reader->first_audio_stream_index;
break;
default:
;
}
if (FAILED(hr = IMFPresentationDescriptor_GetStreamDescriptorByIndex(reader->descriptor, index, &selected, &sd)))
return hr;
hr = IMFStreamDescriptor_GetItem(sd, guid, value);
IMFStreamDescriptor_Release(sd);
return hr;
}
struct IMFSourceReaderVtbl srcreader_vtbl =
{
src_reader_QueryInterface,
src_reader_AddRef,
src_reader_Release,
src_reader_GetStreamSelection,
src_reader_SetStreamSelection,
src_reader_GetNativeMediaType,
src_reader_GetCurrentMediaType,
src_reader_SetCurrentMediaType,
src_reader_SetCurrentPosition,
src_reader_ReadSample,
src_reader_Flush,
src_reader_GetServiceForStream,
src_reader_GetPresentationAttribute
};
static DWORD reader_get_first_stream_index(IMFPresentationDescriptor *descriptor, const GUID *major)
{
unsigned int count, i;
BOOL selected;
HRESULT hr;
GUID guid;
if (FAILED(IMFPresentationDescriptor_GetStreamDescriptorCount(descriptor, &count)))
return MF_SOURCE_READER_INVALID_STREAM_INDEX;
for (i = 0; i < count; ++i)
{
IMFMediaTypeHandler *handler;
IMFStreamDescriptor *sd;
if (SUCCEEDED(IMFPresentationDescriptor_GetStreamDescriptorByIndex(descriptor, i, &selected, &sd)))
{
hr = IMFStreamDescriptor_GetMediaTypeHandler(sd, &handler);
IMFStreamDescriptor_Release(sd);
if (SUCCEEDED(hr))
{
hr = IMFMediaTypeHandler_GetMajorType(handler, &guid);
IMFMediaTypeHandler_Release(handler);
if (FAILED(hr))
{
WARN("Failed to get stream major type, hr %#x.\n", hr);
continue;
}
if (IsEqualGUID(&guid, major))
{
return i;
}
}
}
}
return MF_SOURCE_READER_INVALID_STREAM_INDEX;
}
static HRESULT create_source_reader_from_source(IMFMediaSource *source, IMFAttributes *attributes,
BOOL shutdown_on_release, REFIID riid, void **out)
{
struct source_reader *object;
unsigned int i;
HRESULT hr;
object = heap_alloc_zero(sizeof(*object));
if (!object)
return E_OUTOFMEMORY;
object->IMFSourceReader_iface.lpVtbl = &srcreader_vtbl;
object->source_events_callback.lpVtbl = &source_events_callback_vtbl;
object->stream_events_callback.lpVtbl = &stream_events_callback_vtbl;
object->refcount = 1;
object->source = source;
IMFMediaSource_AddRef(object->source);
InitializeCriticalSection(&object->cs);
if (FAILED(hr = IMFMediaSource_CreatePresentationDescriptor(object->source, &object->descriptor)))
goto failed;
if (FAILED(hr = IMFPresentationDescriptor_GetStreamDescriptorCount(object->descriptor, &object->stream_count)))
goto failed;
if (!(object->streams = heap_alloc_zero(object->stream_count * sizeof(*object->streams))))
{
hr = E_OUTOFMEMORY;
goto failed;
}
/* Set initial current media types. */
for (i = 0; i < object->stream_count; ++i)
{
IMFMediaTypeHandler *handler;
IMFStreamDescriptor *sd;
IMFMediaType *src_type;
BOOL selected;
if (FAILED(hr = MFCreateMediaType(&object->streams[i].current)))
break;
if (FAILED(hr = IMFPresentationDescriptor_GetStreamDescriptorByIndex(object->descriptor, i, &selected, &sd)))
break;
if (FAILED(hr = IMFStreamDescriptor_GetStreamIdentifier(sd, &object->streams[i].id)))
WARN("Failed to get stream identifier, hr %#x.\n", hr);
hr = IMFStreamDescriptor_GetMediaTypeHandler(sd, &handler);
IMFStreamDescriptor_Release(sd);
if (FAILED(hr))
break;
hr = IMFMediaTypeHandler_GetMediaTypeByIndex(handler, 0, &src_type);
IMFMediaTypeHandler_Release(handler);
if (FAILED(hr))
break;
hr = IMFMediaType_CopyAllItems(src_type, (IMFAttributes *)object->streams[i].current);
IMFMediaType_Release(src_type);
if (FAILED(hr))
break;
InitializeCriticalSection(&object->streams[i].cs);
InitializeConditionVariable(&object->streams[i].sample_event);
list_init(&object->streams[i].samples);
}
if (FAILED(hr))
goto failed;
/* At least one major type has to be set. */
object->first_audio_stream_index = reader_get_first_stream_index(object->descriptor, &MFMediaType_Audio);
object->first_video_stream_index = reader_get_first_stream_index(object->descriptor, &MFMediaType_Video);
if (object->first_audio_stream_index == MF_SOURCE_READER_INVALID_STREAM_INDEX &&
object->first_video_stream_index == MF_SOURCE_READER_INVALID_STREAM_INDEX)
{
hr = MF_E_ATTRIBUTENOTFOUND;
}
if (FAILED(hr = IMFMediaSource_BeginGetEvent(object->source, &object->source_events_callback,
(IUnknown *)object->source)))
{
goto failed;
}
if (attributes)
{
IMFAttributes_GetUnknown(attributes, &MF_SOURCE_READER_ASYNC_CALLBACK, &IID_IMFSourceReaderCallback,
(void **)&object->async_callback);
if (object->async_callback)
TRACE("Using async callback %p.\n", object->async_callback);
}
hr = IMFSourceReader_QueryInterface(&object->IMFSourceReader_iface, riid, out);
failed:
IMFSourceReader_Release(&object->IMFSourceReader_iface);
return hr;
}
static HRESULT bytestream_get_url_hint(IMFByteStream *stream, WCHAR const **url)
{
static const UINT8 asfmagic[] = {0x30,0x26,0xb2,0x75,0x8e,0x66,0xcf,0x11,0xa6,0xd9,0x00,0xaa,0x00,0x62,0xce,0x6c};
static const UINT8 wavmagic[] = { 'R', 'I', 'F', 'F',0x00,0x00,0x00,0x00, 'W', 'A', 'V', 'E', 'f', 'm', 't', ' '};
static const UINT8 wavmask[] = {0xff,0xff,0xff,0xff,0x00,0x00,0x00,0x00,0xff,0xff,0xff,0xff,0xff,0xff,0xff,0xff};
static const WCHAR asfW[] = {'.','a','s','f',0};
static const WCHAR wavW[] = {'.','w','a','v',0};
static const struct stream_content_url_hint
{
const UINT8 *magic;
UINT32 magic_len;
const WCHAR *url;
const UINT8 *mask;
}
url_hints[] =
{
{ asfmagic, sizeof(asfmagic), asfW },
{ wavmagic, sizeof(wavmagic), wavW, wavmask },
};
UINT8 buffer[4 * sizeof(unsigned int)];
IMFAttributes *attributes;
UINT32 length = 0;
unsigned int i, j;
DWORD caps = 0;
QWORD position;
HRESULT hr;
*url = NULL;
if (SUCCEEDED(IMFByteStream_QueryInterface(stream, &IID_IMFAttributes, (void **)&attributes)))
{
IMFAttributes_GetStringLength(attributes, &MF_BYTESTREAM_CONTENT_TYPE, &length);
IMFAttributes_Release(attributes);
}
if (length)
return S_OK;
if (FAILED(hr = IMFByteStream_GetCapabilities(stream, &caps)))
return hr;
if (!(caps & MFBYTESTREAM_IS_SEEKABLE))
return S_OK;
if (FAILED(hr = IMFByteStream_GetCurrentPosition(stream, &position)))
return hr;
hr = IMFByteStream_Read(stream, buffer, sizeof(buffer), &length);
IMFByteStream_SetCurrentPosition(stream, position);
if (FAILED(hr))
return hr;
if (length < sizeof(buffer))
return S_OK;
for (i = 0; i < ARRAY_SIZE(url_hints); ++i)
{
if (url_hints[i].mask)
{
unsigned int *mask = (unsigned int *)url_hints[i].mask;
unsigned int *data = (unsigned int *)buffer;
for (j = 0; j < sizeof(buffer) / sizeof(unsigned int); ++j)
data[j] &= mask[j];
}
if (!memcmp(buffer, url_hints[i].magic, min(url_hints[i].magic_len, length)))
{
*url = url_hints[i].url;
break;
}
}
if (!*url)
WARN("Unrecognized content type %s.\n", debugstr_an((char *)buffer, length));
return S_OK;
}
static HRESULT create_source_reader_from_stream(IMFByteStream *stream, IMFAttributes *attributes,
REFIID riid, void **out)
{
IPropertyStore *props = NULL;
IMFSourceResolver *resolver;
MF_OBJECT_TYPE obj_type;
IMFMediaSource *source;
const WCHAR *url;
HRESULT hr;
/* If stream does not have content type set, try to guess from starting byte sequence. */
if (FAILED(hr = bytestream_get_url_hint(stream, &url)))
return hr;
if (FAILED(hr = MFCreateSourceResolver(&resolver)))
return hr;
if (attributes)
IMFAttributes_GetUnknown(attributes, &MF_SOURCE_READER_MEDIASOURCE_CONFIG, &IID_IPropertyStore,
(void **)&props);
hr = IMFSourceResolver_CreateObjectFromByteStream(resolver, stream, url, MF_RESOLUTION_MEDIASOURCE, props,
&obj_type, (IUnknown **)&source);
IMFSourceResolver_Release(resolver);
if (props)
IPropertyStore_Release(props);
if (FAILED(hr))
return hr;
hr = create_source_reader_from_source(source, attributes, TRUE, riid, out);
IMFMediaSource_Release(source);
return hr;
}
static HRESULT create_source_reader_from_url(const WCHAR *url, IMFAttributes *attributes, REFIID riid, void **out)
{
IPropertyStore *props = NULL;
IMFSourceResolver *resolver;
IUnknown *object = NULL;
MF_OBJECT_TYPE obj_type;
IMFMediaSource *source;
HRESULT hr;
if (FAILED(hr = MFCreateSourceResolver(&resolver)))
return hr;
if (attributes)
IMFAttributes_GetUnknown(attributes, &MF_SOURCE_READER_MEDIASOURCE_CONFIG, &IID_IPropertyStore,
(void **)&props);
hr = IMFSourceResolver_CreateObjectFromURL(resolver, url, MF_RESOLUTION_MEDIASOURCE, props, &obj_type,
&object);
if (SUCCEEDED(hr))
{
switch (obj_type)
{
case MF_OBJECT_BYTESTREAM:
hr = IMFSourceResolver_CreateObjectFromByteStream(resolver, (IMFByteStream *)object, NULL,
MF_RESOLUTION_MEDIASOURCE, props, &obj_type, (IUnknown **)&source);
break;
case MF_OBJECT_MEDIASOURCE:
source = (IMFMediaSource *)object;
IMFMediaSource_AddRef(source);
break;
default:
WARN("Unknown object type %d.\n", obj_type);
hr = E_UNEXPECTED;
}
IUnknown_Release(object);
}
IMFSourceResolver_Release(resolver);
if (props)
IPropertyStore_Release(props);
if (FAILED(hr))
return hr;
hr = create_source_reader_from_source(source, attributes, TRUE, riid, out);
IMFMediaSource_Release(source);
return hr;
}
static HRESULT WINAPI sink_writer_QueryInterface(IMFSinkWriter *iface, REFIID riid, void **out)
{
TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), out);
if (IsEqualIID(riid, &IID_IMFSinkWriter) ||
IsEqualIID(riid, &IID_IUnknown))
{
*out = iface;
IMFSinkWriter_AddRef(iface);
return S_OK;
}
WARN("Unsupported %s.\n", debugstr_guid(riid));
*out = NULL;
return E_NOINTERFACE;
}
static ULONG WINAPI sink_writer_AddRef(IMFSinkWriter *iface)
{
struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
ULONG refcount = InterlockedIncrement(&writer->refcount);
TRACE("%p, %u.\n", iface, refcount);
return refcount;
}
static ULONG WINAPI sink_writer_Release(IMFSinkWriter *iface)
{
struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
ULONG refcount = InterlockedDecrement(&writer->refcount);
TRACE("%p, %u.\n", iface, refcount);
if (!refcount)
{
heap_free(writer);
}
return refcount;
}
static HRESULT WINAPI sink_writer_AddStream(IMFSinkWriter *iface, IMFMediaType *type, DWORD *index)
{
FIXME("%p, %p, %p.\n", iface, type, index);
return E_NOTIMPL;
}
static HRESULT WINAPI sink_writer_SetInputMediaType(IMFSinkWriter *iface, DWORD index, IMFMediaType *type,
IMFAttributes *parameters)
{
FIXME("%p, %u, %p, %p.\n", iface, index, type, parameters);
return E_NOTIMPL;
}
static HRESULT WINAPI sink_writer_BeginWriting(IMFSinkWriter *iface)
{
FIXME("%p.\n", iface);
return E_NOTIMPL;
}
static HRESULT WINAPI sink_writer_WriteSample(IMFSinkWriter *iface, DWORD index, IMFSample *sample)
{
FIXME("%p, %u, %p.\n", iface, index, sample);
return E_NOTIMPL;
}
static HRESULT WINAPI sink_writer_SendStreamTick(IMFSinkWriter *iface, DWORD index, LONGLONG timestamp)
{
FIXME("%p, %u, %s.\n", iface, index, wine_dbgstr_longlong(timestamp));
return E_NOTIMPL;
}
static HRESULT WINAPI sink_writer_PlaceMarker(IMFSinkWriter *iface, DWORD index, void *context)
{
FIXME("%p, %u, %p.\n", iface, index, context);
return E_NOTIMPL;
}
static HRESULT WINAPI sink_writer_NotifyEndOfSegment(IMFSinkWriter *iface, DWORD index)
{
FIXME("%p, %u.\n", iface, index);
return E_NOTIMPL;
}
static HRESULT WINAPI sink_writer_Flush(IMFSinkWriter *iface, DWORD index)
{
FIXME("%p, %u.\n", iface, index);
return E_NOTIMPL;
}
static HRESULT WINAPI sink_writer_Finalize(IMFSinkWriter *iface)
{
FIXME("%p.\n", iface);
return E_NOTIMPL;
}
static HRESULT WINAPI sink_writer_GetServiceForStream(IMFSinkWriter *iface, DWORD index, REFGUID service,
REFIID riid, void **object)
{
FIXME("%p, %u, %s, %s, %p.\n", iface, index, debugstr_guid(service), debugstr_guid(riid), object);
return E_NOTIMPL;
}
static HRESULT WINAPI sink_writer_GetStatistics(IMFSinkWriter *iface, DWORD index, MF_SINK_WRITER_STATISTICS *stats)
{
FIXME("%p, %u, %p.\n", iface, index, stats);
return E_NOTIMPL;
}
static const IMFSinkWriterVtbl sink_writer_vtbl =
{
sink_writer_QueryInterface,
sink_writer_AddRef,
sink_writer_Release,
sink_writer_AddStream,
sink_writer_SetInputMediaType,
sink_writer_BeginWriting,
sink_writer_WriteSample,
sink_writer_SendStreamTick,
sink_writer_PlaceMarker,
sink_writer_NotifyEndOfSegment,
sink_writer_Flush,
sink_writer_Finalize,
sink_writer_GetServiceForStream,
sink_writer_GetStatistics,
};
static HRESULT create_sink_writer_from_sink(IMFMediaSink *sink, IMFAttributes *attributes,
REFIID riid, void **out)
{
struct sink_writer *object;
HRESULT hr;
object = heap_alloc(sizeof(*object));
if (!object)
return E_OUTOFMEMORY;
object->IMFSinkWriter_iface.lpVtbl = &sink_writer_vtbl;
object->refcount = 1;
hr = IMFSinkWriter_QueryInterface(&object->IMFSinkWriter_iface, riid, out);
IMFSinkWriter_Release(&object->IMFSinkWriter_iface);
return hr;
}
static HRESULT create_sink_writer_from_stream(IMFByteStream *stream, IMFAttributes *attributes,
REFIID riid, void **out)
{
struct sink_writer *object;
HRESULT hr;
object = heap_alloc(sizeof(*object));
if (!object)
return E_OUTOFMEMORY;
object->IMFSinkWriter_iface.lpVtbl = &sink_writer_vtbl;
object->refcount = 1;
hr = IMFSinkWriter_QueryInterface(&object->IMFSinkWriter_iface, riid, out);
IMFSinkWriter_Release(&object->IMFSinkWriter_iface);
return hr;
}
/***********************************************************************
* MFCreateSinkWriterFromMediaSink (mfreadwrite.@)
*/
HRESULT WINAPI MFCreateSinkWriterFromMediaSink(IMFMediaSink *sink, IMFAttributes *attributes, IMFSinkWriter **writer)
{
TRACE("%p, %p, %p.\n", sink, attributes, writer);
return create_sink_writer_from_sink(sink, attributes, &IID_IMFSinkWriter, (void **)writer);
}
/***********************************************************************
* MFCreateSinkWriterFromURL (mfreadwrite.@)
*/
HRESULT WINAPI MFCreateSinkWriterFromURL(const WCHAR *url, IMFByteStream *bytestream, IMFAttributes *attributes,
IMFSinkWriter **writer)
{
FIXME("%s, %p, %p, %p.\n", debugstr_w(url), bytestream, attributes, writer);
return E_NOTIMPL;
}
static HRESULT create_source_reader_from_object(IUnknown *unk, IMFAttributes *attributes, REFIID riid, void **out)
{
IMFMediaSource *source = NULL;
IMFByteStream *stream = NULL;
HRESULT hr;
hr = IUnknown_QueryInterface(unk, &IID_IMFMediaSource, (void **)&source);
if (FAILED(hr))
hr = IUnknown_QueryInterface(unk, &IID_IMFByteStream, (void **)&stream);
if (source)
{
UINT32 disconnect = 0;
if (attributes)
IMFAttributes_GetUINT32(attributes, &MF_SOURCE_READER_DISCONNECT_MEDIASOURCE_ON_SHUTDOWN, &disconnect);
hr = create_source_reader_from_source(source, attributes, !disconnect, riid, out);
}
else if (stream)
hr = create_source_reader_from_stream(stream, attributes, riid, out);
if (source)
IMFMediaSource_Release(source);
if (stream)
IMFByteStream_Release(stream);
return hr;
}
/***********************************************************************
* MFCreateSourceReaderFromByteStream (mfreadwrite.@)
*/
HRESULT WINAPI MFCreateSourceReaderFromByteStream(IMFByteStream *stream, IMFAttributes *attributes,
IMFSourceReader **reader)
{
TRACE("%p, %p, %p.\n", stream, attributes, reader);
return create_source_reader_from_object((IUnknown *)stream, attributes, &IID_IMFSourceReader, (void **)reader);
}
/***********************************************************************
* MFCreateSourceReaderFromMediaSource (mfreadwrite.@)
*/
HRESULT WINAPI MFCreateSourceReaderFromMediaSource(IMFMediaSource *source, IMFAttributes *attributes,
IMFSourceReader **reader)
{
TRACE("%p, %p, %p.\n", source, attributes, reader);
return create_source_reader_from_object((IUnknown *)source, attributes, &IID_IMFSourceReader, (void **)reader);
}
/***********************************************************************
* MFCreateSourceReaderFromURL (mfreadwrite.@)
*/
HRESULT WINAPI MFCreateSourceReaderFromURL(const WCHAR *url, IMFAttributes *attributes, IMFSourceReader **reader)
{
TRACE("%s, %p, %p.\n", debugstr_w(url), attributes, reader);
return create_source_reader_from_url(url, attributes, &IID_IMFSourceReader, (void **)reader);
}
static HRESULT WINAPI readwrite_factory_QueryInterface(IMFReadWriteClassFactory *iface, REFIID riid, void **out)
{
if (IsEqualIID(riid, &IID_IMFReadWriteClassFactory) ||
IsEqualIID(riid, &IID_IUnknown))
{
*out = iface;
IMFReadWriteClassFactory_AddRef(iface);
return S_OK;
}
WARN("Unsupported interface %s.\n", debugstr_guid(riid));
*out = NULL;
return E_NOINTERFACE;
}
static ULONG WINAPI readwrite_factory_AddRef(IMFReadWriteClassFactory *iface)
{
return 2;
}
static ULONG WINAPI readwrite_factory_Release(IMFReadWriteClassFactory *iface)
{
return 1;
}
static HRESULT WINAPI readwrite_factory_CreateInstanceFromURL(IMFReadWriteClassFactory *iface, REFCLSID clsid,
const WCHAR *url, IMFAttributes *attributes, REFIID riid, void **out)
{
TRACE("%s, %s, %p, %s, %p.\n", debugstr_guid(clsid), debugstr_w(url), attributes, debugstr_guid(riid), out);
if (IsEqualGUID(clsid, &CLSID_MFSourceReader))
{
return create_source_reader_from_url(url, attributes, &IID_IMFSourceReader, out);
}
FIXME("Unsupported %s.\n", debugstr_guid(clsid));
return E_NOTIMPL;
}
static HRESULT WINAPI readwrite_factory_CreateInstanceFromObject(IMFReadWriteClassFactory *iface, REFCLSID clsid,
IUnknown *unk, IMFAttributes *attributes, REFIID riid, void **out)
{
HRESULT hr;
TRACE("%s, %p, %p, %s, %p.\n", debugstr_guid(clsid), unk, attributes, debugstr_guid(riid), out);
if (IsEqualGUID(clsid, &CLSID_MFSourceReader))
{
return create_source_reader_from_object(unk, attributes, riid, out);
}
else if (IsEqualGUID(clsid, &CLSID_MFSinkWriter))
{
IMFByteStream *stream = NULL;
IMFMediaSink *sink = NULL;
hr = IUnknown_QueryInterface(unk, &IID_IMFByteStream, (void **)&stream);
if (FAILED(hr))
hr = IUnknown_QueryInterface(unk, &IID_IMFMediaSink, (void **)&sink);
if (stream)
hr = create_sink_writer_from_stream(stream, attributes, riid, out);
else if (sink)
hr = create_sink_writer_from_sink(sink, attributes, riid, out);
if (sink)
IMFMediaSink_Release(sink);
if (stream)
IMFByteStream_Release(stream);
return hr;
}
else
{
WARN("Unsupported class %s.\n", debugstr_guid(clsid));
*out = NULL;
return E_FAIL;
}
}
static const IMFReadWriteClassFactoryVtbl readwrite_factory_vtbl =
{
readwrite_factory_QueryInterface,
readwrite_factory_AddRef,
readwrite_factory_Release,
readwrite_factory_CreateInstanceFromURL,
readwrite_factory_CreateInstanceFromObject,
};
static IMFReadWriteClassFactory readwrite_factory = { &readwrite_factory_vtbl };
static HRESULT WINAPI classfactory_QueryInterface(IClassFactory *iface, REFIID riid, void **out)
{
TRACE("%s, %p.\n", debugstr_guid(riid), out);
if (IsEqualGUID(riid, &IID_IClassFactory) ||
IsEqualGUID(riid, &IID_IUnknown))
{
IClassFactory_AddRef(iface);
*out = iface;
return S_OK;
}
WARN("interface %s not implemented\n", debugstr_guid(riid));
*out = NULL;
return E_NOINTERFACE;
}
static ULONG WINAPI classfactory_AddRef(IClassFactory *iface)
{
return 2;
}
static ULONG WINAPI classfactory_Release(IClassFactory *iface)
{
return 1;
}
static HRESULT WINAPI classfactory_CreateInstance(IClassFactory *iface, IUnknown *outer, REFIID riid, void **out)
{
TRACE("%p, %s, %p.\n", outer, debugstr_guid(riid), out);
*out = NULL;
if (outer)
return CLASS_E_NOAGGREGATION;
return IMFReadWriteClassFactory_QueryInterface(&readwrite_factory, riid, out);
}
static HRESULT WINAPI classfactory_LockServer(IClassFactory *iface, BOOL dolock)
{
FIXME("%d.\n", dolock);
return S_OK;
}
static const struct IClassFactoryVtbl classfactoryvtbl =
{
classfactory_QueryInterface,
classfactory_AddRef,
classfactory_Release,
classfactory_CreateInstance,
classfactory_LockServer,
};
static IClassFactory classfactory = { &classfactoryvtbl };
HRESULT WINAPI DllGetClassObject(REFCLSID clsid, REFIID riid, void **out)
{
TRACE("%s, %s, %p.\n", debugstr_guid(clsid), debugstr_guid(riid), out);
if (IsEqualGUID(clsid, &CLSID_MFReadWriteClassFactory))
return IClassFactory_QueryInterface(&classfactory, riid, out);
WARN("Unsupported class %s.\n", debugstr_guid(clsid));
*out = NULL;
return CLASS_E_CLASSNOTAVAILABLE;
}