UtilTest.cc 37 KB

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