AuroraRuntime/Source/IO/Protocol/AuProtocolPiece.hpp
Reece 6683d3cfdc [+] IProtocolPiece.GetNextPiece
[+] IProtocolPiece.ReallocateDrainBuffer
[*] Harden protocol piece use after destroy
2023-04-29 07:00:29 +01:00

36 lines
1.1 KiB
C++

/***
Copyright (C) 2022 J Reece Wilson (a/k/a "Reece"). All rights reserved.
File: ProtocolPiece.hpp
Date: 2022-8-24
Author: Reece
***/
#pragma once
namespace Aurora::IO::Protocol
{
struct IProtocolNext;
struct ProtocolStack;
struct ProtocolPiece : IProtocolPiece, AuEnableSharedFromThis<ProtocolPiece>
{
ProtocolStack *pParent;
AuSPtr<IProtocolInterceptor> pInterceptor;
AuSPtr<IProtocolInterceptorEx> pInterceptorEx;
AuSPtr<IProtocolNext> pWriteInteface;
AuSPtr<ProtocolPiece> pNext;
AuByteBuffer outputBuffer;
AuSPtr<IStreamWriter> pOuputWriter;
bool bMultipleTick {};
AuSPtr<IProtocolPiece> GetNextPiece() override;
bool ReallocateDrainBuffer(AuUInt32 uOutputLength) override;
void Remove() override;
AuSPtr<IStreamWriter> ToNextWriter() override;
AuSPtr<IStreamWriter> ToInputWriter() override;
AuSPtr<Memory::ByteBuffer> GetNextPieceBuffer() override;
AuSPtr<IProtocolInterceptorEx> GetExtendedInterceptor() override;
AuSPtr<IProtocolInterceptor> GetShortPipeInterceptor() override;
};
}