UtilTest1.cc 60 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510
  1. #include "util.h"
  2. #include <cmath>
  3. #include <cstring>
  4. #include <string>
  5. #include <cassert>
  6. #include <iostream>
  7. #include <cppunit/extensions/HelperMacros.h>
  8. #include "FixedNumberRandomizer.h"
  9. #include "DlAbortEx.h"
  10. #include "BitfieldMan.h"
  11. #include "ByteArrayDiskWriter.h"
  12. #include "FileEntry.h"
  13. #include "File.h"
  14. #include "array_fun.h"
  15. #include "BufferedFile.h"
  16. #include "TestUtil.h"
  17. #include "SocketCore.h"
  18. namespace aria2 {
  19. class UtilTest1 : public CppUnit::TestFixture {
  20. CPPUNIT_TEST_SUITE(UtilTest1);
  21. CPPUNIT_TEST(testStrip);
  22. CPPUNIT_TEST(testStripIter);
  23. CPPUNIT_TEST(testLstripIter);
  24. CPPUNIT_TEST(testLstripIter_char);
  25. CPPUNIT_TEST(testDivide);
  26. CPPUNIT_TEST(testSplit);
  27. CPPUNIT_TEST(testSplitIter);
  28. CPPUNIT_TEST(testSplitIterM);
  29. CPPUNIT_TEST(testStreq);
  30. CPPUNIT_TEST(testStrieq);
  31. CPPUNIT_TEST(testStrifind);
  32. CPPUNIT_TEST(testEndsWith);
  33. CPPUNIT_TEST(testIendsWith);
  34. CPPUNIT_TEST(testReplace);
  35. CPPUNIT_TEST(testStartsWith);
  36. CPPUNIT_TEST(testIstartsWith);
  37. // may be moved to other helper class in the future.
  38. CPPUNIT_TEST(testGetContentDispositionFilename);
  39. CPPUNIT_TEST(testParseContentDisposition1);
  40. CPPUNIT_TEST(testParseContentDisposition2);
  41. CPPUNIT_TEST_SUITE_END();
  42. private:
  43. public:
  44. void setUp() {}
  45. void testStrip();
  46. void testStripIter();
  47. void testLstripIter();
  48. void testLstripIter_char();
  49. void testDivide();
  50. void testSplit();
  51. void testSplitIter();
  52. void testSplitIterM();
  53. void testStreq();
  54. void testStrieq();
  55. void testStrifind();
  56. void testEndsWith();
  57. void testIendsWith();
  58. void testReplace();
  59. void testStartsWith();
  60. void testIstartsWith();
  61. // may be moved to other helper class in the future.
  62. void testGetContentDispositionFilename();
  63. void testParseContentDisposition1();
  64. void testParseContentDisposition2();
  65. };
  66. CPPUNIT_TEST_SUITE_REGISTRATION(UtilTest1);
  67. void UtilTest1::testStrip()
  68. {
  69. std::string str1 = "aria2";
  70. CPPUNIT_ASSERT_EQUAL(str1, util::strip("aria2"));
  71. CPPUNIT_ASSERT_EQUAL(str1, util::strip(" aria2"));
  72. CPPUNIT_ASSERT_EQUAL(str1, util::strip("aria2 "));
  73. CPPUNIT_ASSERT_EQUAL(str1, util::strip(" aria2 "));
  74. CPPUNIT_ASSERT_EQUAL(str1, util::strip(" aria2 "));
  75. std::string str2 = "aria2 debut";
  76. CPPUNIT_ASSERT_EQUAL(str2, util::strip("aria2 debut"));
  77. CPPUNIT_ASSERT_EQUAL(str2, util::strip(" aria2 debut "));
  78. std::string str3 = "";
  79. CPPUNIT_ASSERT_EQUAL(str3, util::strip(""));
  80. CPPUNIT_ASSERT_EQUAL(str3, util::strip(" "));
  81. CPPUNIT_ASSERT_EQUAL(str3, util::strip(" "));
  82. std::string str4 = "A";
  83. CPPUNIT_ASSERT_EQUAL(str4, util::strip("A"));
  84. CPPUNIT_ASSERT_EQUAL(str4, util::strip(" A "));
  85. CPPUNIT_ASSERT_EQUAL(str4, util::strip(" A "));
  86. }
  87. void UtilTest1::testStripIter()
  88. {
  89. Scip p;
  90. std::string str1 = "aria2";
  91. std::string s = "aria2";
  92. p = util::stripIter(s.begin(), s.end());
  93. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  94. s = " aria2";
  95. p = util::stripIter(s.begin(), s.end());
  96. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  97. s = "aria2 ";
  98. p = util::stripIter(s.begin(), s.end());
  99. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  100. s = " aria2 ";
  101. p = util::stripIter(s.begin(), s.end());
  102. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  103. s = " aria2 ";
  104. p = util::stripIter(s.begin(), s.end());
  105. CPPUNIT_ASSERT_EQUAL(str1, std::string(p.first, p.second));
  106. std::string str2 = "aria2 debut";
  107. s = "aria2 debut";
  108. p = util::stripIter(s.begin(), s.end());
  109. CPPUNIT_ASSERT_EQUAL(str2, std::string(p.first, p.second));
  110. s = " aria2 debut ";
  111. p = util::stripIter(s.begin(), s.end());
  112. CPPUNIT_ASSERT_EQUAL(str2, std::string(p.first, p.second));
  113. std::string str3 = "";
  114. s = "";
  115. p = util::stripIter(s.begin(), s.end());
  116. CPPUNIT_ASSERT_EQUAL(str3, std::string(p.first, p.second));
  117. s = " ";
  118. p = util::stripIter(s.begin(), s.end());
  119. CPPUNIT_ASSERT_EQUAL(str3, std::string(p.first, p.second));
  120. s = " ";
  121. p = util::stripIter(s.begin(), s.end());
  122. CPPUNIT_ASSERT_EQUAL(str3, std::string(p.first, p.second));
  123. std::string str4 = "A";
  124. s = "A";
  125. p = util::stripIter(s.begin(), s.end());
  126. CPPUNIT_ASSERT_EQUAL(str4, std::string(p.first, p.second));
  127. s = " A ";
  128. p = util::stripIter(s.begin(), s.end());
  129. CPPUNIT_ASSERT_EQUAL(str4, std::string(p.first, p.second));
  130. s = " A ";
  131. p = util::stripIter(s.begin(), s.end());
  132. CPPUNIT_ASSERT_EQUAL(str4, std::string(p.first, p.second));
  133. }
  134. void UtilTest1::testLstripIter()
  135. {
  136. std::string::iterator r;
  137. std::string s = "foo";
  138. r = util::lstripIter(s.begin(), s.end());
  139. CPPUNIT_ASSERT_EQUAL(std::string("foo"), std::string(r, s.end()));
  140. s = " foo bar ";
  141. r = util::lstripIter(s.begin(), s.end());
  142. CPPUNIT_ASSERT_EQUAL(std::string("foo bar "), std::string(r, s.end()));
  143. s = "f";
  144. r = util::lstripIter(s.begin(), s.end());
  145. CPPUNIT_ASSERT_EQUAL(std::string("f"), std::string(r, s.end()));
  146. s = "foo ";
  147. r = util::lstripIter(s.begin(), s.end());
  148. CPPUNIT_ASSERT_EQUAL(std::string("foo "), std::string(r, s.end()));
  149. }
  150. void UtilTest1::testLstripIter_char()
  151. {
  152. std::string::iterator r;
  153. std::string s = "foo";
  154. r = util::lstripIter(s.begin(), s.end(), '$');
  155. CPPUNIT_ASSERT_EQUAL(std::string("foo"), std::string(r, s.end()));
  156. s = "$$foo$bar$$";
  157. r = util::lstripIter(s.begin(), s.end(), '$');
  158. CPPUNIT_ASSERT_EQUAL(std::string("foo$bar$$"), std::string(r, s.end()));
  159. s = "f";
  160. r = util::lstripIter(s.begin(), s.end(), '$');
  161. CPPUNIT_ASSERT_EQUAL(std::string("f"), std::string(r, s.end()));
  162. s = "foo$$";
  163. r = util::lstripIter(s.begin(), s.end(), '$');
  164. CPPUNIT_ASSERT_EQUAL(std::string("foo$$"), std::string(r, s.end()));
  165. }
  166. void UtilTest1::testDivide()
  167. {
  168. std::string s = "name=value";
  169. auto p1 = util::divide(std::begin(s), std::end(s), '=');
  170. CPPUNIT_ASSERT_EQUAL(std::string("name"),
  171. std::string(p1.first.first, p1.first.second));
  172. CPPUNIT_ASSERT_EQUAL(std::string("value"),
  173. std::string(p1.second.first, p1.second.second));
  174. s = " name = value ";
  175. p1 = util::divide(std::begin(s), std::end(s), '=');
  176. CPPUNIT_ASSERT_EQUAL(std::string("name"),
  177. std::string(p1.first.first, p1.first.second));
  178. CPPUNIT_ASSERT_EQUAL(std::string("value"),
  179. std::string(p1.second.first, p1.second.second));
  180. s = "=value";
  181. p1 = util::divide(std::begin(s), std::end(s), '=');
  182. CPPUNIT_ASSERT_EQUAL(std::string(""),
  183. std::string(p1.first.first, p1.first.second));
  184. CPPUNIT_ASSERT_EQUAL(std::string("value"),
  185. std::string(p1.second.first, p1.second.second));
  186. s = "name=";
  187. p1 = util::divide(std::begin(s), std::end(s), '=');
  188. CPPUNIT_ASSERT_EQUAL(std::string("name"),
  189. std::string(p1.first.first, p1.first.second));
  190. CPPUNIT_ASSERT_EQUAL(std::string(""),
  191. std::string(p1.second.first, p1.second.second));
  192. s = "name";
  193. p1 = util::divide(std::begin(s), std::end(s), '=');
  194. CPPUNIT_ASSERT_EQUAL(std::string("name"),
  195. std::string(p1.first.first, p1.first.second));
  196. CPPUNIT_ASSERT_EQUAL(std::string(""),
  197. std::string(p1.second.first, p1.second.second));
  198. }
  199. void UtilTest1::testSplit()
  200. {
  201. std::vector<std::string> v;
  202. std::string s = "k1; k2;; k3";
  203. util::split(s.begin(), s.end(), std::back_inserter(v), ';', true);
  204. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  205. std::vector<std::string>::iterator itr = v.begin();
  206. CPPUNIT_ASSERT_EQUAL(std::string("k1"), *itr++);
  207. CPPUNIT_ASSERT_EQUAL(std::string("k2"), *itr++);
  208. CPPUNIT_ASSERT_EQUAL(std::string("k3"), *itr++);
  209. v.clear();
  210. s = "k1; k2; k3";
  211. util::split(s.begin(), s.end(), std::back_inserter(v), ';');
  212. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  213. itr = v.begin();
  214. CPPUNIT_ASSERT_EQUAL(std::string("k1"), *itr++);
  215. CPPUNIT_ASSERT_EQUAL(std::string(" k2"), *itr++);
  216. CPPUNIT_ASSERT_EQUAL(std::string(" k3"), *itr++);
  217. v.clear();
  218. s = "k=v";
  219. util::split(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  220. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  221. itr = v.begin();
  222. CPPUNIT_ASSERT_EQUAL(std::string("k=v"), *itr++);
  223. v.clear();
  224. s = ";;k1;;k2;";
  225. util::split(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  226. CPPUNIT_ASSERT_EQUAL((size_t)6, v.size());
  227. itr = v.begin();
  228. CPPUNIT_ASSERT_EQUAL(std::string(""), *itr++);
  229. CPPUNIT_ASSERT_EQUAL(std::string(""), *itr++);
  230. CPPUNIT_ASSERT_EQUAL(std::string("k1"), *itr++);
  231. CPPUNIT_ASSERT_EQUAL(std::string(""), *itr++);
  232. CPPUNIT_ASSERT_EQUAL(std::string("k2"), *itr++);
  233. CPPUNIT_ASSERT_EQUAL(std::string(""), *itr++);
  234. v.clear();
  235. s = ";;k1;;k2;";
  236. util::split(s.begin(), s.end(), std::back_inserter(v), ';');
  237. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  238. itr = v.begin();
  239. CPPUNIT_ASSERT_EQUAL(std::string("k1"), *itr++);
  240. CPPUNIT_ASSERT_EQUAL(std::string("k2"), *itr++);
  241. v.clear();
  242. s = "k; ";
  243. util::split(s.begin(), s.end(), std::back_inserter(v), ';');
  244. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  245. itr = v.begin();
  246. CPPUNIT_ASSERT_EQUAL(std::string("k"), *itr++);
  247. CPPUNIT_ASSERT_EQUAL(std::string(" "), *itr++);
  248. v.clear();
  249. s = " ";
  250. util::split(s.begin(), s.end(), std::back_inserter(v), ';', true, true);
  251. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  252. CPPUNIT_ASSERT_EQUAL(std::string(""), v[0]);
  253. v.clear();
  254. s = " ";
  255. util::split(s.begin(), s.end(), std::back_inserter(v), ';', true);
  256. CPPUNIT_ASSERT_EQUAL((size_t)0, v.size());
  257. v.clear();
  258. s = " ";
  259. util::split(s.begin(), s.end(), std::back_inserter(v), ';');
  260. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  261. CPPUNIT_ASSERT_EQUAL(std::string(" "), v[0]);
  262. v.clear();
  263. s = ";";
  264. util::split(s.begin(), s.end(), std::back_inserter(v), ';');
  265. CPPUNIT_ASSERT_EQUAL((size_t)0, v.size());
  266. v.clear();
  267. s = ";";
  268. util::split(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  269. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  270. itr = v.begin();
  271. CPPUNIT_ASSERT_EQUAL(std::string(""), *itr++);
  272. CPPUNIT_ASSERT_EQUAL(std::string(""), *itr++);
  273. v.clear();
  274. s = "";
  275. util::split(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  276. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  277. CPPUNIT_ASSERT_EQUAL(std::string(""), v[0]);
  278. }
  279. void UtilTest1::testSplitIter()
  280. {
  281. std::vector<Scip> v;
  282. std::string s = "k1; k2;; k3";
  283. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', true);
  284. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  285. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  286. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[1].first, v[1].second));
  287. CPPUNIT_ASSERT_EQUAL(std::string("k3"), std::string(v[2].first, v[2].second));
  288. v.clear();
  289. s = "k1; k2; k3";
  290. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';');
  291. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  292. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  293. CPPUNIT_ASSERT_EQUAL(std::string(" k2"),
  294. std::string(v[1].first, v[1].second));
  295. CPPUNIT_ASSERT_EQUAL(std::string(" k3"),
  296. std::string(v[2].first, v[2].second));
  297. v.clear();
  298. s = "k=v";
  299. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  300. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  301. CPPUNIT_ASSERT_EQUAL(std::string("k=v"),
  302. std::string(v[0].first, v[0].second));
  303. v.clear();
  304. s = ";;k1;;k2;";
  305. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  306. CPPUNIT_ASSERT_EQUAL((size_t)6, v.size());
  307. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  308. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[1].first, v[1].second));
  309. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[2].first, v[2].second));
  310. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[3].first, v[3].second));
  311. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[4].first, v[4].second));
  312. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[5].first, v[5].second));
  313. v.clear();
  314. s = ";;k1;;k2;";
  315. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';');
  316. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  317. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  318. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[1].first, v[1].second));
  319. v.clear();
  320. s = "k; ";
  321. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';');
  322. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  323. CPPUNIT_ASSERT_EQUAL(std::string("k"), std::string(v[0].first, v[0].second));
  324. CPPUNIT_ASSERT_EQUAL(std::string(" "), std::string(v[1].first, v[1].second));
  325. v.clear();
  326. s = " ";
  327. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', true, true);
  328. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  329. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  330. v.clear();
  331. s = " ";
  332. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', true);
  333. CPPUNIT_ASSERT_EQUAL((size_t)0, v.size());
  334. v.clear();
  335. s = " ";
  336. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';');
  337. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  338. CPPUNIT_ASSERT_EQUAL(std::string(" "), std::string(v[0].first, v[0].second));
  339. v.clear();
  340. s = ";";
  341. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';');
  342. CPPUNIT_ASSERT_EQUAL((size_t)0, v.size());
  343. v.clear();
  344. s = ";";
  345. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  346. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  347. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  348. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[1].first, v[1].second));
  349. v.clear();
  350. s = "";
  351. util::splitIter(s.begin(), s.end(), std::back_inserter(v), ';', false, true);
  352. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  353. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  354. }
  355. void UtilTest1::testSplitIterM()
  356. {
  357. const char d[] = ";";
  358. const char md[] = "; ";
  359. std::vector<Scip> v;
  360. std::string s = "k1; k2;; k3";
  361. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, true);
  362. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  363. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  364. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[1].first, v[1].second));
  365. CPPUNIT_ASSERT_EQUAL(std::string("k3"), std::string(v[2].first, v[2].second));
  366. v.clear();
  367. s = "k1; k2; k3";
  368. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d);
  369. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  370. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  371. CPPUNIT_ASSERT_EQUAL(std::string(" k2"),
  372. std::string(v[1].first, v[1].second));
  373. CPPUNIT_ASSERT_EQUAL(std::string(" k3"),
  374. std::string(v[2].first, v[2].second));
  375. v.clear();
  376. s = "k1; k2; k3";
  377. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), md);
  378. CPPUNIT_ASSERT_EQUAL((size_t)3, v.size());
  379. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  380. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[1].first, v[1].second));
  381. CPPUNIT_ASSERT_EQUAL(std::string("k3"), std::string(v[2].first, v[2].second));
  382. v.clear();
  383. s = "k1; k2; k3;";
  384. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), md, false, true);
  385. CPPUNIT_ASSERT_EQUAL((size_t)6, v.size());
  386. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  387. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[1].first, v[1].second));
  388. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[2].first, v[2].second));
  389. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[3].first, v[3].second));
  390. CPPUNIT_ASSERT_EQUAL(std::string("k3"), std::string(v[4].first, v[4].second));
  391. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[5].first, v[5].second));
  392. v.clear();
  393. s = "k=v";
  394. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, false, true);
  395. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  396. CPPUNIT_ASSERT_EQUAL(std::string("k=v"),
  397. std::string(v[0].first, v[0].second));
  398. v.clear();
  399. s = ";;k1;;k2;";
  400. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, false, true);
  401. CPPUNIT_ASSERT_EQUAL((size_t)6, v.size());
  402. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  403. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[1].first, v[1].second));
  404. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[2].first, v[2].second));
  405. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[3].first, v[3].second));
  406. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[4].first, v[4].second));
  407. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[5].first, v[5].second));
  408. v.clear();
  409. s = ";;k1;;k2;";
  410. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d);
  411. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  412. CPPUNIT_ASSERT_EQUAL(std::string("k1"), std::string(v[0].first, v[0].second));
  413. CPPUNIT_ASSERT_EQUAL(std::string("k2"), std::string(v[1].first, v[1].second));
  414. v.clear();
  415. s = "k; ";
  416. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d);
  417. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  418. CPPUNIT_ASSERT_EQUAL(std::string("k"), std::string(v[0].first, v[0].second));
  419. CPPUNIT_ASSERT_EQUAL(std::string(" "), std::string(v[1].first, v[1].second));
  420. v.clear();
  421. s = " ";
  422. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, true, true);
  423. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  424. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  425. v.clear();
  426. s = " ";
  427. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, true);
  428. CPPUNIT_ASSERT_EQUAL((size_t)0, v.size());
  429. v.clear();
  430. s = " ";
  431. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d);
  432. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  433. CPPUNIT_ASSERT_EQUAL(std::string(" "), std::string(v[0].first, v[0].second));
  434. v.clear();
  435. s = ";";
  436. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d);
  437. CPPUNIT_ASSERT_EQUAL((size_t)0, v.size());
  438. v.clear();
  439. s = ";";
  440. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, false, true);
  441. CPPUNIT_ASSERT_EQUAL((size_t)2, v.size());
  442. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  443. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[1].first, v[1].second));
  444. v.clear();
  445. s = "";
  446. util::splitIterM(s.begin(), s.end(), std::back_inserter(v), d, false, true);
  447. CPPUNIT_ASSERT_EQUAL((size_t)1, v.size());
  448. CPPUNIT_ASSERT_EQUAL(std::string(""), std::string(v[0].first, v[0].second));
  449. }
  450. void UtilTest1::testEndsWith()
  451. {
  452. std::string target = "abcdefg";
  453. std::string part = "fg";
  454. CPPUNIT_ASSERT(
  455. util::endsWith(target.begin(), target.end(), part.begin(), part.end()));
  456. target = "abdefg";
  457. part = "g";
  458. CPPUNIT_ASSERT(
  459. util::endsWith(target.begin(), target.end(), part.begin(), part.end()));
  460. target = "abdefg";
  461. part = "eg";
  462. CPPUNIT_ASSERT(
  463. !util::endsWith(target.begin(), target.end(), part.begin(), part.end()));
  464. target = "g";
  465. part = "eg";
  466. CPPUNIT_ASSERT(
  467. !util::endsWith(target.begin(), target.end(), part.begin(), part.end()));
  468. target = "g";
  469. part = "g";
  470. CPPUNIT_ASSERT(
  471. util::endsWith(target.begin(), target.end(), part.begin(), part.end()));
  472. target = "g";
  473. part = "";
  474. CPPUNIT_ASSERT(
  475. util::endsWith(target.begin(), target.end(), part.begin(), part.end()));
  476. target = "";
  477. part = "";
  478. CPPUNIT_ASSERT(
  479. util::endsWith(target.begin(), target.end(), part.begin(), part.end()));
  480. target = "";
  481. part = "g";
  482. CPPUNIT_ASSERT(
  483. !util::endsWith(target.begin(), target.end(), part.begin(), part.end()));
  484. }
  485. void UtilTest1::testIendsWith()
  486. {
  487. std::string target = "abcdefg";
  488. std::string part = "Fg";
  489. CPPUNIT_ASSERT(
  490. util::iendsWith(target.begin(), target.end(), part.begin(), part.end()));
  491. target = "abdefg";
  492. part = "ef";
  493. CPPUNIT_ASSERT(
  494. !util::iendsWith(target.begin(), target.end(), part.begin(), part.end()));
  495. }
  496. void UtilTest1::testStreq()
  497. {
  498. std::string s1, s2;
  499. s1 = "foo";
  500. s2 = "foo";
  501. CPPUNIT_ASSERT(util::streq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  502. CPPUNIT_ASSERT(util::streq(s1.begin(), s1.end(), s2.c_str()));
  503. s2 = "fooo";
  504. CPPUNIT_ASSERT(!util::streq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  505. CPPUNIT_ASSERT(!util::streq(s1.begin(), s1.end(), s2.c_str()));
  506. s2 = "fo";
  507. CPPUNIT_ASSERT(!util::streq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  508. CPPUNIT_ASSERT(!util::streq(s1.begin(), s1.end(), s2.c_str()));
  509. s2 = "";
  510. CPPUNIT_ASSERT(!util::streq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  511. CPPUNIT_ASSERT(!util::streq(s1.begin(), s1.end(), s2.c_str()));
  512. s1 = "";
  513. CPPUNIT_ASSERT(util::streq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  514. CPPUNIT_ASSERT(util::streq(s1.begin(), s1.end(), s2.c_str()));
  515. }
  516. void UtilTest1::testStrieq()
  517. {
  518. std::string s1, s2;
  519. s1 = "foo";
  520. s2 = "foo";
  521. CPPUNIT_ASSERT(util::strieq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  522. CPPUNIT_ASSERT(util::strieq(s1.begin(), s1.end(), s2.c_str()));
  523. s1 = "FoO";
  524. s2 = "fOo";
  525. CPPUNIT_ASSERT(util::strieq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  526. CPPUNIT_ASSERT(util::strieq(s1.begin(), s1.end(), s2.c_str()));
  527. s2 = "fooo";
  528. CPPUNIT_ASSERT(!util::strieq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  529. CPPUNIT_ASSERT(!util::strieq(s1.begin(), s1.end(), s2.c_str()));
  530. s2 = "fo";
  531. CPPUNIT_ASSERT(!util::strieq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  532. CPPUNIT_ASSERT(!util::strieq(s1.begin(), s1.end(), s2.c_str()));
  533. s2 = "";
  534. CPPUNIT_ASSERT(!util::strieq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  535. CPPUNIT_ASSERT(!util::strieq(s1.begin(), s1.end(), s2.c_str()));
  536. s1 = "";
  537. CPPUNIT_ASSERT(util::strieq(s1.begin(), s1.end(), s2.begin(), s2.end()));
  538. CPPUNIT_ASSERT(util::strieq(s1.begin(), s1.end(), s2.c_str()));
  539. }
  540. void UtilTest1::testStrifind()
  541. {
  542. std::string s1, s2;
  543. s1 = "yamagakani mukashi wo toheba hARU no tuki";
  544. s2 = "HaRu";
  545. CPPUNIT_ASSERT(util::strifind(s1.begin(), s1.end(), s2.begin(), s2.end()) !=
  546. s1.end());
  547. s2 = "aki";
  548. CPPUNIT_ASSERT(util::strifind(s1.begin(), s1.end(), s2.begin(), s2.end()) ==
  549. s1.end());
  550. s1 = "h";
  551. s2 = "HH";
  552. CPPUNIT_ASSERT(util::strifind(s1.begin(), s1.end(), s2.begin(), s2.end()) ==
  553. s1.end());
  554. }
  555. void UtilTest1::testReplace()
  556. {
  557. CPPUNIT_ASSERT_EQUAL(std::string("abc\n"),
  558. util::replace("abc\r\n", "\r", ""));
  559. CPPUNIT_ASSERT_EQUAL(std::string("abc"),
  560. util::replace("abc\r\n", "\r\n", ""));
  561. CPPUNIT_ASSERT_EQUAL(std::string(""), util::replace("", "\r\n", ""));
  562. CPPUNIT_ASSERT_EQUAL(std::string("abc"), util::replace("abc", "", "a"));
  563. CPPUNIT_ASSERT_EQUAL(std::string("xbc"), util::replace("abc", "a", "x"));
  564. }
  565. void UtilTest1::testStartsWith()
  566. {
  567. std::string target;
  568. std::string part;
  569. target = "abcdefg";
  570. part = "abc";
  571. CPPUNIT_ASSERT(
  572. util::startsWith(target.begin(), target.end(), part.begin(), part.end()));
  573. CPPUNIT_ASSERT(util::startsWith(target.begin(), target.end(), part.c_str()));
  574. target = "abcdefg";
  575. part = "abx";
  576. CPPUNIT_ASSERT(!util::startsWith(target.begin(), target.end(), part.begin(),
  577. part.end()));
  578. CPPUNIT_ASSERT(!util::startsWith(target.begin(), target.end(), part.c_str()));
  579. target = "abcdefg";
  580. part = "bcd";
  581. CPPUNIT_ASSERT(!util::startsWith(target.begin(), target.end(), part.begin(),
  582. part.end()));
  583. CPPUNIT_ASSERT(!util::startsWith(target.begin(), target.end(), part.c_str()));
  584. target = "";
  585. part = "a";
  586. CPPUNIT_ASSERT(!util::startsWith(target.begin(), target.end(), part.begin(),
  587. part.end()));
  588. CPPUNIT_ASSERT(!util::startsWith(target.begin(), target.end(), part.c_str()));
  589. target = "";
  590. part = "";
  591. CPPUNIT_ASSERT(
  592. util::startsWith(target.begin(), target.end(), part.begin(), part.end()));
  593. CPPUNIT_ASSERT(util::startsWith(target.begin(), target.end(), part.c_str()));
  594. target = "a";
  595. part = "";
  596. CPPUNIT_ASSERT(
  597. util::startsWith(target.begin(), target.end(), part.begin(), part.end()));
  598. CPPUNIT_ASSERT(util::startsWith(target.begin(), target.end(), part.c_str()));
  599. target = "a";
  600. part = "a";
  601. CPPUNIT_ASSERT(
  602. util::startsWith(target.begin(), target.end(), part.begin(), part.end()));
  603. CPPUNIT_ASSERT(util::startsWith(target.begin(), target.end(), part.c_str()));
  604. }
  605. void UtilTest1::testIstartsWith()
  606. {
  607. std::string target;
  608. std::string part;
  609. target = "abcdefg";
  610. part = "aBc";
  611. CPPUNIT_ASSERT(util::istartsWith(target.begin(), target.end(), part.begin(),
  612. part.end()));
  613. CPPUNIT_ASSERT(util::istartsWith(target.begin(), target.end(), part.c_str()));
  614. target = "abcdefg";
  615. part = "abx";
  616. CPPUNIT_ASSERT(!util::istartsWith(target.begin(), target.end(), part.begin(),
  617. part.end()));
  618. CPPUNIT_ASSERT(
  619. !util::istartsWith(target.begin(), target.end(), part.c_str()));
  620. }
  621. void UtilTest1::testGetContentDispositionFilename()
  622. {
  623. std::string val;
  624. val = "attachment; filename=\"aria2.tar.bz2\"";
  625. CPPUNIT_ASSERT_EQUAL(std::string("aria2.tar.bz2"),
  626. util::getContentDispositionFilename(val, false));
  627. val = "attachment; filename=\"\"";
  628. CPPUNIT_ASSERT_EQUAL(std::string(""),
  629. util::getContentDispositionFilename(val, false));
  630. val = "attachment; filename=\"";
  631. CPPUNIT_ASSERT_EQUAL(std::string(""),
  632. util::getContentDispositionFilename(val, false));
  633. val = "attachment; filename= \" aria2.tar.bz2 \"";
  634. CPPUNIT_ASSERT_EQUAL(std::string(" aria2.tar.bz2 "),
  635. util::getContentDispositionFilename(val, false));
  636. val = "attachment; filename=dir/file";
  637. CPPUNIT_ASSERT_EQUAL(std::string(""),
  638. util::getContentDispositionFilename(val, false));
  639. val = "attachment; filename=dir\\file";
  640. CPPUNIT_ASSERT_EQUAL(std::string(""),
  641. util::getContentDispositionFilename(val, false));
  642. val = "attachment; filename=\"dir/file\"";
  643. CPPUNIT_ASSERT_EQUAL(std::string(""),
  644. util::getContentDispositionFilename(val, false));
  645. val = "attachment; filename=\"dir\\\\file\"";
  646. CPPUNIT_ASSERT_EQUAL(std::string(""),
  647. util::getContentDispositionFilename(val, false));
  648. val = "attachment; filename=\"/etc/passwd\"";
  649. CPPUNIT_ASSERT_EQUAL(std::string(""),
  650. util::getContentDispositionFilename(val, false));
  651. val = "attachment; filename=\"..\"";
  652. CPPUNIT_ASSERT_EQUAL(std::string(""),
  653. util::getContentDispositionFilename(val, false));
  654. val = "attachment; filename=..";
  655. CPPUNIT_ASSERT_EQUAL(std::string(""),
  656. util::getContentDispositionFilename(val, false));
  657. // Unescaping %2E%2E%2F produces "../". But since we won't unescape,
  658. // we just accept it as is.
  659. val = "attachment; filename=\"%2E%2E%2Ffoo.html\"";
  660. CPPUNIT_ASSERT_EQUAL(std::string("%2E%2E%2Ffoo.html"),
  661. util::getContentDispositionFilename(val, false));
  662. // iso-8859-1 string will be converted to utf-8.
  663. val = "attachment; filename*=iso-8859-1''foo-%E4.html";
  664. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  665. util::getContentDispositionFilename(val, false));
  666. val = "attachment; filename*= UTF-8''foo-%c3%a4.html";
  667. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  668. util::getContentDispositionFilename(val, false));
  669. // iso-8859-1 string will be converted to utf-8.
  670. val = "attachment; filename=\"foo-%E4.html\"";
  671. val = util::percentDecode(val.begin(), val.end());
  672. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  673. util::getContentDispositionFilename(val, false));
  674. // allow utf-8 in filename if default_utf8 is set.
  675. val = "attachment; filename=\"foo-ä.html\"";
  676. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  677. util::getContentDispositionFilename(val, true));
  678. // return empty if default_utf8 is set but invalid utf8.
  679. val = "attachment; filename=\"foo-\xc2\x02.html\"";
  680. CPPUNIT_ASSERT_EQUAL(std::string(""),
  681. util::getContentDispositionFilename(val, true));
  682. }
  683. void UtilTest1::testParseContentDisposition1()
  684. {
  685. char dest[1_k];
  686. size_t destlen = sizeof(dest);
  687. const char* cs;
  688. size_t cslen;
  689. std::string val;
  690. // test cases from http://greenbytes.de/tech/tc2231/
  691. // inlonly
  692. val = "inline";
  693. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition(
  694. dest, destlen, &cs, &cslen, val.c_str(),
  695. val.size(), false));
  696. // inlonlyquoted
  697. val = "\"inline\"";
  698. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  699. dest, destlen, &cs, &cslen, val.c_str(),
  700. val.size(), false));
  701. // inlwithasciifilename
  702. val = "inline; filename=\"foo.html\"";
  703. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  704. dest, destlen, &cs, &cslen, val.c_str(),
  705. val.size(), false));
  706. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  707. std::string(&dest[0], &dest[8]));
  708. // inlwithfnattach
  709. val = "inline; filename=\"Not an attachment!\"";
  710. CPPUNIT_ASSERT_EQUAL((ssize_t)18, util::parse_content_disposition(
  711. dest, destlen, &cs, &cslen, val.c_str(),
  712. val.size(), false));
  713. CPPUNIT_ASSERT_EQUAL(std::string("Not an attachment!"),
  714. std::string(&dest[0], &dest[18]));
  715. // inlwithasciifilenamepdf
  716. val = "inline; filename=\"foo.pdf\"";
  717. CPPUNIT_ASSERT_EQUAL((ssize_t)7, util::parse_content_disposition(
  718. dest, destlen, &cs, &cslen, val.c_str(),
  719. val.size(), false));
  720. CPPUNIT_ASSERT_EQUAL(std::string("foo.pdf"), std::string(&dest[0], &dest[7]));
  721. // attwithasciifilename25
  722. val = "attachment; filename=\"0000000000111111111122222\"";
  723. CPPUNIT_ASSERT_EQUAL((ssize_t)25, util::parse_content_disposition(
  724. dest, destlen, &cs, &cslen, val.c_str(),
  725. val.size(), false));
  726. CPPUNIT_ASSERT_EQUAL(std::string("0000000000111111111122222"),
  727. std::string(&dest[0], &dest[25]));
  728. // attwithasciifilename35
  729. val = "attachment; filename=\"00000000001111111111222222222233333\"";
  730. CPPUNIT_ASSERT_EQUAL((ssize_t)35, util::parse_content_disposition(
  731. dest, destlen, &cs, &cslen, val.c_str(),
  732. val.size(), false));
  733. CPPUNIT_ASSERT_EQUAL(std::string("00000000001111111111222222222233333"),
  734. std::string(&dest[0], &dest[35]));
  735. // attwithasciifnescapedchar
  736. val = "attachment; filename=\"f\\oo.html\"";
  737. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  738. dest, destlen, &cs, &cslen, val.c_str(),
  739. val.size(), false));
  740. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  741. std::string(&dest[0], &dest[8]));
  742. // attwithasciifnescapedquote
  743. val = "attachment; filename=\"\\\"quoting\\\" tested.html\"";
  744. CPPUNIT_ASSERT_EQUAL((ssize_t)21, util::parse_content_disposition(
  745. dest, destlen, &cs, &cslen, val.c_str(),
  746. val.size(), false));
  747. CPPUNIT_ASSERT_EQUAL(std::string("\"quoting\" tested.html"),
  748. std::string(&dest[0], &dest[21]));
  749. // attwithquotedsemicolon
  750. val = "attachment; filename=\"Here's a semicolon;.html\"";
  751. CPPUNIT_ASSERT_EQUAL((ssize_t)24, util::parse_content_disposition(
  752. dest, destlen, &cs, &cslen, val.c_str(),
  753. val.size(), false));
  754. CPPUNIT_ASSERT_EQUAL(std::string("Here's a semicolon;.html"),
  755. std::string(&dest[0], &dest[24]));
  756. // attwithfilenameandextparam
  757. val = "attachment; foo=\"bar\"; filename=\"foo.html\"";
  758. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  759. dest, destlen, &cs, &cslen, val.c_str(),
  760. val.size(), false));
  761. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  762. std::string(&dest[0], &dest[8]));
  763. // attwithfilenameandextparamescaped
  764. val = "attachment; foo=\"\\\"\\\\\";filename=\"foo.html\"";
  765. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  766. dest, destlen, &cs, &cslen, val.c_str(),
  767. val.size(), false));
  768. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  769. std::string(&dest[0], &dest[8]));
  770. // attwithasciifilenameucase
  771. val = "attachment; FILENAME=\"foo.html\"";
  772. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  773. dest, destlen, &cs, &cslen, val.c_str(),
  774. val.size(), false));
  775. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  776. std::string(&dest[0], &dest[8]));
  777. // attwithasciifilenamenq
  778. val = "attachment; filename=foo.html";
  779. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  780. dest, destlen, &cs, &cslen, val.c_str(),
  781. val.size(), false));
  782. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  783. std::string(&dest[0], &dest[8]));
  784. // attwithtokfncommanq
  785. val = "attachment; filename=foo,bar.html";
  786. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  787. dest, destlen, &cs, &cslen, val.c_str(),
  788. val.size(), false));
  789. // attwithasciifilenamenqs
  790. val = "attachment; filename=foo.html ;";
  791. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  792. dest, destlen, &cs, &cslen, val.c_str(),
  793. val.size(), false));
  794. // attemptyparam
  795. val = "attachment; ;filename=foo";
  796. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  797. dest, destlen, &cs, &cslen, val.c_str(),
  798. val.size(), false));
  799. // attwithasciifilenamenqws
  800. val = "attachment; filename=foo bar.html";
  801. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  802. dest, destlen, &cs, &cslen, val.c_str(),
  803. val.size(), false));
  804. // attwithfntokensq
  805. val = "attachment; filename='foo.bar'";
  806. CPPUNIT_ASSERT_EQUAL((ssize_t)9, util::parse_content_disposition(
  807. dest, destlen, &cs, &cslen, val.c_str(),
  808. val.size(), false));
  809. CPPUNIT_ASSERT_EQUAL(std::string("'foo.bar'"),
  810. std::string(&dest[0], &dest[9]));
  811. // attwithisofnplain
  812. // attachment; filename="foo-ä.html"
  813. val = "attachment; filename=\"foo-%E4.html\"";
  814. val = util::percentDecode(val.begin(), val.end());
  815. CPPUNIT_ASSERT_EQUAL((ssize_t)10, util::parse_content_disposition(
  816. dest, destlen, &cs, &cslen, val.c_str(),
  817. val.size(), false));
  818. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  819. util::iso8859p1ToUtf8(std::string(&dest[0], &dest[10])));
  820. // attwithutf8fnplain
  821. // attachment; filename="foo-ä.html"
  822. val = "attachment; filename=\"foo-%C3%A4.html\"";
  823. val = util::percentDecode(val.begin(), val.end());
  824. CPPUNIT_ASSERT_EQUAL((ssize_t)11, util::parse_content_disposition(
  825. dest, destlen, &cs, &cslen, val.c_str(),
  826. val.size(), false));
  827. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  828. util::iso8859p1ToUtf8(std::string(&dest[0], &dest[11])));
  829. // attwithfnrawpctenca
  830. val = "attachment; filename=\"foo-%41.html\"";
  831. CPPUNIT_ASSERT_EQUAL((ssize_t)12, util::parse_content_disposition(
  832. dest, destlen, &cs, &cslen, val.c_str(),
  833. val.size(), false));
  834. CPPUNIT_ASSERT_EQUAL(std::string("foo-%41.html"),
  835. std::string(&dest[0], &dest[12]));
  836. // attwithfnusingpct
  837. val = "attachment; filename=\"50%.html\"";
  838. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  839. dest, destlen, &cs, &cslen, val.c_str(),
  840. val.size(), false));
  841. CPPUNIT_ASSERT_EQUAL(std::string("50%.html"),
  842. std::string(&dest[0], &dest[8]));
  843. // attwithfnrawpctencaq
  844. val = "attachment; filename=\"foo-%\\41.html\"";
  845. CPPUNIT_ASSERT_EQUAL((ssize_t)12, util::parse_content_disposition(
  846. dest, destlen, &cs, &cslen, val.c_str(),
  847. val.size(), false));
  848. CPPUNIT_ASSERT_EQUAL(std::string("foo-%41.html"),
  849. std::string(&dest[0], &dest[12]));
  850. // attwithnamepct
  851. val = "attachment; name=\"foo-%41.html\"";
  852. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition(
  853. dest, destlen, &cs, &cslen, val.c_str(),
  854. val.size(), false));
  855. // attwithfilenamepctandiso
  856. // attachment; filename="ä-%41.html"
  857. val = "attachment; filename=\"%E4-%2541.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(),
  861. val.size(), false));
  862. CPPUNIT_ASSERT_EQUAL(std::string("ä-%41.html"),
  863. util::iso8859p1ToUtf8(std::string(&dest[0], &dest[10])));
  864. // attwithfnrawpctenclong
  865. val = "attachment; filename=\"foo-%c3%a4-%e2%82%ac.html\"";
  866. CPPUNIT_ASSERT_EQUAL((ssize_t)25, util::parse_content_disposition(
  867. dest, destlen, &cs, &cslen, val.c_str(),
  868. val.size(), false));
  869. CPPUNIT_ASSERT_EQUAL(std::string("foo-%c3%a4-%e2%82%ac.html"),
  870. std::string(&dest[0], &dest[25]));
  871. // attwithasciifilenamews1
  872. val = "attachment; filename =\"foo.html\"";
  873. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  874. dest, destlen, &cs, &cslen, val.c_str(),
  875. val.size(), false));
  876. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  877. std::string(&dest[0], &dest[8]));
  878. // attwith2filenames
  879. val = "attachment; filename=\"foo.html\"; filename=\"bar.html\"";
  880. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  881. dest, destlen, &cs, &cslen, val.c_str(),
  882. val.size(), false));
  883. // attfnbrokentoken
  884. val = "attachment; filename=foo[1](2).html";
  885. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  886. dest, destlen, &cs, &cslen, val.c_str(),
  887. val.size(), false));
  888. // attfnbrokentokeniso
  889. val = "attachment; filename=foo-%E4.html";
  890. val = util::percentDecode(val.begin(), val.end());
  891. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  892. dest, destlen, &cs, &cslen, val.c_str(),
  893. val.size(), false));
  894. // attfnbrokentokenutf
  895. // attachment; filename=foo-ä.html
  896. val = "attachment; filename=foo-ä.html";
  897. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  898. dest, destlen, &cs, &cslen, val.c_str(),
  899. val.size(), false));
  900. // attmissingdisposition
  901. val = "filename=foo.html";
  902. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  903. dest, destlen, &cs, &cslen, val.c_str(),
  904. val.size(), false));
  905. // attmissingdisposition2
  906. val = "x=y; filename=foo.html";
  907. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  908. dest, destlen, &cs, &cslen, val.c_str(),
  909. val.size(), false));
  910. // attmissingdisposition3
  911. val = "\"foo; filename=bar;baz\"; filename=qux";
  912. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  913. dest, destlen, &cs, &cslen, val.c_str(),
  914. val.size(), false));
  915. // attmissingdisposition4
  916. val = "filename=foo.html, filename=bar.html";
  917. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  918. dest, destlen, &cs, &cslen, val.c_str(),
  919. val.size(), false));
  920. // emptydisposition
  921. val = "; filename=foo.html";
  922. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  923. dest, destlen, &cs, &cslen, val.c_str(),
  924. val.size(), false));
  925. // doublecolon
  926. val = ": inline; attachment; filename=foo.html";
  927. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  928. dest, destlen, &cs, &cslen, val.c_str(),
  929. val.size(), false));
  930. // attandinline
  931. val = "inline; attachment; filename=foo.html";
  932. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  933. dest, destlen, &cs, &cslen, val.c_str(),
  934. val.size(), false));
  935. // attandinline2
  936. val = "attachment; inline; filename=foo.html";
  937. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  938. dest, destlen, &cs, &cslen, val.c_str(),
  939. val.size(), false));
  940. // attbrokenquotedfn
  941. val = "attachment; filename=\"foo.html\".txt";
  942. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  943. dest, destlen, &cs, &cslen, val.c_str(),
  944. val.size(), false));
  945. // attbrokenquotedfn2
  946. val = "attachment; filename=\"bar";
  947. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  948. dest, destlen, &cs, &cslen, val.c_str(),
  949. val.size(), false));
  950. // attbrokenquotedfn3
  951. val = "attachment; filename=foo\"bar;baz\"qux";
  952. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  953. dest, destlen, &cs, &cslen, val.c_str(),
  954. val.size(), false));
  955. // attmultinstances
  956. val = "attachment; filename=foo.html, attachment; filename=bar.html";
  957. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  958. dest, destlen, &cs, &cslen, val.c_str(),
  959. val.size(), false));
  960. }
  961. void UtilTest1::testParseContentDisposition2()
  962. {
  963. char dest[1_k];
  964. size_t destlen = sizeof(dest);
  965. const char* cs;
  966. size_t cslen;
  967. std::string val;
  968. // test cases from http://greenbytes.de/tech/tc2231/
  969. // attmissingdelim
  970. val = "attachment; foo=foo filename=bar";
  971. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  972. dest, destlen, &cs, &cslen, val.c_str(),
  973. val.size(), false));
  974. // attmissingdelim2
  975. val = "attachment; filename=bar foo=foo ";
  976. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  977. dest, destlen, &cs, &cslen, val.c_str(),
  978. val.size(), false));
  979. // attmissingdelim3
  980. val = "attachment filename=bar";
  981. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  982. dest, destlen, &cs, &cslen, val.c_str(),
  983. val.size(), false));
  984. // attreversed
  985. val = "filename=foo.html; attachment";
  986. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  987. dest, destlen, &cs, &cslen, val.c_str(),
  988. val.size(), false));
  989. // attconfusedparam
  990. val = "attachment; xfilename=foo.html";
  991. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition(
  992. dest, destlen, &cs, &cslen, val.c_str(),
  993. val.size(), false));
  994. // attabspath
  995. val = "attachment; filename=\"/foo.html\"";
  996. CPPUNIT_ASSERT_EQUAL((ssize_t)9, util::parse_content_disposition(
  997. dest, destlen, &cs, &cslen, val.c_str(),
  998. val.size(), false));
  999. CPPUNIT_ASSERT_EQUAL(std::string("/foo.html"),
  1000. std::string(&dest[0], &dest[9]));
  1001. // attabspathwin
  1002. val = "attachment; filename=\"\\\\foo.html\"";
  1003. CPPUNIT_ASSERT_EQUAL((ssize_t)9, util::parse_content_disposition(
  1004. dest, destlen, &cs, &cslen, val.c_str(),
  1005. val.size(), false));
  1006. CPPUNIT_ASSERT_EQUAL(std::string("\\foo.html"),
  1007. std::string(&dest[0], &dest[9]));
  1008. // attcdate
  1009. val = "attachment; creation-date=\"Wed, 12 Feb 1997 16:29:51 -0500\"";
  1010. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition(
  1011. dest, destlen, &cs, &cslen, val.c_str(),
  1012. val.size(), false));
  1013. // dispext
  1014. val = "foobar";
  1015. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition(
  1016. dest, destlen, &cs, &cslen, val.c_str(),
  1017. val.size(), false));
  1018. // dispextbadfn
  1019. val = "attachment; example=\"filename=example.txt\"";
  1020. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition(
  1021. dest, destlen, &cs, &cslen, val.c_str(),
  1022. val.size(), false));
  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(),
  1027. val.size(), false));
  1028. CPPUNIT_ASSERT_EQUAL(std::string("iso-8859-1"), std::string(cs, cslen));
  1029. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  1030. util::iso8859p1ToUtf8(std::string(&dest[0], &dest[10])));
  1031. // attwithfn2231utf8
  1032. val = "attachment; filename*=UTF-8''foo-%c3%a4-%e2%82%ac.html";
  1033. CPPUNIT_ASSERT_EQUAL((ssize_t)15, util::parse_content_disposition(
  1034. dest, destlen, &cs, &cslen, val.c_str(),
  1035. val.size(), false));
  1036. CPPUNIT_ASSERT_EQUAL(std::string("UTF-8"), std::string(cs, cslen));
  1037. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä-€.html"),
  1038. std::string(&dest[0], &dest[15]));
  1039. // attwithfn2231noc
  1040. val = "attachment; filename*=''foo-%c3%a4-%e2%82%ac.html";
  1041. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  1042. dest, destlen, &cs, &cslen, val.c_str(),
  1043. val.size(), false));
  1044. // attwithfn2231utf8comp
  1045. val = "attachment; filename*=UTF-8''foo-a%cc%88.html";
  1046. CPPUNIT_ASSERT_EQUAL((ssize_t)12, util::parse_content_disposition(
  1047. dest, destlen, &cs, &cslen, val.c_str(),
  1048. val.size(), false));
  1049. val = "foo-a%cc%88.html";
  1050. CPPUNIT_ASSERT_EQUAL(std::string(util::percentDecode(val.begin(), val.end())),
  1051. std::string(&dest[0], &dest[12]));
  1052. // attwithfn2231utf8-bad
  1053. val = "attachment; filename*=iso-8859-1''foo-%c3%a4-%e2%82%ac.html";
  1054. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  1055. dest, destlen, &cs, &cslen, val.c_str(),
  1056. val.size(), false));
  1057. // attwithfn2231iso-bad
  1058. val = "attachment; filename*=utf-8''foo-%E4.html";
  1059. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  1060. dest, destlen, &cs, &cslen, val.c_str(),
  1061. val.size(), false));
  1062. // attwithfn2231ws1
  1063. val = "attachment; filename *=UTF-8''foo-%c3%a4.html";
  1064. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  1065. dest, destlen, &cs, &cslen, val.c_str(),
  1066. val.size(), false));
  1067. // attwithfn2231ws2
  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(),
  1071. val.size(), false));
  1072. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  1073. std::string(&dest[0], &dest[11]));
  1074. // attwithfn2231ws3
  1075. val = "attachment; filename* =UTF-8''foo-%c3%a4.html";
  1076. CPPUNIT_ASSERT_EQUAL((ssize_t)11, util::parse_content_disposition(
  1077. dest, destlen, &cs, &cslen, val.c_str(),
  1078. val.size(), false));
  1079. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  1080. std::string(&dest[0], &dest[11]));
  1081. // attwithfn2231quot
  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(),
  1085. val.size(), false));
  1086. // attwithfn2231quot2
  1087. val = "attachment; filename*=\"foo%20bar.html\"";
  1088. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  1089. dest, destlen, &cs, &cslen, val.c_str(),
  1090. val.size(), false));
  1091. // attwithfn2231singleqmissing
  1092. val = "attachment; filename*=UTF-8'foo-%c3%a4.html";
  1093. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  1094. dest, destlen, &cs, &cslen, val.c_str(),
  1095. val.size(), false));
  1096. // attwithfn2231nbadpct1
  1097. val = "attachment; filename*=UTF-8''foo%";
  1098. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  1099. dest, destlen, &cs, &cslen, val.c_str(),
  1100. val.size(), false));
  1101. // attwithfn2231nbadpct2
  1102. val = "attachment; filename*=UTF-8''f%oo.html";
  1103. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  1104. dest, destlen, &cs, &cslen, val.c_str(),
  1105. val.size(), false));
  1106. // attwithfn2231dpct
  1107. val = "attachment; filename*=UTF-8''A-%2541.html";
  1108. CPPUNIT_ASSERT_EQUAL((ssize_t)10, util::parse_content_disposition(
  1109. dest, destlen, &cs, &cslen, val.c_str(),
  1110. val.size(), false));
  1111. CPPUNIT_ASSERT_EQUAL(std::string("A-%41.html"),
  1112. std::string(&dest[0], &dest[10]));
  1113. // attwithfn2231abspathdisguised
  1114. val = "attachment; filename*=UTF-8''%5cfoo.html";
  1115. CPPUNIT_ASSERT_EQUAL((ssize_t)9, util::parse_content_disposition(
  1116. dest, destlen, &cs, &cslen, val.c_str(),
  1117. val.size(), false));
  1118. CPPUNIT_ASSERT_EQUAL(std::string("\\foo.html"),
  1119. std::string(&dest[0], &dest[9]));
  1120. // attfnboth
  1121. val =
  1122. "attachment; filename=\"foo-ae.html\"; filename*=UTF-8''foo-%c3%a4.html";
  1123. CPPUNIT_ASSERT_EQUAL((ssize_t)11, util::parse_content_disposition(
  1124. dest, destlen, &cs, &cslen, val.c_str(),
  1125. val.size(), false));
  1126. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  1127. std::string(&dest[0], &dest[11]));
  1128. // attfnboth2
  1129. val =
  1130. "attachment; filename*=UTF-8''foo-%c3%a4.html; filename=\"foo-ae.html\"";
  1131. CPPUNIT_ASSERT_EQUAL((ssize_t)11, util::parse_content_disposition(
  1132. dest, destlen, &cs, &cslen, val.c_str(),
  1133. val.size(), false));
  1134. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  1135. std::string(&dest[0], &dest[11]));
  1136. // attfnboth3
  1137. val = "attachment; filename*0*=ISO-8859-15''euro-sign%3d%a4; "
  1138. "filename*=ISO-8859-1''currency-sign%3d%a4";
  1139. CPPUNIT_ASSERT_EQUAL((ssize_t)15, util::parse_content_disposition(
  1140. dest, destlen, &cs, &cslen, val.c_str(),
  1141. val.size(), false));
  1142. CPPUNIT_ASSERT_EQUAL(std::string("ISO-8859-1"), std::string(cs, cslen));
  1143. CPPUNIT_ASSERT_EQUAL(std::string("currency-sign=¤"),
  1144. util::iso8859p1ToUtf8(std::string(&dest[0], &dest[15])));
  1145. // attnewandfn
  1146. val = "attachment; foobar=x; filename=\"foo.html\"";
  1147. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  1148. dest, destlen, &cs, &cslen, val.c_str(),
  1149. val.size(), false));
  1150. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1151. std::string(&dest[0], &dest[8]));
  1152. // attrfc2047token
  1153. val = "attachment; filename==?ISO-8859-1?Q?foo-=E4.html?=";
  1154. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  1155. dest, destlen, &cs, &cslen, val.c_str(),
  1156. val.size(), false));
  1157. // attrfc2047quoted
  1158. val = "attachment; filename=\"=?ISO-8859-1?Q?foo-=E4.html?=\"";
  1159. CPPUNIT_ASSERT_EQUAL((ssize_t)29, util::parse_content_disposition(
  1160. dest, destlen, &cs, &cslen, val.c_str(),
  1161. val.size(), false));
  1162. CPPUNIT_ASSERT_EQUAL(std::string("=?ISO-8859-1?Q?foo-=E4.html?="),
  1163. std::string(&dest[0], &dest[29]));
  1164. // aria2 original testcases
  1165. // zero-length filename. token cannot be empty, so this is invalid.
  1166. val = "attachment; filename=";
  1167. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  1168. dest, destlen, &cs, &cslen, val.c_str(),
  1169. val.size(), false));
  1170. // zero-length filename. quoted-string can be empty string, so this
  1171. // is ok.
  1172. val = "attachment; filename=\"\"";
  1173. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition(
  1174. dest, destlen, &cs, &cslen, val.c_str(),
  1175. val.size(), false));
  1176. // empty value is not allowed
  1177. val = "attachment; filename=;";
  1178. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  1179. dest, destlen, &cs, &cslen, val.c_str(),
  1180. val.size(), false));
  1181. // / is not valid char in token.
  1182. val = "attachment; filename=dir/file";
  1183. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  1184. dest, destlen, &cs, &cslen, val.c_str(),
  1185. val.size(), false));
  1186. // value-chars is *(pct-encoded / attr-char), so empty string is
  1187. // allowed.
  1188. val = "attachment; filename*=UTF-8''";
  1189. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition(
  1190. dest, destlen, &cs, &cslen, val.c_str(),
  1191. val.size(), false));
  1192. CPPUNIT_ASSERT_EQUAL(std::string("UTF-8"), std::string(cs, cslen));
  1193. val = "attachment; filename*=UTF-8''; filename=foo";
  1194. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition(
  1195. dest, destlen, &cs, &cslen, val.c_str(),
  1196. val.size(), false));
  1197. CPPUNIT_ASSERT_EQUAL(std::string("UTF-8"), std::string(cs, cslen));
  1198. val = "attachment; filename*=UTF-8'' ; filename=foo";
  1199. CPPUNIT_ASSERT_EQUAL((ssize_t)0, util::parse_content_disposition(
  1200. dest, destlen, &cs, &cslen, val.c_str(),
  1201. val.size(), false));
  1202. CPPUNIT_ASSERT_EQUAL(std::string("UTF-8"), std::string(cs, cslen));
  1203. // with language
  1204. val = "attachment; filename*=UTF-8'japanese'konnichiwa";
  1205. CPPUNIT_ASSERT_EQUAL((ssize_t)10, util::parse_content_disposition(
  1206. dest, destlen, &cs, &cslen, val.c_str(),
  1207. val.size(), false));
  1208. CPPUNIT_ASSERT_EQUAL(std::string("konnichiwa"),
  1209. std::string(&dest[0], &dest[10]));
  1210. // lws before and after "="
  1211. val = "attachment; filename = foo.html";
  1212. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  1213. dest, destlen, &cs, &cslen, val.c_str(),
  1214. val.size(), false));
  1215. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1216. std::string(&dest[0], &dest[8]));
  1217. // lws before and after "=" with quoted-string
  1218. val = "attachment; filename = \"foo.html\"";
  1219. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  1220. dest, destlen, &cs, &cslen, val.c_str(),
  1221. val.size(), false));
  1222. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1223. std::string(&dest[0], &dest[8]));
  1224. // lws after parm
  1225. val = "attachment; filename=foo.html ";
  1226. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  1227. dest, destlen, &cs, &cslen, val.c_str(),
  1228. val.size(), false));
  1229. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1230. std::string(&dest[0], &dest[8]));
  1231. val = "attachment; filename=foo.html ; hello=world";
  1232. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  1233. dest, destlen, &cs, &cslen, val.c_str(),
  1234. val.size(), false));
  1235. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1236. std::string(&dest[0], &dest[8]));
  1237. val = "attachment; filename=\"foo.html\" ";
  1238. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  1239. dest, destlen, &cs, &cslen, val.c_str(),
  1240. val.size(), false));
  1241. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1242. std::string(&dest[0], &dest[8]));
  1243. val = "attachment; filename=\"foo.html\" ; hello=world";
  1244. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  1245. dest, destlen, &cs, &cslen, val.c_str(),
  1246. val.size(), false));
  1247. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1248. std::string(&dest[0], &dest[8]));
  1249. val = "attachment; filename*=UTF-8''foo.html ; hello=world";
  1250. CPPUNIT_ASSERT_EQUAL((ssize_t)8, util::parse_content_disposition(
  1251. dest, destlen, &cs, &cslen, val.c_str(),
  1252. val.size(), false));
  1253. CPPUNIT_ASSERT_EQUAL(std::string("foo.html"),
  1254. std::string(&dest[0], &dest[8]));
  1255. // allow utf8 if content-disposition-default-utf8 is set
  1256. val = "attachment; filename=\"foo-ä.html\"";
  1257. CPPUNIT_ASSERT_EQUAL((ssize_t)11, util::parse_content_disposition(
  1258. dest, destlen, &cs, &cslen, val.c_str(),
  1259. val.size(), true));
  1260. CPPUNIT_ASSERT_EQUAL(std::string("foo-ä.html"),
  1261. std::string(&dest[0], &dest[11]));
  1262. // incomplete utf8 sequence must be rejected
  1263. val = "attachment; filename=\"foo-\xc3.html\"";
  1264. CPPUNIT_ASSERT_EQUAL((ssize_t)-1, util::parse_content_disposition(
  1265. dest, destlen, &cs, &cslen, val.c_str(),
  1266. val.size(), true));
  1267. }
  1268. } // namespace aria2