1 files changed, 130 insertions, 0 deletions
diff --git a/dom/media/platforms/ffmpeg/ffvpx/FFVPXRuntimeLinker.cpp b/dom/media/platforms/ffmpeg/ffvpx/FFVPXRuntimeLinker.cpp
new file mode 100644
index 0000000000..44cc4ec4bd
--- /dev/null
+++ b/dom/media/platforms/ffmpeg/ffvpx/FFVPXRuntimeLinker.cpp
@@ -0,0 +1,130 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* 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 "FFVPXRuntimeLinker.h"
+#include "FFmpegLibWrapper.h"
+#include "FFmpegLog.h"
+#include "BinaryPath.h"
+#include "mozilla/FileUtils.h"
+#include "nsLocalFile.h"
+#include "prmem.h"
+#include "prlink.h"
+#ifdef XP_WIN
+#  include <windows.h>
+#endif
+
+namespace mozilla {
+
+template <int V>
+class FFmpegDecoderModule {
+ public:
+  static already_AddRefed<PlatformDecoderModule> Create(FFmpegLibWrapper*);
+};
+
+static FFmpegLibWrapper sFFVPXLib;
+
+FFVPXRuntimeLinker::LinkStatus FFVPXRuntimeLinker::sLinkStatus =
+    LinkStatus_INIT;
+
+static PRLibrary* MozAVLink(nsIFile* aFile) {
+  PRLibSpec lspec;
+  PathString path = aFile->NativePath();
+#ifdef XP_WIN
+  lspec.type = PR_LibSpec_PathnameU;
+  lspec.value.pathname_u = path.get();
+#else
+  lspec.type = PR_LibSpec_Pathname;
+  lspec.value.pathname = path.get();
+#endif
+#ifdef MOZ_WIDGET_ANDROID
+  PRLibrary* lib = PR_LoadLibraryWithFlags(lspec, PR_LD_NOW | PR_LD_GLOBAL);
+#else
+  PRLibrary* lib = PR_LoadLibraryWithFlags(lspec, PR_LD_NOW | PR_LD_LOCAL);
+#endif
+  if (!lib) {
+    FFMPEG_LOG("unable to load library %s", aFile->HumanReadablePath().get());
+  }
+  return lib;
+}
+
+/* static */
+bool FFVPXRuntimeLinker::Init() {
+  if (sLinkStatus) {
+    return sLinkStatus == LinkStatus_SUCCEEDED;
+  }
+
+  MOZ_ASSERT(NS_IsMainThread());
+  sLinkStatus = LinkStatus_FAILED;
+
+#ifdef MOZ_WAYLAND
+  sFFVPXLib.LinkVAAPILibs();
+#endif
+
+  nsCOMPtr<nsIFile> libFile;
+  if (NS_FAILED(mozilla::BinaryPath::GetFile(getter_AddRefs(libFile)))) {
+    return false;
+  }
+
+#ifdef XP_DARWIN
+  if (!XRE_IsParentProcess() &&
+      (XRE_GetChildProcBinPathType(XRE_GetProcessType()) ==
+       BinPathType::PluginContainer)) {
+    // On macOS, PluginContainer processes have their binary in a
+    // plugin-container.app/Content/MacOS/ directory.
+    nsCOMPtr<nsIFile> parentDir1, parentDir2;
+    if (NS_FAILED(libFile->GetParent(getter_AddRefs(parentDir1)))) {
+      return false;
+    }
+    if (NS_FAILED(parentDir1->GetParent(getter_AddRefs(parentDir2)))) {
+      return false;
+    }
+    if (NS_FAILED(parentDir2->GetParent(getter_AddRefs(libFile)))) {
+      return false;
+    }
+  }
+#endif
+
+  if (NS_FAILED(libFile->SetNativeLeafName(MOZ_DLL_PREFIX
+                                           "mozavutil" MOZ_DLL_SUFFIX ""_ns))) {
+    return false;
+  }
+  sFFVPXLib.mAVUtilLib = MozAVLink(libFile);
+
+  if (NS_FAILED(libFile->SetNativeLeafName(
+          MOZ_DLL_PREFIX "mozavcodec" MOZ_DLL_SUFFIX ""_ns))) {
+    return false;
+  }
+  sFFVPXLib.mAVCodecLib = MozAVLink(libFile);
+  if (sFFVPXLib.Link() == FFmpegLibWrapper::LinkResult::Success) {
+    sLinkStatus = LinkStatus_SUCCEEDED;
+    return true;
+  }
+  return false;
+}
+
+/* static */
+already_AddRefed<PlatformDecoderModule> FFVPXRuntimeLinker::Create() {
+  if (!Init()) {
+    return nullptr;
+  }
+  return FFmpegDecoderModule<FFVPX_VERSION>::Create(&sFFVPXLib);
+}
+
+/* static */
+void FFVPXRuntimeLinker::GetRDFTFuncs(FFmpegRDFTFuncs* aOutFuncs) {
+  MOZ_ASSERT(sLinkStatus != LinkStatus_INIT);
+  if (sFFVPXLib.av_rdft_init && sFFVPXLib.av_rdft_calc &&
+      sFFVPXLib.av_rdft_end) {
+    aOutFuncs->init = sFFVPXLib.av_rdft_init;
+    aOutFuncs->calc = sFFVPXLib.av_rdft_calc;
+    aOutFuncs->end = sFFVPXLib.av_rdft_end;
+  } else {
+    NS_WARNING("RDFT functions expected but not found");
+    *aOutFuncs = FFmpegRDFTFuncs();  // zero
+  }
+}
+
+}  // namespace mozilla