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@0:b75d784c7c1a, 2015-12-24 (annotated)

Committer:
alvarolb
Date:
Thu Dec 24 13:18:08 2015 +0000
Revision:
0:b75d784c7c1a
Child:
4:de51256455f7
Initial Commit

Who changed what in which revision?

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