NCDVal.c 63 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177
  1. /**
  2. * @file NCDVal.c
  3. * @author Ambroz Bizjak <ambrop7@gmail.com>
  4. *
  5. * @section LICENSE
  6. *
  7. * Redistribution and use in source and binary forms, with or without
  8. * modification, are permitted provided that the following conditions are met:
  9. * 1. Redistributions of source code must retain the above copyright
  10. * notice, this list of conditions and the following disclaimer.
  11. * 2. Redistributions in binary form must reproduce the above copyright
  12. * notice, this list of conditions and the following disclaimer in the
  13. * documentation and/or other materials provided with the distribution.
  14. * 3. Neither the name of the author nor the
  15. * names of its contributors may be used to endorse or promote products
  16. * derived from this software without specific prior written permission.
  17. *
  18. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  19. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  20. * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  21. * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
  22. * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  23. * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  24. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  25. * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  27. * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28. */
  29. #include <string.h>
  30. #include <limits.h>
  31. #include <stdlib.h>
  32. #include <stddef.h>
  33. #include <stdarg.h>
  34. #include <misc/balloc.h>
  35. #include <misc/strdup.h>
  36. #include <misc/offset.h>
  37. #include <structure/CAvl.h>
  38. #include <base/BLog.h>
  39. #include "NCDVal.h"
  40. #include <generated/blog_channel_NCDVal.h>
  41. #define NCDVAL_FIRST_SIZE 256
  42. #define NCDVAL_MAX_DEPTH 32
  43. #define TYPE_MASK_EXTERNAL_TYPE ((1 << 3) - 1)
  44. #define TYPE_MASK_INTERNAL_TYPE ((1 << 5) - 1)
  45. #define TYPE_SHIFT_DEPTH 5
  46. #define STOREDSTRING_TYPE (NCDVAL_STRING | (0 << 3))
  47. #define IDSTRING_TYPE (NCDVAL_STRING | (1 << 3))
  48. #define EXTERNALSTRING_TYPE (NCDVAL_STRING | (2 << 3))
  49. #define COMPOSEDSTRING_TYPE (NCDVAL_STRING | (3 << 3))
  50. #define NCDVAL_INSTR_PLACEHOLDER 0
  51. #define NCDVAL_INSTR_REINSERT 1
  52. #define NCDVAL_INSTR_BUMPDEPTH 2
  53. struct NCDVal__ref {
  54. NCDVal__idx next;
  55. BRefTarget *target;
  56. };
  57. struct NCDVal__string {
  58. int type;
  59. NCDVal__idx length;
  60. char data[];
  61. };
  62. struct NCDVal__list {
  63. int type;
  64. NCDVal__idx maxcount;
  65. NCDVal__idx count;
  66. NCDVal__idx elem_indices[];
  67. };
  68. struct NCDVal__mapelem {
  69. NCDVal__idx key_idx;
  70. NCDVal__idx val_idx;
  71. NCDVal__idx tree_child[2];
  72. NCDVal__idx tree_parent;
  73. int8_t tree_balance;
  74. };
  75. struct NCDVal__idstring {
  76. int type;
  77. NCD_string_id_t string_id;
  78. NCDStringIndex *string_index;
  79. };
  80. struct NCDVal__externalstring {
  81. int type;
  82. const char *data;
  83. size_t length;
  84. struct NCDVal__ref ref;
  85. };
  86. struct NCDVal__composedstring {
  87. int type;
  88. size_t offset;
  89. size_t length;
  90. void (*func_getptr) (void *, size_t, const char **, size_t *);
  91. void *user;
  92. struct NCDVal__ref ref;
  93. };
  94. struct NCDVal__cms_link {
  95. NCDVal__idx link_idx;
  96. NCDVal__idx next_cms_link;
  97. };
  98. typedef struct NCDVal__mapelem NCDVal__maptree_entry;
  99. typedef NCDValMem *NCDVal__maptree_arg;
  100. #include "NCDVal_maptree.h"
  101. #include <structure/CAvl_decl.h>
  102. struct NCDVal__map {
  103. int type;
  104. NCDVal__idx maxcount;
  105. NCDVal__idx count;
  106. NCDVal__MapTree tree;
  107. struct NCDVal__mapelem elems[];
  108. };
  109. struct NCDVal__instr {
  110. int type;
  111. union {
  112. struct {
  113. NCDVal__idx plid;
  114. NCDVal__idx plidx;
  115. } placeholder;
  116. struct {
  117. NCDVal__idx mapidx;
  118. NCDVal__idx elempos;
  119. } reinsert;
  120. struct {
  121. NCDVal__idx parent_idx;
  122. NCDVal__idx child_idx_idx;
  123. } bumpdepth;
  124. };
  125. };
  126. static int make_type (int internal_type, int depth)
  127. {
  128. ASSERT(internal_type == NCDVAL_LIST ||
  129. internal_type == NCDVAL_MAP ||
  130. internal_type == STOREDSTRING_TYPE ||
  131. internal_type == IDSTRING_TYPE ||
  132. internal_type == EXTERNALSTRING_TYPE ||
  133. internal_type == COMPOSEDSTRING_TYPE)
  134. ASSERT(depth >= 0)
  135. ASSERT(depth <= NCDVAL_MAX_DEPTH)
  136. return (internal_type | (depth << TYPE_SHIFT_DEPTH));
  137. }
  138. static int get_external_type (int type)
  139. {
  140. return (type & TYPE_MASK_EXTERNAL_TYPE);
  141. }
  142. static int get_internal_type (int type)
  143. {
  144. return (type & TYPE_MASK_INTERNAL_TYPE);
  145. }
  146. static int get_depth (int type)
  147. {
  148. return (type >> TYPE_SHIFT_DEPTH);
  149. }
  150. static int bump_depth (int *type_ptr, int elem_depth)
  151. {
  152. if (get_depth(*type_ptr) < elem_depth + 1) {
  153. if (elem_depth + 1 > NCDVAL_MAX_DEPTH) {
  154. return 0;
  155. }
  156. *type_ptr = make_type(get_internal_type(*type_ptr), elem_depth + 1);
  157. }
  158. return 1;
  159. }
  160. static void * NCDValMem__BufAt (NCDValMem *o, NCDVal__idx idx)
  161. {
  162. ASSERT(idx >= 0)
  163. ASSERT(idx < o->used)
  164. return (o->buf ? o->buf : o->fastbuf) + idx;
  165. }
  166. static NCDVal__idx NCDValMem__Alloc (NCDValMem *o, NCDVal__idx alloc_size, NCDVal__idx align)
  167. {
  168. NCDVal__idx mod = o->used % align;
  169. NCDVal__idx align_extra = mod ? (align - mod) : 0;
  170. if (alloc_size > NCDVAL_MAXIDX - align_extra) {
  171. return -1;
  172. }
  173. NCDVal__idx aligned_alloc_size = align_extra + alloc_size;
  174. if (aligned_alloc_size > o->size - o->used) {
  175. NCDVal__idx newsize = (o->buf ? o->size : NCDVAL_FIRST_SIZE);
  176. while (aligned_alloc_size > newsize - o->used) {
  177. if (newsize > NCDVAL_MAXIDX / 2) {
  178. return -1;
  179. }
  180. newsize *= 2;
  181. }
  182. char *newbuf;
  183. if (!o->buf) {
  184. newbuf = malloc(newsize);
  185. if (!newbuf) {
  186. return -1;
  187. }
  188. memcpy(newbuf, o->fastbuf, o->used);
  189. } else {
  190. newbuf = realloc(o->buf, newsize);
  191. if (!newbuf) {
  192. return -1;
  193. }
  194. }
  195. o->buf = newbuf;
  196. o->size = newsize;
  197. }
  198. NCDVal__idx idx = o->used + align_extra;
  199. o->used += aligned_alloc_size;
  200. return idx;
  201. }
  202. static NCDValRef NCDVal__Ref (NCDValMem *mem, NCDVal__idx idx)
  203. {
  204. ASSERT(idx == -1 || mem)
  205. NCDValRef ref = {mem, idx};
  206. return ref;
  207. }
  208. static void NCDVal__AssertMem (NCDValMem *mem)
  209. {
  210. ASSERT(mem)
  211. ASSERT(mem->size >= 0)
  212. ASSERT(mem->used >= 0)
  213. ASSERT(mem->used <= mem->size)
  214. ASSERT(mem->buf || mem->size == NCDVAL_FASTBUF_SIZE)
  215. ASSERT(!mem->buf || mem->size >= NCDVAL_FIRST_SIZE)
  216. }
  217. static void NCDVal_AssertExternal (NCDValMem *mem, const void *e_buf, size_t e_len)
  218. {
  219. #ifndef NDEBUG
  220. const char *e_cbuf = e_buf;
  221. char *buf = (mem->buf ? mem->buf : mem->fastbuf);
  222. ASSERT(e_cbuf >= buf + mem->size || e_cbuf + e_len <= buf)
  223. #endif
  224. }
  225. static void NCDVal__AssertValOnly (NCDValMem *mem, NCDVal__idx idx)
  226. {
  227. // placeholders
  228. if (idx < -1) {
  229. return;
  230. }
  231. ASSERT(idx >= 0)
  232. ASSERT(idx + sizeof(int) <= mem->used)
  233. #ifndef NDEBUG
  234. int *type_ptr = NCDValMem__BufAt(mem, idx);
  235. ASSERT(get_depth(*type_ptr) >= 0)
  236. ASSERT(get_depth(*type_ptr) <= NCDVAL_MAX_DEPTH)
  237. switch (get_internal_type(*type_ptr)) {
  238. case STOREDSTRING_TYPE: {
  239. ASSERT(idx + sizeof(struct NCDVal__string) <= mem->used)
  240. struct NCDVal__string *str_e = NCDValMem__BufAt(mem, idx);
  241. ASSERT(str_e->length >= 0)
  242. ASSERT(idx + sizeof(struct NCDVal__string) + str_e->length + 1 <= mem->used)
  243. } break;
  244. case NCDVAL_LIST: {
  245. ASSERT(idx + sizeof(struct NCDVal__list) <= mem->used)
  246. struct NCDVal__list *list_e = NCDValMem__BufAt(mem, idx);
  247. ASSERT(list_e->maxcount >= 0)
  248. ASSERT(list_e->count >= 0)
  249. ASSERT(list_e->count <= list_e->maxcount)
  250. ASSERT(idx + sizeof(struct NCDVal__list) + list_e->maxcount * sizeof(NCDVal__idx) <= mem->used)
  251. } break;
  252. case NCDVAL_MAP: {
  253. ASSERT(idx + sizeof(struct NCDVal__map) <= mem->used)
  254. struct NCDVal__map *map_e = NCDValMem__BufAt(mem, idx);
  255. ASSERT(map_e->maxcount >= 0)
  256. ASSERT(map_e->count >= 0)
  257. ASSERT(map_e->count <= map_e->maxcount)
  258. ASSERT(idx + sizeof(struct NCDVal__map) + map_e->maxcount * sizeof(struct NCDVal__mapelem) <= mem->used)
  259. } break;
  260. case IDSTRING_TYPE: {
  261. ASSERT(idx + sizeof(struct NCDVal__idstring) <= mem->used)
  262. struct NCDVal__idstring *ids_e = NCDValMem__BufAt(mem, idx);
  263. ASSERT(ids_e->string_id >= 0)
  264. ASSERT(ids_e->string_index)
  265. } break;
  266. case EXTERNALSTRING_TYPE: {
  267. ASSERT(idx + sizeof(struct NCDVal__externalstring) <= mem->used)
  268. struct NCDVal__externalstring *exs_e = NCDValMem__BufAt(mem, idx);
  269. ASSERT(exs_e->data)
  270. ASSERT(!exs_e->ref.target || exs_e->ref.next >= -1)
  271. ASSERT(!exs_e->ref.target || exs_e->ref.next < mem->used)
  272. } break;
  273. case COMPOSEDSTRING_TYPE: {
  274. ASSERT(idx + sizeof(struct NCDVal__composedstring) <= mem->used)
  275. struct NCDVal__composedstring *cms_e = NCDValMem__BufAt(mem, idx);
  276. ASSERT(cms_e->func_getptr)
  277. ASSERT(!cms_e->ref.target || cms_e->ref.next >= -1)
  278. ASSERT(!cms_e->ref.target || cms_e->ref.next < mem->used)
  279. } break;
  280. default: ASSERT(0);
  281. }
  282. #endif
  283. }
  284. static void NCDVal__AssertVal (NCDValRef val)
  285. {
  286. NCDVal__AssertMem(val.mem);
  287. NCDVal__AssertValOnly(val.mem, val.idx);
  288. }
  289. static NCDValMapElem NCDVal__MapElem (NCDVal__idx elemidx)
  290. {
  291. ASSERT(elemidx >= 0 || elemidx == -1)
  292. NCDValMapElem me = {elemidx};
  293. return me;
  294. }
  295. static void NCDVal__MapAssertElemOnly (NCDValRef map, NCDVal__idx elemidx)
  296. {
  297. #ifndef NDEBUG
  298. struct NCDVal__map *map_e = NCDValMem__BufAt(map.mem, map.idx);
  299. ASSERT(elemidx >= map.idx + offsetof(struct NCDVal__map, elems))
  300. ASSERT(elemidx < map.idx + offsetof(struct NCDVal__map, elems) + map_e->count * sizeof(struct NCDVal__mapelem))
  301. struct NCDVal__mapelem *me_e = NCDValMem__BufAt(map.mem, elemidx);
  302. NCDVal__AssertValOnly(map.mem, me_e->key_idx);
  303. NCDVal__AssertValOnly(map.mem, me_e->val_idx);
  304. #endif
  305. }
  306. static void NCDVal__MapAssertElem (NCDValRef map, NCDValMapElem me)
  307. {
  308. ASSERT(NCDVal_IsMap(map))
  309. NCDVal__MapAssertElemOnly(map, me.elemidx);
  310. }
  311. static NCDVal__idx NCDVal__MapElemIdx (NCDVal__idx mapidx, NCDVal__idx pos)
  312. {
  313. return mapidx + offsetof(struct NCDVal__map, elems) + pos * sizeof(struct NCDVal__mapelem);
  314. }
  315. static int NCDVal__Depth (NCDValRef val)
  316. {
  317. ASSERT(val.idx != -1)
  318. // handle placeholders
  319. if (val.idx < 0) {
  320. return 0;
  321. }
  322. int *elem_type_ptr = NCDValMem__BufAt(val.mem, val.idx);
  323. int depth = get_depth(*elem_type_ptr);
  324. ASSERT(depth >= 0)
  325. ASSERT(depth <= NCDVAL_MAX_DEPTH)
  326. return depth;
  327. }
  328. static int NCDValMem__NeedRegisterLink (NCDValMem *mem, NCDVal__idx val_idx)
  329. {
  330. NCDVal__AssertValOnly(mem, val_idx);
  331. return !(val_idx < -1) && get_internal_type(*(int *)NCDValMem__BufAt(mem, val_idx)) == COMPOSEDSTRING_TYPE;
  332. }
  333. static int NCDValMem__RegisterLink (NCDValMem *mem, NCDVal__idx val_idx, NCDVal__idx link_idx)
  334. {
  335. NCDVal__AssertValOnly(mem, val_idx);
  336. ASSERT(NCDValMem__NeedRegisterLink(mem, val_idx))
  337. NCDVal__idx cms_link_idx = NCDValMem__Alloc(mem, sizeof(struct NCDVal__cms_link), __alignof(struct NCDVal__cms_link));
  338. if (cms_link_idx < 0) {
  339. return 0;
  340. }
  341. struct NCDVal__cms_link *cms_link = NCDValMem__BufAt(mem, cms_link_idx);
  342. cms_link->link_idx = link_idx;
  343. cms_link->next_cms_link = mem->first_cms_link;
  344. mem->first_cms_link = cms_link_idx;
  345. return 1;
  346. }
  347. static void NCDValMem__PopLastRegisteredLink (NCDValMem *mem)
  348. {
  349. ASSERT(mem->first_cms_link != -1)
  350. struct NCDVal__cms_link *cms_link = NCDValMem__BufAt(mem, mem->first_cms_link);
  351. mem->first_cms_link = cms_link->next_cms_link;
  352. }
  353. static void NCDValMem__RegisterRef (NCDValMem *o, NCDVal__idx refidx, struct NCDVal__ref *ref)
  354. {
  355. ASSERT(ref == NCDValMem__BufAt(o, refidx))
  356. ASSERT(ref->target)
  357. ref->next = o->first_ref;
  358. o->first_ref = refidx;
  359. }
  360. static NCDValRef NCDVal__CopyComposedStringToStored (NCDValRef val)
  361. {
  362. ASSERT(NCDVal_IsComposedString(val))
  363. struct NCDVal__composedstring cms_e = *(struct NCDVal__composedstring *)NCDValMem__BufAt(val.mem, val.idx);
  364. NCDValRef copy = NCDVal_NewStringUninitialized(val.mem, cms_e.length);
  365. if (NCDVal_IsInvalid(copy)) {
  366. return NCDVal_NewInvalid();
  367. }
  368. char *copy_data = (char *)NCDVal_StringData(copy);
  369. size_t pos = 0;
  370. while (pos < cms_e.length) {
  371. const char *chunk_data;
  372. size_t chunk_len;
  373. cms_e.func_getptr(cms_e.user, cms_e.offset + pos, &chunk_data, &chunk_len);
  374. ASSERT(chunk_data)
  375. ASSERT(chunk_len > 0)
  376. if (chunk_len > cms_e.length - pos) {
  377. chunk_len = cms_e.length - pos;
  378. }
  379. memcpy(copy_data + pos, chunk_data, chunk_len);
  380. pos += chunk_len;
  381. }
  382. return copy;
  383. }
  384. static const char * NCDVal__composedstring_cstring_func (const b_cstring *cstr, size_t offset, size_t *out_length)
  385. {
  386. ASSERT(offset < cstr->length)
  387. ASSERT(out_length)
  388. ASSERT(cstr->func == NCDVal__composedstring_cstring_func)
  389. size_t str_offset = cstr->user1.size;
  390. NCDVal_ComposedString_func_getptr func_getptr = (NCDVal_ComposedString_func_getptr)cstr->user2.fptr;
  391. void *user = cstr->user3.ptr;
  392. const char *data;
  393. func_getptr(user, str_offset + offset, &data, out_length);
  394. ASSERT(data)
  395. ASSERT(*out_length > 0)
  396. return data;
  397. }
  398. #include "NCDVal_maptree.h"
  399. #include <structure/CAvl_impl.h>
  400. void NCDValMem_Init (NCDValMem *o)
  401. {
  402. o->buf = NULL;
  403. o->size = NCDVAL_FASTBUF_SIZE;
  404. o->used = 0;
  405. o->first_ref = -1;
  406. o->first_cms_link = -1;
  407. }
  408. void NCDValMem_Free (NCDValMem *o)
  409. {
  410. NCDVal__AssertMem(o);
  411. NCDVal__idx refidx = o->first_ref;
  412. while (refidx != -1) {
  413. struct NCDVal__ref *ref = NCDValMem__BufAt(o, refidx);
  414. ASSERT(ref->target)
  415. BRefTarget_Deref(ref->target);
  416. refidx = ref->next;
  417. }
  418. if (o->buf) {
  419. BFree(o->buf);
  420. }
  421. }
  422. int NCDValMem_InitCopy (NCDValMem *o, NCDValMem *other)
  423. {
  424. NCDVal__AssertMem(other);
  425. o->size = other->size;
  426. o->used = other->used;
  427. o->first_ref = other->first_ref;
  428. o->first_cms_link = other->first_cms_link;
  429. if (!other->buf) {
  430. o->buf = NULL;
  431. memcpy(o->fastbuf, other->fastbuf, other->used);
  432. } else {
  433. o->buf = BAlloc(other->size);
  434. if (!o->buf) {
  435. goto fail0;
  436. }
  437. memcpy(o->buf, other->buf, other->used);
  438. }
  439. NCDVal__idx refidx = o->first_ref;
  440. while (refidx != -1) {
  441. struct NCDVal__ref *ref = NCDValMem__BufAt(o, refidx);
  442. ASSERT(ref->target)
  443. if (!BRefTarget_Ref(ref->target)) {
  444. goto fail1;
  445. }
  446. refidx = ref->next;
  447. }
  448. return 1;
  449. fail1:;
  450. NCDVal__idx undo_refidx = o->first_ref;
  451. while (undo_refidx != refidx) {
  452. struct NCDVal__ref *ref = NCDValMem__BufAt(o, undo_refidx);
  453. BRefTarget_Deref(ref->target);
  454. undo_refidx = ref->next;
  455. }
  456. if (o->buf) {
  457. BFree(o->buf);
  458. }
  459. fail0:
  460. return 0;
  461. }
  462. int NCDValMem_ConvertNonContinuousStrings (NCDValMem *o, NCDValRef *root_val)
  463. {
  464. NCDVal__AssertMem(o);
  465. ASSERT(root_val)
  466. ASSERT(root_val->mem == o)
  467. NCDVal__AssertValOnly(o, root_val->idx);
  468. while (o->first_cms_link != -1) {
  469. struct NCDVal__cms_link cms_link = *(struct NCDVal__cms_link *)NCDValMem__BufAt(o, o->first_cms_link);
  470. NCDVal__idx val_idx = *(NCDVal__idx *)NCDValMem__BufAt(o, cms_link.link_idx);
  471. NCDValRef val = NCDVal__Ref(o, val_idx);
  472. ASSERT(NCDVal_IsComposedString(val))
  473. NCDValRef copy = NCDVal__CopyComposedStringToStored(val);
  474. if (NCDVal_IsInvalid(copy)) {
  475. return 0;
  476. }
  477. *(int *)NCDValMem__BufAt(o, cms_link.link_idx) = copy.idx;
  478. o->first_cms_link = cms_link.next_cms_link;
  479. }
  480. if (NCDVal_IsComposedString(*root_val)) {
  481. NCDValRef copy = NCDVal__CopyComposedStringToStored(*root_val);
  482. if (NCDVal_IsInvalid(copy)) {
  483. return 0;
  484. }
  485. *root_val = copy;
  486. }
  487. return 1;
  488. }
  489. void NCDVal_Assert (NCDValRef val)
  490. {
  491. ASSERT(val.idx == -1 || (NCDVal__AssertVal(val), 1))
  492. }
  493. int NCDVal_IsInvalid (NCDValRef val)
  494. {
  495. NCDVal_Assert(val);
  496. return (val.idx == -1);
  497. }
  498. int NCDVal_IsPlaceholder (NCDValRef val)
  499. {
  500. NCDVal_Assert(val);
  501. return (val.idx < -1);
  502. }
  503. int NCDVal_Type (NCDValRef val)
  504. {
  505. NCDVal__AssertVal(val);
  506. if (val.idx < -1) {
  507. return NCDVAL_PLACEHOLDER;
  508. }
  509. int *type_ptr = NCDValMem__BufAt(val.mem, val.idx);
  510. return get_external_type(*type_ptr);
  511. }
  512. NCDValRef NCDVal_NewInvalid (void)
  513. {
  514. NCDValRef ref = {NULL, -1};
  515. return ref;
  516. }
  517. NCDValRef NCDVal_NewPlaceholder (NCDValMem *mem, int plid)
  518. {
  519. NCDVal__AssertMem(mem);
  520. ASSERT(plid >= 0)
  521. ASSERT(plid < NCDVAL_TOPPLID)
  522. NCDValRef ref = {mem, NCDVAL_MINIDX + plid};
  523. return ref;
  524. }
  525. int NCDVal_PlaceholderId (NCDValRef val)
  526. {
  527. ASSERT(NCDVal_IsPlaceholder(val))
  528. return (val.idx - NCDVAL_MINIDX);
  529. }
  530. NCDValRef NCDVal_NewCopy (NCDValMem *mem, NCDValRef val)
  531. {
  532. NCDVal__AssertMem(mem);
  533. NCDVal__AssertVal(val);
  534. if (val.idx < -1) {
  535. return NCDVal_NewPlaceholder(mem, NCDVal_PlaceholderId(val));
  536. }
  537. void *ptr = NCDValMem__BufAt(val.mem, val.idx);
  538. switch (get_internal_type(*(int *)ptr)) {
  539. case STOREDSTRING_TYPE: {
  540. struct NCDVal__string *str_e = ptr;
  541. NCDVal__idx size = sizeof(struct NCDVal__string) + str_e->length + 1;
  542. NCDVal__idx idx = NCDValMem__Alloc(mem, size, __alignof(struct NCDVal__string));
  543. if (idx < 0) {
  544. goto fail;
  545. }
  546. str_e = NCDValMem__BufAt(val.mem, val.idx);
  547. struct NCDVal__string *new_str_e = NCDValMem__BufAt(mem, idx);
  548. memcpy(new_str_e, str_e, size);
  549. return NCDVal__Ref(mem, idx);
  550. } break;
  551. case NCDVAL_LIST: {
  552. struct NCDVal__list *list_e = ptr;
  553. NCDVal__idx size = sizeof(struct NCDVal__list) + list_e->maxcount * sizeof(NCDVal__idx);
  554. NCDVal__idx idx = NCDValMem__Alloc(mem, size, __alignof(struct NCDVal__list));
  555. if (idx < 0) {
  556. goto fail;
  557. }
  558. list_e = NCDValMem__BufAt(val.mem, val.idx);
  559. struct NCDVal__list *new_list_e = NCDValMem__BufAt(mem, idx);
  560. *new_list_e = *list_e;
  561. NCDVal__idx count = list_e->count;
  562. for (NCDVal__idx i = 0; i < count; i++) {
  563. NCDValRef elem_copy = NCDVal_NewCopy(mem, NCDVal__Ref(val.mem, list_e->elem_indices[i]));
  564. if (NCDVal_IsInvalid(elem_copy)) {
  565. goto fail;
  566. }
  567. if (NCDValMem__NeedRegisterLink(mem, elem_copy.idx)) {
  568. if (!NCDValMem__RegisterLink(mem, elem_copy.idx, idx + offsetof(struct NCDVal__list, elem_indices) + i * sizeof(NCDVal__idx))) {
  569. goto fail;
  570. }
  571. }
  572. list_e = NCDValMem__BufAt(val.mem, val.idx);
  573. new_list_e = NCDValMem__BufAt(mem, idx);
  574. new_list_e->elem_indices[i] = elem_copy.idx;
  575. }
  576. return NCDVal__Ref(mem, idx);
  577. } break;
  578. case NCDVAL_MAP: {
  579. size_t count = NCDVal_MapCount(val);
  580. NCDValRef copy = NCDVal_NewMap(mem, count);
  581. if (NCDVal_IsInvalid(copy)) {
  582. goto fail;
  583. }
  584. for (NCDValMapElem e = NCDVal_MapFirst(val); !NCDVal_MapElemInvalid(e); e = NCDVal_MapNext(val, e)) {
  585. NCDValRef key_copy = NCDVal_NewCopy(mem, NCDVal_MapElemKey(val, e));
  586. NCDValRef val_copy = NCDVal_NewCopy(mem, NCDVal_MapElemVal(val, e));
  587. if (NCDVal_IsInvalid(key_copy) || NCDVal_IsInvalid(val_copy)) {
  588. goto fail;
  589. }
  590. int inserted;
  591. if (!NCDVal_MapInsert(copy, key_copy, val_copy, &inserted)) {
  592. goto fail;
  593. }
  594. ASSERT_EXECUTE(inserted)
  595. }
  596. return copy;
  597. } break;
  598. case IDSTRING_TYPE: {
  599. NCDVal__idx size = sizeof(struct NCDVal__idstring);
  600. NCDVal__idx idx = NCDValMem__Alloc(mem, size, __alignof(struct NCDVal__idstring));
  601. if (idx < 0) {
  602. goto fail;
  603. }
  604. struct NCDVal__idstring *ids_e = NCDValMem__BufAt(val.mem, val.idx);
  605. struct NCDVal__idstring *new_ids_e = NCDValMem__BufAt(mem, idx);
  606. *new_ids_e = *ids_e;
  607. return NCDVal__Ref(mem, idx);
  608. } break;
  609. case EXTERNALSTRING_TYPE: {
  610. struct NCDVal__externalstring *exs_e = ptr;
  611. return NCDVal_NewExternalString(mem, exs_e->data, exs_e->length, exs_e->ref.target);
  612. } break;
  613. case COMPOSEDSTRING_TYPE: {
  614. struct NCDVal__composedstring *cms_e = ptr;
  615. NCDValComposedStringResource resource;
  616. resource.func_getptr = cms_e->func_getptr;
  617. resource.user = cms_e->user;
  618. resource.ref_target = cms_e->ref.target;
  619. return NCDVal_NewComposedString(mem, resource, cms_e->offset, cms_e->length);
  620. } break;
  621. default: ASSERT(0);
  622. }
  623. ASSERT(0);
  624. fail:
  625. return NCDVal_NewInvalid();
  626. }
  627. int NCDVal_Compare (NCDValRef val1, NCDValRef val2)
  628. {
  629. NCDVal__AssertVal(val1);
  630. NCDVal__AssertVal(val2);
  631. int type1 = NCDVal_Type(val1);
  632. int type2 = NCDVal_Type(val2);
  633. if (type1 != type2) {
  634. return (type1 > type2) - (type1 < type2);
  635. }
  636. switch (type1) {
  637. case NCDVAL_STRING: {
  638. size_t len1 = NCDVal_StringLength(val1);
  639. size_t len2 = NCDVal_StringLength(val2);
  640. size_t min_len = len1 < len2 ? len1 : len2;
  641. int cmp = NCDVal_StringMemCmp(val1, val2, 0, 0, min_len);
  642. if (cmp) {
  643. return (cmp > 0) - (cmp < 0);
  644. }
  645. return (len1 > len2) - (len1 < len2);
  646. } break;
  647. case NCDVAL_LIST: {
  648. size_t count1 = NCDVal_ListCount(val1);
  649. size_t count2 = NCDVal_ListCount(val2);
  650. size_t min_count = count1 < count2 ? count1 : count2;
  651. for (size_t i = 0; i < min_count; i++) {
  652. NCDValRef ev1 = NCDVal_ListGet(val1, i);
  653. NCDValRef ev2 = NCDVal_ListGet(val2, i);
  654. int cmp = NCDVal_Compare(ev1, ev2);
  655. if (cmp) {
  656. return cmp;
  657. }
  658. }
  659. return (count1 > count2) - (count1 < count2);
  660. } break;
  661. case NCDVAL_MAP: {
  662. NCDValMapElem e1 = NCDVal_MapOrderedFirst(val1);
  663. NCDValMapElem e2 = NCDVal_MapOrderedFirst(val2);
  664. while (1) {
  665. int inv1 = NCDVal_MapElemInvalid(e1);
  666. int inv2 = NCDVal_MapElemInvalid(e2);
  667. if (inv1 || inv2) {
  668. return inv2 - inv1;
  669. }
  670. NCDValRef key1 = NCDVal_MapElemKey(val1, e1);
  671. NCDValRef key2 = NCDVal_MapElemKey(val2, e2);
  672. int cmp = NCDVal_Compare(key1, key2);
  673. if (cmp) {
  674. return cmp;
  675. }
  676. NCDValRef value1 = NCDVal_MapElemVal(val1, e1);
  677. NCDValRef value2 = NCDVal_MapElemVal(val2, e2);
  678. cmp = NCDVal_Compare(value1, value2);
  679. if (cmp) {
  680. return cmp;
  681. }
  682. e1 = NCDVal_MapOrderedNext(val1, e1);
  683. e2 = NCDVal_MapOrderedNext(val2, e2);
  684. }
  685. } break;
  686. case NCDVAL_PLACEHOLDER: {
  687. int plid1 = NCDVal_PlaceholderId(val1);
  688. int plid2 = NCDVal_PlaceholderId(val2);
  689. return (plid1 > plid2) - (plid1 < plid2);
  690. } break;
  691. default:
  692. ASSERT(0);
  693. return 0;
  694. }
  695. }
  696. NCDValSafeRef NCDVal_ToSafe (NCDValRef val)
  697. {
  698. NCDVal_Assert(val);
  699. NCDValSafeRef sval = {val.idx};
  700. return sval;
  701. }
  702. NCDValRef NCDVal_FromSafe (NCDValMem *mem, NCDValSafeRef sval)
  703. {
  704. NCDVal__AssertMem(mem);
  705. ASSERT(sval.idx == -1 || (NCDVal__AssertValOnly(mem, sval.idx), 1))
  706. NCDValRef val = {mem, sval.idx};
  707. return val;
  708. }
  709. NCDValRef NCDVal_Moved (NCDValMem *mem, NCDValRef val)
  710. {
  711. NCDVal__AssertMem(mem);
  712. ASSERT(val.idx == -1 || (NCDVal__AssertValOnly(mem, val.idx), 1))
  713. NCDValRef val2 = {mem, val.idx};
  714. return val2;
  715. }
  716. int NCDVal_IsSafeRefPlaceholder (NCDValSafeRef sval)
  717. {
  718. return (sval.idx < -1);
  719. }
  720. int NCDVal_GetSafeRefPlaceholderId (NCDValSafeRef sval)
  721. {
  722. ASSERT(NCDVal_IsSafeRefPlaceholder(sval))
  723. return (sval.idx - NCDVAL_MINIDX);
  724. }
  725. int NCDVal_HasOnlyContinuousStrings (NCDValRef val)
  726. {
  727. NCDVal__AssertVal(val);
  728. switch (NCDVal_Type(val)) {
  729. case NCDVAL_STRING: {
  730. if (!NCDVal_IsContinuousString(val)) {
  731. return 0;
  732. }
  733. } break;
  734. case NCDVAL_LIST: {
  735. size_t count = NCDVal_ListCount(val);
  736. for (size_t i = 0; i < count; i++) {
  737. NCDValRef elem = NCDVal_ListGet(val, i);
  738. if (!NCDVal_HasOnlyContinuousStrings(elem)) {
  739. return 0;
  740. }
  741. }
  742. } break;
  743. case NCDVAL_MAP: {
  744. for (NCDValMapElem me = NCDVal_MapFirst(val); !NCDVal_MapElemInvalid(me); me = NCDVal_MapNext(val, me)) {
  745. NCDValRef e_key = NCDVal_MapElemKey(val, me);
  746. NCDValRef e_val = NCDVal_MapElemVal(val, me);
  747. if (!NCDVal_HasOnlyContinuousStrings(e_key) || !NCDVal_HasOnlyContinuousStrings(e_val)) {
  748. return 0;
  749. }
  750. }
  751. } break;
  752. case NCDVAL_PLACEHOLDER: {
  753. } break;
  754. default:
  755. ASSERT(0);
  756. }
  757. return 1;
  758. }
  759. int NCDVal_IsString (NCDValRef val)
  760. {
  761. NCDVal__AssertVal(val);
  762. return NCDVal_Type(val) == NCDVAL_STRING;
  763. }
  764. int NCDVal_IsContinuousString (NCDValRef val)
  765. {
  766. NCDVal__AssertVal(val);
  767. if (val.idx < -1) {
  768. return 0;
  769. }
  770. switch (get_internal_type(*(int *)NCDValMem__BufAt(val.mem, val.idx))) {
  771. case STOREDSTRING_TYPE:
  772. case IDSTRING_TYPE:
  773. case EXTERNALSTRING_TYPE:
  774. return 1;
  775. default:
  776. return 0;
  777. }
  778. }
  779. int NCDVal_IsStoredString (NCDValRef val)
  780. {
  781. NCDVal__AssertVal(val);
  782. return !(val.idx < -1) && get_internal_type(*(int *)NCDValMem__BufAt(val.mem, val.idx)) == STOREDSTRING_TYPE;
  783. }
  784. int NCDVal_IsIdString (NCDValRef val)
  785. {
  786. NCDVal__AssertVal(val);
  787. return !(val.idx < -1) && get_internal_type(*(int *)NCDValMem__BufAt(val.mem, val.idx)) == IDSTRING_TYPE;
  788. }
  789. int NCDVal_IsExternalString (NCDValRef val)
  790. {
  791. NCDVal__AssertVal(val);
  792. return !(val.idx < -1) && get_internal_type(*(int *)NCDValMem__BufAt(val.mem, val.idx)) == EXTERNALSTRING_TYPE;
  793. }
  794. int NCDVal_IsComposedString (NCDValRef val)
  795. {
  796. NCDVal__AssertVal(val);
  797. return !(val.idx < -1) && get_internal_type(*(int *)NCDValMem__BufAt(val.mem, val.idx)) == COMPOSEDSTRING_TYPE;
  798. }
  799. int NCDVal_IsStringNoNulls (NCDValRef val)
  800. {
  801. NCDVal__AssertVal(val);
  802. return NCDVal_Type(val) == NCDVAL_STRING && !NCDVal_StringHasNulls(val);
  803. }
  804. NCDValRef NCDVal_NewString (NCDValMem *mem, const char *data)
  805. {
  806. NCDVal__AssertMem(mem);
  807. ASSERT(data)
  808. NCDVal_AssertExternal(mem, data, strlen(data));
  809. return NCDVal_NewStringBin(mem, (const uint8_t *)data, strlen(data));
  810. }
  811. NCDValRef NCDVal_NewStringBin (NCDValMem *mem, const uint8_t *data, size_t len)
  812. {
  813. NCDVal__AssertMem(mem);
  814. ASSERT(len == 0 || data)
  815. NCDVal_AssertExternal(mem, data, len);
  816. if (len > NCDVAL_MAXIDX - sizeof(struct NCDVal__string) - 1) {
  817. goto fail;
  818. }
  819. NCDVal__idx size = sizeof(struct NCDVal__string) + len + 1;
  820. NCDVal__idx idx = NCDValMem__Alloc(mem, size, __alignof(struct NCDVal__string));
  821. if (idx < 0) {
  822. goto fail;
  823. }
  824. struct NCDVal__string *str_e = NCDValMem__BufAt(mem, idx);
  825. str_e->type = make_type(STOREDSTRING_TYPE, 0);
  826. str_e->length = len;
  827. if (len > 0) {
  828. memcpy(str_e->data, data, len);
  829. }
  830. str_e->data[len] = '\0';
  831. return NCDVal__Ref(mem, idx);
  832. fail:
  833. return NCDVal_NewInvalid();
  834. }
  835. NCDValRef NCDVal_NewStringUninitialized (NCDValMem *mem, size_t len)
  836. {
  837. NCDVal__AssertMem(mem);
  838. if (len > NCDVAL_MAXIDX - sizeof(struct NCDVal__string) - 1) {
  839. goto fail;
  840. }
  841. NCDVal__idx size = sizeof(struct NCDVal__string) + len + 1;
  842. NCDVal__idx idx = NCDValMem__Alloc(mem, size, __alignof(struct NCDVal__string));
  843. if (idx < 0) {
  844. goto fail;
  845. }
  846. struct NCDVal__string *str_e = NCDValMem__BufAt(mem, idx);
  847. str_e->type = make_type(STOREDSTRING_TYPE, 0);
  848. str_e->length = len;
  849. str_e->data[len] = '\0';
  850. return NCDVal__Ref(mem, idx);
  851. fail:
  852. return NCDVal_NewInvalid();
  853. }
  854. NCDValRef NCDVal_NewIdString (NCDValMem *mem, NCD_string_id_t string_id, NCDStringIndex *string_index)
  855. {
  856. NCDVal__AssertMem(mem);
  857. ASSERT(string_id >= 0)
  858. ASSERT(string_index)
  859. NCDVal__idx size = sizeof(struct NCDVal__idstring);
  860. NCDVal__idx idx = NCDValMem__Alloc(mem, size, __alignof(struct NCDVal__idstring));
  861. if (idx < 0) {
  862. goto fail;
  863. }
  864. struct NCDVal__idstring *ids_e = NCDValMem__BufAt(mem, idx);
  865. ids_e->type = make_type(IDSTRING_TYPE, 0);
  866. ids_e->string_id = string_id;
  867. ids_e->string_index = string_index;
  868. return NCDVal__Ref(mem, idx);
  869. fail:
  870. return NCDVal_NewInvalid();
  871. }
  872. NCDValRef NCDVal_NewExternalString (NCDValMem *mem, const char *data, size_t len,
  873. BRefTarget *ref_target)
  874. {
  875. NCDVal__AssertMem(mem);
  876. ASSERT(data)
  877. NCDVal_AssertExternal(mem, data, len);
  878. NCDVal__idx size = sizeof(struct NCDVal__externalstring);
  879. NCDVal__idx idx = NCDValMem__Alloc(mem, size, __alignof(struct NCDVal__externalstring));
  880. if (idx < 0) {
  881. goto fail;
  882. }
  883. if (ref_target) {
  884. if (!BRefTarget_Ref(ref_target)) {
  885. goto fail;
  886. }
  887. }
  888. struct NCDVal__externalstring *exs_e = NCDValMem__BufAt(mem, idx);
  889. exs_e->type = make_type(EXTERNALSTRING_TYPE, 0);
  890. exs_e->data = data;
  891. exs_e->length = len;
  892. exs_e->ref.target = ref_target;
  893. if (ref_target) {
  894. NCDValMem__RegisterRef(mem, idx + offsetof(struct NCDVal__externalstring, ref), &exs_e->ref);
  895. }
  896. return NCDVal__Ref(mem, idx);
  897. fail:
  898. return NCDVal_NewInvalid();
  899. }
  900. NCDValRef NCDVal_NewComposedString (NCDValMem *mem, NCDValComposedStringResource resource, size_t offset, size_t length)
  901. {
  902. NCDVal__AssertMem(mem);
  903. ASSERT(resource.func_getptr)
  904. NCDVal__idx size = sizeof(struct NCDVal__composedstring);
  905. NCDVal__idx idx = NCDValMem__Alloc(mem, size, __alignof(struct NCDVal__composedstring));
  906. if (idx < 0) {
  907. goto fail;
  908. }
  909. if (resource.ref_target) {
  910. if (!BRefTarget_Ref(resource.ref_target)) {
  911. goto fail;
  912. }
  913. }
  914. struct NCDVal__composedstring *cms_e = NCDValMem__BufAt(mem, idx);
  915. cms_e->type = make_type(COMPOSEDSTRING_TYPE, 0);
  916. cms_e->offset = offset;
  917. cms_e->length = length;
  918. cms_e->func_getptr = resource.func_getptr;
  919. cms_e->user = resource.user;
  920. cms_e->ref.target = resource.ref_target;
  921. if (resource.ref_target) {
  922. NCDValMem__RegisterRef(mem, idx + offsetof(struct NCDVal__composedstring, ref), &cms_e->ref);
  923. }
  924. return NCDVal__Ref(mem, idx);
  925. fail:
  926. return NCDVal_NewInvalid();
  927. }
  928. const char * NCDVal_StringData (NCDValRef contstring)
  929. {
  930. ASSERT(NCDVal_IsContinuousString(contstring))
  931. void *ptr = NCDValMem__BufAt(contstring.mem, contstring.idx);
  932. switch (get_internal_type(*(int *)ptr)) {
  933. case STOREDSTRING_TYPE: {
  934. struct NCDVal__string *str_e = ptr;
  935. return str_e->data;
  936. } break;
  937. case IDSTRING_TYPE: {
  938. struct NCDVal__idstring *ids_e = ptr;
  939. const char *value = NCDStringIndex_Value(ids_e->string_index, ids_e->string_id);
  940. return value;
  941. } break;
  942. case EXTERNALSTRING_TYPE: {
  943. struct NCDVal__externalstring *exs_e = ptr;
  944. return exs_e->data;
  945. } break;
  946. default:
  947. ASSERT(0);
  948. return NULL;
  949. }
  950. }
  951. size_t NCDVal_StringLength (NCDValRef string)
  952. {
  953. ASSERT(NCDVal_IsString(string))
  954. void *ptr = NCDValMem__BufAt(string.mem, string.idx);
  955. switch (get_internal_type(*(int *)ptr)) {
  956. case STOREDSTRING_TYPE: {
  957. struct NCDVal__string *str_e = ptr;
  958. return str_e->length;
  959. } break;
  960. case IDSTRING_TYPE: {
  961. struct NCDVal__idstring *ids_e = ptr;
  962. return NCDStringIndex_Length(ids_e->string_index, ids_e->string_id);
  963. } break;
  964. case EXTERNALSTRING_TYPE: {
  965. struct NCDVal__externalstring *exs_e = ptr;
  966. return exs_e->length;
  967. } break;
  968. case COMPOSEDSTRING_TYPE: {
  969. struct NCDVal__composedstring *cms_e = ptr;
  970. return cms_e->length;
  971. } break;
  972. default:
  973. ASSERT(0);
  974. return 0;
  975. }
  976. }
  977. b_cstring NCDValComposedStringResource_Cstring (NCDValComposedStringResource resource, size_t offset, size_t length)
  978. {
  979. b_cstring cstr;
  980. cstr.length = length;
  981. cstr.func = NCDVal__composedstring_cstring_func;
  982. cstr.user1.size = offset;
  983. cstr.user2.fptr = (void (*) (void))resource.func_getptr;
  984. cstr.user3.ptr = resource.user;
  985. return cstr;
  986. }
  987. b_cstring NCDVal_StringCstring (NCDValRef string)
  988. {
  989. ASSERT(NCDVal_IsString(string))
  990. void *ptr = NCDValMem__BufAt(string.mem, string.idx);
  991. switch (get_internal_type(*(int *)ptr)) {
  992. case STOREDSTRING_TYPE: {
  993. struct NCDVal__string *str_e = ptr;
  994. return b_cstring_make_buf(str_e->data, str_e->length);
  995. } break;
  996. case IDSTRING_TYPE: {
  997. struct NCDVal__idstring *ids_e = ptr;
  998. return b_cstring_make_buf(NCDStringIndex_Value(ids_e->string_index, ids_e->string_id), NCDStringIndex_Length(ids_e->string_index, ids_e->string_id));
  999. } break;
  1000. case EXTERNALSTRING_TYPE: {
  1001. struct NCDVal__externalstring *exs_e = ptr;
  1002. return b_cstring_make_buf(exs_e->data, exs_e->length);
  1003. } break;
  1004. case COMPOSEDSTRING_TYPE: {
  1005. struct NCDVal__composedstring *cms_e = ptr;
  1006. b_cstring cstr;
  1007. cstr.length = cms_e->length;
  1008. cstr.func = NCDVal__composedstring_cstring_func;
  1009. cstr.user1.size = cms_e->offset;
  1010. cstr.user2.fptr = (void (*) (void))cms_e->func_getptr;
  1011. cstr.user3.ptr = cms_e->user;
  1012. return cstr;
  1013. } break;
  1014. default: {
  1015. ASSERT(0);
  1016. return b_cstring_make_empty();
  1017. } break;
  1018. }
  1019. }
  1020. int NCDVal_StringNullTerminate (NCDValRef string, NCDValNullTermString *out)
  1021. {
  1022. ASSERT(NCDVal_IsString(string))
  1023. ASSERT(out)
  1024. void *ptr = NCDValMem__BufAt(string.mem, string.idx);
  1025. switch (get_internal_type(*(int *)ptr)) {
  1026. case STOREDSTRING_TYPE: {
  1027. struct NCDVal__string *str_e = ptr;
  1028. out->data = str_e->data;
  1029. out->is_allocated = 0;
  1030. return 1;
  1031. } break;
  1032. case IDSTRING_TYPE: {
  1033. struct NCDVal__idstring *ids_e = ptr;
  1034. out->data = (char *)NCDStringIndex_Value(ids_e->string_index, ids_e->string_id);
  1035. out->is_allocated = 0;
  1036. return 1;
  1037. } break;
  1038. case EXTERNALSTRING_TYPE: {
  1039. struct NCDVal__externalstring *exs_e = ptr;
  1040. char *copy = b_strdup_bin(exs_e->data, exs_e->length);
  1041. if (!copy) {
  1042. return 0;
  1043. }
  1044. out->data = copy;
  1045. out->is_allocated = 1;
  1046. return 1;
  1047. } break;
  1048. case COMPOSEDSTRING_TYPE: {
  1049. struct NCDVal__composedstring *cms_e = ptr;
  1050. size_t length = cms_e->length;
  1051. if (length == SIZE_MAX) {
  1052. return 0;
  1053. }
  1054. char *copy = BAlloc(length + 1);
  1055. if (!copy) {
  1056. return 0;
  1057. }
  1058. NCDVal_StringCopyOut(string, 0, length, copy);
  1059. copy[length] = '\0';
  1060. out->data = copy;
  1061. out->is_allocated = 1;
  1062. return 1;
  1063. } break;
  1064. default:
  1065. ASSERT(0);
  1066. return 0;
  1067. }
  1068. }
  1069. NCDValNullTermString NCDValNullTermString_NewDummy (void)
  1070. {
  1071. NCDValNullTermString nts;
  1072. nts.data = NULL;
  1073. nts.is_allocated = 0;
  1074. return nts;
  1075. }
  1076. void NCDValNullTermString_Free (NCDValNullTermString *o)
  1077. {
  1078. if (o->is_allocated) {
  1079. BFree(o->data);
  1080. }
  1081. }
  1082. int NCDVal_StringContinuize (NCDValRef string, NCDValContString *out)
  1083. {
  1084. ASSERT(NCDVal_IsString(string))
  1085. ASSERT(out)
  1086. if (NCDVal_IsContinuousString(string)) {
  1087. out->data = (char *)NCDVal_StringData(string);
  1088. out->is_allocated = 0;
  1089. return 1;
  1090. }
  1091. size_t length = NCDVal_StringLength(string);
  1092. char *data = BAlloc(length);
  1093. if (!data) {
  1094. return 0;
  1095. }
  1096. NCDVal_StringCopyOut(string, 0, length, data);
  1097. out->data = data;
  1098. out->is_allocated = 1;
  1099. return 1;
  1100. }
  1101. NCDValContString NCDValContString_NewDummy (void)
  1102. {
  1103. NCDValContString cts;
  1104. cts.data = NULL;
  1105. cts.is_allocated = 0;
  1106. return cts;
  1107. }
  1108. void NCDValContString_Free (NCDValContString *o)
  1109. {
  1110. if (o->is_allocated) {
  1111. BFree(o->data);
  1112. }
  1113. }
  1114. void NCDVal_IdStringGet (NCDValRef idstring, NCD_string_id_t *out_string_id,
  1115. NCDStringIndex **out_string_index)
  1116. {
  1117. ASSERT(NCDVal_IsIdString(idstring))
  1118. ASSERT(out_string_id)
  1119. ASSERT(out_string_index)
  1120. struct NCDVal__idstring *ids_e = NCDValMem__BufAt(idstring.mem, idstring.idx);
  1121. *out_string_id = ids_e->string_id;
  1122. *out_string_index = ids_e->string_index;
  1123. }
  1124. NCD_string_id_t NCDVal_IdStringId (NCDValRef idstring)
  1125. {
  1126. ASSERT(NCDVal_IsIdString(idstring))
  1127. struct NCDVal__idstring *ids_e = NCDValMem__BufAt(idstring.mem, idstring.idx);
  1128. return ids_e->string_id;
  1129. }
  1130. NCDStringIndex * NCDVal_IdStringStringIndex (NCDValRef idstring)
  1131. {
  1132. ASSERT(NCDVal_IsIdString(idstring))
  1133. struct NCDVal__idstring *ids_e = NCDValMem__BufAt(idstring.mem, idstring.idx);
  1134. return ids_e->string_index;
  1135. }
  1136. BRefTarget * NCDVal_ExternalStringTarget (NCDValRef externalstring)
  1137. {
  1138. ASSERT(NCDVal_IsExternalString(externalstring))
  1139. struct NCDVal__externalstring *exs_e = NCDValMem__BufAt(externalstring.mem, externalstring.idx);
  1140. return exs_e->ref.target;
  1141. }
  1142. NCDValComposedStringResource NCDVal_ComposedStringResource (NCDValRef composedstring)
  1143. {
  1144. ASSERT(NCDVal_IsComposedString(composedstring))
  1145. struct NCDVal__composedstring *cms_e = NCDValMem__BufAt(composedstring.mem, composedstring.idx);
  1146. NCDValComposedStringResource res;
  1147. res.func_getptr = cms_e->func_getptr;
  1148. res.user = cms_e->user;
  1149. res.ref_target = cms_e->ref.target;
  1150. return res;
  1151. }
  1152. size_t NCDVal_ComposedStringOffset (NCDValRef composedstring)
  1153. {
  1154. ASSERT(NCDVal_IsComposedString(composedstring))
  1155. struct NCDVal__composedstring *cms_e = NCDValMem__BufAt(composedstring.mem, composedstring.idx);
  1156. return cms_e->offset;
  1157. }
  1158. int NCDVal_StringHasNulls (NCDValRef string)
  1159. {
  1160. ASSERT(NCDVal_IsString(string))
  1161. void *ptr = NCDValMem__BufAt(string.mem, string.idx);
  1162. switch (get_internal_type(*(int *)ptr)) {
  1163. case IDSTRING_TYPE: {
  1164. struct NCDVal__idstring *ids_e = ptr;
  1165. return NCDStringIndex_HasNulls(ids_e->string_index, ids_e->string_id);
  1166. } break;
  1167. case STOREDSTRING_TYPE:
  1168. case EXTERNALSTRING_TYPE: {
  1169. const char *data = NCDVal_StringData(string);
  1170. size_t length = NCDVal_StringLength(string);
  1171. return !!memchr(data, '\0', length);
  1172. } break;
  1173. case COMPOSEDSTRING_TYPE: {
  1174. b_cstring cstr = NCDVal_StringCstring(string);
  1175. return b_cstring_memchr(cstr, 0, cstr.length, '\0', NULL);
  1176. } break;
  1177. default:
  1178. ASSERT(0);
  1179. return 0;
  1180. }
  1181. }
  1182. int NCDVal_StringEquals (NCDValRef string, const char *data)
  1183. {
  1184. ASSERT(NCDVal_IsString(string))
  1185. ASSERT(data)
  1186. size_t data_len = strlen(data);
  1187. return NCDVal_StringLength(string) == data_len && NCDVal_StringRegionEquals(string, 0, data_len, data);
  1188. }
  1189. int NCDVal_StringEqualsId (NCDValRef string, NCD_string_id_t string_id,
  1190. NCDStringIndex *string_index)
  1191. {
  1192. ASSERT(NCDVal_IsString(string))
  1193. ASSERT(string_id >= 0)
  1194. ASSERT(string_index)
  1195. void *ptr = NCDValMem__BufAt(string.mem, string.idx);
  1196. switch (get_internal_type(*(int *)ptr)) {
  1197. case STOREDSTRING_TYPE: {
  1198. struct NCDVal__string *str_e = ptr;
  1199. const char *string_data = NCDStringIndex_Value(string_index, string_id);
  1200. size_t string_length = NCDStringIndex_Length(string_index, string_id);
  1201. return (string_length == str_e->length) && !memcmp(string_data, str_e->data, string_length);
  1202. } break;
  1203. case IDSTRING_TYPE: {
  1204. struct NCDVal__idstring *ids_e = ptr;
  1205. ASSERT(ids_e->string_index == string_index)
  1206. return ids_e->string_id == string_id;
  1207. } break;
  1208. case EXTERNALSTRING_TYPE: {
  1209. struct NCDVal__externalstring *exs_e = ptr;
  1210. const char *string_data = NCDStringIndex_Value(string_index, string_id);
  1211. size_t string_length = NCDStringIndex_Length(string_index, string_id);
  1212. return (string_length == exs_e->length) && !memcmp(string_data, exs_e->data, string_length);
  1213. } break;
  1214. case COMPOSEDSTRING_TYPE: {
  1215. struct NCDVal__composedstring *cms_e = ptr;
  1216. const char *string_data = NCDStringIndex_Value(string_index, string_id);
  1217. size_t string_length = NCDStringIndex_Length(string_index, string_id);
  1218. return (string_length == cms_e->length) && NCDVal_StringRegionEquals(string, 0, string_length, string_data);
  1219. } break;
  1220. default:
  1221. ASSERT(0);
  1222. return 0;
  1223. }
  1224. }
  1225. int NCDVal_StringMemCmp (NCDValRef string1, NCDValRef string2, size_t start1, size_t start2, size_t length)
  1226. {
  1227. ASSERT(NCDVal_IsString(string1))
  1228. ASSERT(NCDVal_IsString(string2))
  1229. ASSERT(start1 <= NCDVal_StringLength(string1))
  1230. ASSERT(start2 <= NCDVal_StringLength(string2))
  1231. ASSERT(length <= NCDVal_StringLength(string1) - start1)
  1232. ASSERT(length <= NCDVal_StringLength(string2) - start2)
  1233. if (NCDVal_IsContinuousString(string1) && NCDVal_IsContinuousString(string2)) {
  1234. return memcmp(NCDVal_StringData(string1) + start1, NCDVal_StringData(string2) + start2, length);
  1235. }
  1236. b_cstring cstr1 = NCDVal_StringCstring(string1);
  1237. b_cstring cstr2 = NCDVal_StringCstring(string2);
  1238. return b_cstring_memcmp(cstr1, cstr2, start1, start2, length);
  1239. }
  1240. void NCDVal_StringCopyOut (NCDValRef string, size_t start, size_t length, char *dst)
  1241. {
  1242. ASSERT(NCDVal_IsString(string))
  1243. ASSERT(start <= NCDVal_StringLength(string))
  1244. ASSERT(length <= NCDVal_StringLength(string) - start)
  1245. if (NCDVal_IsContinuousString(string)) {
  1246. memcpy(dst, NCDVal_StringData(string) + start, length);
  1247. return;
  1248. }
  1249. b_cstring cstr = NCDVal_StringCstring(string);
  1250. b_cstring_copy_to_buf(cstr, start, length, dst);
  1251. }
  1252. int NCDVal_StringRegionEquals (NCDValRef string, size_t start, size_t length, const char *data)
  1253. {
  1254. ASSERT(NCDVal_IsString(string))
  1255. ASSERT(start <= NCDVal_StringLength(string))
  1256. ASSERT(length <= NCDVal_StringLength(string) - start)
  1257. if (NCDVal_IsContinuousString(string)) {
  1258. return !memcmp(NCDVal_StringData(string) + start, data, length);
  1259. }
  1260. b_cstring cstr = NCDVal_StringCstring(string);
  1261. return b_cstring_equals_buffer(cstr, start, length, data);
  1262. }
  1263. int NCDVal_IsList (NCDValRef val)
  1264. {
  1265. NCDVal__AssertVal(val);
  1266. return NCDVal_Type(val) == NCDVAL_LIST;
  1267. }
  1268. NCDValRef NCDVal_NewList (NCDValMem *mem, size_t maxcount)
  1269. {
  1270. NCDVal__AssertMem(mem);
  1271. if (maxcount > (NCDVAL_MAXIDX - sizeof(struct NCDVal__list)) / sizeof(NCDVal__idx)) {
  1272. goto fail;
  1273. }
  1274. NCDVal__idx size = sizeof(struct NCDVal__list) + maxcount * sizeof(NCDVal__idx);
  1275. NCDVal__idx idx = NCDValMem__Alloc(mem, size, __alignof(struct NCDVal__list));
  1276. if (idx < 0) {
  1277. goto fail;
  1278. }
  1279. struct NCDVal__list *list_e = NCDValMem__BufAt(mem, idx);
  1280. list_e->type = make_type(NCDVAL_LIST, 0);
  1281. list_e->maxcount = maxcount;
  1282. list_e->count = 0;
  1283. return NCDVal__Ref(mem, idx);
  1284. fail:
  1285. return NCDVal_NewInvalid();
  1286. }
  1287. int NCDVal_ListAppend (NCDValRef list, NCDValRef elem)
  1288. {
  1289. ASSERT(NCDVal_IsList(list))
  1290. ASSERT(NCDVal_ListCount(list) < NCDVal_ListMaxCount(list))
  1291. ASSERT(elem.mem == list.mem)
  1292. NCDVal__AssertValOnly(list.mem, elem.idx);
  1293. struct NCDVal__list *list_e = NCDValMem__BufAt(list.mem, list.idx);
  1294. int new_type = list_e->type;
  1295. if (!bump_depth(&new_type, NCDVal__Depth(elem))) {
  1296. return 0;
  1297. }
  1298. if (NCDValMem__NeedRegisterLink(list.mem, elem.idx)) {
  1299. if (!NCDValMem__RegisterLink(list.mem, elem.idx, list.idx + offsetof(struct NCDVal__list, elem_indices) + list_e->count * sizeof(NCDVal__idx))) {
  1300. return 0;
  1301. }
  1302. list_e = NCDValMem__BufAt(list.mem, list.idx);
  1303. }
  1304. list_e->type = new_type;
  1305. list_e->elem_indices[list_e->count++] = elem.idx;
  1306. return 1;
  1307. }
  1308. size_t NCDVal_ListCount (NCDValRef list)
  1309. {
  1310. ASSERT(NCDVal_IsList(list))
  1311. struct NCDVal__list *list_e = NCDValMem__BufAt(list.mem, list.idx);
  1312. return list_e->count;
  1313. }
  1314. size_t NCDVal_ListMaxCount (NCDValRef list)
  1315. {
  1316. ASSERT(NCDVal_IsList(list))
  1317. struct NCDVal__list *list_e = NCDValMem__BufAt(list.mem, list.idx);
  1318. return list_e->maxcount;
  1319. }
  1320. NCDValRef NCDVal_ListGet (NCDValRef list, size_t pos)
  1321. {
  1322. ASSERT(NCDVal_IsList(list))
  1323. ASSERT(pos < NCDVal_ListCount(list))
  1324. struct NCDVal__list *list_e = NCDValMem__BufAt(list.mem, list.idx);
  1325. ASSERT(pos < list_e->count)
  1326. NCDVal__AssertValOnly(list.mem, list_e->elem_indices[pos]);
  1327. return NCDVal__Ref(list.mem, list_e->elem_indices[pos]);
  1328. }
  1329. int NCDVal_ListRead (NCDValRef list, int num, ...)
  1330. {
  1331. ASSERT(NCDVal_IsList(list))
  1332. ASSERT(num >= 0)
  1333. struct NCDVal__list *list_e = NCDValMem__BufAt(list.mem, list.idx);
  1334. if (num != list_e->count) {
  1335. return 0;
  1336. }
  1337. va_list ap;
  1338. va_start(ap, num);
  1339. for (int i = 0; i < num; i++) {
  1340. NCDValRef *dest = va_arg(ap, NCDValRef *);
  1341. *dest = NCDVal__Ref(list.mem, list_e->elem_indices[i]);
  1342. }
  1343. va_end(ap);
  1344. return 1;
  1345. }
  1346. int NCDVal_ListReadHead (NCDValRef list, int num, ...)
  1347. {
  1348. ASSERT(NCDVal_IsList(list))
  1349. ASSERT(num >= 0)
  1350. struct NCDVal__list *list_e = NCDValMem__BufAt(list.mem, list.idx);
  1351. if (num > list_e->count) {
  1352. return 0;
  1353. }
  1354. va_list ap;
  1355. va_start(ap, num);
  1356. for (int i = 0; i < num; i++) {
  1357. NCDValRef *dest = va_arg(ap, NCDValRef *);
  1358. *dest = NCDVal__Ref(list.mem, list_e->elem_indices[i]);
  1359. }
  1360. va_end(ap);
  1361. return 1;
  1362. }
  1363. int NCDVal_IsMap (NCDValRef val)
  1364. {
  1365. NCDVal__AssertVal(val);
  1366. return NCDVal_Type(val) == NCDVAL_MAP;
  1367. }
  1368. NCDValRef NCDVal_NewMap (NCDValMem *mem, size_t maxcount)
  1369. {
  1370. NCDVal__AssertMem(mem);
  1371. if (maxcount > (NCDVAL_MAXIDX - sizeof(struct NCDVal__map)) / sizeof(struct NCDVal__mapelem)) {
  1372. goto fail;
  1373. }
  1374. NCDVal__idx size = sizeof(struct NCDVal__map) + maxcount * sizeof(struct NCDVal__mapelem);
  1375. NCDVal__idx idx = NCDValMem__Alloc(mem, size, __alignof(struct NCDVal__map));
  1376. if (idx < 0) {
  1377. goto fail;
  1378. }
  1379. struct NCDVal__map *map_e = NCDValMem__BufAt(mem, idx);
  1380. map_e->type = make_type(NCDVAL_MAP, 0);
  1381. map_e->maxcount = maxcount;
  1382. map_e->count = 0;
  1383. NCDVal__MapTree_Init(&map_e->tree);
  1384. return NCDVal__Ref(mem, idx);
  1385. fail:
  1386. return NCDVal_NewInvalid();
  1387. }
  1388. int NCDVal_MapInsert (NCDValRef map, NCDValRef key, NCDValRef val, int *out_inserted)
  1389. {
  1390. ASSERT(NCDVal_IsMap(map))
  1391. ASSERT(NCDVal_MapCount(map) < NCDVal_MapMaxCount(map))
  1392. ASSERT(key.mem == map.mem)
  1393. ASSERT(val.mem == map.mem)
  1394. NCDVal__AssertValOnly(map.mem, key.idx);
  1395. NCDVal__AssertValOnly(map.mem, val.idx);
  1396. struct NCDVal__map *map_e = NCDValMem__BufAt(map.mem, map.idx);
  1397. int new_type = map_e->type;
  1398. if (!bump_depth(&new_type, NCDVal__Depth(key)) || !bump_depth(&new_type, NCDVal__Depth(val))) {
  1399. goto fail0;
  1400. }
  1401. NCDVal__idx elemidx = NCDVal__MapElemIdx(map.idx, map_e->count);
  1402. if (NCDValMem__NeedRegisterLink(map.mem, key.idx)) {
  1403. if (!NCDValMem__RegisterLink(map.mem, key.idx, elemidx + offsetof(struct NCDVal__mapelem, key_idx))) {
  1404. goto fail0;
  1405. }
  1406. map_e = NCDValMem__BufAt(map.mem, map.idx);
  1407. }
  1408. if (NCDValMem__NeedRegisterLink(map.mem, val.idx)) {
  1409. if (!NCDValMem__RegisterLink(map.mem, val.idx, elemidx + offsetof(struct NCDVal__mapelem, val_idx))) {
  1410. goto fail1;
  1411. }
  1412. map_e = NCDValMem__BufAt(map.mem, map.idx);
  1413. }
  1414. struct NCDVal__mapelem *me_e = NCDValMem__BufAt(map.mem, elemidx);
  1415. ASSERT(me_e == &map_e->elems[map_e->count])
  1416. me_e->key_idx = key.idx;
  1417. me_e->val_idx = val.idx;
  1418. int res = NCDVal__MapTree_Insert(&map_e->tree, map.mem, NCDVal__MapTreeDeref(map.mem, elemidx), NULL);
  1419. if (!res) {
  1420. if (out_inserted) {
  1421. *out_inserted = 0;
  1422. }
  1423. return 1;
  1424. }
  1425. map_e->type = new_type;
  1426. map_e->count++;
  1427. if (out_inserted) {
  1428. *out_inserted = 1;
  1429. }
  1430. return 1;
  1431. fail1:
  1432. if (NCDValMem__NeedRegisterLink(map.mem, key.idx)) {
  1433. NCDValMem__PopLastRegisteredLink(map.mem);
  1434. }
  1435. fail0:
  1436. return 0;
  1437. }
  1438. size_t NCDVal_MapCount (NCDValRef map)
  1439. {
  1440. ASSERT(NCDVal_IsMap(map))
  1441. struct NCDVal__map *map_e = NCDValMem__BufAt(map.mem, map.idx);
  1442. return map_e->count;
  1443. }
  1444. size_t NCDVal_MapMaxCount (NCDValRef map)
  1445. {
  1446. ASSERT(NCDVal_IsMap(map))
  1447. struct NCDVal__map *map_e = NCDValMem__BufAt(map.mem, map.idx);
  1448. return map_e->maxcount;
  1449. }
  1450. int NCDVal_MapElemInvalid (NCDValMapElem me)
  1451. {
  1452. ASSERT(me.elemidx >= 0 || me.elemidx == -1)
  1453. return me.elemidx < 0;
  1454. }
  1455. NCDValMapElem NCDVal_MapFirst (NCDValRef map)
  1456. {
  1457. ASSERT(NCDVal_IsMap(map))
  1458. struct NCDVal__map *map_e = NCDValMem__BufAt(map.mem, map.idx);
  1459. if (map_e->count == 0) {
  1460. return NCDVal__MapElem(-1);
  1461. }
  1462. NCDVal__idx elemidx = NCDVal__MapElemIdx(map.idx, 0);
  1463. NCDVal__MapAssertElemOnly(map, elemidx);
  1464. return NCDVal__MapElem(elemidx);
  1465. }
  1466. NCDValMapElem NCDVal_MapNext (NCDValRef map, NCDValMapElem me)
  1467. {
  1468. NCDVal__MapAssertElem(map, me);
  1469. struct NCDVal__map *map_e = NCDValMem__BufAt(map.mem, map.idx);
  1470. ASSERT(map_e->count > 0)
  1471. NCDVal__idx last_elemidx = NCDVal__MapElemIdx(map.idx, map_e->count - 1);
  1472. ASSERT(me.elemidx <= last_elemidx)
  1473. if (me.elemidx == last_elemidx) {
  1474. return NCDVal__MapElem(-1);
  1475. }
  1476. NCDVal__idx elemidx = me.elemidx + sizeof(struct NCDVal__mapelem);
  1477. NCDVal__MapAssertElemOnly(map, elemidx);
  1478. return NCDVal__MapElem(elemidx);
  1479. }
  1480. NCDValMapElem NCDVal_MapOrderedFirst (NCDValRef map)
  1481. {
  1482. ASSERT(NCDVal_IsMap(map))
  1483. struct NCDVal__map *map_e = NCDValMem__BufAt(map.mem, map.idx);
  1484. NCDVal__MapTreeRef ref = NCDVal__MapTree_GetFirst(&map_e->tree, map.mem);
  1485. ASSERT(ref.link == -1 || (NCDVal__MapAssertElemOnly(map, ref.link), 1))
  1486. return NCDVal__MapElem(ref.link);
  1487. }
  1488. NCDValMapElem NCDVal_MapOrderedNext (NCDValRef map, NCDValMapElem me)
  1489. {
  1490. NCDVal__MapAssertElem(map, me);
  1491. struct NCDVal__map *map_e = NCDValMem__BufAt(map.mem, map.idx);
  1492. NCDVal__MapTreeRef ref = NCDVal__MapTree_GetNext(&map_e->tree, map.mem, NCDVal__MapTreeDeref(map.mem, me.elemidx));
  1493. ASSERT(ref.link == -1 || (NCDVal__MapAssertElemOnly(map, ref.link), 1))
  1494. return NCDVal__MapElem(ref.link);
  1495. }
  1496. NCDValRef NCDVal_MapElemKey (NCDValRef map, NCDValMapElem me)
  1497. {
  1498. NCDVal__MapAssertElem(map, me);
  1499. struct NCDVal__mapelem *me_e = NCDValMem__BufAt(map.mem, me.elemidx);
  1500. return NCDVal__Ref(map.mem, me_e->key_idx);
  1501. }
  1502. NCDValRef NCDVal_MapElemVal (NCDValRef map, NCDValMapElem me)
  1503. {
  1504. NCDVal__MapAssertElem(map, me);
  1505. struct NCDVal__mapelem *me_e = NCDValMem__BufAt(map.mem, me.elemidx);
  1506. return NCDVal__Ref(map.mem, me_e->val_idx);
  1507. }
  1508. NCDValMapElem NCDVal_MapFindKey (NCDValRef map, NCDValRef key)
  1509. {
  1510. ASSERT(NCDVal_IsMap(map))
  1511. NCDVal__AssertVal(key);
  1512. struct NCDVal__map *map_e = NCDValMem__BufAt(map.mem, map.idx);
  1513. NCDVal__MapTreeRef ref = NCDVal__MapTree_LookupExact(&map_e->tree, map.mem, key);
  1514. ASSERT(ref.link == -1 || (NCDVal__MapAssertElemOnly(map, ref.link), 1))
  1515. return NCDVal__MapElem(ref.link);
  1516. }
  1517. NCDValRef NCDVal_MapGetValue (NCDValRef map, const char *key_str)
  1518. {
  1519. ASSERT(NCDVal_IsMap(map))
  1520. ASSERT(key_str)
  1521. NCDValMem mem;
  1522. mem.buf = NULL;
  1523. mem.size = NCDVAL_FASTBUF_SIZE;
  1524. mem.used = sizeof(struct NCDVal__externalstring);
  1525. mem.first_ref = -1;
  1526. struct NCDVal__externalstring *exs_e = (void *)mem.fastbuf;
  1527. exs_e->type = make_type(EXTERNALSTRING_TYPE, 0);
  1528. exs_e->data = key_str;
  1529. exs_e->length = strlen(key_str);
  1530. exs_e->ref.target = NULL;
  1531. NCDValRef key = NCDVal__Ref(&mem, 0);
  1532. NCDValMapElem elem = NCDVal_MapFindKey(map, key);
  1533. if (NCDVal_MapElemInvalid(elem)) {
  1534. return NCDVal_NewInvalid();
  1535. }
  1536. return NCDVal_MapElemVal(map, elem);
  1537. }
  1538. static void replaceprog_build_recurser (NCDValMem *mem, NCDVal__idx idx, size_t *out_num_instr, NCDValReplaceProg *prog)
  1539. {
  1540. ASSERT(idx >= 0)
  1541. NCDVal__AssertValOnly(mem, idx);
  1542. ASSERT(out_num_instr)
  1543. *out_num_instr = 0;
  1544. void *ptr = NCDValMem__BufAt(mem, idx);
  1545. struct NCDVal__instr instr;
  1546. switch (get_internal_type(*((int *)(ptr)))) {
  1547. case STOREDSTRING_TYPE:
  1548. case IDSTRING_TYPE:
  1549. case EXTERNALSTRING_TYPE:
  1550. case COMPOSEDSTRING_TYPE: {
  1551. } break;
  1552. case NCDVAL_LIST: {
  1553. struct NCDVal__list *list_e = ptr;
  1554. for (NCDVal__idx i = 0; i < list_e->count; i++) {
  1555. int elem_changed = 0;
  1556. if (list_e->elem_indices[i] < -1) {
  1557. if (prog) {
  1558. instr.type = NCDVAL_INSTR_PLACEHOLDER;
  1559. instr.placeholder.plid = list_e->elem_indices[i] - NCDVAL_MINIDX;
  1560. instr.placeholder.plidx = idx + offsetof(struct NCDVal__list, elem_indices) + i * sizeof(NCDVal__idx);
  1561. prog->instrs[prog->num_instrs++] = instr;
  1562. }
  1563. (*out_num_instr)++;
  1564. elem_changed = 1;
  1565. } else {
  1566. size_t elem_num_instr;
  1567. replaceprog_build_recurser(mem, list_e->elem_indices[i], &elem_num_instr, prog);
  1568. (*out_num_instr) += elem_num_instr;
  1569. if (elem_num_instr > 0) {
  1570. elem_changed = 1;
  1571. }
  1572. }
  1573. if (elem_changed) {
  1574. if (prog) {
  1575. instr.type = NCDVAL_INSTR_BUMPDEPTH;
  1576. instr.bumpdepth.parent_idx = idx;
  1577. instr.bumpdepth.child_idx_idx = idx + offsetof(struct NCDVal__list, elem_indices) + i * sizeof(NCDVal__idx);
  1578. prog->instrs[prog->num_instrs++] = instr;
  1579. }
  1580. (*out_num_instr)++;
  1581. }
  1582. }
  1583. } break;
  1584. case NCDVAL_MAP: {
  1585. struct NCDVal__map *map_e = ptr;
  1586. for (NCDVal__idx i = 0; i < map_e->count; i++) {
  1587. int key_changed = 0;
  1588. int val_changed = 0;
  1589. if (map_e->elems[i].key_idx < -1) {
  1590. if (prog) {
  1591. instr.type = NCDVAL_INSTR_PLACEHOLDER;
  1592. instr.placeholder.plid = map_e->elems[i].key_idx - NCDVAL_MINIDX;
  1593. instr.placeholder.plidx = idx + offsetof(struct NCDVal__map, elems) + i * sizeof(struct NCDVal__mapelem) + offsetof(struct NCDVal__mapelem, key_idx);
  1594. prog->instrs[prog->num_instrs++] = instr;
  1595. }
  1596. (*out_num_instr)++;
  1597. key_changed = 1;
  1598. } else {
  1599. size_t key_num_instr;
  1600. replaceprog_build_recurser(mem, map_e->elems[i].key_idx, &key_num_instr, prog);
  1601. (*out_num_instr) += key_num_instr;
  1602. if (key_num_instr > 0) {
  1603. key_changed = 1;
  1604. }
  1605. }
  1606. if (map_e->elems[i].val_idx < -1) {
  1607. if (prog) {
  1608. instr.type = NCDVAL_INSTR_PLACEHOLDER;
  1609. instr.placeholder.plid = map_e->elems[i].val_idx - NCDVAL_MINIDX;
  1610. instr.placeholder.plidx = idx + offsetof(struct NCDVal__map, elems) + i * sizeof(struct NCDVal__mapelem) + offsetof(struct NCDVal__mapelem, val_idx);
  1611. prog->instrs[prog->num_instrs++] = instr;
  1612. }
  1613. (*out_num_instr)++;
  1614. val_changed = 1;
  1615. } else {
  1616. size_t val_num_instr;
  1617. replaceprog_build_recurser(mem, map_e->elems[i].val_idx, &val_num_instr, prog);
  1618. (*out_num_instr) += val_num_instr;
  1619. if (val_num_instr > 0) {
  1620. val_changed = 1;
  1621. }
  1622. }
  1623. if (key_changed) {
  1624. if (prog) {
  1625. instr.type = NCDVAL_INSTR_REINSERT;
  1626. instr.reinsert.mapidx = idx;
  1627. instr.reinsert.elempos = i;
  1628. prog->instrs[prog->num_instrs++] = instr;
  1629. }
  1630. (*out_num_instr)++;
  1631. if (prog) {
  1632. instr.type = NCDVAL_INSTR_BUMPDEPTH;
  1633. instr.bumpdepth.parent_idx = idx;
  1634. instr.bumpdepth.child_idx_idx = idx + offsetof(struct NCDVal__map, elems) + i * sizeof(struct NCDVal__mapelem) + offsetof(struct NCDVal__mapelem, key_idx);
  1635. prog->instrs[prog->num_instrs++] = instr;
  1636. }
  1637. (*out_num_instr)++;
  1638. }
  1639. if (val_changed) {
  1640. if (prog) {
  1641. instr.type = NCDVAL_INSTR_BUMPDEPTH;
  1642. instr.bumpdepth.parent_idx = idx;
  1643. instr.bumpdepth.child_idx_idx = idx + offsetof(struct NCDVal__map, elems) + i * sizeof(struct NCDVal__mapelem) + offsetof(struct NCDVal__mapelem, val_idx);
  1644. prog->instrs[prog->num_instrs++] = instr;
  1645. }
  1646. (*out_num_instr)++;
  1647. }
  1648. }
  1649. } break;
  1650. default: ASSERT(0);
  1651. }
  1652. }
  1653. int NCDValReplaceProg_Init (NCDValReplaceProg *o, NCDValRef val)
  1654. {
  1655. NCDVal__AssertVal(val);
  1656. ASSERT(!NCDVal_IsPlaceholder(val))
  1657. size_t num_instrs;
  1658. replaceprog_build_recurser(val.mem, val.idx, &num_instrs, NULL);
  1659. if (!(o->instrs = BAllocArray(num_instrs, sizeof(o->instrs[0])))) {
  1660. BLog(BLOG_ERROR, "BAllocArray failed");
  1661. return 0;
  1662. }
  1663. o->num_instrs = 0;
  1664. size_t num_instrs2;
  1665. replaceprog_build_recurser(val.mem, val.idx, &num_instrs2, o);
  1666. ASSERT(num_instrs2 == num_instrs)
  1667. ASSERT(o->num_instrs == num_instrs)
  1668. return 1;
  1669. }
  1670. void NCDValReplaceProg_Free (NCDValReplaceProg *o)
  1671. {
  1672. BFree(o->instrs);
  1673. }
  1674. int NCDValReplaceProg_Execute (NCDValReplaceProg prog, NCDValMem *mem, NCDVal_replace_func replace, void *arg)
  1675. {
  1676. NCDVal__AssertMem(mem);
  1677. ASSERT(replace)
  1678. for (size_t i = 0; i < prog.num_instrs; i++) {
  1679. struct NCDVal__instr instr = prog.instrs[i];
  1680. switch (instr.type) {
  1681. case NCDVAL_INSTR_PLACEHOLDER: {
  1682. #ifndef NDEBUG
  1683. NCDVal__idx *check_plptr = NCDValMem__BufAt(mem, instr.placeholder.plidx);
  1684. ASSERT(*check_plptr < -1)
  1685. ASSERT(*check_plptr - NCDVAL_MINIDX == instr.placeholder.plid)
  1686. #endif
  1687. NCDValRef repval;
  1688. if (!replace(arg, instr.placeholder.plid, mem, &repval) || NCDVal_IsInvalid(repval)) {
  1689. return 0;
  1690. }
  1691. ASSERT(repval.mem == mem)
  1692. if (NCDValMem__NeedRegisterLink(mem, repval.idx)) {
  1693. NCDValMem__RegisterLink(mem, repval.idx, instr.placeholder.plidx);
  1694. }
  1695. NCDVal__idx *plptr = NCDValMem__BufAt(mem, instr.placeholder.plidx);
  1696. *plptr = repval.idx;
  1697. } break;
  1698. case NCDVAL_INSTR_REINSERT: {
  1699. NCDVal__AssertValOnly(mem, instr.reinsert.mapidx);
  1700. struct NCDVal__map *map_e = NCDValMem__BufAt(mem, instr.reinsert.mapidx);
  1701. ASSERT(get_internal_type(map_e->type) == NCDVAL_MAP)
  1702. ASSERT(instr.reinsert.elempos >= 0)
  1703. ASSERT(instr.reinsert.elempos < map_e->count)
  1704. NCDVal__MapTreeRef ref = {&map_e->elems[instr.reinsert.elempos], NCDVal__MapElemIdx(instr.reinsert.mapidx, instr.reinsert.elempos)};
  1705. NCDVal__MapTree_Remove(&map_e->tree, mem, ref);
  1706. if (!NCDVal__MapTree_Insert(&map_e->tree, mem, ref, NULL)) {
  1707. BLog(BLOG_ERROR, "duplicate key in map");
  1708. return 0;
  1709. }
  1710. } break;
  1711. case NCDVAL_INSTR_BUMPDEPTH: {
  1712. NCDVal__AssertValOnly(mem, instr.bumpdepth.parent_idx);
  1713. int *parent_type_ptr = NCDValMem__BufAt(mem, instr.bumpdepth.parent_idx);
  1714. NCDVal__idx *child_type_idx_ptr = NCDValMem__BufAt(mem, instr.bumpdepth.child_idx_idx);
  1715. NCDVal__AssertValOnly(mem, *child_type_idx_ptr);
  1716. int *child_type_ptr = NCDValMem__BufAt(mem, *child_type_idx_ptr);
  1717. if (!bump_depth(parent_type_ptr, get_depth(*child_type_ptr))) {
  1718. BLog(BLOG_ERROR, "depth limit exceeded");
  1719. return 0;
  1720. }
  1721. } break;
  1722. default: ASSERT(0);
  1723. }
  1724. }
  1725. return 1;
  1726. }