generic-library/msgpack
1
0
Fork 0
mirror of https://github.com/msgpack/msgpack-c.git synced 2025-06-25 22:15:23 +02:00
Code Issues Projects Releases Wiki Activity

Merge branch 'ygj6-cpp_master' into cpp_master

Browse Source
This commit is contained in:
Takatoshi Kondo 2020-07-02 13:58:30 +09:00
commit 0e22de17b1
2 changed files with 42 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/msgpack/v1/sbuffer.hpp
View File

@ -14,6 +14,7 @@
#include <stdexcept>
#include <cstring>
#include <cassert>
namespace msgpack {
@ -68,12 +69,15 @@ public:
void write(const char* buf, size_t len)
{
assert(buf || len == 0);
if(m_alloc - m_size < len) {
expand_buffer(len);
}
if(buf) {
std::memcpy(m_data + m_size, buf, len);
m_size += len;
}
}
char* data()
{

30
include/msgpack/v2/create_object_visitor.hpp
View File

@ -10,6 +10,8 @@
#ifndef MSGPACK_V2_CREATE_OBJECT_VISITOR_HPP
#define MSGPACK_V2_CREATE_OBJECT_VISITOR_HPP
#include <cassert>
#include "msgpack/unpack_decl.hpp"
#include "msgpack/unpack_exception.hpp"
#include "msgpack/v2/create_object_visitor_decl.hpp"
@ -106,19 +108,27 @@ public:
return true;
}
bool visit_str(const char* v, uint32_t size) {
assert(v || size == 0);
if (size > m_limit.str()) throw msgpack::str_size_overflow("str size overflow");
msgpack::object* obj = m_stack.back();
obj->type = msgpack::type::STR;
if (m_func && m_func(obj->type, size, m_user_data)) {
obj->via.str.ptr = v;
obj->via.str.size = size;
set_referenced(true);
}
else {
if (v) {
char* tmp = static_cast<char*>(zone().allocate_align(size, MSGPACK_ZONE_ALIGNOF(char)));
std::memcpy(tmp, v, size);
obj->via.str.ptr = tmp;
}
obj->via.str.size = size;
}
else {
obj->via.str.ptr = MSGPACK_NULLPTR;
obj->via.str.size = 0;
}
}
return true;
}
bool visit_bin(const char* v, uint32_t size) {
@ -127,14 +137,21 @@ public:
obj->type = msgpack::type::BIN;
if (m_func && m_func(obj->type, size, m_user_data)) {
obj->via.bin.ptr = v;
obj->via.bin.size = size;
set_referenced(true);
}
else {
if (v) {
char* tmp = static_cast<char*>(zone().allocate_align(size, MSGPACK_ZONE_ALIGNOF(char)));
std::memcpy(tmp, v, size);
obj->via.bin.ptr = tmp;
}
obj->via.bin.size = size;
}
else {
obj->via.bin.ptr = MSGPACK_NULLPTR;
obj->via.bin.size = 0;
}
}
return true;
}
bool visit_ext(const char* v, uint32_t size) {
@ -143,14 +160,21 @@ public:
obj->type = msgpack::type::EXT;
if (m_func && m_func(obj->type, size, m_user_data)) {
obj->via.ext.ptr = v;
obj->via.ext.size = static_cast<uint32_t>(size - 1);
set_referenced(true);
}
else {
if (v) {
char* tmp = static_cast<char*>(zone().allocate_align(size, MSGPACK_ZONE_ALIGNOF(char)));
std::memcpy(tmp, v, size);
obj->via.ext.ptr = tmp;
}
obj->via.ext.size = static_cast<uint32_t>(size - 1);
}
else {
obj->via.ext.ptr = MSGPACK_NULLPTR;
obj->via.ext.size = 0;
}
}
return true;
}
bool start_array(uint32_t num_elements) {