jstring.cpp 53 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253
  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. const char * StringBuffer::toCharArray() const
  723. {
  724. if (buffer)
  725. {
  726. buffer[curLen] = '\0'; // There is always room for this null
  727. return buffer;
  728. }
  729. return TheNullStr;
  730. }
  731. //===========================================================================
  732. VStringBuffer::VStringBuffer(const char* format, ...)
  733. {
  734. va_list args;
  735. va_start(args,format);
  736. valist_appendf(format,args);
  737. va_end(args);
  738. }
  739. //===========================================================================
  740. String::String()
  741. {
  742. text = (char *)TheNullStr;
  743. }
  744. String::String(const char * value)
  745. {
  746. text = (value ? strdup(value) : (char *)TheNullStr);
  747. }
  748. String::String(const char * value, int offset, int _count)
  749. {
  750. text = (char *)malloc(_count+1);
  751. memcpy(text, value+offset, _count);
  752. text[_count]=0;
  753. }
  754. String::String(String & value)
  755. {
  756. text = strdup(value.toCharArray());
  757. }
  758. String::String(StringBuffer & value)
  759. {
  760. unsigned len = value.length();
  761. text = (char *)malloc(len+1);
  762. value.getChars(0,len,text);
  763. text[len] = 0;
  764. }
  765. String::~String()
  766. {
  767. if (text != TheNullStr) free(text);
  768. }
  769. char String::charAt(size32_t index) const
  770. {
  771. return text[index];
  772. }
  773. int String::compareTo(const String & value) const
  774. {
  775. return strcmp(text, value.toCharArray());
  776. }
  777. int String::compareTo(const char* value) const
  778. {
  779. return strcmp(text,value);
  780. }
  781. String * String::concat(const String & value) const
  782. {
  783. StringBuffer temp(toCharArray());
  784. temp.append(value);
  785. return new String(temp.str());
  786. }
  787. bool String::endsWith(const String & value) const
  788. {
  789. unsigned lenValue = value.length();
  790. unsigned len = (size32_t)strlen(text);
  791. if (len >= lenValue)
  792. return (memcmp(text+(len-lenValue),value.toCharArray(),lenValue) == 0);
  793. return false;
  794. }
  795. bool String::endsWith(const char* value) const
  796. {
  797. return ::endsWith(this->text, value);
  798. }
  799. bool String::equals(String & value) const
  800. {
  801. return strcmp(text, value.toCharArray())==0;
  802. }
  803. bool String::equalsIgnoreCase(const String & value) const
  804. {
  805. return stricmp(text, value.toCharArray())==0;
  806. }
  807. void String::getBytes(int srcBegin, int srcEnd, void * dest, int dstBegin) const
  808. {
  809. memcpy((char *)dest+dstBegin, text+srcBegin, srcEnd-srcBegin);
  810. }
  811. void String::getChars(int srcBegin, int srcEnd, void * dest, int dstBegin) const
  812. {
  813. memcpy((char *)dest+dstBegin, text+srcBegin, srcEnd-srcBegin);
  814. }
  815. int String::hashCode() const
  816. {
  817. return (int)hashc((const byte *)text,length(),0);
  818. }
  819. int String::indexOf(int ch) const
  820. {
  821. char * match = strchr(text, ch);
  822. return match ? (int)(match - text) : -1;
  823. }
  824. int String::indexOf(int ch, int from) const
  825. {
  826. char * match = strchr(text + from, ch);
  827. return match ? (int)(match - text) : -1;
  828. }
  829. int String::indexOf(const String & search) const
  830. {
  831. const char * str = search.toCharArray();
  832. const char * match = strstr(text, str);
  833. return match ? (int)(match - text) : -1;
  834. }
  835. int String::indexOf(const String & search, int from) const
  836. {
  837. const char * str = search.toCharArray();
  838. const char * match = strstr(text + from, str);
  839. return match ? (int)(match - text) : -1;
  840. }
  841. int String::lastIndexOf(int ch) const
  842. {
  843. char * match = strrchr(text, ch);
  844. return match ? (int)(match - text) : -1;
  845. }
  846. int String::lastIndexOf(int ch, int from) const
  847. {
  848. for (;(from > 0);--from)
  849. if (text[from] == ch)
  850. return from;
  851. return -1;
  852. }
  853. int String::lastIndexOf(const String & search) const
  854. {
  855. assertex(!"TBD");
  856. return -1;
  857. }
  858. int String::lastIndexOf(const String & search, int from) const
  859. {
  860. assertex(!"TBD");
  861. return -1;
  862. }
  863. size32_t String::length() const
  864. {
  865. return (size32_t)strlen(text);
  866. }
  867. bool String::startsWith(String & value) const
  868. {
  869. unsigned lenValue = value.length();
  870. const char * search = value.toCharArray();
  871. return (memcmp(text, search, lenValue) == 0);
  872. }
  873. bool String::startsWith(String & value, int offset) const
  874. {
  875. unsigned lenValue = value.length();
  876. const char * search = value.toCharArray();
  877. return (memcmp(text + offset, search, lenValue) == 0);
  878. }
  879. bool String::startsWith(const char* value) const
  880. {
  881. return ::startsWith(this->text,value);
  882. }
  883. String * String::substring(int beginIndex) const
  884. {
  885. return new String(text+beginIndex);
  886. }
  887. String * String::substring(int beginIndex, int endIndex) const
  888. {
  889. return new String(text, beginIndex, endIndex - beginIndex);
  890. }
  891. const char *String::toCharArray() const
  892. {
  893. return text;
  894. }
  895. String * String::toLowerCase() const
  896. {
  897. String *ret = new String();
  898. size32_t l = length();
  899. if (l)
  900. {
  901. ret->text = (char *)malloc(l+1);
  902. for (unsigned i = 0; i < l; i++)
  903. ret->text[i] = tolower(text[i]);
  904. ret->text[l]=0;
  905. }
  906. return ret;
  907. }
  908. String * String::toString()
  909. {
  910. Link();
  911. return this;
  912. }
  913. String * String::toUpperCase() const
  914. {
  915. String *ret = new String();
  916. size32_t l = length();
  917. if (l)
  918. {
  919. ret->text = (char *)malloc(l+1);
  920. for (unsigned i = 0; i < l; i++)
  921. ret->text[i] = toupper(text[i]);
  922. ret->text[l]=0;
  923. }
  924. return ret;
  925. }
  926. String * String::trim() const
  927. {
  928. size32_t l = length();
  929. while (l && isspace(text[l-1]))
  930. l--;
  931. return new String(text, 0, l);
  932. }
  933. //------------------------------------------------
  934. #if 0
  935. String & String::valueOf(char value)
  936. {
  937. return * new String(&value, 0, 1);
  938. }
  939. String & String::valueOf(const char * value)
  940. {
  941. return * new String(value);
  942. }
  943. String & String::valueOf(const char * value, int offset, int count)
  944. {
  945. return * new String(value, offset, count);
  946. }
  947. String & String::valueOf(double value)
  948. {
  949. StringBuffer temp;
  950. return temp.append(value).toString();
  951. }
  952. String & String::valueOf(float value)
  953. {
  954. StringBuffer temp;
  955. return temp.append(value).toString();
  956. }
  957. String & String::valueOf(int value)
  958. {
  959. StringBuffer temp;
  960. return temp.append(value).toString();
  961. }
  962. String & String::valueOf(long value)
  963. {
  964. StringBuffer temp;
  965. return temp.append(value).toString();
  966. }
  967. #endif
  968. //------------------------------------------------
  969. StringAttr::StringAttr(const char * _text)
  970. {
  971. text = _text ? strdup(_text) : NULL;
  972. }
  973. StringAttr::StringAttr(const char * _text, unsigned _len)
  974. {
  975. text = NULL;
  976. set(_text, _len);
  977. }
  978. StringAttr::StringAttr(const StringAttr & src)
  979. {
  980. text = NULL;
  981. set(src.get());
  982. }
  983. void StringAttr::set(const char * _text)
  984. {
  985. free(text);
  986. text = _text ? strdup(_text) : NULL;
  987. }
  988. void StringAttr::set(const char * _text, unsigned _len)
  989. {
  990. if (text)
  991. free(text);
  992. text = (char *)malloc(_len+1);
  993. memcpy(text, _text, _len);
  994. text[_len] = 0;
  995. }
  996. void StringAttr::setown(const char * _text)
  997. {
  998. if (text)
  999. free(text);
  1000. text = (char *)_text;
  1001. }
  1002. void StringAttr::toLowerCase()
  1003. {
  1004. if (text)
  1005. {
  1006. char * cur = text;
  1007. char next;
  1008. while ((next = *cur) != 0)
  1009. {
  1010. if (isupper(next))
  1011. *cur = tolower(next);
  1012. cur++;
  1013. }
  1014. }
  1015. }
  1016. void StringAttr::toUpperCase()
  1017. {
  1018. if (text)
  1019. {
  1020. char * cur = text;
  1021. char next;
  1022. while ((next = *cur) != 0)
  1023. {
  1024. if (islower(next))
  1025. *cur = toupper(next);
  1026. cur++;
  1027. }
  1028. }
  1029. }
  1030. StringAttrItem::StringAttrItem(const char *_text, unsigned _len)
  1031. {
  1032. text.set(_text, _len);
  1033. }
  1034. inline char hex(char c, char lower)
  1035. {
  1036. if (c < 10)
  1037. return '0' + c;
  1038. else if (lower)
  1039. return 'a' + c - 10;
  1040. else
  1041. return 'A' + c - 10;
  1042. }
  1043. StringBuffer & StringBuffer::appendhex(unsigned char c, char lower)
  1044. {
  1045. append(hex(c>>4, lower));
  1046. append(hex(c&0xF, lower));
  1047. return *this;
  1048. }
  1049. void appendURL(StringBuffer *dest, const char *src, size32_t len, char lower)
  1050. {
  1051. if (len == (size32_t)-1)
  1052. len = (size32_t)strlen(src);
  1053. while (len)
  1054. {
  1055. // isalnum seems to give weird results for chars > 127....
  1056. unsigned char c = (unsigned char) *src;
  1057. if (c == ' ')
  1058. dest->append('+');
  1059. else if ((c & 0x80) || !isalnum(*src))
  1060. {
  1061. dest->append('%');
  1062. dest->appendhex(c, lower);
  1063. }
  1064. else
  1065. dest->append(c);
  1066. src++;
  1067. len--;
  1068. }
  1069. }
  1070. static StringBuffer & appendStringExpandControl(StringBuffer &out, unsigned len, const char * src, bool addBreak, bool isCpp)
  1071. {
  1072. const int minBreakPos = 0;
  1073. const int commaBreakPos = 70;
  1074. const int maxBreakPos = 120;
  1075. const char * startLine = src;
  1076. out.ensureCapacity(len+2);
  1077. for (; len > 0; --len)
  1078. {
  1079. unsigned char c = *src++;
  1080. bool insertBreak = false;
  1081. bool allowBreak = true;
  1082. switch (c)
  1083. {
  1084. case '\n':
  1085. {
  1086. out.append("\\n");
  1087. if (src-startLine > minBreakPos)
  1088. insertBreak = true;
  1089. break;
  1090. }
  1091. case ',':
  1092. {
  1093. out.append(c);
  1094. if (src-startLine > commaBreakPos)
  1095. insertBreak = true;
  1096. break;
  1097. }
  1098. case '\r': out.append("\\r"); break;
  1099. case '\t': out.append("\\t"); break;
  1100. case '"':
  1101. if (isCpp)
  1102. out.append("\\");
  1103. out.append(c);
  1104. break;
  1105. case '\'':
  1106. if (!isCpp)
  1107. out.append("\\");
  1108. out.append(c);
  1109. break;
  1110. case '\\': out.append("\\\\"); break;
  1111. case '?':
  1112. if (isCpp)
  1113. {
  1114. //stop trigraphs being generated.... quote the second ?
  1115. out.append(c);
  1116. if ((len!=1) && (*src == '?'))
  1117. {
  1118. out.append('\\');
  1119. allowBreak = false;
  1120. }
  1121. }
  1122. else
  1123. out.append(c);
  1124. break;
  1125. default:
  1126. if ((c >= ' ') && (c <= 126))
  1127. out.append(c);
  1128. else
  1129. out.appendf("\\%03o", c);
  1130. break;
  1131. }
  1132. if (addBreak && (insertBreak || (allowBreak && src-startLine >= maxBreakPos)))
  1133. {
  1134. out.append("\"").newline().append("\t\t\"");
  1135. startLine = src;
  1136. }
  1137. }
  1138. return out;
  1139. }
  1140. StringBuffer & appendStringAsCPP(StringBuffer &out, unsigned len, const char * src, bool addBreak)
  1141. {
  1142. return appendStringExpandControl(out, len, src, addBreak, true);
  1143. }
  1144. StringBuffer & appendStringAsECL(StringBuffer &out, unsigned len, const char * src)
  1145. {
  1146. return appendStringExpandControl(out, len, src, false, false);
  1147. }
  1148. StringBuffer & appendStringAsQuotedCPP(StringBuffer &out, unsigned len, const char * src, bool addBreak)
  1149. {
  1150. out.ensureCapacity(len+2);
  1151. out.append('\"');
  1152. appendStringAsCPP(out, len, src, addBreak);
  1153. return out.append('\"');
  1154. }
  1155. StringBuffer & appendStringAsQuotedECL(StringBuffer &out, unsigned len, const char * src)
  1156. {
  1157. out.ensureCapacity(len+2);
  1158. out.append('\'');
  1159. appendStringAsECL(out, len, src);
  1160. return out.append('\'');
  1161. }
  1162. void extractItem(StringBuffer & res, const char * src, const char * sep, int whichItem, bool caps)
  1163. {
  1164. bool isSeparator[256];
  1165. memset(isSeparator,0,sizeof(isSeparator));
  1166. unsigned char * finger = (unsigned char *)sep;
  1167. while (*finger !=0)
  1168. isSeparator[*finger++] = true;
  1169. isSeparator[0]=true;
  1170. finger = (unsigned char *)src;
  1171. unsigned char next;
  1172. loop
  1173. {
  1174. while (isSeparator[(next = *finger)])
  1175. {
  1176. if (next == 0) return;
  1177. finger++;
  1178. }
  1179. if (whichItem == 0)
  1180. {
  1181. while (!isSeparator[(next = *finger)])
  1182. {
  1183. if (caps)
  1184. next = toupper(next);
  1185. res.append(next);
  1186. finger++;
  1187. }
  1188. return;
  1189. }
  1190. while (!isSeparator[*finger])
  1191. finger++;
  1192. whichItem--;
  1193. }
  1194. }
  1195. int utf8CharLen(const unsigned char *ch)
  1196. {
  1197. //return 1 if this is an ascii character,
  1198. //or 0 if its not a valid utf-8 character
  1199. if (*ch < 128)
  1200. return 1;
  1201. if (*ch < 192)
  1202. return 0;
  1203. unsigned char len = 1;
  1204. for (unsigned char lead = *ch << 1; (lead & 0x80); lead <<=1)
  1205. len++;
  1206. for (unsigned pos = 1; pos < len; pos++)
  1207. if ((ch[pos] < 128) || (ch[pos] >= 192))
  1208. return 0; //its not a valid utf-8 character after all
  1209. return len;
  1210. }
  1211. const char *encodeXML(const char *x, StringBuffer &ret, unsigned flags, unsigned len, bool utf8)
  1212. {
  1213. while (len)
  1214. {
  1215. switch(*x)
  1216. {
  1217. case '&':
  1218. ret.append("&amp;");
  1219. break;
  1220. case '<':
  1221. ret.append("&lt;");
  1222. break;
  1223. case '>':
  1224. ret.append("&gt;");
  1225. break;
  1226. case '\"':
  1227. ret.append("&quot;");
  1228. break;
  1229. case '\'':
  1230. ret.append("&apos;");
  1231. break;
  1232. case ' ':
  1233. ret.append(flags & ENCODE_SPACES?"&#32;":" ");
  1234. break;
  1235. case '\n':
  1236. ret.append(flags & ENCODE_NEWLINES?"&#10;":"\n");
  1237. break;
  1238. case '\r':
  1239. ret.append(flags & ENCODE_NEWLINES?"&#13;":"\r");
  1240. break;
  1241. case '\t':
  1242. ret.append(flags & ENCODE_SPACES?"&#9;":"\t");
  1243. break;
  1244. case '\0':
  1245. if (len == (unsigned) -1)
  1246. return x;
  1247. ret.append("&#xe000;"); // hack!!! Characters below 0x20 are not legal in strict xml, even encoded.
  1248. break;
  1249. default:
  1250. if (*x >= ' ' && ((byte)*x) < 128)
  1251. ret.append(*x);
  1252. else if (*x < ' ' && *x > 0)
  1253. ret.append("&#xe0").appendhex(*x, true).append(';'); // HACK
  1254. else if (utf8)
  1255. {
  1256. unsigned chlen = utf8CharLen((const unsigned char *)x);
  1257. if (chlen==0)
  1258. ret.append("&#").append((unsigned int)*(unsigned char *) x).append(';');
  1259. else
  1260. {
  1261. ret.append(*x);
  1262. while(--chlen)
  1263. {
  1264. if (len != (unsigned) -1)
  1265. len--;
  1266. ret.append(*(++x));
  1267. }
  1268. }
  1269. }
  1270. else
  1271. ret.append("&#").append((unsigned int)*(unsigned char *) x).append(';');
  1272. break;
  1273. }
  1274. if (len != (unsigned) -1)
  1275. len--;
  1276. ++x;
  1277. }
  1278. return x;
  1279. }
  1280. const char *encodeXML(const char *x, IIOStream &out, unsigned flags, unsigned len, bool utf8)
  1281. {
  1282. while (len)
  1283. {
  1284. switch(*x)
  1285. {
  1286. case '&':
  1287. writeStringToStream(out, "&amp;");
  1288. break;
  1289. case '<':
  1290. writeStringToStream(out, "&lt;");
  1291. break;
  1292. case '>':
  1293. writeStringToStream(out, "&gt;");
  1294. break;
  1295. case '\"':
  1296. writeStringToStream(out, "&quot;");
  1297. break;
  1298. case '\'':
  1299. writeStringToStream(out, "&apos;");
  1300. break;
  1301. case ' ':
  1302. writeStringToStream(out, flags & ENCODE_SPACES?"&#32;":" ");
  1303. break;
  1304. case '\n':
  1305. writeStringToStream(out, flags & ENCODE_NEWLINES?"&#10;":"\n");
  1306. break;
  1307. case '\r':
  1308. writeStringToStream(out, flags & ENCODE_NEWLINES?"&#13;":"\r");
  1309. break;
  1310. case '\t':
  1311. writeStringToStream(out, flags & ENCODE_SPACES?"&#9;":"\t");
  1312. break;
  1313. case '\0':
  1314. if (len == (unsigned) -1)
  1315. return x;
  1316. writeStringToStream(out, "&#xe000;"); // hack!!! Characters below 0x20 are not legal in strict xml, even encoded.
  1317. break;
  1318. default:
  1319. if (*x >= ' ' && ((byte)*x) < 128)
  1320. writeCharToStream(out, *x);
  1321. else if (*x < ' ' && *x > 0)
  1322. {
  1323. writeStringToStream(out, "&#xe0");
  1324. unsigned char c = *(unsigned char *)x;
  1325. writeCharToStream(out, hex(c>>4, true));
  1326. writeCharToStream(out, hex(c&0xF, true));
  1327. writeCharToStream(out, ';'); // HACK
  1328. }
  1329. else if (utf8)
  1330. {
  1331. int chlen = utf8CharLen((const unsigned char *)x);
  1332. if (chlen==0)
  1333. {
  1334. writeStringToStream(out, "&#");
  1335. char tmp[12];
  1336. unsigned written = numtostr(tmp, *(unsigned char *)x);
  1337. out.write(written, tmp);
  1338. writeCharToStream(out, ';');
  1339. }
  1340. else
  1341. {
  1342. writeCharToStream(out, *x);
  1343. while(--chlen)
  1344. {
  1345. if (len != (unsigned) -1)
  1346. len--;
  1347. writeCharToStream(out, *(++x));
  1348. }
  1349. }
  1350. }
  1351. else
  1352. {
  1353. writeStringToStream(out, "&#");
  1354. char tmp[12];
  1355. unsigned written = numtostr(tmp, *(unsigned char *)x);
  1356. out.write(written, tmp);
  1357. writeCharToStream(out, ';');
  1358. }
  1359. break;
  1360. }
  1361. if (len != (unsigned) -1)
  1362. len--;
  1363. ++x;
  1364. }
  1365. return x;
  1366. }
  1367. static void writeUtf8(unsigned c, StringBuffer &out)
  1368. {
  1369. if (c < 0x80)
  1370. out.append((char)c);
  1371. else if (c < 0x800)
  1372. {
  1373. out.append((char)(0xC0 | (c>>6)));
  1374. out.append((char)(0x80 | (c & 0x3F)));
  1375. }
  1376. else if (c < 0x10000)
  1377. {
  1378. out.append((char) (0xE0 | (c>>12)));
  1379. out.append((char) (0x80 | (c>>6 & 0x3F)));
  1380. out.append((char) (0x80 | (c & 0x3F)));
  1381. }
  1382. else if (c < 0x200000)
  1383. {
  1384. out.append((char) (0xF0 | (c>>18)));
  1385. out.append((char) (0x80 | (c>>12 & 0x3F)));
  1386. out.append((char) (0x80 | (c>>6 & 0x3F)));
  1387. out.append((char) (0x80 | (c & 0x3F)));
  1388. }
  1389. else if (c < 0x4000000)
  1390. {
  1391. out.append((char) (0xF8 | (c>>24)));
  1392. out.append((char) (0x80 | (c>>18 & 0x3F)));
  1393. out.append((char) (0x80 | (c>>12 & 0x3F)));
  1394. out.append((char) (0x80 | (c>>6 & 0x3F)));
  1395. out.append((char) (0x80 | (c & 0x3F)));
  1396. }
  1397. else if (c < 0x80000000)
  1398. {
  1399. out.append((char) (0xFC | (c>>30)));
  1400. out.append((char) (0x80 | (c>>24 & 0x3F)));
  1401. out.append((char) (0x80 | (c>>18 & 0x3F)));
  1402. out.append((char) (0x80 | (c>>12 & 0x3F)));
  1403. out.append((char) (0x80 | (c>>6 & 0x3F)));
  1404. out.append((char) (0x80 | (c & 0x3F)));
  1405. }
  1406. else
  1407. assertex(false);
  1408. }
  1409. #define JSONSTRICT
  1410. const char *decodeJSON(const char *j, StringBuffer &ret, unsigned len, const char **errMark)
  1411. {
  1412. if (!j)
  1413. return j;
  1414. if ((unsigned)-1 == len)
  1415. len = (unsigned)strlen(j);
  1416. try
  1417. {
  1418. for (const char *end = j+len; j<end && *j; j++)
  1419. {
  1420. if (*j!='\\')
  1421. ret.append(*j);
  1422. else
  1423. {
  1424. switch (*++j)
  1425. {
  1426. case 'u':
  1427. {
  1428. j++;
  1429. if (end-j>=4)
  1430. {
  1431. char *endptr;
  1432. StringAttr s(j, 4);
  1433. unsigned val = strtoul(s.get(), &endptr, 16);
  1434. if (endptr && !*endptr)
  1435. {
  1436. writeUtf8(val, ret);
  1437. j+=3;
  1438. break;
  1439. }
  1440. }
  1441. #ifdef JSONSTRICT
  1442. throw MakeStringException(-1, "invalid json \\u escaped sequence");
  1443. #endif
  1444. ret.append(*j);
  1445. break;
  1446. }
  1447. case '\"':
  1448. case '\\':
  1449. case '/':
  1450. ret.append(*j);
  1451. break;
  1452. case 'b':
  1453. ret.append('\b');
  1454. break;
  1455. case 'f':
  1456. ret.append('\f');
  1457. break;
  1458. case 'n':
  1459. ret.append('\n');
  1460. continue;
  1461. case 'r':
  1462. ret.append('\r');
  1463. break;
  1464. case 't':
  1465. ret.append('\t');
  1466. break;
  1467. default:
  1468. {
  1469. #ifdef JSONSTRICT
  1470. throw MakeStringException(-1, "invalid json escaped sequence");
  1471. #endif
  1472. ret.append('\\');
  1473. ret.append(*j);
  1474. break;
  1475. }
  1476. }
  1477. }
  1478. }
  1479. }
  1480. catch (IException *)
  1481. {
  1482. if (errMark) *errMark = j;
  1483. throw;
  1484. }
  1485. return j;
  1486. }
  1487. void decodeXML(ISimpleReadStream &in, StringBuffer &out, unsigned len)
  1488. {
  1489. // TODO
  1490. UNIMPLEMENTED;
  1491. }
  1492. const char *decodeXML(const char *x, StringBuffer &ret, const char **errMark, IEntityHelper *entityHelper, bool strict)
  1493. {
  1494. if (!x)
  1495. return x;
  1496. try
  1497. {
  1498. while (*x)
  1499. {
  1500. if ('&' == *x)
  1501. {
  1502. switch (x[1])
  1503. {
  1504. case 'a':
  1505. switch (x[2])
  1506. {
  1507. case 'm':
  1508. {
  1509. if ('p' == x[3] && ';' == x[4])
  1510. {
  1511. x += 5;
  1512. ret.append('&');
  1513. continue;
  1514. }
  1515. break;
  1516. }
  1517. case 'p':
  1518. {
  1519. if ('o' == x[3] && 's' == x[4] && ';' == x[5])
  1520. {
  1521. x += 6;
  1522. ret.append('\'');
  1523. continue;
  1524. }
  1525. break;
  1526. }
  1527. }
  1528. break;
  1529. case 'l':
  1530. if ('t' == x[2] && ';' == x[3])
  1531. {
  1532. x += 4;
  1533. ret.append('<');
  1534. continue;
  1535. }
  1536. break;
  1537. case 'g':
  1538. if ('t' == x[2] && ';' == x[3])
  1539. {
  1540. x += 4;
  1541. ret.append('>');
  1542. continue;
  1543. }
  1544. break;
  1545. case 'q':
  1546. if ('u' == x[2] && 'o' == x[3] && 't' == x[4] && ';' == x[5])
  1547. {
  1548. x += 6;
  1549. ret.append('"');
  1550. continue;
  1551. }
  1552. break;
  1553. case 'n':
  1554. if ('b' == x[2] && 's' == x[3] && 'p' == x[4] && ';' == x[5])
  1555. {
  1556. x += 6;
  1557. writeUtf8(0xa0, ret);
  1558. continue;
  1559. }
  1560. break;
  1561. case '#':
  1562. {
  1563. const char *numstart = x+2;
  1564. int base = 10;
  1565. if (*numstart == 'x')
  1566. {
  1567. base = 16;
  1568. numstart++;
  1569. }
  1570. char *numend;
  1571. unsigned val = strtoul(numstart, &numend, base);
  1572. if (numstart==numend || *numend != ';')
  1573. {
  1574. if (strict)
  1575. throw MakeStringException(-1, "invalid escaped sequence");
  1576. }
  1577. else // always convert to utf-8. Should potentially throw error if not marked as utf-8 encoded doc and out of ascii range.
  1578. {
  1579. writeUtf8(val, ret);
  1580. x = numend+1;
  1581. continue;
  1582. }
  1583. break;
  1584. }
  1585. case ';':
  1586. case '\0':
  1587. if (strict)
  1588. throw MakeStringException(-1, "invalid escaped sequence");
  1589. break;
  1590. default:
  1591. if (entityHelper)
  1592. {
  1593. bool error = false;
  1594. const char *start=x+1;
  1595. const char *finger=start;
  1596. while (*finger && *finger != ';')
  1597. ++finger;
  1598. if (*finger == ';')
  1599. {
  1600. StringBuffer entity(finger-start, start);
  1601. if (entityHelper->find(entity, ret))
  1602. {
  1603. x = finger + 1;
  1604. continue;
  1605. }
  1606. }
  1607. }
  1608. if (strict)
  1609. throw MakeStringException(-1, "invalid escaped sequence");
  1610. break;
  1611. }
  1612. }
  1613. ret.append(*x);
  1614. ++x;
  1615. }
  1616. }
  1617. catch (IException *)
  1618. {
  1619. if (errMark) *errMark = x;
  1620. throw;
  1621. }
  1622. return x;
  1623. }
  1624. StringBuffer & appendXMLOpenTag(StringBuffer &xml, const char *tag, const char *prefix, bool complete, bool close, const char *uri)
  1625. {
  1626. if (!tag || !*tag)
  1627. return xml;
  1628. xml.append('<');
  1629. appendXMLTagName(xml, tag, prefix);
  1630. if (uri && *uri)
  1631. {
  1632. xml.append(" xmlns");
  1633. if (prefix && *prefix)
  1634. xml.append(':').append(prefix);
  1635. xml.append("=\"").append(uri).append('\"');
  1636. }
  1637. if (complete)
  1638. {
  1639. if (close)
  1640. xml.append('/');
  1641. xml.append('>');
  1642. }
  1643. return xml;
  1644. }
  1645. jlib_decl StringBuffer &appendJSONName(StringBuffer &s, const char *name)
  1646. {
  1647. if (!name || !*name)
  1648. return s;
  1649. delimitJSON(s);
  1650. return encodeJSON(s.append('"'), name).append("\": ");
  1651. }
  1652. jlib_decl StringBuffer &appendfJSONName(StringBuffer &s, const char *format, ...)
  1653. {
  1654. va_list args;
  1655. va_start(args, format);
  1656. StringBuffer vs;
  1657. vs.valist_appendf(format, args);
  1658. va_end(args);
  1659. return appendJSONName(s, vs);
  1660. }
  1661. static char hexchar[] = "0123456789ABCDEF";
  1662. jlib_decl StringBuffer &appendJSONValue(StringBuffer& s, const char *name, unsigned len, const void *_value)
  1663. {
  1664. appendJSONNameOrDelimit(s, name);
  1665. s.append('"');
  1666. const unsigned char *value = (const unsigned char *) _value;
  1667. for (unsigned int i = 0; i < len; i++)
  1668. s.append(hexchar[value[i] >> 4]).append(hexchar[value[i] & 0x0f]);
  1669. return s.append('"');
  1670. }
  1671. inline StringBuffer &encodeJSON(StringBuffer &s, const char ch)
  1672. {
  1673. switch (ch)
  1674. {
  1675. case '\b':
  1676. s.append("\\b");
  1677. break;
  1678. case '\f':
  1679. s.append("\\f");
  1680. break;
  1681. case '\n':
  1682. s.append("\\n");
  1683. break;
  1684. case '\r':
  1685. s.append("\\r");
  1686. break;
  1687. case '\t':
  1688. s.append("\\t");
  1689. break;
  1690. case '\"':
  1691. case '\\':
  1692. case '/':
  1693. s.append('\\'); //fall through
  1694. default:
  1695. s.append(ch);
  1696. }
  1697. return s;
  1698. }
  1699. StringBuffer &encodeJSON(StringBuffer &s, unsigned len, const char *value)
  1700. {
  1701. if (!value)
  1702. return s;
  1703. unsigned pos=0;
  1704. while(pos<len && value[pos]!=0)
  1705. encodeJSON(s, value[pos++]);
  1706. return s;
  1707. }
  1708. StringBuffer &encodeJSON(StringBuffer &s, const char *value)
  1709. {
  1710. if (!value)
  1711. return s;
  1712. while (*value)
  1713. encodeJSON(s, *value++);
  1714. return s;
  1715. }
  1716. void decodeCppEscapeSequence(StringBuffer & out, const char * in, bool errorIfInvalid)
  1717. {
  1718. out.ensureCapacity((size32_t)strlen(in));
  1719. while (*in)
  1720. {
  1721. char c = *in++;
  1722. if (c == '\\')
  1723. {
  1724. char next = *in;
  1725. if (next)
  1726. {
  1727. in++;
  1728. switch (next)
  1729. {
  1730. case 'a': c = '\a'; break;
  1731. case 'b': c = '\b'; break;
  1732. case 'f': c = '\f'; break;
  1733. case 'n': c = '\n'; break;
  1734. case 'r': c = '\r'; break;
  1735. case 't': c = '\t'; break;
  1736. case 'v': c = '\v'; break;
  1737. case '\\':
  1738. case '\'':
  1739. case '?':
  1740. case '\"': break;
  1741. case '0': case '1': case '2': case '3': case '4': case '5': case '6': case '7':
  1742. {
  1743. c = next - '0';
  1744. if (*in >= '0' && *in <= '7')
  1745. {
  1746. c = c << 3 | (*in++-'0');
  1747. if (*in >= '0' && *in <= '7')
  1748. c = c << 3 | (*in++-'0');
  1749. }
  1750. break;
  1751. }
  1752. case 'x':
  1753. c = 0;
  1754. while (isxdigit(*in))
  1755. {
  1756. next = *in++;
  1757. c = c << 4;
  1758. if (next >= '0' && next <= '9') c |= (next - '0');
  1759. else if (next >= 'A' && next <= 'F') c |= (next - 'A' + 10);
  1760. else if (next >= 'a' && next <= 'f') c |= (next - 'a' + 10);
  1761. }
  1762. break;
  1763. default:
  1764. if (errorIfInvalid)
  1765. throw MakeStringException(1, "unrecognised character escape sequence '\\%c'", next);
  1766. in--; // keep it as is.
  1767. break;
  1768. }
  1769. }
  1770. }
  1771. out.append(c);
  1772. }
  1773. }
  1774. bool isPrintable(unsigned len, const char * src)
  1775. {
  1776. while (len--)
  1777. {
  1778. if (!isprint(*((unsigned char *)src)))
  1779. return false;
  1780. src++;
  1781. }
  1782. return true;
  1783. }
  1784. //make this as fast as possible...
  1785. StringBuffer & appendStringAsSQL(StringBuffer & out, unsigned len, const char * src)
  1786. {
  1787. if (!isPrintable(len, src))
  1788. {
  1789. out.append("X'");
  1790. appendDataAsHex(out, len, src);
  1791. return out.append('\'');
  1792. }
  1793. out.ensureCapacity(2 + len);
  1794. out.append('\'');
  1795. loop
  1796. {
  1797. char * next = (char *)memchr(src, '\'', len);
  1798. if (!next)
  1799. break;
  1800. unsigned chunk=(size32_t)(next-src)+1;
  1801. out.append(chunk, src).append('\'');
  1802. len -= chunk;
  1803. src += chunk;
  1804. }
  1805. return out.append(len, src).append('\'');
  1806. }
  1807. static const char * hexText = "0123456789ABCDEF";
  1808. StringBuffer & appendDataAsHex(StringBuffer &out, unsigned len, const void * data)
  1809. {
  1810. char * target = (char *)out.reserve(len*2);
  1811. unsigned char * start = (unsigned char *)data;
  1812. for (unsigned count=len; count> 0; --count)
  1813. {
  1814. unsigned next = *start++;
  1815. *target++ = hexText[next >>4];
  1816. *target++ = hexText[next & 15];
  1817. }
  1818. return out;
  1819. }
  1820. bool strToBool(size_t len, const char * text)
  1821. {
  1822. switch (len)
  1823. {
  1824. case 4:
  1825. if (memicmp(text, "true", 4) == 0)
  1826. return true;
  1827. break;
  1828. case 3:
  1829. if (memicmp(text, "yes", 3) == 0)
  1830. return true;
  1831. break;
  1832. case 2:
  1833. if (memicmp(text, "on", 2) == 0)
  1834. return true;
  1835. break;
  1836. case 1:
  1837. if ((memicmp(text, "t", 1) == 0) || (memicmp(text, "y", 1) == 0))
  1838. return true;
  1839. break;
  1840. }
  1841. while (len && isspace(*text))
  1842. {
  1843. len--;
  1844. text++;
  1845. }
  1846. while (len-- && isdigit(*text))
  1847. {
  1848. if (*text++ != '0') return true;
  1849. }
  1850. return false;
  1851. }
  1852. bool strToBool(const char * text)
  1853. {
  1854. return strToBool(strlen(text), text);
  1855. }
  1856. bool clipStrToBool(size_t len, const char * text)
  1857. {
  1858. while (len && *text==' ')
  1859. {
  1860. len--;
  1861. text++;
  1862. }
  1863. while (len && text[len-1]== ' ')
  1864. len--;
  1865. return strToBool(len, text);
  1866. }
  1867. bool clipStrToBool(const char * text)
  1868. {
  1869. return clipStrToBool(strlen(text), text);
  1870. }
  1871. StringBuffer & ncnameEscape(char const * in, StringBuffer & out)
  1872. {
  1873. if(!isalpha(*in))
  1874. {
  1875. out.appendf("_%02X", static_cast<unsigned char>(*in));
  1876. in++;
  1877. }
  1878. char const * finger = in;
  1879. while(*finger)
  1880. {
  1881. if(!isalnum(*finger))
  1882. {
  1883. if(finger>in)
  1884. out.append((size32_t)(finger-in), in);
  1885. out.appendf("_%02X", static_cast<unsigned char>(*finger));
  1886. in = ++finger;
  1887. }
  1888. else
  1889. {
  1890. finger++;
  1891. }
  1892. }
  1893. if(finger>in)
  1894. out.append((size32_t)(finger-in), in);
  1895. return out;
  1896. }
  1897. StringBuffer & ncnameUnescape(char const * in, StringBuffer & out)
  1898. {
  1899. char const * finger = in;
  1900. while(*finger)
  1901. {
  1902. if(*finger == '_')
  1903. {
  1904. if(finger>in)
  1905. out.append((size32_t)(finger-in), in);
  1906. unsigned char chr = 16 * hex2num(finger[1]) + hex2num(finger[2]);
  1907. out.append(static_cast<char>(chr));
  1908. in = (finger+=3);
  1909. }
  1910. else
  1911. {
  1912. finger++;
  1913. }
  1914. }
  1915. if(finger>in)
  1916. out.append((size32_t)(finger-in), in);
  1917. return out;
  1918. }
  1919. bool startsWith(const char* src, const char* dst)
  1920. {
  1921. while (*dst && *dst == *src) { src++; dst++; }
  1922. return *dst==0;
  1923. }
  1924. bool startsWithIgnoreCase(const char* src, const char* dst)
  1925. {
  1926. while (*dst && tolower(*dst) == tolower(*src)) { src++; dst++; }
  1927. return *dst==0;
  1928. }
  1929. bool endsWith(const char* src, const char* dst)
  1930. {
  1931. size_t srcLen = strlen(src);
  1932. size_t dstLen = strlen(dst);
  1933. if (dstLen<=srcLen)
  1934. return memcmp(dst, src+srcLen-dstLen, dstLen)==0;
  1935. return false;
  1936. }
  1937. bool endsWithIgnoreCase(const char* src, const char* dst)
  1938. {
  1939. size_t srcLen = strlen(src);
  1940. size_t dstLen = strlen(dst);
  1941. if (dstLen<=srcLen)
  1942. return memicmp(dst, src+srcLen-dstLen, dstLen)==0;
  1943. return false;
  1944. }
  1945. char *j_strtok_r(char *str, const char *delim, char **saveptr)
  1946. {
  1947. if (!str)
  1948. str = *saveptr;
  1949. char c;
  1950. loop {
  1951. c = *str;
  1952. if (!c) {
  1953. *saveptr = str;
  1954. return NULL;
  1955. }
  1956. if (!strchr(delim,c))
  1957. break;
  1958. str++;
  1959. }
  1960. char *ret=str;
  1961. do {
  1962. c = *(++str);
  1963. } while (c&&!strchr(delim,c));
  1964. if (c)
  1965. *(str++) = 0;
  1966. *saveptr = str;
  1967. return ret;
  1968. }
  1969. int j_memicmp (const void *s1, const void *s2, size32_t len)
  1970. {
  1971. const byte *b1 = (const byte *)s1;
  1972. const byte *b2 = (const byte *)s2;
  1973. int ret = 0;
  1974. while (len&&((ret = tolower(*b1)-tolower(*b2)) == 0)) {
  1975. b1++;
  1976. b2++;
  1977. len--;
  1978. }
  1979. return ret;
  1980. }
  1981. size32_t memcount(size32_t len, const char * str, char search)
  1982. {
  1983. size32_t count = 0;
  1984. for (size32_t i=0; i < len; i++)
  1985. {
  1986. if (str[i] == search)
  1987. count++;
  1988. }
  1989. return count;
  1990. }
  1991. StringBuffer & elideString(StringBuffer & s, unsigned maxLength)
  1992. {
  1993. if (s.length() > maxLength)
  1994. {
  1995. s.setLength(maxLength);
  1996. s.append("...");
  1997. }
  1998. return s;
  1999. }