UtilTest.cc 60 KB

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