forked from yaosj2k/dnsforwarder
-
Notifications
You must be signed in to change notification settings - Fork 0
/
extendablebuffer.h
executable file
·129 lines (107 loc) · 4.53 KB
/
extendablebuffer.h
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
#ifndef EXTENDABLEBUFFER_H_INCLUDED
#define EXTENDABLEBUFFER_H_INCLUDED
#include "common.h"
typedef struct _ExtendableBuffer{
/* All datas reside here. */
char *Data;
/* How many bytes have been used. */
uint32_t Used;
/* How many bytes have allocated. */
uint32_t Allocated;
/* Length of bytes allocated when initializing */
uint32_t InitialSize;
/* GuardSize */
int32_t GuardSize;
} ExtendableBuffer;
#define EXTENDABLEBUFFER_INIT_STATIC {NULL, 0, 0, 0, -1}
int ExtendableBuffer_Init( __in ExtendableBuffer *eb,
__in uint32_t InitSize,
__in int32_t GuardSize);
/* Description:
* Initialize an ExtendableBuffer.
* Parameters:
* eb : The ExtendableBuffer to be initialized.
* InitSize : The initial buffer size.
* GuardSize : Set `GuardSize'. This parameter can be negative if it doesn't
* need a guard.
* Return value:
* 0 on success, a non-zero value otherwise.
*/
#define ExtendableBuffer_GetData(eb_ptr) ((char *)((eb_ptr) -> Data))
/* Description:
* Get the whole data of an ExtendableBuffer.
* Parameters:
* eb_ptr : The ExtendableBuffer to be gotten from.
* Return value:
* The data being gotten.
*/
#define ExtendableBuffer_GetUsedBytes(eb_ptr) ((eb_ptr) -> Used)
/* Description:
* Decide how many bytes have been used.
* Parameters:
* eb_ptr : The ExtendableBuffer to be decided.
* Return value:
* The number of used bytes.
*/
BOOL ExtendableBuffer_GuarantyLeft( __in ExtendableBuffer *eb,
__in uint32_t GuarantiedSize);
/* Description:
* Make sure that the number of unused(left) bytes (`Allocated' - `Used') is
* greater than `GuarantiedSize'. Straightforwardly, make sure the unused
* allocated space enough. This function will do realloc if necessary.
* There are 4 cases:
* Case 1 : Left bytes are enough : `Allocated' - `Used' >= `GuarantiedSize' :
* Simply return TRUE.
* Case 2 : Left bytes are not enough and all allocated bytes haven't exceed
* the `GuardSize' : `Allocated' - `Used' < `GuarantiedSize' and
* `GuardSize' >= 0 and `Allocated' <= `GuardSize' :
* Do realloc to make sure `Allocated' - `Used' >= `GuarantiedSize', if it
* success, returns TRUE, or FALSE otherwise.
* Case 3 : Left bytes are not enough but all allocated bytes have exceed the
* `GuardSize' : `Allocated' - `Used' < `GuarantiedSize' and `GuardSize' >= 0
* and `Allocated' > `GuardSize' :
* Simply return FALSE.
* Case 4 : Left bytes are not enough and there is no guard :
* `Allocated' - `Used' < `GuarantiedSize' and `GuardSize' < 0 :
* The same as case 2.
* Parameters:
* eb : The ExtendableBuffer to be guarantied.
* GuarantiedSize : The guarantied size.
* Return value:
* TRUE if it can be guarantied. Or FALSE otherwise.
*/
char *ExtendableBuffer_Expand( __in ExtendableBuffer *eb,
__in uint32_t ExpandedSize,
__out_opt int32_t *Offset
);
/* Description:
* Increase the number of used bytes by `ExpandedSize'. There is a call to
* `ExtendableBuffer_GuarantyLeft' in this function, to make sure there is
* enough legal space to expand.
* Parameters:
* eb : The ExtendableBuffer to expand.
* ExpandedSize : Expanded size.
* Offset : The offset of the newly expanded space. This parameter can be
* NULL if no offset is needed.
* Return value:
* The head address of the newly expanded space.
*/
#define ExtendableBuffer_GetEndOffset(eb_ptr) ((eb_ptr) -> Used)
/* Description:
* Decide the offset (in bytes) of the byte just after the last used byte. We
* call this offset end offset.
* The 'offset' here means the distance between `Data' and the byte, and
* a 'used byte' is such a byte that has an offset less than `Used'.
* Parameters:
* eb_ptr : The ExtendableBuffer to be decided.
* Return value:
* End offset.
*/
#define ExtendableBuffer_SetEndOffset(eb_ptr, val) ((eb_ptr) -> Used = (val))
#define ExtendableBuffer_GetPositionByOffset(eb_ptr, offset) (( ( (char *) ( (eb_ptr) -> Data) ) + (offset) ))
int32_t ExtendableBuffer_Add(ExtendableBuffer *eb, const char *Data, uint32_t DataLength);
char *ExtendableBuffer_Eliminate(ExtendableBuffer *eb, uint32_t Start, uint32_t Length);
#define ExtendableBuffer_Eliminate_Tail(eb_ptr, length) (((eb_ptr) -> Used) -= (length))
void ExtendableBuffer_Reset(ExtendableBuffer *eb);
void ExtendableBuffer_Free(ExtendableBuffer *eb);
#endif // EXTENDABLEBUFFER_H_INCLUDED