server/include/dynbuf.h

#ifndef DYNBUF_H__
#define DYNBUF_H__

#include <cstring>
#include <cassert>

#include "shstr.h"

// this is a "buffer" that can grow fast
// and is still somewhat space-efficient.
// unlike obstacks or other data structures,
// it never moves data around. basically,
// this is a fast strstream without the overhead.

struct dynbuf
{
protected:
  struct chunk
  {
    chunk *next;
    int alloc;
    int size;
    char data[0];
  };

  char *ptr, *end;
  int ext;
  int _size;

  chunk *first, *last;

  void _reserve (int size);
  void clear ();
  void finish ();

public:
  
  dynbuf (int initial = 4096, int extend = 16384);
  ~dynbuf ();

  int  size  () const { return _size + (ptr - last->data); }
  bool empty () const { return !size (); }

  void linearise (void *data);
  char *linearise (); // does not 0-terminate(!)

  int room () const { return end - ptr; }

  char *force (int size)
  {
    if (ptr + size >= end)
      _reserve (size);

    return ptr;
  }

  char *alloc (int size)
  {
    char *res = force (size);
    ptr  += size;
    return res;
  }

  void fadd (char c) { *ptr++ = c; }
  void fadd (unsigned char c) { fadd (char (c)); }

  void add (const void *p, int len)
  {
    memcpy (alloc (len), p, len);
  }

  void add (char c)
  {
    alloc (1)[0] = c;
  }

  void add (const char *s)
  {
     add (s, strlen (s));
  }

  void add (const shstr &s)
  {
    add (s.s, s.length ());
  }

  //TODO
  //void add_destructive (dynbuf &buf);

  dynbuf &operator << (char c) { add (c); return *this; }
  dynbuf &operator << (unsigned char c) { return *this << char (c); }
  dynbuf &operator << (const char *s) { add (s); return *this; }
  dynbuf &operator << (const shstr &s) { add (s); return *this; }
  dynbuf &operator << (const std::string &s) { add (s.data(), s.size ()); return *this; }

  operator std::string ();
};

struct dynbuf_text : dynbuf
{
  dynbuf_text (int initial = 4096, int extend = 16384)
  : dynbuf (initial, extend)
  { }

  using dynbuf::add;

  static const int max_sint32_size = 11;
  static const int max_sint64_size = 20;

  void add (sint32 i);
  void add (sint64 i);

  void printf (const char *format, ...);
};

#endif

Revision:	1.6
Committed:	Mon Apr 30 17:39:58 2007 UTC (17 years, 1 month ago) by root
Content type:	text/plain
Branch:	MAIN
Changes since 1.5:	+11 -10 lines
Log Message:	better dynbuf, some bugfixes, just goofign around, still broken
#	User	Rev	Content
1	root	1.1	#ifndef DYNBUF_H__
2			#define DYNBUF_H__
3
4			#include <cstring>
5			#include <cassert>
6
7	root	1.4	#include "shstr.h"
8
9	root	1.1	// this is a "buffer" that can grow fast
10			// and is still somewhat space-efficient.
11			// unlike obstacks or other data structures,
12			// it never moves data around. basically,
13			// this is a fast strstream without the overhead.
14
15	root	1.4	struct dynbuf
16	root	1.1	{
17	root	1.4	protected:
18	root	1.1	struct chunk
19			{
20			chunk *next;
21	root	1.5	int alloc;
22	root	1.1	int size;
23			char data[0];
24			};
25
26	root	1.6	char ptr, end;
27	root	1.1	int ext;
28			int _size;
29
30			chunk first, last;
31
32			void _reserve (int size);
33			void clear ();
34			void finish ();
35
36			public:
37
38	root	1.2	dynbuf (int initial = 4096, int extend = 16384);
39	root	1.1	~dynbuf ();
40
41	root	1.6	int size () const { return _size + (ptr - last->data); }
42			bool empty () const { return !size (); }
43	root	1.1
44			void linearise (void *data);
45	root	1.6	char *linearise (); // does not 0-terminate(!)
46
47			int room () const { return end - ptr; }
48	root	1.1
49			char *force (int size)
50			{
51	root	1.6	if (ptr + size >= end)
52	root	1.1	_reserve (size);
53
54			return ptr;
55			}
56
57			char *alloc (int size)
58			{
59			char *res = force (size);
60	root	1.6	ptr += size;
61	root	1.1	return res;
62			}
63
64	root	1.6	void fadd (char c) { *ptr++ = c; }
65	root	1.2	void fadd (unsigned char c) { fadd (char (c)); }
66	root	1.1
67			void add (const void *p, int len)
68			{
69			memcpy (alloc (len), p, len);
70			}
71
72			void add (char c)
73			{
74	root	1.3	alloc (1)[0] = c;
75	root	1.1	}
76
77			void add (const char *s)
78			{
79			add (s, strlen (s));
80			}
81
82	root	1.4	void add (const shstr &s)
83			{
84			add (s.s, s.length ());
85			}
86	root	1.1
87			//TODO
88			//void add_destructive (dynbuf &buf);
89
90			dynbuf &operator << (char c) { add (c); return *this; }
91			dynbuf &operator << (unsigned char c) { return *this << char (c); }
92			dynbuf &operator << (const char s) { add (s); return this; }
93	root	1.6	dynbuf &operator << (const shstr &s) { add (s); return *this; }
94	root	1.4	dynbuf &operator << (const std::string &s) { add (s.data(), s.size ()); return *this; }
95
96			operator std::string ();
97			};
98
99			struct dynbuf_text : dynbuf
100			{
101			dynbuf_text (int initial = 4096, int extend = 16384)
102			: dynbuf (initial, extend)
103			{ }
104
105			using dynbuf::add;
106
107			static const int max_sint32_size = 11;
108			static const int max_sint64_size = 20;
109
110			void add (sint32 i);
111			void add (sint64 i);
112
113			void printf (const char *format, ...);
114	root	1.1	};
115
116			#endif
117	root	1.6