UtilTest.cc 40 KB

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