UtilTest.cc 60 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774
  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. #include "BufferedFile.h"
  14. #include "TestUtil.h"
  15. #include "SocketCore.h"
  16. namespace aria2 {
  17. class UtilTest:public CppUnit::TestFixture {
  18. CPPUNIT_TEST_SUITE(UtilTest);
  19. CPPUNIT_TEST(testStrip);
  20. CPPUNIT_TEST(testStripIter);
  21. CPPUNIT_TEST(testLstripIter);
  22. CPPUNIT_TEST(testLstripIter_char);
  23. CPPUNIT_TEST(testDivide);
  24. CPPUNIT_TEST(testSplit);
  25. CPPUNIT_TEST(testSplitIter);
  26. CPPUNIT_TEST(testSplitIterM);
  27. CPPUNIT_TEST(testStreq);
  28. CPPUNIT_TEST(testStrieq);
  29. CPPUNIT_TEST(testStrifind);
  30. CPPUNIT_TEST(testEndsWith);
  31. CPPUNIT_TEST(testIendsWith);
  32. CPPUNIT_TEST(testReplace);
  33. CPPUNIT_TEST(testStartsWith);
  34. CPPUNIT_TEST(testIstartsWith);
  35. // may be moved to other helper class in the future.
  36. CPPUNIT_TEST(testGetContentDispositionFilename);
  37. CPPUNIT_TEST(testRandomAlpha);
  38. CPPUNIT_TEST(testToUpper);
  39. CPPUNIT_TEST(testToLower);
  40. CPPUNIT_TEST(testUppercase);
  41. CPPUNIT_TEST(testLowercase);
  42. CPPUNIT_TEST(testPercentDecode);
  43. CPPUNIT_TEST(testGetRealSize);
  44. CPPUNIT_TEST(testAbbrevSize);
  45. CPPUNIT_TEST(testToStream);
  46. CPPUNIT_TEST(testIsNumber);
  47. CPPUNIT_TEST(testIsLowercase);
  48. CPPUNIT_TEST(testIsUppercase);
  49. CPPUNIT_TEST(testAlphaToNum);
  50. CPPUNIT_TEST(testMkdirs);
  51. CPPUNIT_TEST(testConvertBitfield);
  52. CPPUNIT_TEST(testParseIntSegments);
  53. CPPUNIT_TEST(testParseIntSegments_invalidRange);
  54. CPPUNIT_TEST(testParseIntNoThrow);
  55. CPPUNIT_TEST(testParseUIntNoThrow);
  56. CPPUNIT_TEST(testParseLLIntNoThrow);
  57. CPPUNIT_TEST(testToString_binaryStream);
  58. CPPUNIT_TEST(testItos);
  59. CPPUNIT_TEST(testUitos);
  60. CPPUNIT_TEST(testNtoh64);
  61. CPPUNIT_TEST(testPercentEncode);
  62. CPPUNIT_TEST(testHtmlEscape);
  63. CPPUNIT_TEST(testJoinPath);
  64. CPPUNIT_TEST(testParseIndexPath);
  65. CPPUNIT_TEST(testCreateIndexPaths);
  66. CPPUNIT_TEST(testGenerateRandomData);
  67. CPPUNIT_TEST(testFromHex);
  68. CPPUNIT_TEST(testParsePrioritizePieceRange);
  69. CPPUNIT_TEST(testApplyDir);
  70. CPPUNIT_TEST(testFixTaintedBasename);
  71. CPPUNIT_TEST(testIsNumericHost);
  72. CPPUNIT_TEST(testDetectDirTraversal);
  73. CPPUNIT_TEST(testEscapePath);
  74. CPPUNIT_TEST(testInSameCidrBlock);
  75. CPPUNIT_TEST(testIsUtf8String);
  76. CPPUNIT_TEST(testNextParam);
  77. CPPUNIT_TEST(testNoProxyDomainMatch);
  78. CPPUNIT_TEST(testInPrivateAddress);
  79. CPPUNIT_TEST(testSecfmt);
  80. CPPUNIT_TEST(testTlsHostnameMatch);
  81. CPPUNIT_TEST_SUITE_END();
  82. private:
  83. public:
  84. void setUp() {
  85. }
  86. void testStrip();
  87. void testStripIter();
  88. void testLstripIter();
  89. void testLstripIter_char();
  90. void testDivide();
  91. void testSplit();
  92. void testSplitIter();
  93. void testSplitIterM();
  94. void testStreq();
  95. void testStrieq();
  96. void testStrifind();
  97. void testEndsWith();
  98. void testIendsWith();
  99. void testReplace();
  100. void testStartsWith();
  101. void testIstartsWith();
  102. // may be moved to other helper class in the future.
  103. void testGetContentDispositionFilename();
  104. void testRandomAlpha();
  105. void testToUpper();
  106. void testToLower();
  107. void testUppercase();
  108. void testLowercase();
  109. void testPercentDecode();
  110. void testGetRealSize();
  111. void testAbbrevSize();
  112. void testToStream();
  113. void testIsNumber();
  114. void testIsLowercase();
  115. void testIsUppercase();
  116. void testAlphaToNum();
  117. void testMkdirs();
  118. void testConvertBitfield();
  119. void testParseIntSegments();
  120. void testParseIntSegments_invalidRange();
  121. void testParseIntNoThrow();
  122. void testParseUIntNoThrow();
  123. void testParseLLIntNoThrow();
  124. void testToString_binaryStream();
  125. void testItos();
  126. void testUitos();
  127. void testNtoh64();
  128. void testPercentEncode();
  129. void testHtmlEscape();
  130. void testJoinPath();
  131. void testParseIndexPath();
  132. void testCreateIndexPaths();
  133. void testGenerateRandomData();
  134. void testFromHex();
  135. void testParsePrioritizePieceRange();
  136. void testApplyDir();
  137. void testFixTaintedBasename();
  138. void testIsNumericHost();
  139. void testDetectDirTraversal();
  140. void testEscapePath();
  141. void testInSameCidrBlock();
  142. void testIsUtf8String();
  143. void testNextParam();
  144. void testNoProxyDomainMatch();
  145. void testInPrivateAddress();
  146. void testSecfmt();
  147. void testTlsHostnameMatch();
  148. };
  149. CPPUNIT_TEST_SUITE_REGISTRATION( UtilTest );
  150. void UtilTest::testStrip()
  151. {
  152. std::string str1 = "aria2";
  153. CPPUNIT_ASSERT_EQUAL(str1, util::strip("aria2"));
  154. CPPUNIT_ASSERT_EQUAL(str1, util::strip(" aria2"));
  155. CPPUNIT_ASSERT_EQUAL(str1, util::strip("aria2 "));
  156. CPPUNIT_ASSERT_EQUAL(str1, util::strip(" aria2 "));
  157. CPPUNIT_ASSERT_EQUAL(str1, util::strip(" aria2 "));
  158. std::string str2 = "aria2 debut";
  159. CPPUNIT_ASSERT_EQUAL(str2, util::strip("aria2 debut"));
  160. CPPUNIT_ASSERT_EQUAL(str2, util::strip(" aria2 debut "));
  161. std::string str3 = "";
  162. CPPUNIT_ASSERT_EQUAL(str3, util::strip(""));
  163. CPPUNIT_ASSERT_EQUAL(str3, util::strip(" "));
  164. CPPUNIT_ASSERT_EQUAL(str3, util::strip(" "));
  165. std::string str4 = "A";
  166. CPPUNIT_ASSERT_EQUAL(str4, util::strip("A"));
  167. CPPUNIT_ASSERT_EQUAL(str4, util::strip(" A "));
  168. CPPUNIT_ASSERT_EQUAL(str4, util::strip(" A "));
  169. }
  170. void UtilTest::testStripIter()
  171. {
  172. Scip p;
  173. std::string str1 = "aria2";
  174. std::string s = "aria2";
  175. p = util::stripIter(s.begin(), s.end());
  176. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  177. s = " aria2";
  178. p = util::stripIter(s.begin(), s.end());
  179. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  180. s = "aria2 ";
  181. p = util::stripIter(s.begin(), s.end());
  182. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  183. s = " aria2 ";
  184. p = util::stripIter(s.begin(), s.end());
  185. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  186. s = " aria2 ";
  187. p = util::stripIter(s.begin(), s.end());
  188. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  189. std::string str2 = "aria2 debut";
  190. s = "aria2 debut";
  191. p = util::stripIter(s.begin(), s.end());
  192. CPPUNIT_ASSERT_EQUAL(str2, std::string(p.first, p.second));
  193. s = " aria2 debut ";
  194. p = util::stripIter(s.begin(), s.end());
  195. CPPUNIT_ASSERT_EQUAL(str2, std::string(p.first, p.second));
  196. std::string str3 = "";
  197. s = "";
  198. p = util::stripIter(s.begin(), s.end());
  199. CPPUNIT_ASSERT_EQUAL(str3, std::string(p.first, p.second));
  200. s = " ";
  201. p = util::stripIter(s.begin(), s.end());
  202. CPPUNIT_ASSERT_EQUAL(str3, std::string(p.first, p.second));
  203. s = " ";
  204. p = util::stripIter(s.begin(), s.end());
  205. CPPUNIT_ASSERT_EQUAL(str3, std::string(p.first, p.second));
  206. std::string str4 = "A";
  207. s = "A";
  208. p = util::stripIter(s.begin(), s.end());
  209. CPPUNIT_ASSERT_EQUAL(str4, std::string(p.first, p.second));
  210. s = " A ";
  211. p = util::stripIter(s.begin(), s.end());
  212. CPPUNIT_ASSERT_EQUAL(str4, std::string(p.first, p.second));
  213. s = " A ";
  214. p = util::stripIter(s.begin(), s.end());
  215. CPPUNIT_ASSERT_EQUAL(str4, std::string(p.first, p.second));
  216. }
  217. void UtilTest::testLstripIter()
  218. {
  219. std::string::iterator r;
  220. std::string s = "foo";
  221. r = util::lstripIter(s.begin(), s.end());
  222. CPPUNIT_ASSERT_EQUAL(std::string("foo"), std::string(r, s.end()));
  223. s = " foo bar ";
  224. r = util::lstripIter(s.begin(), s.end());
  225. CPPUNIT_ASSERT_EQUAL(std::string("foo bar "), std::string(r, s.end()));
  226. s = "f";
  227. r = util::lstripIter(s.begin(), s.end());
  228. CPPUNIT_ASSERT_EQUAL(std::string("f"), std::string(r, s.end()));
  229. s = "foo ";
  230. r = util::lstripIter(s.begin(), s.end());
  231. CPPUNIT_ASSERT_EQUAL(std::string("foo "), std::string(r, s.end()));
  232. }
  233. void UtilTest::testLstripIter_char()
  234. {
  235. std::string::iterator r;
  236. std::string s = "foo";
  237. r = util::lstripIter(s.begin(), s.end(), '$');
  238. CPPUNIT_ASSERT_EQUAL(std::string("foo"), std::string(r, s.end()));
  239. s = "$$foo$bar$$";
  240. r = util::lstripIter(s.begin(), s.end(), '$');
  241. CPPUNIT_ASSERT_EQUAL(std::string("foo$bar$$"), std::string(r, s.end()));
  242. s = "f";
  243. r = util::lstripIter(s.begin(), s.end(), '$');
  244. CPPUNIT_ASSERT_EQUAL(std::string("f"), std::string(r, s.end()));
  245. s = "foo$$";
  246. r = util::lstripIter(s.begin(), s.end(), '$');
  247. CPPUNIT_ASSERT_EQUAL(std::string("foo$$"), std::string(r, s.end()));
  248. }
  249. void UtilTest::testDivide() {
  250. std::pair<Sip, Sip> p1;
  251. std::string s = "name=value";
  252. util::divide(p1, s.begin(), s.end(), '=');
  253. CPPUNIT_ASSERT_EQUAL(std::string("name"),
  254. std::string(p1.first.first, p1.first.second));
  255. CPPUNIT_ASSERT_EQUAL(std::string("value"),
  256. std::string(p1.second.first, p1.second.second));
  257. s = " name = value ";
  258. util::divide(p1, s.begin(), s.end(), '=');
  259. CPPUNIT_ASSERT_EQUAL(std::string("name"),
  260. std::string(p1.first.first, p1.first.second));
  261. CPPUNIT_ASSERT_EQUAL(std::string("value"),
  262. std::string(p1.second.first, p1.second.second));
  263. s = "=value";
  264. util::divide(p1, s.begin(), s.end(), '=');
  265. CPPUNIT_ASSERT_EQUAL(std::string(""),
  266. std::string(p1.first.first, p1.first.second));
  267. CPPUNIT_ASSERT_EQUAL(std::string("value"),
  268. std::string(p1.second.first, p1.second.second));
  269. s = "name=";
  270. util::divide(p1, s.begin(), s.end(), '=');
  271. CPPUNIT_ASSERT_EQUAL(std::string("name"),
  272. std::string(p1.first.first, p1.first.second));
  273. CPPUNIT_ASSERT_EQUAL(std::string(""),
  274. std::string(p1.second.first, p1.second.second));
  275. s = "name";
  276. util::divide(p1, s.begin(), s.end(), '=');
  277. CPPUNIT_ASSERT_EQUAL(std::string("name"),
  278. std::string(p1.first.first, p1.first.second));
  279. CPPUNIT_ASSERT_EQUAL(std::string(""),
  280. std::string(p1.second.first, p1.second.second));
  281. }
  282. void UtilTest::testSplit() {
  283. std::vector<std::string> v;
  284. std::string s = "k1; k2;; k3";
  285. util::split(s.begin(), s.end(), std::back_inserter(v), ';', true);
  286. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  287. std::vector<std::string>::iterator itr = v.begin();
  288. CPPUNIT_ASSERT_EQUAL(std::string("k1"), *itr++);
  289. CPPUNIT_ASSERT_EQUAL(std::string("k2"), *itr++);
  290. CPPUNIT_ASSERT_EQUAL(std::string("k3"), *itr++);
  291. v.clear();
  292. s = "k1; k2; k3";
  293. util::split(s.begin(), s.end(), std::back_inserter(v), ';');
  294. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  295. itr = v.begin();
  296. CPPUNIT_ASSERT_EQUAL(std::string("k1"), *itr++);
  297. CPPUNIT_ASSERT_EQUAL(std::string(" k2"), *itr++);
  298. CPPUNIT_ASSERT_EQUAL(std::string(" k3"), *itr++);
  299. v.clear();
  300. s = "k=v";
  301. util::split(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  302. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  303. itr = v.begin();
  304. CPPUNIT_ASSERT_EQUAL(std::string("k=v"), *itr++);
  305. v.clear();
  306. s = ";;k1;;k2;";
  307. util::split(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  308. CPPUNIT_ASSERT_EQUAL((size_t)6, v.size());
  309. itr = v.begin();
  310. CPPUNIT_ASSERT_EQUAL(std::string(""), *itr++);
  311. CPPUNIT_ASSERT_EQUAL(std::string(""), *itr++);
  312. CPPUNIT_ASSERT_EQUAL(std::string("k1"), *itr++);
  313. CPPUNIT_ASSERT_EQUAL(std::string(""), *itr++);
  314. CPPUNIT_ASSERT_EQUAL(std::string("k2"), *itr++);
  315. CPPUNIT_ASSERT_EQUAL(std::string(""), *itr++);
  316. v.clear();
  317. s = ";;k1;;k2;";
  318. util::split(s.begin(), s.end(), std::back_inserter(v), ';');
  319. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  320. itr = v.begin();
  321. CPPUNIT_ASSERT_EQUAL(std::string("k1"), *itr++);
  322. CPPUNIT_ASSERT_EQUAL(std::string("k2"), *itr++);
  323. v.clear();
  324. s = "k; ";
  325. util::split(s.begin(), s.end(), std::back_inserter(v), ';');
  326. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  327. itr = v.begin();
  328. CPPUNIT_ASSERT_EQUAL(std::string("k"), *itr++);
  329. CPPUNIT_ASSERT_EQUAL(std::string(" "), *itr++);
  330. v.clear();
  331. s = " ";
  332. util::split(s.begin(), s.end(), std::back_inserter(v), ';', true, true);
  333. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  334. CPPUNIT_ASSERT_EQUAL(std::string(""), v[0]);
  335. v.clear();
  336. s = " ";
  337. util::split(s.begin(), s.end(), std::back_inserter(v), ';', true);
  338. CPPUNIT_ASSERT_EQUAL((size_t)0, v.size());
  339. v.clear();
  340. s = " ";
  341. util::split(s.begin(), s.end(), std::back_inserter(v), ';');
  342. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  343. CPPUNIT_ASSERT_EQUAL(std::string(" "), v[0]);
  344. v.clear();
  345. s = ";";
  346. util::split(s.begin(), s.end(), std::back_inserter(v), ';');
  347. CPPUNIT_ASSERT_EQUAL((size_t)0, v.size());
  348. v.clear();
  349. s = ";";
  350. util::split(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  351. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  352. itr = v.begin();
  353. CPPUNIT_ASSERT_EQUAL(std::string(""), *itr++);
  354. CPPUNIT_ASSERT_EQUAL(std::string(""), *itr++);
  355. v.clear();
  356. s = "";
  357. util::split(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  358. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  359. CPPUNIT_ASSERT_EQUAL(std::string(""), v[0]);
  360. }
  361. void UtilTest::testSplitIter() {
  362. std::vector<Scip> v;
  363. std::string s = "k1; k2;; k3";
  364. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', true);
  365. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  366. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  367. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[1].first, v[1].second));
  368. CPPUNIT_ASSERT_EQUAL(std::string("k3"), std::string(v[2].first, v[2].second));
  369. v.clear();
  370. s = "k1; k2; k3";
  371. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';');
  372. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  373. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  374. CPPUNIT_ASSERT_EQUAL(std::string(" k2"),
  375. std::string(v[1].first, v[1].second));
  376. CPPUNIT_ASSERT_EQUAL(std::string(" k3"),
  377. std::string(v[2].first, v[2].second));
  378. v.clear();
  379. s = "k=v";
  380. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  381. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  382. CPPUNIT_ASSERT_EQUAL(std::string("k=v"),
  383. std::string(v[0].first, v[0].second));
  384. v.clear();
  385. s = ";;k1;;k2;";
  386. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  387. CPPUNIT_ASSERT_EQUAL((size_t)6, v.size());
  388. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  389. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[1].first, v[1].second));
  390. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[2].first, v[2].second));
  391. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[3].first, v[3].second));
  392. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[4].first, v[4].second));
  393. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[5].first, v[5].second));
  394. v.clear();
  395. s = ";;k1;;k2;";
  396. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';');
  397. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  398. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  399. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[1].first, v[1].second));
  400. v.clear();
  401. s = "k; ";
  402. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';');
  403. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  404. CPPUNIT_ASSERT_EQUAL(std::string("k"), std::string(v[0].first, v[0].second));
  405. CPPUNIT_ASSERT_EQUAL(std::string(" "), std::string(v[1].first, v[1].second));
  406. v.clear();
  407. s = " ";
  408. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', true, true);
  409. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  410. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  411. v.clear();
  412. s = " ";
  413. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', true);
  414. CPPUNIT_ASSERT_EQUAL((size_t)0, v.size());
  415. v.clear();
  416. s = " ";
  417. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';');
  418. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  419. CPPUNIT_ASSERT_EQUAL(std::string(" "), std::string(v[0].first, v[0].second));
  420. v.clear();
  421. s = ";";
  422. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';');
  423. CPPUNIT_ASSERT_EQUAL((size_t)0, v.size());
  424. v.clear();
  425. s = ";";
  426. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  427. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  428. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  429. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[1].first, v[1].second));
  430. v.clear();
  431. s = "";
  432. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  433. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  434. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  435. }
  436. void UtilTest::testSplitIterM() {
  437. const char d[] = ";";
  438. const char md[] = "; ";
  439. std::vector<Scip> v;
  440. std::string s = "k1; k2;; k3";
  441. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, true);
  442. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  443. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  444. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[1].first, v[1].second));
  445. CPPUNIT_ASSERT_EQUAL(std::string("k3"), std::string(v[2].first, v[2].second));
  446. v.clear();
  447. s = "k1; k2; k3";
  448. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d);
  449. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  450. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  451. CPPUNIT_ASSERT_EQUAL(std::string(" k2"),
  452. std::string(v[1].first, v[1].second));
  453. CPPUNIT_ASSERT_EQUAL(std::string(" k3"),
  454. std::string(v[2].first, v[2].second));
  455. v.clear();
  456. s = "k1; k2; k3";
  457. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), md);
  458. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  459. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  460. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[1].first, v[1].second));
  461. CPPUNIT_ASSERT_EQUAL(std::string("k3"), std::string(v[2].first, v[2].second));
  462. v.clear();
  463. s = "k1; k2; k3;";
  464. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), md, false, true);
  465. CPPUNIT_ASSERT_EQUAL((size_t)6, v.size());
  466. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  467. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[1].first, v[1].second));
  468. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[2].first, v[2].second));
  469. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[3].first, v[3].second));
  470. CPPUNIT_ASSERT_EQUAL(std::string("k3"), std::string(v[4].first, v[4].second));
  471. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[5].first, v[5].second));
  472. v.clear();
  473. s = "k=v";
  474. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, false, true);
  475. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  476. CPPUNIT_ASSERT_EQUAL(std::string("k=v"),
  477. std::string(v[0].first, v[0].second));
  478. v.clear();
  479. s = ";;k1;;k2;";
  480. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, false, true);
  481. CPPUNIT_ASSERT_EQUAL((size_t)6, v.size());
  482. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  483. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[1].first, v[1].second));
  484. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[2].first, v[2].second));
  485. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[3].first, v[3].second));
  486. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[4].first, v[4].second));
  487. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[5].first, v[5].second));
  488. v.clear();
  489. s = ";;k1;;k2;";
  490. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d);
  491. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  492. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  493. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[1].first, v[1].second));
  494. v.clear();
  495. s = "k; ";
  496. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d);
  497. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  498. CPPUNIT_ASSERT_EQUAL(std::string("k"), std::string(v[0].first, v[0].second));
  499. CPPUNIT_ASSERT_EQUAL(std::string(" "), std::string(v[1].first, v[1].second));
  500. v.clear();
  501. s = " ";
  502. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, true, true);
  503. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  504. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  505. v.clear();
  506. s = " ";
  507. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, true);
  508. CPPUNIT_ASSERT_EQUAL((size_t)0, v.size());
  509. v.clear();
  510. s = " ";
  511. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d);
  512. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  513. CPPUNIT_ASSERT_EQUAL(std::string(" "), std::string(v[0].first, v[0].second));
  514. v.clear();
  515. s = ";";
  516. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d);
  517. CPPUNIT_ASSERT_EQUAL((size_t)0, v.size());
  518. v.clear();
  519. s = ";";
  520. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, false, true);
  521. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  522. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  523. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[1].first, v[1].second));
  524. v.clear();
  525. s = "";
  526. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, false, true);
  527. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  528. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  529. }
  530. void UtilTest::testEndsWith() {
  531. std::string target = "abcdefg";
  532. std::string part = "fg";
  533. CPPUNIT_ASSERT(util::endsWith(target.begin(), target.end(),
  534. part.begin(), part.end()));
  535. target = "abdefg";
  536. part = "g";
  537. CPPUNIT_ASSERT(util::endsWith(target.begin(), target.end(),
  538. part.begin(), part.end()));
  539. target = "abdefg";
  540. part = "eg";
  541. CPPUNIT_ASSERT(!util::endsWith(target.begin(), target.end(),
  542. part.begin(), part.end()));
  543. target = "g";
  544. part = "eg";
  545. CPPUNIT_ASSERT(!util::endsWith(target.begin(), target.end(),
  546. part.begin(), part.end()));
  547. target = "g";
  548. part = "g";
  549. CPPUNIT_ASSERT(util::endsWith(target.begin(), target.end(),
  550. part.begin(), part.end()));
  551. target = "g";
  552. part = "";
  553. CPPUNIT_ASSERT(util::endsWith(target.begin(), target.end(),
  554. part.begin(), part.end()));
  555. target = "";
  556. part = "";
  557. CPPUNIT_ASSERT(util::endsWith(target.begin(), target.end(),
  558. part.begin(), part.end()));
  559. target = "";
  560. part = "g";
  561. CPPUNIT_ASSERT(!util::endsWith(target.begin(), target.end(),
  562. part.begin(), part.end()));
  563. }
  564. void UtilTest::testIendsWith() {
  565. std::string target = "abcdefg";
  566. std::string part = "Fg";
  567. CPPUNIT_ASSERT(util::iendsWith(target.begin(), target.end(),
  568. part.begin(), part.end()));
  569. target = "abdefg";
  570. part = "ef";
  571. CPPUNIT_ASSERT(!util::iendsWith(target.begin(), target.end(),
  572. part.begin(), part.end()));
  573. }
  574. void UtilTest::testStreq()
  575. {
  576. std::string s1, s2;
  577. s1 = "foo";
  578. s2 = "foo";
  579. CPPUNIT_ASSERT(util::streq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  580. CPPUNIT_ASSERT(util::streq(s1.begin(), s1.end(), s2.c_str()));
  581. s2 = "fooo";
  582. CPPUNIT_ASSERT(!util::streq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  583. CPPUNIT_ASSERT(!util::streq(s1.begin(), s1.end(), s2.c_str()));
  584. s2 = "fo";
  585. CPPUNIT_ASSERT(!util::streq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  586. CPPUNIT_ASSERT(!util::streq(s1.begin(), s1.end(), s2.c_str()));
  587. s2 = "";
  588. CPPUNIT_ASSERT(!util::streq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  589. CPPUNIT_ASSERT(!util::streq(s1.begin(), s1.end(), s2.c_str()));
  590. s1 = "";
  591. CPPUNIT_ASSERT(util::streq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  592. CPPUNIT_ASSERT(util::streq(s1.begin(), s1.end(), s2.c_str()));
  593. }
  594. void UtilTest::testStrieq()
  595. {
  596. std::string s1, s2;
  597. s1 = "foo";
  598. s2 = "foo";
  599. CPPUNIT_ASSERT(util::strieq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  600. CPPUNIT_ASSERT(util::strieq(s1.begin(), s1.end(), s2.c_str()));
  601. s1 = "FoO";
  602. s2 = "fOo";
  603. CPPUNIT_ASSERT(util::strieq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  604. CPPUNIT_ASSERT(util::strieq(s1.begin(), s1.end(), s2.c_str()));
  605. s2 = "fooo";
  606. CPPUNIT_ASSERT(!util::strieq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  607. CPPUNIT_ASSERT(!util::strieq(s1.begin(), s1.end(), s2.c_str()));
  608. s2 = "fo";
  609. CPPUNIT_ASSERT(!util::strieq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  610. CPPUNIT_ASSERT(!util::strieq(s1.begin(), s1.end(), s2.c_str()));
  611. s2 = "";
  612. CPPUNIT_ASSERT(!util::strieq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  613. CPPUNIT_ASSERT(!util::strieq(s1.begin(), s1.end(), s2.c_str()));
  614. s1 = "";
  615. CPPUNIT_ASSERT(util::strieq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  616. CPPUNIT_ASSERT(util::strieq(s1.begin(), s1.end(), s2.c_str()));
  617. }
  618. void UtilTest::testStrifind()
  619. {
  620. std::string s1, s2;
  621. s1 = "yamagakani mukashi wo toheba hARU no tuki";
  622. s2 = "HaRu";
  623. CPPUNIT_ASSERT(util::strifind(s1.begin(), s1.end(), s2.begin(), s2.end())
  624. != s1.end());
  625. s2 = "aki";
  626. CPPUNIT_ASSERT(util::strifind(s1.begin(), s1.end(), s2.begin(), s2.end())
  627. == s1.end());
  628. s1 = "h";
  629. s2 = "HH";
  630. CPPUNIT_ASSERT(util::strifind(s1.begin(), s1.end(), s2.begin(), s2.end())
  631. == s1.end());
  632. }
  633. void UtilTest::testReplace() {
  634. CPPUNIT_ASSERT_EQUAL(std::string("abc\n"), util::replace("abc\r\n", "\r", ""));
  635. CPPUNIT_ASSERT_EQUAL(std::string("abc"), util::replace("abc\r\n", "\r\n", ""));
  636. CPPUNIT_ASSERT_EQUAL(std::string(""), util::replace("", "\r\n", ""));
  637. CPPUNIT_ASSERT_EQUAL(std::string("abc"), util::replace("abc", "", "a"));
  638. CPPUNIT_ASSERT_EQUAL(std::string("xbc"), util::replace("abc", "a", "x"));
  639. }
  640. void UtilTest::testStartsWith() {
  641. std::string target;
  642. std::string part;
  643. target = "abcdefg";
  644. part = "abc";
  645. CPPUNIT_ASSERT(util::startsWith(target.begin(), target.end(),
  646. part.begin(), part.end()));
  647. CPPUNIT_ASSERT(util::startsWith(target.begin(), target.end(), part.c_str()));
  648. target = "abcdefg";
  649. part = "abx";
  650. CPPUNIT_ASSERT(!util::startsWith(target.begin(), target.end(),
  651. part.begin(), part.end()));
  652. CPPUNIT_ASSERT(!util::startsWith(target.begin(), target.end(), part.c_str()));
  653. target = "abcdefg";
  654. part = "bcd";
  655. CPPUNIT_ASSERT(!util::startsWith(target.begin(), target.end(),
  656. part.begin(), part.end()));
  657. CPPUNIT_ASSERT(!util::startsWith(target.begin(), target.end(), part.c_str()));
  658. target = "";
  659. part = "a";
  660. CPPUNIT_ASSERT(!util::startsWith(target.begin(), target.end(),
  661. part.begin(), part.end()));
  662. CPPUNIT_ASSERT(!util::startsWith(target.begin(), target.end(), part.c_str()));
  663. target = "";
  664. part = "";
  665. CPPUNIT_ASSERT(util::startsWith(target.begin(), target.end(),
  666. part.begin(), part.end()));
  667. CPPUNIT_ASSERT(util::startsWith(target.begin(), target.end(), part.c_str()));
  668. target = "a";
  669. part = "";
  670. CPPUNIT_ASSERT(util::startsWith(target.begin(), target.end(),
  671. part.begin(), part.end()));
  672. CPPUNIT_ASSERT(util::startsWith(target.begin(), target.end(), part.c_str()));
  673. target = "a";
  674. part = "a";
  675. CPPUNIT_ASSERT(util::startsWith(target.begin(), target.end(),
  676. part.begin(), part.end()));
  677. CPPUNIT_ASSERT(util::startsWith(target.begin(), target.end(), part.c_str()));
  678. }
  679. void UtilTest::testIstartsWith() {
  680. std::string target;
  681. std::string part;
  682. target = "abcdefg";
  683. part = "aBc";
  684. CPPUNIT_ASSERT(util::istartsWith(target.begin(), target.end(),
  685. part.begin(), part.end()));
  686. CPPUNIT_ASSERT(util::istartsWith(target.begin(), target.end(), part.c_str()));
  687. target = "abcdefg";
  688. part = "abx";
  689. CPPUNIT_ASSERT(!util::istartsWith(target.begin(), target.end(),
  690. part.begin(), part.end()));
  691. CPPUNIT_ASSERT(!util::istartsWith(target.begin(), target.end(), part.c_str()));
  692. }
  693. void UtilTest::testGetContentDispositionFilename() {
  694. std::string h1 = "attachment; filename=\"aria2.tar.bz2\"";
  695. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"), util::getContentDispositionFilename(h1));
  696. std::string h2 = "attachment; filename=\"\"";
  697. CPPUNIT_ASSERT_EQUAL(std::string(""), util::getContentDispositionFilename(h2));
  698. std::string h3 = "attachment; filename=\"";
  699. CPPUNIT_ASSERT_EQUAL(std::string(""), util::getContentDispositionFilename(h3));
  700. std::string h3_2 = "attachment; filename= \" aria2.tar.bz2 \"";
  701. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"),
  702. util::getContentDispositionFilename(h3_2));
  703. std::string h4 = "attachment;";
  704. CPPUNIT_ASSERT_EQUAL(std::string(""), util::getContentDispositionFilename(h4));
  705. std::string h5 = "attachment; filename=aria2.tar.bz2";
  706. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"), util::getContentDispositionFilename(h5));
  707. std::string h6 = "attachment; filename='aria2.tar.bz2'";
  708. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"), util::getContentDispositionFilename(h6));
  709. std::string h7 = "attachment; filename='aria2.tar.bz2";
  710. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"), util::getContentDispositionFilename(h7));
  711. std::string h8 = "attachment; filename=aria2.tar.bz2; creation-date=20 Jun 2007 00:00:00 GMT";
  712. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"), util::getContentDispositionFilename(h8));
  713. std::string h9 = "attachment; filename=\"aria2.tar.bz2; creation-date=20 Jun 2007 00:00:00 GMT\"";
  714. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2; creation-date=20 Jun 2007 00:00:00 GMT"),
  715. util::getContentDispositionFilename(h9));
  716. std::string h10 = "attachment; filename=";
  717. CPPUNIT_ASSERT_EQUAL(std::string(""), util::getContentDispositionFilename(h10));
  718. std::string h11 = "attachment; filename=;";
  719. CPPUNIT_ASSERT_EQUAL(std::string(""), util::getContentDispositionFilename(h11));
  720. std::string filenameWithDir = "attachment; filename=dir/file";
  721. CPPUNIT_ASSERT_EQUAL(std::string(""),
  722. util::getContentDispositionFilename(filenameWithDir));
  723. std::string semicolonInside = "attachment; filename=\"foo;bar\"";
  724. CPPUNIT_ASSERT_EQUAL(std::string("foo;bar"),
  725. util::getContentDispositionFilename(semicolonInside));
  726. // Unescaping %2E%2E%2F produces "../". But since we won't unescape,
  727. // we just accept it as is.
  728. CPPUNIT_ASSERT_EQUAL
  729. (std::string("%2E%2E%2Ffoo.html"),
  730. util::getContentDispositionFilename("filename=\"%2E%2E%2Ffoo.html\""));
  731. // RFC2231 Section4
  732. std::string extparam2 = "filename*=''aria2";
  733. CPPUNIT_ASSERT_EQUAL(std::string("aria2"),
  734. util::getContentDispositionFilename(extparam2));
  735. std::string extparam3 = "filename*='''";
  736. CPPUNIT_ASSERT_EQUAL(std::string(""),
  737. util::getContentDispositionFilename(extparam3));
  738. std::string extparam4 = "filename*='aria2";
  739. CPPUNIT_ASSERT_EQUAL(std::string(""),
  740. util::getContentDispositionFilename(extparam4));
  741. std::string extparam5 = "filename*='''aria2";
  742. CPPUNIT_ASSERT_EQUAL(std::string(""),
  743. util::getContentDispositionFilename(extparam5));
  744. std::string extparam6 = "filename*";
  745. CPPUNIT_ASSERT_EQUAL(std::string(""),
  746. util::getContentDispositionFilename(extparam6));
  747. std::string extparam7 = "filename*=UTF-8''aria2;filename=hello%20world";
  748. CPPUNIT_ASSERT_EQUAL(std::string("aria2"),
  749. util::getContentDispositionFilename(extparam7));
  750. std::string extparam8 = "filename=aria2;filename*=UTF-8''hello%20world";
  751. CPPUNIT_ASSERT_EQUAL(std::string("hello world"),
  752. util::getContentDispositionFilename(extparam8));
  753. std::string extparam9 = "filename*=ISO-8859-1''%A3";
  754. std::string extparam9ans;
  755. extparam9ans += 0xc2;
  756. extparam9ans += 0xa3;
  757. CPPUNIT_ASSERT_EQUAL(extparam9ans,
  758. util::getContentDispositionFilename(extparam9));
  759. CPPUNIT_ASSERT_EQUAL
  760. (std::string(""),
  761. util::getContentDispositionFilename("filename*=UTF-8''foo%2F.html"));
  762. CPPUNIT_ASSERT_EQUAL
  763. (std::string("foo.html"),
  764. util::getContentDispositionFilename("filename*=UTF-8'';filename=\"foo.html\""));
  765. CPPUNIT_ASSERT_EQUAL
  766. (std::string(""),
  767. util::getContentDispositionFilename("filename*=UTF-8''%2E%2E%2Ffoo.html"));
  768. // Tests from http://greenbytes.de/tech/tc2231/
  769. // attwithasciifnescapedchar
  770. CPPUNIT_ASSERT_EQUAL
  771. (std::string("foo.html"),
  772. util::getContentDispositionFilename("filename=\"f\\oo.html\""));
  773. // attwithasciifilenameucase
  774. CPPUNIT_ASSERT_EQUAL
  775. (std::string("foo.html"),
  776. util::getContentDispositionFilename("FILENAME=\"foo.html\""));
  777. // attwithisofn2231iso
  778. CPPUNIT_ASSERT_EQUAL
  779. (std::string("foo-ä.html"),
  780. util::getContentDispositionFilename("filename*=iso-8859-1''foo-%E4.html"));
  781. // attwithfn2231utf8
  782. CPPUNIT_ASSERT_EQUAL
  783. (std::string("foo-ä-€.html"),
  784. util::getContentDispositionFilename
  785. ("filename*=UTF-8''foo-%c3%a4-%e2%82%ac.html"));
  786. // attwithfn2231utf8-bad
  787. CPPUNIT_ASSERT_EQUAL
  788. (std::string(""),
  789. util::getContentDispositionFilename
  790. ("filename*=iso-8859-1''foo-%c3%a4-%e2%82%ac.html"));
  791. // attwithfn2231ws1
  792. CPPUNIT_ASSERT_EQUAL
  793. (std::string(""),
  794. util::getContentDispositionFilename("filename *=UTF-8''foo-%c3%a4.html"));
  795. // attwithfn2231ws2
  796. CPPUNIT_ASSERT_EQUAL
  797. (std::string("foo-ä.html"),
  798. util::getContentDispositionFilename("filename*= UTF-8''foo-%c3%a4.html"));
  799. // attwithfn2231ws3
  800. CPPUNIT_ASSERT_EQUAL
  801. (std::string("foo-ä.html"),
  802. util::getContentDispositionFilename("filename* =UTF-8''foo-%c3%a4.html"));
  803. // attwithfn2231quot
  804. CPPUNIT_ASSERT_EQUAL
  805. (std::string(""),
  806. util::getContentDispositionFilename
  807. ("filename*=\"UTF-8''foo-%c3%a4.html\""));
  808. }
  809. class Printer {
  810. public:
  811. template<class T>
  812. void operator()(T t) {
  813. std::cerr << t << ", ";
  814. }
  815. };
  816. void UtilTest::testRandomAlpha() {
  817. SharedHandle<Randomizer> rand(new FixedNumberRandomizer());
  818. std::string s = util::randomAlpha(8, rand);
  819. CPPUNIT_ASSERT_EQUAL(std::string("AAAAAAAA"), s);
  820. }
  821. void UtilTest::testToUpper() {
  822. std::string src = "608cabc0f2fa18c260cafd974516865c772363d5";
  823. std::string upp = "608CABC0F2FA18C260CAFD974516865C772363D5";
  824. CPPUNIT_ASSERT_EQUAL(upp, util::toUpper(src));
  825. }
  826. void UtilTest::testToLower() {
  827. std::string src = "608CABC0F2FA18C260CAFD974516865C772363D5";
  828. std::string upp = "608cabc0f2fa18c260cafd974516865c772363d5";
  829. CPPUNIT_ASSERT_EQUAL(upp, util::toLower(src));
  830. }
  831. void UtilTest::testUppercase() {
  832. std::string src = "608cabc0f2fa18c260cafd974516865c772363d5";
  833. std::string ans = "608CABC0F2FA18C260CAFD974516865C772363D5";
  834. util::uppercase(src);
  835. CPPUNIT_ASSERT_EQUAL(ans, src);
  836. }
  837. void UtilTest::testLowercase() {
  838. std::string src = "608CABC0F2FA18C260CAFD974516865C772363D5";
  839. std::string ans = "608cabc0f2fa18c260cafd974516865c772363d5";
  840. util::lowercase(src);
  841. CPPUNIT_ASSERT_EQUAL(ans, src);
  842. }
  843. void UtilTest::testPercentDecode() {
  844. std::string src = "http://aria2.sourceforge.net/aria2%200.7.0%20docs.html";
  845. CPPUNIT_ASSERT_EQUAL(std::string("http://aria2.sourceforge.net/aria2 0.7.0 docs.html"),
  846. util::percentDecode(src.begin(), src.end()));
  847. std::string src2 = "aria2+aria2";
  848. CPPUNIT_ASSERT_EQUAL(std::string("aria2+aria2"),
  849. util::percentDecode(src2.begin(), src2.end()));
  850. std::string src3 = "%5t%20";
  851. CPPUNIT_ASSERT_EQUAL(std::string("%5t "),
  852. util::percentDecode(src3.begin(), src3.end()));
  853. std::string src4 = "%";
  854. CPPUNIT_ASSERT_EQUAL(std::string("%"),
  855. util::percentDecode(src4.begin(), src4.end()));
  856. std::string src5 = "%3";
  857. CPPUNIT_ASSERT_EQUAL(std::string("%3"),
  858. util::percentDecode(src5.begin(), src5.end()));
  859. std::string src6 = "%2f";
  860. CPPUNIT_ASSERT_EQUAL(std::string("/"),
  861. util::percentDecode(src6.begin(), src6.end()));
  862. }
  863. void UtilTest::testGetRealSize()
  864. {
  865. CPPUNIT_ASSERT_EQUAL((int64_t)4294967296LL, util::getRealSize("4096M"));
  866. CPPUNIT_ASSERT_EQUAL((int64_t)1024, util::getRealSize("1K"));
  867. try {
  868. util::getRealSize("");
  869. CPPUNIT_FAIL("exception must be thrown.");
  870. } catch(Exception& e) {
  871. std::cerr << e.stackTrace();
  872. }
  873. try {
  874. util::getRealSize("foo");
  875. CPPUNIT_FAIL("exception must be thrown.");
  876. } catch(Exception& e) {
  877. std::cerr << e.stackTrace();
  878. }
  879. try {
  880. util::getRealSize("-1");
  881. CPPUNIT_FAIL("exception must be thrown.");
  882. } catch(Exception& e) {
  883. std::cerr << e.stackTrace();
  884. }
  885. try {
  886. util::getRealSize("9223372036854775807K");
  887. CPPUNIT_FAIL("exception must be thrown.");
  888. } catch(Exception& e) {
  889. std::cerr << e.stackTrace();
  890. }
  891. try {
  892. util::getRealSize("9223372036854775807M");
  893. CPPUNIT_FAIL("exception must be thrown.");
  894. } catch(Exception& e) {
  895. std::cerr << e.stackTrace();
  896. }
  897. }
  898. void UtilTest::testAbbrevSize()
  899. {
  900. CPPUNIT_ASSERT_EQUAL(std::string("4,096.0Mi"), util::abbrevSize(4294967296LL));
  901. CPPUNIT_ASSERT_EQUAL(std::string("1.0Ki"), util::abbrevSize(1024));
  902. CPPUNIT_ASSERT_EQUAL(std::string("1,023"), util::abbrevSize(1023));
  903. CPPUNIT_ASSERT_EQUAL(std::string("0"), util::abbrevSize(0));
  904. CPPUNIT_ASSERT_EQUAL(std::string("1.1Ki"), util::abbrevSize(1127));
  905. CPPUNIT_ASSERT_EQUAL(std::string("1.5Mi"), util::abbrevSize(1572864));
  906. }
  907. void UtilTest::testToStream()
  908. {
  909. std::ostringstream os;
  910. SharedHandle<FileEntry> f1(new FileEntry("aria2.tar.bz2", 12300, 0));
  911. SharedHandle<FileEntry> f2(new FileEntry("aria2.txt", 556, 0));
  912. std::deque<SharedHandle<FileEntry> > entries;
  913. entries.push_back(f1);
  914. entries.push_back(f2);
  915. const char* filename = A2_TEST_OUT_DIR"/aria2_UtilTest_testToStream";
  916. BufferedFile fp(filename, BufferedFile::WRITE);
  917. util::toStream(entries.begin(), entries.end(), fp);
  918. fp.close();
  919. CPPUNIT_ASSERT_EQUAL(
  920. std::string("Files:\n"
  921. "idx|path/length\n"
  922. "===+===========================================================================\n"
  923. " 1|aria2.tar.bz2\n"
  924. " |12.0KiB (12,300)\n"
  925. "---+---------------------------------------------------------------------------\n"
  926. " 2|aria2.txt\n"
  927. " |556B (556)\n"
  928. "---+---------------------------------------------------------------------------\n"),
  929. readFile(filename));
  930. }
  931. void UtilTest::testIsNumber()
  932. {
  933. std::string s = "000";
  934. CPPUNIT_ASSERT_EQUAL(true, util::isNumber(s.begin(),s.end()));
  935. s = "a";
  936. CPPUNIT_ASSERT_EQUAL(false, util::isNumber(s.begin(), s.end()));
  937. s = "0a";
  938. CPPUNIT_ASSERT_EQUAL(false, util::isNumber(s.begin(), s.end()));
  939. s = "";
  940. CPPUNIT_ASSERT_EQUAL(false, util::isNumber(s.begin(), s.end()));
  941. s = " ";
  942. CPPUNIT_ASSERT_EQUAL(false, util::isNumber(s.begin(), s.end()));
  943. }
  944. void UtilTest::testIsLowercase()
  945. {
  946. std::string s = "alpha";
  947. CPPUNIT_ASSERT_EQUAL(true, util::isLowercase(s.begin(), s.end()));
  948. s = "Alpha";
  949. CPPUNIT_ASSERT_EQUAL(false, util::isLowercase(s.begin(), s.end()));
  950. s = "1alpha";
  951. CPPUNIT_ASSERT_EQUAL(false, util::isLowercase(s.begin(), s.end()));
  952. s = "";
  953. CPPUNIT_ASSERT_EQUAL(false, util::isLowercase(s.begin(), s.end()));
  954. s = " ";
  955. CPPUNIT_ASSERT_EQUAL(false, util::isLowercase(s.begin(), s.end()));
  956. }
  957. void UtilTest::testIsUppercase()
  958. {
  959. std::string s = "ALPHA";
  960. CPPUNIT_ASSERT_EQUAL(true, util::isUppercase(s.begin(), s.end()));
  961. s = "Alpha";
  962. CPPUNIT_ASSERT_EQUAL(false, util::isUppercase(s.begin(), s.end()));
  963. s = "1ALPHA";
  964. CPPUNIT_ASSERT_EQUAL(false, util::isUppercase(s.begin(), s.end()));
  965. s = "";
  966. CPPUNIT_ASSERT_EQUAL(false, util::isUppercase(s.begin(), s.end()));
  967. s = " ";
  968. CPPUNIT_ASSERT_EQUAL(false, util::isUppercase(s.begin(), s.end()));
  969. }
  970. void UtilTest::testAlphaToNum()
  971. {
  972. CPPUNIT_ASSERT_EQUAL(0U, util::alphaToNum("a"));
  973. CPPUNIT_ASSERT_EQUAL(0U, util::alphaToNum("aa"));
  974. CPPUNIT_ASSERT_EQUAL(1U, util::alphaToNum("b"));
  975. CPPUNIT_ASSERT_EQUAL(675U, util::alphaToNum("zz")); // 25*26+25
  976. CPPUNIT_ASSERT_EQUAL(675U, util::alphaToNum("ZZ")); // 25*26+25
  977. CPPUNIT_ASSERT_EQUAL(0U, util::alphaToNum(""));
  978. CPPUNIT_ASSERT_EQUAL(4294967295U, util::alphaToNum("NXMRLXV"));
  979. CPPUNIT_ASSERT_EQUAL(0U, util::alphaToNum("NXMRLXW")); // uint32_t overflow
  980. }
  981. void UtilTest::testMkdirs()
  982. {
  983. std::string dir = A2_TEST_OUT_DIR"/aria2-UtilTest-testMkdirs";
  984. File d(dir);
  985. if(d.exists()) {
  986. CPPUNIT_ASSERT(d.remove());
  987. }
  988. CPPUNIT_ASSERT(!d.exists());
  989. util::mkdirs(dir);
  990. CPPUNIT_ASSERT(d.isDir());
  991. std::string file = A2_TEST_DIR"/UtilTest.cc";
  992. File f(file);
  993. CPPUNIT_ASSERT(f.isFile());
  994. try {
  995. util::mkdirs(file);
  996. CPPUNIT_FAIL("exception must be thrown.");
  997. } catch(DlAbortEx& ex) {
  998. std::cerr << ex.stackTrace() << std::endl;
  999. }
  1000. }
  1001. void UtilTest::testConvertBitfield()
  1002. {
  1003. BitfieldMan srcBitfield(384*1024, 256*1024*256+1);
  1004. BitfieldMan destBitfield(512*1024, srcBitfield.getTotalLength());
  1005. srcBitfield.setAllBit();
  1006. srcBitfield.unsetBit(2);// <- range [768, 1152)
  1007. // which corresponds to the index [1,2] in destBitfield
  1008. util::convertBitfield(&destBitfield, &srcBitfield);
  1009. CPPUNIT_ASSERT_EQUAL(std::string("9fffffffffffffffffffffffffffffff80"),
  1010. util::toHex(destBitfield.getBitfield(),
  1011. destBitfield.getBitfieldLength()));
  1012. }
  1013. void UtilTest::testParseIntSegments()
  1014. {
  1015. SegList<int> sgl;
  1016. util::parseIntSegments(sgl, "1,3-8,10");
  1017. CPPUNIT_ASSERT(sgl.hasNext());
  1018. CPPUNIT_ASSERT_EQUAL(1, sgl.next());
  1019. CPPUNIT_ASSERT(sgl.hasNext());
  1020. CPPUNIT_ASSERT_EQUAL(3, sgl.next());
  1021. CPPUNIT_ASSERT(sgl.hasNext());
  1022. CPPUNIT_ASSERT_EQUAL(4, sgl.next());
  1023. CPPUNIT_ASSERT(sgl.hasNext());
  1024. CPPUNIT_ASSERT_EQUAL(5, sgl.next());
  1025. CPPUNIT_ASSERT(sgl.hasNext());
  1026. CPPUNIT_ASSERT_EQUAL(6, sgl.next());
  1027. CPPUNIT_ASSERT(sgl.hasNext());
  1028. CPPUNIT_ASSERT_EQUAL(7, sgl.next());
  1029. CPPUNIT_ASSERT(sgl.hasNext());
  1030. CPPUNIT_ASSERT_EQUAL(8, sgl.next());
  1031. CPPUNIT_ASSERT(sgl.hasNext());
  1032. CPPUNIT_ASSERT_EQUAL(10, sgl.next());
  1033. CPPUNIT_ASSERT(!sgl.hasNext());
  1034. CPPUNIT_ASSERT_EQUAL(0, sgl.next());
  1035. sgl.clear();
  1036. util::parseIntSegments(sgl, ",,,1,,,3,,,");
  1037. CPPUNIT_ASSERT_EQUAL(1, sgl.next());
  1038. CPPUNIT_ASSERT_EQUAL(3, sgl.next());
  1039. CPPUNIT_ASSERT(!sgl.hasNext());
  1040. }
  1041. void UtilTest::testParseIntSegments_invalidRange()
  1042. {
  1043. try {
  1044. SegList<int> sgl;
  1045. util::parseIntSegments(sgl, "-1");
  1046. CPPUNIT_FAIL("exception must be thrown.");
  1047. } catch(Exception& e) {
  1048. }
  1049. try {
  1050. SegList<int> sgl;
  1051. util::parseIntSegments(sgl, "1-");
  1052. CPPUNIT_FAIL("exception must be thrown.");
  1053. } catch(Exception& e) {
  1054. }
  1055. try {
  1056. SegList<int> sgl;
  1057. util::parseIntSegments(sgl, "2147483648");
  1058. CPPUNIT_FAIL("exception must be thrown.");
  1059. } catch(Exception& e) {
  1060. }
  1061. try {
  1062. SegList<int> sgl;
  1063. util::parseIntSegments(sgl, "2147483647-2147483648");
  1064. CPPUNIT_FAIL("exception must be thrown.");
  1065. } catch(Exception& e) {
  1066. }
  1067. try {
  1068. SegList<int> sgl;
  1069. util::parseIntSegments(sgl, "1-2x");
  1070. CPPUNIT_FAIL("exception must be thrown.");
  1071. } catch(Exception& e) {
  1072. }
  1073. try {
  1074. SegList<int> sgl;
  1075. util::parseIntSegments(sgl, "3x-4");
  1076. CPPUNIT_FAIL("exception must be thrown.");
  1077. } catch(Exception& e) {
  1078. }
  1079. }
  1080. void UtilTest::testParseIntNoThrow()
  1081. {
  1082. std::string s;
  1083. int32_t n;
  1084. s = " -1 ";
  1085. CPPUNIT_ASSERT(util::parseIntNoThrow(n, s));
  1086. CPPUNIT_ASSERT_EQUAL((int32_t)-1, n);
  1087. s = "2147483647";
  1088. CPPUNIT_ASSERT(util::parseIntNoThrow(n, s));
  1089. CPPUNIT_ASSERT_EQUAL((int32_t)2147483647, n);
  1090. s = "2147483648";
  1091. CPPUNIT_ASSERT(!util::parseIntNoThrow(n, s));
  1092. s = "-2147483649";
  1093. CPPUNIT_ASSERT(!util::parseIntNoThrow(n, s));
  1094. s = "12x";
  1095. CPPUNIT_ASSERT(!util::parseIntNoThrow(n, s));
  1096. s = "";
  1097. CPPUNIT_ASSERT(!util::parseIntNoThrow(n, s));
  1098. }
  1099. void UtilTest::testParseUIntNoThrow()
  1100. {
  1101. std::string s;
  1102. uint32_t n;
  1103. s = " 2147483647 ";
  1104. CPPUNIT_ASSERT(util::parseUIntNoThrow(n, s));
  1105. CPPUNIT_ASSERT_EQUAL((uint32_t)INT32_MAX, n);
  1106. s = "2147483648";
  1107. CPPUNIT_ASSERT(!util::parseUIntNoThrow(n, s));
  1108. s = "-1";
  1109. CPPUNIT_ASSERT(!util::parseUIntNoThrow(n, s));
  1110. }
  1111. void UtilTest::testParseLLIntNoThrow()
  1112. {
  1113. std::string s;
  1114. int64_t n;
  1115. s = " 9223372036854775807 ";
  1116. CPPUNIT_ASSERT(util::parseLLIntNoThrow(n, s));
  1117. CPPUNIT_ASSERT_EQUAL((int64_t)INT64_MAX, n);
  1118. s = "9223372036854775808";
  1119. CPPUNIT_ASSERT(!util::parseLLIntNoThrow(n, s));
  1120. s = "-9223372036854775808";
  1121. CPPUNIT_ASSERT(util::parseLLIntNoThrow(n, s));
  1122. CPPUNIT_ASSERT_EQUAL((int64_t)INT64_MIN, n);
  1123. s = "-9223372036854775809";
  1124. CPPUNIT_ASSERT(!util::parseLLIntNoThrow(n, s));
  1125. }
  1126. void UtilTest::testToString_binaryStream()
  1127. {
  1128. SharedHandle<DiskWriter> dw(new ByteArrayDiskWriter());
  1129. std::string data(16*1024+256, 'a');
  1130. dw->initAndOpenFile();
  1131. dw->writeData((const unsigned char*)data.c_str(), data.size(), 0);
  1132. std::string readData = util::toString(dw);
  1133. CPPUNIT_ASSERT_EQUAL(data, readData);
  1134. }
  1135. void UtilTest::testItos()
  1136. {
  1137. {
  1138. int i = 0;
  1139. CPPUNIT_ASSERT_EQUAL(std::string("0"), util::itos(i));
  1140. }
  1141. {
  1142. int i = 100;
  1143. CPPUNIT_ASSERT_EQUAL(std::string("100"), util::itos(i, true));
  1144. }
  1145. {
  1146. int i = 100;
  1147. CPPUNIT_ASSERT_EQUAL(std::string("100"), util::itos(i));
  1148. }
  1149. {
  1150. int i = 12345;
  1151. CPPUNIT_ASSERT_EQUAL(std::string("12,345"), util::itos(i, true));
  1152. }
  1153. {
  1154. int i = 12345;
  1155. CPPUNIT_ASSERT_EQUAL(std::string("12345"), util::itos(i));
  1156. }
  1157. {
  1158. int i = -12345;
  1159. CPPUNIT_ASSERT_EQUAL(std::string("-12,345"), util::itos(i, true));
  1160. }
  1161. {
  1162. int64_t i = INT64_MAX;
  1163. CPPUNIT_ASSERT_EQUAL(std::string("9,223,372,036,854,775,807"),
  1164. util::itos(i, true));
  1165. }
  1166. {
  1167. int64_t i = INT64_MIN;
  1168. CPPUNIT_ASSERT_EQUAL(std::string("-9,223,372,036,854,775,808"),
  1169. util::itos(i, true));
  1170. }
  1171. }
  1172. void UtilTest::testUitos()
  1173. {
  1174. {
  1175. uint16_t i = 12345;
  1176. CPPUNIT_ASSERT_EQUAL(std::string("12345"), util::uitos(i));
  1177. }
  1178. {
  1179. int16_t i = -12345;
  1180. CPPUNIT_ASSERT_EQUAL(std::string("/.-,+"), util::uitos(i));
  1181. }
  1182. }
  1183. void UtilTest::testNtoh64()
  1184. {
  1185. uint64_t x = 0xff00ff00ee00ee00LL;
  1186. #ifdef WORDS_BIGENDIAN
  1187. CPPUNIT_ASSERT_EQUAL(x, ntoh64(x));
  1188. CPPUNIT_ASSERT_EQUAL(x, hton64(x));
  1189. #else // !WORDS_BIGENDIAN
  1190. uint64_t y = 0x00ee00ee00ff00ffLL;
  1191. CPPUNIT_ASSERT_EQUAL(y, ntoh64(x));
  1192. CPPUNIT_ASSERT_EQUAL(x, hton64(y));
  1193. #endif // !WORDS_BIGENDIAN
  1194. }
  1195. void UtilTest::testPercentEncode()
  1196. {
  1197. CPPUNIT_ASSERT_EQUAL
  1198. (std::string("%3A%2F%3F%23%5B%5D%40%21%25%26%27%28%29%2A%2B%2C%3B%3D"),
  1199. util::percentEncode(":/?#[]@!%&'()*+,;="));
  1200. std::string unreserved =
  1201. "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
  1202. "abcdefghijklmnopqrstuvwxyz"
  1203. "0123456789"
  1204. "-._~";
  1205. CPPUNIT_ASSERT_EQUAL(unreserved, util::percentEncode(unreserved));
  1206. CPPUNIT_ASSERT_EQUAL(std::string("1%5EA%20"), util::percentEncode("1^A "));
  1207. }
  1208. void UtilTest::testHtmlEscape()
  1209. {
  1210. CPPUNIT_ASSERT_EQUAL(std::string("aria2&lt;&gt;&quot;&#39;util"),
  1211. util::htmlEscape("aria2<>\"'util"));
  1212. }
  1213. void UtilTest::testJoinPath()
  1214. {
  1215. const std::string dir1dir2file[] = { "dir1", "dir2", "file" };
  1216. CPPUNIT_ASSERT_EQUAL
  1217. (std::string("dir1/dir2/file"),
  1218. util::joinPath(vbegin(dir1dir2file), vend(dir1dir2file)));
  1219. const std::string dirparentfile[] = { "dir", "..", "file" };
  1220. CPPUNIT_ASSERT_EQUAL
  1221. (std::string("file"),
  1222. util::joinPath(vbegin(dirparentfile), vend(dirparentfile)));
  1223. const std::string dirparentparentfile[] = { "dir", "..", "..", "file" };
  1224. CPPUNIT_ASSERT_EQUAL
  1225. (std::string("file"),
  1226. util::joinPath(vbegin(dirparentparentfile), vend(dirparentparentfile)));
  1227. const std::string dirdotfile[] = { "dir", ".", "file" };
  1228. CPPUNIT_ASSERT_EQUAL(std::string("dir/file"),
  1229. util::joinPath(vbegin(dirdotfile), vend(dirdotfile)));
  1230. const std::string empty[] = {};
  1231. CPPUNIT_ASSERT_EQUAL(std::string(""), util::joinPath(&empty[0], &empty[0]));
  1232. const std::string parentdot[] = { "..", "." };
  1233. CPPUNIT_ASSERT_EQUAL(std::string(""),
  1234. util::joinPath(vbegin(parentdot), vend(parentdot)));
  1235. }
  1236. void UtilTest::testParseIndexPath()
  1237. {
  1238. std::pair<size_t, std::string> p = util::parseIndexPath("1=foo");
  1239. CPPUNIT_ASSERT_EQUAL((size_t)1, p.first);
  1240. CPPUNIT_ASSERT_EQUAL(std::string("foo"), p.second);
  1241. try {
  1242. util::parseIndexPath("1X=foo");
  1243. CPPUNIT_FAIL("exception must be thrown.");
  1244. } catch(Exception& e) {
  1245. // success
  1246. }
  1247. try {
  1248. util::parseIndexPath("1=");
  1249. CPPUNIT_FAIL("exception must be thrown.");
  1250. } catch(Exception& e) {
  1251. // success
  1252. }
  1253. }
  1254. void UtilTest::testCreateIndexPaths()
  1255. {
  1256. std::stringstream in
  1257. ("1=/tmp/myfile\n"
  1258. "100=/myhome/mypicture.png\n");
  1259. std::vector<std::pair<size_t, std::string> > m = util::createIndexPaths(in);
  1260. CPPUNIT_ASSERT_EQUAL((size_t)2, m.size());
  1261. CPPUNIT_ASSERT_EQUAL((size_t)1, m[0].first);
  1262. CPPUNIT_ASSERT_EQUAL(std::string("/tmp/myfile"), m[0].second);
  1263. CPPUNIT_ASSERT_EQUAL((size_t)100, m[1].first);
  1264. CPPUNIT_ASSERT_EQUAL(std::string("/myhome/mypicture.png"), m[1].second);
  1265. }
  1266. void UtilTest::testGenerateRandomData()
  1267. {
  1268. unsigned char data1[20];
  1269. util::generateRandomData(data1, sizeof(data1));
  1270. unsigned char data2[20];
  1271. util::generateRandomData(data2, sizeof(data2));
  1272. CPPUNIT_ASSERT(memcmp(data1, data2, sizeof(data1)) != 0);
  1273. }
  1274. void UtilTest::testFromHex()
  1275. {
  1276. std::string src;
  1277. std::string dest;
  1278. src = "0011fF";
  1279. dest = util::fromHex(src.begin(), src.end());
  1280. CPPUNIT_ASSERT_EQUAL((size_t)3, dest.size());
  1281. CPPUNIT_ASSERT_EQUAL((char)0x00, dest[0]);
  1282. CPPUNIT_ASSERT_EQUAL((char)0x11, dest[1]);
  1283. CPPUNIT_ASSERT_EQUAL((char)0xff, dest[2]);
  1284. src = "0011f";
  1285. CPPUNIT_ASSERT(util::fromHex(src.begin(), src.end()).empty());
  1286. src = "001g";
  1287. CPPUNIT_ASSERT(util::fromHex(src.begin(), src.end()).empty());
  1288. }
  1289. void UtilTest::testParsePrioritizePieceRange()
  1290. {
  1291. // piece index
  1292. // 0 1 2 3 4 5 6 7
  1293. // | | | |
  1294. // file1 | | |
  1295. // | | |
  1296. // file2 | |
  1297. // file3 |
  1298. // | |
  1299. // file4 |
  1300. size_t pieceLength = 1024;
  1301. std::vector<SharedHandle<FileEntry> > entries(4, SharedHandle<FileEntry>());
  1302. entries[0].reset(new FileEntry("file1", 1024, 0));
  1303. entries[1].reset(new FileEntry("file2",2560,entries[0]->getLastOffset()));
  1304. entries[2].reset(new FileEntry("file3",0,entries[1]->getLastOffset()));
  1305. entries[3].reset(new FileEntry("file4",3584,entries[2]->getLastOffset()));
  1306. std::vector<size_t> result;
  1307. util::parsePrioritizePieceRange(result, "head=1", entries, pieceLength);
  1308. CPPUNIT_ASSERT_EQUAL((size_t)3, result.size());
  1309. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1310. CPPUNIT_ASSERT_EQUAL((size_t)1, result[1]);
  1311. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  1312. result.clear();
  1313. util::parsePrioritizePieceRange(result, "tail=1", entries, pieceLength);
  1314. CPPUNIT_ASSERT_EQUAL((size_t)3, result.size());
  1315. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1316. CPPUNIT_ASSERT_EQUAL((size_t)3, result[1]);
  1317. CPPUNIT_ASSERT_EQUAL((size_t)6, result[2]);
  1318. result.clear();
  1319. util::parsePrioritizePieceRange(result, "head=1K", entries, pieceLength);
  1320. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  1321. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1322. CPPUNIT_ASSERT_EQUAL((size_t)1, result[1]);
  1323. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  1324. CPPUNIT_ASSERT_EQUAL((size_t)4, result[3]);
  1325. result.clear();
  1326. util::parsePrioritizePieceRange(result, "head", entries, pieceLength, 1024);
  1327. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  1328. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1329. CPPUNIT_ASSERT_EQUAL((size_t)1, result[1]);
  1330. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  1331. CPPUNIT_ASSERT_EQUAL((size_t)4, result[3]);
  1332. result.clear();
  1333. util::parsePrioritizePieceRange(result, "tail=1K", entries, pieceLength);
  1334. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  1335. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1336. CPPUNIT_ASSERT_EQUAL((size_t)2, result[1]);
  1337. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  1338. CPPUNIT_ASSERT_EQUAL((size_t)6, result[3]);
  1339. result.clear();
  1340. util::parsePrioritizePieceRange(result, "tail", entries, pieceLength, 1024);
  1341. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  1342. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1343. CPPUNIT_ASSERT_EQUAL((size_t)2, result[1]);
  1344. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  1345. CPPUNIT_ASSERT_EQUAL((size_t)6, result[3]);
  1346. result.clear();
  1347. util::parsePrioritizePieceRange(result, "head=1,tail=1", entries, pieceLength);
  1348. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  1349. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1350. CPPUNIT_ASSERT_EQUAL((size_t)1, result[1]);
  1351. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  1352. CPPUNIT_ASSERT_EQUAL((size_t)6, result[3]);
  1353. result.clear();
  1354. util::parsePrioritizePieceRange(result, "head=300M,tail=300M",
  1355. entries, pieceLength);
  1356. CPPUNIT_ASSERT_EQUAL((size_t)7, result.size());
  1357. for(size_t i = 0; i < 7; ++i) {
  1358. CPPUNIT_ASSERT_EQUAL(i, result[i]);
  1359. }
  1360. result.clear();
  1361. util::parsePrioritizePieceRange(result, "", entries, pieceLength);
  1362. CPPUNIT_ASSERT(result.empty());
  1363. }
  1364. void UtilTest::testApplyDir()
  1365. {
  1366. CPPUNIT_ASSERT_EQUAL(std::string("./pred"), util::applyDir("", "pred"));
  1367. CPPUNIT_ASSERT_EQUAL(std::string("/pred"), util::applyDir("/", "pred"));
  1368. CPPUNIT_ASSERT_EQUAL(std::string("./pred"), util::applyDir(".", "pred"));
  1369. CPPUNIT_ASSERT_EQUAL(std::string("/dl/pred"), util::applyDir("/dl", "pred"));
  1370. }
  1371. void UtilTest::testFixTaintedBasename()
  1372. {
  1373. CPPUNIT_ASSERT_EQUAL(std::string("a%2Fb"), util::fixTaintedBasename("a/b"));
  1374. #ifdef __MINGW32__
  1375. CPPUNIT_ASSERT_EQUAL(std::string("a%5Cb"), util::fixTaintedBasename("a\\b"));
  1376. #else // !__MINGW32__
  1377. CPPUNIT_ASSERT_EQUAL(std::string("a\\b"), util::fixTaintedBasename("a\\b"));
  1378. #endif // !__MINGW32__
  1379. }
  1380. void UtilTest::testIsNumericHost()
  1381. {
  1382. CPPUNIT_ASSERT(util::isNumericHost("192.168.0.1"));
  1383. CPPUNIT_ASSERT(!util::isNumericHost("aria2.sf.net"));
  1384. CPPUNIT_ASSERT(util::isNumericHost("::1"));
  1385. }
  1386. void UtilTest::testDetectDirTraversal()
  1387. {
  1388. CPPUNIT_ASSERT(util::detectDirTraversal("/foo"));
  1389. CPPUNIT_ASSERT(util::detectDirTraversal("./foo"));
  1390. CPPUNIT_ASSERT(util::detectDirTraversal("../foo"));
  1391. CPPUNIT_ASSERT(util::detectDirTraversal("foo/../bar"));
  1392. CPPUNIT_ASSERT(util::detectDirTraversal("foo/./bar"));
  1393. CPPUNIT_ASSERT(util::detectDirTraversal("foo/."));
  1394. CPPUNIT_ASSERT(util::detectDirTraversal("foo/.."));
  1395. CPPUNIT_ASSERT(util::detectDirTraversal("."));
  1396. CPPUNIT_ASSERT(util::detectDirTraversal(".."));
  1397. CPPUNIT_ASSERT(util::detectDirTraversal("/"));
  1398. CPPUNIT_ASSERT(util::detectDirTraversal("foo/"));
  1399. CPPUNIT_ASSERT(util::detectDirTraversal("\t"));
  1400. CPPUNIT_ASSERT(!util::detectDirTraversal("foo/bar"));
  1401. CPPUNIT_ASSERT(!util::detectDirTraversal("foo"));
  1402. }
  1403. void UtilTest::testEscapePath()
  1404. {
  1405. CPPUNIT_ASSERT_EQUAL(std::string("foo%00bar%00%01"),
  1406. util::escapePath(std::string("foo")+(char)0x00+
  1407. std::string("bar")+(char)0x00+
  1408. (char)0x01));
  1409. #ifdef __MINGW32__
  1410. CPPUNIT_ASSERT_EQUAL(std::string("foo%5Cbar"), util::escapePath("foo\\bar"));
  1411. #else // !__MINGW32__
  1412. CPPUNIT_ASSERT_EQUAL(std::string("foo\\bar"), util::escapePath("foo\\bar"));
  1413. #endif // !__MINGW32__
  1414. }
  1415. void UtilTest::testInSameCidrBlock()
  1416. {
  1417. CPPUNIT_ASSERT(util::inSameCidrBlock("192.168.128.1", "192.168.0.1", 16));
  1418. CPPUNIT_ASSERT(!util::inSameCidrBlock("192.168.128.1", "192.168.0.1", 17));
  1419. CPPUNIT_ASSERT(util::inSameCidrBlock("192.168.0.1", "192.168.0.1", 32));
  1420. CPPUNIT_ASSERT(!util::inSameCidrBlock("192.168.0.1", "192.168.0.0", 32));
  1421. CPPUNIT_ASSERT(util::inSameCidrBlock("192.168.0.1", "10.0.0.1", 0));
  1422. CPPUNIT_ASSERT(util::inSameCidrBlock("2001:db8::2:1", "2001:db0::2:2", 28));
  1423. CPPUNIT_ASSERT(!util::inSameCidrBlock("2001:db8::2:1", "2001:db0::2:2", 29));
  1424. CPPUNIT_ASSERT(!util::inSameCidrBlock("2001:db8::2:1", "192.168.0.1", 8));
  1425. }
  1426. void UtilTest::testIsUtf8String()
  1427. {
  1428. CPPUNIT_ASSERT(util::isUtf8("ascii"));
  1429. // "Hello World" in Japanese UTF-8
  1430. CPPUNIT_ASSERT(util::isUtf8
  1431. (fromHex("e38193e38293e381abe381a1e381afe4b896e7958c")));
  1432. // "World" in Shift_JIS
  1433. CPPUNIT_ASSERT(!util::isUtf8(fromHex("90a28a")+"E"));
  1434. // UTF8-2
  1435. CPPUNIT_ASSERT(util::isUtf8(fromHex("c280")));
  1436. CPPUNIT_ASSERT(util::isUtf8(fromHex("dfbf")));
  1437. // UTF8-3
  1438. CPPUNIT_ASSERT(util::isUtf8(fromHex("e0a080")));
  1439. CPPUNIT_ASSERT(util::isUtf8(fromHex("e0bf80")));
  1440. CPPUNIT_ASSERT(util::isUtf8(fromHex("e18080")));
  1441. CPPUNIT_ASSERT(util::isUtf8(fromHex("ec8080")));
  1442. CPPUNIT_ASSERT(util::isUtf8(fromHex("ed8080")));
  1443. CPPUNIT_ASSERT(util::isUtf8(fromHex("ed9f80")));
  1444. CPPUNIT_ASSERT(util::isUtf8(fromHex("ee8080")));
  1445. CPPUNIT_ASSERT(util::isUtf8(fromHex("ef8080")));
  1446. // UTF8-4
  1447. CPPUNIT_ASSERT(util::isUtf8(fromHex("f0908080")));
  1448. CPPUNIT_ASSERT(util::isUtf8(fromHex("f0bf8080")));
  1449. CPPUNIT_ASSERT(util::isUtf8(fromHex("f1808080")));
  1450. CPPUNIT_ASSERT(util::isUtf8(fromHex("f3808080")));
  1451. CPPUNIT_ASSERT(util::isUtf8(fromHex("f4808080")));
  1452. CPPUNIT_ASSERT(util::isUtf8(fromHex("f48f8080")));
  1453. CPPUNIT_ASSERT(util::isUtf8(""));
  1454. CPPUNIT_ASSERT(!util::isUtf8(fromHex("00")));
  1455. }
  1456. void UtilTest::testNextParam()
  1457. {
  1458. std::string s1 = " :a : b=c :d=b::::g::";
  1459. std::pair<std::string::iterator, bool> r;
  1460. std::string name, value;
  1461. r = util::nextParam(name, value, s1.begin(), s1.end(), ':');
  1462. CPPUNIT_ASSERT(r.second);
  1463. CPPUNIT_ASSERT_EQUAL(std::string("a"), name);
  1464. CPPUNIT_ASSERT_EQUAL(std::string(), value);
  1465. r = util::nextParam(name, value, r.first, s1.end(), ':');
  1466. CPPUNIT_ASSERT(r.second);
  1467. CPPUNIT_ASSERT_EQUAL(std::string("b"), name);
  1468. CPPUNIT_ASSERT_EQUAL(std::string("c"), value);
  1469. r = util::nextParam(name, value, r.first, s1.end(), ':');
  1470. CPPUNIT_ASSERT(r.second);
  1471. CPPUNIT_ASSERT_EQUAL(std::string("d"), name);
  1472. CPPUNIT_ASSERT_EQUAL(std::string("b"), value);
  1473. r = util::nextParam(name, value, r.first, s1.end(), ':');
  1474. CPPUNIT_ASSERT(r.second);
  1475. CPPUNIT_ASSERT_EQUAL(std::string("g"), name);
  1476. CPPUNIT_ASSERT_EQUAL(std::string(), value);
  1477. std::string s2 = "";
  1478. r = util::nextParam(name, value, s2.begin(), s2.end(), ':');
  1479. CPPUNIT_ASSERT(!r.second);
  1480. std::string s3 = " ";
  1481. r = util::nextParam(name, value, s3.begin(), s3.end(), ':');
  1482. CPPUNIT_ASSERT(!r.second);
  1483. std::string s4 = ":::";
  1484. r = util::nextParam(name, value, s4.begin(), s4.end(), ':');
  1485. CPPUNIT_ASSERT(!r.second);
  1486. }
  1487. void UtilTest::testNoProxyDomainMatch()
  1488. {
  1489. CPPUNIT_ASSERT(util::noProxyDomainMatch("localhost", "localhost"));
  1490. CPPUNIT_ASSERT(util::noProxyDomainMatch("192.168.0.1", "192.168.0.1"));
  1491. CPPUNIT_ASSERT(util::noProxyDomainMatch("www.example.org", ".example.org"));
  1492. CPPUNIT_ASSERT(!util::noProxyDomainMatch("www.example.org", "example.org"));
  1493. CPPUNIT_ASSERT(!util::noProxyDomainMatch("192.168.0.1", "0.1"));
  1494. CPPUNIT_ASSERT(!util::noProxyDomainMatch("example.org", "example.com"));
  1495. CPPUNIT_ASSERT(!util::noProxyDomainMatch("example.org", "www.example.org"));
  1496. }
  1497. void UtilTest::testInPrivateAddress()
  1498. {
  1499. CPPUNIT_ASSERT(!util::inPrivateAddress("localhost"));
  1500. CPPUNIT_ASSERT(util::inPrivateAddress("192.168.0.1"));
  1501. // Only checks prefix..
  1502. CPPUNIT_ASSERT(util::inPrivateAddress("10."));
  1503. CPPUNIT_ASSERT(!util::inPrivateAddress("172."));
  1504. CPPUNIT_ASSERT(!util::inPrivateAddress("172.15.0.0"));
  1505. CPPUNIT_ASSERT(util::inPrivateAddress("172.16.0.0"));
  1506. CPPUNIT_ASSERT(util::inPrivateAddress("172.31.0.0"));
  1507. CPPUNIT_ASSERT(!util::inPrivateAddress("172.32.0.0"));
  1508. }
  1509. void UtilTest::testSecfmt()
  1510. {
  1511. CPPUNIT_ASSERT_EQUAL(std::string("0s"), util::secfmt(0));
  1512. CPPUNIT_ASSERT_EQUAL(std::string("1s"), util::secfmt(1));
  1513. CPPUNIT_ASSERT_EQUAL(std::string("9s"), util::secfmt(9));
  1514. CPPUNIT_ASSERT_EQUAL(std::string("10s"), util::secfmt(10));
  1515. CPPUNIT_ASSERT_EQUAL(std::string("1m"), util::secfmt(60));
  1516. CPPUNIT_ASSERT_EQUAL(std::string("1m59s"), util::secfmt(119));
  1517. CPPUNIT_ASSERT_EQUAL(std::string("2m"), util::secfmt(120));
  1518. CPPUNIT_ASSERT_EQUAL(std::string("59m59s"), util::secfmt(3599));
  1519. CPPUNIT_ASSERT_EQUAL(std::string("1h"), util::secfmt(3600));
  1520. }
  1521. void UtilTest::testTlsHostnameMatch()
  1522. {
  1523. CPPUNIT_ASSERT(util::tlsHostnameMatch("Foo.com", "foo.com"));
  1524. CPPUNIT_ASSERT(util::tlsHostnameMatch("*.a.com", "foo.a.com"));
  1525. CPPUNIT_ASSERT(!util::tlsHostnameMatch("*.a.com", "bar.foo.a.com"));
  1526. CPPUNIT_ASSERT(!util::tlsHostnameMatch("f*.com", "foo.com"));
  1527. CPPUNIT_ASSERT(!util::tlsHostnameMatch("*.com", "bar.com"));
  1528. CPPUNIT_ASSERT(util::tlsHostnameMatch("com", "com"));
  1529. CPPUNIT_ASSERT(!util::tlsHostnameMatch("foo.*", "foo.com"));
  1530. CPPUNIT_ASSERT(util::tlsHostnameMatch("a.foo.com", "A.foo.com"));
  1531. CPPUNIT_ASSERT(!util::tlsHostnameMatch("a.foo.com", "b.foo.com"));
  1532. CPPUNIT_ASSERT(!util::tlsHostnameMatch("*a.foo.com", "a.foo.com"));
  1533. CPPUNIT_ASSERT(util::tlsHostnameMatch("*a.foo.com", "ba.foo.com"));
  1534. CPPUNIT_ASSERT(!util::tlsHostnameMatch("a*.foo.com", "a.foo.com"));
  1535. CPPUNIT_ASSERT(util::tlsHostnameMatch("a*.foo.com", "ab.foo.com"));
  1536. CPPUNIT_ASSERT(!util::tlsHostnameMatch("foo.b*z.foo.com", "foo.baz.foo.com"));
  1537. CPPUNIT_ASSERT(util::tlsHostnameMatch("B*z.foo.com", "bAZ.Foo.com"));
  1538. CPPUNIT_ASSERT(!util::tlsHostnameMatch("b*z.foo.com", "bz.foo.com"));
  1539. CPPUNIT_ASSERT(!util::tlsHostnameMatch("*", "foo"));
  1540. CPPUNIT_ASSERT(!util::tlsHostnameMatch("*", ""));
  1541. CPPUNIT_ASSERT(util::tlsHostnameMatch("", ""));
  1542. CPPUNIT_ASSERT(!util::tlsHostnameMatch("xn--*.a.b", "xn--c.a.b"));
  1543. }
  1544. } // namespace aria2