lib: Add i_stream_set_blocking()

diff --git a/src/lib/istream.c b/src/lib/istream.c
index eb82c89934f82d9cb44f9aa4565ba0a70091ffff..442dd4eea6f10d17af40a4660e13c17f312dca5a 100644 (file)
--- a/src/lib/istream.c
+++ b/src/lib/istream.c
@@ -4,6 +4,7 @@
 #include "ioloop.h"
 #include "array.h"
 #include "str.h"
+#include "fd-set-nonblock.h"
 #include "istream-private.h"
 
 static bool i_stream_is_buffer_invalid(const struct istream_private *stream);
@@ -153,6 +154,21 @@ void i_stream_set_persistent_buffers(struct istream *stream, bool set)
        } while (stream != NULL);
 }
 
+void i_stream_set_blocking(struct istream *stream, bool blocking)
+{
+       int prev_fd = -1;
+
+       do {
+               stream->blocking = blocking;
+               if (stream->real_stream->fd != -1 &&
+                   stream->real_stream->fd != prev_fd) {
+                       fd_set_nonblock(stream->real_stream->fd, !blocking);
+                       prev_fd = stream->real_stream->fd;
+               }
+               stream = stream->real_stream->parent;
+       } while (stream != NULL);
+}
+
 static void i_stream_update(struct istream_private *stream)
 {
        if (stream->parent == NULL)

diff --git a/src/lib/istream.h b/src/lib/istream.h
index 762ac70faa304230f0d31d1f017342b7f6a8022e..d39ba04bdeb7c921073095be319aba94995b8f71 100644 (file)
--- a/src/lib/istream.h
+++ b/src/lib/istream.h
@@ -130,6 +130,9 @@ void i_stream_set_return_partial_line(struct istream *stream, bool set);
    the memory usage is minimized by freeing the stream's buffers whenever they
    become empty. */
 void i_stream_set_persistent_buffers(struct istream *stream, bool set);
+/* Set the istream blocking or nonblocking, including its parent streams.
+   If any of the istreams have an fd, its O_NONBLOCK flag is changed. */
+void i_stream_set_blocking(struct istream *stream, bool blocking);
 
 /* Returns number of bytes read if read was ok, -1 if EOF or error, -2 if the
    input buffer is full. */