AuroraRuntime/Source/Threading/AuSleep.cpp

/***
    Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.

    File: AuSleep.cpp
    Date: 2021-6-12
    Author: Reece
***/
#include <Source/RuntimeInternal.hpp>
#include "AuSleep.hpp"
#include "AuWaitFor.hpp"

#if defined(AURORA_IS_POSIX_DERIVED)
    #include <unistd.h>
#endif

#if defined(AURORA_IS_MODERNNT_DERIVED)
    #include <Time/Time.hpp>
#endif

namespace Aurora::Threading
{

#if defined(AURORA_IS_MODERNNT_DERIVED)
    using NtDelayExecution_f = DWORD(NTAPI *)(BOOLEAN Alertable, PLARGE_INTEGER       DelayInterval);
    static NtDelayExecution_f NtDelayExecution;
#endif

    AUKN_SYM void Sleep(AuUInt64 qwTimeout)
    {
    #if defined(AURORA_IS_LINUX_DERIVED) 

        SleepNs(AuMSToNS<AuUInt64>(qwTimeout));

    #elif defined(AURORA_IS_MODERNNT_DERIVED)
        
        ::Sleep(qwTimeout);
    
    #else
        std::atomic<bool> value = false;
        BooleanWaitable waitable(value);
        WaitFor(&waitable, qwTimeout);
    #endif 
    }

    AUKN_SYM void SleepNs(AuUInt64 qwTimeout)
    {
    #if defined(AURORA_IS_LINUX_DERIVED) || defined(AURORA_IS_BSD_DERIVED)

        auto qwEndTime = AuTime::SteadyClockNS() + qwTimeout;

        if ((usleep(qwTimeout / 1000) == -1) && 
            (errno == EINTR))
        {
            AuUInt64 qwNow;

            while ((qwNow = AuTime::SteadyClockNS()) < qwEndTime)
            {
                usleep((qwEndTime - qwNow) / 1000);
            }
        }

    #else
    
    #if defined(AURORA_IS_MODERNNT_DERIVED)

        if (NtDelayExecution)
        {
            auto endTimeSteady = AuTime::SteadyClockNS() + qwTimeout;
            auto endTimeWall   = AuTime::CurrentClockNS() + qwTimeout;
            auto targetTimeNt  = AuTime::ConvertTimestampNs(endTimeWall);

            while (AuTime::SteadyClockNS() < endTimeSteady)
            {
                LARGE_INTEGER word;
                word.QuadPart = targetTimeNt;
                NtDelayExecution(FALSE, &word);
            }

            return;
        }

    #endif

        auto status = YieldPollNs(true, qwTimeout + Time::SteadyClockNS(), [=]()
        {
            return false;
        });
    #endif
    }

    void InitSleep()
    {
    #if defined(AURORA_IS_MODERNNT_DERIVED)
        NtDelayExecution = AuReinterpretCast<NtDelayExecution_f>(GetProcAddress(LoadLibraryA("ntdll.dll"), "NtDelayExecution"));
    #endif
    }
}