Justin Marshall
64 lines
1.4 KiB
C++
64 lines
1.4 KiB
C++
#include "win_fibers.h"
|
|
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
|
|
namespace {
|
|
|
|
char* CopyFiberName(const char* source) {
|
|
const char* const text = source == nullptr ? "" : source;
|
|
const std::size_t length = std::strlen(text);
|
|
char* const copy = static_cast<char*>(std::malloc(length + 1));
|
|
if (copy != nullptr) {
|
|
std::memcpy(copy, text, length + 1);
|
|
}
|
|
return copy;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
idSysFiber::idSysFiber(const char* fiberName)
|
|
: name(CopyFiberName(fiberName))
|
|
, alive(true)
|
|
, fiber(nullptr)
|
|
, parent(nullptr) {
|
|
fiber = CreateFiber(0x20000u, &idSysFiber::FiberRoutine, this);
|
|
if (fiber == nullptr) {
|
|
alive = false;
|
|
}
|
|
}
|
|
|
|
idSysFiber::~idSysFiber() {
|
|
std::free(name);
|
|
if (fiber != nullptr) {
|
|
DeleteFiber(fiber);
|
|
}
|
|
}
|
|
|
|
bool idSysFiber::Execute() {
|
|
if (alive && fiber != nullptr) {
|
|
parent = GetCurrentFiber();
|
|
SwitchToFiber(fiber);
|
|
}
|
|
return alive;
|
|
}
|
|
|
|
void idSysFiber::YieldFiber() {
|
|
if (parent != nullptr) {
|
|
SwitchToFiber(parent);
|
|
}
|
|
}
|
|
|
|
void WINAPI idSysFiber::FiberRoutine(void* data) {
|
|
idSysFiber* const self = static_cast<idSysFiber*>(data);
|
|
self->Run();
|
|
self->alive = false;
|
|
self->YieldFiber();
|
|
|
|
// A finished Windows fiber must never return from its entry routine when
|
|
// its parent is expected to retain control. Match the recovered guard.
|
|
for (;;) {
|
|
self->YieldFiber();
|
|
}
|
|
}
|