/* 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 https://mozilla.org/MPL/2.0/. */

#include "mozilla/LoaderAPIInterfaces.h"

#include "freestanding/CheckForCaller.h"
#include "freestanding/LoaderPrivateAPI.h"

namespace mozilla {

extern "C" MOZ_EXPORT nt::LoaderAPI* GetNtLoaderAPI(
    nt::LoaderObserver* aNewObserver) {
  // Make sure the caller is inside mozglue.dll - we don't want to allow
  // external access to this function, as it contains details about
  // the SharedSection which is used to sandbox future child processes.
  const bool isCallerMozglue =
      CheckForAddress(RETURN_ADDRESS(), L"mozglue.dll");
  MOZ_ASSERT(isCallerMozglue);
  if (!isCallerMozglue) {
    return nullptr;
  }

  freestanding::EnsureInitialized();
  freestanding::LoaderPrivateAPI& api = freestanding::gLoaderPrivateAPI;
  api.SetObserver(aNewObserver);

  return &api;
}

}  // namespace mozilla