1 /*
2 * Copyright (c) 1999
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
19 // Implementation of the classes in header <strstream>.
20 // WARNING: The classes defined in <strstream> are DEPRECATED. This
21 // header is defined in section D.7.1 of the C++ standard, and it
22 // MAY BE REMOVED in a future standard revision. You should use the
23 // header <sstream> instead.
24
25 #include "stlport_prefix.h"
26
27 #include <strstream>
28 #include <algorithm>
29 #include <limits>
30
31 _STLP_BEGIN_NAMESPACE
32
33 // strstreambuf constructor, destructor.
strstreambuf(streamsize initial_capacity)34 strstreambuf::strstreambuf(streamsize initial_capacity)
35 : _M_alloc_fun(0), _M_free_fun(0),
36 _M_dynamic(true), _M_frozen(false), _M_constant(false) {
37 size_t n = (sizeof(streamsize) > sizeof(size_t)) ? __STATIC_CAST(size_t, (min)(__STATIC_CAST(streamsize, (numeric_limits<size_t>::max)()),
38 (max)(initial_capacity, streamsize(16))))
39 : __STATIC_CAST(size_t, (max)(initial_capacity, streamsize(16)));
40
41 char* buf = _M_alloc(n);
42 if (buf) {
43 setp(buf, buf + n);
44 setg(buf, buf, buf);
45 }
46 }
47
strstreambuf(__alloc_fn alloc_f,__free_fn free_f)48 strstreambuf::strstreambuf(__alloc_fn alloc_f, __free_fn free_f)
49 : _M_alloc_fun(alloc_f), _M_free_fun(free_f),
50 _M_dynamic(true), _M_frozen(false), _M_constant(false) {
51 size_t n = 16;
52
53 char* buf = _M_alloc(n);
54 if (buf) {
55 setp(buf, buf + n);
56 setg(buf, buf, buf);
57 }
58 }
59
strstreambuf(char * get,streamsize n,char * put)60 strstreambuf::strstreambuf(char* get, streamsize n, char* put)
61 : _M_alloc_fun(0), _M_free_fun(0),
62 _M_dynamic(false), _M_frozen(false), _M_constant(false) {
63 _M_setup(get, put, n);
64 }
65
strstreambuf(signed char * get,streamsize n,signed char * put)66 strstreambuf::strstreambuf(signed char* get, streamsize n, signed char* put)
67 : _M_alloc_fun(0), _M_free_fun(0),
68 _M_dynamic(false), _M_frozen(false), _M_constant(false) {
69 _M_setup(__REINTERPRET_CAST(char*,get), __REINTERPRET_CAST(char*,put), n);
70 }
71
strstreambuf(unsigned char * get,streamsize n,unsigned char * put)72 strstreambuf::strstreambuf(unsigned char* get, streamsize n,
73 unsigned char* put)
74 : _M_alloc_fun(0), _M_free_fun(0),
75 _M_dynamic(false), _M_frozen(false), _M_constant(false) {
76 _M_setup(__REINTERPRET_CAST(char*,get), __REINTERPRET_CAST(char*,put), n);
77 }
78
strstreambuf(const char * get,streamsize n)79 strstreambuf::strstreambuf(const char* get, streamsize n)
80 : _M_alloc_fun(0), _M_free_fun(0),
81 _M_dynamic(false), _M_frozen(false), _M_constant(true) {
82 _M_setup(__CONST_CAST(char*,get), 0, n);
83 }
84
strstreambuf(const signed char * get,streamsize n)85 strstreambuf::strstreambuf(const signed char* get, streamsize n)
86 : _M_alloc_fun(0), _M_free_fun(0),
87 _M_dynamic(false), _M_frozen(false), _M_constant(true) {
88 _M_setup(__REINTERPRET_CAST(char*, __CONST_CAST(signed char*,get)), 0, n);
89 }
90
strstreambuf(const unsigned char * get,streamsize n)91 strstreambuf::strstreambuf(const unsigned char* get, streamsize n)
92 : _M_alloc_fun(0), _M_free_fun(0),
93 _M_dynamic(false), _M_frozen(false), _M_constant(true) {
94 _M_setup(__REINTERPRET_CAST(char*, __CONST_CAST(unsigned char*,get)), 0, n);
95 }
96
~strstreambuf()97 strstreambuf::~strstreambuf() {
98 if (_M_dynamic && !_M_frozen)
99 _M_free(eback());
100 }
101
freeze(bool frozenflag)102 void strstreambuf::freeze(bool frozenflag) {
103 if (_M_dynamic)
104 _M_frozen = frozenflag;
105 }
106
str()107 char* strstreambuf::str() {
108 freeze(true);
109 return eback();
110 }
111
pcount() const112 int strstreambuf::pcount() const {
113 return int(pptr() ? pptr() - pbase() : 0);
114 }
115
overflow(int_type c)116 strstreambuf::int_type strstreambuf::overflow(int_type c) {
117 if (c == traits_type::eof())
118 return traits_type::not_eof(c);
119
120 // Try to expand the buffer.
121 if (pptr() == epptr() && _M_dynamic && !_M_frozen && !_M_constant) {
122 ptrdiff_t old_size = epptr() - pbase();
123 ptrdiff_t new_size = (max)(2 * old_size, ptrdiff_t(1));
124
125 char* buf = _M_alloc(new_size);
126 if (buf) {
127 memcpy(buf, pbase(), old_size);
128
129 char* old_buffer = pbase();
130 bool reposition_get = false;
131 ptrdiff_t old_get_offset;
132 if (gptr() != 0) {
133 reposition_get = true;
134 old_get_offset = gptr() - eback();
135 }
136
137 setp(buf, buf + new_size);
138 pbump((int)old_size);
139
140 if (reposition_get)
141 setg(buf, buf + old_get_offset, buf + (max)(old_get_offset, old_size));
142
143 _M_free(old_buffer);
144 }
145 }
146
147 if (pptr() != epptr()) {
148 *pptr() = traits_type::to_char_type(c);
149 pbump(1);
150 return c;
151 }
152 else
153 return traits_type::eof();
154 }
155
pbackfail(int_type c)156 strstreambuf::int_type strstreambuf::pbackfail(int_type c) {
157 if (gptr() != eback()) {
158 if (c == traits_type::eof()) {
159 gbump(-1);
160 return traits_type::not_eof(c);
161 }
162 else if (c == gptr()[-1]) {
163 gbump(-1);
164 return c;
165 }
166 else if (!_M_constant) {
167 gbump(-1);
168 *gptr() = traits_type::to_char_type(c);
169 return c;
170 }
171 }
172
173 return traits_type::eof();
174 }
175
underflow()176 strstreambuf::int_type strstreambuf::underflow() {
177 if (gptr() == egptr() && pptr() && pptr() > egptr())
178 setg(eback(), gptr(), pptr());
179
180 if (gptr() != egptr())
181 return (unsigned char) *gptr();
182 else
183 return _Traits::eof();
184 }
185
186 basic_streambuf<char, char_traits<char> >*
setbuf(char *,streamsize)187 strstreambuf::setbuf(char*, streamsize) {
188 return this;
189 }
190
191 strstreambuf::pos_type
seekoff(off_type off,ios_base::seekdir dir,ios_base::openmode mode)192 strstreambuf::seekoff(off_type off,
193 ios_base::seekdir dir, ios_base::openmode mode) {
194 bool do_get = false;
195 bool do_put = false;
196
197 if ((mode & (ios_base::in | ios_base::out)) ==
198 (ios_base::in | ios_base::out) &&
199 (dir == ios_base::beg || dir == ios_base::end))
200 do_get = do_put = true;
201 else if (mode & ios_base::in)
202 do_get = true;
203 else if (mode & ios_base::out)
204 do_put = true;
205
206 // !gptr() is here because, according to D.7.1 paragraph 4, the seekable
207 // area is undefined if there is no get area.
208 if ((!do_get && !do_put) || (do_put && !pptr()) || !gptr())
209 return pos_type(off_type(-1));
210
211 char* seeklow = eback();
212 char* seekhigh = epptr() ? epptr() : egptr();
213
214 off_type newoff;
215 switch(dir) {
216 case ios_base::beg:
217 newoff = 0;
218 break;
219 case ios_base::end:
220 newoff = seekhigh - seeklow;
221 break;
222 case ios_base::cur:
223 newoff = do_put ? pptr() - seeklow : gptr() - seeklow;
224 break;
225 default:
226 return pos_type(off_type(-1));
227 }
228
229 off += newoff;
230 if (off < 0 || off > seekhigh - seeklow)
231 return pos_type(off_type(-1));
232
233 if (do_put) {
234 if (seeklow + __STATIC_CAST(ptrdiff_t, off) < pbase()) {
235 setp(seeklow, epptr());
236 pbump((int)off);
237 }
238 else {
239 setp(pbase(), epptr());
240 pbump((int)(off - (pbase() - seeklow)));
241 }
242 }
243 if (do_get) {
244 if (off <= egptr() - seeklow)
245 setg(seeklow, seeklow + __STATIC_CAST(ptrdiff_t, off), egptr());
246 else if (off <= pptr() - seeklow)
247 setg(seeklow, seeklow + __STATIC_CAST(ptrdiff_t, off), pptr());
248 else
249 setg(seeklow, seeklow + __STATIC_CAST(ptrdiff_t, off), epptr());
250 }
251
252 return pos_type(newoff);
253 }
254
255 strstreambuf::pos_type
seekpos(pos_type pos,ios_base::openmode mode)256 strstreambuf::seekpos(pos_type pos, ios_base::openmode mode) {
257 return seekoff(pos - pos_type(off_type(0)), ios_base::beg, mode);
258 }
259
260
_M_alloc(size_t n)261 char* strstreambuf::_M_alloc(size_t n) {
262 if (_M_alloc_fun)
263 return __STATIC_CAST(char*,_M_alloc_fun(n));
264 else
265 return _CHECK_PTR(new char[n]);
266 }
267
_M_free(char * p)268 void strstreambuf::_M_free(char* p) {
269 if (p)
270 if (_M_free_fun)
271 _M_free_fun(p);
272 else
273 delete[] p;
274 }
275
_M_setup(char * get,char * put,streamsize n)276 void strstreambuf::_M_setup(char* get, char* put, streamsize n) {
277 if (get) {
278 size_t N = n > 0 ? size_t(n) : n == 0 ? strlen(get) : size_t(INT_MAX);
279
280 if (put) {
281 setg(get, get, get + N);
282 setp(put, put + N);
283 }
284 else {
285 setg(get, get, get + N);
286 }
287 }
288 }
289
290 //----------------------------------------------------------------------
291 // Class istrstream
292
istrstream(char * s)293 istrstream::istrstream(char* s)
294 : basic_istream<char, char_traits<char> >(0), _M_buf(s, 0) {
295 this->init(&_M_buf);
296 }
297
istrstream(const char * s)298 istrstream::istrstream(const char* s)
299 : basic_istream<char, char_traits<char> >(0), _M_buf(s, 0) {
300 this->init(&_M_buf);
301 }
302
istrstream(char * s,streamsize n)303 istrstream::istrstream(char* s, streamsize n)
304 : basic_istream<char, char_traits<char> >(0), _M_buf(s, n) {
305 this->init(&_M_buf);
306 }
307
istrstream(const char * s,streamsize n)308 istrstream::istrstream(const char* s, streamsize n)
309 : basic_istream<char, char_traits<char> >(0), _M_buf(s, n) {
310 this->init(&_M_buf);
311 }
312
~istrstream()313 istrstream::~istrstream() {}
314
rdbuf() const315 strstreambuf* istrstream::rdbuf() const {
316 return __CONST_CAST(strstreambuf*,&_M_buf);
317 }
318
str()319 char* istrstream::str() { return _M_buf.str(); }
320
321 //----------------------------------------------------------------------
322 // Class ostrstream
323
ostrstream()324 ostrstream::ostrstream()
325 : basic_ostream<char, char_traits<char> >(0), _M_buf() {
326 basic_ios<char, char_traits<char> >::init(&_M_buf);
327 }
328
ostrstream(char * s,int n,ios_base::openmode mode)329 ostrstream::ostrstream(char* s, int n, ios_base::openmode mode)
330 : basic_ostream<char, char_traits<char> >(0),
331 _M_buf(s, n, mode & ios_base::app ? s + strlen(s) : s) {
332 basic_ios<char, char_traits<char> >::init(&_M_buf);
333 }
334
~ostrstream()335 ostrstream::~ostrstream() {}
336
rdbuf() const337 strstreambuf* ostrstream::rdbuf() const {
338 return __CONST_CAST(strstreambuf*,&_M_buf);
339 }
340
freeze(bool freezeflag)341 void ostrstream::freeze(bool freezeflag) {
342 _M_buf.freeze(freezeflag);
343 }
344
str()345 char* ostrstream::str() {
346 return _M_buf.str();
347 }
348
pcount() const349 int ostrstream::pcount() const {
350 return _M_buf.pcount();
351 }
352
353
354 //----------------------------------------------------------------------
355 // Class strstream
356
strstream()357 strstream::strstream()
358 : basic_iostream<char, char_traits<char> >(0), _M_buf() {
359 basic_ios<char, char_traits<char> >::init(&_M_buf);
360 }
361
strstream(char * s,int n,ios_base::openmode mode)362 strstream::strstream(char* s, int n, ios_base::openmode mode)
363 : basic_iostream<char, char_traits<char> >(0),
364 _M_buf(s, n, mode & ios_base::app ? s + strlen(s) : s) {
365 basic_ios<char, char_traits<char> >::init(&_M_buf);
366 }
367
~strstream()368 strstream::~strstream() {}
369
rdbuf() const370 strstreambuf* strstream::rdbuf() const {
371 return __CONST_CAST(strstreambuf*,&_M_buf);
372 }
373
freeze(bool freezeflag)374 void strstream::freeze(bool freezeflag) {
375 _M_buf.freeze(freezeflag);
376 }
377
pcount() const378 int strstream::pcount() const {
379 return _M_buf.pcount();
380 }
381
str()382 char* strstream::str() {
383 return _M_buf.str();
384 }
385
386 _STLP_END_NAMESPACE
387
388 // Local Variables:
389 // mode:C++
390 // End:
391