[+] IFileStream::ToStreamReader()
[+] IFileStreamToStreamWriter() [+] IProcess::ToStreamReader(EStandardHandle stream) [+] IProcess::ToStreamWriter()
This commit is contained in:
parent
9fea60a9ed
commit
baf31e32d0
@ -68,5 +68,15 @@ namespace Aurora::IO::FS
|
||||
* @brief Returns the IO handle backing the file stream, if available.
|
||||
*/
|
||||
virtual AuSPtr<IIOHandle> GetHandle() = 0;
|
||||
|
||||
/**
|
||||
* @brief
|
||||
*/
|
||||
virtual IO::IStreamReader *ToStreamReader() = 0;
|
||||
|
||||
/**
|
||||
* @brief
|
||||
*/
|
||||
virtual IO::IStreamWriter *ToStreamWriter() = 0;
|
||||
};
|
||||
}
|
@ -106,5 +106,17 @@ namespace Aurora::Processes
|
||||
* @return
|
||||
*/
|
||||
virtual AuSPtr<IO::IIOHandle> GetErrorStreamHandle() = 0;
|
||||
|
||||
/**
|
||||
* @brief
|
||||
* @return
|
||||
*/
|
||||
virtual AuSPtr<IO::IStreamReader> ToStreamReader(EStandardHandle stream) = 0;
|
||||
|
||||
/**
|
||||
* @brief
|
||||
* @return
|
||||
*/
|
||||
virtual AuSPtr<IO::IStreamWriter> ToStreamWriter() = 0;
|
||||
};
|
||||
}
|
@ -16,6 +16,13 @@ namespace Aurora::IO::FS
|
||||
{
|
||||
static const AuUInt64 kFileCopyBlock = 0xFFFF * 128;
|
||||
|
||||
WinFileStream::WinFileStream() :
|
||||
reader_(AuUnsafeRaiiToShared(this)),
|
||||
writer_(AuUnsafeRaiiToShared(this))
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
WinFileStream::~WinFileStream()
|
||||
{
|
||||
Close();
|
||||
@ -432,6 +439,16 @@ namespace Aurora::IO::FS
|
||||
return this->pHandle_;
|
||||
}
|
||||
|
||||
IO::IStreamReader *WinFileStream::ToStreamReader()
|
||||
{
|
||||
return &this->reader_;
|
||||
}
|
||||
|
||||
IO::IStreamWriter *WinFileStream::ToStreamWriter()
|
||||
{
|
||||
return &this->writer_;
|
||||
}
|
||||
|
||||
HANDLE WinFileStream::GetWin32Handle(bool bReadOnly)
|
||||
{
|
||||
if (auto pHandle = this->pHandle_)
|
||||
|
@ -13,6 +13,7 @@ namespace Aurora::IO::FS
|
||||
{
|
||||
struct WinFileStream : IFileStream
|
||||
{
|
||||
WinFileStream();
|
||||
~WinFileStream();
|
||||
|
||||
void Init(AuSPtr<IIOHandle> pHandle);
|
||||
@ -27,6 +28,8 @@ namespace Aurora::IO::FS
|
||||
void WriteEoS() override;
|
||||
void MakeTemporary() override;
|
||||
AuSPtr<IIOHandle> GetHandle() override;
|
||||
IO::IStreamReader *ToStreamReader() override;
|
||||
IO::IStreamWriter *ToStreamWriter() override;
|
||||
|
||||
HANDLE GetWin32Handle(bool bReadOnly = false);
|
||||
|
||||
@ -34,6 +37,8 @@ namespace Aurora::IO::FS
|
||||
AuSPtr<IIOHandle> pHandle_;
|
||||
bool bShouldDelete {};
|
||||
HANDLE hEventHandle_ { INVALID_HANDLE_VALUE };
|
||||
AuFS::FileReader reader_;
|
||||
AuFS::FileWriter writer_;
|
||||
};
|
||||
}
|
||||
#endif
|
@ -119,6 +119,12 @@ namespace Aurora::IO::FS
|
||||
return true;
|
||||
}
|
||||
|
||||
PosixFileStream::PosixFileStream() :
|
||||
reader_(AuUnsafeRaiiToShared(this)),
|
||||
writer_(AuUnsafeRaiiToShared(this))
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
PosixFileStream::~PosixFileStream()
|
||||
{
|
||||
@ -416,6 +422,16 @@ namespace Aurora::IO::FS
|
||||
return this->pHandle_;
|
||||
}
|
||||
|
||||
IO::IStreamReader *PosixFileStream::ToStreamReader()
|
||||
{
|
||||
return &this->reader_;
|
||||
}
|
||||
|
||||
IO::IStreamWriter *PosixFileStream::ToStreamWriter()
|
||||
{
|
||||
return &this->writer_;
|
||||
}
|
||||
|
||||
AUKN_SYM AuSPtr<IFileStream> OpenBlockingFileStreamFromHandle(AuSPtr<IIOHandle> pIOHandle)
|
||||
{
|
||||
auto pStream = AuMakeShared<PosixFileStream>();
|
||||
|
@ -19,6 +19,7 @@ namespace Aurora::IO::FS
|
||||
|
||||
struct PosixFileStream : IFileStream
|
||||
{
|
||||
PosixFileStream();
|
||||
~PosixFileStream();
|
||||
|
||||
bool Init(AuSPtr<IIOHandle> pHandle);
|
||||
@ -33,6 +34,8 @@ namespace Aurora::IO::FS
|
||||
void WriteEoS() override;
|
||||
void MakeTemporary() override;
|
||||
AuSPtr<IIOHandle> GetHandle() override;
|
||||
IO::IStreamReader *ToStreamReader() override;
|
||||
IO::IStreamWriter *ToStreamWriter() override;
|
||||
int GetUnixHandle();
|
||||
|
||||
private:
|
||||
@ -40,6 +43,8 @@ namespace Aurora::IO::FS
|
||||
AuString path_;
|
||||
AuThreadPrimitives::SpinLock spinlock_;
|
||||
bool bMadeTemporary {};
|
||||
AuFS::FileReader reader_;
|
||||
AuFS::FileWriter writer_;
|
||||
};
|
||||
}
|
||||
#endif
|
@ -506,6 +506,45 @@ namespace Aurora::Processes
|
||||
return this->fsErrorHandle_;
|
||||
}
|
||||
|
||||
AuSPtr<IO::IStreamReader> ProcessImpl::ToStreamReader(EStandardHandle stream)
|
||||
{
|
||||
if (stream == EStandardHandle::eErrorStream)
|
||||
{
|
||||
if (auto pThat = this->GetErrorStreamHandle())
|
||||
{
|
||||
if (auto pThat2 = AuFS::OpenBlockingFileStreamFromHandle(pThat))
|
||||
{
|
||||
return { pThat2, pThat2->ToStreamReader() };
|
||||
}
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (auto pThat = this->GetOutputAndInputHandles())
|
||||
{
|
||||
if (auto pThat2 = AuFS::OpenBlockingFileStreamFromHandle(pThat))
|
||||
{
|
||||
return { pThat2, pThat2->ToStreamReader() };
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return {};
|
||||
}
|
||||
|
||||
AuSPtr<IO::IStreamWriter> ProcessImpl::ToStreamWriter()
|
||||
{
|
||||
if (auto pThat = this->GetErrorStreamHandle())
|
||||
{
|
||||
if (auto pThat2 = AuFS::OpenBlockingFileStreamFromHandle(pThat))
|
||||
{
|
||||
return { pThat2, pThat2->ToStreamWriter() };
|
||||
}
|
||||
}
|
||||
|
||||
return {};
|
||||
}
|
||||
|
||||
bool ProcessImpl::Start()
|
||||
{
|
||||
if (this->process_ != INVALID_HANDLE_VALUE)
|
||||
|
@ -38,6 +38,9 @@ namespace Aurora::Processes
|
||||
AuSPtr<IO::IIOHandle> GetOutputAndInputHandles() override;
|
||||
AuSPtr<IO::IIOHandle> GetErrorStreamHandle() override;
|
||||
|
||||
AuSPtr<IO::IStreamReader> ToStreamReader(EStandardHandle stream) override;
|
||||
AuSPtr<IO::IStreamWriter> ToStreamWriter() override;
|
||||
|
||||
bool Start() override;
|
||||
|
||||
bool Init();
|
||||
|
@ -481,6 +481,45 @@ namespace Aurora::Processes
|
||||
return this->fsErrorHandle_;
|
||||
}
|
||||
|
||||
AuSPtr<IO::IStreamReader> ProcessImpl::ToStreamReader(EStandardHandle stream)
|
||||
{
|
||||
if (stream == EStandardHandle::eErrorStream)
|
||||
{
|
||||
if (auto pThat = this->GetErrorStreamHandle())
|
||||
{
|
||||
if (auto pThat2 = AuFS::OpenBlockingFileStreamFromHandle(pThat))
|
||||
{
|
||||
return { pThat2, pThat2->ToStreamReader() };
|
||||
}
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (auto pThat = this->GetOutputAndInputHandles())
|
||||
{
|
||||
if (auto pThat2 = AuFS::OpenBlockingFileStreamFromHandle(pThat))
|
||||
{
|
||||
return { pThat2, pThat2->ToStreamReader() };
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return {};
|
||||
}
|
||||
|
||||
AuSPtr<IO::IStreamWriter> ProcessImpl::ToStreamWriter()
|
||||
{
|
||||
if (auto pThat = this->GetErrorStreamHandle())
|
||||
{
|
||||
if (auto pThat2 = AuFS::OpenBlockingFileStreamFromHandle(pThat))
|
||||
{
|
||||
return { pThat2, pThat2->ToStreamWriter() };
|
||||
}
|
||||
}
|
||||
|
||||
return {};
|
||||
}
|
||||
|
||||
void ProcessImpl::ForkMain()
|
||||
{
|
||||
{
|
||||
|
@ -49,6 +49,9 @@ namespace Aurora::Processes
|
||||
AuSPtr<IO::IIOHandle> GetOutputAndInputHandles() override;
|
||||
AuSPtr<IO::IIOHandle> GetErrorStreamHandle() override;
|
||||
|
||||
AuSPtr<IO::IStreamReader> ToStreamReader(EStandardHandle stream) override;
|
||||
AuSPtr<IO::IStreamWriter> ToStreamWriter() override;
|
||||
|
||||
bool Start() override;
|
||||
|
||||
bool Init();
|
||||
|
Loading…
Reference in New Issue
Block a user