BencodeTest.cc 8.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296
  1. #include "bencode.h"
  2. #include <cstring>
  3. #include <iostream>
  4. #include <cppunit/extensions/HelperMacros.h>
  5. #include "Util.h"
  6. #include "RecoverableException.h"
  7. namespace aria2 {
  8. class BencodeTest:public CppUnit::TestFixture {
  9. CPPUNIT_TEST_SUITE(BencodeTest);
  10. CPPUNIT_TEST(testString);
  11. CPPUNIT_TEST(testInteger);
  12. CPPUNIT_TEST(testDict);
  13. CPPUNIT_TEST(testDictIter);
  14. CPPUNIT_TEST(testList);
  15. CPPUNIT_TEST(testListIter);
  16. CPPUNIT_TEST(testDecode);
  17. CPPUNIT_TEST(testEncode);
  18. CPPUNIT_TEST_SUITE_END();
  19. private:
  20. public:
  21. void testString();
  22. void testInteger();
  23. void testDict();
  24. void testDictIter();
  25. void testList();
  26. void testListIter();
  27. void testDecode();
  28. void testEncode();
  29. };
  30. CPPUNIT_TEST_SUITE_REGISTRATION( BencodeTest );
  31. void BencodeTest::testString()
  32. {
  33. bencode::BDE s(std::string("aria2"));
  34. CPPUNIT_ASSERT_EQUAL(std::string("aria2"), s.s());
  35. unsigned char dataWithNull[] = { 0xf0, '\0', 0x0f };
  36. bencode::BDE sWithNull(dataWithNull, sizeof(dataWithNull));
  37. CPPUNIT_ASSERT(memcmp(dataWithNull, sWithNull.s().c_str(),
  38. sizeof(dataWithNull)) == 0);
  39. bencode::BDE zero("");
  40. CPPUNIT_ASSERT_EQUAL(std::string(""), zero.s());
  41. const unsigned char uc[] = { 0x08, 0x19, 0x2a, 0x3b };
  42. bencode::BDE data(uc, sizeof(uc));
  43. CPPUNIT_ASSERT_EQUAL(Util::toHex(uc, sizeof(uc)),
  44. Util::toHex(data.uc(), data.s().size()));
  45. }
  46. void BencodeTest::testInteger()
  47. {
  48. bencode::BDE integer(INT64_MAX);
  49. CPPUNIT_ASSERT_EQUAL(INT64_MAX, integer.i());
  50. }
  51. void BencodeTest::testDict()
  52. {
  53. bencode::BDE dict = bencode::BDE::dict();
  54. CPPUNIT_ASSERT(dict.empty());
  55. dict["ki"] = 7;
  56. dict["ks"] = std::string("abc");
  57. CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(2), dict.size());
  58. CPPUNIT_ASSERT(dict.containsKey("ki"));
  59. CPPUNIT_ASSERT_EQUAL(static_cast<bencode::BDE::Integer>(7), dict["ki"].i());
  60. CPPUNIT_ASSERT(dict.containsKey("ks"));
  61. CPPUNIT_ASSERT_EQUAL(std::string("abc"), dict["ks"].s());
  62. CPPUNIT_ASSERT(dict["kn"].isNone()); // This adds kn key with default value.
  63. CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(3), dict.size());
  64. CPPUNIT_ASSERT(dict.containsKey("kn"));
  65. const bencode::BDE& ref = dict;
  66. ref["kn2"]; // This doesn't add kn2 key.
  67. CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(3), ref.size());
  68. CPPUNIT_ASSERT(!ref.containsKey("kn2"));
  69. dict.removeKey("kn");
  70. CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(2), dict.size());
  71. CPPUNIT_ASSERT(!dict.containsKey("kn"));
  72. }
  73. void BencodeTest::testDictIter()
  74. {
  75. bencode::BDE dict = bencode::BDE::dict();
  76. dict["alpha2"] = std::string("alpha2");
  77. dict["charlie"] = std::string("charlie");
  78. dict["bravo"] = std::string("bravo");
  79. dict["alpha"] = std::string("alpha");
  80. bencode::BDE::Dict::iterator i = dict.dictBegin();
  81. CPPUNIT_ASSERT_EQUAL(std::string("alpha"), (*i++).first);
  82. CPPUNIT_ASSERT_EQUAL(std::string("alpha2"), (*i++).first);
  83. CPPUNIT_ASSERT_EQUAL(std::string("bravo"), (*i++).first);
  84. CPPUNIT_ASSERT_EQUAL(std::string("charlie"), (*i++).first);
  85. CPPUNIT_ASSERT(dict.dictEnd() == i);
  86. const bencode::BDE& ref = dict;
  87. bencode::BDE::Dict::const_iterator ci = ref.dictBegin();
  88. CPPUNIT_ASSERT_EQUAL(std::string("alpha"), (*ci++).first);
  89. std::advance(ci, 3);
  90. CPPUNIT_ASSERT(ref.dictEnd() == ci);
  91. }
  92. void BencodeTest::testList()
  93. {
  94. bencode::BDE list = bencode::BDE::list();
  95. CPPUNIT_ASSERT(list.empty());
  96. list << 7;
  97. list << std::string("aria2");
  98. CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(2), list.size());
  99. CPPUNIT_ASSERT_EQUAL(static_cast<bencode::BDE::Integer>(7), list[0].i());
  100. CPPUNIT_ASSERT_EQUAL(std::string("aria2"), list[1].s());
  101. const bencode::BDE& ref = list;
  102. CPPUNIT_ASSERT_EQUAL(static_cast<bencode::BDE::Integer>(7), ref[0].i());
  103. CPPUNIT_ASSERT_EQUAL(std::string("aria2"), ref[1].s());
  104. }
  105. void BencodeTest::testListIter()
  106. {
  107. bencode::BDE list = bencode::BDE::list();
  108. list << std::string("alpha2");
  109. list << std::string("charlie");
  110. list << std::string("bravo");
  111. list << std::string("alpha");
  112. bencode::BDE::List::iterator i = list.listBegin();
  113. CPPUNIT_ASSERT_EQUAL(std::string("alpha2"), (*i++).s());
  114. CPPUNIT_ASSERT_EQUAL(std::string("charlie"), (*i++).s());
  115. CPPUNIT_ASSERT_EQUAL(std::string("bravo"), (*i++).s());
  116. CPPUNIT_ASSERT_EQUAL(std::string("alpha"), (*i++).s());
  117. CPPUNIT_ASSERT(list.listEnd() == i);
  118. const bencode::BDE& ref = list;
  119. bencode::BDE::List::const_iterator ci = ref.listBegin();
  120. CPPUNIT_ASSERT_EQUAL(std::string("alpha2"), (*ci++).s());
  121. std::advance(ci, 3);
  122. CPPUNIT_ASSERT(ref.listEnd() == ci);
  123. }
  124. void BencodeTest::testDecode()
  125. {
  126. {
  127. // string, integer and list in dict
  128. bencode::BDE dict =
  129. bencode::decode("d4:name5:aria24:sizei12345678900e5:filesl3:bin3:docee");
  130. CPPUNIT_ASSERT(dict.isDict());
  131. CPPUNIT_ASSERT_EQUAL(std::string("aria2"), dict["name"].s());
  132. CPPUNIT_ASSERT_EQUAL(static_cast<bencode::BDE::Integer>(12345678900LL),
  133. dict["size"].i());
  134. bencode::BDE list = dict["files"];
  135. CPPUNIT_ASSERT(list.isList());
  136. CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(2), list.size());
  137. CPPUNIT_ASSERT_EQUAL(std::string("bin"), list[0].s());
  138. CPPUNIT_ASSERT_EQUAL(std::string("doc"), list[1].s());
  139. }
  140. {
  141. // dict in list
  142. bencode::BDE list = bencode::decode("ld1:ki123eee");
  143. CPPUNIT_ASSERT(list.isList());
  144. CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), list.size());
  145. bencode::BDE dict = list[0];
  146. CPPUNIT_ASSERT(dict.isDict());
  147. CPPUNIT_ASSERT_EQUAL(static_cast<bencode::BDE::Integer>(123),
  148. dict["k"].i());
  149. }
  150. {
  151. // empty key is allowed
  152. bencode::BDE s = bencode::decode("d0:1:ve");
  153. }
  154. {
  155. // empty string
  156. bencode::BDE s = bencode::decode("0:");
  157. CPPUNIT_ASSERT_EQUAL(std::string(""), s.s());
  158. }
  159. {
  160. // empty dict
  161. bencode::BDE d = bencode::decode("de");
  162. CPPUNIT_ASSERT(d.empty());
  163. }
  164. {
  165. // empty list
  166. bencode::BDE l = bencode::decode("le");
  167. CPPUNIT_ASSERT(l.empty());
  168. }
  169. {
  170. // integer, without ending 'e'
  171. try {
  172. bencode::decode("i3");
  173. CPPUNIT_FAIL("exception must be thrown.");
  174. } catch(RecoverableException& e) {
  175. CPPUNIT_ASSERT_EQUAL(std::string("Delimiter 'e' not found."),
  176. std::string(e.what()));
  177. }
  178. }
  179. {
  180. // dict, without ending 'e'
  181. try {
  182. bencode::decode("d");
  183. CPPUNIT_FAIL("exception must be thrown.");
  184. } catch(RecoverableException& e) {
  185. CPPUNIT_ASSERT_EQUAL(std::string("Unexpected EOF in dict context."
  186. " 'e' expected."),
  187. std::string(e.what()));
  188. }
  189. }
  190. {
  191. // list, without ending 'e'
  192. try {
  193. bencode::decode("l");
  194. CPPUNIT_FAIL("exception must be thrown.");
  195. } catch(RecoverableException& e) {
  196. CPPUNIT_ASSERT_EQUAL(std::string("Unexpected EOF in list context."
  197. " 'e' expected."),
  198. std::string(e.what()));
  199. }
  200. }
  201. {
  202. // string, less than the specified length.
  203. try {
  204. bencode::decode("3:ab");
  205. CPPUNIT_FAIL("exception must be thrown.");
  206. } catch(RecoverableException& e) {
  207. CPPUNIT_ASSERT_EQUAL(std::string("Expected 3 bytes of data,"
  208. " but only 2 read."),
  209. std::string(e.what()));
  210. }
  211. }
  212. {
  213. // string, but length is invalid
  214. try {
  215. bencode::decode("x:abc");
  216. CPPUNIT_FAIL("exception must be thrown.");
  217. } catch(RecoverableException& e) {
  218. CPPUNIT_ASSERT_EQUAL(std::string("A positive integer expected"
  219. " but none found."),
  220. std::string(e.what()));
  221. }
  222. }
  223. {
  224. // string with minus length
  225. try {
  226. bencode::decode("-1:a");
  227. CPPUNIT_FAIL("exception must be thrown.");
  228. } catch(RecoverableException& e) {
  229. CPPUNIT_ASSERT_EQUAL(std::string("A positive integer expected"
  230. " but none found."),
  231. std::string(e.what()));
  232. }
  233. }
  234. {
  235. // empty encoded data
  236. CPPUNIT_ASSERT(bencode::decode("").isNone());
  237. }
  238. {
  239. // ignore trailing garbage at the end of the input.
  240. bencode::BDE s = bencode::decode("5:aria2trail");
  241. CPPUNIT_ASSERT_EQUAL(std::string("aria2"), s.s());
  242. }
  243. }
  244. void BencodeTest::testEncode()
  245. {
  246. {
  247. bencode::BDE dict = bencode::BDE::dict();
  248. dict["name"] = std::string("aria2");
  249. dict["loc"] = 80000;
  250. dict["files"] = bencode::BDE::list();
  251. dict["files"] << std::string("aria2c");
  252. dict["attrs"] = bencode::BDE::dict();
  253. dict["attrs"]["license"] = std::string("GPL");
  254. CPPUNIT_ASSERT_EQUAL(std::string("d"
  255. "5:attrsd7:license3:GPLe"
  256. "5:filesl6:aria2ce"
  257. "3:loci80000e"
  258. "4:name5:aria2"
  259. "e"),
  260. bencode::encode(dict));
  261. }
  262. }
  263. } // namespace aria2