-#ifndef WIN32
-#include <fcntl.h>
-#include <errno.h>
-#endif
-#include <msp/core/systemerror.h>
-#include <msp/strings/formatter.h>
-#include "handle_private.h"
#include "pipe.h"
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(read_handle, 1024)
{
-#ifdef WIN32
- string name = format("\\\\.\\pipe\\%u.%p", GetCurrentProcessId(), this);
- *handle[0] = CreateNamedPipe(name.c_str(), PIPE_ACCESS_INBOUND|FILE_FLAG_OVERLAPPED, PIPE_TYPE_BYTE, 1, 1024, 1024, 0, 0);
- if(!handle[0])
- throw system_error("CreateNamedPipe");
-
- *handle[1] = CreateFile(name.c_str(), GENERIC_WRITE, 0, 0, OPEN_EXISTING, 0, 0);
- if(!handle[1])
- {
- unsigned err = GetLastError();
- 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)
- throw system_error("pipe");
-
- *handle[0] = pipe_fd[0];
- *handle[1] = pipe_fd[1];
-#endif
+ mode = M_RDWR;
+
+ platform_init();
set_events(P_INPUT);
}
Pipe::~Pipe()
-{
- close();
-#ifdef WIN32
- CloseHandle(*priv->event);
- delete priv->buffer;
-#endif
- delete priv;
-}
-
-void Pipe::close()
{
set_events(P_NONE);
signal_flush_required.emit();
-#ifdef WIN32
- CloseHandle(*handle[0]);
- CloseHandle(*handle[1]);
-#else
- ::close(*handle[0]);
- ::close(*handle[1]);
- signal_closed.emit();
-#endif
+ sys_close(read_handle);
+ sys_close(write_handle);
+}
+
+void Pipe::set_mode(Mode m)
+{
+ m = m&M_RDWR;
+ if(!m)
+ throw invalid_argument("Pipe::set_mode");
+
+ check_access(m);
+
+ Mode close = mode&M_RDWR&~m;
+ if(close&M_READ)
+ sys_close(read_handle);
+ if(close&M_WRITE)
+ sys_close(write_handle);
+ mode = (mode&~M_RDWR)|m;
}
void Pipe::set_block(bool b)
{
- mode = (mode&~M_NONBLOCK);
- if(b)
- mode = (mode|M_NONBLOCK);
-
-#ifndef WIN32
- int flags = fcntl(*handle[0], F_GETFD);
- fcntl(*handle[0], F_SETFL, (flags&O_NONBLOCK)|(b?0:O_NONBLOCK));
- flags = fcntl(*handle[1], F_GETFD);
- fcntl(*handle[1], F_SETFL, (flags&O_NONBLOCK)|(b?0:O_NONBLOCK));
-#endif
+ adjust_mode(mode, M_NONBLOCK, !b);
+ sys_set_blocking(read_handle, b);
+ sys_set_blocking(write_handle, b);
+}
+
+void Pipe::set_inherit(bool i)
+{
+ adjust_mode(mode, M_INHERIT, i);
+ sys_set_inherit(read_handle, i);
+ sys_set_inherit(write_handle, i);
}
unsigned Pipe::do_write(const char *buf, unsigned size)
if(size==0)
return 0;
-#ifdef WIN32
- DWORD ret;
- if(!WriteFile(*handle[1], buf, size, &ret, 0))
- throw system_error("WriteFile");
-#else
- int ret = ::write(*handle[1], buf, size);
- if(ret==-1)
- {
- if(errno==EAGAIN)
- return 0;
- else
- throw system_error("write");
- }
-#endif
-
- return ret;
+ return sys_write(write_handle, buf, size);
}
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
- int ret = ::read(*handle[0], buf, size);
- if(ret==-1)
- {
- if(errno==EAGAIN)
- return 0;
- else
- throw system_error("read");
- }
-#endif
-
+ unsigned ret = reader.read(buf, size);
if(ret==0)
- {
- eof_flag = true;
- signal_end_of_file.emit();
- }
+ set_eof();
return ret;
}
-const Handle &Pipe::get_event_handle()
+const Handle &Pipe::get_handle(Mode m)
{
-#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
+ if(m==M_READ)
+ return read_handle;
+ else if(m==M_WRITE)
+ return write_handle;
+ else
+ throw invalid_argument("Pipe::get_handle");
}
} // namespace IO