142 lines
3.2 KiB
C++
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;
|
|
}
|
|
} |