fdstream.h
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2006 Tommi Maekitalo
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * As a special exception, you may use this file as part of a free
10  * software library without restriction. Specifically, if other files
11  * instantiate templates or use macros or inline functions from this
12  * file, or you compile this file and link it with other files to
13  * produce an executable, this file does not by itself cause the
14  * resulting executable to be covered by the GNU General Public
15  * License. This exception does not however invalidate any other
16  * reasons why the executable file might be covered by the GNU Library
17  * General Public License.
18  *
19  * This library is distributed in the hope that it will be useful,
20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22  * Lesser General Public License for more details.
23  *
24  * You should have received a copy of the GNU Lesser General Public
25  * License along with this library; if not, write to the Free Software
26  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
27  */
28 
29 #ifndef CXXTOOLS_FDSTREAM_H
30 #define CXXTOOLS_FDSTREAM_H
31 
32 #include <iostream>
33 
34 namespace cxxtools
35 {
36  class Fdstreambuf : public std::streambuf
37  {
38  std::streambuf::int_type overflow(std::streambuf::int_type ch);
39  std::streambuf::int_type underflow();
40  int sync();
41 
42  int fd;
43  bool doClose;
44  unsigned bufsize;
45  char* buffer;
46 
47  public:
48  explicit Fdstreambuf(int fd, unsigned bufsize = 8192, bool doClose = false);
49  ~Fdstreambuf();
50 
51  void setClose(bool sw = true) { doClose = sw; }
52  bool isClose() const { return doClose; }
53  void close();
54 
55  int getFd() const { return fd; }
56  };
57 
58  class Fdiostream : public std::iostream
59  {
60  Fdstreambuf streambuf;
61 
62  public:
63  explicit Fdiostream(int fd, unsigned bufsize = 8192, bool close = false)
64  : std::iostream(0),
65  streambuf(fd, bufsize, close)
66  {
67  init(&streambuf);
68  }
69 
70  void setClose(bool sw = true) { streambuf.setClose(sw); }
71  bool isClose() const { return streambuf.isClose(); }
72  void close() { streambuf.close(); }
73 
74  int getFd() const { return streambuf.getFd(); }
75  };
76 }
77 
78 #endif // CXXTOOLS_FDSTREAM_H