UtilTest.cc 38 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179
  1. #include "util.h"
  2. #include <cstring>
  3. #include <string>
  4. #include <iostream>
  5. #include <cppunit/extensions/HelperMacros.h>
  6. #include "FixedNumberRandomizer.h"
  7. #include "DlAbortEx.h"
  8. #include "BitfieldMan.h"
  9. #include "ByteArrayDiskWriter.h"
  10. #include "FileEntry.h"
  11. #include "File.h"
  12. #include "array_fun.h"
  13. namespace aria2 {
  14. class UtilTest:public CppUnit::TestFixture {
  15. CPPUNIT_TEST_SUITE(UtilTest);
  16. CPPUNIT_TEST(testStrip);
  17. CPPUNIT_TEST(testDivide);
  18. CPPUNIT_TEST(testSplit);
  19. CPPUNIT_TEST(testEndsWith);
  20. CPPUNIT_TEST(testReplace);
  21. CPPUNIT_TEST(testStartsWith);
  22. // may be moved to other helper class in the future.
  23. CPPUNIT_TEST(testGetContentDispositionFilename);
  24. CPPUNIT_TEST(testRandomAlpha);
  25. CPPUNIT_TEST(testToUpper);
  26. CPPUNIT_TEST(testToLower);
  27. CPPUNIT_TEST(testUppercase);
  28. CPPUNIT_TEST(testLowercase);
  29. CPPUNIT_TEST(testPercentDecode);
  30. CPPUNIT_TEST(testGetRealSize);
  31. CPPUNIT_TEST(testAbbrevSize);
  32. CPPUNIT_TEST(testToStream);
  33. CPPUNIT_TEST(testIsNumber);
  34. CPPUNIT_TEST(testIsLowercase);
  35. CPPUNIT_TEST(testIsUppercase);
  36. CPPUNIT_TEST(testAlphaToNum);
  37. CPPUNIT_TEST(testMkdirs);
  38. CPPUNIT_TEST(testConvertBitfield);
  39. CPPUNIT_TEST(testParseIntRange);
  40. CPPUNIT_TEST(testParseIntRange_invalidRange);
  41. CPPUNIT_TEST(testParseInt);
  42. CPPUNIT_TEST(testParseUInt);
  43. CPPUNIT_TEST(testParseLLInt);
  44. CPPUNIT_TEST(testParseULLInt);
  45. CPPUNIT_TEST(testParseUIntNoThrow);
  46. CPPUNIT_TEST(testParseLLIntNoThrow);
  47. CPPUNIT_TEST(testToString_binaryStream);
  48. CPPUNIT_TEST(testItos);
  49. CPPUNIT_TEST(testUitos);
  50. CPPUNIT_TEST(testNtoh64);
  51. CPPUNIT_TEST(testPercentEncode);
  52. CPPUNIT_TEST(testHtmlEscape);
  53. CPPUNIT_TEST(testJoinPath);
  54. CPPUNIT_TEST(testParseIndexPath);
  55. CPPUNIT_TEST(testCreateIndexPathMap);
  56. CPPUNIT_TEST(testGenerateRandomData);
  57. CPPUNIT_TEST(testFromHex);
  58. CPPUNIT_TEST(testParsePrioritizePieceRange);
  59. CPPUNIT_TEST(testApplyDir);
  60. CPPUNIT_TEST(testFixTaintedBasename);
  61. CPPUNIT_TEST(testIsNumericHost);
  62. CPPUNIT_TEST(testDetectDirTraversal);
  63. CPPUNIT_TEST(testEscapePath);
  64. CPPUNIT_TEST(testGetCidrPrefix);
  65. CPPUNIT_TEST(testInSameCidrBlock);
  66. CPPUNIT_TEST(testIsUtf8String);
  67. CPPUNIT_TEST_SUITE_END();
  68. private:
  69. public:
  70. void setUp() {
  71. }
  72. void testStrip();
  73. void testDivide();
  74. void testSplit();
  75. void testEndsWith();
  76. void testReplace();
  77. void testStartsWith();
  78. // may be moved to other helper class in the future.
  79. void testGetContentDispositionFilename();
  80. void testRandomAlpha();
  81. void testToUpper();
  82. void testToLower();
  83. void testUppercase();
  84. void testLowercase();
  85. void testPercentDecode();
  86. void testGetRealSize();
  87. void testAbbrevSize();
  88. void testToStream();
  89. void testIsNumber();
  90. void testIsLowercase();
  91. void testIsUppercase();
  92. void testAlphaToNum();
  93. void testMkdirs();
  94. void testConvertBitfield();
  95. void testParseIntRange();
  96. void testParseIntRange_invalidRange();
  97. void testParseInt();
  98. void testParseUInt();
  99. void testParseLLInt();
  100. void testParseULLInt();
  101. void testParseUIntNoThrow();
  102. void testParseLLIntNoThrow();
  103. void testToString_binaryStream();
  104. void testItos();
  105. void testUitos();
  106. void testNtoh64();
  107. void testPercentEncode();
  108. void testHtmlEscape();
  109. void testJoinPath();
  110. void testParseIndexPath();
  111. void testCreateIndexPathMap();
  112. void testGenerateRandomData();
  113. void testFromHex();
  114. void testParsePrioritizePieceRange();
  115. void testApplyDir();
  116. void testFixTaintedBasename();
  117. void testIsNumericHost();
  118. void testDetectDirTraversal();
  119. void testEscapePath();
  120. void testGetCidrPrefix();
  121. void testInSameCidrBlock();
  122. void testIsUtf8String();
  123. };
  124. CPPUNIT_TEST_SUITE_REGISTRATION( UtilTest );
  125. void UtilTest::testStrip()
  126. {
  127. std::string str1 = "aria2";
  128. CPPUNIT_ASSERT_EQUAL(str1, util::strip("aria2"));
  129. CPPUNIT_ASSERT_EQUAL(str1, util::strip(" aria2"));
  130. CPPUNIT_ASSERT_EQUAL(str1, util::strip("aria2 "));
  131. CPPUNIT_ASSERT_EQUAL(str1, util::strip(" aria2 "));
  132. CPPUNIT_ASSERT_EQUAL(str1, util::strip(" aria2 "));
  133. std::string str2 = "aria2 debut";
  134. CPPUNIT_ASSERT_EQUAL(str2, util::strip("aria2 debut"));
  135. CPPUNIT_ASSERT_EQUAL(str2, util::strip(" aria2 debut "));
  136. std::string str3 = "";
  137. CPPUNIT_ASSERT_EQUAL(str3, util::strip(""));
  138. CPPUNIT_ASSERT_EQUAL(str3, util::strip(" "));
  139. CPPUNIT_ASSERT_EQUAL(str3, util::strip(" "));
  140. std::string str4 = "A";
  141. CPPUNIT_ASSERT_EQUAL(str4, util::strip("A"));
  142. CPPUNIT_ASSERT_EQUAL(str4, util::strip(" A "));
  143. CPPUNIT_ASSERT_EQUAL(str4, util::strip(" A "));
  144. }
  145. void UtilTest::testDivide() {
  146. std::pair<std::string, std::string> p1;
  147. util::divide(p1, "name=value", '=');
  148. CPPUNIT_ASSERT_EQUAL(std::string("name"), p1.first);
  149. CPPUNIT_ASSERT_EQUAL(std::string("value"), p1.second);
  150. util::divide(p1, " name = value ", '=');
  151. CPPUNIT_ASSERT_EQUAL(std::string("name"), p1.first);
  152. CPPUNIT_ASSERT_EQUAL(std::string("value"), p1.second);
  153. util::divide(p1, "=value", '=');
  154. CPPUNIT_ASSERT_EQUAL(std::string(""), p1.first);
  155. CPPUNIT_ASSERT_EQUAL(std::string("value"), p1.second);
  156. util::divide(p1, "name=", '=');
  157. CPPUNIT_ASSERT_EQUAL(std::string("name"), p1.first);
  158. CPPUNIT_ASSERT_EQUAL(std::string(""), p1.second);
  159. util::divide(p1, "name", '=');
  160. CPPUNIT_ASSERT_EQUAL(std::string("name"), p1.first);
  161. CPPUNIT_ASSERT_EQUAL(std::string(""), p1.second);
  162. }
  163. void UtilTest::testSplit() {
  164. std::vector<std::string> v1;
  165. util::split("name1=value1; name2=value2; name3=value3",std::back_inserter(v1),
  166. ";", true);
  167. CPPUNIT_ASSERT_EQUAL((size_t)3, v1.size());
  168. std::vector<std::string>::iterator itr = v1.begin();
  169. CPPUNIT_ASSERT_EQUAL(std::string("name1=value1"), *itr++);
  170. CPPUNIT_ASSERT_EQUAL(std::string("name2=value2"), *itr++);
  171. CPPUNIT_ASSERT_EQUAL(std::string("name3=value3"), *itr++);
  172. v1.clear();
  173. util::split("name1=value1; name2=value2; name3=value3",std::back_inserter(v1),
  174. ";", false);
  175. CPPUNIT_ASSERT_EQUAL((size_t)3, v1.size());
  176. itr = v1.begin();
  177. CPPUNIT_ASSERT_EQUAL(std::string("name1=value1"), *itr++);
  178. CPPUNIT_ASSERT_EQUAL(std::string(" name2=value2"), *itr++);
  179. CPPUNIT_ASSERT_EQUAL(std::string(" name3=value3"), *itr++);
  180. v1.clear();
  181. util::split("k=v", std::back_inserter(v1), ";", false, true);
  182. CPPUNIT_ASSERT_EQUAL((size_t)1, v1.size());
  183. CPPUNIT_ASSERT_EQUAL(std::string("k=v"), v1[0]);
  184. v1.clear();
  185. util::split(" ", std::back_inserter(v1), ";", true, true);
  186. CPPUNIT_ASSERT_EQUAL((size_t)1, v1.size());
  187. CPPUNIT_ASSERT_EQUAL(std::string(""), v1[0]);
  188. v1.clear();
  189. util::split(" ", std::back_inserter(v1), ";", true);
  190. CPPUNIT_ASSERT_EQUAL((size_t)0, v1.size());
  191. }
  192. void UtilTest::testEndsWith() {
  193. std::string target = "abcdefg";
  194. std::string part = "fg";
  195. CPPUNIT_ASSERT(util::endsWith(target, part));
  196. target = "abdefg";
  197. part = "g";
  198. CPPUNIT_ASSERT(util::endsWith(target, part));
  199. target = "abdefg";
  200. part = "eg";
  201. CPPUNIT_ASSERT(!util::endsWith(target, part));
  202. target = "g";
  203. part = "eg";
  204. CPPUNIT_ASSERT(!util::endsWith(target, part));
  205. target = "g";
  206. part = "g";
  207. CPPUNIT_ASSERT(util::endsWith(target, part));
  208. target = "g";
  209. part = "";
  210. CPPUNIT_ASSERT(util::endsWith(target, part));
  211. target = "";
  212. part = "";
  213. CPPUNIT_ASSERT(util::endsWith(target, part));
  214. target = "";
  215. part = "g";
  216. CPPUNIT_ASSERT(!util::endsWith(target, part));
  217. }
  218. void UtilTest::testReplace() {
  219. CPPUNIT_ASSERT_EQUAL(std::string("abc\n"), util::replace("abc\r\n", "\r", ""));
  220. CPPUNIT_ASSERT_EQUAL(std::string("abc"), util::replace("abc\r\n", "\r\n", ""));
  221. CPPUNIT_ASSERT_EQUAL(std::string(""), util::replace("", "\r\n", ""));
  222. CPPUNIT_ASSERT_EQUAL(std::string("abc"), util::replace("abc", "", "a"));
  223. CPPUNIT_ASSERT_EQUAL(std::string("xbc"), util::replace("abc", "a", "x"));
  224. }
  225. void UtilTest::testStartsWith() {
  226. std::string target;
  227. std::string part;
  228. target = "abcdefg";
  229. part = "abc";
  230. CPPUNIT_ASSERT(util::startsWith(target, part));
  231. target = "abcdefg";
  232. part = "abx";
  233. CPPUNIT_ASSERT(!util::startsWith(target, part));
  234. target = "abcdefg";
  235. part = "bcd";
  236. CPPUNIT_ASSERT(!util::startsWith(target, part));
  237. target = "";
  238. part = "a";
  239. CPPUNIT_ASSERT(!util::startsWith(target, part));
  240. target = "";
  241. part = "";
  242. CPPUNIT_ASSERT(util::startsWith(target, part));
  243. target = "a";
  244. part = "";
  245. CPPUNIT_ASSERT(util::startsWith(target, part));
  246. target = "a";
  247. part = "a";
  248. CPPUNIT_ASSERT(util::startsWith(target, part));
  249. }
  250. void UtilTest::testGetContentDispositionFilename() {
  251. std::string h1 = "attachment; filename=\"aria2.tar.bz2\"";
  252. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"), util::getContentDispositionFilename(h1));
  253. std::string h2 = "attachment; filename=\"\"";
  254. CPPUNIT_ASSERT_EQUAL(std::string(""), util::getContentDispositionFilename(h2));
  255. std::string h3 = "attachment; filename=\"";
  256. CPPUNIT_ASSERT_EQUAL(std::string(""), util::getContentDispositionFilename(h3));
  257. std::string h4 = "attachment;";
  258. CPPUNIT_ASSERT_EQUAL(std::string(""), util::getContentDispositionFilename(h4));
  259. std::string h5 = "attachment; filename=aria2.tar.bz2";
  260. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"), util::getContentDispositionFilename(h5));
  261. std::string h6 = "attachment; filename='aria2.tar.bz2'";
  262. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"), util::getContentDispositionFilename(h6));
  263. std::string h7 = "attachment; filename='aria2.tar.bz2";
  264. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"), util::getContentDispositionFilename(h7));
  265. std::string h8 = "attachment; filename=aria2.tar.bz2; creation-date=20 Jun 2007 00:00:00 GMT";
  266. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"), util::getContentDispositionFilename(h8));
  267. std::string h9 = "attachment; filename=\"aria2.tar.bz2; creation-date=20 Jun 2007 00:00:00 GMT\"";
  268. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2; creation-date=20 Jun 2007 00:00:00 GMT"),
  269. util::getContentDispositionFilename(h9));
  270. std::string h10 = "attachment; filename=";
  271. CPPUNIT_ASSERT_EQUAL(std::string(""), util::getContentDispositionFilename(h10));
  272. std::string h11 = "attachment; filename=;";
  273. CPPUNIT_ASSERT_EQUAL(std::string(""), util::getContentDispositionFilename(h11));
  274. std::string filenameWithDir = "attachment; filename=dir/file";
  275. CPPUNIT_ASSERT_EQUAL(std::string(""),
  276. util::getContentDispositionFilename(filenameWithDir));
  277. std::string semicolonInside = "attachment; filename=\"foo;bar\"";
  278. CPPUNIT_ASSERT_EQUAL(std::string("foo;bar"),
  279. util::getContentDispositionFilename(semicolonInside));
  280. CPPUNIT_ASSERT_EQUAL
  281. (std::string(""),
  282. util::getContentDispositionFilename("filename=\"%2E%2E%2Ffoo.html\""));
  283. // RFC2231 Section4
  284. std::string extparam2 = "filename*=''aria2";
  285. CPPUNIT_ASSERT_EQUAL(std::string("aria2"),
  286. util::getContentDispositionFilename(extparam2));
  287. std::string extparam3 = "filename*='''";
  288. CPPUNIT_ASSERT_EQUAL(std::string(""),
  289. util::getContentDispositionFilename(extparam3));
  290. std::string extparam4 = "filename*='aria2";
  291. CPPUNIT_ASSERT_EQUAL(std::string(""),
  292. util::getContentDispositionFilename(extparam4));
  293. std::string extparam5 = "filename*='''aria2";
  294. CPPUNIT_ASSERT_EQUAL(std::string(""),
  295. util::getContentDispositionFilename(extparam5));
  296. std::string extparam6 = "filename*";
  297. CPPUNIT_ASSERT_EQUAL(std::string(""),
  298. util::getContentDispositionFilename(extparam6));
  299. std::string extparam7 = "filename*=UTF-8''aria2;filename=hello%20world";
  300. CPPUNIT_ASSERT_EQUAL(std::string("aria2"),
  301. util::getContentDispositionFilename(extparam7));
  302. std::string extparam8 = "filename=aria2;filename*=UTF-8''hello%20world";
  303. CPPUNIT_ASSERT_EQUAL(std::string("hello world"),
  304. util::getContentDispositionFilename(extparam8));
  305. std::string extparam9 = "filename*=ISO-8859-1''%A3";
  306. std::string extparam9ans;
  307. extparam9ans += 0xc2;
  308. extparam9ans += 0xa3;
  309. CPPUNIT_ASSERT_EQUAL(extparam9ans,
  310. util::getContentDispositionFilename(extparam9));
  311. CPPUNIT_ASSERT_EQUAL
  312. (std::string(""),
  313. util::getContentDispositionFilename("filename*=UTF-8''foo%2F.html"));
  314. CPPUNIT_ASSERT_EQUAL
  315. (std::string("foo.html"),
  316. util::getContentDispositionFilename("filename*=UTF-8'';filename=\"foo.html\""));
  317. CPPUNIT_ASSERT_EQUAL
  318. (std::string(""),
  319. util::getContentDispositionFilename("filename*=UTF-8''%2E%2E%2Ffoo.html"));
  320. // Tests from http://greenbytes.de/tech/tc2231/
  321. // attwithasciifnescapedchar
  322. CPPUNIT_ASSERT_EQUAL
  323. (std::string("foo.html"),
  324. util::getContentDispositionFilename("filename=\"f\\oo.html\""));
  325. // attwithasciifilenameucase
  326. CPPUNIT_ASSERT_EQUAL
  327. (std::string("foo.html"),
  328. util::getContentDispositionFilename("FILENAME=\"foo.html\""));
  329. // attwithisofn2231iso
  330. CPPUNIT_ASSERT_EQUAL
  331. (std::string("foo-ä.html"),
  332. util::getContentDispositionFilename("filename*=iso-8859-1''foo-%E4.html"));
  333. // attwithfn2231utf8
  334. CPPUNIT_ASSERT_EQUAL
  335. (std::string("foo-ä-€.html"),
  336. util::getContentDispositionFilename
  337. ("filename*=UTF-8''foo-%c3%a4-%e2%82%ac.html"));
  338. // attwithfn2231utf8-bad
  339. CPPUNIT_ASSERT_EQUAL
  340. (std::string(""),
  341. util::getContentDispositionFilename
  342. ("filename*=iso-8859-1''foo-%c3%a4-%e2%82%ac.html"));
  343. // attwithfn2231ws1
  344. CPPUNIT_ASSERT_EQUAL
  345. (std::string(""),
  346. util::getContentDispositionFilename("filename *=UTF-8''foo-%c3%a4.html"));
  347. // attwithfn2231ws2
  348. CPPUNIT_ASSERT_EQUAL
  349. (std::string("foo-ä.html"),
  350. util::getContentDispositionFilename("filename*= UTF-8''foo-%c3%a4.html"));
  351. // attwithfn2231ws3
  352. CPPUNIT_ASSERT_EQUAL
  353. (std::string("foo-ä.html"),
  354. util::getContentDispositionFilename("filename* =UTF-8''foo-%c3%a4.html"));
  355. // attwithfn2231quot
  356. CPPUNIT_ASSERT_EQUAL
  357. (std::string(""),
  358. util::getContentDispositionFilename
  359. ("filename*=\"UTF-8''foo-%c3%a4.html\""));
  360. }
  361. class Printer {
  362. public:
  363. template<class T>
  364. void operator()(T t) {
  365. std::cerr << t << ", ";
  366. }
  367. };
  368. void UtilTest::testRandomAlpha() {
  369. SharedHandle<Randomizer> rand(new FixedNumberRandomizer());
  370. std::string s = util::randomAlpha(8, rand);
  371. CPPUNIT_ASSERT_EQUAL(std::string("AAAAAAAA"), s);
  372. }
  373. void UtilTest::testToUpper() {
  374. std::string src = "608cabc0f2fa18c260cafd974516865c772363d5";
  375. std::string upp = "608CABC0F2FA18C260CAFD974516865C772363D5";
  376. CPPUNIT_ASSERT_EQUAL(upp, util::toUpper(src));
  377. }
  378. void UtilTest::testToLower() {
  379. std::string src = "608CABC0F2FA18C260CAFD974516865C772363D5";
  380. std::string upp = "608cabc0f2fa18c260cafd974516865c772363d5";
  381. CPPUNIT_ASSERT_EQUAL(upp, util::toLower(src));
  382. }
  383. void UtilTest::testUppercase() {
  384. std::string src = "608cabc0f2fa18c260cafd974516865c772363d5";
  385. std::string ans = "608CABC0F2FA18C260CAFD974516865C772363D5";
  386. util::uppercase(src);
  387. CPPUNIT_ASSERT_EQUAL(ans, src);
  388. }
  389. void UtilTest::testLowercase() {
  390. std::string src = "608CABC0F2FA18C260CAFD974516865C772363D5";
  391. std::string ans = "608cabc0f2fa18c260cafd974516865c772363d5";
  392. util::lowercase(src);
  393. CPPUNIT_ASSERT_EQUAL(ans, src);
  394. }
  395. void UtilTest::testPercentDecode() {
  396. std::string src = "http://aria2.sourceforge.net/aria2%200.7.0%20docs.html";
  397. CPPUNIT_ASSERT_EQUAL(std::string("http://aria2.sourceforge.net/aria2 0.7.0 docs.html"),
  398. util::percentDecode(src));
  399. std::string src2 = "aria2+aria2";
  400. CPPUNIT_ASSERT_EQUAL(std::string("aria2+aria2"), util::percentDecode(src2));
  401. std::string src3 = "%5t%20";
  402. CPPUNIT_ASSERT_EQUAL(std::string("%5t "), util::percentDecode(src3));
  403. std::string src4 = "%";
  404. CPPUNIT_ASSERT_EQUAL(std::string("%"), util::percentDecode(src4));
  405. std::string src5 = "%3";
  406. CPPUNIT_ASSERT_EQUAL(std::string("%3"), util::percentDecode(src5));
  407. std::string src6 = "%2f";
  408. CPPUNIT_ASSERT_EQUAL(std::string("/"), util::percentDecode(src6));
  409. }
  410. void UtilTest::testGetRealSize()
  411. {
  412. CPPUNIT_ASSERT_EQUAL((int64_t)4294967296LL, util::getRealSize("4096M"));
  413. CPPUNIT_ASSERT_EQUAL((int64_t)1024, util::getRealSize("1K"));
  414. try {
  415. util::getRealSize("");
  416. CPPUNIT_FAIL("exception must be thrown.");
  417. } catch(Exception& e) {
  418. std::cerr << e.stackTrace();
  419. }
  420. try {
  421. util::getRealSize("foo");
  422. CPPUNIT_FAIL("exception must be thrown.");
  423. } catch(Exception& e) {
  424. std::cerr << e.stackTrace();
  425. }
  426. try {
  427. util::getRealSize("-1");
  428. CPPUNIT_FAIL("exception must be thrown.");
  429. } catch(Exception& e) {
  430. std::cerr << e.stackTrace();
  431. }
  432. try {
  433. util::getRealSize("9223372036854775807K");
  434. CPPUNIT_FAIL("exception must be thrown.");
  435. } catch(Exception& e) {
  436. std::cerr << e.stackTrace();
  437. }
  438. try {
  439. util::getRealSize("9223372036854775807M");
  440. CPPUNIT_FAIL("exception must be thrown.");
  441. } catch(Exception& e) {
  442. std::cerr << e.stackTrace();
  443. }
  444. }
  445. void UtilTest::testAbbrevSize()
  446. {
  447. CPPUNIT_ASSERT_EQUAL(std::string("4,096.0Mi"), util::abbrevSize(4294967296LL));
  448. CPPUNIT_ASSERT_EQUAL(std::string("1.0Ki"), util::abbrevSize(1024));
  449. CPPUNIT_ASSERT_EQUAL(std::string("1,023"), util::abbrevSize(1023));
  450. CPPUNIT_ASSERT_EQUAL(std::string("0"), util::abbrevSize(0));
  451. CPPUNIT_ASSERT_EQUAL(std::string("1.1Ki"), util::abbrevSize(1127));
  452. CPPUNIT_ASSERT_EQUAL(std::string("1.5Mi"), util::abbrevSize(1572864));
  453. }
  454. void UtilTest::testToStream()
  455. {
  456. std::ostringstream os;
  457. SharedHandle<FileEntry> f1(new FileEntry("aria2.tar.bz2", 12300, 0));
  458. SharedHandle<FileEntry> f2(new FileEntry("aria2.txt", 556, 0));
  459. std::deque<SharedHandle<FileEntry> > entries;
  460. entries.push_back(f1);
  461. entries.push_back(f2);
  462. util::toStream(entries.begin(), entries.end(), os);
  463. CPPUNIT_ASSERT_EQUAL(
  464. std::string("Files:\n"
  465. "idx|path/length\n"
  466. "===+===========================================================================\n"
  467. " 1|aria2.tar.bz2\n"
  468. " |12.0KiB (12,300)\n"
  469. "---+---------------------------------------------------------------------------\n"
  470. " 2|aria2.txt\n"
  471. " |556B (556)\n"
  472. "---+---------------------------------------------------------------------------\n"),
  473. os.str());
  474. }
  475. void UtilTest::testIsNumber()
  476. {
  477. CPPUNIT_ASSERT_EQUAL(true, util::isNumber("000"));
  478. CPPUNIT_ASSERT_EQUAL(false, util::isNumber("a"));
  479. CPPUNIT_ASSERT_EQUAL(false, util::isNumber("0a"));
  480. CPPUNIT_ASSERT_EQUAL(false, util::isNumber(""));
  481. CPPUNIT_ASSERT_EQUAL(false, util::isNumber(" "));
  482. }
  483. void UtilTest::testIsLowercase()
  484. {
  485. CPPUNIT_ASSERT_EQUAL(true, util::isLowercase("alpha"));
  486. CPPUNIT_ASSERT_EQUAL(false, util::isLowercase("Alpha"));
  487. CPPUNIT_ASSERT_EQUAL(false, util::isLowercase("1alpha"));
  488. CPPUNIT_ASSERT_EQUAL(false, util::isLowercase(""));
  489. CPPUNIT_ASSERT_EQUAL(false, util::isLowercase(" "));
  490. }
  491. void UtilTest::testIsUppercase()
  492. {
  493. CPPUNIT_ASSERT_EQUAL(true, util::isUppercase("ALPHA"));
  494. CPPUNIT_ASSERT_EQUAL(false, util::isUppercase("Alpha"));
  495. CPPUNIT_ASSERT_EQUAL(false, util::isUppercase("1ALPHA"));
  496. CPPUNIT_ASSERT_EQUAL(false, util::isUppercase(""));
  497. CPPUNIT_ASSERT_EQUAL(false, util::isUppercase(" "));
  498. }
  499. void UtilTest::testAlphaToNum()
  500. {
  501. CPPUNIT_ASSERT_EQUAL(0U, util::alphaToNum("a"));
  502. CPPUNIT_ASSERT_EQUAL(0U, util::alphaToNum("aa"));
  503. CPPUNIT_ASSERT_EQUAL(1U, util::alphaToNum("b"));
  504. CPPUNIT_ASSERT_EQUAL(675U, util::alphaToNum("zz")); // 25*26+25
  505. CPPUNIT_ASSERT_EQUAL(675U, util::alphaToNum("ZZ")); // 25*26+25
  506. CPPUNIT_ASSERT_EQUAL(0U, util::alphaToNum(""));
  507. CPPUNIT_ASSERT_EQUAL(4294967295U, util::alphaToNum("NXMRLXV"));
  508. CPPUNIT_ASSERT_EQUAL(0U, util::alphaToNum("NXMRLXW")); // uint32_t overflow
  509. }
  510. void UtilTest::testMkdirs()
  511. {
  512. std::string dir = "./aria2-UtilTest-testMkdirs";
  513. File d(dir);
  514. if(d.exists()) {
  515. CPPUNIT_ASSERT(d.remove());
  516. }
  517. CPPUNIT_ASSERT(!d.exists());
  518. util::mkdirs(dir);
  519. CPPUNIT_ASSERT(d.isDir());
  520. std::string file = "./UtilTest.cc";
  521. File f(file);
  522. CPPUNIT_ASSERT(f.isFile());
  523. try {
  524. util::mkdirs(file);
  525. CPPUNIT_FAIL("exception must be thrown.");
  526. } catch(DlAbortEx& ex) {
  527. std::cerr << ex.stackTrace() << std::endl;
  528. }
  529. }
  530. void UtilTest::testConvertBitfield()
  531. {
  532. BitfieldMan srcBitfield(384*1024, 256*1024*256+1);
  533. BitfieldMan destBitfield(512*1024, srcBitfield.getTotalLength());
  534. srcBitfield.setAllBit();
  535. srcBitfield.unsetBit(2);// <- range [768, 1152)
  536. // which corresponds to the index [1,2] in destBitfield
  537. util::convertBitfield(&destBitfield, &srcBitfield);
  538. CPPUNIT_ASSERT_EQUAL(std::string("9fffffffffffffffffffffffffffffff80"),
  539. util::toHex(destBitfield.getBitfield(),
  540. destBitfield.getBitfieldLength()));
  541. }
  542. void UtilTest::testParseIntRange()
  543. {
  544. IntSequence seq = util::parseIntRange("1,3-8,10");
  545. CPPUNIT_ASSERT(seq.hasNext());
  546. CPPUNIT_ASSERT_EQUAL((int32_t)1, seq.next());
  547. CPPUNIT_ASSERT(seq.hasNext());
  548. CPPUNIT_ASSERT_EQUAL((int32_t)3, seq.next());
  549. CPPUNIT_ASSERT(seq.hasNext());
  550. CPPUNIT_ASSERT_EQUAL((int32_t)4, seq.next());
  551. CPPUNIT_ASSERT(seq.hasNext());
  552. CPPUNIT_ASSERT_EQUAL((int32_t)5, seq.next());
  553. CPPUNIT_ASSERT(seq.hasNext());
  554. CPPUNIT_ASSERT_EQUAL((int32_t)6, seq.next());
  555. CPPUNIT_ASSERT(seq.hasNext());
  556. CPPUNIT_ASSERT_EQUAL((int32_t)7, seq.next());
  557. CPPUNIT_ASSERT(seq.hasNext());
  558. CPPUNIT_ASSERT_EQUAL((int32_t)8, seq.next());
  559. CPPUNIT_ASSERT(seq.hasNext());
  560. CPPUNIT_ASSERT_EQUAL((int32_t)10, seq.next());
  561. CPPUNIT_ASSERT(!seq.hasNext());
  562. CPPUNIT_ASSERT_EQUAL((int32_t)0, seq.next());
  563. }
  564. void UtilTest::testParseIntRange_invalidRange()
  565. {
  566. try {
  567. IntSequence seq = util::parseIntRange("-1");
  568. CPPUNIT_FAIL("exception must be thrown.");
  569. } catch(Exception& e) {
  570. std::cerr << e.stackTrace();
  571. }
  572. try {
  573. IntSequence seq = util::parseIntRange("2147483648");
  574. CPPUNIT_FAIL("exception must be thrown.");
  575. } catch(Exception& e) {
  576. std::cerr << e.stackTrace();
  577. }
  578. try {
  579. IntSequence seq = util::parseIntRange("2147483647-2147483648");
  580. CPPUNIT_FAIL("exception must be thrown.");
  581. } catch(Exception& e) {
  582. std::cerr << e.stackTrace();
  583. }
  584. try {
  585. IntSequence seq = util::parseIntRange("1-2x");
  586. CPPUNIT_FAIL("exception must be thrown.");
  587. } catch(Exception& e) {
  588. std::cerr << e.stackTrace();
  589. }
  590. try {
  591. IntSequence seq = util::parseIntRange("3x-4");
  592. CPPUNIT_FAIL("exception must be thrown.");
  593. } catch(Exception& e) {
  594. std::cerr << e.stackTrace();
  595. }
  596. }
  597. void UtilTest::testParseInt()
  598. {
  599. CPPUNIT_ASSERT_EQUAL(-1, util::parseInt(" -1 "));
  600. CPPUNIT_ASSERT_EQUAL(2147483647, util::parseInt("2147483647"));
  601. try {
  602. util::parseInt("2147483648");
  603. CPPUNIT_FAIL("exception must be thrown.");
  604. } catch(Exception& e) {
  605. std::cerr << e.stackTrace();
  606. }
  607. try {
  608. util::parseInt("-2147483649");
  609. CPPUNIT_FAIL("exception must be thrown.");
  610. } catch(Exception& e) {
  611. std::cerr << e.stackTrace();
  612. }
  613. try {
  614. util::parseInt("12x");
  615. CPPUNIT_FAIL("exception must be thrown.");
  616. } catch(Exception& e) {
  617. std::cerr << e.stackTrace();
  618. }
  619. try {
  620. util::parseInt("");
  621. CPPUNIT_FAIL("exception must be thrown.");
  622. } catch(Exception& e) {
  623. std::cerr << e.stackTrace();
  624. }
  625. }
  626. void UtilTest::testParseUInt()
  627. {
  628. CPPUNIT_ASSERT_EQUAL(4294967295U, util::parseUInt(" 4294967295 "));
  629. try {
  630. util::parseUInt("-1");
  631. CPPUNIT_FAIL("exception must be thrown.");
  632. } catch(Exception& e) {
  633. std::cerr << e.stackTrace();
  634. }
  635. try {
  636. util::parseUInt("4294967296");
  637. CPPUNIT_FAIL("exception must be thrown.");
  638. } catch(Exception& e) {
  639. std::cerr << e.stackTrace();
  640. }
  641. }
  642. void UtilTest::testParseLLInt()
  643. {
  644. CPPUNIT_ASSERT_EQUAL((int64_t)-1LL, util::parseLLInt(" -1 "));
  645. CPPUNIT_ASSERT_EQUAL((int64_t)9223372036854775807LL,
  646. util::parseLLInt("9223372036854775807"));
  647. try {
  648. util::parseLLInt("9223372036854775808");
  649. CPPUNIT_FAIL("exception must be thrown.");
  650. } catch(Exception& e) {
  651. std::cerr << e.stackTrace();
  652. }
  653. try {
  654. util::parseLLInt("-9223372036854775809");
  655. CPPUNIT_FAIL("exception must be thrown.");
  656. } catch(Exception& e) {
  657. std::cerr << e.stackTrace();
  658. }
  659. try {
  660. util::parseLLInt("12x");
  661. CPPUNIT_FAIL("exception must be thrown.");
  662. } catch(Exception& e) {
  663. std::cerr << e.stackTrace();
  664. }
  665. try {
  666. util::parseLLInt("");
  667. CPPUNIT_FAIL("exception must be thrown.");
  668. } catch(Exception& e) {
  669. std::cerr << e.stackTrace();
  670. }
  671. }
  672. void UtilTest::testParseULLInt()
  673. {
  674. CPPUNIT_ASSERT_EQUAL((uint64_t)18446744073709551615ULL,
  675. util::parseULLInt("18446744073709551615"));
  676. try {
  677. util::parseUInt("-1");
  678. CPPUNIT_FAIL("exception must be thrown.");
  679. } catch(Exception& e) {
  680. std::cerr << e.stackTrace();
  681. }
  682. try {
  683. util::parseLLInt("18446744073709551616");
  684. CPPUNIT_FAIL("exception must be thrown.");
  685. } catch(Exception& e) {
  686. std::cerr << e.stackTrace();
  687. }
  688. }
  689. void UtilTest::testParseUIntNoThrow()
  690. {
  691. uint32_t n;
  692. CPPUNIT_ASSERT(util::parseUIntNoThrow(n, " 4294967295 "));
  693. CPPUNIT_ASSERT_EQUAL((uint32_t)UINT32_MAX, n);
  694. CPPUNIT_ASSERT(!util::parseUIntNoThrow(n, "4294967296"));
  695. CPPUNIT_ASSERT(!util::parseUIntNoThrow(n, "-1"));
  696. }
  697. void UtilTest::testParseLLIntNoThrow()
  698. {
  699. int64_t n;
  700. CPPUNIT_ASSERT(util::parseLLIntNoThrow(n, " 9223372036854775807 "));
  701. CPPUNIT_ASSERT_EQUAL((int64_t)INT64_MAX, n);
  702. CPPUNIT_ASSERT(!util::parseLLIntNoThrow(n, "9223372036854775808"));
  703. CPPUNIT_ASSERT(util::parseLLIntNoThrow(n, "-9223372036854775808"));
  704. CPPUNIT_ASSERT_EQUAL((int64_t)INT64_MIN, n);
  705. CPPUNIT_ASSERT(!util::parseLLIntNoThrow(n, "-9223372036854775809"));
  706. }
  707. void UtilTest::testToString_binaryStream()
  708. {
  709. SharedHandle<DiskWriter> dw(new ByteArrayDiskWriter());
  710. std::string data(16*1024+256, 'a');
  711. dw->initAndOpenFile();
  712. dw->writeData((const unsigned char*)data.c_str(), data.size(), 0);
  713. std::string readData = util::toString(dw);
  714. CPPUNIT_ASSERT_EQUAL(data, readData);
  715. }
  716. void UtilTest::testItos()
  717. {
  718. {
  719. int i = 0;
  720. CPPUNIT_ASSERT_EQUAL(std::string("0"), util::itos(i));
  721. }
  722. {
  723. int i = 100;
  724. CPPUNIT_ASSERT_EQUAL(std::string("100"), util::itos(i, true));
  725. }
  726. {
  727. int i = 100;
  728. CPPUNIT_ASSERT_EQUAL(std::string("100"), util::itos(i));
  729. }
  730. {
  731. int i = 12345;
  732. CPPUNIT_ASSERT_EQUAL(std::string("12,345"), util::itos(i, true));
  733. }
  734. {
  735. int i = 12345;
  736. CPPUNIT_ASSERT_EQUAL(std::string("12345"), util::itos(i));
  737. }
  738. {
  739. int i = -12345;
  740. CPPUNIT_ASSERT_EQUAL(std::string("-12,345"), util::itos(i, true));
  741. }
  742. {
  743. int64_t i = INT64_MAX;
  744. CPPUNIT_ASSERT_EQUAL(std::string("9,223,372,036,854,775,807"),
  745. util::itos(i, true));
  746. }
  747. {
  748. int64_t i = INT64_MIN;
  749. CPPUNIT_ASSERT_EQUAL(std::string("-9,223,372,036,854,775,808"),
  750. util::itos(i, true));
  751. }
  752. }
  753. void UtilTest::testUitos()
  754. {
  755. {
  756. uint16_t i = 12345;
  757. CPPUNIT_ASSERT_EQUAL(std::string("12345"), util::uitos(i));
  758. }
  759. {
  760. int16_t i = -12345;
  761. CPPUNIT_ASSERT_EQUAL(std::string("/.-,+"), util::uitos(i));
  762. }
  763. }
  764. void UtilTest::testNtoh64()
  765. {
  766. uint64_t x = 0xff00ff00ee00ee00LL;
  767. #ifdef WORDS_BIGENDIAN
  768. CPPUNIT_ASSERT_EQUAL(x, ntoh64(x));
  769. CPPUNIT_ASSERT_EQUAL(x, hton64(x));
  770. #else // !WORDS_BIGENDIAN
  771. uint64_t y = 0x00ee00ee00ff00ffLL;
  772. CPPUNIT_ASSERT_EQUAL(y, ntoh64(x));
  773. CPPUNIT_ASSERT_EQUAL(x, hton64(y));
  774. #endif // !WORDS_BIGENDIAN
  775. }
  776. void UtilTest::testPercentEncode()
  777. {
  778. CPPUNIT_ASSERT_EQUAL
  779. (std::string("%3A%2F%3F%23%5B%5D%40%21%25%26%27%28%29%2A%2B%2C%3B%3D"),
  780. util::percentEncode(":/?#[]@!%&'()*+,;="));
  781. std::string unreserved =
  782. "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
  783. "abcdefghijklmnopqrstuvwxyz"
  784. "0123456789"
  785. "-._~";
  786. CPPUNIT_ASSERT_EQUAL(unreserved, util::percentEncode(unreserved));
  787. CPPUNIT_ASSERT_EQUAL(std::string("1%5EA%20"), util::percentEncode("1^A "));
  788. }
  789. void UtilTest::testHtmlEscape()
  790. {
  791. CPPUNIT_ASSERT_EQUAL(std::string("aria2&lt;&gt;&quot;&#39;util"),
  792. util::htmlEscape("aria2<>\"'util"));
  793. }
  794. void UtilTest::testJoinPath()
  795. {
  796. const std::string dir1dir2file[] = { "dir1", "dir2", "file" };
  797. CPPUNIT_ASSERT_EQUAL
  798. (std::string("dir1/dir2/file"),
  799. util::joinPath(vbegin(dir1dir2file), vend(dir1dir2file)));
  800. const std::string dirparentfile[] = { "dir", "..", "file" };
  801. CPPUNIT_ASSERT_EQUAL
  802. (std::string("file"),
  803. util::joinPath(vbegin(dirparentfile), vend(dirparentfile)));
  804. const std::string dirparentparentfile[] = { "dir", "..", "..", "file" };
  805. CPPUNIT_ASSERT_EQUAL
  806. (std::string("file"),
  807. util::joinPath(vbegin(dirparentparentfile), vend(dirparentparentfile)));
  808. const std::string dirdotfile[] = { "dir", ".", "file" };
  809. CPPUNIT_ASSERT_EQUAL(std::string("dir/file"),
  810. util::joinPath(vbegin(dirdotfile), vend(dirdotfile)));
  811. const std::string empty[] = {};
  812. CPPUNIT_ASSERT_EQUAL(std::string(""), util::joinPath(&empty[0], &empty[0]));
  813. const std::string parentdot[] = { "..", "." };
  814. CPPUNIT_ASSERT_EQUAL(std::string(""),
  815. util::joinPath(vbegin(parentdot), vend(parentdot)));
  816. }
  817. void UtilTest::testParseIndexPath()
  818. {
  819. std::map<size_t, std::string>::value_type p = util::parseIndexPath("1=foo");
  820. CPPUNIT_ASSERT_EQUAL((size_t)1, p.first);
  821. CPPUNIT_ASSERT_EQUAL(std::string("foo"), p.second);
  822. try {
  823. util::parseIndexPath("1X=foo");
  824. CPPUNIT_FAIL("exception must be thrown.");
  825. } catch(Exception& e) {
  826. // success
  827. }
  828. try {
  829. util::parseIndexPath("1=");
  830. CPPUNIT_FAIL("exception must be thrown.");
  831. } catch(Exception& e) {
  832. // success
  833. }
  834. }
  835. void UtilTest::testCreateIndexPathMap()
  836. {
  837. std::stringstream in
  838. ("1=/tmp/myfile\n"
  839. "100=/myhome/mypicture.png\n");
  840. std::map<size_t, std::string> m = util::createIndexPathMap(in);
  841. CPPUNIT_ASSERT_EQUAL((size_t)2, m.size());
  842. CPPUNIT_ASSERT(m.find(1) != m.end());
  843. CPPUNIT_ASSERT_EQUAL(std::string("/tmp/myfile"), m[1]);
  844. CPPUNIT_ASSERT(m.find(100) != m.end());
  845. CPPUNIT_ASSERT_EQUAL(std::string("/myhome/mypicture.png"), m[100]);
  846. }
  847. void UtilTest::testGenerateRandomData()
  848. {
  849. unsigned char data1[20];
  850. util::generateRandomData(data1, sizeof(data1));
  851. unsigned char data2[20];
  852. util::generateRandomData(data2, sizeof(data2));
  853. CPPUNIT_ASSERT(memcmp(data1, data2, sizeof(data1)) != 0);
  854. }
  855. void UtilTest::testFromHex()
  856. {
  857. std::string src;
  858. std::string dest;
  859. src = "0011fF";
  860. dest = util::fromHex(src);
  861. CPPUNIT_ASSERT_EQUAL((size_t)3, dest.size());
  862. CPPUNIT_ASSERT_EQUAL((char)0x00, dest[0]);
  863. CPPUNIT_ASSERT_EQUAL((char)0x11, dest[1]);
  864. CPPUNIT_ASSERT_EQUAL((char)0xff, dest[2]);
  865. src = "0011f";
  866. CPPUNIT_ASSERT(util::fromHex(src).empty());
  867. src = "001g";
  868. CPPUNIT_ASSERT(util::fromHex(src).empty());
  869. }
  870. void UtilTest::testParsePrioritizePieceRange()
  871. {
  872. // piece index
  873. // 0 1 2 3 4 5 6 7
  874. // | | | |
  875. // file1 | | |
  876. // | | |
  877. // file2 | |
  878. // file3 |
  879. // | |
  880. // file4 |
  881. size_t pieceLength = 1024;
  882. std::vector<SharedHandle<FileEntry> > entries(4, SharedHandle<FileEntry>());
  883. entries[0].reset(new FileEntry("file1", 1024, 0));
  884. entries[1].reset(new FileEntry("file2",2560,entries[0]->getLastOffset()));
  885. entries[2].reset(new FileEntry("file3",0,entries[1]->getLastOffset()));
  886. entries[3].reset(new FileEntry("file4",3584,entries[2]->getLastOffset()));
  887. std::vector<size_t> result;
  888. util::parsePrioritizePieceRange(result, "head=1", entries, pieceLength);
  889. CPPUNIT_ASSERT_EQUAL((size_t)3, result.size());
  890. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  891. CPPUNIT_ASSERT_EQUAL((size_t)1, result[1]);
  892. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  893. result.clear();
  894. util::parsePrioritizePieceRange(result, "tail=1", entries, pieceLength);
  895. CPPUNIT_ASSERT_EQUAL((size_t)3, result.size());
  896. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  897. CPPUNIT_ASSERT_EQUAL((size_t)3, result[1]);
  898. CPPUNIT_ASSERT_EQUAL((size_t)6, result[2]);
  899. result.clear();
  900. util::parsePrioritizePieceRange(result, "head=1K", entries, pieceLength);
  901. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  902. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  903. CPPUNIT_ASSERT_EQUAL((size_t)1, result[1]);
  904. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  905. CPPUNIT_ASSERT_EQUAL((size_t)4, result[3]);
  906. result.clear();
  907. util::parsePrioritizePieceRange(result, "head", entries, pieceLength, 1024);
  908. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  909. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  910. CPPUNIT_ASSERT_EQUAL((size_t)1, result[1]);
  911. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  912. CPPUNIT_ASSERT_EQUAL((size_t)4, result[3]);
  913. result.clear();
  914. util::parsePrioritizePieceRange(result, "tail=1K", entries, pieceLength);
  915. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  916. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  917. CPPUNIT_ASSERT_EQUAL((size_t)2, result[1]);
  918. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  919. CPPUNIT_ASSERT_EQUAL((size_t)6, result[3]);
  920. result.clear();
  921. util::parsePrioritizePieceRange(result, "tail", entries, pieceLength, 1024);
  922. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  923. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  924. CPPUNIT_ASSERT_EQUAL((size_t)2, result[1]);
  925. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  926. CPPUNIT_ASSERT_EQUAL((size_t)6, result[3]);
  927. result.clear();
  928. util::parsePrioritizePieceRange(result, "head=1,tail=1", entries, pieceLength);
  929. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  930. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  931. CPPUNIT_ASSERT_EQUAL((size_t)1, result[1]);
  932. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  933. CPPUNIT_ASSERT_EQUAL((size_t)6, result[3]);
  934. result.clear();
  935. util::parsePrioritizePieceRange(result, "head=300M,tail=300M",
  936. entries, pieceLength);
  937. CPPUNIT_ASSERT_EQUAL((size_t)7, result.size());
  938. for(size_t i = 0; i < 7; ++i) {
  939. CPPUNIT_ASSERT_EQUAL(i, result[i]);
  940. }
  941. result.clear();
  942. util::parsePrioritizePieceRange(result, "", entries, pieceLength);
  943. CPPUNIT_ASSERT(result.empty());
  944. }
  945. void UtilTest::testApplyDir()
  946. {
  947. CPPUNIT_ASSERT_EQUAL(std::string("./pred"), util::applyDir("", "pred"));
  948. CPPUNIT_ASSERT_EQUAL(std::string("/pred"), util::applyDir("/", "pred"));
  949. CPPUNIT_ASSERT_EQUAL(std::string("./pred"), util::applyDir(".", "pred"));
  950. CPPUNIT_ASSERT_EQUAL(std::string("/dl/pred"), util::applyDir("/dl", "pred"));
  951. }
  952. void UtilTest::testFixTaintedBasename()
  953. {
  954. CPPUNIT_ASSERT_EQUAL(std::string("a%2Fb"), util::fixTaintedBasename("a/b"));
  955. #ifdef __MINGW32__
  956. CPPUNIT_ASSERT_EQUAL(std::string("a%5Cb"), util::fixTaintedBasename("a\\b"));
  957. #else // !__MINGW32__
  958. CPPUNIT_ASSERT_EQUAL(std::string("a\\b"), util::fixTaintedBasename("a\\b"));
  959. #endif // !__MINGW32__
  960. }
  961. void UtilTest::testIsNumericHost()
  962. {
  963. CPPUNIT_ASSERT(util::isNumericHost("192.168.0.1"));
  964. CPPUNIT_ASSERT(!util::isNumericHost("aria2.sf.net"));
  965. CPPUNIT_ASSERT(util::isNumericHost("::1"));
  966. }
  967. void UtilTest::testDetectDirTraversal()
  968. {
  969. CPPUNIT_ASSERT(util::detectDirTraversal("/foo"));
  970. CPPUNIT_ASSERT(util::detectDirTraversal("./foo"));
  971. CPPUNIT_ASSERT(util::detectDirTraversal("../foo"));
  972. CPPUNIT_ASSERT(util::detectDirTraversal("foo/../bar"));
  973. CPPUNIT_ASSERT(util::detectDirTraversal("foo/./bar"));
  974. CPPUNIT_ASSERT(util::detectDirTraversal("foo/."));
  975. CPPUNIT_ASSERT(util::detectDirTraversal("foo/.."));
  976. CPPUNIT_ASSERT(util::detectDirTraversal("."));
  977. CPPUNIT_ASSERT(util::detectDirTraversal(".."));
  978. CPPUNIT_ASSERT(util::detectDirTraversal("/"));
  979. CPPUNIT_ASSERT(util::detectDirTraversal("foo/"));
  980. CPPUNIT_ASSERT(util::detectDirTraversal("\t"));
  981. CPPUNIT_ASSERT(!util::detectDirTraversal("foo/bar"));
  982. CPPUNIT_ASSERT(!util::detectDirTraversal("foo"));
  983. }
  984. void UtilTest::testEscapePath()
  985. {
  986. CPPUNIT_ASSERT_EQUAL(std::string("foo%00bar%00%01"),
  987. util::escapePath(std::string("foo")+(char)0x00+
  988. std::string("bar")+(char)0x00+
  989. (char)0x01));
  990. #ifdef __MINGW32__
  991. CPPUNIT_ASSERT_EQUAL(std::string("foo%5Cbar"), util::escapePath("foo\\bar"));
  992. #else // !__MINGW32__
  993. CPPUNIT_ASSERT_EQUAL(std::string("foo\\bar"), util::escapePath("foo\\bar"));
  994. #endif // !__MINGW32__
  995. }
  996. void UtilTest::testGetCidrPrefix()
  997. {
  998. struct in_addr in;
  999. CPPUNIT_ASSERT(util::getCidrPrefix(in, "192.168.0.1", 16));
  1000. CPPUNIT_ASSERT_EQUAL(std::string("192.168.0.0"), std::string(inet_ntoa(in)));
  1001. CPPUNIT_ASSERT(util::getCidrPrefix(in, "192.168.255.255", 17));
  1002. CPPUNIT_ASSERT_EQUAL(std::string("192.168.128.0"),std::string(inet_ntoa(in)));
  1003. CPPUNIT_ASSERT(util::getCidrPrefix(in, "192.168.128.1", 16));
  1004. CPPUNIT_ASSERT_EQUAL(std::string("192.168.0.0"), std::string(inet_ntoa(in)));
  1005. CPPUNIT_ASSERT(util::getCidrPrefix(in, "192.168.0.1", 32));
  1006. CPPUNIT_ASSERT_EQUAL(std::string("192.168.0.1"), std::string(inet_ntoa(in)));
  1007. CPPUNIT_ASSERT(util::getCidrPrefix(in, "192.168.0.1", 0));
  1008. CPPUNIT_ASSERT_EQUAL(std::string("0.0.0.0"), std::string(inet_ntoa(in)));
  1009. CPPUNIT_ASSERT(util::getCidrPrefix(in, "10.10.1.44", 27));
  1010. CPPUNIT_ASSERT_EQUAL(std::string("10.10.1.32"), std::string(inet_ntoa(in)));
  1011. CPPUNIT_ASSERT(!util::getCidrPrefix(in, "::1", 32));
  1012. }
  1013. void UtilTest::testInSameCidrBlock()
  1014. {
  1015. CPPUNIT_ASSERT(util::inSameCidrBlock("192.168.128.1", "192.168.0.1", 16));
  1016. CPPUNIT_ASSERT(!util::inSameCidrBlock("192.168.128.1", "192.168.0.1", 17));
  1017. }
  1018. void UtilTest::testIsUtf8String()
  1019. {
  1020. CPPUNIT_ASSERT(util::isUtf8("ascii"));
  1021. // "Hello World" in Japanese UTF-8
  1022. CPPUNIT_ASSERT(util::isUtf8
  1023. (util::fromHex("e38193e38293e381abe381a1e381afe4b896e7958c")));
  1024. // "World" in Shift_JIS
  1025. CPPUNIT_ASSERT(!util::isUtf8(util::fromHex("90a28a")+"E"));
  1026. // UTF8-2
  1027. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("c280")));
  1028. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("dfbf")));
  1029. // UTF8-3
  1030. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("e0a080")));
  1031. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("e0bf80")));
  1032. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("e18080")));
  1033. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("ec8080")));
  1034. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("ed8080")));
  1035. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("ed9f80")));
  1036. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("ee8080")));
  1037. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("ef8080")));
  1038. // UTF8-4
  1039. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("f0908080")));
  1040. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("f0bf8080")));
  1041. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("f1808080")));
  1042. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("f3808080")));
  1043. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("f4808080")));
  1044. CPPUNIT_ASSERT(util::isUtf8(util::fromHex("f48f8080")));
  1045. CPPUNIT_ASSERT(util::isUtf8(""));
  1046. CPPUNIT_ASSERT(!util::isUtf8(util::fromHex("00")));
  1047. }
  1048. } // namespace aria2