pack_unpack.cpp 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599
  1. #include <msgpack.hpp>
  2. #include <gtest/gtest.h>
  3. #include <sstream>
  4. TEST(pack, num)
  5. {
  6. msgpack::sbuffer sbuf;
  7. msgpack::pack(sbuf, 1);
  8. }
  9. TEST(pack, vector)
  10. {
  11. msgpack::sbuffer sbuf;
  12. std::vector<int> vec;
  13. vec.push_back(1);
  14. vec.push_back(2);
  15. vec.push_back(3);
  16. msgpack::pack(sbuf, vec);
  17. }
  18. TEST(pack, to_ostream)
  19. {
  20. std::ostringstream stream;
  21. msgpack::pack(stream, 1);
  22. }
  23. struct myclass {
  24. myclass() : num(0), str("default") { }
  25. myclass(int num, const std::string& str) :
  26. num(num), str(str) { }
  27. ~myclass() { }
  28. int num;
  29. std::string str;
  30. MSGPACK_DEFINE(num, str);
  31. };
  32. TEST(pack, myclass)
  33. {
  34. msgpack::sbuffer sbuf;
  35. myclass m(1, "msgpack");
  36. msgpack::pack(sbuf, m);
  37. }
  38. TEST(unpack, int_ret_no_offset_no_ref)
  39. {
  40. msgpack::sbuffer sbuf;
  41. msgpack::pack(sbuf, 1);
  42. msgpack::object_handle oh = msgpack::unpack(sbuf.data(), sbuf.size());
  43. EXPECT_EQ(1, oh.get().as<int>());
  44. }
  45. TEST(unpack, int_ret_offset_no_ref)
  46. {
  47. msgpack::sbuffer sbuf;
  48. msgpack::pack(sbuf, 1);
  49. std::size_t off = 0;
  50. msgpack::object_handle oh = msgpack::unpack(sbuf.data(), sbuf.size(), off);
  51. EXPECT_EQ(1, oh.get().as<int>());
  52. EXPECT_EQ(off, sbuf.size());
  53. }
  54. TEST(unpack, int_ret_no_offset_ref)
  55. {
  56. msgpack::sbuffer sbuf;
  57. msgpack::pack(sbuf, 1);
  58. bool referenced;
  59. msgpack::object_handle oh = msgpack::unpack(sbuf.data(), sbuf.size(), referenced);
  60. EXPECT_EQ(1, oh.get().as<int>());
  61. EXPECT_FALSE(referenced);
  62. }
  63. TEST(unpack, int_ret_offset_ref)
  64. {
  65. msgpack::sbuffer sbuf;
  66. msgpack::pack(sbuf, 1);
  67. std::size_t off = 0;
  68. bool referenced;
  69. msgpack::object_handle oh = msgpack::unpack(sbuf.data(), sbuf.size(), off, referenced);
  70. EXPECT_EQ(1, oh.get().as<int>());
  71. EXPECT_FALSE(referenced);
  72. EXPECT_EQ(off, sbuf.size());
  73. }
  74. TEST(unpack, int_no_offset_no_ref)
  75. {
  76. msgpack::sbuffer sbuf;
  77. msgpack::pack(sbuf, 1);
  78. msgpack::object_handle oh;
  79. msgpack::unpack(oh, sbuf.data(), sbuf.size());
  80. EXPECT_EQ(1, oh.get().as<int>());
  81. }
  82. TEST(unpack, int_offset_no_ref)
  83. {
  84. msgpack::sbuffer sbuf;
  85. msgpack::pack(sbuf, 1);
  86. msgpack::object_handle oh;
  87. std::size_t off = 0;
  88. msgpack::unpack(oh, sbuf.data(), sbuf.size(), off);
  89. EXPECT_EQ(1, oh.get().as<int>());
  90. EXPECT_EQ(off, sbuf.size());
  91. }
  92. TEST(unpack, int_no_offset_ref)
  93. {
  94. msgpack::sbuffer sbuf;
  95. msgpack::pack(sbuf, 1);
  96. msgpack::object_handle oh;
  97. bool referenced;
  98. msgpack::unpack(oh, sbuf.data(), sbuf.size(), referenced);
  99. EXPECT_EQ(1, oh.get().as<int>());
  100. EXPECT_FALSE(referenced);
  101. }
  102. TEST(unpack, int_offset_ref)
  103. {
  104. msgpack::sbuffer sbuf;
  105. msgpack::pack(sbuf, 1);
  106. msgpack::object_handle oh;
  107. std::size_t off = 0;
  108. bool referenced;
  109. msgpack::unpack(oh, sbuf.data(), sbuf.size(), off, referenced);
  110. EXPECT_EQ(1, oh.get().as<int>());
  111. EXPECT_FALSE(referenced);
  112. EXPECT_EQ(off, sbuf.size());
  113. }
  114. #if MSGPACK_DEFAULT_API_VERSION == 1
  115. TEST(unpack, int_pointer_off_no_ref)
  116. {
  117. msgpack::sbuffer sbuf;
  118. msgpack::pack(sbuf, 1);
  119. msgpack::object_handle oh;
  120. std::size_t off = 0;
  121. // obsolete
  122. #if (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  123. #pragma GCC diagnostic push
  124. #pragma GCC diagnostic ignored "-Wdeprecated-declarations"
  125. #endif // (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  126. msgpack::unpack(&oh, sbuf.data(), sbuf.size(), &off);
  127. #if (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  128. #pragma GCC diagnostic pop
  129. #endif // (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  130. EXPECT_EQ(1, oh.get().as<int>());
  131. EXPECT_EQ(off, sbuf.size());
  132. }
  133. TEST(unpack, int_pointer_off_no_ref_explicit)
  134. {
  135. msgpack::sbuffer sbuf;
  136. msgpack::pack(sbuf, 1);
  137. msgpack::object_handle oh;
  138. std::size_t off = 0;
  139. // obsolete
  140. #if (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  141. #pragma GCC diagnostic push
  142. #pragma GCC diagnostic ignored "-Wdeprecated-declarations"
  143. #endif // (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  144. msgpack::unpack(&oh, sbuf.data(), sbuf.size(), &off, MSGPACK_NULLPTR);
  145. #if (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  146. #pragma GCC diagnostic pop
  147. #endif // (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  148. EXPECT_EQ(1, oh.get().as<int>());
  149. EXPECT_EQ(off, sbuf.size());
  150. }
  151. TEST(unpack, int_pointer_no_off_ref)
  152. {
  153. msgpack::sbuffer sbuf;
  154. msgpack::pack(sbuf, 1);
  155. msgpack::object_handle oh;
  156. bool referenced;
  157. // obsolete
  158. #if (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  159. #pragma GCC diagnostic push
  160. #pragma GCC diagnostic ignored "-Wdeprecated-declarations"
  161. #endif // (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  162. msgpack::unpack(&oh, sbuf.data(), sbuf.size(), MSGPACK_NULLPTR, &referenced);
  163. #if (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  164. #pragma GCC diagnostic pop
  165. #endif // (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  166. EXPECT_EQ(1, oh.get().as<int>());
  167. EXPECT_FALSE(referenced);
  168. }
  169. TEST(unpack, int_pointer_off_ref)
  170. {
  171. msgpack::sbuffer sbuf;
  172. msgpack::pack(sbuf, 1);
  173. msgpack::object_handle oh;
  174. bool referenced;
  175. std::size_t off = 0;
  176. // obsolete
  177. #if (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  178. #pragma GCC diagnostic push
  179. #pragma GCC diagnostic ignored "-Wdeprecated-declarations"
  180. #endif // (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  181. msgpack::unpack(&oh, sbuf.data(), sbuf.size(), &off, &referenced);
  182. #if (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  183. #pragma GCC diagnostic pop
  184. #endif // (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  185. EXPECT_EQ(1, oh.get().as<int>());
  186. EXPECT_EQ(off, sbuf.size());
  187. EXPECT_FALSE(referenced);
  188. }
  189. TEST(unpack, int_default_null_pointer)
  190. {
  191. msgpack::sbuffer sbuf;
  192. msgpack::pack(sbuf, 1);
  193. msgpack::object_handle oh;
  194. // obsolete
  195. #if (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  196. #pragma GCC diagnostic push
  197. #pragma GCC diagnostic ignored "-Wdeprecated-declarations"
  198. #endif // (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  199. msgpack::unpack(&oh, sbuf.data(), sbuf.size());
  200. #if (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  201. #pragma GCC diagnostic pop
  202. #endif // (__GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 2))
  203. EXPECT_EQ(1, oh.get().as<int>());
  204. }
  205. #endif // MSGPACK_DEFAULT_API_VERSION == 1
  206. TEST(unpack, int_zone_no_offset_no_ref)
  207. {
  208. msgpack::sbuffer sbuf;
  209. msgpack::pack(sbuf, 1);
  210. msgpack::zone z;
  211. msgpack::object obj = msgpack::unpack(z, sbuf.data(), sbuf.size());
  212. EXPECT_EQ(1, obj.as<int>());
  213. }
  214. TEST(unpack, int_zone_offset_no_ref)
  215. {
  216. msgpack::sbuffer sbuf;
  217. msgpack::pack(sbuf, 1);
  218. std::size_t off = 0;
  219. msgpack::zone z;
  220. msgpack::object obj = msgpack::unpack(z, sbuf.data(), sbuf.size(), off);
  221. EXPECT_EQ(1, obj.as<int>());
  222. EXPECT_EQ(off, sbuf.size());
  223. }
  224. TEST(unpack, int_zone_no_offset_ref)
  225. {
  226. msgpack::sbuffer sbuf;
  227. msgpack::pack(sbuf, 1);
  228. bool referenced;
  229. msgpack::zone z;
  230. msgpack::object obj = msgpack::unpack(z, sbuf.data(), sbuf.size(), referenced);
  231. EXPECT_EQ(1, obj.as<int>());
  232. EXPECT_FALSE(referenced);
  233. }
  234. TEST(unpack, int_zone_offset_ref)
  235. {
  236. msgpack::sbuffer sbuf;
  237. msgpack::pack(sbuf, 1);
  238. std::size_t off = 0;
  239. bool referenced;
  240. msgpack::zone z;
  241. msgpack::object obj = msgpack::unpack(z, sbuf.data(), sbuf.size(), off, referenced);
  242. EXPECT_EQ(1, obj.as<int>());
  243. EXPECT_FALSE(referenced);
  244. EXPECT_EQ(off, sbuf.size());
  245. }
  246. TEST(unpack, sequence)
  247. {
  248. msgpack::sbuffer sbuf;
  249. msgpack::pack(sbuf, 1);
  250. msgpack::pack(sbuf, 2);
  251. msgpack::pack(sbuf, 3);
  252. std::size_t off = 0;
  253. msgpack::object_handle oh;
  254. msgpack::unpack(oh, sbuf.data(), sbuf.size(), off);
  255. EXPECT_EQ(1, oh.get().as<int>());
  256. msgpack::unpack(oh, sbuf.data(), sbuf.size(), off);
  257. EXPECT_EQ(2, oh.get().as<int>());
  258. msgpack::unpack(oh, sbuf.data(), sbuf.size(), off);
  259. EXPECT_EQ(3, oh.get().as<int>());
  260. EXPECT_EQ(off, sbuf.size());
  261. }
  262. TEST(unpack, convert_to_object_handle)
  263. {
  264. msgpack::sbuffer sbuf;
  265. msgpack::pack(sbuf, 1);
  266. msgpack::unpacked msg;
  267. msgpack::unpack(msg, sbuf.data(), sbuf.size());
  268. msgpack::object_handle oh(msgpack::move(msg));
  269. EXPECT_EQ(1, oh.get().as<int>());
  270. }
  271. TEST(unpack, convert_to_object_handle_direct)
  272. {
  273. msgpack::sbuffer sbuf;
  274. msgpack::pack(sbuf, 1);
  275. msgpack::object_handle oh(msgpack::unpack(sbuf.data(), sbuf.size()));
  276. EXPECT_EQ(1, oh.get().as<int>());
  277. }
  278. TEST(unpack, convert_to_object_handle_direct_implicit)
  279. {
  280. msgpack::sbuffer sbuf;
  281. msgpack::pack(sbuf, 1);
  282. msgpack::object_handle oh = msgpack::unpack(sbuf.data(), sbuf.size());
  283. EXPECT_EQ(1, oh.get().as<int>());
  284. }
  285. TEST(unpack, insufficient_bytes_ref)
  286. {
  287. msgpack::sbuffer sbuf;
  288. msgpack::pack(sbuf, 255); // uint8 (2bytes)
  289. std::size_t off = 0;
  290. msgpack::object_handle oh;
  291. try {
  292. msgpack::unpack(oh, sbuf.data(), 1, off);
  293. EXPECT_TRUE(false);
  294. }
  295. catch (msgpack::insufficient_bytes const&) {
  296. EXPECT_TRUE(true);
  297. #if MSGPACK_DEFAULT_API_VERSION < 3
  298. EXPECT_EQ(off, 0u);
  299. #else // MSGPACK_DEFAULT_API_VERSION < 3
  300. EXPECT_EQ(1u, off);
  301. #endif // MSGPACK_DEFAULT_API_VERSION < 3
  302. }
  303. }
  304. TEST(unpack, insufficient_bytes_object_handle)
  305. {
  306. msgpack::sbuffer sbuf;
  307. msgpack::pack(sbuf, 255); // uint8 (2bytes)
  308. std::size_t off = 0;
  309. try {
  310. msgpack::object_handle oh(msgpack::unpack(sbuf.data(), 1, off));
  311. EXPECT_TRUE(false);
  312. }
  313. catch (msgpack::insufficient_bytes const&) {
  314. EXPECT_TRUE(true);
  315. #if MSGPACK_DEFAULT_API_VERSION < 3
  316. EXPECT_EQ(off, 0u);
  317. #else // MSGPACK_DEFAULT_API_VERSION < 3
  318. EXPECT_EQ(1u, off);
  319. #endif // MSGPACK_DEFAULT_API_VERSION < 3
  320. }
  321. }
  322. TEST(unpack, insufficient_bytes_zone)
  323. {
  324. msgpack::sbuffer sbuf;
  325. msgpack::pack(sbuf, 255); // uint8 (2bytes)
  326. std::size_t off = 0;
  327. try {
  328. msgpack::zone z;
  329. msgpack::unpack(z, sbuf.data(), 1, off);
  330. EXPECT_TRUE(false);
  331. }
  332. catch (msgpack::insufficient_bytes const&) {
  333. EXPECT_TRUE(true);
  334. #if MSGPACK_DEFAULT_API_VERSION < 3
  335. EXPECT_EQ(off, 0u);
  336. #else // MSGPACK_DEFAULT_API_VERSION < 3
  337. EXPECT_EQ(1u, off);
  338. #endif // MSGPACK_DEFAULT_API_VERSION < 3
  339. }
  340. }
  341. TEST(unpack, parse_error)
  342. {
  343. msgpack::sbuffer sbuf;
  344. char c = '\xc1';
  345. sbuf.write(&c, 1);
  346. bool thrown = false;
  347. msgpack::object_handle oh;
  348. try {
  349. msgpack::unpack(oh, sbuf.data(), sbuf.size());
  350. EXPECT_TRUE(false);
  351. }
  352. catch (msgpack::parse_error const&) {
  353. thrown = true;
  354. }
  355. EXPECT_TRUE(thrown);
  356. }
  357. TEST(unpack, returned_parse_error)
  358. {
  359. msgpack::sbuffer sbuf;
  360. char c = '\xc1';
  361. sbuf.write(&c, 1);
  362. bool thrown = false;
  363. try {
  364. msgpack::unpack(sbuf.data(), sbuf.size());
  365. EXPECT_TRUE(false);
  366. }
  367. catch (msgpack::parse_error const&) {
  368. thrown = true;
  369. }
  370. EXPECT_TRUE(thrown);
  371. }
  372. TEST(unpack, zone_parse_error)
  373. {
  374. msgpack::sbuffer sbuf;
  375. char c = '\xc1';
  376. sbuf.write(&c, 1);
  377. bool thrown = false;
  378. msgpack::zone z;
  379. try {
  380. msgpack::unpack(z, sbuf.data(), sbuf.size());
  381. EXPECT_TRUE(false);
  382. }
  383. catch (msgpack::parse_error const&) {
  384. thrown = true;
  385. }
  386. EXPECT_TRUE(thrown);
  387. }
  388. TEST(unpack, extra_bytes)
  389. {
  390. msgpack::sbuffer sbuf;
  391. msgpack::pack(sbuf, 1);
  392. msgpack::object_handle oh = msgpack::unpack(sbuf.data(), sbuf.size() + 1);
  393. EXPECT_EQ(1, oh.get().as<int>());
  394. }
  395. TEST(unpack, zone_extra_bytes)
  396. {
  397. msgpack::sbuffer sbuf;
  398. msgpack::pack(sbuf, 1);
  399. msgpack::zone z;
  400. msgpack::object obj = msgpack::unpack(z, sbuf.data(), sbuf.size() + 1);
  401. EXPECT_EQ(1, obj.as<int>());
  402. }
  403. TEST(unpack, int_off_larger_than_length)
  404. {
  405. msgpack::sbuffer sbuf;
  406. msgpack::pack(sbuf, 1);
  407. std::size_t off = 2;
  408. bool thrown = false;
  409. try {
  410. msgpack::object_handle oh = msgpack::unpack(sbuf.data(), sbuf.size(), off);
  411. }
  412. catch (msgpack::insufficient_bytes const&) {
  413. thrown = true;
  414. }
  415. EXPECT_TRUE(thrown);
  416. EXPECT_EQ(off, 2u);
  417. }
  418. TEST(unpack, empty_array_fix)
  419. {
  420. std::string buf;
  421. buf.push_back(static_cast<unsigned char>(0x90));
  422. std::size_t off = 0;
  423. msgpack::object_handle oh = msgpack::unpack(buf.data(), buf.size(), off);
  424. EXPECT_EQ(oh.get().type, msgpack::type::ARRAY);
  425. EXPECT_EQ(oh.get().via.array.size, 0u);
  426. EXPECT_EQ(off, 1u);
  427. }
  428. TEST(unpack, empty_array_16)
  429. {
  430. std::string buf;
  431. buf.push_back(static_cast<unsigned char>(0xdc));
  432. buf.push_back(static_cast<unsigned char>(0x00));
  433. buf.push_back(static_cast<unsigned char>(0x00));
  434. std::size_t off = 0;
  435. msgpack::object_handle oh = msgpack::unpack(buf.data(), buf.size(), off);
  436. EXPECT_EQ(oh.get().type, msgpack::type::ARRAY);
  437. EXPECT_EQ(oh.get().via.array.size, 0u);
  438. EXPECT_EQ(off, 3u);
  439. }
  440. TEST(unpack, empty_array_32)
  441. {
  442. std::string buf;
  443. buf.push_back(static_cast<unsigned char>(0xdd));
  444. buf.push_back(static_cast<unsigned char>(0x00));
  445. buf.push_back(static_cast<unsigned char>(0x00));
  446. buf.push_back(static_cast<unsigned char>(0x00));
  447. buf.push_back(static_cast<unsigned char>(0x00));
  448. std::size_t off = 0;
  449. msgpack::object_handle oh = msgpack::unpack(buf.data(), buf.size(), off);
  450. EXPECT_EQ(oh.get().type, msgpack::type::ARRAY);
  451. EXPECT_EQ(oh.get().via.array.size, 0u);
  452. EXPECT_EQ(off, 5u);
  453. }
  454. TEST(unpack, empty_map_fix)
  455. {
  456. std::string buf;
  457. buf.push_back(static_cast<unsigned char>(0x80));
  458. std::size_t off = 0;
  459. msgpack::object_handle oh = msgpack::unpack(buf.data(), buf.size(), off);
  460. EXPECT_EQ(oh.get().type, msgpack::type::MAP);
  461. EXPECT_EQ(oh.get().via.map.size, 0u);
  462. EXPECT_EQ(off, 1u);
  463. }
  464. TEST(unpack, empty_map_16)
  465. {
  466. std::string buf;
  467. buf.push_back(static_cast<unsigned char>(0xde));
  468. buf.push_back(static_cast<unsigned char>(0x00));
  469. buf.push_back(static_cast<unsigned char>(0x00));
  470. std::size_t off = 0;
  471. msgpack::object_handle oh = msgpack::unpack(buf.data(), buf.size(), off);
  472. EXPECT_EQ(oh.get().type, msgpack::type::MAP);
  473. EXPECT_EQ(oh.get().via.map.size, 0u);
  474. EXPECT_EQ(off, 3u);
  475. }
  476. TEST(unpack, empty_map_32)
  477. {
  478. std::string buf;
  479. buf.push_back(static_cast<unsigned char>(0xdf));
  480. buf.push_back(static_cast<unsigned char>(0x00));
  481. buf.push_back(static_cast<unsigned char>(0x00));
  482. buf.push_back(static_cast<unsigned char>(0x00));
  483. buf.push_back(static_cast<unsigned char>(0x00));
  484. std::size_t off = 0;
  485. msgpack::object_handle oh = msgpack::unpack(buf.data(), buf.size(), off);
  486. EXPECT_EQ(oh.get().type, msgpack::type::MAP);
  487. EXPECT_EQ(oh.get().via.map.size, 0u);
  488. EXPECT_EQ(off, 5u);
  489. }