1 /*
2  * Copyright (c) 1996,1997
3  * Silicon Graphics Computer Systems, Inc.
4  *
5  * Copyright (c) 1999
6  * Boris Fomitchev
7  *
8  * This material is provided "as is", with absolutely no warranty expressed
9  * or implied. Any use is at your own risk.
10  *
11  * Permission to use or copy this software for any purpose is hereby granted
12  * without fee, provided the above notices are retained on all copies.
13  * Permission to modify the code and to distribute modified code is granted,
14  * provided the above notices are retained, and a notice that the code was
15  * modified is included with the above copyright notice.
16  *
17  */
18 #ifndef _STLP_FSTREAM_C
19 #define _STLP_FSTREAM_C
20 
21 #ifndef _STLP_INTERNAL_FSTREAM_H
22 #  include <stl/_fstream.h>
23 #endif
24 
25 #ifndef _STLP_INTERNAL_LIMITS
26 #  include <stl/_limits.h>
27 #endif
28 
29 _STLP_BEGIN_NAMESPACE
30 
31 # if defined ( _STLP_NESTED_TYPE_PARAM_BUG )
32 // no wchar_t is supported for this mode
33 # define __BF_int_type__ int
34 # define __BF_pos_type__ streampos
35 # define __BF_off_type__ streamoff
36 # else
37 # define __BF_int_type__ _STLP_TYPENAME_ON_RETURN_TYPE basic_filebuf<_CharT, _Traits>::int_type
38 # define __BF_pos_type__ _STLP_TYPENAME_ON_RETURN_TYPE basic_filebuf<_CharT, _Traits>::pos_type
39 # define __BF_off_type__ _STLP_TYPENAME_ON_RETURN_TYPE basic_filebuf<_CharT, _Traits>::off_type
40 # endif
41 
42 
43 //----------------------------------------------------------------------
44 // Public basic_filebuf<> member functions
45 
46 template <class _CharT, class _Traits>
basic_filebuf()47 basic_filebuf<_CharT, _Traits>::basic_filebuf()
48      :  basic_streambuf<_CharT, _Traits>(), _M_base(),
49     _M_constant_width(false), _M_always_noconv(false),
50     _M_int_buf_dynamic(false),
51     _M_in_input_mode(false), _M_in_output_mode(false),
52     _M_in_error_mode(false), _M_in_putback_mode(false),
53     _M_int_buf(0), _M_int_buf_EOS(0),
54     _M_ext_buf(0), _M_ext_buf_EOS(0),
55     _M_ext_buf_converted(0), _M_ext_buf_end(0),
56     _M_state(_STLP_DEFAULT_CONSTRUCTED(_State_type)),
57     _M_end_state(_STLP_DEFAULT_CONSTRUCTED(_State_type)),
58     _M_mmap_base(0), _M_mmap_len(0),
59     _M_saved_eback(0), _M_saved_gptr(0), _M_saved_egptr(0),
60     _M_codecvt(0),
61     _M_width(1), _M_max_width(1)
62 {
63   this->_M_setup_codecvt(locale(), false);
64 }
65 
66 template <class _CharT, class _Traits>
~basic_filebuf()67 basic_filebuf<_CharT, _Traits>::~basic_filebuf() {
68   this->close();
69   _M_deallocate_buffers();
70 }
71 
72 
73 template <class _CharT, class _Traits>
74 _STLP_TYPENAME_ON_RETURN_TYPE basic_filebuf<_CharT, _Traits>::int_type
underflow()75 basic_filebuf<_CharT, _Traits>::underflow() {
76   return _Underflow<_CharT, _Traits>::_M_doit(this);
77 }
78 
79 template <class _CharT, class _Traits>
80 basic_filebuf<_CharT, _Traits>*
close()81 basic_filebuf<_CharT, _Traits>::close() {
82   bool __ok = this->is_open();
83 
84   if (_M_in_output_mode) {
85     __ok = __ok && !_Traits::eq_int_type(this->overflow(traits_type::eof()),
86                                          traits_type::eof());
87     __ok = __ok && this->_M_unshift();
88   }
89   else if (_M_in_input_mode)
90       this->_M_exit_input_mode();
91 
92   // Note order of arguments.  We close the file even if __ok is false.
93   __ok = _M_base._M_close() && __ok;
94 
95   // Restore the initial state, except that we don't deallocate the buffer
96   // or mess with the cached codecvt information.
97   _M_state = _M_end_state = _State_type();
98   _M_ext_buf_converted = _M_ext_buf_end = 0;
99 
100   _M_mmap_base = 0;
101   _M_mmap_len = 0;
102 
103   this->setg(0, 0, 0);
104   this->setp(0, 0);
105 
106   _M_saved_eback = _M_saved_gptr = _M_saved_egptr = 0;
107 
108   _M_in_input_mode = _M_in_output_mode = _M_in_error_mode = _M_in_putback_mode
109     = false;
110 
111   return __ok ? this : 0;
112 }
113 
114 // This member function is called whenever we exit input mode.
115 // It unmaps the memory-mapped file, if any, and sets
116 // _M_in_input_mode to false.
117 template <class _CharT, class _Traits>
_M_exit_input_mode()118 void basic_filebuf<_CharT, _Traits>::_M_exit_input_mode() {
119   if (_M_mmap_base != 0) {
120     _M_base._M_unmap(_M_mmap_base, _M_mmap_len);
121     _M_mmap_base = 0;
122     _M_mmap_len = 0;
123   }
124   _M_in_input_mode = false;
125 }
126 
127 
128 //----------------------------------------------------------------------
129 // basic_filebuf<> overridden protected virtual member functions
130 
131 template <class _CharT, class _Traits>
showmanyc()132 streamsize basic_filebuf<_CharT, _Traits>::showmanyc() {
133   // Is there any possibility that reads can succeed?
134   if (!this->is_open() || _M_in_output_mode || _M_in_error_mode)
135     return -1;
136   else if (_M_in_putback_mode)
137     return this->egptr() - this->gptr();
138   else if (_M_constant_width) {
139     streamoff __pos  = _M_base._M_seek(0, ios_base::cur);
140     streamoff __size = _M_base._M_file_size();
141     return __pos >= 0 && __size > __pos ? __size - __pos : 0;
142   }
143   else
144     return 0;
145 }
146 
147 
148 // Make a putback position available, if necessary, by switching to a
149 // special internal buffer used only for putback.  The buffer is
150 // [_M_pback_buf, _M_pback_buf + _S_pback_buf_size), but the base
151 // class only sees a piece of it at a time.  (We want to make sure
152 // that we don't try to read a character that hasn't been initialized.)
153 // The end of the putback buffer is always _M_pback_buf + _S_pback_buf_size,
154 // but the beginning is usually not _M_pback_buf.
155 template <class _CharT, class _Traits>
156 __BF_int_type__
pbackfail(int_type __c)157 basic_filebuf<_CharT, _Traits>::pbackfail(int_type __c) {
158   const int_type __eof = traits_type::eof();
159 
160   // If we aren't already in input mode, pushback is impossible.
161   if (!_M_in_input_mode)
162     return __eof;
163 
164   // We can use the ordinary get buffer if there's enough space, and
165   // if it's a buffer that we're allowed to write to.
166   if (this->gptr() != this->eback() &&
167       (traits_type::eq_int_type(__c, __eof) ||
168        traits_type::eq(traits_type::to_char_type(__c), this->gptr()[-1]) ||
169        !_M_mmap_base)) {
170     this->gbump(-1);
171     if (traits_type::eq_int_type(__c, __eof) ||
172         traits_type::eq(traits_type::to_char_type(__c), *this->gptr()))
173       return traits_type::to_int_type(*this->gptr());
174   }
175   else if (!traits_type::eq_int_type(__c, __eof)) {
176     // Are we in the putback buffer already?
177     _CharT* __pback_end = _M_pback_buf + __STATIC_CAST(int,_S_pback_buf_size);
178     if (_M_in_putback_mode) {
179       // Do we have more room in the putback buffer?
180       if (this->eback() != _M_pback_buf)
181         this->setg(this->egptr() - 1, this->egptr() - 1, __pback_end);
182       else
183         return __eof;           // No more room in the buffer, so fail.
184     }
185     else {                      // We're not yet in the putback buffer.
186       _M_saved_eback = this->eback();
187       _M_saved_gptr  = this->gptr();
188       _M_saved_egptr = this->egptr();
189       this->setg(__pback_end - 1, __pback_end - 1, __pback_end);
190       _M_in_putback_mode = true;
191     }
192   }
193   else
194     return __eof;
195 
196   // We have made a putback position available.  Assign to it, and return.
197   *this->gptr() = traits_type::to_char_type(__c);
198   return __c;
199 }
200 
201 // This member function flushes the put area, and also outputs the
202 // character __c (unless __c is eof).  Invariant: we always leave room
203 // in the internal buffer for one character more than the base class knows
204 // about.  We see the internal buffer as [_M_int_buf, _M_int_buf_EOS), but
205 // the base class only sees [_M_int_buf, _M_int_buf_EOS - 1).
206 template <class _CharT, class _Traits>
207 __BF_int_type__
overflow(int_type __c)208 basic_filebuf<_CharT, _Traits>::overflow(int_type __c) {
209   // Switch to output mode, if necessary.
210   if (!_M_in_output_mode)
211     if (!_M_switch_to_output_mode())
212       return traits_type::eof();
213 
214   _CharT* __ibegin = this->_M_int_buf;
215   _CharT* __iend   = this->pptr();
216   this->setp(_M_int_buf, _M_int_buf_EOS - 1);
217 
218   // Put __c at the end of the internal buffer.
219   if (!traits_type::eq_int_type(__c, traits_type::eof()))
220     *__iend++ = _Traits::to_char_type(__c);
221 
222   // For variable-width encodings, output may take more than one pass.
223   while (__ibegin != __iend) {
224     const _CharT* __inext = __ibegin;
225     char* __enext         = _M_ext_buf;
226     typename _Codecvt::result __status
227       = _M_codecvt->out(_M_state, __ibegin, __iend, __inext,
228                         _M_ext_buf, _M_ext_buf_EOS, __enext);
229     if (__status == _Codecvt::noconv) {
230       return _Noconv_output<_Traits>::_M_doit(this, __ibegin, __iend)
231         ? traits_type::not_eof(__c)
232         : _M_output_error();
233     }
234 
235     // For a constant-width encoding we know that the external buffer
236     // is large enough, so failure to consume the entire internal buffer
237     // or to produce the correct number of external characters, is an error.
238     // For a variable-width encoding, however, we require only that we
239     // consume at least one internal character
240     else if (__status != _Codecvt::error &&
241              (((__inext == __iend) &&
242                (__enext - _M_ext_buf == _M_width * (__iend - __ibegin))) ||
243               (!_M_constant_width && __inext != __ibegin))) {
244         // We successfully converted part or all of the internal buffer.
245       ptrdiff_t __n = __enext - _M_ext_buf;
246       if (_M_write(_M_ext_buf, __n))
247         __ibegin += __inext - __ibegin;
248       else
249         return _M_output_error();
250     }
251     else
252       return _M_output_error();
253   }
254 
255   return traits_type::not_eof(__c);
256 }
257 
258 // This member function must be called before any I/O has been
259 // performed on the stream, otherwise it has no effect.
260 //
261 // __buf == 0 && __n == 0 means to make this stream unbuffered.
262 // __buf != 0 && __n > 0 means to use __buf as the stream's internal
263 // buffer, rather than the buffer that would otherwise be allocated
264 // automatically.  __buf must be a pointer to an array of _CharT whose
265 // size is at least __n.
266 template <class _CharT, class _Traits>
267 basic_streambuf<_CharT, _Traits>*
setbuf(_CharT * __buf,streamsize __n)268 basic_filebuf<_CharT, _Traits>::setbuf(_CharT* __buf, streamsize __n) {
269   if (!_M_in_input_mode &&! _M_in_output_mode && !_M_in_error_mode &&
270       _M_int_buf == 0) {
271     if (__buf == 0 && __n == 0)
272       _M_allocate_buffers(0, 1);
273     else if (__buf != 0 && __n > 0)
274       _M_allocate_buffers(__buf, __n);
275   }
276   return this;
277 }
278 
279 #if defined (_STLP_ASSERTIONS)
280 // helper class.
281 template <class _CharT>
282 struct _Filebuf_Tmp_Buf {
283   _CharT* _M_ptr;
_Filebuf_Tmp_Buf_Filebuf_Tmp_Buf284   _Filebuf_Tmp_Buf(ptrdiff_t __n) : _M_ptr(0) { _M_ptr = new _CharT[__n]; }
~_Filebuf_Tmp_Buf_Filebuf_Tmp_Buf285   ~_Filebuf_Tmp_Buf() { delete[] _M_ptr; }
286 };
287 #endif
288 
289 template <class _CharT, class _Traits>
290 __BF_pos_type__
seekoff(off_type __off,ios_base::seekdir __whence,ios_base::openmode)291 basic_filebuf<_CharT, _Traits>::seekoff(off_type __off,
292                                         ios_base::seekdir __whence,
293                                         ios_base::openmode /* dummy */) {
294   if (!this->is_open())
295     return pos_type(-1);
296 
297   if (!_M_constant_width && __off != 0)
298     return pos_type(-1);
299 
300   if (!_M_seek_init(__off != 0 || __whence != ios_base::cur))
301     return pos_type(-1);
302 
303   // Seek to beginning or end, regardless of whether we're in input mode.
304   if (__whence == ios_base::beg || __whence == ios_base::end)
305     return _M_seek_return(_M_base._M_seek(_M_width * __off, __whence),
306                           _State_type());
307 
308   // Seek relative to current position.  Complicated if we're in input mode.
309   _STLP_ASSERT(__whence == ios_base::cur)
310   if (!_M_in_input_mode)
311     return _M_seek_return(_M_base._M_seek(_M_width * __off, __whence),
312                           _State_type());
313 
314   if (_M_mmap_base != 0) {
315     // __off is relative to gptr().  We need to do a bit of arithmetic
316     // to get an offset relative to the external file pointer.
317     streamoff __adjust = _M_mmap_len - (this->gptr() - (_CharT*) _M_mmap_base);
318 
319     // if __off == 0, we do not need to exit input mode and to shift file pointer
320     return __off == 0 ? pos_type(_M_base._M_seek(0, ios_base::cur) - __adjust)
321                       : _M_seek_return(_M_base._M_seek(__off - __adjust, ios_base::cur), _State_type());
322   }
323 
324   if (_M_constant_width) { // Get or set the position.
325     streamoff __iadj = _M_width * (this->gptr() - this->eback());
326 
327     // Compensate for offset relative to gptr versus offset relative
328     // to external pointer.  For a text-oriented stream, where the
329     // compensation is more than just pointer arithmetic, we may get
330     // but not set the current position.
331 
332     if (__iadj <= _M_ext_buf_end - _M_ext_buf) {
333       streamoff __eadj =  _M_base._M_get_offset(_M_ext_buf + __STATIC_CAST(ptrdiff_t, __iadj), _M_ext_buf_end);
334 
335       return __off == 0 ? pos_type(_M_base._M_seek(0, ios_base::cur) - __eadj)
336                         : _M_seek_return(_M_base._M_seek(__off - __eadj, ios_base::cur), _State_type());
337     }
338   }
339   else {                    // Get the position.  Encoding is var width.
340     // Get position in internal buffer.
341     ptrdiff_t __ipos = this->gptr() - this->eback();
342 
343     // Get corresponding position in external buffer.
344     _State_type __state = _M_state;
345     int __epos = _M_codecvt->length(__state, _M_ext_buf, _M_ext_buf_converted,
346                                     __ipos);
347 #if defined (_STLP_ASSERTIONS)
348     // Sanity check (expensive): make sure __epos is the right answer.
349     _STLP_ASSERT(__epos >= 0)
350     _State_type __tmp_state = _M_state;
351     _Filebuf_Tmp_Buf<_CharT> __buf(__ipos);
352     _CharT* __ibegin = __buf._M_ptr;
353     _CharT* __inext  = __ibegin;
354     const char* __dummy;
355     typename _Codecvt::result __status
356       = _M_codecvt->in(__tmp_state,
357                        _M_ext_buf, _M_ext_buf + __epos, __dummy,
358                        __ibegin, __ibegin + __ipos, __inext);
359     // The result code is necessarily ok because:
360     // - noconv: impossible for a variable encoding
361     // - error: length method is supposed to count until it reach max value or find an error so we cannot have
362     //          an error again when decoding an external buffer up to length return value.
363     // - partial: idem error, it is also a reason for length to stop counting.
364     _STLP_ASSERT(__status == _Codecvt::ok)
365     _STLP_ASSERT(__inext == __ibegin + __ipos)
366     _STLP_ASSERT(equal(this->eback(), this->gptr(), __ibegin, _STLP_PRIV _Eq_traits<traits_type>()))
367 #endif
368     // Get the current position (at the end of the external buffer),
369     // then adjust it.  Again, it might be a text-oriented stream.
370     streamoff __cur = _M_base._M_seek(0, ios_base::cur);
371     streamoff __adj = _M_base._M_get_offset(_M_ext_buf, _M_ext_buf + __epos) -
372                       _M_base._M_get_offset(_M_ext_buf, _M_ext_buf_end);
373     if (__cur != -1 && __cur + __adj >= 0)
374       return __off == 0 ? pos_type(__cur + __adj)
375                         : _M_seek_return(__cur + __adj, __state);
376   }
377 
378   return pos_type(-1);
379 }
380 
381 
382 template <class _CharT, class _Traits>
383 __BF_pos_type__
seekpos(pos_type __pos,ios_base::openmode)384 basic_filebuf<_CharT, _Traits>::seekpos(pos_type __pos,
385                                         ios_base::openmode /* dummy */) {
386   if (this->is_open()) {
387     if (!_M_seek_init(true))
388       return pos_type(-1);
389 
390     streamoff __off = off_type(__pos);
391     if (__off != -1 && _M_base._M_seek(__off, ios_base::beg) != -1) {
392       _M_state = __pos.state();
393       return _M_seek_return(__off, __pos.state());
394     }
395   }
396 
397   return pos_type(-1);
398 }
399 
400 
401 template <class _CharT, class _Traits>
sync()402 int basic_filebuf<_CharT, _Traits>::sync() {
403   if (_M_in_output_mode)
404     return traits_type::eq_int_type(this->overflow(traits_type::eof()),
405                                     traits_type::eof()) ? -1 : 0;
406   return 0;
407 }
408 
409 
410 // Change the filebuf's locale.  This member function has no effect
411 // unless it is called before any I/O is performed on the stream.
412 template <class _CharT, class _Traits>
imbue(const locale & __loc)413 void basic_filebuf<_CharT, _Traits>::imbue(const locale& __loc) {
414   if (!_M_in_input_mode && !_M_in_output_mode && !_M_in_error_mode) {
415     this->_M_setup_codecvt(__loc);
416   }
417 }
418 
419 //----------------------------------------------------------------------
420 // basic_filebuf<> helper functions.
421 
422 //----------------------------------------
423 // Helper functions for switching between modes.
424 
425 // This member function is called if we're performing the first I/O
426 // operation on a filebuf, or if we're performing an input operation
427 // immediately after a seek.
428 template <class _CharT, class _Traits>
_M_switch_to_input_mode()429 bool basic_filebuf<_CharT, _Traits>::_M_switch_to_input_mode() {
430   if (this->is_open() && (((int)_M_base.__o_mode() & (int)ios_base::in) !=0)
431       && (_M_in_output_mode == 0) && (_M_in_error_mode == 0)) {
432     if (!_M_int_buf && !_M_allocate_buffers())
433       return false;
434 
435     _M_ext_buf_converted = _M_ext_buf;
436     _M_ext_buf_end       = _M_ext_buf;
437 
438     _M_end_state    = _M_state;
439 
440     _M_in_input_mode = true;
441     return true;
442   }
443 
444   return false;
445 }
446 
447 
448 // This member function is called if we're performing the first I/O
449 // operation on a filebuf, or if we're performing an output operation
450 // immediately after a seek.
451 template <class _CharT, class _Traits>
_M_switch_to_output_mode()452 bool basic_filebuf<_CharT, _Traits>::_M_switch_to_output_mode() {
453   if (this->is_open() && (_M_base.__o_mode() & (int)ios_base::out) &&
454       _M_in_input_mode == 0 && _M_in_error_mode == 0) {
455 
456     if (!_M_int_buf && !_M_allocate_buffers())
457       return false;
458 
459     // In append mode, every write does an implicit seek to the end
460     // of the file.  Whenever leaving output mode, the end of file
461     // get put in the initial shift state.
462     if (_M_base.__o_mode() & ios_base::app)
463       _M_state = _State_type();
464 
465     this->setp(_M_int_buf, _M_int_buf_EOS - 1);
466     _M_in_output_mode = true;
467     return true;
468   }
469 
470   return false;
471 }
472 
473 
474 //----------------------------------------
475 // Helper functions for input
476 
477 // This member function is called if there is an error during input.
478 // It puts the filebuf in error mode, clear the get area buffer, and
479 // returns eof.
480 // returns eof.  Error mode is sticky; it is cleared only by close or
481 // seek.
482 
483 template <class _CharT, class _Traits>
484 __BF_int_type__
_M_input_error()485 basic_filebuf<_CharT, _Traits>::_M_input_error() {
486    this->_M_exit_input_mode();
487   _M_in_output_mode = false;
488   _M_in_error_mode = true;
489   this->setg(0, 0, 0);
490   return traits_type::eof();
491 }
492 
493 template <class _CharT, class _Traits>
494 __BF_int_type__
_M_underflow_aux()495 basic_filebuf<_CharT, _Traits>::_M_underflow_aux() {
496   // We have the state and file position from the end of the internal
497   // buffer.  This round, they become the beginning of the internal buffer.
498   _M_state    = _M_end_state;
499 
500   // Fill the external buffer.  Start with any leftover characters that
501   // didn't get converted last time.
502   if (_M_ext_buf_end > _M_ext_buf_converted)
503 
504     _M_ext_buf_end = _STLP_STD::copy(_M_ext_buf_converted, _M_ext_buf_end, _M_ext_buf);
505     // boris : copy_backward did not work
506     //_M_ext_buf_end = copy_backward(_M_ext_buf_converted, _M_ext_buf_end,
507     //_M_ext_buf+ (_M_ext_buf_end - _M_ext_buf_converted));
508   else
509     _M_ext_buf_end = _M_ext_buf;
510 
511   // Now fill the external buffer with characters from the file.  This is
512   // a loop because occasionally we don't get enough external characters
513   // to make progress.
514   for (;;) {
515     ptrdiff_t __n = _M_base._M_read(_M_ext_buf_end, _M_ext_buf_EOS - _M_ext_buf_end);
516     if (__n < 0) {
517       // Read failed, maybe we should set err bit on associated stream...
518       this->setg(0, 0, 0);
519       return traits_type::eof();
520     }
521 
522     _M_ext_buf_end += __n;
523 
524     // If external buffer is empty there is nothing to do.
525     if (_M_ext_buf == _M_ext_buf_end) {
526       this->setg(0, 0, 0);
527       return traits_type::eof();
528     }
529 
530     // Convert the external buffer to internal characters.
531     const char* __enext;
532     _CharT* __inext;
533 
534     typename _Codecvt::result __status
535       = _M_codecvt->in(_M_end_state,
536                        _M_ext_buf, _M_ext_buf_end, __enext,
537                        _M_int_buf, _M_int_buf_EOS, __inext);
538 
539     /* Error conditions:
540      * (1) Return value of error.
541      * (2) Producing internal characters without consuming external characters.
542      * (3) In fixed-width encodings, producing an internal sequence whose length
543      * is inconsistent with that of the internal sequence.
544      * (4) Failure to produce any characters if we have enough characters in
545      * the external buffer, where "enough" means the largest possible width
546      * of a single character. */
547     if (__status == _Codecvt::noconv)
548       return _Noconv_input<_Traits>::_M_doit(this);
549     else if (__status == _Codecvt::error ||
550             (__inext != _M_int_buf && __enext == _M_ext_buf) ||
551             (_M_constant_width && (__inext - _M_int_buf) *  _M_width != (__enext - _M_ext_buf)) ||
552             (__inext == _M_int_buf && __enext - _M_ext_buf >= _M_max_width))
553       return _M_input_error();
554     else if (__inext != _M_int_buf) {
555       _M_ext_buf_converted = _M_ext_buf + (__enext - _M_ext_buf);
556       this->setg(_M_int_buf, _M_int_buf, __inext);
557       return traits_type::to_int_type(*_M_int_buf);
558     }
559     /* We need to go around the loop again to get more external characters.
560      * But if the previous read failed then don't try again for now.
561      * Don't enter error mode for a failed read. Error mode is sticky,
562      * and we might succeed if we try again. */
563     if (__n <= 0) {
564       this->setg(0, 0, 0);
565       return traits_type::eof();
566     }
567   }
568 }
569 
570 //----------------------------------------
571 // Helper functions for output
572 
573 // This member function is called if there is an error during output.
574 // It puts the filebuf in error mode, clear the put area buffer, and
575 // returns eof.  Error mode is sticky; it is cleared only by close or
576 // seek.
577 template <class _CharT, class _Traits>
578 __BF_int_type__
_M_output_error()579 basic_filebuf<_CharT, _Traits>::_M_output_error() {
580   _M_in_output_mode = false;
581   _M_in_input_mode = false;
582   _M_in_error_mode = true;
583   this->setp(0, 0);
584   return traits_type::eof();
585 }
586 
587 
588 // Write whatever sequence of characters is necessary to get back to
589 // the initial shift state.  This function overwrites the external
590 // buffer, changes the external file position, and changes the state.
591 // Precondition: the internal buffer is empty.
592 template <class _CharT, class _Traits>
_M_unshift()593 bool basic_filebuf<_CharT, _Traits>::_M_unshift() {
594   if (_M_in_output_mode && !_M_constant_width) {
595     typename _Codecvt::result __status;
596     do {
597       char* __enext = _M_ext_buf;
598       __status = _M_codecvt->unshift(_M_state,
599                                      _M_ext_buf, _M_ext_buf_EOS, __enext);
600       if (__status == _Codecvt::noconv ||
601           (__enext == _M_ext_buf && __status == _Codecvt::ok))
602         return true;
603       else if (__status == _Codecvt::error)
604         return false;
605       else if (!_M_write(_M_ext_buf, __enext - _M_ext_buf))
606         return false;
607     } while (__status == _Codecvt::partial);
608   }
609 
610   return true;
611 }
612 
613 
614 //----------------------------------------
615 // Helper functions for buffer allocation and deallocation
616 
617 // This member function is called when we're initializing a filebuf's
618 // internal and external buffers.  The argument is the size of the
619 // internal buffer; the external buffer is sized using the character
620 // width in the current encoding.  Preconditions: the buffers are currently
621 // null.  __n >= 1.  __buf is either a null pointer or a pointer to an
622 // array show size is at least __n.
623 
624 // We need __n >= 1 for two different reasons.  For input, the base
625 // class always needs a buffer because of the semantics of underflow().
626 // For output, we want to have an internal buffer that's larger by one
627 // element than the buffer that the base class knows about.  (See
628 // basic_filebuf<>::overflow() for the reason.)
629 template <class _CharT, class _Traits>
_M_allocate_buffers(_CharT * __buf,streamsize __n)630 bool basic_filebuf<_CharT, _Traits>::_M_allocate_buffers(_CharT* __buf, streamsize __n) {
631   //The major hypothesis in the following implementation is that size_t is unsigned.
632   //We also need streamsize byte representation to be larger or equal to the int
633   //representation to correctly store the encoding information.
634   _STLP_STATIC_ASSERT(!numeric_limits<size_t>::is_signed &&
635                       sizeof(streamsize) >= sizeof(int))
636 
637   if (__buf == 0) {
638     streamsize __bufsize = __n * sizeof(_CharT);
639     //We first check that the streamsize representation can't overflow a size_t one.
640     //If it can, we check that __bufsize is not higher than the size_t max value.
641     if ((sizeof(streamsize) > sizeof(size_t)) &&
642         (__bufsize > __STATIC_CAST(streamsize, (numeric_limits<size_t>::max)())))
643       return false;
644     _M_int_buf = __STATIC_CAST(_CharT*, malloc(__STATIC_CAST(size_t, __bufsize)));
645     if (!_M_int_buf)
646       return false;
647     _M_int_buf_dynamic = true;
648   }
649   else {
650     _M_int_buf = __buf;
651     _M_int_buf_dynamic = false;
652   }
653 
654   streamsize __ebufsiz = (max)(__n * __STATIC_CAST(streamsize, _M_width),
655                                __STATIC_CAST(streamsize, _M_codecvt->max_length()));
656 
657   _M_ext_buf = 0;
658   if ((sizeof(streamsize) < sizeof(size_t)) ||
659       ((sizeof(streamsize) == sizeof(size_t)) && numeric_limits<streamsize>::is_signed) ||
660       (__ebufsiz <= __STATIC_CAST(streamsize, (numeric_limits<size_t>::max)()))) {
661     _M_ext_buf = __STATIC_CAST(char*, malloc(__STATIC_CAST(size_t, __ebufsiz)));
662   }
663 
664   if (!_M_ext_buf) {
665     _M_deallocate_buffers();
666     return false;
667   }
668 
669   _M_int_buf_EOS = _M_int_buf + __STATIC_CAST(ptrdiff_t, __n);
670   _M_ext_buf_EOS = _M_ext_buf + __STATIC_CAST(ptrdiff_t, __ebufsiz);
671   return true;
672 }
673 
674 // Abbreviation for the most common case.
675 template <class _CharT, class _Traits>
_M_allocate_buffers()676 bool basic_filebuf<_CharT, _Traits>::_M_allocate_buffers() {
677   // Choose a buffer that's at least 4096 characters long and that's a
678   // multiple of the page size.
679   streamsize __default_bufsiz =
680     ((_M_base.__page_size() + 4095UL) / _M_base.__page_size()) * _M_base.__page_size();
681   return _M_allocate_buffers(0, __default_bufsiz);
682 }
683 
684 template <class _CharT, class _Traits>
_M_deallocate_buffers()685 void basic_filebuf<_CharT, _Traits>::_M_deallocate_buffers() {
686   if (_M_int_buf_dynamic)
687     free(_M_int_buf);
688   free(_M_ext_buf);
689   _M_int_buf     = 0;
690   _M_int_buf_EOS = 0;
691   _M_ext_buf     = 0;
692   _M_ext_buf_EOS = 0;
693 }
694 
695 
696 //----------------------------------------
697 // Helper functiosn for seek and imbue
698 
699 template <class _CharT, class _Traits>
_M_seek_init(bool __do_unshift)700 bool basic_filebuf<_CharT, _Traits>::_M_seek_init(bool __do_unshift) {
701   // If we're in error mode, leave it.
702    _M_in_error_mode = false;
703 
704   // Flush the output buffer if we're in output mode, and (conditionally)
705   // emit an unshift sequence.
706   if (_M_in_output_mode) {
707     bool __ok = !traits_type::eq_int_type(this->overflow(traits_type::eof()),
708                                           traits_type::eof());
709     if (__do_unshift)
710       __ok = __ok && this->_M_unshift();
711     if (!__ok) {
712       _M_in_output_mode = false;
713       _M_in_error_mode = true;
714       this->setp(0, 0);
715       return false;
716     }
717   }
718 
719   // Discard putback characters, if any.
720   if (_M_in_input_mode && _M_in_putback_mode)
721     _M_exit_putback_mode();
722 
723   return true;
724 }
725 
726 
727 /* Change the filebuf's locale.  This member function has no effect
728  * unless it is called before any I/O is performed on the stream.
729  * This function is called on construction and on an imbue call. In the
730  * case of the construction the codecvt facet might be a custom one if
731  * the basic_filebuf user has instanciate it with a custom char_traits.
732  * The user will have to call imbue before any I/O operation.
733  */
734 template <class _CharT, class _Traits>
_M_setup_codecvt(const locale & __loc,bool __on_imbue)735 void basic_filebuf<_CharT, _Traits>::_M_setup_codecvt(const locale& __loc, bool __on_imbue) {
736   if (has_facet<_Codecvt>(__loc)) {
737     _M_codecvt = &use_facet<_Codecvt>(__loc) ;
738     int __encoding    = _M_codecvt->encoding();
739 
740     _M_width          = (max)(__encoding, 1);
741     _M_max_width      = _M_codecvt->max_length();
742     _M_constant_width = __encoding > 0;
743     _M_always_noconv  = _M_codecvt->always_noconv();
744   }
745   else {
746     _M_codecvt = 0;
747     _M_width = _M_max_width = 1;
748     _M_constant_width = _M_always_noconv  = false;
749     if (__on_imbue) {
750       //This call will generate an exception reporting the problem.
751       use_facet<_Codecvt>(__loc);
752     }
753   }
754 }
755 
756 _STLP_END_NAMESPACE
757 
758 # undef __BF_int_type__
759 # undef __BF_pos_type__
760 # undef __BF_off_type__
761 
762 #endif /* _STLP_FSTREAM_C */
763 
764 // Local Variables:
765 // mode:C++
766 // End:
767