msgpack/cpp/unpack.hpp

#ifndef MSGPACK_UNPACK_HPP__
#define MSGPACK_UNPACK_HPP__

#include "msgpack/object.hpp"
#include "msgpack/zone.hpp"
#include <stdexcept>

#ifndef MSGPACK_UNPACKER_INITIAL_BUFFER_SIZE
#define MSGPACK_UNPACKER_INITIAL_BUFFER_SIZE 8*1024
#endif

namespace msgpack {


static const size_t UNPACKER_INITIAL_BUFFER_SIZE = MSGPACK_UNPACKER_INITIAL_BUFFER_SIZE;


struct unpack_error : public std::runtime_error {
	unpack_error(const std::string& msg) :
		std::runtime_error(msg) { }
};


class unpacker {
public:
	unpacker();
	~unpacker();

public:
	/*! 1. reserve buffer. at least `len' bytes of capacity will be ready */
	void reserve_buffer(size_t len);

	/*! 2. read data to the buffer() up to buffer_capacity() bytes */
	void* buffer();
	size_t buffer_capacity() const;

	/*! 3. specify the number of bytes actually copied */
	void buffer_consumed(size_t len);

	/*! 4. repeat execute() until it retunrs false */
	bool execute();

	/*! 5.1. if execute() returns true, take out the parsed object */
	object data();

	/*! 5.2. the parsed object is valid until the zone is deleted */
	// Note that once release_zone() from unpacker, you must delete it
	// otherwise the memrory will leak.
	zone* release_zone();

	/*! 5.3. after release_zone(), re-initialize unpacker */
	void reset();

public:
	// These functions are usable when non-MessagePack message follows after
	// MessagePack message.
	// Note that there are no parsed buffer when execute() returned true.

	/*! get address of buffer that is not parsed */
	void* nonparsed_buffer();
	size_t nonparsed_size() const;

	/*! get the number of bytes that is already parsed */
	size_t parsed_size() const;

	/*! remove unparsed buffer from unpacker */
	// Note that reset() leaves non-parsed buffer.
	void remove_nonparsed_buffer();

private:
	zone* m_zone;

	struct context;
	context* m_ctx;

	void* m_buffer;
	size_t m_used;
	size_t m_free;
	size_t m_off;
	void expand_buffer(size_t len);

private:
	unpacker(const unpacker&);

public:
	static object unpack(const void* data, size_t len, zone& z);
};


inline void unpacker::reserve_buffer(size_t len)
{
	if(m_free >= len) { return; }
	expand_buffer(len);
}

inline void* unpacker::buffer()
	{ return (void*)(((char*)m_buffer)+m_used); }

inline size_t unpacker::buffer_capacity() const
	{ return m_free; }

inline void unpacker::buffer_consumed(size_t len)
{
	m_used += len;
	m_free -= len;
}


inline void* unpacker::nonparsed_buffer()
	{ return (void*)(((char*)m_buffer)+m_off); }

inline size_t unpacker::nonparsed_size() const
	{ return m_used - m_off; }

inline size_t unpacker::parsed_size() const
	{ return m_off; }

inline void unpacker::remove_nonparsed_buffer()
	{ m_used = m_off; }


inline object unpack(const void* data, size_t len, zone& z)
{
	return unpacker::unpack(data, len, z);
}


}  // namespace msgpack

#endif /* msgpack/unpack.hpp */