Metalink2RequestGroupTest.cc 8.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227
  1. #include "Metalink2RequestGroup.h"
  2. #include <algorithm>
  3. #include <cppunit/extensions/HelperMacros.h>
  4. #include "DownloadContext.h"
  5. #include "prefs.h"
  6. #include "Option.h"
  7. #include "RequestGroup.h"
  8. #include "FileEntry.h"
  9. namespace aria2 {
  10. class Metalink2RequestGroupTest:public CppUnit::TestFixture {
  11. CPPUNIT_TEST_SUITE(Metalink2RequestGroupTest);
  12. CPPUNIT_TEST(testGenerate);
  13. CPPUNIT_TEST(testGenerate_groupByMetaurl);
  14. CPPUNIT_TEST(testGenerate_dosDirTraversal);
  15. CPPUNIT_TEST_SUITE_END();
  16. private:
  17. SharedHandle<Option> option_;
  18. public:
  19. void setUp()
  20. {
  21. option_.reset(new Option());
  22. option_->put(PREF_SPLIT, "1");
  23. }
  24. void testGenerate();
  25. void testGenerate_groupByMetaurl();
  26. void testGenerate_dosDirTraversal();
  27. };
  28. CPPUNIT_TEST_SUITE_REGISTRATION( Metalink2RequestGroupTest );
  29. void Metalink2RequestGroupTest::testGenerate()
  30. {
  31. std::vector<SharedHandle<RequestGroup> > groups;
  32. option_->put(PREF_DIR, "/tmp");
  33. Metalink2RequestGroup().generate(groups, "test.xml", option_);
  34. // first file
  35. {
  36. SharedHandle<RequestGroup> rg = groups[0];
  37. std::vector<std::string> uris;
  38. rg->getDownloadContext()->getFirstFileEntry()->getUris(uris);
  39. std::sort(uris.begin(), uris.end());
  40. CPPUNIT_ASSERT_EQUAL((size_t)2, uris.size());
  41. CPPUNIT_ASSERT_EQUAL
  42. (std::string("ftp://ftphost/aria2-0.5.2.tar.bz2"), uris[0]);
  43. CPPUNIT_ASSERT_EQUAL
  44. (std::string("http://httphost/aria2-0.5.2.tar.bz2"), uris[1]);
  45. const SharedHandle<DownloadContext>& dctx = rg->getDownloadContext();
  46. CPPUNIT_ASSERT(!dctx.isNull());
  47. CPPUNIT_ASSERT_EQUAL((uint64_t)0ULL, dctx->getTotalLength());
  48. CPPUNIT_ASSERT_EQUAL(std::string("/tmp"), dctx->getDir());
  49. #ifdef ENABLE_MESSAGE_DIGEST
  50. CPPUNIT_ASSERT_EQUAL(std::string("sha-1"), dctx->getChecksumHashAlgo());
  51. CPPUNIT_ASSERT_EQUAL
  52. (std::string("a96cf3f0266b91d87d5124cf94326422800b627d"),
  53. dctx->getChecksum());
  54. #endif // ENABLE_MESSAGE_DIGEST
  55. CPPUNIT_ASSERT(!dctx->getSignature().isNull());
  56. CPPUNIT_ASSERT_EQUAL(std::string("pgp"), dctx->getSignature()->getType());
  57. }
  58. // second file
  59. {
  60. SharedHandle<RequestGroup> rg = groups[1];
  61. std::vector<std::string> uris;
  62. rg->getDownloadContext()->getFirstFileEntry()->getUris(uris);
  63. CPPUNIT_ASSERT_EQUAL((size_t)2, uris.size());
  64. const SharedHandle<DownloadContext>& dctx = rg->getDownloadContext();
  65. CPPUNIT_ASSERT(!dctx.isNull());
  66. CPPUNIT_ASSERT_EQUAL(std::string("/tmp"), dctx->getDir());
  67. #ifdef ENABLE_MESSAGE_DIGEST
  68. CPPUNIT_ASSERT_EQUAL(std::string("sha-1"), dctx->getPieceHashAlgo());
  69. CPPUNIT_ASSERT_EQUAL((size_t)2, dctx->getPieceHashes().size());
  70. CPPUNIT_ASSERT_EQUAL((size_t)262144, dctx->getPieceLength());
  71. CPPUNIT_ASSERT_EQUAL(std::string("sha-1"), dctx->getChecksumHashAlgo());
  72. CPPUNIT_ASSERT_EQUAL
  73. (std::string("4c255b0ed130f5ea880f0aa061c3da0487e251cc"),
  74. dctx->getChecksum());
  75. #endif // ENABLE_MESSAGE_DIGEST
  76. CPPUNIT_ASSERT(dctx->getSignature().isNull());
  77. }
  78. #ifdef ENABLE_BITTORRENT
  79. // fifth file <- downloading .torrent file
  80. {
  81. SharedHandle<RequestGroup> rg = groups[4];
  82. std::vector<std::string> uris;
  83. rg->getDownloadContext()->getFirstFileEntry()->getUris(uris);
  84. CPPUNIT_ASSERT_EQUAL((size_t)1, uris.size());
  85. CPPUNIT_ASSERT_EQUAL
  86. (std::string("http://host/torrent-http.integrated.torrent"), uris[0]);
  87. const SharedHandle<DownloadContext>& dctx = rg->getDownloadContext();
  88. CPPUNIT_ASSERT(!dctx.isNull());
  89. CPPUNIT_ASSERT_EQUAL(groups[5]->getGID(), rg->belongsTo());
  90. }
  91. #endif // ENABLE_BITTORRENT
  92. // sixth file <- depends on fifth file to download .torrent file.
  93. {
  94. #ifdef ENABLE_BITTORRENT
  95. SharedHandle<RequestGroup> rg = groups[5];
  96. #else
  97. SharedHandle<RequestGroup> rg = groups[4];
  98. #endif // ENABLE_BITTORRENT
  99. std::vector<std::string> uris;
  100. rg->getDownloadContext()->getFirstFileEntry()->getUris(uris);
  101. CPPUNIT_ASSERT_EQUAL((size_t)1, uris.size());
  102. CPPUNIT_ASSERT_EQUAL
  103. (std::string("http://host/torrent-http.integrated"), uris[0]);
  104. const SharedHandle<DownloadContext>& dctx = rg->getDownloadContext();
  105. CPPUNIT_ASSERT(!dctx.isNull());
  106. CPPUNIT_ASSERT_EQUAL(std::string("/tmp"), dctx->getDir());
  107. }
  108. }
  109. void Metalink2RequestGroupTest::testGenerate_groupByMetaurl()
  110. {
  111. std::vector<SharedHandle<RequestGroup> > groups;
  112. Metalink2RequestGroup().generate(groups, "metalink4-groupbymetaurl.xml",
  113. option_);
  114. CPPUNIT_ASSERT_EQUAL((size_t)3, groups.size());
  115. #ifdef ENABLE_BITTORRENT
  116. // first RequestGroup is torrent for second RequestGroup
  117. {
  118. SharedHandle<RequestGroup> rg = groups[0];
  119. std::vector<std::string> uris;
  120. rg->getDownloadContext()->getFirstFileEntry()->getUris(uris);
  121. CPPUNIT_ASSERT_EQUAL((size_t)1, uris.size());
  122. CPPUNIT_ASSERT_EQUAL(std::string("http://torrent"), uris[0]);
  123. }
  124. // second
  125. {
  126. SharedHandle<RequestGroup> rg = groups[1];
  127. SharedHandle<DownloadContext> dctx = rg->getDownloadContext();
  128. const std::vector<SharedHandle<FileEntry> >& fileEntries =
  129. dctx->getFileEntries();
  130. CPPUNIT_ASSERT_EQUAL((size_t)2, fileEntries.size());
  131. CPPUNIT_ASSERT_EQUAL(std::string("./file1"), fileEntries[0]->getPath());
  132. CPPUNIT_ASSERT_EQUAL(std::string("file1"), fileEntries[0]->getOriginalName());
  133. CPPUNIT_ASSERT_EQUAL((size_t)1, fileEntries[0]->getRemainingUris().size());
  134. CPPUNIT_ASSERT_EQUAL(std::string("http://file1p1"),
  135. fileEntries[0]->getRemainingUris()[0]);
  136. CPPUNIT_ASSERT_EQUAL(std::string("./file3"), fileEntries[1]->getPath());
  137. CPPUNIT_ASSERT_EQUAL(std::string("file3"), fileEntries[1]->getOriginalName());
  138. CPPUNIT_ASSERT_EQUAL((size_t)1, fileEntries[1]->getRemainingUris().size());
  139. CPPUNIT_ASSERT_EQUAL(std::string("http://file3p1"),
  140. fileEntries[1]->getRemainingUris()[0]);
  141. }
  142. // third
  143. {
  144. SharedHandle<RequestGroup> rg = groups[2];
  145. SharedHandle<DownloadContext> dctx = rg->getDownloadContext();
  146. const std::vector<SharedHandle<FileEntry> >& fileEntries =
  147. dctx->getFileEntries();
  148. CPPUNIT_ASSERT_EQUAL((size_t)1, fileEntries.size());
  149. CPPUNIT_ASSERT_EQUAL(std::string("./file2"), fileEntries[0]->getPath());
  150. CPPUNIT_ASSERT_EQUAL((size_t)1, fileEntries[0]->getRemainingUris().size());
  151. CPPUNIT_ASSERT_EQUAL(std::string("http://file2p1"),
  152. fileEntries[0]->getRemainingUris()[0]);
  153. }
  154. #else // !ENABLE_BITTORRENT
  155. {
  156. SharedHandle<RequestGroup> rg = groups[0];
  157. std::vector<std::string> uris;
  158. rg->getDownloadContext()->getFirstFileEntry()->getUris(uris);
  159. CPPUNIT_ASSERT_EQUAL((size_t)1, uris.size());
  160. CPPUNIT_ASSERT_EQUAL(std::string("http://file1p1"), uris[0]);
  161. }
  162. {
  163. SharedHandle<RequestGroup> rg = groups[1];
  164. std::vector<std::string> uris;
  165. rg->getDownloadContext()->getFirstFileEntry()->getUris(uris);
  166. CPPUNIT_ASSERT_EQUAL((size_t)1, uris.size());
  167. CPPUNIT_ASSERT_EQUAL(std::string("http://file2p1"), uris[0]);
  168. }
  169. {
  170. SharedHandle<RequestGroup> rg = groups[2];
  171. std::vector<std::string> uris;
  172. rg->getDownloadContext()->getFirstFileEntry()->getUris(uris);
  173. CPPUNIT_ASSERT_EQUAL((size_t)1, uris.size());
  174. CPPUNIT_ASSERT_EQUAL(std::string("http://file3p1"), uris[0]);
  175. }
  176. #endif // !ENABLE_BITTORRENT
  177. }
  178. void Metalink2RequestGroupTest::testGenerate_dosDirTraversal()
  179. {
  180. #ifdef __MINGW32__
  181. #ifdef ENABLE_BITTORRENT
  182. std::vector<SharedHandle<RequestGroup> > groups;
  183. option_->put(PREF_DIR, "/tmp");
  184. Metalink2RequestGroup().generate
  185. (groups, "metalink4-dosdirtraversal.xml", option_);
  186. CPPUNIT_ASSERT_EQUAL((size_t)3, groups.size());
  187. SharedHandle<RequestGroup> rg = groups[0];
  188. SharedHandle<FileEntry> file = rg->getDownloadContext()->getFirstFileEntry();
  189. CPPUNIT_ASSERT_EQUAL(std::string("/tmp/..%5C..%5Cexample.ext"),
  190. file->getPath());
  191. rg = groups[2];
  192. file = rg->getDownloadContext()->getFileEntries()[0];
  193. CPPUNIT_ASSERT_EQUAL(std::string("/tmp/..%5C..%5Cfile1.ext"),
  194. file->getPath());
  195. file = rg->getDownloadContext()->getFileEntries()[1];
  196. CPPUNIT_ASSERT_EQUAL(std::string("/tmp/..%5C..%5Cfile2.ext"),
  197. file->getPath());
  198. #endif // ENABLE_BITTORRENT
  199. #endif // __MINGW32__
  200. }
  201. } // namespace aria2