Add a helper function for adjusting mode flags

[libs/core.git] / source / io / pipe.cpp

diff --git a/source/io/pipe.cpp b/source/io/pipe.cpp
index 831076f0086b2ffd756ec9e86a3b1b1364e9b1cc..444f8c72a3c3bcc986935f698c3d4b62e61b1a37 100644 (file)
--- a/source/io/pipe.cpp
+++ b/source/io/pipe.cpp
@@ -24,12 +24,25 @@ Pipe::~Pipe()
        sys_close(write_handle);
 }
 
-void Pipe::set_block(bool b)
+void Pipe::set_mode(Mode m)
 {
-       mode = (mode&~M_NONBLOCK);
-       if(b)
-               mode = (mode|M_NONBLOCK);
+       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)
+{
+       adjust_mode(mode, M_NONBLOCK, !b);
        sys_set_blocking(read_handle, b);
        sys_set_blocking(write_handle, b);
 }