network_ydxlib/stream_buffer.h at master · crspecter/network_ydxlib · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#ifndef __YDX_STREAM_BUFFER_H__
#define __YDX_STREAM_BUFFER_H__
#include <stdint.h>
#include <stdlib.h>
#include <boost/noncopyable.hpp>
namespace ydx
{

#define _byte(n)     (n)                    //B
#define _kilobyte(n) ((n)*1024UL)           //KB:2的10次方,1024 BYTE
#define _megabyte(n) ((n)*1048576UL)        //MB:2的20次方,1048576 BYTE 或 1024KB.
#define _gigabyte(n) ((n)*1073741824ULL)    //GB:2的30次方,1073741824 BYTE 或 1024MB.
#define _terabyte(n) ((n)*1099511627776ULL) //TB:2的40次方,1099511627776 BYTE 或 1024GB.

#ifndef MIN
#define MIN(a, b) (((a)>(b))?(b):(a))
#endif

#ifndef MAX
#define MAX(a, b) (((a)>(b))?(a):(b))
#endif


class StreamBuffer : boost::noncopyable
{
public:
	StreamBuffer()
		: in_(0),
		  out_(0),
		  size_(0),
		  data_(NULL)
	{

	}
	~StreamBuffer()
	{
		if(NULL != data_)
		{
			::free(data_);
			data_ = NULL;
		}

	}

	int alloc(uint64_t size = _megabyte(16));

	void clear()
	{
		in_ = out_ = 0;
	}

	uint64_t used()
	{
		int64_t n = in_ - out_;
		return (n < 0)? n + size_ :  n;
	}

	uint64_t unused()
	{
		return size_ - used();
	}

	void copy_out(void *buf, uint64_t len);
	void copy_in(const void *data, uint64_t len);

	uint64_t copy_in_must(const void *data, uint64_t len);
	uint64_t copy_out_must(void* buf, uint64_t len);

	uint64_t peek_out_must(void* buf, uint64_t len);

public:
	bool write_buffer(const void* buf, int len);
	bool read_buffer(void* buf, int& len);
	bool has_writen(uint64_t len);
	bool has_read(uint64_t len);

private:
    uint64_t    in_;     ///< in  offset
    uint64_t    out_;    ///< out offset
    //uint64_t    mask;   ///< the mask of power of two sub one
    uint64_t    size_;  ///< granularity or element's size
    void*       data_;   ///< data pointer
	//__stream_info* stream_info_;
};


}

#endif