1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072 |
- //
- // MessagePack for C++ deserializing routine
- //
- // Copyright (C) 2016-2017 KONDO Takatoshi
- //
- // Distributed under the Boost Software License, Version 1.0.
- // (See accompanying file LICENSE_1_0.txt or copy at
- // http://www.boost.org/LICENSE_1_0.txt)
- //
- #ifndef MSGPACK_V2_PARSE_HPP
- #define MSGPACK_V2_PARSE_HPP
- #if MSGPACK_DEFAULT_API_VERSION >= 2
- #include <cstddef>
- #include "msgpack/unpack_define.h"
- #include "msgpack/parse_return.hpp"
- #include "msgpack/unpack_exception.hpp"
- #include "msgpack/unpack_decl.hpp"
- namespace msgpack {
- /// @cond
- MSGPACK_API_VERSION_NAMESPACE(v2) {
- /// @endcond
- namespace detail {
- using v1::detail::fix_tag;
- using v1::detail::value;
- using v1::detail::load;
- template <typename VisitorHolder>
- class context {
- public:
- context()
- :m_trail(0), m_cs(MSGPACK_CS_HEADER)
- {
- }
- void init()
- {
- m_cs = MSGPACK_CS_HEADER;
- m_trail = 0;
- m_stack.clear();
- holder().visitor().init();
- }
- parse_return execute(const char* data, std::size_t len, std::size_t& off);
- private:
- template <typename T>
- static uint32_t next_cs(T p)
- {
- return static_cast<uint32_t>(*p) & 0x1f;
- }
- VisitorHolder& holder() {
- return static_cast<VisitorHolder&>(*this);
- }
- template <typename T, typename StartVisitor, typename EndVisitor>
- parse_return start_aggregate(
- StartVisitor const& sv,
- EndVisitor const& ev,
- const char* load_pos,
- std::size_t& off) {
- typename value<T>::type size;
- load<T>(size, load_pos);
- ++m_current;
- if (size == 0) {
- if (!sv(size)) {
- off = m_current - m_start;
- return PARSE_STOP_VISITOR;
- }
- if (!ev()) {
- off = m_current - m_start;
- return PARSE_STOP_VISITOR;
- }
- parse_return ret = m_stack.consume(holder());
- if (ret != PARSE_CONTINUE) {
- off = m_current - m_start;
- return ret;
- }
- }
- else {
- if (!sv(size)) {
- off = m_current - m_start;
- return PARSE_STOP_VISITOR;
- }
- parse_return ret = m_stack.push(holder(), sv.type(), static_cast<uint32_t>(size));
- if (ret != PARSE_CONTINUE) {
- off = m_current - m_start;
- return ret;
- }
- }
- m_cs = MSGPACK_CS_HEADER;
- return PARSE_CONTINUE;
- }
- parse_return after_visit_proc(bool visit_result, std::size_t& off) {
- ++m_current;
- if (!visit_result) {
- off = m_current - m_start;
- return PARSE_STOP_VISITOR;
- }
- parse_return ret = m_stack.consume(holder());
- if (ret != PARSE_CONTINUE) {
- off = m_current - m_start;
- }
- m_cs = MSGPACK_CS_HEADER;
- return ret;
- }
- struct array_sv {
- array_sv(VisitorHolder& visitor_holder):m_visitor_holder(visitor_holder) {}
- bool operator()(uint32_t size) const {
- return m_visitor_holder.visitor().start_array(size);
- }
- msgpack_container_type type() const { return MSGPACK_CT_ARRAY_ITEM; }
- private:
- VisitorHolder& m_visitor_holder;
- };
- struct array_ev {
- array_ev(VisitorHolder& visitor_holder):m_visitor_holder(visitor_holder) {}
- bool operator()() const {
- return m_visitor_holder.visitor().end_array();
- }
- private:
- VisitorHolder& m_visitor_holder;
- };
- struct map_sv {
- map_sv(VisitorHolder& visitor_holder):m_visitor_holder(visitor_holder) {}
- bool operator()(uint32_t size) const {
- return m_visitor_holder.visitor().start_map(size);
- }
- msgpack_container_type type() const { return MSGPACK_CT_MAP_KEY; }
- private:
- VisitorHolder& m_visitor_holder;
- };
- struct map_ev {
- map_ev(VisitorHolder& visitor_holder):m_visitor_holder(visitor_holder) {}
- bool operator()() const {
- return m_visitor_holder.visitor().end_map();
- }
- private:
- VisitorHolder& m_visitor_holder;
- };
- struct unpack_stack {
- struct stack_elem {
- stack_elem(msgpack_container_type type, uint32_t rest):m_type(type), m_rest(rest) {}
- msgpack_container_type m_type;
- uint32_t m_rest;
- };
- unpack_stack() {
- m_stack.reserve(MSGPACK_EMBED_STACK_SIZE);
- }
- parse_return push(VisitorHolder& visitor_holder, msgpack_container_type type, uint32_t rest) {
- m_stack.push_back(stack_elem(type, rest));
- switch (type) {
- case MSGPACK_CT_ARRAY_ITEM:
- return visitor_holder.visitor().start_array_item() ? PARSE_CONTINUE : PARSE_STOP_VISITOR;
- case MSGPACK_CT_MAP_KEY:
- return visitor_holder.visitor().start_map_key() ? PARSE_CONTINUE : PARSE_STOP_VISITOR;
- case MSGPACK_CT_MAP_VALUE:
- assert(0);
- return PARSE_STOP_VISITOR;
- }
- assert(0);
- return PARSE_STOP_VISITOR;
- }
- parse_return consume(VisitorHolder& visitor_holder) {
- while (!m_stack.empty()) {
- stack_elem& e = m_stack.back();
- switch (e.m_type) {
- case MSGPACK_CT_ARRAY_ITEM:
- if (!visitor_holder.visitor().end_array_item()) return PARSE_STOP_VISITOR;
- if (--e.m_rest == 0) {
- m_stack.pop_back();
- if (!visitor_holder.visitor().end_array()) return PARSE_STOP_VISITOR;
- }
- else {
- if (!visitor_holder.visitor().start_array_item()) return PARSE_STOP_VISITOR;
- return PARSE_CONTINUE;
- }
- break;
- case MSGPACK_CT_MAP_KEY:
- if (!visitor_holder.visitor().end_map_key()) return PARSE_STOP_VISITOR;
- if (!visitor_holder.visitor().start_map_value()) return PARSE_STOP_VISITOR;
- e.m_type = MSGPACK_CT_MAP_VALUE;
- return PARSE_CONTINUE;
- case MSGPACK_CT_MAP_VALUE:
- if (!visitor_holder.visitor().end_map_value()) return PARSE_STOP_VISITOR;
- if (--e.m_rest == 0) {
- m_stack.pop_back();
- if (!visitor_holder.visitor().end_map()) return PARSE_STOP_VISITOR;
- }
- else {
- e.m_type = MSGPACK_CT_MAP_KEY;
- if (!visitor_holder.visitor().start_map_key()) return PARSE_STOP_VISITOR;
- return PARSE_CONTINUE;
- }
- break;
- }
- }
- return PARSE_SUCCESS;
- }
- bool empty() const { return m_stack.empty(); }
- void clear() { m_stack.clear(); }
- private:
- std::vector<stack_elem> m_stack;
- };
- char const* m_start;
- char const* m_current;
- std::size_t m_trail;
- uint32_t m_cs;
- uint32_t m_num_elements;
- unpack_stack m_stack;
- };
- template <std::size_t N>
- inline void check_ext_size(std::size_t /*size*/) {
- }
- template <>
- inline void check_ext_size<4>(std::size_t size) {
- if (size == 0xffffffff) throw msgpack::ext_size_overflow("ext size overflow");
- }
- template <typename VisitorHolder>
- inline parse_return context<VisitorHolder>::execute(const char* data, std::size_t len, std::size_t& off)
- {
- assert(len >= off);
- m_start = data;
- m_current = data + off;
- const char* const pe = data + len;
- const char* n = MSGPACK_NULLPTR;
- msgpack::object obj;
- if(m_current == pe) {
- off = m_current - m_start;
- return PARSE_CONTINUE;
- }
- bool fixed_trail_again = false;
- do {
- if (m_cs == MSGPACK_CS_HEADER) {
- fixed_trail_again = false;
- int selector = *reinterpret_cast<const unsigned char*>(m_current);
- if (0x00 <= selector && selector <= 0x7f) { // Positive Fixnum
- uint8_t tmp = *reinterpret_cast<const uint8_t*>(m_current);
- bool visret = holder().visitor().visit_positive_integer(tmp);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } else if(0xe0 <= selector && selector <= 0xff) { // Negative Fixnum
- int8_t tmp = *reinterpret_cast<const int8_t*>(m_current);
- bool visret = holder().visitor().visit_negative_integer(tmp);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } else if (0xc4 <= selector && selector <= 0xdf) {
- const uint32_t trail[] = {
- 1, // bin 8 0xc4
- 2, // bin 16 0xc5
- 4, // bin 32 0xc6
- 1, // ext 8 0xc7
- 2, // ext 16 0xc8
- 4, // ext 32 0xc9
- 4, // float 32 0xca
- 8, // float 64 0xcb
- 1, // uint 8 0xcc
- 2, // uint 16 0xcd
- 4, // uint 32 0xce
- 8, // uint 64 0xcf
- 1, // int 8 0xd0
- 2, // int 16 0xd1
- 4, // int 32 0xd2
- 8, // int 64 0xd3
- 2, // fixext 1 0xd4
- 3, // fixext 2 0xd5
- 5, // fixext 4 0xd6
- 9, // fixext 8 0xd7
- 17,// fixext 16 0xd8
- 1, // str 8 0xd9
- 2, // str 16 0xda
- 4, // str 32 0xdb
- 2, // array 16 0xdc
- 4, // array 32 0xdd
- 2, // map 16 0xde
- 4, // map 32 0xdf
- };
- m_trail = trail[selector - 0xc4];
- m_cs = next_cs(m_current);
- fixed_trail_again = true;
- } else if(0xa0 <= selector && selector <= 0xbf) { // FixStr
- m_trail = static_cast<uint32_t>(*m_current) & 0x1f;
- if(m_trail == 0) {
- bool visret = holder().visitor().visit_str(n, static_cast<uint32_t>(m_trail));
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- }
- else {
- m_cs = MSGPACK_ACS_STR_VALUE;
- fixed_trail_again = true;
- }
- } else if(0x90 <= selector && selector <= 0x9f) { // FixArray
- parse_return ret = start_aggregate<fix_tag>(array_sv(holder()), array_ev(holder()), m_current, off);
- if (ret != PARSE_CONTINUE) return ret;
- } else if(0x80 <= selector && selector <= 0x8f) { // FixMap
- parse_return ret = start_aggregate<fix_tag>(map_sv(holder()), map_ev(holder()), m_current, off);
- if (ret != PARSE_CONTINUE) return ret;
- } else if(selector == 0xc2) { // false
- bool visret = holder().visitor().visit_boolean(false);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } else if(selector == 0xc3) { // true
- bool visret = holder().visitor().visit_boolean(true);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } else if(selector == 0xc0) { // nil
- bool visret = holder().visitor().visit_nil();
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } else {
- off = m_current - m_start;
- holder().visitor().parse_error(off - 1, off);
- return PARSE_PARSE_ERROR;
- }
- // end MSGPACK_CS_HEADER
- }
- if (m_cs != MSGPACK_CS_HEADER || fixed_trail_again) {
- if (fixed_trail_again) {
- ++m_current;
- fixed_trail_again = false;
- }
- if(static_cast<std::size_t>(pe - m_current) < m_trail) {
- off = m_current - m_start;
- return PARSE_CONTINUE;
- }
- n = m_current;
- m_current += m_trail - 1;
- switch(m_cs) {
- //case MSGPACK_CS_
- //case MSGPACK_CS_
- case MSGPACK_CS_FLOAT: {
- union { uint32_t i; float f; } mem;
- load<uint32_t>(mem.i, n);
- bool visret = holder().visitor().visit_float32(mem.f);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_DOUBLE: {
- union { uint64_t i; double f; } mem;
- load<uint64_t>(mem.i, n);
- #if defined(TARGET_OS_IPHONE)
- // ok
- #elif defined(__arm__) && !(__ARM_EABI__) // arm-oabi
- // https://github.com/msgpack/msgpack-perl/pull/1
- mem.i = (mem.i & 0xFFFFFFFFUL) << 32UL | (mem.i >> 32UL);
- #endif
- bool visret = holder().visitor().visit_float64(mem.f);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_UINT_8: {
- uint8_t tmp;
- load<uint8_t>(tmp, n);
- bool visret = holder().visitor().visit_positive_integer(tmp);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_UINT_16: {
- uint16_t tmp;
- load<uint16_t>(tmp, n);
- bool visret = holder().visitor().visit_positive_integer(tmp);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_UINT_32: {
- uint32_t tmp;
- load<uint32_t>(tmp, n);
- bool visret = holder().visitor().visit_positive_integer(tmp);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_UINT_64: {
- uint64_t tmp;
- load<uint64_t>(tmp, n);
- bool visret = holder().visitor().visit_positive_integer(tmp);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_INT_8: {
- int8_t tmp;
- load<int8_t>(tmp, n);
- bool visret = holder().visitor().visit_negative_integer(tmp);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_INT_16: {
- int16_t tmp;
- load<int16_t>(tmp, n);
- bool visret = holder().visitor().visit_negative_integer(tmp);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_INT_32: {
- int32_t tmp;
- load<int32_t>(tmp, n);
- bool visret = holder().visitor().visit_negative_integer(tmp);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_INT_64: {
- int64_t tmp;
- load<int64_t>(tmp, n);
- bool visret = holder().visitor().visit_negative_integer(tmp);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_FIXEXT_1: {
- bool visret = holder().visitor().visit_ext(n, 1+1);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_FIXEXT_2: {
- bool visret = holder().visitor().visit_ext(n, 2+1);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_FIXEXT_4: {
- bool visret = holder().visitor().visit_ext(n, 4+1);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_FIXEXT_8: {
- bool visret = holder().visitor().visit_ext(n, 8+1);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_FIXEXT_16: {
- bool visret = holder().visitor().visit_ext(n, 16+1);
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_STR_8: {
- uint8_t tmp;
- load<uint8_t>(tmp, n);
- m_trail = tmp;
- if(m_trail == 0) {
- bool visret = holder().visitor().visit_str(n, static_cast<uint32_t>(m_trail));
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- }
- else {
- m_cs = MSGPACK_ACS_STR_VALUE;
- fixed_trail_again = true;
- }
- } break;
- case MSGPACK_CS_BIN_8: {
- uint8_t tmp;
- load<uint8_t>(tmp, n);
- m_trail = tmp;
- if(m_trail == 0) {
- bool visret = holder().visitor().visit_bin(n, static_cast<uint32_t>(m_trail));
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- }
- else {
- m_cs = MSGPACK_ACS_BIN_VALUE;
- fixed_trail_again = true;
- }
- } break;
- case MSGPACK_CS_EXT_8: {
- uint8_t tmp;
- load<uint8_t>(tmp, n);
- m_trail = tmp + 1;
- if(m_trail == 0) {
- bool visret = holder().visitor().visit_ext(n, static_cast<uint32_t>(m_trail));
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- }
- else {
- m_cs = MSGPACK_ACS_EXT_VALUE;
- fixed_trail_again = true;
- }
- } break;
- case MSGPACK_CS_STR_16: {
- uint16_t tmp;
- load<uint16_t>(tmp, n);
- m_trail = tmp;
- if(m_trail == 0) {
- bool visret = holder().visitor().visit_str(n, static_cast<uint32_t>(m_trail));
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- }
- else {
- m_cs = MSGPACK_ACS_STR_VALUE;
- fixed_trail_again = true;
- }
- } break;
- case MSGPACK_CS_BIN_16: {
- uint16_t tmp;
- load<uint16_t>(tmp, n);
- m_trail = tmp;
- if(m_trail == 0) {
- bool visret = holder().visitor().visit_bin(n, static_cast<uint32_t>(m_trail));
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- }
- else {
- m_cs = MSGPACK_ACS_BIN_VALUE;
- fixed_trail_again = true;
- }
- } break;
- case MSGPACK_CS_EXT_16: {
- uint16_t tmp;
- load<uint16_t>(tmp, n);
- m_trail = tmp + 1;
- if(m_trail == 0) {
- bool visret = holder().visitor().visit_ext(n, static_cast<uint32_t>(m_trail));
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- }
- else {
- m_cs = MSGPACK_ACS_EXT_VALUE;
- fixed_trail_again = true;
- }
- } break;
- case MSGPACK_CS_STR_32: {
- uint32_t tmp;
- load<uint32_t>(tmp, n);
- m_trail = tmp;
- if(m_trail == 0) {
- bool visret = holder().visitor().visit_str(n, static_cast<uint32_t>(m_trail));
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- }
- else {
- m_cs = MSGPACK_ACS_STR_VALUE;
- fixed_trail_again = true;
- }
- } break;
- case MSGPACK_CS_BIN_32: {
- uint32_t tmp;
- load<uint32_t>(tmp, n);
- m_trail = tmp;
- if(m_trail == 0) {
- bool visret = holder().visitor().visit_bin(n, static_cast<uint32_t>(m_trail));
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- }
- else {
- m_cs = MSGPACK_ACS_BIN_VALUE;
- fixed_trail_again = true;
- }
- } break;
- case MSGPACK_CS_EXT_32: {
- uint32_t tmp;
- load<uint32_t>(tmp, n);
- check_ext_size<sizeof(std::size_t)>(tmp);
- m_trail = tmp;
- ++m_trail;
- if(m_trail == 0) {
- bool visret = holder().visitor().visit_ext(n, static_cast<uint32_t>(m_trail));
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- }
- else {
- m_cs = MSGPACK_ACS_EXT_VALUE;
- fixed_trail_again = true;
- }
- } break;
- case MSGPACK_ACS_STR_VALUE: {
- bool visret = holder().visitor().visit_str(n, static_cast<uint32_t>(m_trail));
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_ACS_BIN_VALUE: {
- bool visret = holder().visitor().visit_bin(n, static_cast<uint32_t>(m_trail));
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_ACS_EXT_VALUE: {
- bool visret = holder().visitor().visit_ext(n, static_cast<uint32_t>(m_trail));
- parse_return upr = after_visit_proc(visret, off);
- if (upr != PARSE_CONTINUE) return upr;
- } break;
- case MSGPACK_CS_ARRAY_16: {
- parse_return ret = start_aggregate<uint16_t>(array_sv(holder()), array_ev(holder()), n, off);
- if (ret != PARSE_CONTINUE) return ret;
- } break;
- case MSGPACK_CS_ARRAY_32: {
- parse_return ret = start_aggregate<uint32_t>(array_sv(holder()), array_ev(holder()), n, off);
- if (ret != PARSE_CONTINUE) return ret;
- } break;
- case MSGPACK_CS_MAP_16: {
- parse_return ret = start_aggregate<uint16_t>(map_sv(holder()), map_ev(holder()), n, off);
- if (ret != PARSE_CONTINUE) return ret;
- } break;
- case MSGPACK_CS_MAP_32: {
- parse_return ret = start_aggregate<uint32_t>(map_sv(holder()), map_ev(holder()), n, off);
- if (ret != PARSE_CONTINUE) return ret;
- } break;
- default:
- off = m_current - m_start;
- holder().visitor().parse_error(n - m_start - 1, n - m_start);
- return PARSE_PARSE_ERROR;
- }
- }
- } while(m_current != pe);
- off = m_current - m_start;
- return PARSE_CONTINUE;
- }
- } // detail
- /// Parsing class for a stream deserialization.
- template <typename VisitorHolder, typename ReferencedBufferHook>
- class parser : public detail::context<VisitorHolder> {
- typedef parser<VisitorHolder, ReferencedBufferHook> this_type;
- typedef detail::context<VisitorHolder> context_type;
- public:
- /// Constructor
- /**
- * @param hook The handler that is called when buffer is allocated internally.
- * `hook` should be callable with char* parameter.
- * `parser` gives a chance to prepare finalizer.
- * See https://github.com/msgpack/msgpack-c/wiki/v2_0_cpp_visitor#parse-api
- * @param initial_buffer_size The memory size to allocate when unpacker is constructed.
- *
- */
- parser(ReferencedBufferHook& hook,
- std::size_t initial_buffer_size = MSGPACK_UNPACKER_INIT_BUFFER_SIZE);
- #if !defined(MSGPACK_USE_CPP03)
- parser(this_type&& other);
- this_type& operator=(this_type&& other);
- #endif // !defined(MSGPACK_USE_CPP03)
- ~parser();
- public:
- /// Reserve a buffer memory.
- /**
- * @param size The size of allocating memory.
- *
- * After returning this function, buffer_capacity() returns at least 'size'.
- * See:
- * https://github.com/msgpack/msgpack-c/wiki/v1_1_cpp_unpacker#msgpack-controls-a-buffer
- */
- void reserve_buffer(std::size_t size = MSGPACK_UNPACKER_RESERVE_SIZE);
- /// Get buffer pointer.
- /**
- * You need to care about the memory is enable between buffer() and buffer() + buffer_capacity()
- * See:
- * https://github.com/msgpack/msgpack-c/wiki/v1_1_cpp_unpacker#msgpack-controls-a-buffer
- */
- char* buffer();
- /// Get buffer capacity.
- /**
- * @return The memory size that you can write.
- *
- * See:
- * https://github.com/msgpack/msgpack-c/wiki/v1_1_cpp_unpacker#msgpack-controls-a-buffer
- */
- std::size_t buffer_capacity() const;
- /// Notify a buffer consumed information to msgpack::unpacker.
- /**
- * @param size The size of memory that you consumed.
- *
- * After copying the data to the memory that is pointed by buffer(), you need to call the
- * function to notify how many bytes are consumed. Then you can call next() functions.
- *
- * See:
- * https://github.com/msgpack/msgpack-c/wiki/v1_1_cpp_unpacker#msgpack-controls-a-buffer
- */
- void buffer_consumed(std::size_t size);
- /// Unpack one msgpack::object.
- /**
- *
- *
- * @return If one msgpack::object is unpacked, then return true, if msgpack::object is incomplete
- * and additional data is required, then return false. If data format is invalid, throw
- * msgpack::parse_error.
- *
- * See:
- * https://github.com/msgpack/msgpack-c/wiki/v1_1_cpp_unpacker#msgpack-controls-a-buffer
- */
- bool next();
- /// Get message size.
- /**
- * @return Returns parsed_size() + nonparsed_size()
- */
- std::size_t message_size() const;
- public:
- /// Get parsed message size.
- /**
- * @return Parsed message size.
- *
- * This function is usable when non-MessagePack message follows after
- * MessagePack message.
- */
- std::size_t parsed_size() const;
- /// Get the address that is not parsed in the buffer.
- /**
- * @return Address of the buffer that is not parsed
- *
- * This function is usable when non-MessagePack message follows after
- * MessagePack message.
- */
- char* nonparsed_buffer();
- /// Get the size of the buffer that is not parsed.
- /**
- * @return Size of the buffer that is not parsed
- *
- * This function is usable when non-MessagePack message follows after
- * MessagePack message.
- */
- std::size_t nonparsed_size() const;
- /// Skip the specified size of non-parsed buffer.
- /**
- * @param size to skip
- *
- * Note that the `size' argument must be smaller than nonparsed_size().
- * This function is usable when non-MessagePack message follows after
- * MessagePack message.
- */
- void skip_nonparsed_buffer(std::size_t size);
- /// Remove nonparsed buffer and reset the current position as a new start point.
- /**
- * This function is usable when non-MessagePack message follows after
- * MessagePack message.
- */
- void remove_nonparsed_buffer();
- void reset();
- protected:
- char* get_raw_buffer() {
- return m_buffer;
- }
- private:
- void expand_buffer(std::size_t size);
- parse_return execute_imp();
- private:
- char* m_buffer;
- std::size_t m_used;
- std::size_t m_free;
- std::size_t m_off;
- std::size_t m_parsed;
- std::size_t m_initial_buffer_size;
- ReferencedBufferHook& m_referenced_buffer_hook;
- #if defined(MSGPACK_USE_CPP03)
- private:
- parser(const this_type&);
- this_type& operator=(const this_type&);
- #else // defined(MSGPACK_USE_CPP03)
- public:
- parser(const this_type&) = delete;
- this_type& operator=(const this_type&) = delete;
- #endif // defined(MSGPACK_USE_CPP03)
- };
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline parser<VisitorHolder, ReferencedBufferHook>::parser(
- ReferencedBufferHook& hook,
- std::size_t initial_buffer_size)
- :m_referenced_buffer_hook(hook)
- {
- if(initial_buffer_size < COUNTER_SIZE) {
- initial_buffer_size = COUNTER_SIZE;
- }
- char* buffer = static_cast<char*>(::malloc(initial_buffer_size));
- if(!buffer) {
- throw std::bad_alloc();
- }
- m_buffer = buffer;
- m_used = COUNTER_SIZE;
- m_free = initial_buffer_size - m_used;
- m_off = COUNTER_SIZE;
- m_parsed = 0;
- m_initial_buffer_size = initial_buffer_size;
- detail::init_count(m_buffer);
- }
- #if !defined(MSGPACK_USE_CPP03)
- // Move constructor and move assignment operator
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline parser<VisitorHolder, ReferencedBufferHook>::parser(this_type&& other)
- :context_type(std::move(other)),
- m_buffer(other.m_buffer),
- m_used(other.m_used),
- m_free(other.m_free),
- m_off(other.m_off),
- m_parsed(other.m_parsed),
- m_initial_buffer_size(other.m_initial_buffer_size),
- m_referenced_buffer_hook(other.m_referenced_buffer_hook) {
- other.m_buffer = MSGPACK_NULLPTR;
- other.m_used = 0;
- other.m_free = 0;
- other.m_off = 0;
- other.m_parsed = 0;
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline parser<VisitorHolder, ReferencedBufferHook>& parser<VisitorHolder, ReferencedBufferHook>::operator=(this_type&& other) {
- this->~parser();
- new (this) this_type(std::move(other));
- return *this;
- }
- #endif // !defined(MSGPACK_USE_CPP03)
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline parser<VisitorHolder, ReferencedBufferHook>::~parser()
- {
- // These checks are required for move operations.
- if (m_buffer) detail::decr_count(m_buffer);
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline void parser<VisitorHolder, ReferencedBufferHook>::reserve_buffer(std::size_t size)
- {
- if(m_free >= size) return;
- expand_buffer(size);
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline void parser<VisitorHolder, ReferencedBufferHook>::expand_buffer(std::size_t size)
- {
- if(m_used == m_off && detail::get_count(m_buffer) == 1
- && !static_cast<VisitorHolder&>(*this).visitor().referenced()) {
- // rewind buffer
- m_free += m_used - COUNTER_SIZE;
- m_used = COUNTER_SIZE;
- m_off = COUNTER_SIZE;
- if(m_free >= size) return;
- }
- if(m_off == COUNTER_SIZE) {
- std::size_t next_size = (m_used + m_free) * 2; // include COUNTER_SIZE
- while(next_size < size + m_used) {
- std::size_t tmp_next_size = next_size * 2;
- if (tmp_next_size <= next_size) {
- next_size = size + m_used;
- break;
- }
- next_size = tmp_next_size;
- }
- char* tmp = static_cast<char*>(::realloc(m_buffer, next_size));
- if(!tmp) {
- throw std::bad_alloc();
- }
- m_buffer = tmp;
- m_free = next_size - m_used;
- } else {
- std::size_t next_size = m_initial_buffer_size; // include COUNTER_SIZE
- std::size_t not_parsed = m_used - m_off;
- while(next_size < size + not_parsed + COUNTER_SIZE) {
- std::size_t tmp_next_size = next_size * 2;
- if (tmp_next_size <= next_size) {
- next_size = size + not_parsed + COUNTER_SIZE;
- break;
- }
- next_size = tmp_next_size;
- }
- char* tmp = static_cast<char*>(::malloc(next_size));
- if(!tmp) {
- throw std::bad_alloc();
- }
- detail::init_count(tmp);
- std::memcpy(tmp+COUNTER_SIZE, m_buffer + m_off, not_parsed);
- if(static_cast<VisitorHolder&>(*this).referenced()) {
- try {
- m_referenced_buffer_hook(m_buffer);
- }
- catch (...) {
- ::free(tmp);
- throw;
- }
- static_cast<VisitorHolder&>(*this).set_referenced(false);
- } else {
- detail::decr_count(m_buffer);
- }
- m_buffer = tmp;
- m_used = not_parsed + COUNTER_SIZE;
- m_free = next_size - m_used;
- m_off = COUNTER_SIZE;
- }
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline char* parser<VisitorHolder, ReferencedBufferHook>::buffer()
- {
- return m_buffer + m_used;
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline std::size_t parser<VisitorHolder, ReferencedBufferHook>::buffer_capacity() const
- {
- return m_free;
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline void parser<VisitorHolder, ReferencedBufferHook>::buffer_consumed(std::size_t size)
- {
- m_used += size;
- m_free -= size;
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline bool parser<VisitorHolder, ReferencedBufferHook>::next()
- {
- parse_return ret = execute_imp();
- return ret == PARSE_SUCCESS;
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline parse_return parser<VisitorHolder, ReferencedBufferHook>::execute_imp()
- {
- std::size_t off = m_off;
- parse_return ret = context_type::execute(m_buffer, m_used, m_off);
- if(m_off > off) {
- m_parsed += m_off - off;
- }
- return ret;
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline void parser<VisitorHolder, ReferencedBufferHook>::reset()
- {
- context_type::init();
- // don't reset referenced flag
- m_parsed = 0;
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline std::size_t parser<VisitorHolder, ReferencedBufferHook>::message_size() const
- {
- return m_parsed - m_off + m_used;
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline std::size_t parser<VisitorHolder, ReferencedBufferHook>::parsed_size() const
- {
- return m_parsed;
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline char* parser<VisitorHolder, ReferencedBufferHook>::nonparsed_buffer()
- {
- return m_buffer + m_off;
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline std::size_t parser<VisitorHolder, ReferencedBufferHook>::nonparsed_size() const
- {
- return m_used - m_off;
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline void parser<VisitorHolder, ReferencedBufferHook>::skip_nonparsed_buffer(std::size_t size)
- {
- m_off += size;
- }
- template <typename VisitorHolder, typename ReferencedBufferHook>
- inline void parser<VisitorHolder, ReferencedBufferHook>::remove_nonparsed_buffer()
- {
- m_used = m_off;
- }
- template <typename Visitor>
- inline bool parse(const char* data, size_t len, size_t& off, Visitor& v) {
- parse_return ret = msgpack::detail::parse_imp(data, len, off, v);
- return ret == PARSE_SUCCESS || ret == PARSE_EXTRA_BYTES;
- }
- template <typename Visitor>
- inline bool parse(const char* data, size_t len, Visitor& v) {
- std::size_t off = 0;
- return msgpack::parse(data, len, off, v);
- }
- namespace detail {
- template <typename Visitor>
- struct parse_helper : detail::context<parse_helper<Visitor> > {
- parse_helper(Visitor& v):m_visitor(v) {}
- parse_return execute(const char* data, std::size_t len, std::size_t& off) {
- return detail::context<parse_helper<Visitor> >::execute(data, len, off);
- }
- Visitor& visitor() const { return m_visitor; }
- Visitor& m_visitor;
- };
- template <typename Visitor>
- inline parse_return
- parse_imp(const char* data, size_t len, size_t& off, Visitor& v) {
- std::size_t noff = off;
- if(len <= noff) {
- // FIXME
- v.insufficient_bytes(noff, noff);
- return PARSE_CONTINUE;
- }
- detail::parse_helper<Visitor> h(v);
- parse_return ret = h.execute(data, len, noff);
- switch (ret) {
- case PARSE_CONTINUE:
- off = noff;
- v.insufficient_bytes(noff - 1, noff);
- return ret;
- case PARSE_SUCCESS:
- off = noff;
- if(noff < len) {
- return PARSE_EXTRA_BYTES;
- }
- return ret;
- default:
- return ret;
- }
- }
- } // detail
- /// @cond
- } // MSGPACK_API_VERSION_NAMESPACE(v2)
- /// @endcond
- } // namespace msgpack
- #endif // MSGPACK_DEFAULT_API_VERSION >= 2
- #endif // MSGPACK_V2_PARSE_HPP
|