NCDVal.c 63 KB

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