dlls/mfreadwrite/writer.c

   1 /*
   2  * Copyright 2020 Nikolay Sivov for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   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  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include <assert.h>
  20
  21 #define COBJMACROS
  22 #define NONAMELESSUNION
  23
  24 #include "mfapi.h"
  25 #include "mfidl.h"
  26 #include "mfreadwrite.h"
  27 #include "pathcch.h"
  28 #include "wine/mfinternal.h"
  29 #include "mf_private.h"
  30
  31 #include "wine/debug.h"
  32 #include "wine/list.h"
  33
  34 WINE_DEFAULT_DEBUG_CHANNEL(mfplat);
  35
  36 enum writer_state
  37 {
  38     SINK_WRITER_STATE_INITIAL = 0,
  39     SINK_WRITER_STATE_WRITING,
  40 };
  41
  42 struct marker_context
  43 {
  44     IUnknown IUnknown_iface;
  45     LONG refcount;
  46
  47     unsigned int marker_type;
  48     void *user_context;
  49 };
  50
  51 struct pending_item
  52 {
  53     struct list entry;
  54     unsigned int marker_type;
  55     IMFSample *sample;
  56     LONGLONG timestamp;
  57     void *user_context;
  58 };
  59
  60 struct stream
  61 {
  62     IMFStreamSink *stream_sink;
  63     IMFTransform *encoder;
  64     MF_SINK_WRITER_STATISTICS stats;
  65     struct list queue;
  66 };
  67
  68 struct sink_writer
  69 {
  70     IMFSinkWriter IMFSinkWriter_iface;
  71     IMFAsyncCallback events_callback;
  72     LONG refcount;
  73
  74     struct
  75     {
  76         struct stream *items;
  77         size_t count;
  78         size_t capacity;
  79         DWORD next_id;
  80     } streams;
  81
  82     IMFPresentationClock *clock;
  83     IMFMediaSink *sink;
  84     enum writer_state state;
  85     HRESULT status;
  86     MF_SINK_WRITER_STATISTICS stats;
  87
  88     IMFSinkWriterCallback *callback;
  89
  90     CRITICAL_SECTION cs;
  91 };
  92
  93 static struct sink_writer *impl_from_IMFSinkWriter(IMFSinkWriter *iface)
  94 {
  95     return CONTAINING_RECORD(iface, struct sink_writer, IMFSinkWriter_iface);
  96 }
  97
  98 static struct sink_writer *impl_from_events_callback_IMFAsyncCallback(IMFAsyncCallback *iface)
  99 {
 100     return CONTAINING_RECORD(iface, struct sink_writer, events_callback);
 101 }
 102
 103 static struct marker_context *impl_from_marker_context_IUnknown(IUnknown *iface)
 104 {
 105     return CONTAINING_RECORD(iface, struct marker_context, IUnknown_iface);
 106 }
 107
 108 static HRESULT WINAPI marker_context_QueryInterface(IUnknown *iface, REFIID riid, void **out)
 109 {
 110     if (IsEqualIID(riid, &IID_IUnknown))
 111     {
 112         *out = iface;
 113         IUnknown_AddRef(iface);
 114         return S_OK;
 115     }
 116
 117     *out = NULL;
 118     return E_NOINTERFACE;
 119 }
 120
 121 static ULONG WINAPI marker_context_AddRef(IUnknown *iface)
 122 {
 123     struct marker_context *context = impl_from_marker_context_IUnknown(iface);
 124     return InterlockedIncrement(&context->refcount);
 125 }
 126
 127 static ULONG WINAPI marker_context_Release(IUnknown *iface)
 128 {
 129     struct marker_context *context = impl_from_marker_context_IUnknown(iface);
 130     LONG refcount = InterlockedDecrement(&context->refcount);
 131
 132     if (!refcount)
 133         free(context);
 134
 135     return refcount;
 136 }
 137
 138 static const IUnknownVtbl marker_context_vtbl =
 139 {
 140     marker_context_QueryInterface,
 141     marker_context_AddRef,
 142     marker_context_Release,
 143 };
 144
 145 static struct marker_context *unsafe_impl_from_marker_context_IUnknown(IUnknown *iface)
 146 {
 147     if (!iface)
 148         return NULL;
 149     assert(iface->lpVtbl == &marker_context_vtbl);
 150     return CONTAINING_RECORD(iface, struct marker_context, IUnknown_iface);
 151 }
 152
 153 static HRESULT create_marker_context(unsigned int marker_type, void *user_context,
 154         IUnknown **ret)
 155 {
 156     struct marker_context *object;
 157
 158     if (!(object = calloc(1, sizeof(*object))))
 159         return E_OUTOFMEMORY;
 160
 161     object->IUnknown_iface.lpVtbl = &marker_context_vtbl;
 162     object->refcount = 1;
 163     object->marker_type = marker_type;
 164     object->user_context = user_context;
 165
 166     *ret = &object->IUnknown_iface;
 167
 168     return S_OK;
 169 }
 170
 171 static void sink_writer_release_pending_item(struct pending_item *item)
 172 {
 173     list_remove(&item->entry);
 174     if (item->sample)
 175         IMFSample_Release(item->sample);
 176     free(item);
 177 }
 178
 179 static HRESULT WINAPI sink_writer_QueryInterface(IMFSinkWriter *iface, REFIID riid, void **out)
 180 {
 181     TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), out);
 182
 183     if (IsEqualIID(riid, &IID_IMFSinkWriter) ||
 184             IsEqualIID(riid, &IID_IUnknown))
 185     {
 186         *out = iface;
 187         IMFSinkWriter_AddRef(iface);
 188         return S_OK;
 189     }
 190
 191     WARN("Unsupported %s.\n", debugstr_guid(riid));
 192     *out = NULL;
 193     return E_NOINTERFACE;
 194 }
 195
 196 static ULONG WINAPI sink_writer_AddRef(IMFSinkWriter *iface)
 197 {
 198     struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
 199     ULONG refcount = InterlockedIncrement(&writer->refcount);
 200
 201     TRACE("%p, %lu.\n", iface, refcount);
 202
 203     return refcount;
 204 }
 205
 206 static void sink_writer_drop_pending_items(struct stream *stream)
 207 {
 208     struct pending_item *item, *next;
 209
 210     LIST_FOR_EACH_ENTRY_SAFE(item, next, &stream->queue, struct pending_item, entry)
 211     {
 212         sink_writer_release_pending_item(item);
 213     }
 214 }
 215
 216 static ULONG WINAPI sink_writer_Release(IMFSinkWriter *iface)
 217 {
 218     struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
 219     ULONG refcount = InterlockedDecrement(&writer->refcount);
 220     unsigned int i;
 221
 222     TRACE("%p, %lu.\n", iface, refcount);
 223
 224     if (!refcount)
 225     {
 226         if (writer->clock)
 227             IMFPresentationClock_Release(writer->clock);
 228         if (writer->sink)
 229             IMFMediaSink_Release(writer->sink);
 230         if (writer->callback)
 231             IMFSinkWriterCallback_Release(writer->callback);
 232         for (i = 0; i < writer->streams.count; ++i)
 233         {
 234             struct stream *stream = &writer->streams.items[i];
 235
 236             if (stream->stream_sink)
 237                 IMFStreamSink_Release(stream->stream_sink);
 238             if (stream->encoder)
 239                 IMFTransform_Release(stream->encoder);
 240             sink_writer_drop_pending_items(stream);
 241         }
 242         DeleteCriticalSection(&writer->cs);
 243         free(writer);
 244     }
 245
 246     return refcount;
 247 }
 248
 249 static HRESULT sink_writer_add_stream(struct sink_writer *writer, IMFStreamSink *stream_sink, DWORD *index)
 250 {
 251     struct stream *stream;
 252     DWORD id = 0;
 253     HRESULT hr;
 254
 255     if (!mf_array_reserve((void **)&writer->streams.items, &writer->streams.capacity, writer->streams.count + 1,
 256             sizeof(*writer->streams.items)))
 257     {
 258         return E_OUTOFMEMORY;
 259     }
 260
 261     if (FAILED(hr = IMFStreamSink_GetIdentifier(stream_sink, &id))) return hr;
 262
 263     *index = writer->streams.count++;
 264
 265     stream = &writer->streams.items[*index];
 266     memset(stream, 0, sizeof(*stream));
 267     stream->stream_sink = stream_sink;
 268     IMFStreamSink_AddRef(stream_sink);
 269     stream->stats.cb = sizeof(stream->stats);
 270     list_init(&stream->queue);
 271
 272     writer->streams.next_id = max(writer->streams.next_id, id);
 273
 274     return hr;
 275 }
 276
 277 static HRESULT WINAPI sink_writer_AddStream(IMFSinkWriter *iface, IMFMediaType *media_type, DWORD *index)
 278 {
 279     struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
 280     HRESULT hr = MF_E_INVALIDREQUEST;
 281     IMFStreamSink *stream_sink;
 282     DWORD id;
 283
 284     TRACE("%p, %p, %p.\n", iface, media_type, index);
 285
 286     if (!media_type)
 287         return E_INVALIDARG;
 288
 289     if (!index)
 290         return E_POINTER;
 291
 292     EnterCriticalSection(&writer->cs);
 293
 294     if (writer->state == SINK_WRITER_STATE_INITIAL)
 295     {
 296         id = writer->streams.next_id + 1;
 297         if (SUCCEEDED(hr = IMFMediaSink_AddStreamSink(writer->sink, id, media_type, &stream_sink)))
 298         {
 299             if (FAILED(hr = sink_writer_add_stream(writer, stream_sink, index)))
 300                 IMFMediaSink_RemoveStreamSink(writer->sink, id);
 301         }
 302     }
 303
 304     LeaveCriticalSection(&writer->cs);
 305
 306     return hr;
 307 }
 308
 309 static HRESULT WINAPI sink_writer_SetInputMediaType(IMFSinkWriter *iface, DWORD index, IMFMediaType *type,
 310         IMFAttributes *parameters)
 311 {
 312     FIXME("%p, %lu, %p, %p.\n", iface, index, type, parameters);
 313
 314     return E_NOTIMPL;
 315 }
 316
 317 static HRESULT sink_writer_set_presentation_clock(struct sink_writer *writer)
 318 {
 319     IMFPresentationTimeSource *time_source = NULL;
 320     HRESULT hr;
 321
 322     if (FAILED(hr = MFCreatePresentationClock(&writer->clock))) return hr;
 323
 324     if (FAILED(IMFMediaSink_QueryInterface(writer->sink, &IID_IMFPresentationTimeSource, (void **)&time_source)))
 325         hr = MFCreateSystemTimeSource(&time_source);
 326
 327     if (SUCCEEDED(hr = IMFPresentationClock_SetTimeSource(writer->clock, time_source)))
 328         hr = IMFMediaSink_SetPresentationClock(writer->sink, writer->clock);
 329
 330     if (time_source)
 331         IMFPresentationTimeSource_Release(time_source);
 332
 333     return hr;
 334 }
 335
 336 static HRESULT WINAPI sink_writer_BeginWriting(IMFSinkWriter *iface)
 337 {
 338     struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
 339     HRESULT hr = S_OK;
 340     unsigned int i;
 341
 342     TRACE("%p.\n", iface);
 343
 344     EnterCriticalSection(&writer->cs);
 345
 346     if (!writer->streams.count)
 347         hr = MF_E_INVALIDREQUEST;
 348     else if (writer->state != SINK_WRITER_STATE_INITIAL)
 349         hr = MF_E_INVALIDREQUEST;
 350     else if (SUCCEEDED(hr = sink_writer_set_presentation_clock(writer)))
 351     {
 352         for (i = 0; i < writer->streams.count; ++i)
 353         {
 354             struct stream *stream = &writer->streams.items[i];
 355
 356             if (FAILED(hr = IMFStreamSink_BeginGetEvent(stream->stream_sink, &writer->events_callback,
 357                     (IUnknown *)stream->stream_sink)))
 358             {
 359                 WARN("Failed to subscribe to events for steam %u, hr %#lx.\n", i, hr);
 360             }
 361
 362             if (stream->encoder)
 363                 IMFTransform_ProcessMessage(stream->encoder, MFT_MESSAGE_NOTIFY_BEGIN_STREAMING, 0);
 364         }
 365
 366         if (SUCCEEDED(hr))
 367             hr = IMFPresentationClock_Start(writer->clock, 0);
 368
 369         writer->state = SINK_WRITER_STATE_WRITING;
 370     }
 371
 372     LeaveCriticalSection(&writer->cs);
 373
 374     return hr;
 375 }
 376
 377 static struct stream * sink_writer_get_stream(const struct sink_writer *writer, DWORD index)
 378 {
 379     if (index >= writer->streams.count) return NULL;
 380     return &writer->streams.items[index];
 381 }
 382
 383 static HRESULT sink_writer_get_buffer_length(IMFSample *sample, LONGLONG *timestamp, DWORD *length)
 384 {
 385     IMFMediaBuffer *buffer;
 386     HRESULT hr;
 387
 388     *timestamp = 0;
 389     *length = 0;
 390
 391     if (FAILED(hr = IMFSample_ConvertToContiguousBuffer(sample, &buffer))) return hr;
 392     hr = IMFMediaBuffer_GetCurrentLength(buffer, length);
 393     IMFMediaBuffer_Release(buffer);
 394     if (FAILED(hr)) return hr;
 395     if (!*length) return E_INVALIDARG;
 396     IMFSample_GetSampleTime(sample, timestamp);
 397
 398     return hr;
 399 }
 400
 401 static HRESULT sink_writer_place_marker(struct sink_writer *writer, struct stream *stream, unsigned int marker_type,
 402         LONGLONG timestamp, void *user_context)
 403 {
 404     PROPVARIANT value, context;
 405     IUnknown *context_obj;
 406     HRESULT hr;
 407
 408     if (FAILED(hr = create_marker_context(marker_type, user_context, &context_obj))) return hr;
 409
 410     context.vt = VT_UNKNOWN;
 411     context.punkVal = context_obj;
 412     value.vt = VT_I8;
 413     value.hVal.QuadPart = timestamp;
 414     hr = IMFStreamSink_PlaceMarker(stream->stream_sink, marker_type, marker_type == MFSTREAMSINK_MARKER_TICK ? &value : NULL,
 415             &context);
 416     IUnknown_Release(context_obj);
 417     return hr;
 418 }
 419
 420 static HRESULT sink_writer_queue_marker(struct sink_writer *writer, struct stream *stream, unsigned int marker_type,
 421         LONGLONG timestamp, void *user_context)
 422 {
 423     struct pending_item *item;
 424
 425     if (list_empty(&stream->queue))
 426         return sink_writer_place_marker(writer, stream, marker_type, timestamp, user_context);
 427
 428     if (!(item = calloc(1, sizeof(*item))))
 429         return E_OUTOFMEMORY;
 430
 431     item->marker_type = marker_type;
 432     item->timestamp = timestamp;
 433     list_add_tail(&stream->queue, &item->entry);
 434
 435     return S_OK;
 436 }
 437
 438 static HRESULT sink_writer_send_stream_tick(struct sink_writer *writer, unsigned int index, LONGLONG timestamp)
 439 {
 440     struct stream *stream;
 441
 442     if (!(stream = sink_writer_get_stream(writer, index))) return MF_E_INVALIDSTREAMNUMBER;
 443
 444     writer->stats.llLastStreamTickReceived = timestamp;
 445     writer->stats.qwNumStreamTicksReceived++;
 446     stream->stats.llLastStreamTickReceived = timestamp;
 447     stream->stats.qwNumStreamTicksReceived++;
 448
 449     return sink_writer_queue_marker(writer, stream, MFSTREAMSINK_MARKER_TICK, timestamp, NULL);
 450 }
 451
 452 static HRESULT sink_writer_notify_end_of_segment(struct sink_writer *writer, unsigned int index)
 453 {
 454     struct stream *stream;
 455
 456     if (!(stream = sink_writer_get_stream(writer, index))) return MF_E_INVALIDSTREAMNUMBER;
 457
 458     return sink_writer_queue_marker(writer, stream, MFSTREAMSINK_MARKER_ENDOFSEGMENT, 0, NULL);
 459 }
 460
 461 static HRESULT sink_writer_flush(struct sink_writer *writer, unsigned int index)
 462 {
 463     struct stream *stream;
 464
 465     if (!(stream = sink_writer_get_stream(writer, index))) return MF_E_INVALIDSTREAMNUMBER;
 466
 467     sink_writer_drop_pending_items(stream);
 468
 469     IMFStreamSink_Flush(stream->stream_sink);
 470
 471     if (stream->encoder)
 472         IMFTransform_ProcessMessage(stream->encoder, MFT_MESSAGE_COMMAND_FLUSH, 0);
 473
 474     return sink_writer_place_marker(writer, stream, MFSTREAMSINK_MARKER_ENDOFSEGMENT, 0, NULL);
 475 }
 476
 477 static HRESULT sink_writer_process_sample(struct sink_writer *writer, struct stream *stream)
 478 {
 479     struct pending_item *item, *next;
 480     LONGLONG timestamp;
 481     IMFSample *sample;
 482     HRESULT hr;
 483
 484     if (list_empty(&stream->queue)) return S_OK;
 485     item = LIST_ENTRY(list_head(&stream->queue), struct pending_item, entry);
 486     if (!item->sample) return S_OK;
 487
 488     IMFSample_AddRef((sample = item->sample));
 489     sink_writer_release_pending_item(item);
 490
 491     writer->stats.dwNumOutstandingSinkSampleRequests--;
 492     stream->stats.dwNumOutstandingSinkSampleRequests--;
 493
 494     if (FAILED(hr = IMFSample_GetSampleTime(sample, &timestamp)))
 495     {
 496         IMFSample_Release(sample);
 497         writer->status = hr;
 498         return S_OK;
 499     }
 500
 501     writer->stats.llLastTimestampProcessed = timestamp;
 502     stream->stats.llLastTimestampProcessed = timestamp;
 503
 504     hr = IMFStreamSink_ProcessSample(stream->stream_sink, sample);
 505     IMFSample_Release(sample);
 506     if (FAILED(hr)) return hr;
 507
 508     LIST_FOR_EACH_ENTRY_SAFE(item, next, &stream->queue, struct pending_item, entry)
 509     {
 510         if (item->sample) break;
 511         sink_writer_place_marker(writer, stream, item->marker_type, item->timestamp, item->user_context);
 512         sink_writer_release_pending_item(item);
 513     }
 514
 515     return hr;
 516 }
 517
 518 static HRESULT sink_writer_encode_sample(struct sink_writer *writer, struct stream *stream, IMFSample *sample)
 519 {
 520     struct pending_item *item;
 521
 522     /* FIXME: call the encoder, queue its output */
 523
 524     if (!(item = calloc(1, sizeof(*item))))
 525         return E_OUTOFMEMORY;
 526
 527     item->sample = sample;
 528     IMFSample_AddRef(item->sample);
 529     list_add_tail(&stream->queue, &item->entry);
 530
 531     return S_OK;
 532 }
 533
 534 static HRESULT sink_writer_write_sample(struct sink_writer *writer, struct stream *stream, IMFSample *sample)
 535 {
 536     LONGLONG timestamp;
 537     DWORD length;
 538     HRESULT hr;
 539
 540     if (FAILED(hr = sink_writer_get_buffer_length(sample, &timestamp, &length))) return hr;
 541
 542     stream->stats.llLastTimestampReceived = timestamp;
 543     stream->stats.qwNumSamplesReceived++;
 544     stream->stats.dwByteCountQueued += length;
 545
 546     writer->stats.llLastTimestampReceived = timestamp;
 547     writer->stats.qwNumSamplesReceived++;
 548     writer->stats.dwByteCountQueued += length;
 549
 550     if (FAILED(hr = sink_writer_encode_sample(writer, stream, sample))) return hr;
 551
 552     if (stream->stats.dwNumOutstandingSinkSampleRequests)
 553         hr = sink_writer_process_sample(writer, stream);
 554
 555     return hr;
 556 }
 557
 558 static HRESULT WINAPI sink_writer_WriteSample(IMFSinkWriter *iface, DWORD index, IMFSample *sample)
 559 {
 560     struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
 561     struct stream *stream;
 562     LONGLONG timestamp;
 563     DWORD length;
 564     HRESULT hr;
 565
 566     TRACE("%p, %lu, %p.\n", iface, index, sample);
 567
 568     if (!sample)
 569         return E_INVALIDARG;
 570
 571     EnterCriticalSection(&writer->cs);
 572
 573     if (writer->state != SINK_WRITER_STATE_WRITING)
 574         hr = MF_E_INVALIDREQUEST;
 575     else if (!(stream = sink_writer_get_stream(writer, index)))
 576     {
 577         hr = MF_E_INVALIDSTREAMNUMBER;
 578     }
 579     else if (SUCCEEDED(hr = sink_writer_get_buffer_length(sample, &timestamp, &length)))
 580     {
 581         stream->stats.llLastTimestampReceived = timestamp;
 582         stream->stats.qwNumSamplesReceived++;
 583         stream->stats.dwByteCountQueued += length;
 584
 585         writer->stats.llLastTimestampReceived = timestamp;
 586         writer->stats.qwNumSamplesReceived++;
 587         writer->stats.dwByteCountQueued += length;
 588
 589         hr = sink_writer_write_sample(writer, stream, sample);
 590     }
 591
 592     LeaveCriticalSection(&writer->cs);
 593
 594     return hr;
 595 }
 596
 597 static HRESULT WINAPI sink_writer_SendStreamTick(IMFSinkWriter *iface, DWORD index, LONGLONG timestamp)
 598 {
 599     struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
 600     HRESULT hr;
 601
 602     TRACE("%p, %lu, %s.\n", iface, index, wine_dbgstr_longlong(timestamp));
 603
 604     EnterCriticalSection(&writer->cs);
 605
 606     if (writer->state != SINK_WRITER_STATE_WRITING)
 607         hr = MF_E_INVALIDREQUEST;
 608     else
 609         hr = sink_writer_send_stream_tick(writer, index, timestamp);
 610
 611     LeaveCriticalSection(&writer->cs);
 612
 613     return hr;
 614 }
 615
 616 static HRESULT WINAPI sink_writer_PlaceMarker(IMFSinkWriter *iface, DWORD index, void *user_context)
 617 {
 618     struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
 619     struct stream *stream;
 620     HRESULT hr;
 621
 622     TRACE("%p, %lu, %p.\n", iface, index, user_context);
 623
 624     EnterCriticalSection(&writer->cs);
 625
 626     if (!writer->callback)
 627         hr = MF_E_INVALIDREQUEST;
 628     else if (writer->state != SINK_WRITER_STATE_WRITING)
 629         hr = MF_E_INVALIDREQUEST;
 630     else if (!(stream = sink_writer_get_stream(writer, index)))
 631         hr = MF_E_INVALIDSTREAMNUMBER;
 632     else
 633         hr = sink_writer_queue_marker(writer, stream, MFSTREAMSINK_MARKER_DEFAULT, 0, user_context);
 634
 635     LeaveCriticalSection(&writer->cs);
 636
 637     return hr;
 638 }
 639
 640 static HRESULT WINAPI sink_writer_NotifyEndOfSegment(IMFSinkWriter *iface, DWORD index)
 641 {
 642     struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
 643     HRESULT hr = S_OK;
 644     unsigned int i;
 645
 646     TRACE("%p, %lu.\n", iface, index);
 647
 648     EnterCriticalSection(&writer->cs);
 649
 650     if (writer->state != SINK_WRITER_STATE_WRITING)
 651         hr = MF_E_INVALIDREQUEST;
 652     else if (index == MF_SINK_WRITER_ALL_STREAMS)
 653     {
 654         for (i = 0; i < writer->streams.count; ++i)
 655         {
 656             if (FAILED(hr = sink_writer_notify_end_of_segment(writer, index)))
 657             {
 658                 WARN("Failed to place a marker for stream %u.\n", i);
 659                 break;
 660             }
 661         }
 662     }
 663     else
 664         hr = sink_writer_notify_end_of_segment(writer, index);
 665
 666     LeaveCriticalSection(&writer->cs);
 667
 668     return hr;
 669 }
 670
 671 static HRESULT WINAPI sink_writer_Flush(IMFSinkWriter *iface, DWORD index)
 672 {
 673     struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
 674     HRESULT hr = S_OK;
 675     unsigned int i;
 676
 677     TRACE("%p, %lu.\n", iface, index);
 678
 679     EnterCriticalSection(&writer->cs);
 680
 681     if (writer->state != SINK_WRITER_STATE_WRITING)
 682         hr = MF_E_INVALIDREQUEST;
 683     else if (index == MF_SINK_WRITER_ALL_STREAMS)
 684     {
 685         for (i = 0; i < writer->streams.count; ++i)
 686         {
 687             if (FAILED(hr = sink_writer_flush(writer, i)))
 688             {
 689                 WARN("Failed to flush stream %u.\n", i);
 690                 break;
 691             }
 692         }
 693     }
 694     else
 695         hr = sink_writer_flush(writer, index);
 696
 697     LeaveCriticalSection(&writer->cs);
 698
 699     return hr;
 700 }
 701
 702 static HRESULT WINAPI sink_writer_Finalize(IMFSinkWriter *iface)
 703 {
 704     FIXME("%p.\n", iface);
 705
 706     return E_NOTIMPL;
 707 }
 708
 709 static HRESULT sink_writer_get_service(void *object, REFGUID service, REFIID riid, void **ret)
 710 {
 711     IUnknown *iface = object;
 712     IMFGetService *gs;
 713     HRESULT hr;
 714
 715     if (!iface) return MF_E_UNSUPPORTED_SERVICE;
 716
 717     if (IsEqualGUID(service, &GUID_NULL))
 718         return IUnknown_QueryInterface(iface, riid, ret);
 719
 720     if (FAILED(hr = IUnknown_QueryInterface(iface, &IID_IMFGetService, (void **)&gs)))
 721         return hr;
 722
 723     hr = IMFGetService_GetService(gs, service, riid, ret);
 724     IMFGetService_Release(gs);
 725     return hr;
 726 }
 727
 728 static HRESULT WINAPI sink_writer_GetServiceForStream(IMFSinkWriter *iface, DWORD index, REFGUID service,
 729         REFIID riid, void **object)
 730 {
 731     struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
 732     HRESULT hr = E_UNEXPECTED;
 733     struct stream *stream;
 734
 735     TRACE("%p, %lu, %s, %s, %p.\n", iface, index, debugstr_guid(service), debugstr_guid(riid), object);
 736
 737     EnterCriticalSection(&writer->cs);
 738
 739     if (index == MF_SINK_WRITER_MEDIASINK)
 740         hr = sink_writer_get_service(writer->sink, service, riid, object);
 741     else if ((stream = sink_writer_get_stream(writer, index)))
 742     {
 743         if (stream->encoder)
 744             hr = sink_writer_get_service(stream->encoder, service, riid, object);
 745         if (FAILED(hr))
 746             hr = sink_writer_get_service(stream->stream_sink, service, riid, object);
 747     }
 748     else
 749         hr = MF_E_INVALIDSTREAMNUMBER;
 750
 751     LeaveCriticalSection(&writer->cs);
 752
 753     return hr;
 754 }
 755
 756 static HRESULT WINAPI sink_writer_GetStatistics(IMFSinkWriter *iface, DWORD index, MF_SINK_WRITER_STATISTICS *stats)
 757 {
 758     struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
 759     struct stream *stream;
 760     HRESULT hr = S_OK;
 761
 762     TRACE("%p, %lu, %p.\n", iface, index, stats);
 763
 764     if (!stats)
 765         return E_POINTER;
 766
 767     if (stats->cb != sizeof(*stats))
 768         return E_INVALIDARG;
 769
 770     EnterCriticalSection(&writer->cs);
 771
 772     if (FAILED(writer->status))
 773         hr = writer->status;
 774     else if (index == MF_SINK_WRITER_ALL_STREAMS)
 775         *stats = writer->stats;
 776     else if ((stream = sink_writer_get_stream(writer, index)))
 777         *stats = stream->stats;
 778     else
 779         hr = MF_E_INVALIDSTREAMNUMBER;
 780
 781     LeaveCriticalSection(&writer->cs);
 782
 783     return hr;
 784 }
 785
 786 static const IMFSinkWriterVtbl sink_writer_vtbl =
 787 {
 788     sink_writer_QueryInterface,
 789     sink_writer_AddRef,
 790     sink_writer_Release,
 791     sink_writer_AddStream,
 792     sink_writer_SetInputMediaType,
 793     sink_writer_BeginWriting,
 794     sink_writer_WriteSample,
 795     sink_writer_SendStreamTick,
 796     sink_writer_PlaceMarker,
 797     sink_writer_NotifyEndOfSegment,
 798     sink_writer_Flush,
 799     sink_writer_Finalize,
 800     sink_writer_GetServiceForStream,
 801     sink_writer_GetStatistics,
 802 };
 803
 804 static HRESULT WINAPI sink_writer_callback_QueryInterface(IMFAsyncCallback *iface,
 805         REFIID riid, void **out)
 806 {
 807     if (IsEqualIID(riid, &IID_IMFAsyncCallback) ||
 808             IsEqualIID(riid, &IID_IUnknown))
 809     {
 810         *out = iface;
 811         IMFAsyncCallback_AddRef(iface);
 812         return S_OK;
 813     }
 814
 815     WARN("Unsupported %s.\n", debugstr_guid(riid));
 816     *out = NULL;
 817     return E_NOINTERFACE;
 818 }
 819
 820 static ULONG WINAPI sink_writer_events_callback_AddRef(IMFAsyncCallback *iface)
 821 {
 822     struct sink_writer *writer = impl_from_events_callback_IMFAsyncCallback(iface);
 823     return IMFSinkWriter_AddRef(&writer->IMFSinkWriter_iface);
 824 }
 825
 826 static ULONG WINAPI sink_writer_events_callback_Release(IMFAsyncCallback *iface)
 827 {
 828     struct sink_writer *writer = impl_from_events_callback_IMFAsyncCallback(iface);
 829     return IMFSinkWriter_Release(&writer->IMFSinkWriter_iface);
 830 }
 831
 832 static HRESULT WINAPI sink_writer_callback_GetParameters(IMFAsyncCallback *iface,
 833         DWORD *flags, DWORD *queue)
 834 {
 835     return E_NOTIMPL;
 836 }
 837
 838 static struct stream *sink_writer_get_stream_for_stream_sink(struct sink_writer *writer,
 839         IMFStreamSink *stream_sink, DWORD *index)
 840 {
 841     unsigned int i;
 842
 843     for (i = 0; i < writer->streams.count; ++i)
 844     {
 845         if (writer->streams.items[i].stream_sink == stream_sink)
 846         {
 847             *index = i;
 848             return &writer->streams.items[i];
 849         }
 850     }
 851
 852     return NULL;
 853 }
 854
 855 static HRESULT WINAPI sink_writer_events_callback_Invoke(IMFAsyncCallback *iface, IMFAsyncResult *result)
 856 {
 857     struct sink_writer *writer = impl_from_events_callback_IMFAsyncCallback(iface);
 858     struct marker_context *context;
 859     IMFStreamSink *stream_sink;
 860     MediaEventType event_type;
 861     struct stream *stream;
 862     IMFMediaEvent *event;
 863     LONGLONG timestamp;
 864     HRESULT status, hr;
 865     PROPVARIANT value;
 866     DWORD index;
 867
 868     TRACE("%p, %p.\n", iface, result);
 869
 870     stream_sink = (IMFStreamSink *)IMFAsyncResult_GetStateNoAddRef(result);
 871
 872     if (FAILED(hr = IMFStreamSink_EndGetEvent(stream_sink, result, &event)))
 873         return hr;
 874
 875     IMFMediaEvent_GetType(event, &event_type);
 876     IMFMediaEvent_GetStatus(event, &status);
 877
 878     TRACE("Got event %lu.\n", event_type);
 879
 880     PropVariantInit(&value);
 881
 882     EnterCriticalSection(&writer->cs);
 883
 884     if (writer->status == S_OK && FAILED(status))
 885         writer->status = status;
 886
 887     if (writer->status == S_OK && (stream = sink_writer_get_stream_for_stream_sink(writer, stream_sink, &index)))
 888     {
 889         switch (event_type)
 890         {
 891             case MEStreamSinkRequestSample:
 892
 893                 timestamp = MFGetSystemTime();
 894
 895                 writer->stats.llLastSinkSampleRequest = timestamp;
 896                 writer->stats.dwNumOutstandingSinkSampleRequests++;
 897                 stream->stats.llLastSinkSampleRequest = timestamp;
 898                 stream->stats.dwNumOutstandingSinkSampleRequests++;
 899
 900                 sink_writer_process_sample(writer, stream);
 901
 902                 break;
 903
 904             case MEStreamSinkMarker:
 905                 if (FAILED(hr = IMFMediaEvent_GetValue(event, &value))) break;
 906                 if (value.vt != VT_UNKNOWN || !(context = unsafe_impl_from_marker_context_IUnknown(value.punkVal))) break;
 907
 908                 /* This relies on the fact that default marker type is only used for PlaceMarker(). */
 909                 if (context->marker_type == MFSTREAMSINK_MARKER_DEFAULT)
 910                     IMFSinkWriterCallback_OnMarker(writer->callback, index, context->user_context);
 911
 912                 break;
 913
 914             default:
 915                 ;
 916         }
 917     }
 918
 919     LeaveCriticalSection(&writer->cs);
 920
 921     PropVariantClear(&value);
 922
 923     IMFMediaEvent_Release(event);
 924
 925     IMFStreamSink_BeginGetEvent(stream_sink, iface, (IUnknown *)stream_sink);
 926
 927     return S_OK;
 928 }
 929
 930 static const IMFAsyncCallbackVtbl sink_writer_events_callback_vtbl =
 931 {
 932     sink_writer_callback_QueryInterface,
 933     sink_writer_events_callback_AddRef,
 934     sink_writer_events_callback_Release,
 935     sink_writer_callback_GetParameters,
 936     sink_writer_events_callback_Invoke,
 937 };
 938
 939 static HRESULT sink_writer_initialize_existing_streams(struct sink_writer *writer, IMFMediaSink *sink)
 940 {
 941     IMFStreamSink *stream_sink;
 942     DWORD count = 0, i, index;
 943     HRESULT hr;
 944
 945     if (FAILED(hr = IMFMediaSink_GetStreamSinkCount(sink, &count))) return hr;
 946     if (!count) return S_OK;
 947
 948     for (i = 0; i < count; ++i)
 949     {
 950         if (FAILED(hr = IMFMediaSink_GetStreamSinkByIndex(sink, i, &stream_sink))) break;
 951         hr = sink_writer_add_stream(writer, stream_sink, &index);
 952         IMFStreamSink_Release(stream_sink);
 953         if (FAILED(hr)) break;
 954     }
 955
 956     return hr;
 957 }
 958
 959 HRESULT create_sink_writer_from_sink(IMFMediaSink *sink, IMFAttributes *attributes,
 960         REFIID riid, void **out)
 961 {
 962     struct sink_writer *object;
 963     HRESULT hr;
 964
 965     *out = NULL;
 966
 967     if (!sink)
 968         return E_INVALIDARG;
 969
 970     if (!(object = calloc(1, sizeof(*object))))
 971         return E_OUTOFMEMORY;
 972
 973     object->IMFSinkWriter_iface.lpVtbl = &sink_writer_vtbl;
 974     object->events_callback.lpVtbl = &sink_writer_events_callback_vtbl;
 975     object->refcount = 1;
 976     object->sink = sink;
 977     IMFMediaSink_AddRef(sink);
 978     object->stats.cb = sizeof(object->stats);
 979     InitializeCriticalSection(&object->cs);
 980
 981     if (attributes)
 982     {
 983         IMFAttributes_GetUnknown(attributes, &MF_SINK_WRITER_ASYNC_CALLBACK,
 984                 &IID_IMFSinkWriterCallback, (void **)&object->callback);
 985     }
 986
 987     if (FAILED(hr = sink_writer_initialize_existing_streams(object, sink)))
 988     {
 989         IMFSinkWriter_Release(&object->IMFSinkWriter_iface);
 990         return hr;
 991     }
 992
 993     hr = IMFSinkWriter_QueryInterface(&object->IMFSinkWriter_iface, riid, out);
 994     IMFSinkWriter_Release(&object->IMFSinkWriter_iface);
 995     return hr;
 996 }
 997
 998 /***********************************************************************
 999  *      MFCreateSinkWriterFromMediaSink (mfreadwrite.@)
1000  */
1001 HRESULT WINAPI MFCreateSinkWriterFromMediaSink(IMFMediaSink *sink, IMFAttributes *attributes, IMFSinkWriter **writer)
1002 {
1003     TRACE("%p, %p, %p.\n", sink, attributes, writer);
1004
1005     if (!writer)
1006         return E_INVALIDARG;
1007
1008     return create_sink_writer_from_sink(sink, attributes, &IID_IMFSinkWriter, (void **)writer);
1009 }
1010
1011 static HRESULT sink_writer_get_sink_factory_class(const WCHAR *url, IMFAttributes *attributes, CLSID *clsid)
1012 {
1013     static const struct extension_map
1014     {
1015         const WCHAR *ext;
1016         const GUID *guid;
1017     } ext_map[] =
1018     {
1019         { L".mp4", &MFTranscodeContainerType_MPEG4 },
1020         { L".mp3", &MFTranscodeContainerType_MP3 },
1021         { L".wav", &MFTranscodeContainerType_WAVE },
1022         { L".avi", &MFTranscodeContainerType_AVI },
1023     };
1024     static const struct
1025     {
1026         const GUID *container;
1027         const CLSID *clsid;
1028     } class_map[] =
1029     {
1030         { &MFTranscodeContainerType_MPEG4, &CLSID_MFMPEG4SinkClassFactory },
1031         { &MFTranscodeContainerType_MP3, &CLSID_MFMP3SinkClassFactory },
1032         { &MFTranscodeContainerType_WAVE, &CLSID_MFWAVESinkClassFactory },
1033         { &MFTranscodeContainerType_AVI, &CLSID_MFAVISinkClassFactory },
1034     };
1035     const WCHAR *extension;
1036     GUID container;
1037     unsigned int i;
1038
1039     if (url)
1040     {
1041         if (!attributes || FAILED(IMFAttributes_GetGUID(attributes, &MF_TRANSCODE_CONTAINERTYPE, &container)))
1042         {
1043             const struct extension_map *map = NULL;
1044
1045             if (FAILED(PathCchFindExtension(url, PATHCCH_MAX_CCH, &extension))) return E_INVALIDARG;
1046             if (!extension || !*extension) return E_INVALIDARG;
1047
1048             for (i = 0; i < ARRAY_SIZE(ext_map); ++i)
1049             {
1050                 map = &ext_map[i];
1051
1052                 if (!wcsicmp(map->ext, extension))
1053                     break;
1054             }
1055
1056             if (!map)
1057             {
1058                 WARN("Couldn't find container type for extension %s.\n", debugstr_w(extension));
1059                 return E_INVALIDARG;
1060             }
1061
1062             container = *map->guid;
1063         }
1064     }
1065     else
1066     {
1067         if (!attributes) return E_INVALIDARG;
1068         if (FAILED(IMFAttributes_GetGUID(attributes, &MF_TRANSCODE_CONTAINERTYPE, &container))) return E_INVALIDARG;
1069     }
1070
1071     for (i = 0; i < ARRAY_SIZE(class_map); ++i)
1072     {
1073         if (IsEqualGUID(&container, &class_map[i].container))
1074         {
1075             *clsid = *class_map[i].clsid;
1076             return S_OK;
1077         }
1078     }
1079
1080     WARN("Couldn't find factory class for container %s.\n", debugstr_guid(&container));
1081     return E_INVALIDARG;
1082 }
1083
1084 HRESULT create_sink_writer_from_url(const WCHAR *url, IMFByteStream *bytestream, IMFAttributes *attributes,
1085         REFIID riid, void **out)
1086 {
1087     IMFSinkClassFactory *factory;
1088     IMFMediaSink *sink;
1089     CLSID clsid;
1090     HRESULT hr;
1091
1092     *out = NULL;
1093
1094     if (!url && !bytestream)
1095         return E_INVALIDARG;
1096
1097     if (FAILED(hr = sink_writer_get_sink_factory_class(url, attributes, &clsid))) return hr;
1098
1099     if (FAILED(hr = CoCreateInstance(&clsid, NULL, CLSCTX_INPROC_SERVER, &IID_IMFSinkClassFactory, (void **)&factory)))
1100     {
1101         WARN("Failed to create a sink factory, hr %#lx.\n", hr);
1102         return hr;
1103     }
1104
1105     if (bytestream)
1106         IMFByteStream_AddRef(bytestream);
1107     else if (FAILED(hr = MFCreateFile(MF_ACCESSMODE_WRITE, MF_OPENMODE_DELETE_IF_EXIST, 0, url, &bytestream)))
1108     {
1109         WARN("Failed to create output file stream, hr %#lx.\n", hr);
1110         IMFSinkClassFactory_Release(factory);
1111         return hr;
1112     }
1113
1114     hr = IMFSinkClassFactory_CreateMediaSink(factory, bytestream, NULL, NULL, &sink);
1115     IMFSinkClassFactory_Release(factory);
1116     IMFByteStream_Release(bytestream);
1117     if (FAILED(hr))
1118     {
1119         WARN("Failed to create a sink, hr %#lx.\n", hr);
1120         return hr;
1121     }
1122
1123     hr = create_sink_writer_from_sink(sink, attributes, riid, out);
1124     IMFMediaSink_Release(sink);
1125
1126     return hr;
1127 }
1128
1129 /***********************************************************************
1130  *      MFCreateSinkWriterFromURL (mfreadwrite.@)
1131  */
1132 HRESULT WINAPI MFCreateSinkWriterFromURL(const WCHAR *url, IMFByteStream *bytestream, IMFAttributes *attributes,
1133         IMFSinkWriter **writer)
1134 {
1135     TRACE("%s, %p, %p, %p.\n", debugstr_w(url), bytestream, attributes, writer);
1136
1137     if (!writer)
1138         return E_INVALIDARG;
1139
1140     return create_sink_writer_from_url(url, bytestream, attributes, &IID_IMFSinkWriter, (void **)writer);
1141 }