123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464 |
- #include <msgpack.hpp>
- #include <gtest/gtest.h>
- TEST(reference, unpack_int)
- {
- msgpack::sbuffer sbuf;
- msgpack::pack(sbuf, 1);
- bool referenced = false;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced);
- EXPECT_FALSE(referenced);
- }
- TEST(reference, unpack_string)
- {
- msgpack::sbuffer sbuf;
- msgpack::pack(sbuf, std::string("abcdefg"));
- bool referenced = false;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced);
- EXPECT_FALSE(referenced);
- }
- TEST(reference, unpack_bin)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char c[] = { 1, 2, 3, 4, 5, 6 };
- packer.pack_bin(sizeof(c));
- packer.pack_bin_body(c, sizeof(c));
- bool referenced = false;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced);
- EXPECT_FALSE(referenced);
- }
- TEST(reference, unpack_ext)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char const buf [] = { 2 };
- packer.pack_ext(sizeof(buf), 1);
- packer.pack_ext_body(buf, sizeof(buf));
- bool referenced = false;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced);
- EXPECT_FALSE(referenced);
- }
- bool never_called(msgpack::type::object_type, std::size_t, void*)
- {
- EXPECT_TRUE(false);
- return false;
- }
- bool always_reference(msgpack::type::object_type, std::size_t, void*)
- {
- return true;
- }
- TEST(reference, unpack_int_ref)
- {
- msgpack::sbuffer sbuf;
- msgpack::pack(sbuf, 1);
- bool referenced = false;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced, never_called);
- EXPECT_FALSE(referenced);
- }
- TEST(reference, unpack_string_ref)
- {
- msgpack::sbuffer sbuf;
- msgpack::pack(sbuf, std::string("abcdefg"));
- bool referenced = false;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced, always_reference);
- EXPECT_TRUE(referenced);
- }
- TEST(reference, unpack_bin_ref)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char c[] = { 1, 2, 3, 4, 5, 6 };
- packer.pack_bin(sizeof(c));
- packer.pack_bin_body(c, sizeof(c));
- bool referenced = false;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced, always_reference);
- EXPECT_TRUE(referenced);
- }
- TEST(reference, unpack_ext_ref)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char const buf [] = { 2 };
- packer.pack_ext(sizeof(buf), 1);
- packer.pack_ext_body(buf, sizeof(buf));
- bool referenced = false;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced, always_reference);
- EXPECT_TRUE(referenced);
- }
- static void* s_p;
- bool sized_reference(msgpack::type::object_type t, std::size_t s, void* p)
- {
- s_p = p;
- switch (t) {
- case msgpack::type::STR:
- if (s >= 5) return true;
- break;
- case msgpack::type::BIN:
- if (s >= 6) return true;
- break;
- case msgpack::type::EXT:
- if (s >= 7) return true;
- break;
- default:
- EXPECT_TRUE(false);
- }
- return false;
- }
- TEST(reference, unpack_int_sized_ref)
- {
- msgpack::sbuffer sbuf;
- msgpack::pack(sbuf, 1);
- bool referenced = false;
- s_p = MSGPACK_NULLPTR;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced, never_called, &sbuf);
- EXPECT_FALSE(referenced);
- EXPECT_EQ(MSGPACK_NULLPTR, s_p);
- }
- TEST(reference, unpack_string_sized_ref_4)
- {
- msgpack::sbuffer sbuf;
- msgpack::pack(sbuf, std::string("1234"));
- bool referenced = false;
- s_p = MSGPACK_NULLPTR;
- // the last argument sbuf is any pointer as a user data.
- // That is stored to s_p in sized_reference
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced, sized_reference, &sbuf);
- EXPECT_FALSE(referenced);
- // compare the passed argument with stored s_p.
- EXPECT_EQ(&sbuf, s_p);
- }
- TEST(reference, unpack_string_sized_ref_5)
- {
- msgpack::sbuffer sbuf;
- msgpack::pack(sbuf, std::string("12345"));
- bool referenced = false;
- s_p = MSGPACK_NULLPTR;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced, sized_reference, &sbuf);
- EXPECT_TRUE(referenced);
- EXPECT_EQ(&sbuf, s_p);
- }
- TEST(reference, unpack_bin_sized_ref_5)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char c[] = { 1, 2, 3, 4, 5 };
- packer.pack_bin(sizeof(c));
- packer.pack_bin_body(c, sizeof(c));
- bool referenced = false;
- s_p = MSGPACK_NULLPTR;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced, sized_reference, &sbuf);
- EXPECT_FALSE(referenced);
- EXPECT_EQ(&sbuf, s_p);
- }
- TEST(reference, unpack_bin_sized_ref_6)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char c[] = { 1, 2, 3, 4, 5, 6 };
- packer.pack_bin(sizeof(c));
- packer.pack_bin_body(c, sizeof(c));
- bool referenced = false;
- s_p = MSGPACK_NULLPTR;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced, sized_reference, &sbuf);
- EXPECT_TRUE(referenced);
- EXPECT_EQ(&sbuf, s_p);
- }
- TEST(reference, unpack_ext_sized_ref_6)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char const buf [] = { 1, 2, 3, 4, 5 };
- packer.pack_ext(sizeof(buf), 1); // 5 + 1(type) = 6
- packer.pack_ext_body(buf, sizeof(buf));
- bool referenced = false;
- s_p = MSGPACK_NULLPTR;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced, sized_reference, &sbuf);
- EXPECT_FALSE(referenced);
- EXPECT_EQ(&sbuf, s_p);
- }
- TEST(reference, unpack_ext_sized_ref_7)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char const buf [] = { 1, 2, 3, 4, 5, 6 };
- packer.pack_ext(sizeof(buf), 1); // 6 + 1(type) = 7
- packer.pack_ext_body(buf, sizeof(buf));
- bool referenced = false;
- s_p = MSGPACK_NULLPTR;
- msgpack::object_handle oh =
- msgpack::unpack(sbuf.data(), sbuf.size(), referenced, sized_reference, &sbuf);
- EXPECT_TRUE(referenced);
- EXPECT_EQ(&sbuf, s_p);
- }
- // default reference function
- // STR, BIN, and EXT are always referenced, otherwise copied (converted).
- TEST(reference, unpacker_int)
- {
- msgpack::sbuffer sbuf;
- msgpack::pack(sbuf, 1);
- msgpack::unpacker unp;
- msgpack::object_handle oh;
- bool referenced = false;
- unp.reserve_buffer(sbuf.size());
- std::memcpy(unp.buffer(), sbuf.data(), sbuf.size());
- unp.buffer_consumed(sbuf.size());
- bool b = unp.next(oh, referenced);
- EXPECT_TRUE(b);
- EXPECT_FALSE(referenced);
- }
- TEST(reference, unpacker_string)
- {
- msgpack::sbuffer sbuf;
- msgpack::pack(sbuf, std::string("a"));
- msgpack::unpacker unp;
- msgpack::object_handle oh;
- bool referenced = false;
- unp.reserve_buffer(sbuf.size());
- std::memcpy(unp.buffer(), sbuf.data(), sbuf.size());
- unp.buffer_consumed(sbuf.size());
- bool b = unp.next(oh, referenced);
- EXPECT_TRUE(b);
- EXPECT_TRUE(referenced);
- }
- TEST(reference, unpacker_bin)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char c[] = { 1 };
- packer.pack_bin(sizeof(c));
- packer.pack_bin_body(c, sizeof(c));
- msgpack::unpacker unp;
- msgpack::object_handle oh;
- bool referenced = false;
- unp.reserve_buffer(sbuf.size());
- std::memcpy(unp.buffer(), sbuf.data(), sbuf.size());
- unp.buffer_consumed(sbuf.size());
- bool b = unp.next(oh, referenced);
- EXPECT_TRUE(b);
- EXPECT_TRUE(referenced);
- }
- TEST(reference, unpacker_ext)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char const buf [] = { 2 };
- packer.pack_ext(sizeof(buf), 1);
- packer.pack_ext_body(buf, sizeof(buf));
- msgpack::unpacker unp;
- msgpack::object_handle oh;
- bool referenced = false;
- unp.reserve_buffer(sbuf.size());
- std::memcpy(unp.buffer(), sbuf.data(), sbuf.size());
- unp.buffer_consumed(sbuf.size());
- bool b = unp.next(oh, referenced);
- EXPECT_TRUE(b);
- EXPECT_TRUE(referenced);
- }
- // pass user custom reference function
- TEST(reference, unpacker_int_sized_ref)
- {
- msgpack::sbuffer sbuf;
- msgpack::pack(sbuf, 1);
- msgpack::unpacker unp(never_called, &sbuf);
- msgpack::object_handle oh;
- bool referenced = false;
- unp.reserve_buffer(sbuf.size());
- std::memcpy(unp.buffer(), sbuf.data(), sbuf.size());
- unp.buffer_consumed(sbuf.size());
- s_p = MSGPACK_NULLPTR;
- bool b = unp.next(oh, referenced);
- EXPECT_TRUE(b);
- EXPECT_FALSE(referenced);
- EXPECT_EQ(MSGPACK_NULLPTR, s_p);
- }
- TEST(reference, unpacker_string_sized_ref_4)
- {
- msgpack::sbuffer sbuf;
- msgpack::pack(sbuf, std::string("1234"));
- msgpack::unpacker unp(sized_reference, &sbuf);
- msgpack::object_handle oh;
- bool referenced = false;
- unp.reserve_buffer(sbuf.size());
- std::memcpy(unp.buffer(), sbuf.data(), sbuf.size());
- unp.buffer_consumed(sbuf.size());
- s_p = MSGPACK_NULLPTR;
- bool b = unp.next(oh, referenced);
- EXPECT_TRUE(b);
- EXPECT_FALSE(referenced);
- EXPECT_EQ(&sbuf, s_p);
- }
- TEST(reference, unpacker_string_sized_ref_5)
- {
- msgpack::sbuffer sbuf;
- msgpack::pack(sbuf, std::string("12345"));
- msgpack::unpacker unp(sized_reference, &sbuf);
- msgpack::object_handle oh;
- bool referenced = false;
- unp.reserve_buffer(sbuf.size());
- std::memcpy(unp.buffer(), sbuf.data(), sbuf.size());
- unp.buffer_consumed(sbuf.size());
- s_p = MSGPACK_NULLPTR;
- bool b = unp.next(oh, referenced);
- EXPECT_TRUE(b);
- EXPECT_TRUE(referenced);
- EXPECT_EQ(&sbuf, s_p);
- }
- TEST(reference, unpacker_bin_sized_ref_5)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char c[] = { 1, 2, 3, 4, 5 };
- packer.pack_bin(sizeof(c));
- packer.pack_bin_body(c, sizeof(c));
- msgpack::unpacker unp(sized_reference, &sbuf);
- msgpack::object_handle oh;
- bool referenced = false;
- unp.reserve_buffer(sbuf.size());
- std::memcpy(unp.buffer(), sbuf.data(), sbuf.size());
- unp.buffer_consumed(sbuf.size());
- s_p = MSGPACK_NULLPTR;
- bool b = unp.next(oh, referenced);
- EXPECT_TRUE(b);
- EXPECT_FALSE(referenced);
- EXPECT_EQ(&sbuf, s_p);
- }
- TEST(reference, unpacker_bin_sized_ref_6)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char c[] = { 1, 2, 3, 4, 5, 6 };
- packer.pack_bin(sizeof(c));
- packer.pack_bin_body(c, sizeof(c));
- msgpack::unpacker unp(sized_reference, &sbuf);
- msgpack::object_handle oh;
- bool referenced = false;
- unp.reserve_buffer(sbuf.size());
- std::memcpy(unp.buffer(), sbuf.data(), sbuf.size());
- unp.buffer_consumed(sbuf.size());
- s_p = MSGPACK_NULLPTR;
- bool b = unp.next(oh, referenced);
- EXPECT_TRUE(b);
- EXPECT_TRUE(referenced);
- EXPECT_EQ(&sbuf, s_p);
- }
- TEST(reference, unpacker_ext_sized_ref_6)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char const buf [] = { 1, 2, 3, 4, 5 };
- packer.pack_ext(sizeof(buf), 1); // 5 + 1(type) = 6
- packer.pack_ext_body(buf, sizeof(buf));
- msgpack::unpacker unp(sized_reference, &sbuf);
- msgpack::object_handle oh;
- bool referenced = false;
- unp.reserve_buffer(sbuf.size());
- std::memcpy(unp.buffer(), sbuf.data(), sbuf.size());
- unp.buffer_consumed(sbuf.size());
- s_p = MSGPACK_NULLPTR;
- bool b = unp.next(oh, referenced);
- EXPECT_TRUE(b);
- EXPECT_FALSE(referenced);
- EXPECT_EQ(&sbuf, s_p);
- }
- TEST(reference, unpacker_ext_sized_ref_7)
- {
- msgpack::sbuffer sbuf;
- msgpack::packer<msgpack::sbuffer> packer(sbuf);
- char const buf [] = { 1, 2, 3, 4, 5, 6 };
- packer.pack_ext(sizeof(buf), 1); // 6 + 1(type) = 7
- packer.pack_ext_body(buf, sizeof(buf));
- msgpack::unpacker unp(sized_reference, &sbuf);
- msgpack::object_handle oh;
- bool referenced = false;
- unp.reserve_buffer(sbuf.size());
- std::memcpy(unp.buffer(), sbuf.data(), sbuf.size());
- unp.buffer_consumed(sbuf.size());
- s_p = MSGPACK_NULLPTR;
- bool b = unp.next(oh, referenced);
- EXPECT_TRUE(b);
- EXPECT_TRUE(referenced);
- EXPECT_EQ(&sbuf, s_p);
- }
|