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