1 // Short-string-optimized versatile string base -*- C++ -*-
2 
3 // Copyright (C) 2005, 2006 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 2, 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 // You should have received a copy of the GNU General Public License along
17 // with this library; see the file COPYING.  If not, write to the Free
18 // Software Foundation, 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
19 // USA.
20 
21 // As a special exception, you may use this file as part of a free software
22 // library without restriction.  Specifically, if other files instantiate
23 // templates or use macros or inline functions from this file, or you compile
24 // this file and link it with other files to produce an executable, this
25 // file does not by itself cause the resulting executable to be covered by
26 // the GNU General Public License.  This exception does not however
27 // invalidate any other reasons why the executable file might be covered by
28 // the GNU General Public License.
29 
30 /** @file ext/sso_string_base.h
31  *  This file is a GNU extension to the Standard C++ Library.
32  *  This is an internal header file, included by other library headers.
33  *  You should not attempt to use it directly.
34  */
35 
36 #ifndef _SSO_STRING_BASE_H
37 #define _SSO_STRING_BASE_H 1
38 
39 _GLIBCXX_BEGIN_NAMESPACE(__gnu_cxx)
40 
41   template<typename _CharT, typename _Traits, typename _Alloc>
42     class __sso_string_base
43     : protected __vstring_utility<_CharT, _Traits, _Alloc>
44     {
45     public:
46       typedef _Traits					    traits_type;
47       typedef typename _Traits::char_type		    value_type;
48 
49       typedef __vstring_utility<_CharT, _Traits, _Alloc>    _Util_Base;
50       typedef typename _Util_Base::_CharT_alloc_type        _CharT_alloc_type;
51       typedef typename _CharT_alloc_type::size_type	    size_type;
52 
53     private:
54       // Data Members:
55       typename _Util_Base::template _Alloc_hider<_CharT_alloc_type>
56                                                             _M_dataplus;
57       size_type                                             _M_string_length;
58 
59       enum { _S_local_capacity = 15 };
60 
61       union
62       {
63 	_CharT           _M_local_data[_S_local_capacity + 1];
64 	size_type        _M_allocated_capacity;
65       };
66 
67       void
68       _M_data(_CharT* __p)
69       { _M_dataplus._M_p = __p; }
70 
71       void
72       _M_length(size_type __length)
73       { _M_string_length = __length; }
74 
75       void
76       _M_capacity(size_type __capacity)
77       { _M_allocated_capacity = __capacity; }
78 
79       bool
80       _M_is_local() const
81       { return _M_data() == _M_local_data; }
82 
83       // Create & Destroy
84       _CharT*
85       _M_create(size_type&, size_type);
86 
87       void
88       _M_dispose()
89       {
90 	if (!_M_is_local())
91 	  _M_destroy(_M_allocated_capacity);
92       }
93 
94       void
95       _M_destroy(size_type __size) throw()
96       { _M_get_allocator().deallocate(_M_data(), __size + 1); }
97 
98       // _M_construct_aux is used to implement the 21.3.1 para 15 which
99       // requires special behaviour if _InIterator is an integral type
100       template<typename _InIterator>
101         void
102         _M_construct_aux(_InIterator __beg, _InIterator __end,
103 			 std::__false_type)
104 	{
105           typedef typename iterator_traits<_InIterator>::iterator_category _Tag;
106           _M_construct(__beg, __end, _Tag());
107 	}
108 
109       template<typename _InIterator>
110         void
111         _M_construct_aux(_InIterator __beg, _InIterator __end,
112 			 std::__true_type)
113 	{ _M_construct(static_cast<size_type>(__beg),
114 		       static_cast<value_type>(__end)); }
115 
116       template<typename _InIterator>
117         void
118         _M_construct(_InIterator __beg, _InIterator __end)
119 	{
120 	  typedef typename std::__is_integer<_InIterator>::__type _Integral;
121 	  _M_construct_aux(__beg, __end, _Integral());
122         }
123 
124       // For Input Iterators, used in istreambuf_iterators, etc.
125       template<typename _InIterator>
126         void
127         _M_construct(_InIterator __beg, _InIterator __end,
128 		     std::input_iterator_tag);
129 
130       // For forward_iterators up to random_access_iterators, used for
131       // string::iterator, _CharT*, etc.
132       template<typename _FwdIterator>
133         void
134         _M_construct(_FwdIterator __beg, _FwdIterator __end,
135 		     std::forward_iterator_tag);
136 
137       void
138       _M_construct(size_type __req, _CharT __c);
139 
140     public:
141       size_type
142       _M_max_size() const
143       { return (_M_get_allocator().max_size() - 1) / 2; }
144 
145       _CharT*
146       _M_data() const
147       { return _M_dataplus._M_p; }
148 
149       size_type
150       _M_length() const
151       { return _M_string_length; }
152 
153       size_type
154       _M_capacity() const
155       {
156 	return _M_is_local() ? size_type(_S_local_capacity)
157 	                     : _M_allocated_capacity;
158       }
159 
160       bool
161       _M_is_shared() const
162       { return false; }
163 
164       void
165       _M_set_leaked() { }
166 
167       void
168       _M_leak() { }
169 
170       void
171       _M_set_length(size_type __n)
172       {
173 	_M_length(__n);
174 	traits_type::assign(_M_data()[__n], _CharT());
175       }
176 
177       __sso_string_base()
178       : _M_dataplus(_Alloc(), _M_local_data)
179       { _M_set_length(0); }
180 
181       __sso_string_base(const _Alloc& __a);
182 
183       __sso_string_base(const __sso_string_base& __rcs);
184 
185       __sso_string_base(size_type __n, _CharT __c, const _Alloc& __a);
186 
187       template<typename _InputIterator>
188         __sso_string_base(_InputIterator __beg, _InputIterator __end,
189 			  const _Alloc& __a);
190 
191       ~__sso_string_base()
192       { _M_dispose(); }
193 
194       _CharT_alloc_type&
195       _M_get_allocator()
196       { return _M_dataplus; }
197 
198       const _CharT_alloc_type&
199       _M_get_allocator() const
200       { return _M_dataplus; }
201 
202       void
203       _M_swap(__sso_string_base& __rcs);
204 
205       void
206       _M_assign(const __sso_string_base& __rcs);
207 
208       void
209       _M_reserve(size_type __res);
210 
211       void
212       _M_mutate(size_type __pos, size_type __len1, const _CharT* __s,
213 		size_type __len2);
214 
215       void
216       _M_erase(size_type __pos, size_type __n);
217 
218       void
219       _M_clear()
220       { _M_set_length(0); }
221 
222       bool
223       _M_compare(const __sso_string_base&) const
224       { return false; }
225     };
226 
227   template<typename _CharT, typename _Traits, typename _Alloc>
228     void
229     __sso_string_base<_CharT, _Traits, _Alloc>::
230     _M_swap(__sso_string_base& __rcs)
231     {
232       // _GLIBCXX_RESOLVE_LIB_DEFECTS
233       // 431. Swapping containers with unequal allocators.
234       std::__alloc_swap<_CharT_alloc_type>::_S_do_it(_M_get_allocator(),
235 						     __rcs._M_get_allocator());
236 
237       if (_M_is_local())
238 	if (__rcs._M_is_local())
239 	  {
240 	    if (_M_length() && __rcs._M_length())
241 	      {
242 		_CharT __tmp_data[_S_local_capacity + 1];
243 		traits_type::copy(__tmp_data, __rcs._M_local_data,
244 				  _S_local_capacity + 1);
245 		traits_type::copy(__rcs._M_local_data, _M_local_data,
246 				  _S_local_capacity + 1);
247 		traits_type::copy(_M_local_data, __tmp_data,
248 				  _S_local_capacity + 1);
249 	      }
250 	    else if (__rcs._M_length())
251 	      {
252 		traits_type::copy(_M_local_data, __rcs._M_local_data,
253 				  _S_local_capacity + 1);
254 		_M_length(__rcs._M_length());
255 		__rcs._M_set_length(0);
256 		return;
257 	      }
258 	    else if (_M_length())
259 	      {
260 		traits_type::copy(__rcs._M_local_data, _M_local_data,
261 				  _S_local_capacity + 1);
262 		__rcs._M_length(_M_length());
263 		_M_set_length(0);
264 		return;
265 	      }
266 	  }
267 	else
268 	  {
269 	    const size_type __tmp_capacity = __rcs._M_allocated_capacity;
270 	    traits_type::copy(__rcs._M_local_data, _M_local_data,
271 			      _S_local_capacity + 1);
272 	    _M_data(__rcs._M_data());
273 	    __rcs._M_data(__rcs._M_local_data);
274 	    _M_capacity(__tmp_capacity);
275 	  }
276       else
277 	{
278 	  const size_type __tmp_capacity = _M_allocated_capacity;
279 	  if (__rcs._M_is_local())
280 	    {
281 	      traits_type::copy(_M_local_data, __rcs._M_local_data,
282 				_S_local_capacity + 1);
283 	      __rcs._M_data(_M_data());
284 	      _M_data(_M_local_data);
285 	    }
286 	  else
287 	    {
288 	      _CharT* __tmp_ptr = _M_data();
289 	      _M_data(__rcs._M_data());
290 	      __rcs._M_data(__tmp_ptr);
291 	      _M_capacity(__rcs._M_allocated_capacity);
292 	    }
293 	  __rcs._M_capacity(__tmp_capacity);
294 	}
295 
296       const size_type __tmp_length = _M_length();
297       _M_length(__rcs._M_length());
298       __rcs._M_length(__tmp_length);
299     }
300 
301   template<typename _CharT, typename _Traits, typename _Alloc>
302     _CharT*
303     __sso_string_base<_CharT, _Traits, _Alloc>::
304     _M_create(size_type& __capacity, size_type __old_capacity)
305     {
306       // _GLIBCXX_RESOLVE_LIB_DEFECTS
307       // 83.  String::npos vs. string::max_size()
308       if (__capacity > _M_max_size())
309 	std::__throw_length_error(__N("__sso_string_base::_M_create"));
310 
311       // The below implements an exponential growth policy, necessary to
312       // meet amortized linear time requirements of the library: see
313       // http://gcc.gnu.org/ml/libstdc++/2001-07/msg00085.html.
314       if (__capacity > __old_capacity && __capacity < 2 * __old_capacity)
315 	{
316 	  __capacity = 2 * __old_capacity;
317 	  // Never allocate a string bigger than max_size.
318 	  if (__capacity > _M_max_size())
319 	    __capacity = _M_max_size();
320 	}
321 
322       // NB: Need an array of char_type[__capacity], plus a terminating
323       // null char_type() element.
324       return _M_get_allocator().allocate(__capacity + 1);
325     }
326 
327   template<typename _CharT, typename _Traits, typename _Alloc>
328     __sso_string_base<_CharT, _Traits, _Alloc>::
329     __sso_string_base(const _Alloc& __a)
330     : _M_dataplus(__a, _M_local_data)
331     { _M_set_length(0); }
332 
333   template<typename _CharT, typename _Traits, typename _Alloc>
334     __sso_string_base<_CharT, _Traits, _Alloc>::
335     __sso_string_base(const __sso_string_base& __rcs)
336     : _M_dataplus(__rcs._M_get_allocator(), _M_local_data)
337     { _M_construct(__rcs._M_data(), __rcs._M_data() + __rcs._M_length()); }
338 
339   template<typename _CharT, typename _Traits, typename _Alloc>
340     __sso_string_base<_CharT, _Traits, _Alloc>::
341     __sso_string_base(size_type __n, _CharT __c, const _Alloc& __a)
342     : _M_dataplus(__a, _M_local_data)
343     { _M_construct(__n, __c); }
344 
345   template<typename _CharT, typename _Traits, typename _Alloc>
346     template<typename _InputIterator>
347     __sso_string_base<_CharT, _Traits, _Alloc>::
348     __sso_string_base(_InputIterator __beg, _InputIterator __end,
349 		      const _Alloc& __a)
350     : _M_dataplus(__a, _M_local_data)
351     { _M_construct(__beg, __end); }
352 
353   // NB: This is the special case for Input Iterators, used in
354   // istreambuf_iterators, etc.
355   // Input Iterators have a cost structure very different from
356   // pointers, calling for a different coding style.
357   template<typename _CharT, typename _Traits, typename _Alloc>
358     template<typename _InIterator>
359       void
360       __sso_string_base<_CharT, _Traits, _Alloc>::
361       _M_construct(_InIterator __beg, _InIterator __end,
362 		   std::input_iterator_tag)
363       {
364 	size_type __len = 0;
365 	size_type __capacity = size_type(_S_local_capacity);
366 
367 	while (__beg != __end && __len < __capacity)
368 	  {
369 	    _M_data()[__len++] = *__beg;
370 	    ++__beg;
371 	  }
372 
373 	try
374 	  {
375 	    while (__beg != __end)
376 	      {
377 		if (__len == __capacity)
378 		  {
379 		    // Allocate more space.
380 		    __capacity = __len + 1;
381 		    _CharT* __another = _M_create(__capacity, __len);
382 		    _S_copy(__another, _M_data(), __len);
383 		    _M_dispose();
384 		    _M_data(__another);
385 		    _M_capacity(__capacity);
386 		  }
387 		_M_data()[__len++] = *__beg;
388 		++__beg;
389 	      }
390 	  }
391 	catch(...)
392 	  {
393 	    _M_dispose();
394 	    __throw_exception_again;
395 	  }
396 
397 	_M_set_length(__len);
398       }
399 
400   template<typename _CharT, typename _Traits, typename _Alloc>
401     template<typename _InIterator>
402       void
403       __sso_string_base<_CharT, _Traits, _Alloc>::
404       _M_construct(_InIterator __beg, _InIterator __end,
405 		   std::forward_iterator_tag)
406       {
407 	// NB: Not required, but considered best practice.
408 	if (__builtin_expect(_S_is_null_pointer(__beg) && __beg != __end, 0))
409 	  std::__throw_logic_error(__N("__sso_string_base::"
410 				       "_M_construct NULL not valid"));
411 
412 	size_type __dnew = static_cast<size_type>(std::distance(__beg, __end));
413 
414 	if (__dnew > size_type(_S_local_capacity))
415 	  {
416 	    _M_data(_M_create(__dnew, size_type(0)));
417 	    _M_capacity(__dnew);
418 	  }
419 
420 	// Check for out_of_range and length_error exceptions.
421 	try
422 	  { _S_copy_chars(_M_data(), __beg, __end); }
423 	catch(...)
424 	  {
425 	    _M_dispose();
426 	    __throw_exception_again;
427 	  }
428 
429 	_M_set_length(__dnew);
430       }
431 
432   template<typename _CharT, typename _Traits, typename _Alloc>
433     void
434     __sso_string_base<_CharT, _Traits, _Alloc>::
435     _M_construct(size_type __n, _CharT __c)
436     {
437       if (__n > size_type(_S_local_capacity))
438 	{
439 	  _M_data(_M_create(__n, size_type(0)));
440 	  _M_capacity(__n);
441 	}
442 
443       if (__n)
444 	_S_assign(_M_data(), __n, __c);
445 
446       _M_set_length(__n);
447     }
448 
449   template<typename _CharT, typename _Traits, typename _Alloc>
450     void
451     __sso_string_base<_CharT, _Traits, _Alloc>::
452     _M_assign(const __sso_string_base& __rcs)
453     {
454       if (this != &__rcs)
455 	{
456 	  const size_type __rsize = __rcs._M_length();
457 	  const size_type __capacity = _M_capacity();
458 
459 	  if (__rsize > __capacity)
460 	    {
461 	      size_type __new_capacity = __rsize;
462 	      _CharT* __tmp = _M_create(__new_capacity, __capacity);
463 	      _M_dispose();
464 	      _M_data(__tmp);
465 	      _M_capacity(__new_capacity);
466 	    }
467 
468 	  if (__rsize)
469 	    _S_copy(_M_data(), __rcs._M_data(), __rsize);
470 
471 	  _M_set_length(__rsize);
472 	}
473     }
474 
475   template<typename _CharT, typename _Traits, typename _Alloc>
476     void
477     __sso_string_base<_CharT, _Traits, _Alloc>::
478     _M_reserve(size_type __res)
479     {
480       // Make sure we don't shrink below the current size.
481       if (__res < _M_length())
482 	__res = _M_length();
483 
484       const size_type __capacity = _M_capacity();
485       if (__res != __capacity)
486 	{
487 	  if (__res > __capacity
488 	      || __res > size_type(_S_local_capacity))
489 	    {
490 	      _CharT* __tmp = _M_create(__res, __capacity);
491 	      _S_copy(__tmp, _M_data(), _M_length() + 1);
492 	      _M_dispose();
493 	      _M_data(__tmp);
494 	      _M_capacity(__res);
495 	    }
496 	  else if (!_M_is_local())
497 	    {
498 	      _S_copy(_M_local_data, _M_data(), _M_length() + 1);
499 	      _M_destroy(__capacity);
500 	      _M_data(_M_local_data);
501 	    }
502 	}
503     }
504 
505   template<typename _CharT, typename _Traits, typename _Alloc>
506     void
507     __sso_string_base<_CharT, _Traits, _Alloc>::
508     _M_mutate(size_type __pos, size_type __len1, const _CharT* __s,
509 	      const size_type __len2)
510     {
511       const size_type __how_much = _M_length() - __pos - __len1;
512 
513       size_type __new_capacity = _M_length() + __len2 - __len1;
514       _CharT* __r = _M_create(__new_capacity, _M_capacity());
515 
516       if (__pos)
517 	_S_copy(__r, _M_data(), __pos);
518       if (__s && __len2)
519 	_S_copy(__r + __pos, __s, __len2);
520       if (__how_much)
521 	_S_copy(__r + __pos + __len2,
522 		_M_data() + __pos + __len1, __how_much);
523 
524       _M_dispose();
525       _M_data(__r);
526       _M_capacity(__new_capacity);
527     }
528 
529   template<typename _CharT, typename _Traits, typename _Alloc>
530     void
531     __sso_string_base<_CharT, _Traits, _Alloc>::
532     _M_erase(size_type __pos, size_type __n)
533     {
534       const size_type __how_much = _M_length() - __pos - __n;
535 
536       if (__how_much && __n)
537 	_S_move(_M_data() + __pos, _M_data() + __pos + __n,
538 		__how_much);
539 
540       _M_set_length(_M_length() - __n);
541     }
542 
543   template<>
544     inline bool
545     __sso_string_base<char, std::char_traits<char>,
546 		      std::allocator<char> >::
547     _M_compare(const __sso_string_base& __rcs) const
548     {
549       if (this == &__rcs)
550 	return true;
551       return false;
552     }
553 
554 #ifdef _GLIBCXX_USE_WCHAR_T
555   template<>
556     inline bool
557     __sso_string_base<wchar_t, std::char_traits<wchar_t>,
558 		      std::allocator<wchar_t> >::
559     _M_compare(const __sso_string_base& __rcs) const
560     {
561       if (this == &__rcs)
562 	return true;
563       return false;
564     }
565 #endif
566 
567 _GLIBCXX_END_NAMESPACE
568 
569 #endif /* _SSO_STRING_BASE_H */
570