strbuf.hxx

C++ 编写的EROS RTOS

#ifndef __STRBUF_HXX__
#define __STRBUF_HXX__

/*
 * Copyright (C) 2001, Jonathan S. Shapiro.
 *
 * This file is part of the EROS Operating System.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2,
 * or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

#include <eros/target.h>

class StrBuf {
  char *s;
  uint32_t limit;
  uint32_t m_len;
  
  void Grow()
    {
      char *news = new char[limit + 4096];
      limit += 4096;
      strcpy(s, news);
      delete [] s;
      s = news;
    }
public:
  StrBuf()
    {
      s = new char[4096];
      limit = 4096;
      m_len = 0;
    }

  StrBuf& operator << (char c)
    {
      if (m_len + 1 >= limit)
	Grow();

      s[m_len++] = c;
      s[m_len] = 0;
      return *this;
    }

  StrBuf& operator << (int i)
    {
      char str[20];
      sprintf(str, "%d", i);
      int str_len = strlen(str);
      
      while (m_len + str_len >= limit)
	Grow();

      strcat(s, str);
      m_len += str_len;
      return *this;
    }

  StrBuf& operator << (unsigned u)
    {
      char str[20];
      sprintf(str, "%u", u);
      int str_len = strlen(str);
      
      while (m_len + str_len >= limit)
	Grow();

      strcat(s, str);
      m_len += str_len;
      return *this;
    }

  StrBuf& operator << (const char *str)
    {
      uint32_t str_len = strlen(str);
      while (m_len + str_len >= limit)
	Grow();

      strcat(s, str);
      m_len += str_len;
      return *this;
    }

  StrBuf& operator << (const InternedString& is)
    {
      (*this) << is.str();
      return *this;
    }

  const char *str() const
    {
      return s;
    }

  int len() const
    {
      return m_len;
    }

  StrBuf& operator << (const StrBuf& sb)
    {
      (*this) << sb.str();
      return *this;
    }
};

#endif /* __STRBUF_HXX__ */