//
// internal_shared.h
//
//      Copyright (c) 2024 Timo Kreuzer
//
// Header for definitions shared between vcruntime and UCRT.
//
// SPDX-License-Identifier: MIT
//

#pragma once

#include <corecrt_startup.h>
#include <crtdbg.h>
#include <windef.h>
#include <winbase.h>

#ifdef __cplusplus
extern "C" {
#endif // __cplusplus

#if defined(_MSC_VER)
#define _CRTALLOC(x) __declspec(allocate(x))
#else
#define _CRTALLOC(x) __attribute__((section(x)))
#endif

#ifdef _MSC_VER
#pragma section(".CRT$XIC", long, read)
#pragma section(".CRT$XPX", long, read)
#pragma section(".CRT$XPXA", long, read)

#pragma section(".CRT$XIA", long, read)
#pragma section(".CRT$XIAA", long, read) // CRT startup
#pragma section(".CRT$XIZ", long, read)
#pragma section(".CRT$XCA", long, read)
#pragma section(".CRT$XCZ", long, read)
#pragma section(".CRT$XPA", long, read)
#pragma section(".CRT$XPZ", long, read)
#pragma section(".CRT$XTA", long, read)
#pragma section(".CRT$XTZ", long, read)


#pragma section(".CRT$XDA",long,read)
#pragma section(".CRT$XDC",long,read)
#pragma section(".CRT$XDZ",long,read)

#pragma section(".CRT$XLA",long,read) // TLS callback start
#pragma section(".CRT$XLC",long,read) // TLS constructors
#pragma section(".CRT$XLD",long,read) // TLS destructors
#pragma section(".CRT$XLZ",long,read) // TLS callback end

#pragma section(".tls",long,read,write)
#pragma section(".tls$AAA",long,read,write)
#pragma section(".tls$ZZZ",long,read,write)
#endif

extern _PIFV __xi_a[];
extern _PIFV __xi_z[];
extern _PVFV __xc_a[];
extern _PVFV __xc_z[];
extern _PVFV __xp_a[];
extern _PVFV __xp_z[];
extern _PVFV __xt_a[];
extern _PVFV __xt_z[];

extern char __ImageBase;

extern const IMAGE_TLS_DIRECTORY _tls_used;

#define CRT_WARNING_DISABLE_PUSH(wn, message) \
    __pragma(warning(push)) \
    __pragma(warning(disable: wn))

#define CRT_WARNING_POP \
    __pragma(warning(pop))

#define _BEGIN_SECURE_CRT_DEPRECATION_DISABLE \
    __pragma(warning(push)) \
    __pragma(warning(disable: 4996))

#define _END_SECURE_CRT_DEPRECATION_DISABLE \
    __pragma(warning(pop))

#define _CRT_DEBUGGER_IGNORE -1
#define _CRT_DEBUGGER_GSFAILURE 1
#define _CRT_DEBUGGER_INVALIDPARAMETER 2
#define _CRT_DEBUGGER_ABORT 3

#define _CRT_DEBUGGER_HOOK(x)

#define _CRT_SECURITYCRITICAL_ATTRIBUTE
#define _CRT_SECURITYSAFECRITICAL_ATTRIBUTE

//-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
//
// Parameter validation macros
// Partly duplicated in corecrt_internal_strtox.h
//
//-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+

#ifdef _DEBUG
    #define _INVALID_PARAMETER(expr) _invalid_parameter(expr, __FUNCTIONW__, __FILEW__, __LINE__, 0)
#else
    #define _INVALID_PARAMETER(expr) _invalid_parameter_noinfo()
#endif

#define _VALIDATE_RETURN(expr, errorcode, retexpr)                             \
    {                                                                          \
        int _Expr_val = !!(expr);                                              \
        _ASSERT_EXPR((_Expr_val), _CRT_WIDE(#expr));                           \
        if (!(_Expr_val))                                                      \
        {                                                                      \
            *_errno() = (errorcode);                                           \
            _INVALID_PARAMETER(_CRT_WIDE(#expr));                              \
            return (retexpr);                                                  \
        }                                                                      \
    }

#define _VALIDATE_RETURN_VOID(expr, errorcode)                                 \
    {                                                                          \
        int _Expr_val = !!(expr);                                              \
        _ASSERT_EXPR((_Expr_val), _CRT_WIDE(#expr));                           \
        if (!(_Expr_val))                                                      \
        {                                                                      \
            *_errno() = (errorcode);                                           \
            _INVALID_PARAMETER(_CRT_WIDE(#expr));                              \
            return;                                                            \
        }                                                                      \
    }

#define _VALIDATE_RETURN_NOERRNO(expr, retexpr)                                \
    {                                                                          \
        int _Expr_val = !!(expr);                                              \
        _ASSERT_EXPR((_Expr_val), _CRT_WIDE(#expr));                           \
        if (!(_Expr_val))                                                      \
        {                                                                      \
            _INVALID_PARAMETER(_CRT_WIDE(#expr));                              \
            return (retexpr);                                                  \
        }                                                                      \
    }

#define _VALIDATE_RETURN_ERRCODE(expr, errorcode)                              \
    _VALIDATE_RETURN(expr, errorcode, errorcode)

#define _VALIDATE_CLEAR_OSSERR_RETURN(expr, errorcode, retexpr)                \
    {                                                                          \
        int _Expr_val = !!(expr);                                              \
        _ASSERT_EXPR((_Expr_val), _CRT_WIDE(#expr));                           \
        if (!(_Expr_val))                                                      \
        {                                                                      \
            *__doserrno() = 0L;                                                \
            *_errno() = errorcode;                                             \
            _INVALID_PARAMETER(_CRT_WIDE(#expr) );                             \
            return (retexpr);                                                  \
        }                                                                      \
    }

#define _VALIDATE_CLEAR_OSSERR_RETURN_ERRCODE(expr, errorcode)                 \
    _VALIDATE_CLEAR_OSSERR_RETURN(expr, errorcode, errorcode)

#define _VALIDATE_RETURN_NOEXC(expr, errorcode, retexpr)                       \
    {                                                                          \
        if (!(expr))                                                           \
        {                                                                      \
            *_errno() = errorcode;                                             \
            return (retexpr);                                                  \
        }                                                                      \
    }

#define _VALIDATE_RETURN_ERRCODE_NOEXC(expr, errorcode)                        \
    _VALIDATE_RETURN_NOEXC(expr, errorcode, errorcode)

#define _malloc_crt malloc
#define _free_crt free
#define _calloc_crt calloc
#define _recalloc_crt _recalloc
#define _malloca_crt _malloca
#define _freea_crt _freea

#ifdef __cplusplus
} // extern "C"

template<typename T>
__forceinline
T __crt_fast_encode_pointer(T ptr)
{
    union { T Ptr; uintptr_t Uint; } u = { ptr };
    u.Uint ^= __security_cookie;
    return u.Ptr;
}

template<typename T>
__forceinline
T __crt_fast_decode_pointer(T ptr)
{
    union { T Ptr; uintptr_t Uint; } u = { ptr };
    u.Uint ^= __security_cookie;
    return u.Ptr;
}

template<typename T>
T __crt_interlocked_read(const T volatile* ptr)
{
    return *ptr;
}

template<typename T>
T __crt_interlocked_read_pointer(const T volatile* ptr)
{
    return *ptr;
}

template<typename T>
T __crt_interlocked_exchange_pointer(T *ptr, void* value)
{
    return (T)_InterlockedExchangePointer((void* volatile*)ptr, value);
}

template<typename T>
struct __crt_seh_guarded_call
{
    template<typename Init, typename Action, typename Cleanup>
    T operator()(Init init, Action action, Cleanup cleanup)
    {
        T result;
        init();
        __try
        {
            result = action();
        }
        __finally
        {
            cleanup();
        }
        __endtry
        return result;
    }
};

template<>
struct __crt_seh_guarded_call<void>
{
    template<typename Init, typename Action, typename Cleanup>
    void operator()(Init init, Action action, Cleanup cleanup)
    {
        init();
        __try
        {
            action();
        }
        __finally
        {
            cleanup();
        }
        __endtry
    }
};

template<typename FuncPtr>
FuncPtr __crt_get_proc_address(HMODULE module, const char* name)
{
    return (FuncPtr)GetProcAddress(module, name);
}

template<typename Action, typename Cleanup>
void __crt_call_and_cleanup(Action action, Cleanup cleanup)
{
    action();
    cleanup();
}

struct __crt_malloc_free_policy
{
    template<typename T>
    void operator()(T* const ptr) throw()
    {
        _free_crt((void*)ptr);
    }
};

struct __crt_public_free_policy
{
    template<typename T>
    void operator()(T* const ptr) throw()
    {
        _free_crt((void*)ptr);
    }
};

struct __crt_malloca_free_policy
{
    void operator()(void* const ptr) throw()
    {
        _freea_crt(ptr);
    }
};

template <typename T, typename FreePolicy = __crt_malloc_free_policy>
class __crt_unique_heap_ptr
{
    T* _ptr;
public:
    __crt_unique_heap_ptr() : _ptr(nullptr) {}
    __crt_unique_heap_ptr(T* ptr) : _ptr(ptr) {}
    __crt_unique_heap_ptr(__crt_unique_heap_ptr&& from) : _ptr(from._ptr) { from._ptr = nullptr; }
     ~__crt_unique_heap_ptr() { FreePolicy()(_ptr); }
    void attach(T* ptr) { FreePolicy()(_ptr); _ptr = ptr; }
    T* detach() { T* ptr = _ptr; _ptr = nullptr; return ptr; }
    operator bool() const { return _ptr != nullptr; }
    T* get() const { return _ptr; }
    T** get_address_of() { return &_ptr; }

    // Move assignment
    __crt_unique_heap_ptr& operator=(__crt_unique_heap_ptr&& from)
    {
        FreePolicy()(_ptr);
        _ptr = from._ptr;
        from._ptr = nullptr;
        return *this;
    }
};

template <typename T>
using __crt_scoped_stack_ptr = __crt_unique_heap_ptr<T, __crt_malloca_free_policy>;

#define _malloc_crt_t(t, n) (__crt_unique_heap_ptr<t>(static_cast<t*>(_malloc_crt((n) * sizeof(t)))))
#define _calloc_crt_t(t, n) (__crt_unique_heap_ptr<t>(static_cast<t*>(_calloc_crt((n),  sizeof(t)))))
#define _recalloc_crt_t(t, p, n) (__crt_unique_heap_ptr<t>(static_cast<t*>(_recalloc_crt(p, (n), sizeof(t)))))
#define _malloca_crt_t(t, n) (__crt_scoped_stack_ptr<t>(static_cast<t*>(_malloca_crt((n) * sizeof(t)))))

#endif // __cplusplus