00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025 #include <core/exceptions/system.h>
00026 #include <core/exceptions/software.h>
00027 #include <netcomm/utils/dynamic_buffer.h>
00028
00029 #include <cstdlib>
00030 #include <cstring>
00031 #include <netinet/in.h>
00032
00033 namespace fawkes {
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047
00048
00049
00050
00051
00052
00053
00054
00055
00056
00057
00058
00059
00060
00061
00062
00063
00064 DynamicBuffer::DynamicBuffer(dynamic_list_t *db, size_t initial_buffer_size)
00065 {
00066 _read_only = false;
00067 _db = db;
00068 _buffer_size = initial_buffer_size;
00069 _buffer = malloc(_buffer_size);
00070 _curhead = (element_header_t *)_buffer;
00071 _curdata = (void *)((size_t)_buffer + sizeof(element_header_t));
00072
00073 _db->size = htonl(0);
00074 _db->num_elements = htonl(0);
00075
00076 _it_curhead = NULL;
00077 _it_curdata = NULL;
00078 _it_curel = 0;
00079 }
00080
00081
00082
00083
00084
00085
00086
00087
00088 DynamicBuffer::DynamicBuffer(dynamic_list_t *db, void *buf, size_t size)
00089 {
00090 _read_only = true;
00091 _db = db;
00092 _buffer_size = size;
00093 _buffer = buf;
00094
00095 _curhead = (element_header_t *)_buffer;
00096 _curdata = (void *)((size_t)_buffer + sizeof(element_header_t));
00097
00098 _it_curhead = _curhead;
00099 _it_curdata = _curdata;
00100 _it_curel = 0;
00101 }
00102
00103
00104
00105
00106
00107 DynamicBuffer::~DynamicBuffer()
00108 {
00109 if (! _read_only) free(_buffer);
00110 }
00111
00112
00113
00114
00115
00116
00117
00118
00119
00120
00121
00122 void
00123 DynamicBuffer::append(const void *data, size_t data_size)
00124 {
00125 if ( _read_only ) throw AccessViolationException("DynamicBuffer is read-only");
00126
00127 if ( data_size > (0xFFFF - sizeof(element_header_t)) ) {
00128 throw IllegalArgumentException("Buffer size too big, max 65535 bytes");
00129 }
00130
00131 size_t cur_size = ntohl(_db->size);
00132 if ( (cur_size + data_size + sizeof(element_header_t)) > _buffer_size ) {
00133 try {
00134 increase();
00135 } catch (OutOfMemoryException &e) {
00136 throw;
00137 }
00138 if ( (cur_size + data_size + sizeof(element_header_t)) > _buffer_size ) {
00139 throw OutOfMemoryException("Could not increase buffer far enough to hold data");
00140 }
00141 }
00142
00143 *_curhead = htons(data_size);
00144 memcpy(_curdata, data, data_size);
00145
00146 _curhead = (element_header_t *)((size_t)_curhead + data_size
00147 + sizeof(element_header_t));
00148 _curdata = (void *)((size_t)_curdata + data_size + sizeof(element_header_t));
00149 _db->size = htonl(cur_size + sizeof(element_header_t) + data_size);
00150 uint16_t tmp = ntohl(_db->num_elements) + 1;
00151 _db->num_elements = htonl(tmp);
00152 }
00153
00154
00155
00156
00157
00158 void *
00159 DynamicBuffer::buffer()
00160 {
00161 return _buffer;
00162 }
00163
00164
00165
00166
00167
00168 void
00169 DynamicBuffer::increase()
00170 {
00171 size_t new_buffer_size;
00172
00173 if ( (_buffer_size) >= 0xFFFFFFFF / 2 ) {
00174 if ( _buffer_size == 0xFFFFFFFF ) {
00175 throw OutOfMemoryException("Dynamic buffer may not be greater than 64KB");
00176 } else {
00177 new_buffer_size = 0xFFFFFFFF;
00178 }
00179 } else {
00180 new_buffer_size = _buffer_size * 2;
00181 }
00182
00183 void *tmp = realloc(_buffer, new_buffer_size);
00184 if ( tmp == NULL ) {
00185 throw OutOfMemoryException();
00186 } else {
00187 _buffer_size = new_buffer_size;
00188 _curhead = (element_header_t *)((size_t)tmp + ((size_t)_curhead - (size_t)_buffer));
00189 _curdata = (void *)((size_t)tmp + ((size_t)_curdata - (size_t)_buffer));
00190 _buffer = tmp;
00191 }
00192 }
00193
00194
00195
00196
00197
00198
00199
00200 size_t
00201 DynamicBuffer::buffer_size()
00202 {
00203 return ntohl(_db->size);
00204 }
00205
00206
00207
00208
00209
00210
00211
00212 size_t
00213 DynamicBuffer::real_buffer_size()
00214 {
00215 return _buffer_size;
00216 }
00217
00218
00219
00220
00221
00222 unsigned int
00223 DynamicBuffer::num_elements()
00224 {
00225 return ntohl(_db->num_elements);
00226 }
00227
00228
00229
00230
00231 void
00232 DynamicBuffer::reset_iterator()
00233 {
00234 _it_curhead = _curhead;
00235 _it_curdata = _curdata;
00236
00237 _it_curel = 0;
00238 }
00239
00240
00241
00242
00243
00244 bool
00245 DynamicBuffer::has_next()
00246 {
00247 return (_read_only && (_it_curel < (ntohl(_db->num_elements))));
00248 }
00249
00250
00251
00252
00253
00254
00255
00256
00257 void *
00258 DynamicBuffer::next(size_t *size)
00259 {
00260
00261 if ( ! has_next() ) {
00262 throw OutOfBoundsException("No next element while iterator DynamicBuffer");
00263 }
00264
00265 if ( _it_curel > 0 ) {
00266 size_t offset = ntohs(*_it_curhead) + sizeof(element_header_t);
00267 _it_curhead = (element_header_t *)((size_t)_it_curhead + offset);
00268 _it_curdata = (void *)((size_t)_it_curdata + offset);
00269 }
00270 ++_it_curel;
00271 *size = ntohs(*_it_curhead);
00272
00273 return _it_curdata;
00274 }
00275
00276 }