From d5549ccac174725e54d1be940a5404b70ced559a Mon Sep 17 00:00:00 2001 From: Marshall Clow Date: Thu, 17 Jul 2014 15:32:20 +0000 Subject: [PATCH] Fix bug #20335 - memory leak when move-constructing a string with unequal allocator. Thanks to Thomas Koeppe for the report git-svn-id: https://llvm.org/svn/llvm-project/libcxx/trunk@213269 91177308-0d34-0410-b5e6-96231b3b80d8 --- include/string | 30 ++++++++++--------- .../string.cons/move_alloc.pass.cpp | 10 +++++++ 2 files changed, 26 insertions(+), 14 deletions(-) diff --git a/include/string b/include/string index d07ff289..d6c148ac 100644 --- a/include/string +++ b/include/string @@ -1031,16 +1031,16 @@ __str_rfind(const _CharT *__p, _SizeT __sz, _CharT __c, _SizeT __pos) _NOEXCEPT { if (__sz < 1) - return __npos; - if (__pos < __sz) - ++__pos; - else - __pos = __sz; - for (const _CharT* __ps = __p + __pos; __ps != __p;) - { - if (_Traits::eq(*--__ps, __c)) - return static_cast<_SizeT>(__ps - __p); - } + return __npos; + if (__pos < __sz) + ++__pos; + else + __pos = __sz; + for (const _CharT* __ps = __p + __pos; __ps != __p;) + { + if (_Traits::eq(*--__ps, __c)) + return static_cast<_SizeT>(__ps - __p); + } return __npos; } @@ -2136,11 +2136,13 @@ inline _LIBCPP_INLINE_VISIBILITY basic_string<_CharT, _Traits, _Allocator>::basic_string(basic_string&& __str, const allocator_type& __a) : __r_(__a) { - if (__a == __str.__alloc() || !__str.__is_long()) - __r_.first().__r = __str.__r_.first().__r; - else + if (__str.__is_long() && __a != __str.__alloc()) // copy, not move __init(_VSTD::__to_raw_pointer(__str.__get_long_pointer()), __str.__get_long_size()); - __str.__zero(); + else + { + __r_.first().__r = __str.__r_.first().__r; + __str.__zero(); + } #if _LIBCPP_DEBUG_LEVEL >= 2 __get_db()->__insert_c(this); if (__is_long()) diff --git a/test/strings/basic.string/string.cons/move_alloc.pass.cpp b/test/strings/basic.string/string.cons/move_alloc.pass.cpp index 4e8b038e..1f963144 100644 --- a/test/strings/basic.string/string.cons/move_alloc.pass.cpp +++ b/test/strings/basic.string/string.cons/move_alloc.pass.cpp @@ -45,6 +45,16 @@ int main() test(S("1"), A(5)); test(S("1234567890123456789012345678901234567890123456789012345678901234567890"), A(7)); } + + int alloc_count = test_alloc_base::alloc_count; + { + typedef test_allocator A; + typedef std::basic_string, A> S; + S s1 ( "Twas brillig, and the slivy toves did gyre and gymbal in the wabe" ); + S s2 (std::move(s1), A(1)); + } + assert ( test_alloc_base::alloc_count == alloc_count ); + #if __cplusplus >= 201103L { typedef min_allocator A;