AuroraRuntime/Source/IO/Net/AuNetSocketOverlappedOperation.Unix.cpp
Jamie Reece Wilson 5cafefae5d [*] LSLocalSemaphore bug fix for scriptability. (TODO: reimpl like the event?)
[*] Fix: IO objects not using the explicit slow sync primitives.
Dunno how these werent refactored; then again, i never properly got around to finishing the factories for fast/slow io objects. In addition, some of these arent even accessed by the ILoopSource interface, so it's not as critical of a failure.
2024-07-30 17:53:58 +01:00

184 lines
4.5 KiB
C++

/***
Copyright (C) 2022 J Reece Wilson (a/k/a "Reece"). All rights reserved.
File: AuNetSocketOverlappedOperation.Unix.cpp
Date: 2022-8-26
Author: Reece
***/
#include "Networking.hpp"
#include "AuNetSocketOverlappedOperation.hpp"
#include <Source/IO/Loop/LSEvent.hpp>
#include "AuNetSocket.Unix.hpp"
#include "AuNetError.hpp"
namespace Aurora::IO::Net
{
SocketOverlappedOperation::SocketOverlappedOperation(bool bMultipleTrigger) :
bMultipleTrigger(bMultipleTrigger)
{
this->pEvent = AuLoop::NewLSEventSlow(false, true);
if (!this->pEvent)
{
return;
}
}
SocketOverlappedOperation::SocketOverlappedOperation(const AuSPtr<AuLoop::ILoopSource> &pTrigger)
{
this->pEvent = pTrigger;
}
bool SocketOverlappedOperation::FinishOperation(const AuSPtr<void> &pThat,
const AuSPtr<INetWorker> &pWorker,
bool bReturnCode)
{
return this->FinishOperationEx(pThat, pWorker, bReturnCode, errno);
}
bool SocketOverlappedOperation::FinishOperationEx(const AuSPtr<void> &pThat,
const AuSPtr<INetWorker> &pWorker,
bool bReturnCode,
AuUInt32 dwErrorCode)
{
if (bReturnCode)
{
this->OnIOComplete();
return true;
}
else
{
if ((dwErrorCode == EAGAIN) ||
(dwErrorCode == EINPROGRESS))
{
return this->BeginOperation(pThat, pWorker);
}
else
{
// ERROR
SysPushErrorIO("Couldn't connect [overlapped] (0x{:x})", dwErrorCode);
this->uOsError = dwErrorCode;
return false;
}
}
}
bool SocketOverlappedOperation::BeginOperation(const AuSPtr<void> &that,
const AuSPtr<INetWorker> &pWorker)
{
this->pSocket = that;
if (!this->pEvent)
{
return false;
}
auto bStatus = pWorker->ToProcessor()->StartSimpleLSWatchEx(this->pEvent,
AuSPtr<AuIO::IIOSimpleEventListener>(that, this),
true);
if (!bStatus)
{
this->pSocket.reset();
return false;
}
return true;
}
void SocketOverlappedOperation::UpdateTrigger(const AuSPtr<AuLoop::ILoopSource> &pTrigger)
{
this->pEvent = pTrigger;
}
bool SocketOverlappedOperation::IsValid()
{
return bool(this->pEvent);
}
void SocketOverlappedOperation::OnIOTick()
{
if (!this->bMultipleTrigger)
{
return;
}
auto pTemp = AuExchange(this->pSocket, {});
if (this->HasFailed())
{
this->OnOverlappedFailure(this->ToError());
}
else
{
this->OnOverlappedComplete();
}
}
void SocketOverlappedOperation::OnIOComplete()
{
if (this->bMultipleTrigger)
{
return;
}
if (AuExchange(this->bHasFlipped, true))
{
return;
}
if (this->HasFailed())
{
this->OnOverlappedFailure(this->ToError());
}
else
{
this->OnOverlappedComplete();
}
this->pSocket.reset();
}
void SocketOverlappedOperation::OnIOFailure()
{
if (AuExchange(this->bHasFlipped, true))
{
return;
}
this->OnOverlappedFailure(this->ToError());
this->pSocket.reset();
}
bool SocketOverlappedOperation::HasFailed()
{
return this->ToOsError() != 0;
}
AuUInt SocketOverlappedOperation::ToOsError()
{
if (this->uOsError)
{
return this->uOsError;
}
return this->uOsError;
}
bool SocketOverlappedOperation::HasComplete()
{
if (this->HasFailed())
{
return false;
}
return this->ToOsError() == 0;
}
NetError SocketOverlappedOperation::ToError()
{
NetError error;
NetError_SetOsError(error, this->ToOsError());
return error;
}
}