core/StringBuffer.cpp

   1 /* -*- Mode: C++; c-basic-offset: 4; indent-tabs-mode: nil; tab-width: 4 -*- */
   2 /* vi: set ts=4 sw=4 expandtab: (add to ~/.vimrc: set modeline modelines=5) */
   3 /* ***** BEGIN LICENSE BLOCK *****
   4  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   5  *
   6  * The contents of this file are subject to the Mozilla Public License Version
   7  * 1.1 (the "License"); you may not use this file except in compliance with
   8  * the License. You may obtain a copy of the License at
   9  * http://www.mozilla.org/MPL/
  10  *
  11  * Software distributed under the License is distributed on an "AS IS" basis,
  12  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  13  * for the specific language governing rights and limitations under the
  14  * License.
  15  *
  16  * The Original Code is [Open Source Virtual Machine.].
  17  *
  18  * The Initial Developer of the Original Code is
  19  * Adobe System Incorporated.
  20  * Portions created by the Initial Developer are Copyright (C) 2004-2006
  21  * the Initial Developer. All Rights Reserved.
  22  *
  23  * Contributor(s):
  24  *   Adobe AS3 Team
  25  *
  26  * Alternatively, the contents of this file may be used under the terms of
  27  * either the GNU General Public License Version 2 or later (the "GPL"), or
  28  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  29  * in which case the provisions of the GPL or the LGPL are applicable instead
  30  * of those above. If you wish to allow use of your version of this file only
  31  * under the terms of either the GPL or the LGPL, and not to allow others to
  32  * use your version of this file under the terms of the MPL, indicate your
  33  * decision by deleting the provisions above and replace them with the notice
  34  * and other provisions required by the GPL or the LGPL. If you do not delete
  35  * the provisions above, a recipient may use your version of this file under
  36  * the terms of any one of the MPL, the GPL or the LGPL.
  37  *
  38  * ***** END LICENSE BLOCK ***** */
  39
  40 #include "avmplus.h"
  41
  42 namespace avmplus
  43 {
  44     using namespace MMgc;
  45
  46     StringOutputStream::StringOutputStream(GC* gc)
  47     {
  48         m_gc = gc;
  49         m_buffer = (char*) gc->Alloc(kInitialCapacity);
  50         m_buffer[0] = 0;
  51         m_length = 0;
  52     }
  53
  54     StringOutputStream::~StringOutputStream()
  55     {
  56         if (m_gc && m_buffer) {
  57             char* buffer = m_buffer;
  58             m_buffer = 0;   // avoid dangling pointer
  59             m_gc->Free(buffer);
  60         }
  61         m_gc = 0;
  62         m_buffer = 0;   // might have already been cleared above, but that's ok
  63         m_length = 0;
  64     }
  65
  66     void StringOutputStream::write(const char* utf8)
  67     {
  68         writeN(utf8, String::Length(utf8));
  69     }
  70
  71     void StringOutputStream::writeN(const char* buffer, size_t count)
  72     {
  73         if (m_length + count >= (size_t)GC::Size(m_buffer))
  74         {
  75             size_t newCapacity = (m_length+count+1)*2;
  76             char* newBuffer = (char*) m_gc->Alloc(newCapacity);
  77             VMPI_memcpy(newBuffer, m_buffer, m_length);
  78             m_gc->Free(m_buffer);
  79             if (m_gc->IsPointerToGCPage(this))
  80                 WB(m_gc, m_gc->FindBeginningFast(this), &m_buffer, newBuffer);
  81             else
  82                 m_buffer = newBuffer;
  83         }
  84         VMPI_memcpy(m_buffer+m_length, buffer, count);
  85         m_length += (int)count;
  86         m_buffer[m_length] = 0;
  87     }
  88
  89     void StringBuffer::writeN(const char* buffer, size_t count)
  90     {
  91         m_stream.writeN(buffer, count);
  92     }
  93 }