Thinger.io / ThingerClient

Thinger.io Client Library for ARM mbed platform. This is a generic library that provides a base class that can be used to other develop hardware specific libraries.

Fork of ThingerClient by Alvaro Luis Bustamante

thinger/pson.h@4:de51256455f7, 2015-12-26 (annotated)

Committer:
alvarolb
Date:
Sat Dec 26 13:18:01 2015 +0000
Revision:
4:de51256455f7
Parent:
0:b75d784c7c1a 
Adapter pson to properly work in ARM Mbed old compiler

Who changed what in which revision?

UserRevisionLine numberNew contents of line
alvarolb 4:de51256455f7 1 // The MIT License (MIT)
alvarolb 4:de51256455f7 2 //
alvarolb 4:de51256455f7 3 // Copyright (c) 2015 THINGER LTD
alvarolb 4:de51256455f7 4 // Author: alvarolb@gmail.com (Alvaro Luis Bustamante)
alvarolb 4:de51256455f7 5 //
alvarolb 4:de51256455f7 6 // Permission is hereby granted, free of charge, to any person obtaining a copy
alvarolb 4:de51256455f7 7 // of this software and associated documentation files (the "Software"), to deal
alvarolb 4:de51256455f7 8 // in the Software without restriction, including without limitation the rights
alvarolb 4:de51256455f7 9 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
alvarolb 4:de51256455f7 10 // copies of the Software, and to permit persons to whom the Software is
alvarolb 4:de51256455f7 11 // furnished to do so, subject to the following conditions:
alvarolb 4:de51256455f7 12 //
alvarolb 4:de51256455f7 13 // The above copyright notice and this permission notice shall be included in
alvarolb 4:de51256455f7 14 // all copies or substantial portions of the Software.
alvarolb 4:de51256455f7 15 //
alvarolb 4:de51256455f7 16 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
alvarolb 4:de51256455f7 17 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
alvarolb 4:de51256455f7 18 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
alvarolb 4:de51256455f7 19 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
alvarolb 4:de51256455f7 20 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
alvarolb 4:de51256455f7 21 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
alvarolb 4:de51256455f7 22 // THE SOFTWARE.
alvarolb 4:de51256455f7 23
alvarolb 4:de51256455f7 24 #ifndef PSON_HPP
alvarolb 4:de51256455f7 25 #define PSON_HPP
alvarolb 4:de51256455f7 26
alvarolb 4:de51256455f7 27 #include <stdint.h>
alvarolb 4:de51256455f7 28 #include <string.h>
alvarolb 4:de51256455f7 29 #include <math.h>
alvarolb 4:de51256455f7 30 #include <stdlib.h>
alvarolb 4:de51256455f7 31
alvarolb 4:de51256455f7 32 namespace protoson {
alvarolb 4:de51256455f7 33
alvarolb 4:de51256455f7 34 class memory_allocator{
alvarolb 4:de51256455f7 35 public:
alvarolb 4:de51256455f7 36 virtual void *allocate(size_t size) = 0;
alvarolb 4:de51256455f7 37 virtual void deallocate(void *) = 0;
alvarolb 4:de51256455f7 38 };
alvarolb 4:de51256455f7 39
alvarolb 4:de51256455f7 40 template<size_t buffer_size>
alvarolb 4:de51256455f7 41 class circular_memory_allocator : public memory_allocator{
alvarolb 4:de51256455f7 42 private:
alvarolb 4:de51256455f7 43 uint8_t buffer_[buffer_size];
alvarolb 4:de51256455f7 44 size_t index_;
alvarolb 4:de51256455f7 45 public:
alvarolb 4:de51256455f7 46 circular_memory_allocator() : index_(0) {
alvarolb 4:de51256455f7 47 }
alvarolb 4:de51256455f7 48
alvarolb 4:de51256455f7 49 virtual void *allocate(size_t size) {
alvarolb 4:de51256455f7 50 if (index_ + size > buffer_size) {
alvarolb 4:de51256455f7 51 index_ = 0;
alvarolb 4:de51256455f7 52 }
alvarolb 4:de51256455f7 53 void *position = &buffer_[index_];
alvarolb 4:de51256455f7 54 index_ += size;
alvarolb 4:de51256455f7 55 return position;
alvarolb 4:de51256455f7 56 }
alvarolb 4:de51256455f7 57
alvarolb 4:de51256455f7 58 virtual void deallocate(void *) {
alvarolb 4:de51256455f7 59
alvarolb 4:de51256455f7 60 }
alvarolb 4:de51256455f7 61 };
alvarolb 4:de51256455f7 62
alvarolb 4:de51256455f7 63 class dynamic_memory_allocator : public memory_allocator{
alvarolb 4:de51256455f7 64 public:
alvarolb 4:de51256455f7 65 virtual void *allocate(size_t size) {
alvarolb 4:de51256455f7 66 return malloc(size);
alvarolb 4:de51256455f7 67 }
alvarolb 4:de51256455f7 68
alvarolb 4:de51256455f7 69 virtual void deallocate(void *ptr) {
alvarolb 4:de51256455f7 70 free(ptr);
alvarolb 4:de51256455f7 71 }
alvarolb 4:de51256455f7 72 };
alvarolb 4:de51256455f7 73
alvarolb 4:de51256455f7 74 extern memory_allocator& pool;
alvarolb 4:de51256455f7 75 }
alvarolb 4:de51256455f7 76
alvarolb 4:de51256455f7 77 inline void* operator new(size_t sz, protoson::memory_allocator& a)
alvarolb 4:de51256455f7 78 {
alvarolb 4:de51256455f7 79 return a.allocate(sz);
alvarolb 4:de51256455f7 80 }
alvarolb 4:de51256455f7 81
alvarolb 4:de51256455f7 82 inline void operator delete(void* p, protoson::memory_allocator& a)
alvarolb 4:de51256455f7 83 {
alvarolb 4:de51256455f7 84 a.deallocate(p);
alvarolb 4:de51256455f7 85 }
alvarolb 4:de51256455f7 86
alvarolb 4:de51256455f7 87 template<class T>
alvarolb 4:de51256455f7 88 void destroy(T* p, protoson::memory_allocator& a)
alvarolb 4:de51256455f7 89 {
alvarolb 4:de51256455f7 90 if (p) {
alvarolb 4:de51256455f7 91 p->~T();
alvarolb 4:de51256455f7 92 a.deallocate(p);
alvarolb 4:de51256455f7 93 }
alvarolb 4:de51256455f7 94 }
alvarolb 4:de51256455f7 95
alvarolb 4:de51256455f7 96 namespace protoson {
alvarolb 4:de51256455f7 97
alvarolb 4:de51256455f7 98 enum pb_wire_type{
alvarolb 4:de51256455f7 99 varint = 0,
alvarolb 4:de51256455f7 100 fixed_64 = 1,
alvarolb 4:de51256455f7 101 length_delimited = 2,
alvarolb 4:de51256455f7 102 fixed_32 = 5,
alvarolb 4:de51256455f7 103 pson_type = 6
alvarolb 4:de51256455f7 104 };
alvarolb 4:de51256455f7 105
alvarolb 4:de51256455f7 106 template<class T>
alvarolb 4:de51256455f7 107 class pson_container {
alvarolb 4:de51256455f7 108
alvarolb 4:de51256455f7 109 struct list_item{
alvarolb 4:de51256455f7 110 T item_;
alvarolb 4:de51256455f7 111 list_item* next_;
alvarolb 4:de51256455f7 112 };
alvarolb 4:de51256455f7 113
alvarolb 4:de51256455f7 114 public:
alvarolb 4:de51256455f7 115
alvarolb 4:de51256455f7 116 class iterator{
alvarolb 4:de51256455f7 117 public:
alvarolb 4:de51256455f7 118 iterator(list_item *current) : current(current) {
alvarolb 4:de51256455f7 119 }
alvarolb 4:de51256455f7 120
alvarolb 4:de51256455f7 121 private:
alvarolb 4:de51256455f7 122 list_item * current;
alvarolb 4:de51256455f7 123
alvarolb 4:de51256455f7 124 public:
alvarolb 4:de51256455f7 125 bool next(){
alvarolb 4:de51256455f7 126 if(current==NULL) return false;
alvarolb 4:de51256455f7 127 current = current->next_;
alvarolb 4:de51256455f7 128 return true;
alvarolb 4:de51256455f7 129 }
alvarolb 4:de51256455f7 130
alvarolb 4:de51256455f7 131 bool has_next(){
alvarolb 4:de51256455f7 132 return current!=NULL && current->next_!=NULL;
alvarolb 4:de51256455f7 133 }
alvarolb 4:de51256455f7 134
alvarolb 4:de51256455f7 135 bool valid(){
alvarolb 4:de51256455f7 136 return current!=NULL;
alvarolb 4:de51256455f7 137 }
alvarolb 4:de51256455f7 138
alvarolb 4:de51256455f7 139 T& item(){
alvarolb 4:de51256455f7 140 return current->item_;
alvarolb 4:de51256455f7 141 }
alvarolb 4:de51256455f7 142 };
alvarolb 4:de51256455f7 143
alvarolb 4:de51256455f7 144 private:
alvarolb 4:de51256455f7 145 list_item* item_;
alvarolb 4:de51256455f7 146
alvarolb 4:de51256455f7 147 public:
alvarolb 4:de51256455f7 148 iterator begin() const{
alvarolb 4:de51256455f7 149 return iterator(item_);
alvarolb 4:de51256455f7 150 }
alvarolb 4:de51256455f7 151
alvarolb 4:de51256455f7 152 pson_container() : item_(NULL) {
alvarolb 4:de51256455f7 153 }
alvarolb 4:de51256455f7 154
alvarolb 4:de51256455f7 155 ~pson_container(){
alvarolb 4:de51256455f7 156 clear();
alvarolb 4:de51256455f7 157 }
alvarolb 4:de51256455f7 158
alvarolb 4:de51256455f7 159 size_t size() const{
alvarolb 4:de51256455f7 160 size_t size = 0;
alvarolb 4:de51256455f7 161 list_item* current = item_;
alvarolb 4:de51256455f7 162 while(current!=NULL){
alvarolb 4:de51256455f7 163 current = current->next_;
alvarolb 4:de51256455f7 164 size++;
alvarolb 4:de51256455f7 165 }
alvarolb 4:de51256455f7 166 return size;
alvarolb 4:de51256455f7 167 }
alvarolb 4:de51256455f7 168
alvarolb 4:de51256455f7 169 T* operator[](size_t index){
alvarolb 4:de51256455f7 170 list_item* current = item_;
alvarolb 4:de51256455f7 171 size_t current_index = 0;
alvarolb 4:de51256455f7 172 while(current!=NULL){
alvarolb 4:de51256455f7 173 if(current_index==index){
alvarolb 4:de51256455f7 174 return &current->item_;
alvarolb 4:de51256455f7 175 }
alvarolb 4:de51256455f7 176 current = current->next_;
alvarolb 4:de51256455f7 177 current_index++;
alvarolb 4:de51256455f7 178 }
alvarolb 4:de51256455f7 179 return NULL;
alvarolb 4:de51256455f7 180 }
alvarolb 4:de51256455f7 181
alvarolb 4:de51256455f7 182 void clear(){
alvarolb 4:de51256455f7 183 list_item* current = item_;
alvarolb 4:de51256455f7 184 while(current!=NULL){
alvarolb 4:de51256455f7 185 list_item* next = current->next_;
alvarolb 4:de51256455f7 186 destroy(current, pool);
alvarolb 4:de51256455f7 187 current = next;
alvarolb 4:de51256455f7 188 }
alvarolb 4:de51256455f7 189 item_ = NULL;
alvarolb 4:de51256455f7 190 }
alvarolb 4:de51256455f7 191
alvarolb 4:de51256455f7 192 T& create_item(){
alvarolb 4:de51256455f7 193 list_item* new_list_item = new(pool) list_item();
alvarolb 4:de51256455f7 194 if(item_==NULL){
alvarolb 4:de51256455f7 195 item_ = new_list_item;
alvarolb 4:de51256455f7 196 }
alvarolb 4:de51256455f7 197 else{
alvarolb 4:de51256455f7 198 list_item * last = item_;
alvarolb 4:de51256455f7 199 while(last->next_!=NULL) last = last->next_;
alvarolb 4:de51256455f7 200 last->next_ = new_list_item;
alvarolb 4:de51256455f7 201 }
alvarolb 4:de51256455f7 202 return new_list_item->item_;
alvarolb 4:de51256455f7 203 }
alvarolb 4:de51256455f7 204 };
alvarolb 4:de51256455f7 205
alvarolb 4:de51256455f7 206 class pson_object;
alvarolb 4:de51256455f7 207 class pson_array;
alvarolb 4:de51256455f7 208
alvarolb 4:de51256455f7 209 class pson {
alvarolb 4:de51256455f7 210 public:
alvarolb 4:de51256455f7 211 enum field_type {
alvarolb 4:de51256455f7 212 null_field = 0,
alvarolb 4:de51256455f7 213 varint_field = 1,
alvarolb 4:de51256455f7 214 svarint_field = 2,
alvarolb 4:de51256455f7 215 float_field = 3,
alvarolb 4:de51256455f7 216 double_field = 4,
alvarolb 4:de51256455f7 217 true_field = 5,
alvarolb 4:de51256455f7 218 false_field = 6,
alvarolb 4:de51256455f7 219 zero_field = 7,
alvarolb 4:de51256455f7 220 one_field = 8,
alvarolb 4:de51256455f7 221 string_field = 9,
alvarolb 4:de51256455f7 222 empty_string = 10,
alvarolb 4:de51256455f7 223 bytes_field = 11,
alvarolb 4:de51256455f7 224 empty_bytes = 12,
alvarolb 4:de51256455f7 225 object_field = 13,
alvarolb 4:de51256455f7 226 array_field = 14,
alvarolb 4:de51256455f7 227 empty = 15,
alvarolb 4:de51256455f7 228 // a message tag is encoded in a 128-base varint [1-bit][3-bit wire type][4-bit field]
alvarolb 4:de51256455f7 229 // we have up to 4 bits (0-15) for encoding fields in the first byte
alvarolb 4:de51256455f7 230 };
alvarolb 4:de51256455f7 231
alvarolb 4:de51256455f7 232 bool is_boolean() const{
alvarolb 4:de51256455f7 233 return field_type_ == true_field || field_type_ == false_field;
alvarolb 4:de51256455f7 234 }
alvarolb 4:de51256455f7 235
alvarolb 4:de51256455f7 236 bool is_string() const{
alvarolb 4:de51256455f7 237 return field_type_ == string_field;
alvarolb 4:de51256455f7 238 }
alvarolb 4:de51256455f7 239
alvarolb 4:de51256455f7 240 bool is_bytes() const{
alvarolb 4:de51256455f7 241 return field_type_ == bytes_field;
alvarolb 4:de51256455f7 242 }
alvarolb 4:de51256455f7 243
alvarolb 4:de51256455f7 244 bool is_number() const{
alvarolb 4:de51256455f7 245 return field_type_ == varint_field ||
alvarolb 4:de51256455f7 246 field_type_ == svarint_field ||
alvarolb 4:de51256455f7 247 field_type_ == float_field ||
alvarolb 4:de51256455f7 248 field_type_ == double_field ||
alvarolb 4:de51256455f7 249 field_type_ == zero_field ||
alvarolb 4:de51256455f7 250 field_type_ == one_field;
alvarolb 4:de51256455f7 251 }
alvarolb 4:de51256455f7 252
alvarolb 4:de51256455f7 253 bool is_object() const{
alvarolb 4:de51256455f7 254 return field_type_ == object_field;
alvarolb 4:de51256455f7 255 }
alvarolb 4:de51256455f7 256
alvarolb 4:de51256455f7 257 bool is_array() const{
alvarolb 4:de51256455f7 258 return field_type_ == array_field;
alvarolb 4:de51256455f7 259 }
alvarolb 4:de51256455f7 260
alvarolb 4:de51256455f7 261 bool is_null() const{
alvarolb 4:de51256455f7 262 return field_type_ == null_field;
alvarolb 4:de51256455f7 263 }
alvarolb 4:de51256455f7 264
alvarolb 4:de51256455f7 265 bool is_empty() const{
alvarolb 4:de51256455f7 266 return field_type_ == empty;
alvarolb 4:de51256455f7 267 }
alvarolb 4:de51256455f7 268
alvarolb 4:de51256455f7 269 pson() : value_(NULL), field_type_(empty) {
alvarolb 4:de51256455f7 270 }
alvarolb 4:de51256455f7 271
alvarolb 4:de51256455f7 272 template<class T>
alvarolb 4:de51256455f7 273 pson(T value) : value_(NULL), field_type_(empty){
alvarolb 4:de51256455f7 274 *this = value;
alvarolb 4:de51256455f7 275 }
alvarolb 4:de51256455f7 276
alvarolb 4:de51256455f7 277 ~pson(){
alvarolb 4:de51256455f7 278 if(field_type_==object_field){
alvarolb 4:de51256455f7 279 destroy((pson_object *) value_, pool);
alvarolb 4:de51256455f7 280 }else if(field_type_==array_field) {
alvarolb 4:de51256455f7 281 destroy((pson_array *) value_, pool);
alvarolb 4:de51256455f7 282 }else{
alvarolb 4:de51256455f7 283 pool.deallocate(value_);
alvarolb 4:de51256455f7 284 }
alvarolb 4:de51256455f7 285 }
alvarolb 4:de51256455f7 286
alvarolb 4:de51256455f7 287 template<class T>
alvarolb 4:de51256455f7 288 void operator=(T value)
alvarolb 4:de51256455f7 289 {
alvarolb 4:de51256455f7 290 if(value==0){
alvarolb 4:de51256455f7 291 field_type_ = zero_field;
alvarolb 4:de51256455f7 292 }else if(value==1) {
alvarolb 4:de51256455f7 293 field_type_ = one_field;
alvarolb 4:de51256455f7 294 }else{
alvarolb 4:de51256455f7 295 field_type_ = value>0 ? varint_field : svarint_field;
alvarolb 4:de51256455f7 296 uint64_t uint_value = value>0 ? value : -value;
alvarolb 4:de51256455f7 297 value_ = pool.allocate(pson::get_varint_size(uint_value));
alvarolb 4:de51256455f7 298 pb_encode_varint(uint_value);
alvarolb 4:de51256455f7 299 }
alvarolb 4:de51256455f7 300 }
alvarolb 4:de51256455f7 301
alvarolb 4:de51256455f7 302 void operator=(bool value){
alvarolb 4:de51256455f7 303 field_type_ = value ? true_field : false_field;
alvarolb 4:de51256455f7 304 }
alvarolb 4:de51256455f7 305
alvarolb 4:de51256455f7 306 void operator=(float value) {
alvarolb 4:de51256455f7 307 if(value==(int32_t)value){
alvarolb 4:de51256455f7 308 *this = (int32_t) value;
alvarolb 4:de51256455f7 309 }else{
alvarolb 4:de51256455f7 310 field_type_ = float_field;
alvarolb 4:de51256455f7 311 set(value);
alvarolb 4:de51256455f7 312 }
alvarolb 4:de51256455f7 313 }
alvarolb 4:de51256455f7 314
alvarolb 4:de51256455f7 315 void operator=(double value) {
alvarolb 4:de51256455f7 316 if(value==(int64_t)value) {
alvarolb 4:de51256455f7 317 *this = (int64_t) value;
alvarolb 4:de51256455f7 318 }else if(fabs(value-(float)value)<=0.00001){
alvarolb 4:de51256455f7 319 field_type_ = float_field;
alvarolb 4:de51256455f7 320 set((float)value);
alvarolb 4:de51256455f7 321 }else{
alvarolb 4:de51256455f7 322 field_type_ = double_field;
alvarolb 4:de51256455f7 323 set(value);
alvarolb 4:de51256455f7 324 }
alvarolb 4:de51256455f7 325 }
alvarolb 4:de51256455f7 326
alvarolb 4:de51256455f7 327 void operator=(const char *str) {
alvarolb 4:de51256455f7 328 size_t str_size = strlen(str);
alvarolb 4:de51256455f7 329 if(str_size==0){
alvarolb 4:de51256455f7 330 field_type_ = empty_string;
alvarolb 4:de51256455f7 331 }else{
alvarolb 4:de51256455f7 332 field_type_ = string_field;
alvarolb 4:de51256455f7 333 memcpy(allocate(str_size+1), str, str_size+1);
alvarolb 4:de51256455f7 334 }
alvarolb 4:de51256455f7 335 }
alvarolb 4:de51256455f7 336
alvarolb 4:de51256455f7 337 void set_bytes(const void* bytes, size_t size) {
alvarolb 4:de51256455f7 338 if(size>0){
alvarolb 4:de51256455f7 339 size_t varint_size = get_varint_size(size);
alvarolb 4:de51256455f7 340 value_ = pool.allocate(varint_size+size);
alvarolb 4:de51256455f7 341 pb_encode_varint(size);
alvarolb 4:de51256455f7 342 memcpy(((uint8_t*)value_)+varint_size, bytes, size);
alvarolb 4:de51256455f7 343 field_type_ = bytes_field;
alvarolb 4:de51256455f7 344 }else{
alvarolb 4:de51256455f7 345 field_type_ = empty_bytes;
alvarolb 4:de51256455f7 346 }
alvarolb 4:de51256455f7 347 }
alvarolb 4:de51256455f7 348
alvarolb 4:de51256455f7 349 bool get_bytes(const void*& bytes, size_t& size){
alvarolb 4:de51256455f7 350 switch(field_type_){
alvarolb 4:de51256455f7 351 case bytes_field:
alvarolb 4:de51256455f7 352 size = pb_decode_varint();
alvarolb 4:de51256455f7 353 bytes = (uint8_t*) value_ + get_varint_size(size);
alvarolb 4:de51256455f7 354 return true;
alvarolb 4:de51256455f7 355 case empty:
alvarolb 4:de51256455f7 356 field_type_ = empty_bytes;
alvarolb 4:de51256455f7 357 default:
alvarolb 4:de51256455f7 358 return false;
alvarolb 4:de51256455f7 359 }
alvarolb 4:de51256455f7 360 }
alvarolb 4:de51256455f7 361
alvarolb 4:de51256455f7 362 void* allocate(size_t size){
alvarolb 4:de51256455f7 363 value_ = pool.allocate(size);
alvarolb 4:de51256455f7 364 return value_;
alvarolb 4:de51256455f7 365 }
alvarolb 4:de51256455f7 366
alvarolb 4:de51256455f7 367 operator pson_object &();
alvarolb 4:de51256455f7 368 operator pson_array &();
alvarolb 4:de51256455f7 369 pson & operator[](const char *name);
alvarolb 4:de51256455f7 370
alvarolb 4:de51256455f7 371 operator const char *() {
alvarolb 4:de51256455f7 372 switch(field_type_){
alvarolb 4:de51256455f7 373 case string_field:
alvarolb 4:de51256455f7 374 return (const char*) value_;
alvarolb 4:de51256455f7 375 case empty:
alvarolb 4:de51256455f7 376 field_type_ = empty_string;
alvarolb 4:de51256455f7 377 default:
alvarolb 4:de51256455f7 378 return "";
alvarolb 4:de51256455f7 379 }
alvarolb 4:de51256455f7 380 }
alvarolb 4:de51256455f7 381
alvarolb 4:de51256455f7 382 operator bool(){
alvarolb 4:de51256455f7 383 switch(field_type_){
alvarolb 4:de51256455f7 384 case zero_field:
alvarolb 4:de51256455f7 385 case false_field:
alvarolb 4:de51256455f7 386 return false;
alvarolb 4:de51256455f7 387 case one_field:
alvarolb 4:de51256455f7 388 case true_field:
alvarolb 4:de51256455f7 389 return true;
alvarolb 4:de51256455f7 390 case empty:
alvarolb 4:de51256455f7 391 field_type_ = false_field;
alvarolb 4:de51256455f7 392 default:
alvarolb 4:de51256455f7 393 return 0;
alvarolb 4:de51256455f7 394 }
alvarolb 4:de51256455f7 395 }
alvarolb 4:de51256455f7 396
alvarolb 4:de51256455f7 397 operator char(){
alvarolb 4:de51256455f7 398 return get_value<char>();
alvarolb 4:de51256455f7 399 }
alvarolb 4:de51256455f7 400
alvarolb 4:de51256455f7 401 operator unsigned char(){
alvarolb 4:de51256455f7 402 return get_value<unsigned char>();
alvarolb 4:de51256455f7 403 }
alvarolb 4:de51256455f7 404
alvarolb 4:de51256455f7 405 operator short(){
alvarolb 4:de51256455f7 406 return get_value<short>();
alvarolb 4:de51256455f7 407 }
alvarolb 4:de51256455f7 408
alvarolb 4:de51256455f7 409 operator unsigned short(){
alvarolb 4:de51256455f7 410 return get_value<unsigned short>();
alvarolb 4:de51256455f7 411 }
alvarolb 4:de51256455f7 412
alvarolb 4:de51256455f7 413 operator int(){
alvarolb 4:de51256455f7 414 return get_value<int>();
alvarolb 4:de51256455f7 415 }
alvarolb 4:de51256455f7 416
alvarolb 4:de51256455f7 417 operator unsigned int(){
alvarolb 4:de51256455f7 418 return get_value<int>();
alvarolb 4:de51256455f7 419 }
alvarolb 4:de51256455f7 420
alvarolb 4:de51256455f7 421 operator long(){
alvarolb 4:de51256455f7 422 return get_value<long>();
alvarolb 4:de51256455f7 423 }
alvarolb 4:de51256455f7 424
alvarolb 4:de51256455f7 425 operator unsigned long(){
alvarolb 4:de51256455f7 426 return get_value<unsigned long>();
alvarolb 4:de51256455f7 427 }
alvarolb 4:de51256455f7 428
alvarolb 4:de51256455f7 429 operator float(){
alvarolb 4:de51256455f7 430 return get_value<float>();
alvarolb 4:de51256455f7 431 }
alvarolb 4:de51256455f7 432
alvarolb 4:de51256455f7 433 operator double(){
alvarolb 4:de51256455f7 434 return get_value<double>();
alvarolb 4:de51256455f7 435 }
alvarolb 4:de51256455f7 436
alvarolb 4:de51256455f7 437 template<class T>
alvarolb 4:de51256455f7 438 operator T() {
alvarolb 4:de51256455f7 439 return get_value<T>();
alvarolb 4:de51256455f7 440 }
alvarolb 4:de51256455f7 441
alvarolb 4:de51256455f7 442 template<class T>
alvarolb 4:de51256455f7 443 T get_value(){
alvarolb 4:de51256455f7 444 switch(field_type_){
alvarolb 4:de51256455f7 445 case one_field:
alvarolb 4:de51256455f7 446 case true_field:
alvarolb 4:de51256455f7 447 return 1;
alvarolb 4:de51256455f7 448 case float_field:
alvarolb 4:de51256455f7 449 return *(float*)value_;
alvarolb 4:de51256455f7 450 case double_field:
alvarolb 4:de51256455f7 451 return *(double*)value_;
alvarolb 4:de51256455f7 452 case varint_field:
alvarolb 4:de51256455f7 453 return pb_decode_varint();
alvarolb 4:de51256455f7 454 case svarint_field:
alvarolb 4:de51256455f7 455 return -pb_decode_varint();
alvarolb 4:de51256455f7 456 case empty:
alvarolb 4:de51256455f7 457 field_type_ = zero_field;
alvarolb 4:de51256455f7 458 default:
alvarolb 4:de51256455f7 459 return 0;
alvarolb 4:de51256455f7 460 }
alvarolb 4:de51256455f7 461 }
alvarolb 4:de51256455f7 462
alvarolb 4:de51256455f7 463 void* get_value(){
alvarolb 4:de51256455f7 464 return value_;
alvarolb 4:de51256455f7 465 }
alvarolb 4:de51256455f7 466
alvarolb 4:de51256455f7 467 void set_value(void* value){
alvarolb 4:de51256455f7 468 value_ = value;
alvarolb 4:de51256455f7 469 }
alvarolb 4:de51256455f7 470
alvarolb 4:de51256455f7 471 field_type get_type() const{
alvarolb 4:de51256455f7 472 return field_type_;
alvarolb 4:de51256455f7 473 }
alvarolb 4:de51256455f7 474
alvarolb 4:de51256455f7 475 void set_null(){
alvarolb 4:de51256455f7 476 field_type_ = null_field;
alvarolb 4:de51256455f7 477 }
alvarolb 4:de51256455f7 478
alvarolb 4:de51256455f7 479 void set_type(field_type type){
alvarolb 4:de51256455f7 480 field_type_ = type;
alvarolb 4:de51256455f7 481 }
alvarolb 4:de51256455f7 482
alvarolb 4:de51256455f7 483 uint8_t get_varint_size(uint64_t value) const{
alvarolb 4:de51256455f7 484 uint8_t size = 1;
alvarolb 4:de51256455f7 485 while(value>>=7) size++;
alvarolb 4:de51256455f7 486 return size;
alvarolb 4:de51256455f7 487 }
alvarolb 4:de51256455f7 488
alvarolb 4:de51256455f7 489 void pb_encode_varint(uint64_t value) const
alvarolb 4:de51256455f7 490 {
alvarolb 4:de51256455f7 491 uint8_t count = 0;
alvarolb 4:de51256455f7 492 do
alvarolb 4:de51256455f7 493 {
alvarolb 4:de51256455f7 494 uint8_t byte = (uint8_t)(value & 0x7F);
alvarolb 4:de51256455f7 495 value >>= 7;
alvarolb 4:de51256455f7 496 if(value) byte |= 0x80;
alvarolb 4:de51256455f7 497 ((uint8_t*)value_)[count] = byte;
alvarolb 4:de51256455f7 498 count++;
alvarolb 4:de51256455f7 499 }while(value);
alvarolb 4:de51256455f7 500 }
alvarolb 4:de51256455f7 501
alvarolb 4:de51256455f7 502 uint64_t pb_decode_varint() const
alvarolb 4:de51256455f7 503 {
alvarolb 4:de51256455f7 504 if(value_==NULL) return 0;
alvarolb 4:de51256455f7 505 uint64_t value = 0;
alvarolb 4:de51256455f7 506 uint8_t pos = 0;
alvarolb 4:de51256455f7 507 uint8_t byte = 0;
alvarolb 4:de51256455f7 508 do{
alvarolb 4:de51256455f7 509 byte = ((uint8_t*)value_)[pos];
alvarolb 4:de51256455f7 510 value |= (uint64_t)(byte&0x7F) << pos*7;
alvarolb 4:de51256455f7 511 pos++;
alvarolb 4:de51256455f7 512 }while(byte>=0x80);
alvarolb 4:de51256455f7 513 return value;
alvarolb 4:de51256455f7 514 }
alvarolb 4:de51256455f7 515
alvarolb 4:de51256455f7 516 private:
alvarolb 4:de51256455f7 517 void* value_;
alvarolb 4:de51256455f7 518 field_type field_type_;
alvarolb 4:de51256455f7 519
alvarolb 4:de51256455f7 520 template<class T>
alvarolb 4:de51256455f7 521 void set(T value) {
alvarolb 4:de51256455f7 522 memcpy(allocate(sizeof(T)), &value, sizeof(T));
alvarolb 4:de51256455f7 523 }
alvarolb 4:de51256455f7 524 };
alvarolb 4:de51256455f7 525
alvarolb 4:de51256455f7 526 class pson_pair{
alvarolb 4:de51256455f7 527 private:
alvarolb 4:de51256455f7 528 char* name_;
alvarolb 4:de51256455f7 529 pson value_;
alvarolb 4:de51256455f7 530 public:
alvarolb 4:de51256455f7 531 pson_pair() : name_(NULL){
alvarolb 4:de51256455f7 532 }
alvarolb 4:de51256455f7 533
alvarolb 4:de51256455f7 534 ~pson_pair(){
alvarolb 4:de51256455f7 535 destroy(name_, pool);
alvarolb 4:de51256455f7 536 }
alvarolb 4:de51256455f7 537
alvarolb 4:de51256455f7 538 void set_name(const char *name) {
alvarolb 4:de51256455f7 539 size_t name_size = strlen(name) + 1;
alvarolb 4:de51256455f7 540 memcpy(allocate_name(name_size), name, name_size);
alvarolb 4:de51256455f7 541 }
alvarolb 4:de51256455f7 542
alvarolb 4:de51256455f7 543 char* allocate_name(size_t size){
alvarolb 4:de51256455f7 544 name_ = (char*)pool.allocate(size);
alvarolb 4:de51256455f7 545 return name_;
alvarolb 4:de51256455f7 546 }
alvarolb 4:de51256455f7 547
alvarolb 4:de51256455f7 548 pson& value(){
alvarolb 4:de51256455f7 549 return value_;
alvarolb 4:de51256455f7 550 }
alvarolb 4:de51256455f7 551
alvarolb 4:de51256455f7 552 char* name() const{
alvarolb 4:de51256455f7 553 return name_;
alvarolb 4:de51256455f7 554 }
alvarolb 4:de51256455f7 555 };
alvarolb 4:de51256455f7 556
alvarolb 4:de51256455f7 557 class pson_object : public pson_container<pson_pair> {
alvarolb 4:de51256455f7 558 public:
alvarolb 4:de51256455f7 559 pson &operator[](const char *name) {
alvarolb 4:de51256455f7 560 for(iterator it=begin(); it.valid(); it.next()){
alvarolb 4:de51256455f7 561 if(strcmp(it.item().name(), name)==0){
alvarolb 4:de51256455f7 562 return it.item().value();
alvarolb 4:de51256455f7 563 }
alvarolb 4:de51256455f7 564 }
alvarolb 4:de51256455f7 565 pson_pair & pair = create_item();
alvarolb 4:de51256455f7 566 pair.set_name(name);
alvarolb 4:de51256455f7 567 return pair.value();
alvarolb 4:de51256455f7 568 };
alvarolb 4:de51256455f7 569 };
alvarolb 4:de51256455f7 570
alvarolb 4:de51256455f7 571 class pson_array : public pson_container<pson> {
alvarolb 4:de51256455f7 572 public:
alvarolb 4:de51256455f7 573 template<class T>
alvarolb 4:de51256455f7 574 pson_array& add(T item_value){
alvarolb 4:de51256455f7 575 create_item() = item_value;
alvarolb 4:de51256455f7 576 return *this;
alvarolb 4:de51256455f7 577 }
alvarolb 4:de51256455f7 578 };
alvarolb 4:de51256455f7 579
alvarolb 4:de51256455f7 580 inline pson::operator pson_object &() {
alvarolb 4:de51256455f7 581 if (field_type_ != object_field) {
alvarolb 4:de51256455f7 582 value_ = new(pool) pson_object;
alvarolb 4:de51256455f7 583 field_type_ = object_field;
alvarolb 4:de51256455f7 584 }
alvarolb 4:de51256455f7 585 return *((pson_object *)value_);
alvarolb 4:de51256455f7 586 }
alvarolb 4:de51256455f7 587
alvarolb 4:de51256455f7 588 inline pson::operator pson_array &() {
alvarolb 4:de51256455f7 589 if (field_type_ != array_field) {
alvarolb 4:de51256455f7 590 value_ = new(pool) pson_array;
alvarolb 4:de51256455f7 591 field_type_ = array_field;
alvarolb 4:de51256455f7 592 }
alvarolb 4:de51256455f7 593 return *((pson_array *)value_);
alvarolb 4:de51256455f7 594 }
alvarolb 4:de51256455f7 595
alvarolb 4:de51256455f7 596 inline pson &pson::operator[](const char *name) {
alvarolb 4:de51256455f7 597 return ((pson_object &) *this)[name];
alvarolb 4:de51256455f7 598 }
alvarolb 4:de51256455f7 599
alvarolb 4:de51256455f7 600 ////////////////////////////
alvarolb 4:de51256455f7 601 /////// PSON_DECODER ///////
alvarolb 4:de51256455f7 602 ////////////////////////////
alvarolb 4:de51256455f7 603
alvarolb 4:de51256455f7 604 class pson_decoder {
alvarolb 4:de51256455f7 605
alvarolb 4:de51256455f7 606 protected:
alvarolb 4:de51256455f7 607 size_t read_;
alvarolb 4:de51256455f7 608
alvarolb 4:de51256455f7 609 virtual bool read(void* buffer, size_t size){
alvarolb 4:de51256455f7 610 read_+=size;
alvarolb 4:de51256455f7 611 return true;
alvarolb 4:de51256455f7 612 }
alvarolb 4:de51256455f7 613
alvarolb 4:de51256455f7 614 public:
alvarolb 4:de51256455f7 615
alvarolb 4:de51256455f7 616 pson_decoder() : read_(0) {
alvarolb 4:de51256455f7 617
alvarolb 4:de51256455f7 618 }
alvarolb 4:de51256455f7 619
alvarolb 4:de51256455f7 620 void reset(){
alvarolb 4:de51256455f7 621 read_ = 0;
alvarolb 4:de51256455f7 622 }
alvarolb 4:de51256455f7 623
alvarolb 4:de51256455f7 624 size_t bytes_read(){
alvarolb 4:de51256455f7 625 return read_;
alvarolb 4:de51256455f7 626 }
alvarolb 4:de51256455f7 627
alvarolb 4:de51256455f7 628 bool pb_decode_tag(pb_wire_type& wire_type, uint32_t& field_number)
alvarolb 4:de51256455f7 629 {
alvarolb 4:de51256455f7 630 uint32_t temp = pb_decode_varint32();
alvarolb 4:de51256455f7 631 wire_type = (pb_wire_type)(temp & 0x07);
alvarolb 4:de51256455f7 632 field_number = temp >> 3;
alvarolb 4:de51256455f7 633 return true;
alvarolb 4:de51256455f7 634 }
alvarolb 4:de51256455f7 635
alvarolb 4:de51256455f7 636 uint32_t pb_decode_varint32()
alvarolb 4:de51256455f7 637 {
alvarolb 4:de51256455f7 638 uint32_t varint = 0;
alvarolb 4:de51256455f7 639 if(try_pb_decode_varint32(varint)){
alvarolb 4:de51256455f7 640 return varint;
alvarolb 4:de51256455f7 641 }
alvarolb 4:de51256455f7 642 return 0;
alvarolb 4:de51256455f7 643 }
alvarolb 4:de51256455f7 644
alvarolb 4:de51256455f7 645 bool try_pb_decode_varint32(uint32_t& varint){
alvarolb 4:de51256455f7 646 varint = 0;
alvarolb 4:de51256455f7 647 uint8_t byte;
alvarolb 4:de51256455f7 648 uint8_t bit_pos = 0;
alvarolb 4:de51256455f7 649 do{
alvarolb 4:de51256455f7 650 if(!read(&byte, 1) || bit_pos>=32){
alvarolb 4:de51256455f7 651 return false;
alvarolb 4:de51256455f7 652 }
alvarolb 4:de51256455f7 653 varint |= (uint32_t)(byte&0x7F) << bit_pos;
alvarolb 4:de51256455f7 654 bit_pos += 7;
alvarolb 4:de51256455f7 655 }while(byte>=0x80);
alvarolb 4:de51256455f7 656 return true;
alvarolb 4:de51256455f7 657 }
alvarolb 4:de51256455f7 658
alvarolb 4:de51256455f7 659 uint64_t pb_decode_varint64()
alvarolb 4:de51256455f7 660 {
alvarolb 4:de51256455f7 661 uint64_t varint = 0;
alvarolb 4:de51256455f7 662 uint8_t byte;
alvarolb 4:de51256455f7 663 uint8_t bit_pos = 0;
alvarolb 4:de51256455f7 664 do{
alvarolb 4:de51256455f7 665 if(!read(&byte, 1) || bit_pos>=64){
alvarolb 4:de51256455f7 666 return varint;
alvarolb 4:de51256455f7 667 }
alvarolb 4:de51256455f7 668 varint |= (uint32_t)(byte&0x7F) << bit_pos;
alvarolb 4:de51256455f7 669 bit_pos += 7;
alvarolb 4:de51256455f7 670 }while(byte>=0x80);
alvarolb 4:de51256455f7 671 return varint;
alvarolb 4:de51256455f7 672 }
alvarolb 4:de51256455f7 673
alvarolb 4:de51256455f7 674 bool pb_skip(size_t size){
alvarolb 4:de51256455f7 675 uint8_t byte;
alvarolb 4:de51256455f7 676 bool success = true;
alvarolb 4:de51256455f7 677 for(size_t i=0; i<size; i++){
alvarolb 4:de51256455f7 678 success &= read(&byte, 1);
alvarolb 4:de51256455f7 679 }
alvarolb 4:de51256455f7 680 return success;
alvarolb 4:de51256455f7 681 }
alvarolb 4:de51256455f7 682
alvarolb 4:de51256455f7 683 bool pb_skip_varint(){
alvarolb 4:de51256455f7 684 uint8_t byte;
alvarolb 4:de51256455f7 685 bool success = true;
alvarolb 4:de51256455f7 686 do{
alvarolb 4:de51256455f7 687 success &= read(&byte, 1);
alvarolb 4:de51256455f7 688 }while(byte>0x80);
alvarolb 4:de51256455f7 689 return success;
alvarolb 4:de51256455f7 690 }
alvarolb 4:de51256455f7 691
alvarolb 4:de51256455f7 692 bool pb_read_string(char *str, size_t size){
alvarolb 4:de51256455f7 693 bool success = read(str, size);
alvarolb 4:de51256455f7 694 str[size]=0;
alvarolb 4:de51256455f7 695 return success;
alvarolb 4:de51256455f7 696 }
alvarolb 4:de51256455f7 697
alvarolb 4:de51256455f7 698 bool pb_read_varint(pson& value)
alvarolb 4:de51256455f7 699 {
alvarolb 4:de51256455f7 700 uint8_t temp[10];
alvarolb 4:de51256455f7 701 uint8_t byte=0;
alvarolb 4:de51256455f7 702 uint8_t bytes_read=0;
alvarolb 4:de51256455f7 703 do{
alvarolb 4:de51256455f7 704 if(!read(&byte, 1)) return false;
alvarolb 4:de51256455f7 705 temp[bytes_read] = byte;
alvarolb 4:de51256455f7 706 bytes_read++;
alvarolb 4:de51256455f7 707 }while(byte>=0x80);
alvarolb 4:de51256455f7 708 memcpy(value.allocate(bytes_read), temp, bytes_read);
alvarolb 4:de51256455f7 709 return true;
alvarolb 4:de51256455f7 710 }
alvarolb 4:de51256455f7 711
alvarolb 4:de51256455f7 712 public:
alvarolb 4:de51256455f7 713
alvarolb 4:de51256455f7 714 void decode(pson_object & object, size_t size){
alvarolb 4:de51256455f7 715 size_t start_read = bytes_read();
alvarolb 4:de51256455f7 716 while(size-(bytes_read()-start_read)>0){
alvarolb 4:de51256455f7 717 decode(object.create_item());
alvarolb 4:de51256455f7 718 }
alvarolb 4:de51256455f7 719 }
alvarolb 4:de51256455f7 720
alvarolb 4:de51256455f7 721 void decode(pson_array & array, size_t size){
alvarolb 4:de51256455f7 722 size_t start_read = bytes_read();
alvarolb 4:de51256455f7 723 while(size-(bytes_read()-start_read)>0){
alvarolb 4:de51256455f7 724 decode(array.create_item());
alvarolb 4:de51256455f7 725 }
alvarolb 4:de51256455f7 726 }
alvarolb 4:de51256455f7 727
alvarolb 4:de51256455f7 728 void decode(pson_pair & pair){
alvarolb 4:de51256455f7 729 uint32_t name_size = pb_decode_varint32();
alvarolb 4:de51256455f7 730 pb_read_string(pair.allocate_name(name_size+1), name_size);
alvarolb 4:de51256455f7 731 decode(pair.value());
alvarolb 4:de51256455f7 732 }
alvarolb 4:de51256455f7 733
alvarolb 4:de51256455f7 734 void decode(pson& value) {
alvarolb 4:de51256455f7 735 uint32_t field_number;
alvarolb 4:de51256455f7 736 pb_wire_type wire_type;
alvarolb 4:de51256455f7 737 pb_decode_tag(wire_type, field_number);
alvarolb 4:de51256455f7 738 value.set_type((pson::field_type)field_number);
alvarolb 4:de51256455f7 739 if(wire_type==length_delimited){
alvarolb 4:de51256455f7 740 uint32_t size = pb_decode_varint32();
alvarolb 4:de51256455f7 741 switch(field_number){
alvarolb 4:de51256455f7 742 case pson::string_field:
alvarolb 4:de51256455f7 743 pb_read_string((char*)value.allocate(size + 1), size);
alvarolb 4:de51256455f7 744 break;
alvarolb 4:de51256455f7 745 case pson::bytes_field: {
alvarolb 4:de51256455f7 746 uint8_t varint_size = value.get_varint_size(size);
alvarolb 4:de51256455f7 747 read((char*)value.allocate(size + varint_size) + varint_size, size);
alvarolb 4:de51256455f7 748 value.pb_encode_varint(size);
alvarolb 4:de51256455f7 749 }
alvarolb 4:de51256455f7 750 break;
alvarolb 4:de51256455f7 751 case pson::object_field:
alvarolb 4:de51256455f7 752 value.set_value(new (pool) pson_object);
alvarolb 4:de51256455f7 753 decode(*(pson_object *)value.get_value(), size);
alvarolb 4:de51256455f7 754 break;
alvarolb 4:de51256455f7 755 case pson::array_field:
alvarolb 4:de51256455f7 756 value.set_value(new (pool) pson_array);
alvarolb 4:de51256455f7 757 decode(*(pson_array *)value.get_value(), size);
alvarolb 4:de51256455f7 758 break;
alvarolb 4:de51256455f7 759 default:
alvarolb 4:de51256455f7 760 pb_skip(size);
alvarolb 4:de51256455f7 761 break;
alvarolb 4:de51256455f7 762 }
alvarolb 4:de51256455f7 763 }else {
alvarolb 4:de51256455f7 764 switch (field_number) {
alvarolb 4:de51256455f7 765 case pson::svarint_field:
alvarolb 4:de51256455f7 766 case pson::varint_field:
alvarolb 4:de51256455f7 767 pb_read_varint(value);
alvarolb 4:de51256455f7 768 break;
alvarolb 4:de51256455f7 769 case pson::float_field:
alvarolb 4:de51256455f7 770 read(value.allocate(4), 4);
alvarolb 4:de51256455f7 771 break;
alvarolb 4:de51256455f7 772 case pson::double_field:
alvarolb 4:de51256455f7 773 read(value.allocate(8), 8);
alvarolb 4:de51256455f7 774 break;
alvarolb 4:de51256455f7 775 default:
alvarolb 4:de51256455f7 776 break;
alvarolb 4:de51256455f7 777 }
alvarolb 4:de51256455f7 778 }
alvarolb 4:de51256455f7 779 }
alvarolb 4:de51256455f7 780 };
alvarolb 4:de51256455f7 781
alvarolb 4:de51256455f7 782 ////////////////////////////
alvarolb 4:de51256455f7 783 /////// PSON_ENCODER ///////
alvarolb 4:de51256455f7 784 ////////////////////////////
alvarolb 4:de51256455f7 785
alvarolb 4:de51256455f7 786 class pson_encoder {
alvarolb 4:de51256455f7 787
alvarolb 4:de51256455f7 788 protected:
alvarolb 4:de51256455f7 789 size_t written_;
alvarolb 4:de51256455f7 790
alvarolb 4:de51256455f7 791 virtual void write(const void* buffer, size_t size){
alvarolb 4:de51256455f7 792 written_+=size;
alvarolb 4:de51256455f7 793 }
alvarolb 4:de51256455f7 794
alvarolb 4:de51256455f7 795 public:
alvarolb 4:de51256455f7 796
alvarolb 4:de51256455f7 797 pson_encoder() : written_(0) {
alvarolb 4:de51256455f7 798 }
alvarolb 4:de51256455f7 799
alvarolb 4:de51256455f7 800 void reset(){
alvarolb 4:de51256455f7 801 written_ = 0;
alvarolb 4:de51256455f7 802 }
alvarolb 4:de51256455f7 803
alvarolb 4:de51256455f7 804 size_t bytes_written(){
alvarolb 4:de51256455f7 805 return written_;
alvarolb 4:de51256455f7 806 }
alvarolb 4:de51256455f7 807
alvarolb 4:de51256455f7 808 void pb_encode_tag(pb_wire_type wire_type, uint32_t field_number){
alvarolb 4:de51256455f7 809 uint64_t tag = ((uint64_t)field_number << 3) | wire_type;
alvarolb 4:de51256455f7 810 pb_encode_varint(tag);
alvarolb 4:de51256455f7 811 }
alvarolb 4:de51256455f7 812
alvarolb 4:de51256455f7 813 void pb_encode_varint(uint32_t field, uint64_t value)
alvarolb 4:de51256455f7 814 {
alvarolb 4:de51256455f7 815 pb_encode_tag(varint, field);
alvarolb 4:de51256455f7 816 pb_encode_varint(value);
alvarolb 4:de51256455f7 817 }
alvarolb 4:de51256455f7 818
alvarolb 4:de51256455f7 819 uint8_t pb_write_varint(void * buffer)
alvarolb 4:de51256455f7 820 {
alvarolb 4:de51256455f7 821 uint8_t byte=0;
alvarolb 4:de51256455f7 822 uint8_t bytes_written=0;
alvarolb 4:de51256455f7 823 do{
alvarolb 4:de51256455f7 824 byte = *((uint8_t*)buffer + bytes_written);
alvarolb 4:de51256455f7 825 bytes_written++;
alvarolb 4:de51256455f7 826 }while(byte>=0x80);
alvarolb 4:de51256455f7 827 write(buffer, bytes_written);
alvarolb 4:de51256455f7 828 return bytes_written;
alvarolb 4:de51256455f7 829 }
alvarolb 4:de51256455f7 830
alvarolb 4:de51256455f7 831 void pb_encode_varint(uint64_t value)
alvarolb 4:de51256455f7 832 {
alvarolb 4:de51256455f7 833 do
alvarolb 4:de51256455f7 834 {
alvarolb 4:de51256455f7 835 uint8_t byte = (uint8_t)(value & 0x7F);
alvarolb 4:de51256455f7 836 value >>= 7;
alvarolb 4:de51256455f7 837 if(value>0) byte |= 0x80;
alvarolb 4:de51256455f7 838 write(&byte, 1);
alvarolb 4:de51256455f7 839 }while(value>0);
alvarolb 4:de51256455f7 840 }
alvarolb 4:de51256455f7 841
alvarolb 4:de51256455f7 842 void pb_encode_string(const char* str, uint32_t field_number){
alvarolb 4:de51256455f7 843 pb_encode_tag(length_delimited, field_number);
alvarolb 4:de51256455f7 844 pb_encode_string(str);
alvarolb 4:de51256455f7 845 }
alvarolb 4:de51256455f7 846
alvarolb 4:de51256455f7 847 void pb_encode_string(const char* str){
alvarolb 4:de51256455f7 848 size_t string_size = strlen(str);
alvarolb 4:de51256455f7 849 pb_encode_varint(string_size);
alvarolb 4:de51256455f7 850 write(str, string_size);
alvarolb 4:de51256455f7 851 }
alvarolb 4:de51256455f7 852
alvarolb 4:de51256455f7 853 template<class T>
alvarolb 4:de51256455f7 854 void pb_encode_submessage(T& element, uint32_t field_number)
alvarolb 4:de51256455f7 855 {
alvarolb 4:de51256455f7 856 pb_encode_tag(length_delimited, field_number);
alvarolb 4:de51256455f7 857 pson_encoder sink;
alvarolb 4:de51256455f7 858 sink.encode(element);
alvarolb 4:de51256455f7 859 pb_encode_varint(sink.bytes_written());
alvarolb 4:de51256455f7 860 encode(element);
alvarolb 4:de51256455f7 861 }
alvarolb 4:de51256455f7 862
alvarolb 4:de51256455f7 863 void pb_encode_fixed32(void* value){
alvarolb 4:de51256455f7 864 write(value, 4);
alvarolb 4:de51256455f7 865 }
alvarolb 4:de51256455f7 866
alvarolb 4:de51256455f7 867 void pb_encode_fixed64(void* value){
alvarolb 4:de51256455f7 868 write(value, 8);
alvarolb 4:de51256455f7 869 }
alvarolb 4:de51256455f7 870
alvarolb 4:de51256455f7 871 void pb_encode_fixed32(uint32_t field, void*value)
alvarolb 4:de51256455f7 872 {
alvarolb 4:de51256455f7 873 pb_encode_tag(fixed_32, field);
alvarolb 4:de51256455f7 874 pb_encode_fixed32(value);
alvarolb 4:de51256455f7 875 }
alvarolb 4:de51256455f7 876
alvarolb 4:de51256455f7 877 void pb_encode_fixed64(uint32_t field, void*value)
alvarolb 4:de51256455f7 878 {
alvarolb 4:de51256455f7 879 pb_encode_tag(fixed_64, field);
alvarolb 4:de51256455f7 880 pb_encode_fixed64(value);
alvarolb 4:de51256455f7 881 }
alvarolb 4:de51256455f7 882
alvarolb 4:de51256455f7 883 public:
alvarolb 4:de51256455f7 884
alvarolb 4:de51256455f7 885 void encode(pson_object & object){
alvarolb 4:de51256455f7 886 pson_container<pson_pair>::iterator it = object.begin();
alvarolb 4:de51256455f7 887 while(it.valid()){
alvarolb 4:de51256455f7 888 encode(it.item());
alvarolb 4:de51256455f7 889 it.next();
alvarolb 4:de51256455f7 890 }
alvarolb 4:de51256455f7 891 }
alvarolb 4:de51256455f7 892
alvarolb 4:de51256455f7 893 void encode(pson_array & array){
alvarolb 4:de51256455f7 894 pson_container<pson>::iterator it = array.begin();
alvarolb 4:de51256455f7 895 while(it.valid()){
alvarolb 4:de51256455f7 896 encode(it.item());
alvarolb 4:de51256455f7 897 it.next();
alvarolb 4:de51256455f7 898 }
alvarolb 4:de51256455f7 899 }
alvarolb 4:de51256455f7 900
alvarolb 4:de51256455f7 901 void encode(pson_pair & pair){
alvarolb 4:de51256455f7 902 pb_encode_string(pair.name());
alvarolb 4:de51256455f7 903 encode(pair.value());
alvarolb 4:de51256455f7 904 }
alvarolb 4:de51256455f7 905
alvarolb 4:de51256455f7 906 void encode(pson & value) {
alvarolb 4:de51256455f7 907 switch (value.get_type()) {
alvarolb 4:de51256455f7 908 case pson::string_field:
alvarolb 4:de51256455f7 909 pb_encode_string((const char*)value.get_value(), pson::string_field);
alvarolb 4:de51256455f7 910 break;
alvarolb 4:de51256455f7 911 case pson::bytes_field:
alvarolb 4:de51256455f7 912 pb_encode_tag(length_delimited, pson::bytes_field);
alvarolb 4:de51256455f7 913 write(((const char *) value.get_value()) + pb_write_varint(value.get_value()), value.pb_decode_varint());
alvarolb 4:de51256455f7 914 break;
alvarolb 4:de51256455f7 915 case pson::svarint_field:
alvarolb 4:de51256455f7 916 case pson::varint_field:
alvarolb 4:de51256455f7 917 pb_encode_tag(varint, value.get_type());
alvarolb 4:de51256455f7 918 pb_write_varint(value.get_value());
alvarolb 4:de51256455f7 919 break;
alvarolb 4:de51256455f7 920 case pson::float_field:
alvarolb 4:de51256455f7 921 pb_encode_fixed32(pson::float_field, value.get_value());
alvarolb 4:de51256455f7 922 break;
alvarolb 4:de51256455f7 923 case pson::double_field:
alvarolb 4:de51256455f7 924 pb_encode_fixed64(pson::double_field, value.get_value());
alvarolb 4:de51256455f7 925 break;
alvarolb 4:de51256455f7 926 case pson::object_field:
alvarolb 4:de51256455f7 927 pb_encode_submessage(*(pson_object *) value.get_value(), pson::object_field);
alvarolb 4:de51256455f7 928 break;
alvarolb 4:de51256455f7 929 case pson::array_field:
alvarolb 4:de51256455f7 930 pb_encode_submessage(*(pson_array *) value.get_value(), pson::array_field);
alvarolb 4:de51256455f7 931 break;
alvarolb 4:de51256455f7 932 default:
alvarolb 4:de51256455f7 933 pb_encode_tag(varint, value.get_type());
alvarolb 4:de51256455f7 934 break;
alvarolb 4:de51256455f7 935 }
alvarolb 4:de51256455f7 936 }
alvarolb 4:de51256455f7 937 };
alvarolb 4:de51256455f7 938 }
alvarolb 4:de51256455f7 939
alvarolb 4:de51256455f7 940 #endif