115 lines
4.1 KiB
C++
115 lines
4.1 KiB
C++
/***
|
|
Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.
|
|
|
|
File: AuOSThread.hpp
|
|
Date: 2021-6-12
|
|
Author: Reece
|
|
***/
|
|
#pragma once
|
|
|
|
#if defined(AURORA_IS_POSIX_DERIVED)
|
|
#include <setjmp.h>
|
|
#endif
|
|
|
|
namespace Aurora::Threading::Threads
|
|
{
|
|
struct OSThread : IAuroraThread
|
|
{
|
|
OSThread();
|
|
OSThread(const ThreadInfo &info);
|
|
OSThread(AuUInt64);
|
|
~OSThread();
|
|
|
|
bool Run() override;
|
|
void Exit() override;
|
|
bool Exiting() override;
|
|
void SendExitSignal() override;
|
|
|
|
void UnsafeForceTerminateSignal() override;
|
|
|
|
void SetPriority(EThreadPriority prio) override;
|
|
void SetThrottle(EThreadThrottle prio) override;
|
|
void SetAffinity(const HWInfo::CpuBitId &mask) override;
|
|
void SetName(const AuString &name) override;
|
|
|
|
EThreadPriority GetPriority() override;
|
|
EThreadThrottle GetThrottle() override;
|
|
HWInfo::CpuBitId GetMask() override;
|
|
AuString GetName() override;
|
|
|
|
AuSPtr<TLSView> GetTlsView() override;
|
|
|
|
void ExecuteInDeadThread(AuFunction<void()> callback) override;
|
|
AuSPtr<IWaitable> AsWaitable() override;
|
|
|
|
void AddLastHopeTlsHook(const AuSPtr<Threading::Threads::IThreadFeature> &feature) override;
|
|
|
|
void Detach() override;
|
|
|
|
void _ThreadEP();
|
|
AuSPtr<AuLoop::ILoopSource> AsLoopSource() override;
|
|
AuSPtr<IWaitable> GetShutdownSignalWaitable() override;
|
|
AuSPtr<AuLoop::ILoopSource> GetShutdownSignalLS() override;
|
|
|
|
bool InternalKillForceNtfy();
|
|
bool InternalKill(bool locked);
|
|
|
|
void AffinityPrioThrottleTickAmendECores();
|
|
protected:
|
|
|
|
bool Exit(bool willReturnToOS, bool isEOL = false);
|
|
bool ExecuteNewOSContext(AuFunction<void()> task);
|
|
void UpdatePrio(EThreadThrottle throttle, EThreadPriority prio);
|
|
void UpdateAffinity(const HWInfo::CpuBitId &mask);
|
|
void UpdateName();
|
|
void OSAttach();
|
|
void OSDeatach();
|
|
void TeminateOSContext(bool calledFromThis);
|
|
void FreeOSContext();
|
|
|
|
void HookReleaseThreadResources();
|
|
void HookOnExit();
|
|
|
|
private:
|
|
#if defined(AURORA_IS_POSIX_DERIVED)
|
|
jmp_buf env;
|
|
#endif
|
|
Primitives::Mutex tlsLock_;
|
|
AuUInt32 tls_ {};
|
|
OSThread * tlsReferenceThread_ {};
|
|
AuString name_;
|
|
ThreadInfo info_;
|
|
HWInfo::CpuBitId mask_ = HWInfo::CpuBitId().Not();
|
|
HWInfo::CpuBitId throttleMask_ = HWInfo::CpuBitId().Not();
|
|
EThreadPriority prio_ = EThreadPriority::ePrioNormal;
|
|
EThreadThrottle throttle_ = EThreadThrottle::eNormal;
|
|
bool bNotOwned {};
|
|
bool bSupportsAltKill {};
|
|
bool userManagingAffinity_ {};
|
|
bool exiting_{};
|
|
bool contextUsed_{}; // can this thread instance execute code again?
|
|
Primitives::EventShared_t terminated_;
|
|
Primitives::EventShared_t terminateSignal_;
|
|
AuSPtr<AuLoop::ILSEvent> terminateSignalLs_;
|
|
AuSPtr<AuLoop::ILSEvent> terminatedSignalLs_;
|
|
bool bLongJmpOnce {};
|
|
|
|
Primitives::CriticalSection exitOnlyOnce_;
|
|
|
|
AuList<AuSPtr<IThreadFeature>> threadFeatures_;
|
|
|
|
AuFunction<void()> task_;
|
|
|
|
#if defined(AURORA_IS_MODERNNT_DERIVED)
|
|
HANDLE handle_ = INVALID_HANDLE_VALUE;
|
|
#elif defined(AURORA_HAS_PTHREADS)
|
|
pthread_t handle_ {};
|
|
#endif
|
|
|
|
AuUInt64 unixThreadId_ = 0;
|
|
bool detached_ {};
|
|
|
|
};
|
|
|
|
void InitThreading();
|
|
} |