// Copyright (c) Microsoft Corporation.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

// set_new_handler

#include <mutex>
#include <new.h>
#include <new>

namespace {
    _STD new_handler _New_handler;
    constinit _STD mutex _New_handler_mutex;

    int __cdecl _New_handler_interface(size_t) { // interface to existing Microsoft _callnewh mechanism
        _New_handler();
        return 1;
    }
} // namespace

_STD_BEGIN

_CRTIMP2 new_handler __cdecl set_new_handler(_In_opt_ new_handler pnew) noexcept { // remove current handler
    lock_guard _Lock{_New_handler_mutex};
    new_handler pold = _New_handler;
    _New_handler     = pnew;
    _set_new_handler(pnew ? _New_handler_interface : nullptr);
    return pold;
}

_CRTIMP2 new_handler __cdecl get_new_handler() noexcept { // get current new handler
    lock_guard _Lock{_New_handler_mutex};
    return _New_handler;
}

_STD_END