1 // Streambuf iterators
2
3 // Copyright (C) 1997-2019 Free Software Foundation, Inc.
4 //
5 // This file is part of the GNU ISO C++ Library. This library is free
6 // software; you can redistribute it and/or modify it under the
7 // terms of the GNU General Public License as published by the
8 // Free Software Foundation; either version 3, or (at your option)
9 // any later version.
10
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 // GNU General Public License for more details.
15
16 // Under Section 7 of GPL version 3, you are granted additional
17 // permissions described in the GCC Runtime Library Exception, version
18 // 3.1, as published by the Free Software Foundation.
19
20 // You should have received a copy of the GNU General Public License and
21 // a copy of the GCC Runtime Library Exception along with this program;
22 // see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
23 // <http://www.gnu.org/licenses/>.
24
25 /** @file bits/streambuf_iterator.h
26 * This is an internal header file, included by other library headers.
27 * Do not attempt to use it directly. @headername{iterator}
28 */
29
30 #ifndef _STREAMBUF_ITERATOR_H
31 #define _STREAMBUF_ITERATOR_H 1
32
33 #pragma GCC system_header
34
35 #include <streambuf>
36 #include <debug/debug.h>
37
_GLIBCXX_VISIBILITY(default)38 namespace std _GLIBCXX_VISIBILITY(default)
39 {
40 _GLIBCXX_BEGIN_NAMESPACE_VERSION
41
42 /**
43 * @addtogroup iterators
44 * @{
45 */
46
47 // 24.5.3 Template class istreambuf_iterator
48 /// Provides input iterator semantics for streambufs.
49 template<typename _CharT, typename _Traits>
50 class istreambuf_iterator
51 : public iterator<input_iterator_tag, _CharT, typename _Traits::off_type,
52 _CharT*,
53 #if __cplusplus >= 201103L
54 // LWG 445.
55 _CharT>
56 #else
57 _CharT&>
58 #endif
59 {
60 public:
61 // Types:
62 ///@{
63 /// Public typedefs
64 typedef _CharT char_type;
65 typedef _Traits traits_type;
66 typedef typename _Traits::int_type int_type;
67 typedef basic_streambuf<_CharT, _Traits> streambuf_type;
68 typedef basic_istream<_CharT, _Traits> istream_type;
69 ///@}
70
71 template<typename _CharT2>
72 friend typename __gnu_cxx::__enable_if<__is_char<_CharT2>::__value,
73 ostreambuf_iterator<_CharT2> >::__type
74 copy(istreambuf_iterator<_CharT2>, istreambuf_iterator<_CharT2>,
75 ostreambuf_iterator<_CharT2>);
76
77 template<bool _IsMove, typename _CharT2>
78 friend typename __gnu_cxx::__enable_if<__is_char<_CharT2>::__value,
79 _CharT2*>::__type
80 __copy_move_a2(istreambuf_iterator<_CharT2>,
81 istreambuf_iterator<_CharT2>, _CharT2*);
82
83 template<typename _CharT2>
84 friend typename __gnu_cxx::__enable_if<__is_char<_CharT2>::__value,
85 istreambuf_iterator<_CharT2> >::__type
86 find(istreambuf_iterator<_CharT2>, istreambuf_iterator<_CharT2>,
87 const _CharT2&);
88
89 template<typename _CharT2, typename _Distance>
90 friend typename __gnu_cxx::__enable_if<__is_char<_CharT2>::__value,
91 void>::__type
92 advance(istreambuf_iterator<_CharT2>&, _Distance);
93
94 private:
95 // 24.5.3 istreambuf_iterator
96 // p 1
97 // If the end of stream is reached (streambuf_type::sgetc()
98 // returns traits_type::eof()), the iterator becomes equal to
99 // the "end of stream" iterator value.
100 // NB: This implementation assumes the "end of stream" value
101 // is EOF, or -1.
102 mutable streambuf_type* _M_sbuf;
103 int_type _M_c;
104
105 public:
106 /// Construct end of input stream iterator.
107 _GLIBCXX_CONSTEXPR istreambuf_iterator() _GLIBCXX_USE_NOEXCEPT
108 : _M_sbuf(0), _M_c(traits_type::eof()) { }
109
110 #if __cplusplus >= 201103L
111 istreambuf_iterator(const istreambuf_iterator&) noexcept = default;
112
113 ~istreambuf_iterator() = default;
114 #endif
115
116 /// Construct start of input stream iterator.
117 istreambuf_iterator(istream_type& __s) _GLIBCXX_USE_NOEXCEPT
118 : _M_sbuf(__s.rdbuf()), _M_c(traits_type::eof()) { }
119
120 /// Construct start of streambuf iterator.
121 istreambuf_iterator(streambuf_type* __s) _GLIBCXX_USE_NOEXCEPT
122 : _M_sbuf(__s), _M_c(traits_type::eof()) { }
123
124 #if __cplusplus >= 201103L
125 istreambuf_iterator&
126 operator=(const istreambuf_iterator&) noexcept = default;
127 #endif
128
129 /// Return the current character pointed to by iterator. This returns
130 /// streambuf.sgetc(). It cannot be assigned. NB: The result of
131 /// operator*() on an end of stream is undefined.
132 char_type
133 operator*() const
134 {
135 int_type __c = _M_get();
136
137 #ifdef _GLIBCXX_DEBUG_PEDANTIC
138 // Dereferencing a past-the-end istreambuf_iterator is a
139 // libstdc++ extension
140 __glibcxx_requires_cond(!_S_is_eof(__c),
141 _M_message(__gnu_debug::__msg_deref_istreambuf)
142 ._M_iterator(*this));
143 #endif
144 return traits_type::to_char_type(__c);
145 }
146
147 /// Advance the iterator. Calls streambuf.sbumpc().
148 istreambuf_iterator&
149 operator++()
150 {
151 __glibcxx_requires_cond(_M_sbuf &&
152 (!_S_is_eof(_M_c) || !_S_is_eof(_M_sbuf->sgetc())),
153 _M_message(__gnu_debug::__msg_inc_istreambuf)
154 ._M_iterator(*this));
155
156 _M_sbuf->sbumpc();
157 _M_c = traits_type::eof();
158 return *this;
159 }
160
161 /// Advance the iterator. Calls streambuf.sbumpc().
162 istreambuf_iterator
163 operator++(int)
164 {
165 __glibcxx_requires_cond(_M_sbuf &&
166 (!_S_is_eof(_M_c) || !_S_is_eof(_M_sbuf->sgetc())),
167 _M_message(__gnu_debug::__msg_inc_istreambuf)
168 ._M_iterator(*this));
169
170 istreambuf_iterator __old = *this;
171 __old._M_c = _M_sbuf->sbumpc();
172 _M_c = traits_type::eof();
173 return __old;
174 }
175
176 // _GLIBCXX_RESOLVE_LIB_DEFECTS
177 // 110 istreambuf_iterator::equal not const
178 // NB: there is also number 111 (NAD) relevant to this function.
179 /// Return true both iterators are end or both are not end.
180 bool
181 equal(const istreambuf_iterator& __b) const
182 { return _M_at_eof() == __b._M_at_eof(); }
183
184 private:
185 int_type
186 _M_get() const
187 {
188 int_type __ret = _M_c;
189 if (_M_sbuf && _S_is_eof(__ret) && _S_is_eof(__ret = _M_sbuf->sgetc()))
190 _M_sbuf = 0;
191 return __ret;
192 }
193
194 bool
195 _M_at_eof() const
196 { return _S_is_eof(_M_get()); }
197
198 static bool
199 _S_is_eof(int_type __c)
200 {
201 const int_type __eof = traits_type::eof();
202 return traits_type::eq_int_type(__c, __eof);
203 }
204 };
205
206 template<typename _CharT, typename _Traits>
207 inline bool
208 operator==(const istreambuf_iterator<_CharT, _Traits>& __a,
209 const istreambuf_iterator<_CharT, _Traits>& __b)
210 { return __a.equal(__b); }
211
212 template<typename _CharT, typename _Traits>
213 inline bool
214 operator!=(const istreambuf_iterator<_CharT, _Traits>& __a,
215 const istreambuf_iterator<_CharT, _Traits>& __b)
216 { return !__a.equal(__b); }
217
218 /// Provides output iterator semantics for streambufs.
219 template<typename _CharT, typename _Traits>
220 class ostreambuf_iterator
221 : public iterator<output_iterator_tag, void, void, void, void>
222 {
223 public:
224 // Types:
225 ///@{
226 /// Public typedefs
227 typedef _CharT char_type;
228 typedef _Traits traits_type;
229 typedef basic_streambuf<_CharT, _Traits> streambuf_type;
230 typedef basic_ostream<_CharT, _Traits> ostream_type;
231 ///@}
232
233 template<typename _CharT2>
234 friend typename __gnu_cxx::__enable_if<__is_char<_CharT2>::__value,
235 ostreambuf_iterator<_CharT2> >::__type
236 copy(istreambuf_iterator<_CharT2>, istreambuf_iterator<_CharT2>,
237 ostreambuf_iterator<_CharT2>);
238
239 private:
240 streambuf_type* _M_sbuf;
241 bool _M_failed;
242
243 public:
244 /// Construct output iterator from ostream.
245 ostreambuf_iterator(ostream_type& __s) _GLIBCXX_USE_NOEXCEPT
246 : _M_sbuf(__s.rdbuf()), _M_failed(!_M_sbuf) { }
247
248 /// Construct output iterator from streambuf.
249 ostreambuf_iterator(streambuf_type* __s) _GLIBCXX_USE_NOEXCEPT
250 : _M_sbuf(__s), _M_failed(!_M_sbuf) { }
251
252 /// Write character to streambuf. Calls streambuf.sputc().
253 ostreambuf_iterator&
254 operator=(_CharT __c)
255 {
256 if (!_M_failed &&
257 _Traits::eq_int_type(_M_sbuf->sputc(__c), _Traits::eof()))
258 _M_failed = true;
259 return *this;
260 }
261
262 /// Return *this.
263 ostreambuf_iterator&
264 operator*()
265 { return *this; }
266
267 /// Return *this.
268 ostreambuf_iterator&
269 operator++(int)
270 { return *this; }
271
272 /// Return *this.
273 ostreambuf_iterator&
274 operator++()
275 { return *this; }
276
277 /// Return true if previous operator=() failed.
278 bool
279 failed() const _GLIBCXX_USE_NOEXCEPT
280 { return _M_failed; }
281
282 ostreambuf_iterator&
283 _M_put(const _CharT* __ws, streamsize __len)
284 {
285 if (__builtin_expect(!_M_failed, true)
286 && __builtin_expect(this->_M_sbuf->sputn(__ws, __len) != __len,
287 false))
288 _M_failed = true;
289 return *this;
290 }
291 };
292
293 // Overloads for streambuf iterators.
294 template<typename _CharT>
295 typename __gnu_cxx::__enable_if<__is_char<_CharT>::__value,
296 ostreambuf_iterator<_CharT> >::__type
297 copy(istreambuf_iterator<_CharT> __first,
298 istreambuf_iterator<_CharT> __last,
299 ostreambuf_iterator<_CharT> __result)
300 {
301 if (__first._M_sbuf && !__last._M_sbuf && !__result._M_failed)
302 {
303 bool __ineof;
304 __copy_streambufs_eof(__first._M_sbuf, __result._M_sbuf, __ineof);
305 if (!__ineof)
306 __result._M_failed = true;
307 }
308 return __result;
309 }
310
311 template<bool _IsMove, typename _CharT>
312 typename __gnu_cxx::__enable_if<__is_char<_CharT>::__value,
313 ostreambuf_iterator<_CharT> >::__type
314 __copy_move_a2(_CharT* __first, _CharT* __last,
315 ostreambuf_iterator<_CharT> __result)
316 {
317 const streamsize __num = __last - __first;
318 if (__num > 0)
319 __result._M_put(__first, __num);
320 return __result;
321 }
322
323 template<bool _IsMove, typename _CharT>
324 typename __gnu_cxx::__enable_if<__is_char<_CharT>::__value,
325 ostreambuf_iterator<_CharT> >::__type
326 __copy_move_a2(const _CharT* __first, const _CharT* __last,
327 ostreambuf_iterator<_CharT> __result)
328 {
329 const streamsize __num = __last - __first;
330 if (__num > 0)
331 __result._M_put(__first, __num);
332 return __result;
333 }
334
335 template<bool _IsMove, typename _CharT>
336 typename __gnu_cxx::__enable_if<__is_char<_CharT>::__value,
337 _CharT*>::__type
338 __copy_move_a2(istreambuf_iterator<_CharT> __first,
339 istreambuf_iterator<_CharT> __last, _CharT* __result)
340 {
341 typedef istreambuf_iterator<_CharT> __is_iterator_type;
342 typedef typename __is_iterator_type::traits_type traits_type;
343 typedef typename __is_iterator_type::streambuf_type streambuf_type;
344 typedef typename traits_type::int_type int_type;
345
346 if (__first._M_sbuf && !__last._M_sbuf)
347 {
348 streambuf_type* __sb = __first._M_sbuf;
349 int_type __c = __sb->sgetc();
350 while (!traits_type::eq_int_type(__c, traits_type::eof()))
351 {
352 const streamsize __n = __sb->egptr() - __sb->gptr();
353 if (__n > 1)
354 {
355 traits_type::copy(__result, __sb->gptr(), __n);
356 __sb->__safe_gbump(__n);
357 __result += __n;
358 __c = __sb->underflow();
359 }
360 else
361 {
362 *__result++ = traits_type::to_char_type(__c);
363 __c = __sb->snextc();
364 }
365 }
366 }
367 return __result;
368 }
369
370 template<typename _CharT>
371 typename __gnu_cxx::__enable_if<__is_char<_CharT>::__value,
372 istreambuf_iterator<_CharT> >::__type
373 find(istreambuf_iterator<_CharT> __first,
374 istreambuf_iterator<_CharT> __last, const _CharT& __val)
375 {
376 typedef istreambuf_iterator<_CharT> __is_iterator_type;
377 typedef typename __is_iterator_type::traits_type traits_type;
378 typedef typename __is_iterator_type::streambuf_type streambuf_type;
379 typedef typename traits_type::int_type int_type;
380 const int_type __eof = traits_type::eof();
381
382 if (__first._M_sbuf && !__last._M_sbuf)
383 {
384 const int_type __ival = traits_type::to_int_type(__val);
385 streambuf_type* __sb = __first._M_sbuf;
386 int_type __c = __sb->sgetc();
387 while (!traits_type::eq_int_type(__c, __eof)
388 && !traits_type::eq_int_type(__c, __ival))
389 {
390 streamsize __n = __sb->egptr() - __sb->gptr();
391 if (__n > 1)
392 {
393 const _CharT* __p = traits_type::find(__sb->gptr(),
394 __n, __val);
395 if (__p)
396 __n = __p - __sb->gptr();
397 __sb->__safe_gbump(__n);
398 __c = __sb->sgetc();
399 }
400 else
401 __c = __sb->snextc();
402 }
403
404 __first._M_c = __eof;
405 }
406
407 return __first;
408 }
409
410 template<typename _CharT, typename _Distance>
411 typename __gnu_cxx::__enable_if<__is_char<_CharT>::__value,
412 void>::__type
413 advance(istreambuf_iterator<_CharT>& __i, _Distance __n)
414 {
415 if (__n == 0)
416 return;
417
418 __glibcxx_assert(__n > 0);
419 __glibcxx_requires_cond(!__i._M_at_eof(),
420 _M_message(__gnu_debug::__msg_inc_istreambuf)
421 ._M_iterator(__i));
422
423 typedef istreambuf_iterator<_CharT> __is_iterator_type;
424 typedef typename __is_iterator_type::traits_type traits_type;
425 typedef typename __is_iterator_type::streambuf_type streambuf_type;
426 typedef typename traits_type::int_type int_type;
427 const int_type __eof = traits_type::eof();
428
429 streambuf_type* __sb = __i._M_sbuf;
430 while (__n > 0)
431 {
432 streamsize __size = __sb->egptr() - __sb->gptr();
433 if (__size > __n)
434 {
435 __sb->__safe_gbump(__n);
436 break;
437 }
438
439 __sb->__safe_gbump(__size);
440 __n -= __size;
441 if (traits_type::eq_int_type(__sb->underflow(), __eof))
442 {
443 __glibcxx_requires_cond(__n == 0,
444 _M_message(__gnu_debug::__msg_inc_istreambuf)
445 ._M_iterator(__i));
446 break;
447 }
448 }
449
450 __i._M_c = __eof;
451 }
452
453 /// @} group iterators
454
455 _GLIBCXX_END_NAMESPACE_VERSION
456 } // namespace
457
458 #endif
459