From 6a15c8431f8fbf078003a8f000f21e1968cb7da8 Mon Sep 17 00:00:00 2001
From: Boris Kolpackov <boris@codesynthesis.com>
Date: Wed, 25 Nov 2015 12:50:09 +0200
Subject: Implement ofdstream

---
 butl/fdstream | 50 ++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 38 insertions(+), 12 deletions(-)

(limited to 'butl/fdstream')

diff --git a/butl/fdstream b/butl/fdstream
index 01a19b7..2814d4b 100644
--- a/butl/fdstream
+++ b/butl/fdstream
@@ -6,18 +6,19 @@
 #define BUTL_FDSTREAM
 
 #include <istream>
+#include <ostream>
 
 namespace butl
 {
-  // An input stream that is initialized with a file descriptor
-  // rather than a file name.
+  // An iostream that is initialized with a file descriptor rather than
+  // a file name.
   //
   // Notes and limitations:
   //
   // - char only
-  // - input only
+  // - input or output but not both
   // - no support for put back
-  // - throws std::system_error in case of a read() error
+  // - throws std::system_error in case of a read()/write() error
   // - not movable, though can be easily supported
   //
   class fdbuf: public std::basic_streambuf<char>
@@ -35,17 +36,18 @@ namespace butl
     close ();
 
     void
-    open (int fd) {close (); fd_ = fd; setg (buf_, buf_, buf_);}
+    open (int fd);
 
     bool
     is_open () const {return fd_ != -1;}
 
-    // basic_streambuf input interface.
-    //
   public:
     using int_type = std::basic_streambuf<char>::int_type;
     using traits_type = std::basic_streambuf<char>::traits_type;
 
+    // basic_streambuf input interface.
+    //
+  public:
     virtual std::streamsize
     showmanyc ();
 
@@ -56,26 +58,50 @@ namespace butl
     bool
     load ();
 
+    // basic_streambuf output interface.
+    //
+  public:
+    virtual int_type
+    overflow (int_type);
+
+    virtual int
+    sync ();
+
+  private:
+    bool
+    save ();
+
   private:
     int fd_ = -1;
     char buf_[2048];
   };
 
-  class ifdstream_base
+  class fdstream_base
   {
   protected:
-    ifdstream_base () = default;
-    ifdstream_base (int fd): buf_ (fd) {}
+    fdstream_base () = default;
+    fdstream_base (int fd): buf_ (fd) {}
 
   protected:
     fdbuf buf_;
   };
 
-  class ifdstream: ifdstream_base, public std::istream
+  class ifdstream: fdstream_base, public std::istream
   {
   public:
     ifdstream (): std::istream (&buf_) {}
-    ifdstream (int fd): ifdstream_base (fd), std::istream (&buf_) {}
+    ifdstream (int fd): fdstream_base (fd), std::istream (&buf_) {}
+
+    void close () {buf_.close ();}
+    void open (int fd) {buf_.open (fd);}
+    bool is_open () const {return buf_.is_open ();}
+  };
+
+  class ofdstream: fdstream_base, public std::ostream
+  {
+  public:
+    ofdstream (): std::ostream (&buf_) {}
+    ofdstream (int fd): fdstream_base (fd), std::ostream (&buf_) {}
 
     void close () {buf_.close ();}
     void open (int fd) {buf_.open (fd);}
-- 
cgit v1.1