/* 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/. */

#ifndef mozilla_windowsdllblocklist_h
#define mozilla_windowsdllblocklist_h

#if (defined(_MSC_VER) || defined(__MINGW32__)) && \
    (defined(_M_IX86) || defined(_M_X64) || defined(_M_ARM64))

#  include <windows.h>
#  include "mozilla/ProcessType.h"
#  include "mozilla/Types.h"
#  include <algorithm>

#  define HAS_DLL_BLOCKLIST

enum DllBlocklistInitFlags {
  eDllBlocklistInitFlagDefault = 0,
  eDllBlocklistInitFlagIsChildProcess = 1 << 0,
  eDllBlocklistInitFlagWasBootstrapped = 1 << 1,
  eDllBlocklistInitFlagIsUtilityProcess = 1 << 2,
  eDllBlocklistInitFlagIsSocketProcess = 1 << 3,
  eDllBlocklistInitFlagIsGPUProcess = 1 << 4,
  eDllBlocklistInitFlagIsGMPluginProcess = 1 << 5,
  eDllBlocklistInitFlagIsRDDProcess = 1 << 6,
};

inline void SetDllBlocklistProcessTypeFlags(uint32_t& aFlags,
                                            GeckoProcessType aProcessType) {
  if (aProcessType == GeckoProcessType_Utility) {
    aFlags |= eDllBlocklistInitFlagIsUtilityProcess;
  } else if (aProcessType == GeckoProcessType_Socket) {
    aFlags |= eDllBlocklistInitFlagIsSocketProcess;
  } else if (aProcessType == GeckoProcessType_GPU) {
    aFlags |= eDllBlocklistInitFlagIsGPUProcess;
  } else if (aProcessType == GeckoProcessType_GMPlugin) {
    aFlags |= eDllBlocklistInitFlagIsGMPluginProcess;
  } else if (aProcessType == GeckoProcessType_RDD) {
    aFlags |= eDllBlocklistInitFlagIsRDDProcess;
  }
}

// Only available from within firefox.exe
#  if !defined(IMPL_MFBT) && !defined(MOZILLA_INTERNAL_API)
extern uint32_t gBlocklistInitFlags;
#  endif  // !defined(IMPL_MFBT) && !defined(MOZILLA_INTERNAL_API)

MFBT_API void DllBlocklist_Initialize(
    uint32_t aInitFlags = eDllBlocklistInitFlagDefault);
MFBT_API void DllBlocklist_WriteNotes();
MFBT_API bool DllBlocklist_CheckStatus();

MFBT_API bool* DllBlocklist_GetBlocklistInitFailedPointer();
MFBT_API bool* DllBlocklist_GetUser32BeforeBlocklistPointer();
MFBT_API const char* DllBlocklist_GetBlocklistWriterData();

// This export intends to clean up after DllBlocklist_Initialize().
// It's disabled in release builds for performance and to limit callers' ability
// to interfere with dll blocking.
#  ifdef DEBUG
MFBT_API void DllBlocklist_Shutdown();
#  endif  // DEBUG

namespace mozilla {
namespace glue {
namespace detail {
// Forward declaration
class DllServicesBase;

template <size_t N>
class WritableBuffer {
  char mBuffer[N];
  size_t mLen;

  size_t Available() const { return sizeof(mBuffer) - mLen; }

 public:
  constexpr WritableBuffer() : mBuffer{0}, mLen(0) {}

  void Write(const char* aData, size_t aLen) {
    size_t writable_len = std::min(aLen, Available());
    memcpy(mBuffer + mLen, aData, writable_len);
    mLen += writable_len;
  }

  size_t Length() const { return mLen; }
  const char* Data() const { return mBuffer; }
};
}  // namespace detail
}  // namespace glue
}  // namespace mozilla

MFBT_API void DllBlocklist_SetFullDllServices(
    mozilla::glue::detail::DllServicesBase* aSvc);
MFBT_API void DllBlocklist_SetBasicDllServices(
    mozilla::glue::detail::DllServicesBase* aSvc);

#endif  // defined(_MSC_VER) && (defined(_M_IX86) || defined(_M_X64))
#endif  // mozilla_windowsdllblocklist_h
