StringBuffer.h 3.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. /*
  2. * Copyright (C) 2008, 2010 Apple Inc. All rights reserved.
  3. *
  4. * Redistribution and use in source and binary forms, with or without
  5. * modification, are permitted provided that the following conditions
  6. * are met:
  7. *
  8. * 1. Redistributions of source code must retain the above copyright
  9. * notice, this list of conditions and the following disclaimer.
  10. * 2. Redistributions in binary form must reproduce the above copyright
  11. * notice, this list of conditions and the following disclaimer in the
  12. * documentation and/or other materials provided with the distribution.
  13. * 3. Neither the name of Apple Inc. ("Apple") nor the names of its
  14. * contributors may be used to endorse or promote products derived
  15. * from this software without specific prior written permission.
  16. *
  17. * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
  18. * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  19. * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  20. * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
  21. * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  22. * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  23. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  24. * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  26. * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27. */
  28. #ifndef StringBuffer_h
  29. #define StringBuffer_h
  30. #include <wtf/Assertions.h>
  31. #include <wtf/unicode/Unicode.h>
  32. #include <limits>
  33. namespace WTF {
  34. template <typename CharType>
  35. class StringBuffer {
  36. WTF_MAKE_NONCOPYABLE(StringBuffer);
  37. public:
  38. explicit StringBuffer(unsigned length)
  39. : m_length(length)
  40. {
  41. if (m_length > std::numeric_limits<unsigned>::max() / sizeof(CharType))
  42. CRASH();
  43. m_data = static_cast<CharType*>(fastMalloc(m_length * sizeof(CharType)));
  44. }
  45. ~StringBuffer()
  46. {
  47. fastFree(m_data);
  48. }
  49. void shrink(unsigned newLength)
  50. {
  51. ASSERT(newLength <= m_length);
  52. m_length = newLength;
  53. }
  54. void resize(unsigned newLength)
  55. {
  56. if (newLength > m_length) {
  57. if (newLength > std::numeric_limits<unsigned>::max() / sizeof(UChar))
  58. CRASH();
  59. m_data = static_cast<UChar*>(fastRealloc(m_data, newLength * sizeof(UChar)));
  60. }
  61. m_length = newLength;
  62. }
  63. unsigned length() const { return m_length; }
  64. CharType* characters() { return m_data; }
  65. CharType& operator[](unsigned i) { ASSERT_WITH_SECURITY_IMPLICATION(i < m_length); return m_data[i]; }
  66. CharType* release() { CharType* data = m_data; m_data = 0; return data; }
  67. private:
  68. unsigned m_length;
  69. CharType* m_data;
  70. };
  71. } // namespace WTF
  72. using WTF::StringBuffer;
  73. #endif // StringBuffer_h