UtilTest.cc 84 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303
  1. #include "util.h"
  2. #include <cmath>
  3. #include <cstring>
  4. #include <string>
  5. #include <iostream>
  6. #include <cppunit/extensions/HelperMacros.h>
  7. #include "FixedNumberRandomizer.h"
  8. #include "DlAbortEx.h"
  9. #include "BitfieldMan.h"
  10. #include "ByteArrayDiskWriter.h"
  11. #include "FileEntry.h"
  12. #include "File.h"
  13. #include "array_fun.h"
  14. #include "BufferedFile.h"
  15. #include "TestUtil.h"
  16. #include "SocketCore.h"
  17. namespace aria2 {
  18. class UtilTest:public CppUnit::TestFixture {
  19. CPPUNIT_TEST_SUITE(UtilTest);
  20. CPPUNIT_TEST(testStrip);
  21. CPPUNIT_TEST(testStripIter);
  22. CPPUNIT_TEST(testLstripIter);
  23. CPPUNIT_TEST(testLstripIter_char);
  24. CPPUNIT_TEST(testDivide);
  25. CPPUNIT_TEST(testSplit);
  26. CPPUNIT_TEST(testSplitIter);
  27. CPPUNIT_TEST(testSplitIterM);
  28. CPPUNIT_TEST(testStreq);
  29. CPPUNIT_TEST(testStrieq);
  30. CPPUNIT_TEST(testStrifind);
  31. CPPUNIT_TEST(testEndsWith);
  32. CPPUNIT_TEST(testIendsWith);
  33. CPPUNIT_TEST(testReplace);
  34. CPPUNIT_TEST(testStartsWith);
  35. CPPUNIT_TEST(testIstartsWith);
  36. // may be moved to other helper class in the future.
  37. CPPUNIT_TEST(testGetContentDispositionFilename);
  38. CPPUNIT_TEST(testParseContentDisposition);
  39. CPPUNIT_TEST(testToUpper);
  40. CPPUNIT_TEST(testToLower);
  41. CPPUNIT_TEST(testUppercase);
  42. CPPUNIT_TEST(testLowercase);
  43. CPPUNIT_TEST(testPercentDecode);
  44. CPPUNIT_TEST(testGetRealSize);
  45. CPPUNIT_TEST(testAbbrevSize);
  46. CPPUNIT_TEST(testToStream);
  47. CPPUNIT_TEST(testIsNumber);
  48. CPPUNIT_TEST(testIsLowercase);
  49. CPPUNIT_TEST(testIsUppercase);
  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(testPercentEncodeMini);
  63. CPPUNIT_TEST(testHtmlEscape);
  64. CPPUNIT_TEST(testJoinPath);
  65. CPPUNIT_TEST(testParseIndexPath);
  66. CPPUNIT_TEST(testCreateIndexPaths);
  67. CPPUNIT_TEST(testGenerateRandomData);
  68. CPPUNIT_TEST(testFromHex);
  69. CPPUNIT_TEST(testParsePrioritizePieceRange);
  70. CPPUNIT_TEST(testApplyDir);
  71. CPPUNIT_TEST(testFixTaintedBasename);
  72. CPPUNIT_TEST(testIsNumericHost);
  73. CPPUNIT_TEST(testDetectDirTraversal);
  74. CPPUNIT_TEST(testEscapePath);
  75. CPPUNIT_TEST(testInSameCidrBlock);
  76. CPPUNIT_TEST(testIsUtf8String);
  77. CPPUNIT_TEST(testNextParam);
  78. CPPUNIT_TEST(testNoProxyDomainMatch);
  79. CPPUNIT_TEST(testInPrivateAddress);
  80. CPPUNIT_TEST(testSecfmt);
  81. CPPUNIT_TEST(testTlsHostnameMatch);
  82. CPPUNIT_TEST_SUITE_END();
  83. private:
  84. public:
  85. void setUp() {
  86. }
  87. void testStrip();
  88. void testStripIter();
  89. void testLstripIter();
  90. void testLstripIter_char();
  91. void testDivide();
  92. void testSplit();
  93. void testSplitIter();
  94. void testSplitIterM();
  95. void testStreq();
  96. void testStrieq();
  97. void testStrifind();
  98. void testEndsWith();
  99. void testIendsWith();
  100. void testReplace();
  101. void testStartsWith();
  102. void testIstartsWith();
  103. // may be moved to other helper class in the future.
  104. void testGetContentDispositionFilename();
  105. void testParseContentDisposition();
  106. void testToUpper();
  107. void testToLower();
  108. void testUppercase();
  109. void testLowercase();
  110. void testPercentDecode();
  111. void testGetRealSize();
  112. void testAbbrevSize();
  113. void testToStream();
  114. void testIsNumber();
  115. void testIsLowercase();
  116. void testIsUppercase();
  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 testPercentEncodeMini();
  130. void testHtmlEscape();
  131. void testJoinPath();
  132. void testParseIndexPath();
  133. void testCreateIndexPaths();
  134. void testGenerateRandomData();
  135. void testFromHex();
  136. void testParsePrioritizePieceRange();
  137. void testApplyDir();
  138. void testFixTaintedBasename();
  139. void testIsNumericHost();
  140. void testDetectDirTraversal();
  141. void testEscapePath();
  142. void testInSameCidrBlock();
  143. void testIsUtf8String();
  144. void testNextParam();
  145. void testNoProxyDomainMatch();
  146. void testInPrivateAddress();
  147. void testSecfmt();
  148. void testTlsHostnameMatch();
  149. };
  150. CPPUNIT_TEST_SUITE_REGISTRATION( UtilTest );
  151. void UtilTest::testStrip()
  152. {
  153. std::string str1 = "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. CPPUNIT_ASSERT_EQUAL(str1, util::strip(" aria2 "));
  159. std::string str2 = "aria2 debut";
  160. CPPUNIT_ASSERT_EQUAL(str2, util::strip("aria2 debut"));
  161. CPPUNIT_ASSERT_EQUAL(str2, util::strip(" aria2 debut "));
  162. std::string str3 = "";
  163. CPPUNIT_ASSERT_EQUAL(str3, util::strip(""));
  164. CPPUNIT_ASSERT_EQUAL(str3, util::strip(" "));
  165. CPPUNIT_ASSERT_EQUAL(str3, util::strip(" "));
  166. std::string str4 = "A";
  167. CPPUNIT_ASSERT_EQUAL(str4, util::strip("A"));
  168. CPPUNIT_ASSERT_EQUAL(str4, util::strip(" A "));
  169. CPPUNIT_ASSERT_EQUAL(str4, util::strip(" A "));
  170. }
  171. void UtilTest::testStripIter()
  172. {
  173. Scip p;
  174. std::string str1 = "aria2";
  175. std::string s = "aria2";
  176. p = util::stripIter(s.begin(), s.end());
  177. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  178. s = " aria2";
  179. p = util::stripIter(s.begin(), s.end());
  180. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  181. s = "aria2 ";
  182. p = util::stripIter(s.begin(), s.end());
  183. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  184. s = " aria2 ";
  185. p = util::stripIter(s.begin(), s.end());
  186. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  187. s = " aria2 ";
  188. p = util::stripIter(s.begin(), s.end());
  189. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  190. std::string str2 = "aria2 debut";
  191. s = "aria2 debut";
  192. p = util::stripIter(s.begin(), s.end());
  193. CPPUNIT_ASSERT_EQUAL(str2, std::string(p.first, p.second));
  194. s = " aria2 debut ";
  195. p = util::stripIter(s.begin(), s.end());
  196. CPPUNIT_ASSERT_EQUAL(str2, std::string(p.first, p.second));
  197. std::string str3 = "";
  198. s = "";
  199. p = util::stripIter(s.begin(), s.end());
  200. CPPUNIT_ASSERT_EQUAL(str3, std::string(p.first, p.second));
  201. s = " ";
  202. p = util::stripIter(s.begin(), s.end());
  203. CPPUNIT_ASSERT_EQUAL(str3, std::string(p.first, p.second));
  204. s = " ";
  205. p = util::stripIter(s.begin(), s.end());
  206. CPPUNIT_ASSERT_EQUAL(str3, std::string(p.first, p.second));
  207. std::string str4 = "A";
  208. s = "A";
  209. p = util::stripIter(s.begin(), s.end());
  210. CPPUNIT_ASSERT_EQUAL(str4, std::string(p.first, p.second));
  211. s = " A ";
  212. p = util::stripIter(s.begin(), s.end());
  213. CPPUNIT_ASSERT_EQUAL(str4, std::string(p.first, p.second));
  214. s = " A ";
  215. p = util::stripIter(s.begin(), s.end());
  216. CPPUNIT_ASSERT_EQUAL(str4, std::string(p.first, p.second));
  217. }
  218. void UtilTest::testLstripIter()
  219. {
  220. std::string::iterator r;
  221. std::string s = "foo";
  222. r = util::lstripIter(s.begin(), s.end());
  223. CPPUNIT_ASSERT_EQUAL(std::string("foo"), std::string(r, s.end()));
  224. s = " foo bar ";
  225. r = util::lstripIter(s.begin(), s.end());
  226. CPPUNIT_ASSERT_EQUAL(std::string("foo bar "), std::string(r, s.end()));
  227. s = "f";
  228. r = util::lstripIter(s.begin(), s.end());
  229. CPPUNIT_ASSERT_EQUAL(std::string("f"), std::string(r, s.end()));
  230. s = "foo ";
  231. r = util::lstripIter(s.begin(), s.end());
  232. CPPUNIT_ASSERT_EQUAL(std::string("foo "), std::string(r, s.end()));
  233. }
  234. void UtilTest::testLstripIter_char()
  235. {
  236. std::string::iterator r;
  237. std::string s = "foo";
  238. r = util::lstripIter(s.begin(), s.end(), '$');
  239. CPPUNIT_ASSERT_EQUAL(std::string("foo"), std::string(r, s.end()));
  240. s = "$$foo$bar$$";
  241. r = util::lstripIter(s.begin(), s.end(), '$');
  242. CPPUNIT_ASSERT_EQUAL(std::string("foo$bar$$"), std::string(r, s.end()));
  243. s = "f";
  244. r = util::lstripIter(s.begin(), s.end(), '$');
  245. CPPUNIT_ASSERT_EQUAL(std::string("f"), std::string(r, s.end()));
  246. s = "foo$$";
  247. r = util::lstripIter(s.begin(), s.end(), '$');
  248. CPPUNIT_ASSERT_EQUAL(std::string("foo$$"), std::string(r, s.end()));
  249. }
  250. void UtilTest::testDivide() {
  251. std::string s = "name=value";
  252. auto p1 = util::divide(std::begin(s), std::end(s), '=');
  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. p1 = util::divide(std::begin(s), std::end(s), '=');
  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. p1 = util::divide(std::begin(s), std::end(s), '=');
  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. p1 = util::divide(std::begin(s), std::end(s), '=');
  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. p1 = util::divide(std::begin(s), std::end(s), '=');
  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 val;
  695. val = "attachment; filename=\"aria2.tar.bz2\"";
  696. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"),
  697. util::getContentDispositionFilename(val));
  698. val = "attachment; filename=\"\"";
  699. CPPUNIT_ASSERT_EQUAL(std::string(""),
  700. util::getContentDispositionFilename(val));
  701. val = "attachment; filename=\"";
  702. CPPUNIT_ASSERT_EQUAL(std::string(""),
  703. util::getContentDispositionFilename(val));
  704. val = "attachment; filename= \" aria2.tar.bz2 \"";
  705. CPPUNIT_ASSERT_EQUAL(std::string(" aria2.tar.bz2 "),
  706. util::getContentDispositionFilename(val));
  707. val = "attachment; filename=dir/file";
  708. CPPUNIT_ASSERT_EQUAL(std::string(""),
  709. util::getContentDispositionFilename(val));
  710. val = "attachment; filename=dir\\file";
  711. CPPUNIT_ASSERT_EQUAL(std::string(""),
  712. util::getContentDispositionFilename(val));
  713. val = "attachment; filename=\"dir/file\"";
  714. CPPUNIT_ASSERT_EQUAL(std::string(""),
  715. util::getContentDispositionFilename(val));
  716. val = "attachment; filename=\"dir\\\\file\"";
  717. CPPUNIT_ASSERT_EQUAL(std::string(""),
  718. util::getContentDispositionFilename(val));
  719. val = "attachment; filename=\"/etc/passwd\"";
  720. CPPUNIT_ASSERT_EQUAL(std::string(""),
  721. util::getContentDispositionFilename(val));
  722. val = "attachment; filename=\"..\"";
  723. CPPUNIT_ASSERT_EQUAL(std::string(""),
  724. util::getContentDispositionFilename(val));
  725. val = "attachment; filename=..";
  726. CPPUNIT_ASSERT_EQUAL(std::string(""),
  727. util::getContentDispositionFilename(val));
  728. // Unescaping %2E%2E%2F produces "../". But since we won't unescape,
  729. // we just accept it as is.
  730. val = "attachment; filename=\"%2E%2E%2Ffoo.html\"";
  731. CPPUNIT_ASSERT_EQUAL(std::string("%2E%2E%2Ffoo.html"),
  732. util::getContentDispositionFilename(val));
  733. // iso-8859-1 string will be converted to utf-8.
  734. val = "attachment; filename*=iso-8859-1''foo-%E4.html";
  735. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  736. util::getContentDispositionFilename(val));
  737. val = "attachment; filename*= UTF-8''foo-%c3%a4.html";
  738. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  739. util::getContentDispositionFilename(val));
  740. // iso-8859-1 string will be converted to utf-8.
  741. val = "attachment; filename=\"foo-%E4.html\"";
  742. val = util::percentDecode(val.begin(), val.end());
  743. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  744. util::getContentDispositionFilename(val));
  745. }
  746. void UtilTest::testParseContentDisposition() {
  747. char dest[1024];
  748. size_t destlen = sizeof(dest);
  749. const char *cs;
  750. size_t cslen;
  751. std::string val;
  752. // test cases from http://greenbytes.de/tech/tc2231/
  753. // inlonly
  754. val = "inline";
  755. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition
  756. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  757. // inlonlyquoted
  758. val = "\"inline\"";
  759. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  760. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  761. // inlwithasciifilename
  762. val = "inline; filename=\"foo.html\"";
  763. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  764. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  765. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  766. std::string(&dest[0], &dest[8]));
  767. // inlwithfnattach
  768. val = "inline; filename=\"Not an attachment!\"";
  769. CPPUNIT_ASSERT_EQUAL((ssize_t)18, util::parse_content_disposition
  770. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  771. CPPUNIT_ASSERT_EQUAL(std::string("Not an attachment!"),
  772. std::string(&dest[0], &dest[18]));
  773. // inlwithasciifilenamepdf
  774. val = "inline; filename=\"foo.pdf\"";
  775. CPPUNIT_ASSERT_EQUAL((ssize_t)7, util::parse_content_disposition
  776. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  777. CPPUNIT_ASSERT_EQUAL(std::string("foo.pdf"),
  778. std::string(&dest[0], &dest[7]));
  779. // attwithasciifilename25
  780. val = "attachment; filename=\"0000000000111111111122222\"";
  781. CPPUNIT_ASSERT_EQUAL((ssize_t)25, util::parse_content_disposition
  782. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  783. CPPUNIT_ASSERT_EQUAL(std::string("0000000000111111111122222"),
  784. std::string(&dest[0], &dest[25]));
  785. // attwithasciifilename35
  786. val = "attachment; filename=\"00000000001111111111222222222233333\"";
  787. CPPUNIT_ASSERT_EQUAL((ssize_t)35, util::parse_content_disposition
  788. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  789. CPPUNIT_ASSERT_EQUAL(std::string("00000000001111111111222222222233333"),
  790. std::string(&dest[0], &dest[35]));
  791. // attwithasciifnescapedchar
  792. val = "attachment; filename=\"f\\oo.html\"";
  793. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  794. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  795. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  796. std::string(&dest[0], &dest[8]));
  797. // attwithasciifnescapedquote
  798. val = "attachment; filename=\"\\\"quoting\\\" tested.html\"";
  799. CPPUNIT_ASSERT_EQUAL((ssize_t)21, util::parse_content_disposition
  800. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  801. CPPUNIT_ASSERT_EQUAL(std::string("\"quoting\" tested.html"),
  802. std::string(&dest[0], &dest[21]));
  803. // attwithquotedsemicolon
  804. val = "attachment; filename=\"Here's a semicolon;.html\"";
  805. CPPUNIT_ASSERT_EQUAL((ssize_t)24, util::parse_content_disposition
  806. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  807. CPPUNIT_ASSERT_EQUAL(std::string("Here's a semicolon;.html"),
  808. std::string(&dest[0], &dest[24]));
  809. // attwithfilenameandextparam
  810. val = "attachment; foo=\"bar\"; filename=\"foo.html\"";
  811. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  812. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  813. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  814. std::string(&dest[0], &dest[8]));
  815. // attwithfilenameandextparamescaped
  816. val = "attachment; foo=\"\\\"\\\\\";filename=\"foo.html\"";
  817. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  818. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  819. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  820. std::string(&dest[0], &dest[8]));
  821. // attwithasciifilenameucase
  822. val = "attachment; FILENAME=\"foo.html\"";
  823. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  824. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  825. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  826. std::string(&dest[0], &dest[8]));
  827. // attwithasciifilenamenq
  828. val = "attachment; filename=foo.html";
  829. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  830. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  831. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  832. std::string(&dest[0], &dest[8]));
  833. // attwithtokfncommanq
  834. val = "attachment; filename=foo,bar.html";
  835. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  836. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  837. // attwithasciifilenamenqs
  838. val = "attachment; filename=foo.html ;";
  839. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  840. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  841. // attemptyparam
  842. val = "attachment; ;filename=foo";
  843. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  844. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  845. // attwithasciifilenamenqws
  846. val = "attachment; filename=foo bar.html";
  847. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  848. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  849. // attwithfntokensq
  850. val = "attachment; filename='foo.bar'";
  851. CPPUNIT_ASSERT_EQUAL((ssize_t)9, util::parse_content_disposition
  852. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  853. CPPUNIT_ASSERT_EQUAL(std::string("'foo.bar'"),
  854. std::string(&dest[0], &dest[9]));
  855. // attwithisofnplain
  856. // attachment; filename="foo-ä.html"
  857. val = "attachment; filename=\"foo-%E4.html\"";
  858. val = util::percentDecode(val.begin(), val.end());
  859. CPPUNIT_ASSERT_EQUAL((ssize_t)10, util::parse_content_disposition
  860. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  861. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  862. util::iso8859p1ToUtf8(std::string(&dest[0], &dest[10])));
  863. // attwithutf8fnplain
  864. // attachment; filename="foo-ä.html"
  865. val = "attachment; filename=\"foo-%C3%A4.html\"";
  866. val = util::percentDecode(val.begin(), val.end());
  867. CPPUNIT_ASSERT_EQUAL((ssize_t)11, util::parse_content_disposition
  868. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  869. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  870. util::iso8859p1ToUtf8(std::string(&dest[0], &dest[11])));
  871. // attwithfnrawpctenca
  872. val = "attachment; filename=\"foo-%41.html\"";
  873. CPPUNIT_ASSERT_EQUAL((ssize_t)12, util::parse_content_disposition
  874. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  875. CPPUNIT_ASSERT_EQUAL(std::string("foo-%41.html"),
  876. std::string(&dest[0], &dest[12]));
  877. // attwithfnusingpct
  878. val = "attachment; filename=\"50%.html\"";
  879. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  880. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  881. CPPUNIT_ASSERT_EQUAL(std::string("50%.html"),
  882. std::string(&dest[0], &dest[8]));
  883. // attwithfnrawpctencaq
  884. val = "attachment; filename=\"foo-%\\41.html\"";
  885. CPPUNIT_ASSERT_EQUAL((ssize_t)12, util::parse_content_disposition
  886. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  887. CPPUNIT_ASSERT_EQUAL(std::string("foo-%41.html"),
  888. std::string(&dest[0], &dest[12]));
  889. // attwithnamepct
  890. val = "attachment; name=\"foo-%41.html\"";
  891. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition
  892. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  893. // attwithfilenamepctandiso
  894. // attachment; filename="ä-%41.html"
  895. val = "attachment; filename=\"%E4-%2541.html\"";
  896. val = util::percentDecode(val.begin(), val.end());
  897. CPPUNIT_ASSERT_EQUAL((ssize_t)10, util::parse_content_disposition
  898. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  899. CPPUNIT_ASSERT_EQUAL(std::string("ä-%41.html"),
  900. util::iso8859p1ToUtf8(std::string(&dest[0], &dest[10])));
  901. // attwithfnrawpctenclong
  902. val = "attachment; filename=\"foo-%c3%a4-%e2%82%ac.html\"";
  903. CPPUNIT_ASSERT_EQUAL((ssize_t)25, util::parse_content_disposition
  904. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  905. CPPUNIT_ASSERT_EQUAL(std::string("foo-%c3%a4-%e2%82%ac.html"),
  906. std::string(&dest[0], &dest[25]));
  907. // attwithasciifilenamews1
  908. val = "attachment; filename =\"foo.html\"";
  909. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  910. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  911. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  912. std::string(&dest[0], &dest[8]));
  913. // attwith2filenames
  914. val = "attachment; filename=\"foo.html\"; filename=\"bar.html\"";
  915. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  916. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  917. // attfnbrokentoken
  918. val = "attachment; filename=foo[1](2).html";
  919. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  920. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  921. // attfnbrokentokeniso
  922. val = "attachment; filename=foo-%E4.html";
  923. val = util::percentDecode(val.begin(), val.end());
  924. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  925. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  926. // attfnbrokentokenutf
  927. // attachment; filename=foo-ä.html
  928. val = "attachment; filename=foo-ä.html";
  929. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  930. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  931. // attmissingdisposition
  932. val = "filename=foo.html";
  933. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  934. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  935. // attmissingdisposition2
  936. val = "x=y; filename=foo.html";
  937. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  938. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  939. // attmissingdisposition3
  940. val = "\"foo; filename=bar;baz\"; filename=qux";
  941. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  942. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  943. // attmissingdisposition4
  944. val = "filename=foo.html, filename=bar.html";
  945. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  946. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  947. // emptydisposition
  948. val = "; filename=foo.html";
  949. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  950. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  951. // doublecolon
  952. val = ": inline; attachment; filename=foo.html";
  953. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  954. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  955. // attandinline
  956. val = "inline; attachment; filename=foo.html";
  957. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  958. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  959. // attandinline2
  960. val = "attachment; inline; filename=foo.html";
  961. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  962. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  963. // attbrokenquotedfn
  964. val = "attachment; filename=\"foo.html\".txt";
  965. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  966. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  967. // attbrokenquotedfn2
  968. val = "attachment; filename=\"bar";
  969. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  970. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  971. // attbrokenquotedfn3
  972. val = "attachment; filename=foo\"bar;baz\"qux";
  973. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  974. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  975. // attmultinstances
  976. val = "attachment; filename=foo.html, attachment; filename=bar.html";
  977. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  978. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  979. // attmissingdelim
  980. val = "attachment; foo=foo filename=bar";
  981. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  982. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  983. // attmissingdelim2
  984. val = "attachment; filename=bar foo=foo ";
  985. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  986. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  987. // attmissingdelim3
  988. val = "attachment filename=bar";
  989. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  990. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  991. // attreversed
  992. val = "filename=foo.html; attachment";
  993. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  994. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  995. // attconfusedparam
  996. val = "attachment; xfilename=foo.html";
  997. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition
  998. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  999. // attabspath
  1000. val = "attachment; filename=\"/foo.html\"";
  1001. CPPUNIT_ASSERT_EQUAL((ssize_t)9, util::parse_content_disposition
  1002. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1003. CPPUNIT_ASSERT_EQUAL(std::string("/foo.html"),
  1004. std::string(&dest[0], &dest[9]));
  1005. // attabspathwin
  1006. val = "attachment; filename=\"\\\\foo.html\"";
  1007. CPPUNIT_ASSERT_EQUAL((ssize_t)9, util::parse_content_disposition
  1008. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1009. CPPUNIT_ASSERT_EQUAL(std::string("\\foo.html"),
  1010. std::string(&dest[0], &dest[9]));
  1011. // attcdate
  1012. val = "attachment; creation-date=\"Wed, 12 Feb 1997 16:29:51 -0500\"";
  1013. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition
  1014. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1015. // dispext
  1016. val = "foobar";
  1017. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition
  1018. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1019. // dispextbadfn
  1020. val = "attachment; example=\"filename=example.txt\"";
  1021. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition
  1022. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1023. // attwithisofn2231iso
  1024. val = "attachment; filename*=iso-8859-1''foo-%E4.html";
  1025. CPPUNIT_ASSERT_EQUAL((ssize_t)10, util::parse_content_disposition
  1026. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1027. CPPUNIT_ASSERT_EQUAL(std::string("iso-8859-1"), std::string(cs, cslen));
  1028. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  1029. util::iso8859p1ToUtf8(std::string(&dest[0], &dest[10])));
  1030. // attwithfn2231utf8
  1031. val = "attachment; filename*=UTF-8''foo-%c3%a4-%e2%82%ac.html";
  1032. CPPUNIT_ASSERT_EQUAL((ssize_t)15, util::parse_content_disposition
  1033. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1034. CPPUNIT_ASSERT_EQUAL(std::string("UTF-8"), std::string(cs, cslen));
  1035. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä-€.html"),
  1036. std::string(&dest[0], &dest[15]));
  1037. // attwithfn2231noc
  1038. val = "attachment; filename*=''foo-%c3%a4-%e2%82%ac.html";
  1039. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  1040. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1041. // attwithfn2231utf8comp
  1042. val = "attachment; filename*=UTF-8''foo-a%cc%88.html";
  1043. CPPUNIT_ASSERT_EQUAL((ssize_t)12, util::parse_content_disposition
  1044. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1045. val = "foo-a%cc%88.html";
  1046. CPPUNIT_ASSERT_EQUAL(std::string(util::percentDecode(val.begin(),
  1047. val.end())),
  1048. std::string(&dest[0], &dest[12]));
  1049. // attwithfn2231utf8-bad
  1050. val = "attachment; filename*=iso-8859-1''foo-%c3%a4-%e2%82%ac.html";
  1051. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  1052. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1053. // attwithfn2231iso-bad
  1054. val = "attachment; filename*=utf-8''foo-%E4.html";
  1055. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  1056. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1057. // attwithfn2231ws1
  1058. val = "attachment; filename *=UTF-8''foo-%c3%a4.html";
  1059. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  1060. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1061. // attwithfn2231ws2
  1062. val = "attachment; filename*= UTF-8''foo-%c3%a4.html";
  1063. CPPUNIT_ASSERT_EQUAL((ssize_t)11, util::parse_content_disposition
  1064. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1065. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  1066. std::string(&dest[0], &dest[11]));
  1067. // attwithfn2231ws3
  1068. val = "attachment; filename* =UTF-8''foo-%c3%a4.html";
  1069. CPPUNIT_ASSERT_EQUAL((ssize_t)11, util::parse_content_disposition
  1070. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1071. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  1072. std::string(&dest[0], &dest[11]));
  1073. // attwithfn2231quot
  1074. val = "attachment; filename*=\"UTF-8''foo-%c3%a4.html\"";
  1075. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  1076. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1077. // attwithfn2231quot2
  1078. val = "attachment; filename*=\"foo%20bar.html\"";
  1079. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  1080. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1081. // attwithfn2231singleqmissing
  1082. val = "attachment; filename*=UTF-8'foo-%c3%a4.html";
  1083. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  1084. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1085. // attwithfn2231nbadpct1
  1086. val = "attachment; filename*=UTF-8''foo%";
  1087. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  1088. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1089. // attwithfn2231nbadpct2
  1090. val = "attachment; filename*=UTF-8''f%oo.html";
  1091. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  1092. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1093. // attwithfn2231dpct
  1094. val = "attachment; filename*=UTF-8''A-%2541.html";
  1095. CPPUNIT_ASSERT_EQUAL((ssize_t)10, util::parse_content_disposition
  1096. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1097. CPPUNIT_ASSERT_EQUAL(std::string("A-%41.html"),
  1098. std::string(&dest[0], &dest[10]));
  1099. // attwithfn2231abspathdisguised
  1100. val = "attachment; filename*=UTF-8''%5cfoo.html";
  1101. CPPUNIT_ASSERT_EQUAL((ssize_t)9, util::parse_content_disposition
  1102. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1103. CPPUNIT_ASSERT_EQUAL(std::string("\\foo.html"),
  1104. std::string(&dest[0], &dest[9]));
  1105. // attfnboth
  1106. val = "attachment; filename=\"foo-ae.html\"; filename*=UTF-8''foo-%c3%a4.html";
  1107. CPPUNIT_ASSERT_EQUAL((ssize_t)11, util::parse_content_disposition
  1108. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1109. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  1110. std::string(&dest[0], &dest[11]));
  1111. // attfnboth2
  1112. val = "attachment; filename*=UTF-8''foo-%c3%a4.html; filename=\"foo-ae.html\"";
  1113. CPPUNIT_ASSERT_EQUAL((ssize_t)11, util::parse_content_disposition
  1114. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1115. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  1116. std::string(&dest[0], &dest[11]));
  1117. // attfnboth3
  1118. val = "attachment; filename*0*=ISO-8859-15''euro-sign%3d%a4; filename*=ISO-8859-1''currency-sign%3d%a4";
  1119. CPPUNIT_ASSERT_EQUAL((ssize_t)15, util::parse_content_disposition
  1120. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1121. CPPUNIT_ASSERT_EQUAL(std::string("ISO-8859-1"), std::string(cs, cslen));
  1122. CPPUNIT_ASSERT_EQUAL(std::string("currency-sign=¤"),
  1123. util::iso8859p1ToUtf8(std::string(&dest[0], &dest[15])));
  1124. // attnewandfn
  1125. val = "attachment; foobar=x; filename=\"foo.html\"";
  1126. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  1127. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1128. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1129. std::string(&dest[0], &dest[8]));
  1130. // attrfc2047token
  1131. val = "attachment; filename==?ISO-8859-1?Q?foo-=E4.html?=";
  1132. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  1133. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1134. // attrfc2047quoted
  1135. val = "attachment; filename=\"=?ISO-8859-1?Q?foo-=E4.html?=\"";
  1136. CPPUNIT_ASSERT_EQUAL((ssize_t)29, util::parse_content_disposition
  1137. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1138. CPPUNIT_ASSERT_EQUAL(std::string("=?ISO-8859-1?Q?foo-=E4.html?="),
  1139. std::string(&dest[0], &dest[29]));
  1140. // aria2 original testcases
  1141. // zero-length filename. token cannot be empty, so this is invalid.
  1142. val = "attachment; filename=";
  1143. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  1144. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1145. // zero-length filename. quoted-string can be empty string, so this
  1146. // is ok.
  1147. val = "attachment; filename=\"\"";
  1148. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition
  1149. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1150. // empty value is not allowed
  1151. val = "attachment; filename=;";
  1152. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  1153. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1154. // / is not valid char in token.
  1155. val = "attachment; filename=dir/file";
  1156. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition
  1157. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1158. // value-chars is *(pct-encoded / attr-char), so empty string is
  1159. // allowed.
  1160. val = "attachment; filename*=UTF-8''";
  1161. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition
  1162. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1163. CPPUNIT_ASSERT_EQUAL(std::string("UTF-8"), std::string(cs, cslen));
  1164. val = "attachment; filename*=UTF-8''; filename=foo";
  1165. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition
  1166. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1167. CPPUNIT_ASSERT_EQUAL(std::string("UTF-8"), std::string(cs, cslen));
  1168. val = "attachment; filename*=UTF-8'' ; filename=foo";
  1169. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition
  1170. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1171. CPPUNIT_ASSERT_EQUAL(std::string("UTF-8"), std::string(cs, cslen));
  1172. // with language
  1173. val = "attachment; filename*=UTF-8'japanese'konnichiwa";
  1174. CPPUNIT_ASSERT_EQUAL((ssize_t)10, util::parse_content_disposition
  1175. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1176. CPPUNIT_ASSERT_EQUAL(std::string("konnichiwa"),
  1177. std::string(&dest[0], &dest[10]));
  1178. // lws before and after "="
  1179. val = "attachment; filename = foo.html";
  1180. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  1181. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1182. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1183. std::string(&dest[0], &dest[8]));
  1184. // lws before and after "=" with quoted-string
  1185. val = "attachment; filename = \"foo.html\"";
  1186. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  1187. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1188. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1189. std::string(&dest[0], &dest[8]));
  1190. // lws after parm
  1191. val = "attachment; filename=foo.html ";
  1192. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  1193. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1194. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1195. std::string(&dest[0], &dest[8]));
  1196. val = "attachment; filename=foo.html ; hello=world";
  1197. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  1198. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1199. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1200. std::string(&dest[0], &dest[8]));
  1201. val = "attachment; filename=\"foo.html\" ";
  1202. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  1203. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1204. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1205. std::string(&dest[0], &dest[8]));
  1206. val = "attachment; filename=\"foo.html\" ; hello=world";
  1207. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  1208. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1209. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1210. std::string(&dest[0], &dest[8]));
  1211. val = "attachment; filename*=UTF-8''foo.html ; hello=world";
  1212. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition
  1213. (dest, destlen, &cs, &cslen, val.c_str(), val.size()));
  1214. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1215. std::string(&dest[0], &dest[8]));
  1216. }
  1217. class Printer {
  1218. public:
  1219. template<class T>
  1220. void operator()(T t) {
  1221. std::cerr << t << ", ";
  1222. }
  1223. };
  1224. void UtilTest::testToUpper() {
  1225. std::string src = "608cabc0f2fa18c260cafd974516865c772363d5";
  1226. std::string upp = "608CABC0F2FA18C260CAFD974516865C772363D5";
  1227. CPPUNIT_ASSERT_EQUAL(upp, util::toUpper(src));
  1228. }
  1229. void UtilTest::testToLower() {
  1230. std::string src = "608CABC0F2FA18C260CAFD974516865C772363D5";
  1231. std::string upp = "608cabc0f2fa18c260cafd974516865c772363d5";
  1232. CPPUNIT_ASSERT_EQUAL(upp, util::toLower(src));
  1233. }
  1234. void UtilTest::testUppercase() {
  1235. std::string src = "608cabc0f2fa18c260cafd974516865c772363d5";
  1236. std::string ans = "608CABC0F2FA18C260CAFD974516865C772363D5";
  1237. util::uppercase(src);
  1238. CPPUNIT_ASSERT_EQUAL(ans, src);
  1239. }
  1240. void UtilTest::testLowercase() {
  1241. std::string src = "608CABC0F2FA18C260CAFD974516865C772363D5";
  1242. std::string ans = "608cabc0f2fa18c260cafd974516865c772363d5";
  1243. util::lowercase(src);
  1244. CPPUNIT_ASSERT_EQUAL(ans, src);
  1245. }
  1246. void UtilTest::testPercentDecode() {
  1247. std::string src = "http://aria2.sourceforge.net/aria2%200.7.0%20docs.html";
  1248. CPPUNIT_ASSERT_EQUAL(std::string("http://aria2.sourceforge.net/aria2 0.7.0 docs.html"),
  1249. util::percentDecode(src.begin(), src.end()));
  1250. std::string src2 = "aria2+aria2";
  1251. CPPUNIT_ASSERT_EQUAL(std::string("aria2+aria2"),
  1252. util::percentDecode(src2.begin(), src2.end()));
  1253. std::string src3 = "%5t%20";
  1254. CPPUNIT_ASSERT_EQUAL(std::string("%5t "),
  1255. util::percentDecode(src3.begin(), src3.end()));
  1256. std::string src4 = "%";
  1257. CPPUNIT_ASSERT_EQUAL(std::string("%"),
  1258. util::percentDecode(src4.begin(), src4.end()));
  1259. std::string src5 = "%3";
  1260. CPPUNIT_ASSERT_EQUAL(std::string("%3"),
  1261. util::percentDecode(src5.begin(), src5.end()));
  1262. std::string src6 = "%2f";
  1263. CPPUNIT_ASSERT_EQUAL(std::string("/"),
  1264. util::percentDecode(src6.begin(), src6.end()));
  1265. }
  1266. void UtilTest::testGetRealSize()
  1267. {
  1268. CPPUNIT_ASSERT_EQUAL((int64_t)4294967296LL, util::getRealSize("4096M"));
  1269. CPPUNIT_ASSERT_EQUAL((int64_t)1024, util::getRealSize("1K"));
  1270. CPPUNIT_ASSERT_EQUAL((int64_t)4294967296LL, util::getRealSize("4096m"));
  1271. CPPUNIT_ASSERT_EQUAL((int64_t)1024, util::getRealSize("1k"));
  1272. try {
  1273. util::getRealSize("");
  1274. CPPUNIT_FAIL("exception must be thrown.");
  1275. } catch(Exception& e) {
  1276. std::cerr << e.stackTrace();
  1277. }
  1278. try {
  1279. util::getRealSize("foo");
  1280. CPPUNIT_FAIL("exception must be thrown.");
  1281. } catch(Exception& e) {
  1282. std::cerr << e.stackTrace();
  1283. }
  1284. try {
  1285. util::getRealSize("-1");
  1286. CPPUNIT_FAIL("exception must be thrown.");
  1287. } catch(Exception& e) {
  1288. std::cerr << e.stackTrace();
  1289. }
  1290. try {
  1291. util::getRealSize("9223372036854775807K");
  1292. CPPUNIT_FAIL("exception must be thrown.");
  1293. } catch(Exception& e) {
  1294. std::cerr << e.stackTrace();
  1295. }
  1296. try {
  1297. util::getRealSize("9223372036854775807M");
  1298. CPPUNIT_FAIL("exception must be thrown.");
  1299. } catch(Exception& e) {
  1300. std::cerr << e.stackTrace();
  1301. }
  1302. }
  1303. void UtilTest::testAbbrevSize()
  1304. {
  1305. CPPUNIT_ASSERT_EQUAL(std::string("8,589,934,591Gi"),
  1306. util::abbrevSize(9223372036854775807LL));
  1307. CPPUNIT_ASSERT_EQUAL(std::string("4.0Gi"), util::abbrevSize(4294967296LL));
  1308. CPPUNIT_ASSERT_EQUAL(std::string("1.0Ki"), util::abbrevSize(1024));
  1309. CPPUNIT_ASSERT_EQUAL(std::string("0.9Ki"), util::abbrevSize(1023));
  1310. CPPUNIT_ASSERT_EQUAL(std::string("511"), util::abbrevSize(511));
  1311. CPPUNIT_ASSERT_EQUAL(std::string("0"), util::abbrevSize(0));
  1312. CPPUNIT_ASSERT_EQUAL(std::string("1.1Ki"), util::abbrevSize(1127));
  1313. CPPUNIT_ASSERT_EQUAL(std::string("1.5Mi"), util::abbrevSize(1572864));
  1314. }
  1315. void UtilTest::testToStream()
  1316. {
  1317. std::ostringstream os;
  1318. std::shared_ptr<FileEntry> f1(new FileEntry("aria2.tar.bz2", 12300, 0));
  1319. std::shared_ptr<FileEntry> f2(new FileEntry("aria2.txt", 556, 0));
  1320. std::deque<std::shared_ptr<FileEntry> > entries;
  1321. entries.push_back(f1);
  1322. entries.push_back(f2);
  1323. const char* filename = A2_TEST_OUT_DIR"/aria2_UtilTest_testToStream";
  1324. BufferedFile fp(filename, BufferedFile::WRITE);
  1325. util::toStream(entries.begin(), entries.end(), fp);
  1326. fp.close();
  1327. CPPUNIT_ASSERT_EQUAL(
  1328. std::string("Files:\n"
  1329. "idx|path/length\n"
  1330. "===+===========================================================================\n"
  1331. " 1|aria2.tar.bz2\n"
  1332. " |12KiB (12,300)\n"
  1333. "---+---------------------------------------------------------------------------\n"
  1334. " 2|aria2.txt\n"
  1335. " |556B (556)\n"
  1336. "---+---------------------------------------------------------------------------\n"),
  1337. readFile(filename));
  1338. }
  1339. void UtilTest::testIsNumber()
  1340. {
  1341. std::string s = "000";
  1342. CPPUNIT_ASSERT_EQUAL(true, util::isNumber(s.begin(),s.end()));
  1343. s = "a";
  1344. CPPUNIT_ASSERT_EQUAL(false, util::isNumber(s.begin(), s.end()));
  1345. s = "0a";
  1346. CPPUNIT_ASSERT_EQUAL(false, util::isNumber(s.begin(), s.end()));
  1347. s = "";
  1348. CPPUNIT_ASSERT_EQUAL(false, util::isNumber(s.begin(), s.end()));
  1349. s = " ";
  1350. CPPUNIT_ASSERT_EQUAL(false, util::isNumber(s.begin(), s.end()));
  1351. }
  1352. void UtilTest::testIsLowercase()
  1353. {
  1354. std::string s = "alpha";
  1355. CPPUNIT_ASSERT_EQUAL(true, util::isLowercase(s.begin(), s.end()));
  1356. s = "Alpha";
  1357. CPPUNIT_ASSERT_EQUAL(false, util::isLowercase(s.begin(), s.end()));
  1358. s = "1alpha";
  1359. CPPUNIT_ASSERT_EQUAL(false, util::isLowercase(s.begin(), s.end()));
  1360. s = "";
  1361. CPPUNIT_ASSERT_EQUAL(false, util::isLowercase(s.begin(), s.end()));
  1362. s = " ";
  1363. CPPUNIT_ASSERT_EQUAL(false, util::isLowercase(s.begin(), s.end()));
  1364. }
  1365. void UtilTest::testIsUppercase()
  1366. {
  1367. std::string s = "ALPHA";
  1368. CPPUNIT_ASSERT_EQUAL(true, util::isUppercase(s.begin(), s.end()));
  1369. s = "Alpha";
  1370. CPPUNIT_ASSERT_EQUAL(false, util::isUppercase(s.begin(), s.end()));
  1371. s = "1ALPHA";
  1372. CPPUNIT_ASSERT_EQUAL(false, util::isUppercase(s.begin(), s.end()));
  1373. s = "";
  1374. CPPUNIT_ASSERT_EQUAL(false, util::isUppercase(s.begin(), s.end()));
  1375. s = " ";
  1376. CPPUNIT_ASSERT_EQUAL(false, util::isUppercase(s.begin(), s.end()));
  1377. }
  1378. void UtilTest::testMkdirs()
  1379. {
  1380. std::string dir = A2_TEST_OUT_DIR"/aria2-UtilTest-testMkdirs";
  1381. File d(dir);
  1382. if(d.exists()) {
  1383. CPPUNIT_ASSERT(d.remove());
  1384. }
  1385. CPPUNIT_ASSERT(!d.exists());
  1386. util::mkdirs(dir);
  1387. CPPUNIT_ASSERT(d.isDir());
  1388. std::string file = A2_TEST_DIR"/UtilTest.cc";
  1389. File f(file);
  1390. CPPUNIT_ASSERT(f.isFile());
  1391. try {
  1392. util::mkdirs(file);
  1393. CPPUNIT_FAIL("exception must be thrown.");
  1394. } catch(DlAbortEx& ex) {
  1395. std::cerr << ex.stackTrace() << std::endl;
  1396. }
  1397. }
  1398. void UtilTest::testConvertBitfield()
  1399. {
  1400. BitfieldMan srcBitfield(384*1024, 256*1024*256+1);
  1401. BitfieldMan destBitfield(512*1024, srcBitfield.getTotalLength());
  1402. srcBitfield.setAllBit();
  1403. srcBitfield.unsetBit(2);// <- range [768, 1152)
  1404. // which corresponds to the index [1,2] in destBitfield
  1405. util::convertBitfield(&destBitfield, &srcBitfield);
  1406. CPPUNIT_ASSERT_EQUAL(std::string("9fffffffffffffffffffffffffffffff80"),
  1407. util::toHex(destBitfield.getBitfield(),
  1408. destBitfield.getBitfieldLength()));
  1409. }
  1410. void UtilTest::testParseIntSegments()
  1411. {
  1412. {
  1413. auto sgl = util::parseIntSegments("1,3-8,10");
  1414. CPPUNIT_ASSERT(sgl.hasNext());
  1415. CPPUNIT_ASSERT_EQUAL(1, sgl.next());
  1416. CPPUNIT_ASSERT(sgl.hasNext());
  1417. CPPUNIT_ASSERT_EQUAL(3, sgl.next());
  1418. CPPUNIT_ASSERT(sgl.hasNext());
  1419. CPPUNIT_ASSERT_EQUAL(4, sgl.next());
  1420. CPPUNIT_ASSERT(sgl.hasNext());
  1421. CPPUNIT_ASSERT_EQUAL(5, sgl.next());
  1422. CPPUNIT_ASSERT(sgl.hasNext());
  1423. CPPUNIT_ASSERT_EQUAL(6, sgl.next());
  1424. CPPUNIT_ASSERT(sgl.hasNext());
  1425. CPPUNIT_ASSERT_EQUAL(7, sgl.next());
  1426. CPPUNIT_ASSERT(sgl.hasNext());
  1427. CPPUNIT_ASSERT_EQUAL(8, sgl.next());
  1428. CPPUNIT_ASSERT(sgl.hasNext());
  1429. CPPUNIT_ASSERT_EQUAL(10, sgl.next());
  1430. CPPUNIT_ASSERT(!sgl.hasNext());
  1431. CPPUNIT_ASSERT_EQUAL(0, sgl.next());
  1432. }
  1433. {
  1434. auto sgl = util::parseIntSegments(",,,1,,,3,,,");
  1435. CPPUNIT_ASSERT_EQUAL(1, sgl.next());
  1436. CPPUNIT_ASSERT_EQUAL(3, sgl.next());
  1437. CPPUNIT_ASSERT(!sgl.hasNext());
  1438. }
  1439. }
  1440. void UtilTest::testParseIntSegments_invalidRange()
  1441. {
  1442. try {
  1443. auto sgl = util::parseIntSegments("-1");
  1444. CPPUNIT_FAIL("exception must be thrown.");
  1445. } catch(Exception& e) {
  1446. }
  1447. try {
  1448. auto sgl = util::parseIntSegments("1-");
  1449. CPPUNIT_FAIL("exception must be thrown.");
  1450. } catch(Exception& e) {
  1451. }
  1452. try {
  1453. auto sgl = util::parseIntSegments("2147483648");
  1454. CPPUNIT_FAIL("exception must be thrown.");
  1455. } catch(Exception& e) {
  1456. }
  1457. try {
  1458. auto sgl = util::parseIntSegments("2147483647-2147483648");
  1459. CPPUNIT_FAIL("exception must be thrown.");
  1460. } catch(Exception& e) {
  1461. }
  1462. try {
  1463. auto sgl = util::parseIntSegments("1-2x");
  1464. CPPUNIT_FAIL("exception must be thrown.");
  1465. } catch(Exception& e) {
  1466. }
  1467. try {
  1468. auto sgl = util::parseIntSegments("3x-4");
  1469. CPPUNIT_FAIL("exception must be thrown.");
  1470. } catch(Exception& e) {
  1471. }
  1472. }
  1473. void UtilTest::testParseIntNoThrow()
  1474. {
  1475. std::string s;
  1476. int32_t n;
  1477. s = " -1 ";
  1478. CPPUNIT_ASSERT(util::parseIntNoThrow(n, s));
  1479. CPPUNIT_ASSERT_EQUAL((int32_t)-1, n);
  1480. s = "2147483647";
  1481. CPPUNIT_ASSERT(util::parseIntNoThrow(n, s));
  1482. CPPUNIT_ASSERT_EQUAL((int32_t)2147483647, n);
  1483. s = "2147483648";
  1484. CPPUNIT_ASSERT(!util::parseIntNoThrow(n, s));
  1485. s = "-2147483649";
  1486. CPPUNIT_ASSERT(!util::parseIntNoThrow(n, s));
  1487. s = "12x";
  1488. CPPUNIT_ASSERT(!util::parseIntNoThrow(n, s));
  1489. s = "";
  1490. CPPUNIT_ASSERT(!util::parseIntNoThrow(n, s));
  1491. }
  1492. void UtilTest::testParseUIntNoThrow()
  1493. {
  1494. std::string s;
  1495. uint32_t n;
  1496. s = " 2147483647 ";
  1497. CPPUNIT_ASSERT(util::parseUIntNoThrow(n, s));
  1498. CPPUNIT_ASSERT_EQUAL((uint32_t)INT32_MAX, n);
  1499. s = "2147483648";
  1500. CPPUNIT_ASSERT(!util::parseUIntNoThrow(n, s));
  1501. s = "-1";
  1502. CPPUNIT_ASSERT(!util::parseUIntNoThrow(n, s));
  1503. }
  1504. void UtilTest::testParseLLIntNoThrow()
  1505. {
  1506. std::string s;
  1507. int64_t n;
  1508. s = " 9223372036854775807 ";
  1509. CPPUNIT_ASSERT(util::parseLLIntNoThrow(n, s));
  1510. CPPUNIT_ASSERT_EQUAL((int64_t)INT64_MAX, n);
  1511. s = "9223372036854775808";
  1512. CPPUNIT_ASSERT(!util::parseLLIntNoThrow(n, s));
  1513. s = "-9223372036854775808";
  1514. CPPUNIT_ASSERT(util::parseLLIntNoThrow(n, s));
  1515. CPPUNIT_ASSERT_EQUAL((int64_t)INT64_MIN, n);
  1516. s = "-9223372036854775809";
  1517. CPPUNIT_ASSERT(!util::parseLLIntNoThrow(n, s));
  1518. }
  1519. void UtilTest::testToString_binaryStream()
  1520. {
  1521. std::shared_ptr<DiskWriter> dw(new ByteArrayDiskWriter());
  1522. std::string data(16*1024+256, 'a');
  1523. dw->initAndOpenFile();
  1524. dw->writeData((const unsigned char*)data.c_str(), data.size(), 0);
  1525. std::string readData = util::toString(dw);
  1526. CPPUNIT_ASSERT_EQUAL(data, readData);
  1527. }
  1528. void UtilTest::testItos()
  1529. {
  1530. {
  1531. int i = 0;
  1532. CPPUNIT_ASSERT_EQUAL(std::string("0"), util::itos(i));
  1533. }
  1534. {
  1535. int i = 100;
  1536. CPPUNIT_ASSERT_EQUAL(std::string("100"), util::itos(i, true));
  1537. }
  1538. {
  1539. int i = 100;
  1540. CPPUNIT_ASSERT_EQUAL(std::string("100"), util::itos(i));
  1541. }
  1542. {
  1543. int i = 12345;
  1544. CPPUNIT_ASSERT_EQUAL(std::string("12,345"), util::itos(i, true));
  1545. }
  1546. {
  1547. int i = 12345;
  1548. CPPUNIT_ASSERT_EQUAL(std::string("12345"), util::itos(i));
  1549. }
  1550. {
  1551. int i = -12345;
  1552. CPPUNIT_ASSERT_EQUAL(std::string("-12,345"), util::itos(i, true));
  1553. }
  1554. {
  1555. int64_t i = INT64_MAX;
  1556. CPPUNIT_ASSERT_EQUAL(std::string("9,223,372,036,854,775,807"),
  1557. util::itos(i, true));
  1558. }
  1559. {
  1560. int64_t i = INT64_MIN;
  1561. CPPUNIT_ASSERT_EQUAL(std::string("-9,223,372,036,854,775,808"),
  1562. util::itos(i, true));
  1563. }
  1564. }
  1565. void UtilTest::testUitos()
  1566. {
  1567. {
  1568. uint16_t i = 12345;
  1569. CPPUNIT_ASSERT_EQUAL(std::string("12345"), util::uitos(i));
  1570. }
  1571. {
  1572. int16_t i = -12345;
  1573. CPPUNIT_ASSERT_EQUAL(std::string("/.-,+"), util::uitos(i));
  1574. }
  1575. }
  1576. void UtilTest::testNtoh64()
  1577. {
  1578. uint64_t x = 0xff00ff00ee00ee00LL;
  1579. #ifdef WORDS_BIGENDIAN
  1580. CPPUNIT_ASSERT_EQUAL(x, ntoh64(x));
  1581. CPPUNIT_ASSERT_EQUAL(x, hton64(x));
  1582. #else // !WORDS_BIGENDIAN
  1583. uint64_t y = 0x00ee00ee00ff00ffLL;
  1584. CPPUNIT_ASSERT_EQUAL(y, ntoh64(x));
  1585. CPPUNIT_ASSERT_EQUAL(x, hton64(y));
  1586. #endif // !WORDS_BIGENDIAN
  1587. }
  1588. void UtilTest::testPercentEncode()
  1589. {
  1590. CPPUNIT_ASSERT_EQUAL
  1591. (std::string("%3A%2F%3F%23%5B%5D%40%21%25%26%27%28%29%2A%2B%2C%3B%3D"),
  1592. util::percentEncode(":/?#[]@!%&'()*+,;="));
  1593. std::string unreserved =
  1594. "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
  1595. "abcdefghijklmnopqrstuvwxyz"
  1596. "0123456789"
  1597. "-._~";
  1598. CPPUNIT_ASSERT_EQUAL(unreserved, util::percentEncode(unreserved));
  1599. CPPUNIT_ASSERT_EQUAL(std::string("1%5EA%20"), util::percentEncode("1^A "));
  1600. }
  1601. void UtilTest::testPercentEncodeMini()
  1602. {
  1603. CPPUNIT_ASSERT_EQUAL(std::string("%80"),
  1604. util::percentEncodeMini({(char)0x80}));
  1605. }
  1606. void UtilTest::testHtmlEscape()
  1607. {
  1608. CPPUNIT_ASSERT_EQUAL(std::string("aria2&lt;&gt;&quot;&#39;util"),
  1609. util::htmlEscape("aria2<>\"'util"));
  1610. }
  1611. void UtilTest::testJoinPath()
  1612. {
  1613. const std::string dir1dir2file[] = { "dir1", "dir2", "file" };
  1614. CPPUNIT_ASSERT_EQUAL
  1615. (std::string("dir1/dir2/file"),
  1616. util::joinPath(std::begin(dir1dir2file), std::end(dir1dir2file)));
  1617. const std::string dirparentfile[] = { "dir", "..", "file" };
  1618. CPPUNIT_ASSERT_EQUAL
  1619. (std::string("file"),
  1620. util::joinPath(std::begin(dirparentfile), std::end(dirparentfile)));
  1621. const std::string dirparentparentfile[] = { "dir", "..", "..", "file" };
  1622. CPPUNIT_ASSERT_EQUAL
  1623. (std::string("file"),
  1624. util::joinPath(std::begin(dirparentparentfile),
  1625. std::end(dirparentparentfile)));
  1626. const std::string dirdotfile[] = { "dir", ".", "file" };
  1627. CPPUNIT_ASSERT_EQUAL(std::string("dir/file"),
  1628. util::joinPath(std::begin(dirdotfile),
  1629. std::end(dirdotfile)));
  1630. const std::string empty[] = {};
  1631. CPPUNIT_ASSERT_EQUAL(std::string(""), util::joinPath(&empty[0], &empty[0]));
  1632. const std::string parentdot[] = { "..", "." };
  1633. CPPUNIT_ASSERT_EQUAL(std::string(""),
  1634. util::joinPath(std::begin(parentdot),
  1635. std::end(parentdot)));
  1636. }
  1637. void UtilTest::testParseIndexPath()
  1638. {
  1639. std::pair<size_t, std::string> p = util::parseIndexPath("1=foo");
  1640. CPPUNIT_ASSERT_EQUAL((size_t)1, p.first);
  1641. CPPUNIT_ASSERT_EQUAL(std::string("foo"), p.second);
  1642. try {
  1643. util::parseIndexPath("1X=foo");
  1644. CPPUNIT_FAIL("exception must be thrown.");
  1645. } catch(Exception& e) {
  1646. // success
  1647. }
  1648. try {
  1649. util::parseIndexPath("1=");
  1650. CPPUNIT_FAIL("exception must be thrown.");
  1651. } catch(Exception& e) {
  1652. // success
  1653. }
  1654. }
  1655. void UtilTest::testCreateIndexPaths()
  1656. {
  1657. std::stringstream in
  1658. ("1=/tmp/myfile\n"
  1659. "100=/myhome/mypicture.png\n");
  1660. std::vector<std::pair<size_t, std::string> > m = util::createIndexPaths(in);
  1661. CPPUNIT_ASSERT_EQUAL((size_t)2, m.size());
  1662. CPPUNIT_ASSERT_EQUAL((size_t)1, m[0].first);
  1663. CPPUNIT_ASSERT_EQUAL(std::string("/tmp/myfile"), m[0].second);
  1664. CPPUNIT_ASSERT_EQUAL((size_t)100, m[1].first);
  1665. CPPUNIT_ASSERT_EQUAL(std::string("/myhome/mypicture.png"), m[1].second);
  1666. }
  1667. void UtilTest::testGenerateRandomData()
  1668. {
  1669. using namespace std;
  1670. // Simple sanity check
  1671. unsigned char data1[25];
  1672. memset(data1, 0, sizeof(data1));
  1673. util::generateRandomData(data1, sizeof(data1));
  1674. unsigned char data2[25];
  1675. memset(data2, 0, sizeof(data2));
  1676. util::generateRandomData(data2, sizeof(data2));
  1677. CPPUNIT_ASSERT(memcmp(data1, data2, sizeof(data1)) != 0);
  1678. // Simple stddev/mean tests
  1679. map<uint8_t, size_t> counts;
  1680. uint8_t bytes[1 << 20];
  1681. for (auto i = 0; i < 10; ++i) {
  1682. util::generateRandomData(bytes, sizeof(bytes));
  1683. for (auto b : bytes) {
  1684. counts[b]++;
  1685. }
  1686. }
  1687. CPPUNIT_ASSERT_MESSAGE("Should see all kinds of bytes", counts.size() == 256);
  1688. double sum = accumulate(
  1689. counts.begin(),
  1690. counts.end(),
  1691. 0.0,
  1692. [](double total, const decltype(counts)::value_type & elem) {
  1693. return total + elem.second;
  1694. });
  1695. double mean = sum / counts.size();
  1696. vector<double> diff(counts.size());
  1697. transform(
  1698. counts.begin(),
  1699. counts.end(),
  1700. diff.begin(),
  1701. [&](const decltype(counts)::value_type & elem) -> double {
  1702. return (double)elem.second - mean;
  1703. });
  1704. double sq_sum = inner_product(diff.begin(), diff.end(), diff.begin(), 0.0);
  1705. double stddev = sqrt(sq_sum / counts.size());
  1706. cout << "stddev: " << fixed << stddev << endl;
  1707. CPPUNIT_ASSERT_MESSAGE("stddev makes sense (lower)", stddev <= 320);
  1708. CPPUNIT_ASSERT_MESSAGE("stddev makes sense (upper)", stddev >= 100);
  1709. }
  1710. void UtilTest::testFromHex()
  1711. {
  1712. std::string src;
  1713. std::string dest;
  1714. src = "0011fF";
  1715. dest = util::fromHex(src.begin(), src.end());
  1716. CPPUNIT_ASSERT_EQUAL((size_t)3, dest.size());
  1717. CPPUNIT_ASSERT_EQUAL((char)0x00, dest[0]);
  1718. CPPUNIT_ASSERT_EQUAL((char)0x11, dest[1]);
  1719. CPPUNIT_ASSERT_EQUAL((char)0xff, dest[2]);
  1720. src = "0011f";
  1721. CPPUNIT_ASSERT(util::fromHex(src.begin(), src.end()).empty());
  1722. src = "001g";
  1723. CPPUNIT_ASSERT(util::fromHex(src.begin(), src.end()).empty());
  1724. }
  1725. void UtilTest::testParsePrioritizePieceRange()
  1726. {
  1727. // piece index
  1728. // 0 1 2 3 4 5 6 7
  1729. // | | | |
  1730. // file1 | | |
  1731. // | | |
  1732. // file2 | |
  1733. // file3 |
  1734. // | |
  1735. // file4 |
  1736. size_t pieceLength = 1024;
  1737. std::vector<std::shared_ptr<FileEntry> > entries(4, std::shared_ptr<FileEntry>());
  1738. entries[0].reset(new FileEntry("file1", 1024, 0));
  1739. entries[1].reset(new FileEntry("file2",2560,entries[0]->getLastOffset()));
  1740. entries[2].reset(new FileEntry("file3",0,entries[1]->getLastOffset()));
  1741. entries[3].reset(new FileEntry("file4",3584,entries[2]->getLastOffset()));
  1742. std::vector<size_t> result;
  1743. util::parsePrioritizePieceRange(result, "head=1", entries, pieceLength);
  1744. CPPUNIT_ASSERT_EQUAL((size_t)3, result.size());
  1745. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1746. CPPUNIT_ASSERT_EQUAL((size_t)1, result[1]);
  1747. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  1748. result.clear();
  1749. util::parsePrioritizePieceRange(result, "tail=1", entries, pieceLength);
  1750. CPPUNIT_ASSERT_EQUAL((size_t)3, result.size());
  1751. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1752. CPPUNIT_ASSERT_EQUAL((size_t)3, result[1]);
  1753. CPPUNIT_ASSERT_EQUAL((size_t)6, result[2]);
  1754. result.clear();
  1755. util::parsePrioritizePieceRange(result, "head=1K", entries, pieceLength);
  1756. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  1757. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1758. CPPUNIT_ASSERT_EQUAL((size_t)1, result[1]);
  1759. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  1760. CPPUNIT_ASSERT_EQUAL((size_t)4, result[3]);
  1761. result.clear();
  1762. util::parsePrioritizePieceRange(result, "head", entries, pieceLength, 1024);
  1763. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  1764. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1765. CPPUNIT_ASSERT_EQUAL((size_t)1, result[1]);
  1766. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  1767. CPPUNIT_ASSERT_EQUAL((size_t)4, result[3]);
  1768. result.clear();
  1769. util::parsePrioritizePieceRange(result, "tail=1K", entries, pieceLength);
  1770. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  1771. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1772. CPPUNIT_ASSERT_EQUAL((size_t)2, result[1]);
  1773. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  1774. CPPUNIT_ASSERT_EQUAL((size_t)6, result[3]);
  1775. result.clear();
  1776. util::parsePrioritizePieceRange(result, "tail", entries, pieceLength, 1024);
  1777. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  1778. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1779. CPPUNIT_ASSERT_EQUAL((size_t)2, result[1]);
  1780. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  1781. CPPUNIT_ASSERT_EQUAL((size_t)6, result[3]);
  1782. result.clear();
  1783. util::parsePrioritizePieceRange(result, "head=1,tail=1", entries, pieceLength);
  1784. CPPUNIT_ASSERT_EQUAL((size_t)4, result.size());
  1785. CPPUNIT_ASSERT_EQUAL((size_t)0, result[0]);
  1786. CPPUNIT_ASSERT_EQUAL((size_t)1, result[1]);
  1787. CPPUNIT_ASSERT_EQUAL((size_t)3, result[2]);
  1788. CPPUNIT_ASSERT_EQUAL((size_t)6, result[3]);
  1789. result.clear();
  1790. util::parsePrioritizePieceRange(result, "head=300M,tail=300M",
  1791. entries, pieceLength);
  1792. CPPUNIT_ASSERT_EQUAL((size_t)7, result.size());
  1793. for(size_t i = 0; i < 7; ++i) {
  1794. CPPUNIT_ASSERT_EQUAL(i, result[i]);
  1795. }
  1796. result.clear();
  1797. util::parsePrioritizePieceRange(result, "", entries, pieceLength);
  1798. CPPUNIT_ASSERT(result.empty());
  1799. }
  1800. void UtilTest::testApplyDir()
  1801. {
  1802. CPPUNIT_ASSERT_EQUAL(std::string("./pred"), util::applyDir("", "pred"));
  1803. CPPUNIT_ASSERT_EQUAL(std::string("/pred"), util::applyDir("/", "pred"));
  1804. CPPUNIT_ASSERT_EQUAL(std::string("./pred"), util::applyDir(".", "pred"));
  1805. CPPUNIT_ASSERT_EQUAL(std::string("/dl/pred"), util::applyDir("/dl", "pred"));
  1806. }
  1807. void UtilTest::testFixTaintedBasename()
  1808. {
  1809. CPPUNIT_ASSERT_EQUAL(std::string("a%2Fb"), util::fixTaintedBasename("a/b"));
  1810. #ifdef __MINGW32__
  1811. CPPUNIT_ASSERT_EQUAL(std::string("a%5Cb"), util::fixTaintedBasename("a\\b"));
  1812. #else // !__MINGW32__
  1813. CPPUNIT_ASSERT_EQUAL(std::string("a\\b"), util::fixTaintedBasename("a\\b"));
  1814. #endif // !__MINGW32__
  1815. }
  1816. void UtilTest::testIsNumericHost()
  1817. {
  1818. CPPUNIT_ASSERT(util::isNumericHost("192.168.0.1"));
  1819. CPPUNIT_ASSERT(!util::isNumericHost("aria2.sf.net"));
  1820. CPPUNIT_ASSERT(util::isNumericHost("::1"));
  1821. }
  1822. void UtilTest::testDetectDirTraversal()
  1823. {
  1824. CPPUNIT_ASSERT(util::detectDirTraversal("/foo"));
  1825. CPPUNIT_ASSERT(util::detectDirTraversal("./foo"));
  1826. CPPUNIT_ASSERT(util::detectDirTraversal("../foo"));
  1827. CPPUNIT_ASSERT(util::detectDirTraversal("foo/../bar"));
  1828. CPPUNIT_ASSERT(util::detectDirTraversal("foo/./bar"));
  1829. CPPUNIT_ASSERT(util::detectDirTraversal("foo/."));
  1830. CPPUNIT_ASSERT(util::detectDirTraversal("foo/.."));
  1831. CPPUNIT_ASSERT(util::detectDirTraversal("."));
  1832. CPPUNIT_ASSERT(util::detectDirTraversal(".."));
  1833. CPPUNIT_ASSERT(util::detectDirTraversal("/"));
  1834. CPPUNIT_ASSERT(util::detectDirTraversal("foo/"));
  1835. CPPUNIT_ASSERT(util::detectDirTraversal("\t"));
  1836. CPPUNIT_ASSERT(!util::detectDirTraversal("foo/bar"));
  1837. CPPUNIT_ASSERT(!util::detectDirTraversal("foo"));
  1838. }
  1839. void UtilTest::testEscapePath()
  1840. {
  1841. CPPUNIT_ASSERT_EQUAL(std::string("foo%00bar%00%01"),
  1842. util::escapePath(std::string("foo")+(char)0x00+
  1843. std::string("bar")+(char)0x00+
  1844. (char)0x01));
  1845. #ifdef __MINGW32__
  1846. CPPUNIT_ASSERT_EQUAL(std::string("foo%5Cbar"), util::escapePath("foo\\bar"));
  1847. #else // !__MINGW32__
  1848. CPPUNIT_ASSERT_EQUAL(std::string("foo\\bar"), util::escapePath("foo\\bar"));
  1849. #endif // !__MINGW32__
  1850. }
  1851. void UtilTest::testInSameCidrBlock()
  1852. {
  1853. CPPUNIT_ASSERT(util::inSameCidrBlock("192.168.128.1", "192.168.0.1", 16));
  1854. CPPUNIT_ASSERT(!util::inSameCidrBlock("192.168.128.1", "192.168.0.1", 17));
  1855. CPPUNIT_ASSERT(util::inSameCidrBlock("192.168.0.1", "192.168.0.1", 32));
  1856. CPPUNIT_ASSERT(!util::inSameCidrBlock("192.168.0.1", "192.168.0.0", 32));
  1857. CPPUNIT_ASSERT(util::inSameCidrBlock("192.168.0.1", "10.0.0.1", 0));
  1858. CPPUNIT_ASSERT(util::inSameCidrBlock("2001:db8::2:1", "2001:db0::2:2", 28));
  1859. CPPUNIT_ASSERT(!util::inSameCidrBlock("2001:db8::2:1", "2001:db0::2:2", 29));
  1860. CPPUNIT_ASSERT(!util::inSameCidrBlock("2001:db8::2:1", "192.168.0.1", 8));
  1861. }
  1862. void UtilTest::testIsUtf8String()
  1863. {
  1864. CPPUNIT_ASSERT(util::isUtf8("ascii"));
  1865. // "Hello World" in Japanese UTF-8
  1866. CPPUNIT_ASSERT(util::isUtf8
  1867. (fromHex("e38193e38293e381abe381a1e381afe4b896e7958c")));
  1868. // "World" in Shift_JIS
  1869. CPPUNIT_ASSERT(!util::isUtf8(fromHex("90a28a")+"E"));
  1870. // UTF8-2
  1871. CPPUNIT_ASSERT(util::isUtf8(fromHex("c280")));
  1872. CPPUNIT_ASSERT(util::isUtf8(fromHex("dfbf")));
  1873. // UTF8-3
  1874. CPPUNIT_ASSERT(util::isUtf8(fromHex("e0a080")));
  1875. CPPUNIT_ASSERT(util::isUtf8(fromHex("e0bf80")));
  1876. CPPUNIT_ASSERT(util::isUtf8(fromHex("e18080")));
  1877. CPPUNIT_ASSERT(util::isUtf8(fromHex("ec8080")));
  1878. CPPUNIT_ASSERT(util::isUtf8(fromHex("ed8080")));
  1879. CPPUNIT_ASSERT(util::isUtf8(fromHex("ed9f80")));
  1880. CPPUNIT_ASSERT(util::isUtf8(fromHex("ee8080")));
  1881. CPPUNIT_ASSERT(util::isUtf8(fromHex("ef8080")));
  1882. // UTF8-4
  1883. CPPUNIT_ASSERT(util::isUtf8(fromHex("f0908080")));
  1884. CPPUNIT_ASSERT(util::isUtf8(fromHex("f0bf8080")));
  1885. CPPUNIT_ASSERT(util::isUtf8(fromHex("f1808080")));
  1886. CPPUNIT_ASSERT(util::isUtf8(fromHex("f3808080")));
  1887. CPPUNIT_ASSERT(util::isUtf8(fromHex("f4808080")));
  1888. CPPUNIT_ASSERT(util::isUtf8(fromHex("f48f8080")));
  1889. CPPUNIT_ASSERT(util::isUtf8(""));
  1890. CPPUNIT_ASSERT(!util::isUtf8(fromHex("00")));
  1891. }
  1892. void UtilTest::testNextParam()
  1893. {
  1894. std::string s1 = " :a : b=c :d=b::::g::";
  1895. std::pair<std::string::iterator, bool> r;
  1896. std::string name, value;
  1897. r = util::nextParam(name, value, s1.begin(), s1.end(), ':');
  1898. CPPUNIT_ASSERT(r.second);
  1899. CPPUNIT_ASSERT_EQUAL(std::string("a"), name);
  1900. CPPUNIT_ASSERT_EQUAL(std::string(), value);
  1901. r = util::nextParam(name, value, r.first, s1.end(), ':');
  1902. CPPUNIT_ASSERT(r.second);
  1903. CPPUNIT_ASSERT_EQUAL(std::string("b"), name);
  1904. CPPUNIT_ASSERT_EQUAL(std::string("c"), value);
  1905. r = util::nextParam(name, value, r.first, s1.end(), ':');
  1906. CPPUNIT_ASSERT(r.second);
  1907. CPPUNIT_ASSERT_EQUAL(std::string("d"), name);
  1908. CPPUNIT_ASSERT_EQUAL(std::string("b"), value);
  1909. r = util::nextParam(name, value, r.first, s1.end(), ':');
  1910. CPPUNIT_ASSERT(r.second);
  1911. CPPUNIT_ASSERT_EQUAL(std::string("g"), name);
  1912. CPPUNIT_ASSERT_EQUAL(std::string(), value);
  1913. std::string s2 = "";
  1914. r = util::nextParam(name, value, s2.begin(), s2.end(), ':');
  1915. CPPUNIT_ASSERT(!r.second);
  1916. std::string s3 = " ";
  1917. r = util::nextParam(name, value, s3.begin(), s3.end(), ':');
  1918. CPPUNIT_ASSERT(!r.second);
  1919. std::string s4 = ":::";
  1920. r = util::nextParam(name, value, s4.begin(), s4.end(), ':');
  1921. CPPUNIT_ASSERT(!r.second);
  1922. }
  1923. void UtilTest::testNoProxyDomainMatch()
  1924. {
  1925. CPPUNIT_ASSERT(util::noProxyDomainMatch("localhost", "localhost"));
  1926. CPPUNIT_ASSERT(util::noProxyDomainMatch("192.168.0.1", "192.168.0.1"));
  1927. CPPUNIT_ASSERT(util::noProxyDomainMatch("www.example.org", ".example.org"));
  1928. CPPUNIT_ASSERT(!util::noProxyDomainMatch("www.example.org", "example.org"));
  1929. CPPUNIT_ASSERT(!util::noProxyDomainMatch("192.168.0.1", "0.1"));
  1930. CPPUNIT_ASSERT(!util::noProxyDomainMatch("example.org", "example.com"));
  1931. CPPUNIT_ASSERT(!util::noProxyDomainMatch("example.org", "www.example.org"));
  1932. }
  1933. void UtilTest::testInPrivateAddress()
  1934. {
  1935. CPPUNIT_ASSERT(!util::inPrivateAddress("localhost"));
  1936. CPPUNIT_ASSERT(util::inPrivateAddress("192.168.0.1"));
  1937. // Only checks prefix..
  1938. CPPUNIT_ASSERT(util::inPrivateAddress("10."));
  1939. CPPUNIT_ASSERT(!util::inPrivateAddress("172."));
  1940. CPPUNIT_ASSERT(!util::inPrivateAddress("172.15.0.0"));
  1941. CPPUNIT_ASSERT(util::inPrivateAddress("172.16.0.0"));
  1942. CPPUNIT_ASSERT(util::inPrivateAddress("172.31.0.0"));
  1943. CPPUNIT_ASSERT(!util::inPrivateAddress("172.32.0.0"));
  1944. }
  1945. void UtilTest::testSecfmt()
  1946. {
  1947. CPPUNIT_ASSERT_EQUAL(std::string("0s"), util::secfmt(0));
  1948. CPPUNIT_ASSERT_EQUAL(std::string("1s"), util::secfmt(1));
  1949. CPPUNIT_ASSERT_EQUAL(std::string("9s"), util::secfmt(9));
  1950. CPPUNIT_ASSERT_EQUAL(std::string("10s"), util::secfmt(10));
  1951. CPPUNIT_ASSERT_EQUAL(std::string("1m"), util::secfmt(60));
  1952. CPPUNIT_ASSERT_EQUAL(std::string("1m59s"), util::secfmt(119));
  1953. CPPUNIT_ASSERT_EQUAL(std::string("2m"), util::secfmt(120));
  1954. CPPUNIT_ASSERT_EQUAL(std::string("59m59s"), util::secfmt(3599));
  1955. CPPUNIT_ASSERT_EQUAL(std::string("1h"), util::secfmt(3600));
  1956. }
  1957. void UtilTest::testTlsHostnameMatch()
  1958. {
  1959. CPPUNIT_ASSERT(util::tlsHostnameMatch("Foo.com", "foo.com"));
  1960. CPPUNIT_ASSERT(util::tlsHostnameMatch("*.a.com", "foo.a.com"));
  1961. CPPUNIT_ASSERT(!util::tlsHostnameMatch("*.a.com", "bar.foo.a.com"));
  1962. CPPUNIT_ASSERT(!util::tlsHostnameMatch("f*.com", "foo.com"));
  1963. CPPUNIT_ASSERT(!util::tlsHostnameMatch("*.com", "bar.com"));
  1964. CPPUNIT_ASSERT(util::tlsHostnameMatch("com", "com"));
  1965. CPPUNIT_ASSERT(!util::tlsHostnameMatch("foo.*", "foo.com"));
  1966. CPPUNIT_ASSERT(util::tlsHostnameMatch("a.foo.com", "A.foo.com"));
  1967. CPPUNIT_ASSERT(!util::tlsHostnameMatch("a.foo.com", "b.foo.com"));
  1968. CPPUNIT_ASSERT(!util::tlsHostnameMatch("*a.foo.com", "a.foo.com"));
  1969. CPPUNIT_ASSERT(util::tlsHostnameMatch("*a.foo.com", "ba.foo.com"));
  1970. CPPUNIT_ASSERT(!util::tlsHostnameMatch("a*.foo.com", "a.foo.com"));
  1971. CPPUNIT_ASSERT(util::tlsHostnameMatch("a*.foo.com", "ab.foo.com"));
  1972. CPPUNIT_ASSERT(!util::tlsHostnameMatch("foo.b*z.foo.com", "foo.baz.foo.com"));
  1973. CPPUNIT_ASSERT(util::tlsHostnameMatch("B*z.foo.com", "bAZ.Foo.com"));
  1974. CPPUNIT_ASSERT(!util::tlsHostnameMatch("b*z.foo.com", "bz.foo.com"));
  1975. CPPUNIT_ASSERT(!util::tlsHostnameMatch("*", "foo"));
  1976. CPPUNIT_ASSERT(!util::tlsHostnameMatch("*", ""));
  1977. CPPUNIT_ASSERT(util::tlsHostnameMatch("", ""));
  1978. CPPUNIT_ASSERT(!util::tlsHostnameMatch("xn--*.a.b", "xn--c.a.b"));
  1979. }
  1980. } // namespace aria2