jstring.cpp 56 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359
  1. /*##############################################################################
  2. HPCC SYSTEMS software Copyright (C) 2012 HPCC Systems.
  3. Licensed under the Apache License, Version 2.0 (the "License");
  4. you may not use this file except in compliance with the License.
  5. You may obtain a copy of the License at
  6. http://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. ############################################################################## */
  13. #include "platform.h"
  14. #include <stdio.h>
  15. #include <stdlib.h>
  16. #include <stdarg.h>
  17. #include <assert.h>
  18. #include <string.h>
  19. #include <ctype.h>
  20. #include <time.h>
  21. #include <math.h>
  22. #include "jstring.hpp"
  23. #include "jexcept.hpp"
  24. #include "jhash.hpp"
  25. #include "jlog.hpp"
  26. #include "jfile.hpp"
  27. #include "jdebug.hpp"
  28. #include "jutil.hpp"
  29. #define DOUBLE_FORMAT "%.16g"
  30. #define FLOAT_FORMAT "%.7g"
  31. #ifndef va_copy
  32. /* WARNING - DANGER - ASSUMES TYPICAL STACK MACHINE */
  33. #define va_copy(dst, src) ((void)((dst) = (src)))
  34. #endif
  35. static const char * TheNullStr = "";
  36. #define FIRST_CHUNK_SIZE 8
  37. #define DOUBLE_LIMIT 0x100000 // must be a power of 2
  38. #define DETACH_GRANULARITY 16
  39. //===========================================================================
  40. StringBuffer::StringBuffer()
  41. {
  42. init();
  43. }
  44. #if 0
  45. StringBuffer::StringBuffer(int initial)
  46. {
  47. init();
  48. ensureCapacity(initial);
  49. }
  50. #endif
  51. StringBuffer::StringBuffer(String & value)
  52. {
  53. init();
  54. append(value);
  55. }
  56. StringBuffer::StringBuffer(const char *value)
  57. {
  58. init();
  59. append(value);
  60. }
  61. StringBuffer::StringBuffer(unsigned len, const char *value)
  62. {
  63. init();
  64. append(len, value);
  65. }
  66. StringBuffer::StringBuffer(const StringBuffer & value)
  67. {
  68. init();
  69. append(value);
  70. }
  71. void StringBuffer::setBuffer(size32_t buffLen, char * newBuff, size32_t strLen)
  72. {
  73. assertex(buffLen>0 && newBuff!=NULL && strLen<buffLen);
  74. if (buffer)
  75. free(buffer);
  76. buffer = newBuff;
  77. maxLen=buffLen;
  78. curLen=strLen;
  79. }
  80. void StringBuffer::_realloc(size32_t newLen)
  81. {
  82. if (newLen >= maxLen)
  83. {
  84. size32_t newMax = maxLen;
  85. if (newMax == 0)
  86. newMax = FIRST_CHUNK_SIZE;
  87. if (newLen > DOUBLE_LIMIT)
  88. {
  89. newMax = (newLen + DOUBLE_LIMIT) & ~(DOUBLE_LIMIT-1);
  90. if (newLen >= newMax)
  91. throw MakeStringException(MSGAUD_operator, -1, "StringBuffer::_realloc: Request for %d bytes oldMax = %d", newLen, maxLen);
  92. }
  93. else
  94. {
  95. while (newLen >= newMax)
  96. newMax += newMax;
  97. }
  98. char * newStr;
  99. if(!newMax || !(newStr=(char *)realloc(buffer, newMax)))
  100. {
  101. DBGLOG("StringBuffer::_realloc: Failed to realloc = %d, oldMax = %d", newMax, maxLen);
  102. PrintStackReport();
  103. PrintMemoryReport();
  104. throw MakeStringException(MSGAUD_operator, -1, "StringBuffer::_realloc: Failed to realloc = %d, oldMax = %d", newMax, maxLen);
  105. }
  106. buffer = newStr;
  107. maxLen = newMax;
  108. }
  109. }
  110. char * StringBuffer::detach()
  111. {
  112. if (buffer)
  113. {
  114. if (maxLen>curLen+1+DETACH_GRANULARITY)
  115. buffer = (char *)realloc(buffer,curLen+1); // shrink
  116. buffer[curLen] = '\0'; // There is always room for this null
  117. char *ret = buffer;
  118. init();
  119. return ret;
  120. }
  121. return strdup(TheNullStr);
  122. }
  123. StringBuffer & StringBuffer::append(char value)
  124. {
  125. ensureCapacity(1);
  126. buffer[curLen] = value;
  127. ++curLen;
  128. return *this;
  129. }
  130. StringBuffer & StringBuffer::append(unsigned char value)
  131. {
  132. ensureCapacity(1);
  133. buffer[curLen] = value;
  134. ++curLen;
  135. return *this;
  136. }
  137. StringBuffer & StringBuffer::append(const char * value)
  138. {
  139. if (value)
  140. {
  141. size32_t SourceLen = (size32_t)::strlen(value);
  142. ensureCapacity(SourceLen);
  143. memcpy(buffer + curLen, value, SourceLen);
  144. curLen += SourceLen;
  145. }
  146. return *this;
  147. }
  148. StringBuffer & StringBuffer::append(unsigned len, const char * value)
  149. {
  150. if (len)
  151. {
  152. ensureCapacity(len);
  153. memcpy(buffer + curLen, value, len);
  154. curLen += len;
  155. }
  156. return *this;
  157. }
  158. StringBuffer & StringBuffer::append(const unsigned char * value)
  159. {
  160. return append((const char *) value);
  161. }
  162. StringBuffer & StringBuffer::append(const char * value, int offset, int len)
  163. {
  164. ensureCapacity(len);
  165. memcpy(buffer + curLen, value+offset, len);
  166. curLen += len;
  167. return *this;
  168. }
  169. StringBuffer & StringBuffer::append(const IAtom * value)
  170. {
  171. if (value)
  172. append(value->getAtomNamePtr());
  173. return *this;
  174. }
  175. StringBuffer & StringBuffer::append(double value)
  176. {
  177. int len = length();
  178. int newlen = appendf(DOUBLE_FORMAT, value).length();
  179. while (len < newlen)
  180. {
  181. switch (charAt(len))
  182. {
  183. case '.':
  184. case 'E':
  185. case 'e':
  186. case 'N': // Not a number/infinity
  187. case 'n':
  188. return *this;
  189. }
  190. len++;
  191. }
  192. return append(".0");
  193. }
  194. StringBuffer & StringBuffer::append(float value)
  195. {
  196. int len = length();
  197. int newlen = appendf(FLOAT_FORMAT, value).length();
  198. while (len < newlen)
  199. {
  200. switch (charAt(len))
  201. {
  202. case '.':
  203. case 'E':
  204. case 'e':
  205. case 'N': // Not a number/infinity
  206. case 'n':
  207. return *this;
  208. }
  209. len++;
  210. }
  211. return append(".0");
  212. }
  213. StringBuffer & StringBuffer::append(int value)
  214. {
  215. char temp[12];
  216. unsigned written = numtostr(temp, value);
  217. return append(written, temp);
  218. }
  219. StringBuffer & StringBuffer::append(unsigned value)
  220. {
  221. char temp[12];
  222. unsigned written = numtostr(temp, value);
  223. return append(written, temp);
  224. }
  225. StringBuffer & StringBuffer::appendlong(long value)
  226. {
  227. char temp[24];
  228. unsigned written = numtostr(temp, value);
  229. return append(written, temp);
  230. }
  231. StringBuffer & StringBuffer::appendulong(unsigned long value)
  232. {
  233. char temp[24];
  234. unsigned written = numtostr(temp, value);
  235. return append(written, temp);
  236. }
  237. StringBuffer & StringBuffer::append(__int64 value)
  238. {
  239. char temp[24];
  240. unsigned written = numtostr(temp, value);
  241. return append(written, temp);
  242. }
  243. StringBuffer & StringBuffer::append(unsigned __int64 value)
  244. {
  245. char temp[24];
  246. unsigned written = numtostr(temp, value);
  247. return append(written, temp);
  248. }
  249. StringBuffer & StringBuffer::append(const String & value)
  250. {
  251. size32_t SourceLen = value.length();
  252. ensureCapacity(SourceLen);
  253. value.getChars(0, SourceLen, buffer, curLen);
  254. curLen += SourceLen;
  255. return *this;
  256. }
  257. StringBuffer & StringBuffer::append(const IStringVal & value)
  258. {
  259. return append(value.str());
  260. }
  261. StringBuffer & StringBuffer::append(const IStringVal * value)
  262. {
  263. if (value)
  264. return append(value->str());
  265. else
  266. return *this;
  267. }
  268. StringBuffer & StringBuffer::append(const StringBuffer & value)
  269. {
  270. size32_t SourceLen = value.length();
  271. ensureCapacity(SourceLen);
  272. value.getChars(0, SourceLen, buffer + curLen);
  273. curLen += SourceLen;
  274. return *this;
  275. }
  276. StringBuffer & StringBuffer::appendf(const char *format, ...)
  277. {
  278. va_list args;
  279. va_start(args, format);
  280. valist_appendf(format, args);
  281. va_end(args);
  282. return *this;
  283. }
  284. StringBuffer & StringBuffer::appendLower(unsigned len, const char * value)
  285. {
  286. if (len)
  287. {
  288. ensureCapacity(len);
  289. const byte * from = reinterpret_cast<const byte *>(value);
  290. for (unsigned i = 0; i < len; i++)
  291. buffer[curLen + i] = tolower(from[i]);
  292. curLen += len;
  293. }
  294. return *this;
  295. }
  296. StringBuffer & StringBuffer::setf(const char *format, ...)
  297. {
  298. clear();
  299. va_list args;
  300. va_start(args, format);
  301. valist_appendf(format, args);
  302. va_end(args);
  303. return *this;
  304. }
  305. StringBuffer & StringBuffer::limited_valist_appendf(unsigned szLimit, const char *format, va_list args)
  306. {
  307. #define BUF_SIZE 1024
  308. #define MAX_BUF_SIZE (1024*1024) // limit buffer size to 1MB when doubling
  309. // handle string that is bigger that BUF_SIZE bytes
  310. unsigned size = (0 == szLimit||szLimit>BUF_SIZE)?BUF_SIZE:szLimit;
  311. int len;
  312. va_list args2;
  313. va_copy(args2, args);
  314. try { ensureCapacity(size); }
  315. catch (IException *e)
  316. {
  317. StringBuffer eMsg;
  318. IException *e2 = MakeStringException(-1, "StringBuffer::valist_appendf(\"%s\"): vsnprintf failed or result exceeds limit (%d): %s", format, size, e->errorMessage(eMsg).str());
  319. e->Release();
  320. throw e2;
  321. }
  322. len = _vsnprintf(buffer+curLen,size,format,args);
  323. if (len >= 0)
  324. {
  325. if ((unsigned)len >= size)
  326. {
  327. if (szLimit && (unsigned)len >= szLimit)
  328. {
  329. if ((unsigned)len>szLimit)
  330. {
  331. len = size;
  332. if (len>3) memcpy(buffer+len-3, "...", 3);
  333. }
  334. }
  335. else
  336. {
  337. ensureCapacity(len);
  338. // no need for _vsnprintf since the buffer is already made big enough
  339. vsprintf(buffer+curLen,format,args2);
  340. }
  341. }
  342. }
  343. else if (size == szLimit)
  344. {
  345. len = size;
  346. if (len>3) memcpy(buffer+len-3, "...", 3);
  347. }
  348. else
  349. {
  350. size = BUF_SIZE * 2;
  351. loop
  352. {
  353. if (0 != szLimit && size>szLimit) size = szLimit; // if so, will be last attempt
  354. if (size>MAX_BUF_SIZE)
  355. {
  356. WARNLOG("StringBuffer::valist_appendf(\"%s\"): vsnprintf exceeds limit (%d)", format, size);
  357. size = szLimit = MAX_BUF_SIZE;
  358. }
  359. try { ensureCapacity(size); }
  360. catch (IException *e)
  361. {
  362. StringBuffer eMsg;
  363. IException *e2 = MakeStringException(-1, "StringBuffer::valist_appendf(\"%s\"): vsnprintf failed (%d): %s", format, size, e->errorMessage(eMsg).str());
  364. e->Release();
  365. throw e2;
  366. }
  367. va_list args3;
  368. va_copy(args3, args2);
  369. len = _vsnprintf(buffer+curLen,size,format,args3);
  370. va_end(args3);
  371. if (len>=0) // NB: len>size not possible, 1st _vsnprintf would have handled.
  372. break;
  373. if (size == szLimit)
  374. {
  375. len = size;
  376. if (len>3) memcpy(buffer+len-3, "...", 3);
  377. break;
  378. }
  379. size <<= 1;
  380. }
  381. }
  382. va_end(args2);
  383. curLen += len;
  384. return *this;
  385. }
  386. StringBuffer & StringBuffer::appendN(size32_t count, char fill)
  387. {
  388. ensureCapacity(count);
  389. memset(buffer+curLen, fill, count);
  390. curLen += count;
  391. return *this;
  392. }
  393. void StringBuffer::setLength(unsigned len)
  394. {
  395. if (len > curLen)
  396. {
  397. ensureCapacity(len-curLen);
  398. }
  399. curLen = len;
  400. }
  401. char * StringBuffer::reserve(size32_t size)
  402. {
  403. ensureCapacity(size);
  404. char *ret = buffer+curLen;
  405. curLen += size;
  406. return ret;
  407. }
  408. char * StringBuffer::reserveTruncate(size32_t size)
  409. {
  410. size32_t newMax = curLen+size+1;
  411. if (newMax != maxLen) {
  412. char * newStr = (char *) realloc(buffer, newMax);
  413. if (!newStr)
  414. throw MakeStringException(-1, "StringBuffer::_realloc: Failed to realloc newMax = %d, oldMax = %d", newMax, maxLen);
  415. buffer = newStr;
  416. maxLen = newMax;
  417. }
  418. char *ret = buffer+curLen;
  419. curLen += size;
  420. return ret;
  421. }
  422. void StringBuffer::swapWith(StringBuffer &other)
  423. {
  424. size32_t tmpsz = curLen;
  425. curLen = other.curLen;
  426. other.curLen = tmpsz;
  427. tmpsz = maxLen;
  428. maxLen = other.maxLen;
  429. other.maxLen = tmpsz;
  430. char *tmpbuf = buffer;
  431. buffer = other.buffer;
  432. other.buffer = tmpbuf;
  433. }
  434. void StringBuffer::kill()
  435. {
  436. if (buffer)
  437. free(buffer);
  438. init();
  439. }
  440. void StringBuffer::getChars(int srcBegin, int srcEnd, char * target) const
  441. {
  442. const int len = srcEnd - srcBegin;
  443. if (target && buffer && len > 0)
  444. memcpy(target, buffer + srcBegin, len);
  445. }
  446. void StringBuffer::_insert(unsigned offset, size32_t insertLen)
  447. {
  448. ensureCapacity(insertLen);
  449. memmove(buffer + offset + insertLen, buffer + offset, curLen - offset);
  450. curLen += insertLen;
  451. }
  452. StringBuffer & StringBuffer::insert(int offset, char value)
  453. {
  454. _insert(offset, 1);
  455. buffer[offset] = value;
  456. return *this;
  457. }
  458. StringBuffer & StringBuffer::insert(int offset, const char * value)
  459. {
  460. if (!value) return *this;
  461. unsigned len = (size32_t)strlen(value);
  462. _insert(offset, len);
  463. memcpy(buffer + offset, value, len);
  464. return *this;
  465. }
  466. StringBuffer & StringBuffer::insert(int offset, double value)
  467. {
  468. char temp[36];
  469. sprintf(temp, "%f", value);
  470. insert(offset, temp);
  471. return *this;
  472. }
  473. StringBuffer & StringBuffer::insert(int offset, float value)
  474. {
  475. return insert(offset, (double)value);
  476. }
  477. StringBuffer & StringBuffer::insert(int offset, int value)
  478. {
  479. char temp[12];
  480. numtostr(temp, value);
  481. return insert(offset, temp);
  482. }
  483. StringBuffer & StringBuffer::insert(int offset, unsigned value)
  484. {
  485. char temp[12];
  486. numtostr(temp, value);
  487. return insert(offset, temp);
  488. }
  489. #if 0
  490. StringBuffer & StringBuffer::insert(int offset, long value)
  491. {
  492. char temp[24];
  493. numtostr(temp, value);
  494. return insert(offset, temp);
  495. }
  496. #endif
  497. StringBuffer & StringBuffer::insert(int offset, __int64 value)
  498. {
  499. char temp[24];
  500. numtostr(temp, value);
  501. return insert(offset, temp);
  502. }
  503. StringBuffer & StringBuffer::insert(int offset, const String & value)
  504. {
  505. size32_t len = value.length();
  506. _insert(offset, len);
  507. value.getChars(0, len, buffer, offset);
  508. return *this;
  509. }
  510. StringBuffer & StringBuffer::insert(int offset, const StringBuffer & value)
  511. {
  512. size32_t len = value.length();
  513. _insert(offset, len);
  514. value.getChars(0, len, buffer+offset);
  515. return *this;
  516. }
  517. StringBuffer & StringBuffer::insert(int offset, const IStringVal & value)
  518. {
  519. return insert(offset, value.str());
  520. }
  521. StringBuffer & StringBuffer::insert(int offset, const IStringVal * value)
  522. {
  523. if (value)
  524. return insert(offset, value->str());
  525. else
  526. return *this;
  527. }
  528. StringBuffer & StringBuffer::newline()
  529. {
  530. return append("\n");
  531. }
  532. StringBuffer & StringBuffer::pad(unsigned count)
  533. {
  534. ensureCapacity(count);
  535. memset(buffer + curLen, ' ', count);
  536. curLen += count;
  537. return *this;
  538. }
  539. StringBuffer & StringBuffer::padTo(unsigned count)
  540. {
  541. if (curLen<count)
  542. pad(count-curLen);
  543. return *this;
  544. }
  545. StringBuffer & StringBuffer::clip()
  546. {
  547. while (curLen && isspace(buffer[curLen-1]))
  548. curLen--;
  549. return *this;
  550. }
  551. StringBuffer & StringBuffer::trim()
  552. {
  553. return clip().trimLeft();
  554. }
  555. StringBuffer & StringBuffer::trimLeft()
  556. {
  557. char *p;
  558. if (curLen==0)
  559. return *this;
  560. buffer[curLen] = 0;
  561. for(p = buffer;isspace(*p);p++)
  562. ;
  563. if (p!=buffer)
  564. {
  565. curLen -= p-buffer;
  566. memmove(buffer,p,curLen);
  567. }
  568. return *this;
  569. }
  570. StringBuffer & StringBuffer::remove(unsigned start, unsigned len)
  571. {
  572. if (start > curLen) start = curLen;
  573. if (start + len > curLen) len = curLen - start;
  574. unsigned start2 = start + len;
  575. memmove(buffer + start, buffer + start2, curLen - start2);
  576. setLength(curLen - len);
  577. return *this;
  578. }
  579. StringBuffer &StringBuffer::reverse()
  580. {
  581. unsigned max = curLen/2;
  582. char * end = buffer + curLen;
  583. unsigned idx;
  584. for (idx = 0; idx < max; idx++)
  585. {
  586. char temp = buffer[idx];
  587. end--;
  588. buffer[idx] = *end;
  589. *end = temp;
  590. }
  591. return *this;
  592. }
  593. MemoryBuffer & StringBuffer::deserialize(MemoryBuffer & in)
  594. {
  595. unsigned len;
  596. in.read(len);
  597. append(len, (const char *)in.readDirect(len));
  598. return in;
  599. }
  600. MemoryBuffer & StringBuffer::serialize(MemoryBuffer & out) const
  601. {
  602. return out.append(curLen).append(curLen, buffer);
  603. }
  604. StringBuffer &StringBuffer::loadFile(const char *filename, bool binaryMode)
  605. {
  606. FILE *in = fopen(filename, binaryMode?"rb":"rt");
  607. if (in)
  608. {
  609. char buffer[1024];
  610. int bytes;
  611. for (;;)
  612. {
  613. bytes = (size32_t)fread(buffer, 1, sizeof(buffer), in);
  614. if (!bytes)
  615. break;
  616. append(buffer, 0, bytes);
  617. }
  618. fclose(in);
  619. return *this;
  620. }
  621. else
  622. throw MakeStringException(errno, "File %s could not be opened", filename);
  623. }
  624. StringBuffer & StringBuffer::loadFile(IFile* f)
  625. {
  626. if(!f)
  627. return *this;
  628. Owned<IFileIO> io = f->open(IFOread);
  629. if(!io)
  630. throw MakeStringException(errno, "file %s could not be opened for reading", f->queryFilename());
  631. char buf[2048];
  632. const unsigned requestedSize = sizeof(buf);
  633. offset_t pos = 0;
  634. loop
  635. {
  636. size32_t len = io->read(pos, requestedSize, buf);
  637. if (len == 0)
  638. break;
  639. append(len, buf);
  640. pos += len;
  641. if (len != requestedSize)
  642. break;
  643. }
  644. return *this;
  645. }
  646. void StringBuffer::setCharAt(unsigned offset, char value)
  647. {
  648. if (offset < curLen)
  649. buffer[offset] = value;
  650. }
  651. StringBuffer & StringBuffer::toLowerCase()
  652. {
  653. if (buffer)
  654. {
  655. int l = curLen;
  656. for (int i = 0; i < l; i++)
  657. if (isupper(buffer[i]))
  658. buffer[i] = tolower(buffer[i]);
  659. }
  660. return *this;
  661. }
  662. StringBuffer & StringBuffer::toUpperCase()
  663. {
  664. if (buffer)
  665. {
  666. int l = curLen;
  667. for (int i = 0; i < l; i++)
  668. if (islower(buffer[i]))
  669. buffer[i] = toupper(buffer[i]);
  670. }
  671. return *this;
  672. }
  673. StringBuffer & StringBuffer::replace(char oldChar, char newChar)
  674. {
  675. if (buffer)
  676. {
  677. int l = curLen;
  678. for (int i = 0; i < l; i++)
  679. if (buffer[i] == oldChar)
  680. {
  681. buffer[i] = newChar;
  682. if (newChar == '\0')
  683. {
  684. curLen = i;
  685. break;
  686. }
  687. }
  688. }
  689. return *this;
  690. }
  691. // this method will replace all occurrances of "oldStr" with "newStr"
  692. StringBuffer & StringBuffer::replaceString(const char* oldStr, const char* newStr)
  693. {
  694. if (buffer)
  695. {
  696. const char* s = str(); // get null terminated version of the string
  697. int left = length();
  698. int oldStr_len = (size32_t)strlen(oldStr);
  699. StringBuffer tempbuff;
  700. while (left >= oldStr_len)
  701. {
  702. if ( memcmp(s, oldStr, oldStr_len) == 0)
  703. {
  704. tempbuff.append(newStr);
  705. s += oldStr_len;
  706. left -= oldStr_len;
  707. }
  708. else
  709. {
  710. tempbuff.append(*s);
  711. s++;
  712. left--;
  713. }
  714. }
  715. // there are no more possible replacements, make sure we keep the end of the original buffer
  716. tempbuff.append(s);
  717. //*this = tempbuff;
  718. swapWith(tempbuff);
  719. }
  720. return *this;
  721. }
  722. StringBuffer & StringBuffer::stripChar(char oldChar)
  723. {
  724. if (buffer)
  725. {
  726. size32_t delta = 0;
  727. size32_t l = curLen;
  728. for (size32_t i = 0; i < l; i++)
  729. {
  730. if (buffer[i] == oldChar)
  731. delta++;
  732. else if (delta)
  733. buffer[i-delta] = buffer[i];
  734. }
  735. curLen = curLen - delta;
  736. }
  737. return *this;
  738. }
  739. const char * StringBuffer::toCharArray() const
  740. {
  741. if (buffer)
  742. {
  743. buffer[curLen] = '\0'; // There is always room for this null
  744. return buffer;
  745. }
  746. return TheNullStr;
  747. }
  748. //===========================================================================
  749. VStringBuffer::VStringBuffer(const char* format, ...)
  750. {
  751. va_list args;
  752. va_start(args,format);
  753. valist_appendf(format,args);
  754. va_end(args);
  755. }
  756. //===========================================================================
  757. String::String()
  758. {
  759. text = (char *)TheNullStr;
  760. }
  761. String::String(const char * value)
  762. {
  763. text = (value ? strdup(value) : (char *)TheNullStr);
  764. }
  765. String::String(const char * value, int offset, int _count)
  766. {
  767. text = (char *)malloc(_count+1);
  768. memcpy(text, value+offset, _count);
  769. text[_count]=0;
  770. }
  771. String::String(String & value)
  772. {
  773. text = strdup(value.toCharArray());
  774. }
  775. String::String(StringBuffer & value)
  776. {
  777. unsigned len = value.length();
  778. text = (char *)malloc(len+1);
  779. value.getChars(0,len,text);
  780. text[len] = 0;
  781. }
  782. String::~String()
  783. {
  784. if (text != TheNullStr) free(text);
  785. }
  786. char String::charAt(size32_t index) const
  787. {
  788. return text[index];
  789. }
  790. int String::compareTo(const String & value) const
  791. {
  792. return strcmp(text, value.toCharArray());
  793. }
  794. int String::compareTo(const char* value) const
  795. {
  796. return strcmp(text,value);
  797. }
  798. String * String::concat(const String & value) const
  799. {
  800. StringBuffer temp(toCharArray());
  801. temp.append(value);
  802. return new String(temp.str());
  803. }
  804. bool String::endsWith(const String & value) const
  805. {
  806. unsigned lenValue = value.length();
  807. unsigned len = (size32_t)strlen(text);
  808. if (len >= lenValue)
  809. return (memcmp(text+(len-lenValue),value.toCharArray(),lenValue) == 0);
  810. return false;
  811. }
  812. bool String::endsWith(const char* value) const
  813. {
  814. return ::endsWith(this->text, value);
  815. }
  816. bool String::equals(String & value) const
  817. {
  818. return strcmp(text, value.toCharArray())==0;
  819. }
  820. bool String::equalsIgnoreCase(const String & value) const
  821. {
  822. return stricmp(text, value.toCharArray())==0;
  823. }
  824. void String::getBytes(int srcBegin, int srcEnd, void * dest, int dstBegin) const
  825. {
  826. memcpy((char *)dest+dstBegin, text+srcBegin, srcEnd-srcBegin);
  827. }
  828. void String::getChars(int srcBegin, int srcEnd, void * dest, int dstBegin) const
  829. {
  830. memcpy((char *)dest+dstBegin, text+srcBegin, srcEnd-srcBegin);
  831. }
  832. int String::hashCode() const
  833. {
  834. return (int)hashc((const byte *)text,length(),0);
  835. }
  836. int String::indexOf(int ch) const
  837. {
  838. char * match = strchr(text, ch);
  839. return match ? (int)(match - text) : -1;
  840. }
  841. int String::indexOf(int ch, int from) const
  842. {
  843. char * match = strchr(text + from, ch);
  844. return match ? (int)(match - text) : -1;
  845. }
  846. int String::indexOf(const String & search) const
  847. {
  848. const char * str = search.toCharArray();
  849. const char * match = strstr(text, str);
  850. return match ? (int)(match - text) : -1;
  851. }
  852. int String::indexOf(const String & search, int from) const
  853. {
  854. const char * str = search.toCharArray();
  855. const char * match = strstr(text + from, str);
  856. return match ? (int)(match - text) : -1;
  857. }
  858. int String::lastIndexOf(int ch) const
  859. {
  860. char * match = strrchr(text, ch);
  861. return match ? (int)(match - text) : -1;
  862. }
  863. int String::lastIndexOf(int ch, int from) const
  864. {
  865. for (;(from > 0);--from)
  866. if (text[from] == ch)
  867. return from;
  868. return -1;
  869. }
  870. int String::lastIndexOf(const String & search) const
  871. {
  872. assertex(!"TBD");
  873. return -1;
  874. }
  875. int String::lastIndexOf(const String & search, int from) const
  876. {
  877. assertex(!"TBD");
  878. return -1;
  879. }
  880. size32_t String::length() const
  881. {
  882. return (size32_t)strlen(text);
  883. }
  884. bool String::startsWith(String & value) const
  885. {
  886. unsigned lenValue = value.length();
  887. const char * search = value.toCharArray();
  888. return (memcmp(text, search, lenValue) == 0);
  889. }
  890. bool String::startsWith(String & value, int offset) const
  891. {
  892. unsigned lenValue = value.length();
  893. const char * search = value.toCharArray();
  894. return (memcmp(text + offset, search, lenValue) == 0);
  895. }
  896. bool String::startsWith(const char* value) const
  897. {
  898. return ::startsWith(this->text,value);
  899. }
  900. String * String::substring(int beginIndex) const
  901. {
  902. return new String(text+beginIndex);
  903. }
  904. String * String::substring(int beginIndex, int endIndex) const
  905. {
  906. return new String(text, beginIndex, endIndex - beginIndex);
  907. }
  908. const char *String::toCharArray() const
  909. {
  910. return text;
  911. }
  912. String * String::toLowerCase() const
  913. {
  914. String *ret = new String();
  915. size32_t l = length();
  916. if (l)
  917. {
  918. ret->text = (char *)malloc(l+1);
  919. for (unsigned i = 0; i < l; i++)
  920. ret->text[i] = tolower(text[i]);
  921. ret->text[l]=0;
  922. }
  923. return ret;
  924. }
  925. String * String::toString()
  926. {
  927. Link();
  928. return this;
  929. }
  930. String * String::toUpperCase() const
  931. {
  932. String *ret = new String();
  933. size32_t l = length();
  934. if (l)
  935. {
  936. ret->text = (char *)malloc(l+1);
  937. for (unsigned i = 0; i < l; i++)
  938. ret->text[i] = toupper(text[i]);
  939. ret->text[l]=0;
  940. }
  941. return ret;
  942. }
  943. String * String::trim() const
  944. {
  945. size32_t l = length();
  946. while (l && isspace(text[l-1]))
  947. l--;
  948. return new String(text, 0, l);
  949. }
  950. //------------------------------------------------
  951. #if 0
  952. String & String::valueOf(char value)
  953. {
  954. return * new String(&value, 0, 1);
  955. }
  956. String & String::valueOf(const char * value)
  957. {
  958. return * new String(value);
  959. }
  960. String & String::valueOf(const char * value, int offset, int count)
  961. {
  962. return * new String(value, offset, count);
  963. }
  964. String & String::valueOf(double value)
  965. {
  966. StringBuffer temp;
  967. return temp.append(value).toString();
  968. }
  969. String & String::valueOf(float value)
  970. {
  971. StringBuffer temp;
  972. return temp.append(value).toString();
  973. }
  974. String & String::valueOf(int value)
  975. {
  976. StringBuffer temp;
  977. return temp.append(value).toString();
  978. }
  979. String & String::valueOf(long value)
  980. {
  981. StringBuffer temp;
  982. return temp.append(value).toString();
  983. }
  984. #endif
  985. //------------------------------------------------
  986. StringAttr::StringAttr(const char * _text)
  987. {
  988. text = _text ? strdup(_text) : NULL;
  989. }
  990. StringAttr::StringAttr(const char * _text, unsigned _len)
  991. {
  992. text = NULL;
  993. set(_text, _len);
  994. }
  995. StringAttr::StringAttr(const StringAttr & src)
  996. {
  997. text = NULL;
  998. set(src.get());
  999. }
  1000. void StringAttr::set(const char * _text)
  1001. {
  1002. free(text);
  1003. text = _text ? strdup(_text) : NULL;
  1004. }
  1005. void StringAttr::set(const char * _text, unsigned _len)
  1006. {
  1007. if (text)
  1008. free(text);
  1009. text = (char *)malloc(_len+1);
  1010. memcpy(text, _text, _len);
  1011. text[_len] = 0;
  1012. }
  1013. void StringAttr::setown(const char * _text)
  1014. {
  1015. if (text)
  1016. free(text);
  1017. text = (char *)_text;
  1018. }
  1019. void StringAttr::toLowerCase()
  1020. {
  1021. if (text)
  1022. {
  1023. char * cur = text;
  1024. char next;
  1025. while ((next = *cur) != 0)
  1026. {
  1027. if (isupper(next))
  1028. *cur = tolower(next);
  1029. cur++;
  1030. }
  1031. }
  1032. }
  1033. void StringAttr::toUpperCase()
  1034. {
  1035. if (text)
  1036. {
  1037. char * cur = text;
  1038. char next;
  1039. while ((next = *cur) != 0)
  1040. {
  1041. if (islower(next))
  1042. *cur = toupper(next);
  1043. cur++;
  1044. }
  1045. }
  1046. }
  1047. StringAttrItem::StringAttrItem(const char *_text, unsigned _len)
  1048. {
  1049. text.set(_text, _len);
  1050. }
  1051. inline char hex(char c, char lower)
  1052. {
  1053. if (c < 10)
  1054. return '0' + c;
  1055. else if (lower)
  1056. return 'a' + c - 10;
  1057. else
  1058. return 'A' + c - 10;
  1059. }
  1060. StringBuffer & StringBuffer::appendhex(unsigned char c, char lower)
  1061. {
  1062. append(hex(c>>4, lower));
  1063. append(hex(c&0xF, lower));
  1064. return *this;
  1065. }
  1066. void appendURL(StringBuffer *dest, const char *src, size32_t len, char lower)
  1067. {
  1068. if (len == (size32_t)-1)
  1069. len = (size32_t)strlen(src);
  1070. while (len)
  1071. {
  1072. // isalnum seems to give weird results for chars > 127....
  1073. unsigned char c = (unsigned char) *src;
  1074. if (c == ' ')
  1075. dest->append('+');
  1076. else if ((c & 0x80) || !isalnum(*src))
  1077. {
  1078. dest->append('%');
  1079. dest->appendhex(c, lower);
  1080. }
  1081. else
  1082. dest->append(c);
  1083. src++;
  1084. len--;
  1085. }
  1086. }
  1087. inline char translateHex(char hex)
  1088. {
  1089. if(hex >= 'A')
  1090. return (hex & 0xdf) - 'A' + 10;
  1091. else
  1092. return hex - '0';
  1093. }
  1094. inline char translateHex(char h1, char h2)
  1095. {
  1096. return (translateHex(h1) * 16 + translateHex(h2));
  1097. }
  1098. StringBuffer &appendDecodedURL(StringBuffer &s, const char *url)
  1099. {
  1100. if(!url)
  1101. return s;
  1102. while (*url)
  1103. {
  1104. char c = *url++;
  1105. if (c == '+')
  1106. c = ' ';
  1107. else if (c == '%')
  1108. {
  1109. if (isxdigit(url[0]) && isxdigit(url[1]))
  1110. {
  1111. c = translateHex(url[0], url[1]);
  1112. url+=2;
  1113. }
  1114. }
  1115. s.append(c);
  1116. }
  1117. return s;
  1118. }
  1119. static StringBuffer & appendStringExpandControl(StringBuffer &out, unsigned len, const char * src, bool addBreak, bool isCpp, bool isUtf8)
  1120. {
  1121. const int minBreakPos = 0;
  1122. const int commaBreakPos = 70;
  1123. const int maxBreakPos = 120;
  1124. const char * startLine = src;
  1125. out.ensureCapacity(len+2);
  1126. for (; len > 0; --len)
  1127. {
  1128. unsigned char c = *src++;
  1129. bool insertBreak = false;
  1130. bool allowBreak = true;
  1131. switch (c)
  1132. {
  1133. case '\n':
  1134. {
  1135. out.append("\\n");
  1136. if (src-startLine > minBreakPos)
  1137. insertBreak = true;
  1138. break;
  1139. }
  1140. case ',':
  1141. {
  1142. out.append(c);
  1143. if (src-startLine > commaBreakPos)
  1144. insertBreak = true;
  1145. break;
  1146. }
  1147. case '\r': out.append("\\r"); break;
  1148. case '\t': out.append("\\t"); break;
  1149. case '"':
  1150. if (isCpp)
  1151. out.append("\\");
  1152. out.append(c);
  1153. break;
  1154. case '\'':
  1155. if (!isCpp)
  1156. out.append("\\");
  1157. out.append(c);
  1158. break;
  1159. case '\\': out.append("\\\\"); break;
  1160. case '?':
  1161. if (isCpp)
  1162. {
  1163. //stop trigraphs being generated.... quote the second ?
  1164. out.append(c);
  1165. if ((len!=1) && (*src == '?'))
  1166. {
  1167. out.append('\\');
  1168. allowBreak = false;
  1169. }
  1170. }
  1171. else
  1172. out.append(c);
  1173. break;
  1174. default:
  1175. if (isUtf8 || ((c >= ' ') && (c <= 126)))
  1176. out.append(c);
  1177. else
  1178. out.appendf("\\%03o", c);
  1179. break;
  1180. }
  1181. if (addBreak && (insertBreak || (allowBreak && src-startLine >= maxBreakPos)))
  1182. {
  1183. out.append("\"").newline().append("\t\t\"");
  1184. startLine = src;
  1185. }
  1186. }
  1187. return out;
  1188. }
  1189. StringBuffer & appendStringAsCPP(StringBuffer &out, unsigned len, const char * src, bool addBreak)
  1190. {
  1191. return appendStringExpandControl(out, len, src, addBreak, true, false);
  1192. }
  1193. StringBuffer & appendStringAsECL(StringBuffer &out, unsigned len, const char * src)
  1194. {
  1195. return appendStringExpandControl(out, len, src, false, false, false);
  1196. }
  1197. StringBuffer & appendUtf8AsECL(StringBuffer &out, unsigned len, const char * src)
  1198. {
  1199. return appendStringExpandControl(out, len, src, false, false, true);
  1200. }
  1201. StringBuffer & appendStringAsQuotedCPP(StringBuffer &out, unsigned len, const char * src, bool addBreak)
  1202. {
  1203. out.ensureCapacity(len+2);
  1204. out.append('\"');
  1205. appendStringAsCPP(out, len, src, addBreak);
  1206. return out.append('\"');
  1207. }
  1208. StringBuffer & appendStringAsQuotedECL(StringBuffer &out, unsigned len, const char * src)
  1209. {
  1210. out.ensureCapacity(len+2);
  1211. out.append('\'');
  1212. appendStringAsECL(out, len, src);
  1213. return out.append('\'');
  1214. }
  1215. void extractItem(StringBuffer & res, const char * src, const char * sep, int whichItem, bool caps)
  1216. {
  1217. bool isSeparator[256];
  1218. memset(isSeparator,0,sizeof(isSeparator));
  1219. unsigned char * finger = (unsigned char *)sep;
  1220. while (*finger !=0)
  1221. isSeparator[*finger++] = true;
  1222. isSeparator[0]=true;
  1223. finger = (unsigned char *)src;
  1224. unsigned char next;
  1225. loop
  1226. {
  1227. while (isSeparator[(next = *finger)])
  1228. {
  1229. if (next == 0) return;
  1230. finger++;
  1231. }
  1232. if (whichItem == 0)
  1233. {
  1234. while (!isSeparator[(next = *finger)])
  1235. {
  1236. if (caps)
  1237. next = toupper(next);
  1238. res.append(next);
  1239. finger++;
  1240. }
  1241. return;
  1242. }
  1243. while (!isSeparator[*finger])
  1244. finger++;
  1245. whichItem--;
  1246. }
  1247. }
  1248. int utf8CharLen(unsigned char ch)
  1249. {
  1250. //return 1 if this is an ascii character,
  1251. //or 0 if its not a valid utf-8 character
  1252. if (ch < 128)
  1253. return 1;
  1254. if (ch < 192)
  1255. return 0;
  1256. unsigned char len = 1;
  1257. for (unsigned char lead = ch << 1; (lead & 0x80); lead <<=1)
  1258. len++;
  1259. return len;
  1260. }
  1261. int utf8CharLen(const unsigned char *ch)
  1262. {
  1263. //return 1 if this is an ascii character,
  1264. //or 0 if its not a valid utf-8 character
  1265. if (*ch < 128)
  1266. return 1;
  1267. unsigned char len = utf8CharLen(*ch);
  1268. for (unsigned pos = 1; pos < len; pos++)
  1269. if ((ch[pos] < 128) || (ch[pos] >= 192))
  1270. return 0; //its not a valid utf-8 character after all
  1271. return len;
  1272. }
  1273. const char *encodeXML(const char *x, StringBuffer &ret, unsigned flags, unsigned len, bool utf8)
  1274. {
  1275. while (len)
  1276. {
  1277. switch(*x)
  1278. {
  1279. case '&':
  1280. ret.append("&amp;");
  1281. break;
  1282. case '<':
  1283. ret.append("&lt;");
  1284. break;
  1285. case '>':
  1286. ret.append("&gt;");
  1287. break;
  1288. case '\"':
  1289. ret.append("&quot;");
  1290. break;
  1291. case '\'':
  1292. ret.append("&apos;");
  1293. break;
  1294. case ' ':
  1295. ret.append(flags & ENCODE_SPACES?"&#32;":" ");
  1296. break;
  1297. case '\n':
  1298. ret.append(flags & ENCODE_NEWLINES?"&#10;":"\n");
  1299. break;
  1300. case '\r':
  1301. ret.append(flags & ENCODE_NEWLINES?"&#13;":"\r");
  1302. break;
  1303. case '\t':
  1304. ret.append(flags & ENCODE_SPACES?"&#9;":"\t");
  1305. break;
  1306. case '\0':
  1307. if (len == (unsigned) -1)
  1308. return x;
  1309. ret.append("&#xe000;"); // hack!!! Characters below 0x20 are not legal in strict xml, even encoded.
  1310. break;
  1311. default:
  1312. if (*x >= ' ' && ((byte)*x) < 128)
  1313. ret.append(*x);
  1314. else if (*x < ' ' && *x > 0)
  1315. ret.append("&#xe0").appendhex(*x, true).append(';'); // HACK
  1316. else if (utf8)
  1317. {
  1318. unsigned chlen = utf8CharLen((const unsigned char *)x);
  1319. if (chlen==0)
  1320. ret.append("&#").append((unsigned int)*(unsigned char *) x).append(';');
  1321. else
  1322. {
  1323. ret.append(*x);
  1324. while(--chlen)
  1325. {
  1326. if (len != (unsigned) -1)
  1327. len--;
  1328. ret.append(*(++x));
  1329. }
  1330. }
  1331. }
  1332. else
  1333. ret.append("&#").append((unsigned int)*(unsigned char *) x).append(';');
  1334. break;
  1335. }
  1336. if (len != (unsigned) -1)
  1337. len--;
  1338. ++x;
  1339. }
  1340. return x;
  1341. }
  1342. const char *encodeXML(const char *x, IIOStream &out, unsigned flags, unsigned len, bool utf8)
  1343. {
  1344. while (len)
  1345. {
  1346. switch(*x)
  1347. {
  1348. case '&':
  1349. writeStringToStream(out, "&amp;");
  1350. break;
  1351. case '<':
  1352. writeStringToStream(out, "&lt;");
  1353. break;
  1354. case '>':
  1355. writeStringToStream(out, "&gt;");
  1356. break;
  1357. case '\"':
  1358. writeStringToStream(out, "&quot;");
  1359. break;
  1360. case '\'':
  1361. writeStringToStream(out, "&apos;");
  1362. break;
  1363. case ' ':
  1364. writeStringToStream(out, flags & ENCODE_SPACES?"&#32;":" ");
  1365. break;
  1366. case '\n':
  1367. writeStringToStream(out, flags & ENCODE_NEWLINES?"&#10;":"\n");
  1368. break;
  1369. case '\r':
  1370. writeStringToStream(out, flags & ENCODE_NEWLINES?"&#13;":"\r");
  1371. break;
  1372. case '\t':
  1373. writeStringToStream(out, flags & ENCODE_SPACES?"&#9;":"\t");
  1374. break;
  1375. case '\0':
  1376. if (len == (unsigned) -1)
  1377. return x;
  1378. writeStringToStream(out, "&#xe000;"); // hack!!! Characters below 0x20 are not legal in strict xml, even encoded.
  1379. break;
  1380. default:
  1381. if (*x >= ' ' && ((byte)*x) < 128)
  1382. writeCharToStream(out, *x);
  1383. else if (*x < ' ' && *x > 0)
  1384. {
  1385. writeStringToStream(out, "&#xe0");
  1386. unsigned char c = *(unsigned char *)x;
  1387. writeCharToStream(out, hex(c>>4, true));
  1388. writeCharToStream(out, hex(c&0xF, true));
  1389. writeCharToStream(out, ';'); // HACK
  1390. }
  1391. else if (utf8)
  1392. {
  1393. int chlen = utf8CharLen((const unsigned char *)x);
  1394. if (chlen==0)
  1395. {
  1396. writeStringToStream(out, "&#");
  1397. char tmp[12];
  1398. unsigned written = numtostr(tmp, *(unsigned char *)x);
  1399. out.write(written, tmp);
  1400. writeCharToStream(out, ';');
  1401. }
  1402. else
  1403. {
  1404. writeCharToStream(out, *x);
  1405. while(--chlen)
  1406. {
  1407. if (len != (unsigned) -1)
  1408. len--;
  1409. writeCharToStream(out, *(++x));
  1410. }
  1411. }
  1412. }
  1413. else
  1414. {
  1415. writeStringToStream(out, "&#");
  1416. char tmp[12];
  1417. unsigned written = numtostr(tmp, *(unsigned char *)x);
  1418. out.write(written, tmp);
  1419. writeCharToStream(out, ';');
  1420. }
  1421. break;
  1422. }
  1423. if (len != (unsigned) -1)
  1424. len--;
  1425. ++x;
  1426. }
  1427. return x;
  1428. }
  1429. static void writeUtf8(unsigned c, StringBuffer &out)
  1430. {
  1431. if (c < 0x80)
  1432. out.append((char)c);
  1433. else if (c < 0x800)
  1434. {
  1435. out.append((char)(0xC0 | (c>>6)));
  1436. out.append((char)(0x80 | (c & 0x3F)));
  1437. }
  1438. else if (c < 0x10000)
  1439. {
  1440. out.append((char) (0xE0 | (c>>12)));
  1441. out.append((char) (0x80 | (c>>6 & 0x3F)));
  1442. out.append((char) (0x80 | (c & 0x3F)));
  1443. }
  1444. else if (c < 0x200000)
  1445. {
  1446. out.append((char) (0xF0 | (c>>18)));
  1447. out.append((char) (0x80 | (c>>12 & 0x3F)));
  1448. out.append((char) (0x80 | (c>>6 & 0x3F)));
  1449. out.append((char) (0x80 | (c & 0x3F)));
  1450. }
  1451. else if (c < 0x4000000)
  1452. {
  1453. out.append((char) (0xF8 | (c>>24)));
  1454. out.append((char) (0x80 | (c>>18 & 0x3F)));
  1455. out.append((char) (0x80 | (c>>12 & 0x3F)));
  1456. out.append((char) (0x80 | (c>>6 & 0x3F)));
  1457. out.append((char) (0x80 | (c & 0x3F)));
  1458. }
  1459. else if (c < 0x80000000)
  1460. {
  1461. out.append((char) (0xFC | (c>>30)));
  1462. out.append((char) (0x80 | (c>>24 & 0x3F)));
  1463. out.append((char) (0x80 | (c>>18 & 0x3F)));
  1464. out.append((char) (0x80 | (c>>12 & 0x3F)));
  1465. out.append((char) (0x80 | (c>>6 & 0x3F)));
  1466. out.append((char) (0x80 | (c & 0x3F)));
  1467. }
  1468. else
  1469. assertex(false);
  1470. }
  1471. #define JSONSTRICT
  1472. const char *decodeJSON(const char *j, StringBuffer &ret, unsigned len, const char **errMark)
  1473. {
  1474. if (!j)
  1475. return j;
  1476. if ((unsigned)-1 == len)
  1477. len = (unsigned)strlen(j);
  1478. try
  1479. {
  1480. for (const char *end = j+len; j<end && *j; j++)
  1481. {
  1482. if (*j!='\\')
  1483. ret.append(*j);
  1484. else
  1485. {
  1486. switch (*++j)
  1487. {
  1488. case 'u':
  1489. {
  1490. j++;
  1491. if (end-j>=4)
  1492. {
  1493. char *endptr;
  1494. StringAttr s(j, 4);
  1495. unsigned val = strtoul(s.get(), &endptr, 16);
  1496. if (endptr && !*endptr)
  1497. {
  1498. writeUtf8(val, ret);
  1499. j+=3;
  1500. break;
  1501. }
  1502. }
  1503. #ifdef JSONSTRICT
  1504. throw MakeStringException(-1, "invalid json \\u escaped sequence");
  1505. #endif
  1506. ret.append(*j);
  1507. break;
  1508. }
  1509. case '\"':
  1510. case '\\':
  1511. case '/':
  1512. ret.append(*j);
  1513. break;
  1514. case 'b':
  1515. ret.append('\b');
  1516. break;
  1517. case 'f':
  1518. ret.append('\f');
  1519. break;
  1520. case 'n':
  1521. ret.append('\n');
  1522. continue;
  1523. case 'r':
  1524. ret.append('\r');
  1525. break;
  1526. case 't':
  1527. ret.append('\t');
  1528. break;
  1529. default:
  1530. {
  1531. #ifdef JSONSTRICT
  1532. throw MakeStringException(-1, "invalid json escaped sequence");
  1533. #endif
  1534. ret.append('\\');
  1535. ret.append(*j);
  1536. break;
  1537. }
  1538. }
  1539. }
  1540. }
  1541. }
  1542. catch (IException *)
  1543. {
  1544. if (errMark) *errMark = j;
  1545. throw;
  1546. }
  1547. return j;
  1548. }
  1549. void decodeXML(ISimpleReadStream &in, StringBuffer &out, unsigned len)
  1550. {
  1551. // TODO
  1552. UNIMPLEMENTED;
  1553. }
  1554. const char *decodeXML(const char *x, StringBuffer &ret, const char **errMark, IEntityHelper *entityHelper, bool strict)
  1555. {
  1556. if (!x)
  1557. return x;
  1558. try
  1559. {
  1560. while (*x)
  1561. {
  1562. if ('&' == *x)
  1563. {
  1564. switch (x[1])
  1565. {
  1566. case 'a':
  1567. switch (x[2])
  1568. {
  1569. case 'm':
  1570. {
  1571. if ('p' == x[3] && ';' == x[4])
  1572. {
  1573. x += 5;
  1574. ret.append('&');
  1575. continue;
  1576. }
  1577. break;
  1578. }
  1579. case 'p':
  1580. {
  1581. if ('o' == x[3] && 's' == x[4] && ';' == x[5])
  1582. {
  1583. x += 6;
  1584. ret.append('\'');
  1585. continue;
  1586. }
  1587. break;
  1588. }
  1589. }
  1590. break;
  1591. case 'l':
  1592. if ('t' == x[2] && ';' == x[3])
  1593. {
  1594. x += 4;
  1595. ret.append('<');
  1596. continue;
  1597. }
  1598. break;
  1599. case 'g':
  1600. if ('t' == x[2] && ';' == x[3])
  1601. {
  1602. x += 4;
  1603. ret.append('>');
  1604. continue;
  1605. }
  1606. break;
  1607. case 'q':
  1608. if ('u' == x[2] && 'o' == x[3] && 't' == x[4] && ';' == x[5])
  1609. {
  1610. x += 6;
  1611. ret.append('"');
  1612. continue;
  1613. }
  1614. break;
  1615. case 'n':
  1616. if ('b' == x[2] && 's' == x[3] && 'p' == x[4] && ';' == x[5])
  1617. {
  1618. x += 6;
  1619. writeUtf8(0xa0, ret);
  1620. continue;
  1621. }
  1622. break;
  1623. case '#':
  1624. {
  1625. const char *numstart = x+2;
  1626. int base = 10;
  1627. if (*numstart == 'x')
  1628. {
  1629. base = 16;
  1630. numstart++;
  1631. }
  1632. char *numend;
  1633. unsigned val = strtoul(numstart, &numend, base);
  1634. if (numstart==numend || *numend != ';')
  1635. {
  1636. if (strict)
  1637. throw MakeStringException(-1, "invalid escaped sequence");
  1638. }
  1639. else // always convert to utf-8. Should potentially throw error if not marked as utf-8 encoded doc and out of ascii range.
  1640. {
  1641. writeUtf8(val, ret);
  1642. x = numend+1;
  1643. continue;
  1644. }
  1645. break;
  1646. }
  1647. case ';':
  1648. case '\0':
  1649. if (strict)
  1650. throw MakeStringException(-1, "invalid escaped sequence");
  1651. break;
  1652. default:
  1653. if (entityHelper)
  1654. {
  1655. bool error = false;
  1656. const char *start=x+1;
  1657. const char *finger=start;
  1658. while (*finger && *finger != ';')
  1659. ++finger;
  1660. if (*finger == ';')
  1661. {
  1662. StringBuffer entity(finger-start, start);
  1663. if (entityHelper->find(entity, ret))
  1664. {
  1665. x = finger + 1;
  1666. continue;
  1667. }
  1668. }
  1669. }
  1670. if (strict)
  1671. throw MakeStringException(-1, "invalid escaped sequence");
  1672. break;
  1673. }
  1674. }
  1675. ret.append(*x);
  1676. ++x;
  1677. }
  1678. }
  1679. catch (IException *)
  1680. {
  1681. if (errMark) *errMark = x;
  1682. throw;
  1683. }
  1684. return x;
  1685. }
  1686. StringBuffer & appendXMLOpenTag(StringBuffer &xml, const char *tag, const char *prefix, bool complete, bool close, const char *uri)
  1687. {
  1688. if (!tag || !*tag)
  1689. return xml;
  1690. xml.append('<');
  1691. appendXMLTagName(xml, tag, prefix);
  1692. if (uri && *uri)
  1693. {
  1694. xml.append(" xmlns");
  1695. if (prefix && *prefix)
  1696. xml.append(':').append(prefix);
  1697. xml.append("=\"").append(uri).append('\"');
  1698. }
  1699. if (complete)
  1700. {
  1701. if (close)
  1702. xml.append('/');
  1703. xml.append('>');
  1704. }
  1705. return xml;
  1706. }
  1707. jlib_decl StringBuffer &appendJSONName(StringBuffer &s, const char *name)
  1708. {
  1709. if (!name || !*name)
  1710. return s;
  1711. delimitJSON(s);
  1712. return encodeJSON(s.append('"'), name).append("\": ");
  1713. }
  1714. jlib_decl StringBuffer &appendfJSONName(StringBuffer &s, const char *format, ...)
  1715. {
  1716. va_list args;
  1717. va_start(args, format);
  1718. StringBuffer vs;
  1719. vs.valist_appendf(format, args);
  1720. va_end(args);
  1721. return appendJSONName(s, vs);
  1722. }
  1723. static char hexchar[] = "0123456789ABCDEF";
  1724. jlib_decl StringBuffer &appendJSONDataValue(StringBuffer& s, const char *name, unsigned len, const void *_value)
  1725. {
  1726. appendJSONNameOrDelimit(s, name);
  1727. s.append('"');
  1728. const unsigned char *value = (const unsigned char *) _value;
  1729. for (unsigned int i = 0; i < len; i++)
  1730. s.append(hexchar[value[i] >> 4]).append(hexchar[value[i] & 0x0f]);
  1731. return s.append('"');
  1732. }
  1733. StringBuffer &appendJSONRealValue(StringBuffer& s, const char *name, double value)
  1734. {
  1735. appendJSONNameOrDelimit(s, name);
  1736. bool quoted = j_isnan(value) || j_isinf(value);
  1737. if (quoted)
  1738. s.append('"');
  1739. s.append(value);
  1740. if (quoted)
  1741. s.append('"');
  1742. return s;
  1743. }
  1744. inline StringBuffer &encodeJSONChar(StringBuffer &s, const char *&ch)
  1745. {
  1746. switch (*ch)
  1747. {
  1748. case '\b':
  1749. s.append("\\b");
  1750. break;
  1751. case '\f':
  1752. s.append("\\f");
  1753. break;
  1754. case '\n':
  1755. s.append("\\n");
  1756. break;
  1757. case '\r':
  1758. s.append("\\r");
  1759. break;
  1760. case '\t':
  1761. s.append("\\t");
  1762. break;
  1763. case '\"':
  1764. case '\\':
  1765. case '/':
  1766. s.append('\\');
  1767. s.append(*ch);
  1768. break;
  1769. default:
  1770. if (*ch >= ' ' && ((byte)*ch) < 128)
  1771. s.append(*ch);
  1772. else if (*ch < ' ' && *ch > 0)
  1773. s.append("\\u00").appendhex(*ch, true);
  1774. else //json is always supposed to be utf8 (or other unicode formats)
  1775. {
  1776. unsigned chlen = utf8CharLen((const unsigned char *)ch);
  1777. if (chlen==0)
  1778. s.append("\\u00").appendhex(*ch, true);
  1779. else
  1780. {
  1781. s.append(*ch);
  1782. while(--chlen)
  1783. s.append(*(++ch));
  1784. }
  1785. }
  1786. break;
  1787. }
  1788. ch++;
  1789. return s;
  1790. }
  1791. StringBuffer &encodeJSON(StringBuffer &s, unsigned len, const char *value)
  1792. {
  1793. if (!value)
  1794. return s;
  1795. while (len-- && *value)
  1796. encodeJSONChar(s, value);
  1797. return s;
  1798. }
  1799. StringBuffer &encodeJSON(StringBuffer &s, const char *value)
  1800. {
  1801. if (!value)
  1802. return s;
  1803. while (*value)
  1804. encodeJSONChar(s, value);
  1805. return s;
  1806. }
  1807. void decodeCppEscapeSequence(StringBuffer & out, const char * in, bool errorIfInvalid)
  1808. {
  1809. out.ensureCapacity((size32_t)strlen(in));
  1810. while (*in)
  1811. {
  1812. char c = *in++;
  1813. if (c == '\\')
  1814. {
  1815. char next = *in;
  1816. if (next)
  1817. {
  1818. in++;
  1819. switch (next)
  1820. {
  1821. case 'a': c = '\a'; break;
  1822. case 'b': c = '\b'; break;
  1823. case 'f': c = '\f'; break;
  1824. case 'n': c = '\n'; break;
  1825. case 'r': c = '\r'; break;
  1826. case 't': c = '\t'; break;
  1827. case 'v': c = '\v'; break;
  1828. case '\\':
  1829. case '\'':
  1830. case '?':
  1831. case '\"': break;
  1832. case '0': case '1': case '2': case '3': case '4': case '5': case '6': case '7':
  1833. {
  1834. c = next - '0';
  1835. if (*in >= '0' && *in <= '7')
  1836. {
  1837. c = c << 3 | (*in++-'0');
  1838. if (*in >= '0' && *in <= '7')
  1839. c = c << 3 | (*in++-'0');
  1840. }
  1841. break;
  1842. }
  1843. case 'x':
  1844. c = 0;
  1845. while (isxdigit(*in))
  1846. {
  1847. next = *in++;
  1848. c = c << 4;
  1849. if (next >= '0' && next <= '9') c |= (next - '0');
  1850. else if (next >= 'A' && next <= 'F') c |= (next - 'A' + 10);
  1851. else if (next >= 'a' && next <= 'f') c |= (next - 'a' + 10);
  1852. }
  1853. break;
  1854. default:
  1855. if (errorIfInvalid)
  1856. throw MakeStringException(1, "unrecognised character escape sequence '\\%c'", next);
  1857. in--; // keep it as is.
  1858. break;
  1859. }
  1860. }
  1861. }
  1862. out.append(c);
  1863. }
  1864. }
  1865. bool isPrintable(unsigned len, const char * src)
  1866. {
  1867. while (len--)
  1868. {
  1869. if (!isprint(*((unsigned char *)src)))
  1870. return false;
  1871. src++;
  1872. }
  1873. return true;
  1874. }
  1875. //make this as fast as possible...
  1876. StringBuffer & appendStringAsSQL(StringBuffer & out, unsigned len, const char * src)
  1877. {
  1878. if (!isPrintable(len, src))
  1879. {
  1880. out.append("X'");
  1881. appendDataAsHex(out, len, src);
  1882. return out.append('\'');
  1883. }
  1884. out.ensureCapacity(2 + len);
  1885. out.append('\'');
  1886. loop
  1887. {
  1888. char * next = (char *)memchr(src, '\'', len);
  1889. if (!next)
  1890. break;
  1891. unsigned chunk=(size32_t)(next-src)+1;
  1892. out.append(chunk, src).append('\'');
  1893. len -= chunk;
  1894. src += chunk;
  1895. }
  1896. return out.append(len, src).append('\'');
  1897. }
  1898. static const char * hexText = "0123456789ABCDEF";
  1899. StringBuffer & appendDataAsHex(StringBuffer &out, unsigned len, const void * data)
  1900. {
  1901. char * target = (char *)out.reserve(len*2);
  1902. unsigned char * start = (unsigned char *)data;
  1903. for (unsigned count=len; count> 0; --count)
  1904. {
  1905. unsigned next = *start++;
  1906. *target++ = hexText[next >>4];
  1907. *target++ = hexText[next & 15];
  1908. }
  1909. return out;
  1910. }
  1911. bool strToBool(size_t len, const char * text)
  1912. {
  1913. switch (len)
  1914. {
  1915. case 4:
  1916. if (memicmp(text, "true", 4) == 0)
  1917. return true;
  1918. break;
  1919. case 3:
  1920. if (memicmp(text, "yes", 3) == 0)
  1921. return true;
  1922. break;
  1923. case 2:
  1924. if (memicmp(text, "on", 2) == 0)
  1925. return true;
  1926. break;
  1927. case 1:
  1928. if ((memicmp(text, "t", 1) == 0) || (memicmp(text, "y", 1) == 0))
  1929. return true;
  1930. break;
  1931. }
  1932. while (len && isspace(*text))
  1933. {
  1934. len--;
  1935. text++;
  1936. }
  1937. while (len-- && isdigit(*text))
  1938. {
  1939. if (*text++ != '0') return true;
  1940. }
  1941. return false;
  1942. }
  1943. bool strToBool(const char * text)
  1944. {
  1945. return strToBool(strlen(text), text);
  1946. }
  1947. bool clipStrToBool(size_t len, const char * text)
  1948. {
  1949. while (len && *text==' ')
  1950. {
  1951. len--;
  1952. text++;
  1953. }
  1954. while (len && text[len-1]== ' ')
  1955. len--;
  1956. return strToBool(len, text);
  1957. }
  1958. bool clipStrToBool(const char * text)
  1959. {
  1960. return clipStrToBool(strlen(text), text);
  1961. }
  1962. StringBuffer & ncnameEscape(char const * in, StringBuffer & out)
  1963. {
  1964. if(!isalpha(*in))
  1965. {
  1966. out.appendf("_%02X", static_cast<unsigned char>(*in));
  1967. in++;
  1968. }
  1969. char const * finger = in;
  1970. while(*finger)
  1971. {
  1972. if(!isalnum(*finger))
  1973. {
  1974. if(finger>in)
  1975. out.append((size32_t)(finger-in), in);
  1976. out.appendf("_%02X", static_cast<unsigned char>(*finger));
  1977. in = ++finger;
  1978. }
  1979. else
  1980. {
  1981. finger++;
  1982. }
  1983. }
  1984. if(finger>in)
  1985. out.append((size32_t)(finger-in), in);
  1986. return out;
  1987. }
  1988. StringBuffer & ncnameUnescape(char const * in, StringBuffer & out)
  1989. {
  1990. char const * finger = in;
  1991. while(*finger)
  1992. {
  1993. if(*finger == '_')
  1994. {
  1995. if(finger>in)
  1996. out.append((size32_t)(finger-in), in);
  1997. unsigned char chr = 16 * hex2num(finger[1]) + hex2num(finger[2]);
  1998. out.append(static_cast<char>(chr));
  1999. in = (finger+=3);
  2000. }
  2001. else
  2002. {
  2003. finger++;
  2004. }
  2005. }
  2006. if(finger>in)
  2007. out.append((size32_t)(finger-in), in);
  2008. return out;
  2009. }
  2010. bool startsWith(const char* src, const char* dst)
  2011. {
  2012. while (*dst && *dst == *src) { src++; dst++; }
  2013. return *dst==0;
  2014. }
  2015. bool startsWithIgnoreCase(const char* src, const char* dst)
  2016. {
  2017. while (*dst && tolower(*dst) == tolower(*src)) { src++; dst++; }
  2018. return *dst==0;
  2019. }
  2020. bool endsWith(const char* src, const char* dst)
  2021. {
  2022. size_t srcLen = strlen(src);
  2023. size_t dstLen = strlen(dst);
  2024. if (dstLen<=srcLen)
  2025. return memcmp(dst, src+srcLen-dstLen, dstLen)==0;
  2026. return false;
  2027. }
  2028. bool endsWithIgnoreCase(const char* src, const char* dst)
  2029. {
  2030. size_t srcLen = strlen(src);
  2031. size_t dstLen = strlen(dst);
  2032. if (dstLen<=srcLen)
  2033. return memicmp(dst, src+srcLen-dstLen, dstLen)==0;
  2034. return false;
  2035. }
  2036. char *j_strtok_r(char *str, const char *delim, char **saveptr)
  2037. {
  2038. if (!str)
  2039. str = *saveptr;
  2040. char c;
  2041. loop {
  2042. c = *str;
  2043. if (!c) {
  2044. *saveptr = str;
  2045. return NULL;
  2046. }
  2047. if (!strchr(delim,c))
  2048. break;
  2049. str++;
  2050. }
  2051. char *ret=str;
  2052. do {
  2053. c = *(++str);
  2054. } while (c&&!strchr(delim,c));
  2055. if (c)
  2056. *(str++) = 0;
  2057. *saveptr = str;
  2058. return ret;
  2059. }
  2060. int j_memicmp (const void *s1, const void *s2, size32_t len)
  2061. {
  2062. const byte *b1 = (const byte *)s1;
  2063. const byte *b2 = (const byte *)s2;
  2064. int ret = 0;
  2065. while (len&&((ret = tolower(*b1)-tolower(*b2)) == 0)) {
  2066. b1++;
  2067. b2++;
  2068. len--;
  2069. }
  2070. return ret;
  2071. }
  2072. size32_t memcount(size32_t len, const char * str, char search)
  2073. {
  2074. size32_t count = 0;
  2075. for (size32_t i=0; i < len; i++)
  2076. {
  2077. if (str[i] == search)
  2078. count++;
  2079. }
  2080. return count;
  2081. }
  2082. StringBuffer & elideString(StringBuffer & s, unsigned maxLength)
  2083. {
  2084. if (s.length() > maxLength)
  2085. {
  2086. s.setLength(maxLength);
  2087. s.append("...");
  2088. }
  2089. return s;
  2090. }
  2091. const char * nullText(const char * text)
  2092. {
  2093. if (text) return text;
  2094. return "(null)";
  2095. }