AuroraRuntime/Source/IO/Protocol/AuProtocolPiece.cpp

142 lines
3.2 KiB
C++

/***
Copyright (C) 2022 J Reece Wilson (a/k/a "Reece"). All rights reserved.
File: AuProtocolPiece.cpp
Date: 2022-8-24
Author: Reece
***/
#include "Protocol.hpp"
#include "AuProtocolPiece.hpp"
#include "AuProtocolStack.hpp"
#include "IProtocolNext.hpp"
// TODO: check correct thread or throw api
// restrict this to one thread
namespace Aurora::IO::Protocol
{
AuSPtr<ProtocolStack> ProtocolPiece::GetParent()
{
if (this->pParent)
{
return this->pParent->SharedFromThis();
}
else
{
return {};
}
}
AuSPtr<IStreamWriter> ProtocolPiece::ToNextWriter()
{
if (!this->pParent)
{
return {};
}
if (this->pParent->pTopPiece.get() == this)
{
if (this->pParent->pDrainBuffer)
{
return AuMakeShared<Buffered::BlobWriter>(this->pParent->pDrainBuffer);
}
}
return this->pNext ?
this->pNext->pWriteInteface->GetStreamWriter() :
this->pOuputWriter;
}
AuSPtr<IStreamWriter> ProtocolPiece::ToInputWriter()
{
return this->pWriteInteface ? this->pWriteInteface->GetStreamWriter() : AuSPtr<IStreamWriter> {};
}
AuSPtr<IProtocolPiece> ProtocolPiece::GetNextPiece()
{
return this->pNext;
}
bool ProtocolPiece::ReallocateDrainBuffer(AuUInt32 uOutputLength)
{
if (!this->pParent)
{
return false;
}
if (!this->pNext)
{
if (this->pParent->pDrainBuffer)
{
return false;
}
}
return this->outputBuffer.Resize(uOutputLength);
}
void ProtocolPiece::Remove()
{
this->PrivateUserDataClear();
if (!this->pParent)
{
return;
}
auto &pBottomPiece = this->pParent->pBottomPiece;
// fix chain
AuSPtr<IProtocolPiece> pLast;
auto pCurrent = pBottomPiece;
while (true)
{
if (!pCurrent)
{
break;
}
if (pCurrent.get() == this)
{
AuReinterpretCast<ProtocolPiece>(pLast)->pNext = pCurrent->pNext;
break;
}
pLast = pCurrent;
pCurrent = pCurrent->pNext;
}
// fix head pointers
if (this->pParent->pTopPiece.get() == this)
{
this->pParent->pTopPiece = AuReinterpretCast<ProtocolPiece>(pLast);
this->pParent->bWrittenEnd = false;
}
if (this->pParent->pBottomPiece.get() == this)
{
this->pParent->pBottomPiece = this->pNext;
}
this->PrivateUserDataClear();
this->pParent->DiscardAllocCaches();
this->pParent = nullptr;
}
AuSPtr<Memory::ByteBuffer> ProtocolPiece::GetNextPieceBuffer()
{
return AuSPtr<AuByteBuffer>(AuSharedFromThis(), &this->outputBuffer);
}
AuSPtr<IProtocolInterceptorEx> ProtocolPiece::GetExtendedInterceptor()
{
return this->pInterceptorEx;
}
AuSPtr<IProtocolInterceptor> ProtocolPiece::GetShortPipeInterceptor()
{
return this->pInterceptor;
}
}