-
Notifications
You must be signed in to change notification settings - Fork 2
/
buffer.cpp
141 lines (120 loc) · 3.19 KB
/
buffer.cpp
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
130
131
132
133
134
135
136
137
138
139
140
/*
Source for MSP430 emulator IdaPro plugin
File: buffer.cpp
Copyright (c) 2005-2010 Chris Eagle
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 of the License, 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, Inc., 59 Temple
Place, Suite 330, Boston, MA 02111-1307 USA
*/
#include <stdlib.h>
#include <string.h>
#include "buffer.h"
#define BLOCK_SIZE 0x100 //keep this a power of two
Buffer::Buffer() {
init(BLOCK_SIZE);
}
Buffer::Buffer(unsigned int magic) {
init(BLOCK_SIZE);
this->magic = magic;
write(&magic, sizeof(magic));
}
Buffer::Buffer(unsigned char *buf, unsigned int len) {
init(len);
if (!error) {
if (len >= 4) { //check for presence of BUFFER_MAGIC
unsigned int m = *(unsigned int*)buf;
if ((m & BUFFER_MAGIC_MASK) == BUFFER_MAGIC) {
magic = m;
len -= 4; //adjust length
buf += 4; //adjust buffer start
}
}
memcpy(bptr, buf, len);
}
wptr = sz;
}
void Buffer::init(unsigned int size) {
bptr = (unsigned char *)malloc(size);
sz = bptr ? size : 0;
rptr = wptr = 0;
error = sz != size;
magic = 0;
}
Buffer::~Buffer() {
free(bptr);
}
int Buffer::read(void *data, unsigned int len) {
if ((rptr + len) <= sz) {
memcpy(data, bptr + rptr, len);
rptr += len;
return 0;
}
error = true;
return 1;
}
bool Buffer::rewind(unsigned int amt) {
if (rptr >= amt) {
rptr -= amt;
return true;
}
return false;
}
int Buffer::write(const void *data, unsigned int len) {
if (!check_size(wptr + len)) {
memcpy(bptr + wptr, data, len);
wptr += len;
return 0;
}
error = true;
return 1;
}
int Buffer::readString(char **str) {
unsigned int len;
if (read(&len, sizeof(len)) == 0) {
*str = (char*)malloc(len);
if (*str && read(*str, len) == 0) return 0;
free(*str);
}
error = true;
return 1;
}
int Buffer::writeString(const char *str) {
unsigned int len = strlen(str) + 1;
if (write(&len, sizeof(len)) == 0) {
return write(str, len);
}
error = true;
return 1;
}
unsigned char *Buffer::get_buf() {
return bptr;
}
unsigned int Buffer::get_wlen() {
return wptr;
}
unsigned int Buffer::get_rlen() {
return rptr;
}
int Buffer::check_size(unsigned int max) {
if (max <= sz) return 0;
max = (max + BLOCK_SIZE) & ~(BLOCK_SIZE - 1); //round up to next BLOCK_SIZE
unsigned char *tmp = (unsigned char *)realloc(bptr, max);
if (tmp) {
bptr = tmp;
sz = max;
return 0;
}
error = true;
return 1;
}
unsigned int Buffer::getVersion() {
return ((magic & BUFFER_MAGIC_MASK) == BUFFER_MAGIC) ? (magic & ~BUFFER_MAGIC_MASK) : 0;
}