1 files changed, 369 insertions, 0 deletions
diff --git a/dom/media/directshow/DirectShowUtils.cpp b/dom/media/directshow/DirectShowUtils.cpp
new file mode 100644
index 000000000..b2afa7528
--- /dev/null
+++ b/dom/media/directshow/DirectShowUtils.cpp
@@ -0,0 +1,369 @@
+/* vim:set ts=2 sw=2 sts=2 et cindent: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "DirectShowUtils.h"
+#include "dmodshow.h"
+#include "wmcodecdsp.h"
+#include "dmoreg.h"
+#include "mozilla/ArrayUtils.h"
+#include "mozilla/RefPtr.h"
+#include "nsPrintfCString.h"
+
+#define WARN(...) NS_WARNING(nsPrintfCString(__VA_ARGS__).get())
+
+namespace mozilla {
+
+// Create a table which maps GUIDs to a string representation of the GUID.
+// This is useful for debugging purposes, for logging the GUIDs of media types.
+// This is only available when logging is enabled, i.e. not in release builds.
+struct GuidToName {
+  const char* name;
+  const GUID guid;
+};
+
+#pragma push_macro("OUR_GUID_ENTRY")
+#undef OUR_GUID_ENTRY
+#define OUR_GUID_ENTRY(name, l, w1, w2, b1, b2, b3, b4, b5, b6, b7, b8) \
+  { #name, {l, w1, w2, {b1, b2, b3, b4, b5, b6, b7, b8}} },
+
+static const GuidToName GuidToNameTable[] = {
+#include <uuids.h>
+};
+
+#pragma pop_macro("OUR_GUID_ENTRY")
+
+const char*
+GetDirectShowGuidName(const GUID& aGuid)
+{
+  const size_t len = ArrayLength(GuidToNameTable);
+  for (unsigned i = 0; i < len; i++) {
+    if (IsEqualGUID(aGuid, GuidToNameTable[i].guid)) {
+      return GuidToNameTable[i].name;
+    }
+  }
+  return "Unknown";
+}
+
+void
+RemoveGraphFromRunningObjectTable(DWORD aRotRegister)
+{
+  RefPtr<IRunningObjectTable> runningObjectTable;
+  if (SUCCEEDED(GetRunningObjectTable(0, getter_AddRefs(runningObjectTable)))) {
+    runningObjectTable->Revoke(aRotRegister);
+  }
+}
+
+HRESULT
+AddGraphToRunningObjectTable(IUnknown *aUnkGraph, DWORD *aOutRotRegister)
+{
+  HRESULT hr;
+
+  RefPtr<IMoniker> moniker;
+  RefPtr<IRunningObjectTable> runningObjectTable;
+
+  hr = GetRunningObjectTable(0, getter_AddRefs(runningObjectTable));
+  NS_ENSURE_TRUE(SUCCEEDED(hr), hr);
+
+  const size_t STRING_LENGTH = 256;
+  WCHAR wsz[STRING_LENGTH];
+
+  StringCchPrintfW(wsz,
+                   STRING_LENGTH,
+                   L"FilterGraph %08x pid %08x",
+                   (DWORD_PTR)aUnkGraph,
+                   GetCurrentProcessId());
+
+  hr = CreateItemMoniker(L"!", wsz, getter_AddRefs(moniker));
+  NS_ENSURE_TRUE(SUCCEEDED(hr), hr);
+
+  hr = runningObjectTable->Register(ROTFLAGS_REGISTRATIONKEEPSALIVE,
+                                    aUnkGraph,
+                                    moniker,
+                                    aOutRotRegister);
+  NS_ENSURE_TRUE(SUCCEEDED(hr), hr);
+
+  return S_OK;
+}
+
+const char*
+GetGraphNotifyString(long evCode)
+{
+#define CASE(x) case x: return #x
+  switch(evCode) {
+    CASE(EC_ACTIVATE); // A video window is being activated or deactivated.
+    CASE(EC_BANDWIDTHCHANGE); // Not supported.
+    CASE(EC_BUFFERING_DATA); // The graph is buffering data, or has stopped buffering data.
+    CASE(EC_BUILT); // Send by the Video Control when a graph has been built. Not forwarded to applications.
+    CASE(EC_CLOCK_CHANGED); // The reference clock has changed.
+    CASE(EC_CLOCK_UNSET); // The clock provider was disconnected.
+    CASE(EC_CODECAPI_EVENT); // Sent by an encoder to signal an encoding event.
+    CASE(EC_COMPLETE); // All data from a particular stream has been rendered.
+    CASE(EC_CONTENTPROPERTY_CHANGED); // Not supported.
+    CASE(EC_DEVICE_LOST); // A Plug and Play device was removed or has become available again.
+    CASE(EC_DISPLAY_CHANGED); // The display mode has changed.
+    CASE(EC_END_OF_SEGMENT); // The end of a segment has been reached.
+    CASE(EC_EOS_SOON); // Not supported.
+    CASE(EC_ERROR_STILLPLAYING); // An asynchronous command to run the graph has failed.
+    CASE(EC_ERRORABORT); // An operation was aborted because of an error.
+    CASE(EC_ERRORABORTEX); // An operation was aborted because of an error.
+    CASE(EC_EXTDEVICE_MODE_CHANGE); // Not supported.
+    CASE(EC_FILE_CLOSED); // The source file was closed because of an unexpected event.
+    CASE(EC_FULLSCREEN_LOST); // The video renderer is switching out of full-screen mode.
+    CASE(EC_GRAPH_CHANGED); // The filter graph has changed.
+    CASE(EC_LENGTH_CHANGED); // The length of a source has changed.
+    CASE(EC_LOADSTATUS); // Notifies the application of progress when opening a network file.
+    CASE(EC_MARKER_HIT); // Not supported.
+    CASE(EC_NEED_RESTART); // A filter is requesting that the graph be restarted.
+    CASE(EC_NEW_PIN); // Not supported.
+    CASE(EC_NOTIFY_WINDOW); // Notifies a filter of the video renderer's window.
+    CASE(EC_OLE_EVENT); // A filter is passing a text string to the application.
+    CASE(EC_OPENING_FILE); // The graph is opening a file, or has finished opening a file.
+    CASE(EC_PALETTE_CHANGED); // The video palette has changed.
+    CASE(EC_PAUSED); // A pause request has completed.
+    CASE(EC_PLEASE_REOPEN); // The source file has changed.
+    CASE(EC_PREPROCESS_COMPLETE); // Sent by the WM ASF Writer filter when it completes the pre-processing for multipass encoding.
+    CASE(EC_PROCESSING_LATENCY); // Indicates the amount of time that a component is taking to process each sample.
+    CASE(EC_QUALITY_CHANGE); // The graph is dropping samples, for quality control.
+    //CASE(EC_RENDER_FINISHED); // Not supported.
+    CASE(EC_REPAINT); // A video renderer requires a repaint.
+    CASE(EC_SAMPLE_LATENCY); // Specifies how far behind schedule a component is for processing samples.
+    //CASE(EC_SAMPLE_NEEDED); // Requests a new input sample from the Enhanced Video Renderer (EVR) filter.
+    CASE(EC_SCRUB_TIME); // Specifies the time stamp for the most recent frame step.
+    CASE(EC_SEGMENT_STARTED); // A new segment has started.
+    CASE(EC_SHUTTING_DOWN); // The filter graph is shutting down, prior to being destroyed.
+    CASE(EC_SNDDEV_IN_ERROR); // A device error has occurred in an audio capture filter.
+    CASE(EC_SNDDEV_OUT_ERROR); // A device error has occurred in an audio renderer filter.
+    CASE(EC_STARVATION); // A filter is not receiving enough data.
+    CASE(EC_STATE_CHANGE); // The filter graph has changed state.
+    CASE(EC_STATUS); // Contains two arbitrary status strings.
+    CASE(EC_STEP_COMPLETE); // A filter performing frame stepping has stepped the specified number of frames.
+    CASE(EC_STREAM_CONTROL_STARTED); // A stream-control start command has taken effect.
+    CASE(EC_STREAM_CONTROL_STOPPED); // A stream-control stop command has taken effect.
+    CASE(EC_STREAM_ERROR_STILLPLAYING); // An error has occurred in a stream. The stream is still playing.
+    CASE(EC_STREAM_ERROR_STOPPED); // A stream has stopped because of an error.
+    CASE(EC_TIMECODE_AVAILABLE); // Not supported.
+    CASE(EC_UNBUILT); // Send by the Video Control when a graph has been torn down. Not forwarded to applications.
+    CASE(EC_USERABORT); // The user has terminated playback.
+    CASE(EC_VIDEO_SIZE_CHANGED); // The native video size has changed.
+    CASE(EC_VIDEOFRAMEREADY); // A video frame is ready for display.
+    CASE(EC_VMR_RECONNECTION_FAILED); // Sent by the VMR-7 and the VMR-9 when it was unable to accept a dynamic format change request from the upstream decoder.
+    CASE(EC_VMR_RENDERDEVICE_SET); // Sent when the VMR has selected its rendering mechanism.
+    CASE(EC_VMR_SURFACE_FLIPPED); // Sent when the VMR-7's allocator presenter has called the DirectDraw Flip method on the surface being presented.
+    CASE(EC_WINDOW_DESTROYED); // The video renderer was destroyed or removed from the graph.
+    CASE(EC_WMT_EVENT); // Sent by the WM ASF Reader filter when it reads ASF files protected by digital rights management (DRM).
+    CASE(EC_WMT_INDEX_EVENT); // Sent when an application uses the WM ASF Writer to index Windows Media Video files.
+    CASE(S_OK); // Success.
+    CASE(VFW_S_AUDIO_NOT_RENDERED); // Partial success; the audio was not rendered.
+    CASE(VFW_S_DUPLICATE_NAME); // Success; the Filter Graph Manager modified a filter name to avoid duplication.
+    CASE(VFW_S_PARTIAL_RENDER); // Partial success; some of the streams in this movie are in an unsupported format.
+    CASE(VFW_S_VIDEO_NOT_RENDERED); // Partial success; the video was not rendered.
+    CASE(E_ABORT); // Operation aborted.
+    CASE(E_OUTOFMEMORY); // Insufficient memory.
+    CASE(E_POINTER); // Null pointer argument.
+    CASE(VFW_E_CANNOT_CONNECT); // No combination of intermediate filters could be found to make the connection.
+    CASE(VFW_E_CANNOT_RENDER); // No combination of filters could be found to render the stream.
+    CASE(VFW_E_NO_ACCEPTABLE_TYPES); // There is no common media type between these pins.
+    CASE(VFW_E_NOT_IN_GRAPH);
+
+    default:
+      return "Unknown Code";
+  };
+#undef CASE
+}
+
+HRESULT
+CreateAndAddFilter(IGraphBuilder* aGraph,
+                   REFGUID aFilterClsId,
+                   LPCWSTR aFilterName,
+                   IBaseFilter **aOutFilter)
+{
+  NS_ENSURE_TRUE(aGraph, E_POINTER);
+  NS_ENSURE_TRUE(aOutFilter, E_POINTER);
+  HRESULT hr;
+
+  RefPtr<IBaseFilter> filter;
+  hr = CoCreateInstance(aFilterClsId,
+                        nullptr,
+                        CLSCTX_INPROC_SERVER,
+                        IID_IBaseFilter,
+                        getter_AddRefs(filter));
+  if (FAILED(hr)) {
+    // Object probably not available on this system.
+    WARN("CoCreateInstance failed, hr=%x", hr);
+    return hr;
+  }
+
+  hr = aGraph->AddFilter(filter, aFilterName);
+  if (FAILED(hr)) {
+    WARN("AddFilter failed, hr=%x", hr);
+    return hr;
+  }
+
+  filter.forget(aOutFilter);
+
+  return S_OK;
+}
+
+HRESULT
+CreateMP3DMOWrapperFilter(IBaseFilter **aOutFilter)
+{
+  NS_ENSURE_TRUE(aOutFilter, E_POINTER);
+  HRESULT hr;
+
+  // Create the wrapper filter.
+  RefPtr<IBaseFilter> filter;
+  hr = CoCreateInstance(CLSID_DMOWrapperFilter,
+                        nullptr,
+                        CLSCTX_INPROC_SERVER,
+                        IID_IBaseFilter,
+                        getter_AddRefs(filter));
+  if (FAILED(hr)) {
+    WARN("CoCreateInstance failed, hr=%x", hr);
+    return hr;
+  }
+
+  // Query for IDMOWrapperFilter.
+  RefPtr<IDMOWrapperFilter> dmoWrapper;
+  hr = filter->QueryInterface(IID_IDMOWrapperFilter,
+                              getter_AddRefs(dmoWrapper));
+  if (FAILED(hr)) {
+    WARN("QueryInterface failed, hr=%x", hr);
+    return hr;
+  }
+
+  hr = dmoWrapper->Init(CLSID_CMP3DecMediaObject, DMOCATEGORY_AUDIO_DECODER);
+  if (FAILED(hr)) {
+    // Can't instantiate MP3 DMO. It doesn't exist on Windows XP, we're
+    // probably hitting that. Don't log warning to console, this is an
+    // expected error.
+    WARN("dmoWrapper Init failed, hr=%x", hr);
+    return hr;
+  }
+
+  filter.forget(aOutFilter);
+
+  return S_OK;
+}
+
+HRESULT
+AddMP3DMOWrapperFilter(IGraphBuilder* aGraph,
+                       IBaseFilter **aOutFilter)
+{
+  NS_ENSURE_TRUE(aGraph, E_POINTER);
+  NS_ENSURE_TRUE(aOutFilter, E_POINTER);
+  HRESULT hr;
+
+  // Create the wrapper filter.
+  RefPtr<IBaseFilter> filter;
+  hr = CreateMP3DMOWrapperFilter(getter_AddRefs(filter));
+  NS_ENSURE_TRUE(SUCCEEDED(hr), hr);
+
+  // Add the wrapper filter to graph.
+  hr = aGraph->AddFilter(filter, L"MP3 Decoder DMO");
+  if (FAILED(hr)) {
+    WARN("AddFilter failed, hr=%x", hr);
+    return hr;
+  }
+
+  filter.forget(aOutFilter);
+
+  return S_OK;
+}
+
+bool
+CanDecodeMP3UsingDirectShow()
+{
+  RefPtr<IBaseFilter> filter;
+
+  // Can we create the MP3 demuxer filter?
+  if (FAILED(CoCreateInstance(CLSID_MPEG1Splitter,
+                              nullptr,
+                              CLSCTX_INPROC_SERVER,
+                              IID_IBaseFilter,
+                              getter_AddRefs(filter)))) {
+    return false;
+  }
+
+  // Can we create either the WinXP MP3 decoder filter or the MP3 DMO decoder?
+  if (FAILED(CoCreateInstance(DirectShowReader::CLSID_MPEG_LAYER_3_DECODER_FILTER,
+                              nullptr,
+                              CLSCTX_INPROC_SERVER,
+                              IID_IBaseFilter,
+                              getter_AddRefs(filter))) &&
+      FAILED(CreateMP3DMOWrapperFilter(getter_AddRefs(filter)))) {
+    return false;
+  }
+
+  // Else, we can create all of the components we need. Assume
+  // DirectShow is going to work...
+  return true;
+}
+
+// Match a pin by pin direction and connection state.
+HRESULT
+MatchUnconnectedPin(IPin* aPin,
+                    PIN_DIRECTION aPinDir,
+                    bool *aOutMatches)
+{
+  NS_ENSURE_TRUE(aPin, E_POINTER);
+  NS_ENSURE_TRUE(aOutMatches, E_POINTER);
+
+  // Ensure the pin is unconnected.
+  RefPtr<IPin> peer;
+  HRESULT hr = aPin->ConnectedTo(getter_AddRefs(peer));
+  if (hr != VFW_E_NOT_CONNECTED) {
+    *aOutMatches = false;
+    return hr;
+  }
+
+  // Ensure the pin is of the specified direction.
+  PIN_DIRECTION pinDir;
+  hr = aPin->QueryDirection(&pinDir);
+  NS_ENSURE_TRUE(SUCCEEDED(hr), hr);
+
+  *aOutMatches = (pinDir == aPinDir);
+  return S_OK;
+}
+
+// Return the first unconnected input pin or output pin.
+already_AddRefed<IPin>
+GetUnconnectedPin(IBaseFilter* aFilter, PIN_DIRECTION aPinDir)
+{
+  RefPtr<IEnumPins> enumPins;
+
+  HRESULT hr = aFilter->EnumPins(getter_AddRefs(enumPins));
+  NS_ENSURE_TRUE(SUCCEEDED(hr), nullptr);
+
+  // Test each pin to see if it matches the direction we're looking for.
+  RefPtr<IPin> pin;
+  while (S_OK == enumPins->Next(1, getter_AddRefs(pin), nullptr)) {
+    bool matches = FALSE;
+    if (SUCCEEDED(MatchUnconnectedPin(pin, aPinDir, &matches)) &&
+        matches) {
+      return pin.forget();
+    }
+  }
+
+  return nullptr;
+}
+
+HRESULT
+ConnectFilters(IGraphBuilder* aGraph,
+               IBaseFilter* aOutputFilter,
+               IBaseFilter* aInputFilter)
+{
+  RefPtr<IPin> output = GetUnconnectedPin(aOutputFilter, PINDIR_OUTPUT);
+  NS_ENSURE_TRUE(output, E_FAIL);
+
+  RefPtr<IPin> input = GetUnconnectedPin(aInputFilter, PINDIR_INPUT);
+  NS_ENSURE_TRUE(output, E_FAIL);
+
+  return aGraph->Connect(output, input);
+}
+
+} // namespace mozilla
+
+// avoid redefined macro in unified build
+#undef WARN