From a27ea413e861a30c61b29d77d58ea3d15a9de6a1 Mon Sep 17 00:00:00 2001 From: Mikko Rasa Date: Sun, 12 Jun 2011 18:03:33 +0300 Subject: [PATCH] Use EventReader in pipe --- source/io/pipe.cpp | 88 ++-------------------------------------------- source/io/pipe.h | 7 ++-- 2 files changed, 5 insertions(+), 90 deletions(-) diff --git a/source/io/pipe.cpp b/source/io/pipe.cpp index a848786..f52801c 100644 --- a/source/io/pipe.cpp +++ b/source/io/pipe.cpp @@ -12,21 +12,8 @@ using namespace std; namespace Msp { namespace IO { -struct Pipe::Private -{ -#ifdef WIN32 - OVERLAPPED *overlapped; - Handle event; - unsigned buf_size; - char *buffer; - unsigned buf_avail; - char *buf_next; -#endif -}; - - Pipe::Pipe(): - priv(0) + reader(handle[0], 1024) { #ifdef WIN32 string name = format("\\\\.\\pipe\\%u.%p", GetCurrentProcessId(), this); @@ -41,14 +28,6 @@ Pipe::Pipe(): CloseHandle(*handle[0]); throw system_error(format("CreateFile(%s)", name), err); } - - priv = new Private; - priv->overlapped = 0; - *priv->event = CreateEvent(0, true, false, 0); - priv->buf_size = 1024; - priv->buffer = new char[priv->buf_size]; - priv->buf_avail = 0; - priv->buf_next = priv->buffer; #else int pipe_fd[2]; if(pipe(pipe_fd)==-1) @@ -64,11 +43,6 @@ Pipe::Pipe(): Pipe::~Pipe() { close(); -#ifdef WIN32 - CloseHandle(*priv->event); - delete priv->buffer; -#endif - delete priv; } void Pipe::close() @@ -108,33 +82,7 @@ unsigned Pipe::do_read(char *buf, unsigned size) if(size==0) return 0; -#ifdef WIN32 - // Initiate overlapped read if needed - get_event_handle(); - - if(priv->overlapped) - { - DWORD ret; - if(!GetOverlappedResult(*handle[0], priv->overlapped, &ret, !priv->buf_avail)) - throw system_error("GetOverlappedResult"); - else - { - priv->buf_avail += ret; - delete priv->overlapped; - priv->overlapped = 0; - } - } - - unsigned ret = min(priv->buf_avail, size); - memcpy(buf, priv->buf_next, ret); - priv->buf_next += ret; - priv->buf_avail -= ret; - - // Initiate another overlapped read in case someone is polling us - get_event_handle(); -#else - unsigned ret = sys_read(handle[0], buf, size); -#endif + unsigned ret = reader.read(buf, size); if(ret==0) { @@ -145,37 +93,5 @@ unsigned Pipe::do_read(char *buf, unsigned size) return ret; } -const Handle &Pipe::get_event_handle() -{ -#ifdef WIN32 - if(!priv->overlapped && !priv->buf_avail) - { - priv->overlapped = new OVERLAPPED; - memset(priv->overlapped, 0, sizeof(OVERLAPPED)); - priv->overlapped->hEvent = *priv->event; - - DWORD ret; - priv->buf_next = priv->buffer; - if(!ReadFile(*handle[0], priv->buffer, priv->buf_size, &ret, priv->overlapped)) - { - unsigned err = GetLastError(); - if(err!=ERROR_IO_PENDING) - throw system_error("ReadFile"); - } - else - { - priv->buf_avail = ret; - delete priv->overlapped; - priv->overlapped = 0; - SetEvent(*priv->event); - } - } - - return priv->event; -#else - return handle[0]; -#endif -} - } // namespace IO } // namespace Msp diff --git a/source/io/pipe.h b/source/io/pipe.h index eb77e6a..813c4d8 100644 --- a/source/io/pipe.h +++ b/source/io/pipe.h @@ -2,6 +2,7 @@ #define MSP_IO_PIPE_H_ #include "eventobject.h" +#include "eventreader.h" #include "handle.h" namespace Msp { @@ -10,10 +11,8 @@ namespace IO { class Pipe: public EventObject { private: - struct Private; - Handle handle[2]; - Private *priv; + EventReader reader; public: Pipe(); @@ -28,7 +27,7 @@ protected: virtual unsigned do_read(char *, unsigned); public: - virtual const Handle &get_event_handle(); + virtual const Handle &get_event_handle() { return reader.get_event(); } }; } // namespace IO -- 2.43.0