AuroraSupport/AuroraRuntime
2
1
Fork 0
Code Issues 29 Pull Requests Releases Wiki Activity

[*] TLS Stack: use new AuByteBuffer apis

Browse Source
This commit is contained in:
Reece Wilson 2022-08-31 00:06:50 +01:00
parent d68fc7fc91
commit 2b72228b9e
4 changed files with 9 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Source/IO/TLS/TLS.hpp
View File

@ -10,6 +10,8 @@
#include <Source/RuntimeInternal.hpp>
#include <Aurora/IO/TLS/TLS.hpp>
#include <mbedtls/entropy.h>
#include <mbedtls/ctr_drbg.h>
#include <mbedtls/ssl.h>
#include <mbedtls/x509.h>

2
Source/IO/TLS/TLSPrivateKeyPair.cpp
View File

@ -56,7 +56,7 @@ namespace Aurora::IO::TLS
keyPair.privateKey.RemainingBytes(),
keyPair.sPassword.size() ? (const unsigned char *)keyPair.sPassword.c_str() : nullptr,
keyPair.sPassword.size(),
gEntropy,
mbedtls_ctr_drbg_random,
&gCtrDrbg);
if (iRet != 0)
{

37
Source/IO/TLS/TLSProtocolRecv.cpp
View File

@ -117,42 +117,11 @@ namespace Aurora::IO::TLS
bool TLSProtocolRecv::DoOneTick(const AuSPtr<Memory::ByteBuffer> &pWriteOutByteBuffer)
{
AuUInt8 *pBase {};
AuUInt uCount {};
while (true)
{
// sanitize end index as index zero
if (pWriteOutByteBuffer->flagCircular &&
pWriteOutByteBuffer->writePtr == pWriteOutByteBuffer->base + pWriteOutByteBuffer->length)
{
pWriteOutByteBuffer->writePtr = pWriteOutByteBuffer->base;
if (pWriteOutByteBuffer->readPtr == pWriteOutByteBuffer->writePtr)
{
return true;
}
}
// fetch linear bytes, readable
// i do not care what anybody says. you do not touch this EVER.
// this is the most correct linear-access in unsanitized state to-date.
// no concept in the bytebuffer exists yet.
if (pWriteOutByteBuffer->flagCircular && pWriteOutByteBuffer->readPtr > pWriteOutByteBuffer->writePtr)
{
uCount = pWriteOutByteBuffer->writePtr - pWriteOutByteBuffer->readPtr;
pBase = pWriteOutByteBuffer->writePtr;
}
else if (pWriteOutByteBuffer->readPtr == pWriteOutByteBuffer->writePtr)
{
uCount = (pWriteOutByteBuffer->base + pWriteOutByteBuffer->length) - pWriteOutByteBuffer->writePtr;
pBase = pWriteOutByteBuffer->writePtr;
}
else if (pWriteOutByteBuffer->flagCircular)
{
uCount = (pWriteOutByteBuffer->base + pWriteOutByteBuffer->length) - pWriteOutByteBuffer->writePtr;
pBase = pWriteOutByteBuffer->writePtr;
}
auto pDest = pWriteOutByteBuffer->GetNextLinearWrite();
AuUInt8 *pBase { pDest.ToPointer() };
AuUInt uCount { pDest.length };
// ...
if (!uCount)

21
Source/IO/TLS/TLSProtocolSend.cpp
View File

@ -30,24 +30,9 @@ namespace Aurora::IO::TLS
while (pReadInByteBuffer->writePtr != pReadInByteBuffer->readPtr)
{
AuUInt8 *pBase {};
AuUInt uCount {};
if (pReadInByteBuffer->base + pReadInByteBuffer->length == pReadInByteBuffer->readPtr)
{
pReadInByteBuffer->readPtr = pReadInByteBuffer->base;
}
if (pReadInByteBuffer->writePtr >= pReadInByteBuffer->readPtr)
{
uCount = pReadInByteBuffer->writePtr - pReadInByteBuffer->readPtr;
pBase = pReadInByteBuffer->readPtr;
}
else if (pReadInByteBuffer->flagCircular)
{
uCount = (pReadInByteBuffer->base + pReadInByteBuffer->length) - pReadInByteBuffer->readPtr;
pBase = pReadInByteBuffer->readPtr;
}
auto pDest = pReadInByteBuffer->GetNextLinearRead();
const AuUInt8 *pBase { pDest.ToPointer() };
AuUInt uCount { pDest.length };
if (!uCount)
{