xs.h 37 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637
  1. /* copyright (c) 2022 - 2026 grunfink et al. / MIT license */
  2. #ifndef _XS_H
  3. #define _XS_H
  4. #include <stdio.h>
  5. #include <string.h>
  6. #include <stdlib.h>
  7. #include <ctype.h>
  8. #include <unistd.h>
  9. #include <stdarg.h>
  10. #include <signal.h>
  11. #include <errno.h>
  12. #include <stdint.h>
  13. typedef enum {
  14. XSTYPE_STRING = 0x02, /* C string (\0 delimited) (NOT STORED) */
  15. XSTYPE_NUMBER = 0x17, /* double in spirit, stored as a C string (\0 delimited) */
  16. XSTYPE_NULL = 0x18, /* Special NULL value */
  17. XSTYPE_TRUE = 0x06, /* Boolean */
  18. XSTYPE_FALSE = 0x15, /* Boolean */
  19. XSTYPE_LIST = 0x1d, /* Sequence of LITEMs up to EOM (with size) */
  20. XSTYPE_LITEM = 0x1f, /* Element of a list (any type) */
  21. XSTYPE_DICT = 0x1c, /* Sequence of KEYVALs up to EOM (with size) */
  22. XSTYPE_KEYVAL = 0x1e, /* key + value (STRING key + any type) */
  23. XSTYPE_DATA = 0x10 /* A block of anonymous data */
  24. } xstype;
  25. /* types */
  26. typedef char xs_val;
  27. typedef char xs_str;
  28. typedef char xs_list;
  29. typedef char xs_keyval;
  30. typedef char xs_dict;
  31. typedef char xs_number;
  32. typedef char xs_data;
  33. /* size in bytes of the type size */
  34. #define _XS_TYPE_SIZE 4
  35. /* auto-destroyable strings */
  36. #define xs __attribute__ ((__cleanup__ (_xs_destroy))) xs_val
  37. /* not really all, just very much */
  38. #define XS_ALL 0xfffffff
  39. #ifndef xs_countof
  40. #define xs_countof(a) (sizeof((a)) / sizeof((*a)))
  41. #endif
  42. void *xs_free(void *ptr);
  43. void *_xs_realloc(void *ptr, size_t size, const char *file, int line, const char *func);
  44. #define xs_realloc(ptr, size) _xs_realloc(ptr, size, __FILE__, __LINE__, __func__)
  45. int _xs_blk_size(int sz);
  46. void _xs_destroy(char **var);
  47. #define xs_debug() raise(SIGTRAP)
  48. xstype xs_type(const xs_val *data);
  49. int xs_size(const xs_val *data);
  50. int xs_is_null(const xs_val *data);
  51. int xs_cmp(const xs_val *v1, const xs_val *v2);
  52. const xs_val *xs_or(const xs_val *v1, const xs_val *v2);
  53. xs_val *xs_dup(const xs_val *data);
  54. xs_val *xs_expand(xs_val *data, int offset, int size);
  55. xs_val *xs_collapse(xs_val *data, int offset, int size);
  56. xs_val *xs_insert_m(xs_val *data, int offset, const char *mem, int size);
  57. #define xs_insert(data, offset, data2) xs_insert_m(data, offset, data2, xs_size(data2))
  58. #define xs_append_m(data, mem, size) xs_insert_m(data, xs_size(data) - 1, mem, size)
  59. xs_val *xs_stock(int type);
  60. xs_str *xs_str_new(const char *str);
  61. xs_str *xs_str_new_sz(const char *mem, int sz);
  62. xs_str *xs_str_wrap_i(const char *prefix, xs_str *str, const char *suffix);
  63. #define xs_str_prepend_i(str, prefix) xs_str_wrap_i(prefix, str, NULL)
  64. xs_str *_xs_str_cat(xs_str *str, const char *strs[]);
  65. #define xs_str_cat(str, ...) _xs_str_cat(str, (const char *[]){ __VA_ARGS__, NULL })
  66. xs_str *xs_replace_in(xs_str *str, const char *sfrom, const char *sto, int times);
  67. #define xs_replace_i(str, sfrom, sto) xs_replace_in(str, sfrom, sto, XS_ALL)
  68. #define xs_replace(str, sfrom, sto) xs_replace_in(xs_dup(str), sfrom, sto, XS_ALL)
  69. #define xs_replace_n(str, sfrom, sto, times) xs_replace_in(xs_dup(str), sfrom, sto, times)
  70. xs_str *xs_fmt(const char *fmt, ...);
  71. int xs_str_in(const char *haystack, const char *needle);
  72. int xs_between(const char *prefix, const char *str, const char *suffix);
  73. #define xs_startswith(str, prefix) xs_between(prefix, str, NULL)
  74. #define xs_endswith(str, suffix) xs_between(NULL, str, suffix)
  75. xs_str *xs_crop_i(xs_str *str, int start, int end);
  76. xs_str *xs_lstrip_chars_i(xs_str *str, const char *chars);
  77. xs_str *xs_rstrip_chars_i(xs_str *str, const char *chars);
  78. xs_str *xs_strip_chars_i(xs_str *str, const char *chars);
  79. #define xs_strip_i(str) xs_strip_chars_i(str, " \r\n\t\v\f")
  80. xs_str *xs_tolower_i(xs_str *str);
  81. xs_str *xs_toupper_i(xs_str *str);
  82. xs_list *xs_list_new(void);
  83. xs_list *xs_list_append_m(xs_list *list, const char *mem, int dsz);
  84. xs_list *_xs_list_append(xs_list *list, const xs_val *vals[]);
  85. #define xs_list_append(list, ...) _xs_list_append(list, (const xs_val *[]){ __VA_ARGS__, NULL })
  86. int xs_list_iter(xs_list **list, const xs_val **value);
  87. int xs_list_next(const xs_list *list, const xs_val **value, int *ctxt);
  88. int xs_list_cap(xs_list *list, int max);
  89. xs_list *xs_list_reverse(const xs_list *l);
  90. int xs_list_len(const xs_list *list);
  91. const xs_val *xs_list_get(const xs_list *list, int num);
  92. xs_list *xs_list_del(xs_list *list, int num);
  93. xs_list *xs_list_insert(xs_list *list, int num, const xs_val *data);
  94. xs_list *xs_list_set(xs_list *list, int num, const xs_val *data);
  95. xs_list *xs_list_dequeue(xs_list *list, xs_val **data, int last);
  96. #define xs_list_pop(list, data) xs_list_dequeue(list, data, 1)
  97. #define xs_list_shift(list, data) xs_list_dequeue(list, data, 0)
  98. int xs_list_in(const xs_list *list, const xs_val *val);
  99. xs_str *xs_join(const xs_list *list, const char *sep);
  100. xs_list *xs_split_n(const char *str, const char *sep, int times);
  101. #define xs_split(str, sep) xs_split_n(str, sep, XS_ALL)
  102. xs_list *xs_list_cat(xs_list *l1, const xs_list *l2);
  103. int xs_keyval_size(const xs_str *key, const xs_val *value);
  104. xs_str *xs_keyval_key(const xs_keyval *keyval);
  105. xs_val *xs_keyval_value(const xs_keyval *keyval);
  106. xs_keyval *xs_keyval_make(xs_keyval *keyval, const xs_str *key, const xs_val *value);
  107. xs_dict *xs_dict_new(void);
  108. xs_dict *xs_dict_append(xs_dict *dict, const xs_str *key, const xs_val *value);
  109. xs_dict *xs_dict_prepend(xs_dict *dict, const xs_str *key, const xs_val *value);
  110. int xs_dict_next(const xs_dict *dict, const xs_str **key, const xs_val **value, int *ctxt);
  111. const xs_val *xs_dict_get(const xs_dict *dict, const xs_str *key);
  112. #define xs_dict_get_def(dict, key, def) xs_or(xs_dict_get(dict, key), def)
  113. xs_dict *xs_dict_del(xs_dict *dict, const xs_str *key);
  114. xs_dict *xs_dict_set(xs_dict *dict, const xs_str *key, const xs_val *data);
  115. xs_dict *xs_dict_gc(const xs_dict *dict);
  116. const xs_val *xs_dict_get_path_sep(const xs_dict *dict, const char *path, const char *sep);
  117. #define xs_dict_get_path(dict, path) xs_dict_get_path_sep(dict, path, ".")
  118. xs_dict *xs_dict_set_path_sep(xs_dict *dict, const char *path, const xs_val *value, const char *sep);
  119. #define xs_dict_set_path(dict, path, value) xs_dict_set_path_sep(dict, path, value, ".")
  120. xs_val *xs_val_new(xstype t);
  121. xs_number *xs_number_new(double f);
  122. xs_number *xs_number_new_l(long l);
  123. double xs_number_get(const xs_number *v);
  124. long xs_number_get_l(const xs_number *v);
  125. const char *xs_number_str(const xs_number *v);
  126. xs_data *xs_data_new(const void *data, int size);
  127. int xs_data_size(const xs_data *value);
  128. const void *xs_data_ptr(const xs_data *value);
  129. void xs_data_get(void *data, const xs_data *value);
  130. void *xs_memmem(const char *haystack, int h_size, const char *needle, int n_size);
  131. unsigned int xs_hash_func(const char *data, int size);
  132. uint64_t xs_hash64_func(const char *data, int size);
  133. #ifdef XS_ASSERT
  134. #include <assert.h>
  135. #define XS_ASSERT_TYPE(v, t) assert(xs_type(v) == t)
  136. #define XS_ASSERT_TYPE_NULL(v, t) assert(v == NULL || xs_type(v) == t)
  137. #else
  138. #define XS_ASSERT_TYPE(v, t) (void)(0)
  139. #define XS_ASSERT_TYPE_NULL(v, t) (void)(0)
  140. #endif
  141. #define xs_return(v) xs_val *__r = v; v = NULL; return __r
  142. #define xs_is_true(v) (xs_type((v)) == XSTYPE_TRUE)
  143. #define xs_is_false(v) (xs_type((v)) == XSTYPE_FALSE)
  144. #define xs_not(v) xs_stock(xs_is_true((v)) ? XSTYPE_FALSE : XSTYPE_TRUE)
  145. #define xs_is_string(v) (xs_type((v)) == XSTYPE_STRING)
  146. #define xs_is_list(v) (xs_type((v)) == XSTYPE_LIST)
  147. #define xs_is_dict(v) (xs_type((v)) == XSTYPE_DICT)
  148. #define xs_is_number(v) (xs_type((v)) == XSTYPE_NUMBER)
  149. #define xs_list_foreach(l, v) for (int ct_##__LINE__ = 0; xs_list_next(l, &v, &ct_##__LINE__); )
  150. #define xs_dict_foreach(l, k, v) for (int ct_##__LINE__ = 0; xs_dict_next(l, &k, &v, &ct_##__LINE__); )
  151. #ifdef XS_IMPLEMENTATION
  152. void *_xs_realloc(void *ptr, size_t size, const char *file, int line, const char *func)
  153. {
  154. xs_val *ndata = realloc(ptr, size);
  155. if (ndata == NULL) {
  156. fprintf(stderr, "ERROR: out of memory at %s:%d: %s()\n", file, line, func);
  157. abort();
  158. }
  159. #ifdef XS_DEBUG
  160. if (ndata != ptr) {
  161. int n;
  162. FILE *f = fopen("xs_memory.out", "a");
  163. if (ptr != NULL)
  164. fprintf(f, "%p r\n", ptr);
  165. fprintf(f, "%p a %ld %s:%d: %s", ndata, size, file, line, func);
  166. if (ptr != NULL) {
  167. fprintf(f, " [");
  168. for (n = 0; n < 32 && ndata[n]; n++) {
  169. if (ndata[n] >= 32 && ndata[n] <= 127)
  170. fprintf(f, "%c", ndata[n]);
  171. else
  172. fprintf(f, "\\%02x", (unsigned char)ndata[n]);
  173. }
  174. fprintf(f, "]");
  175. }
  176. fprintf(f, "\n");
  177. fclose(f);
  178. }
  179. #else
  180. (void)file;
  181. (void)line;
  182. (void)func;
  183. #endif
  184. return ndata;
  185. }
  186. void *xs_free(void *ptr)
  187. {
  188. #ifdef XS_DEBUG
  189. if (ptr != NULL) {
  190. FILE *f = fopen("xs_memory.out", "a");
  191. fprintf(f, "%p b\n", ptr);
  192. fclose(f);
  193. }
  194. #endif
  195. free(ptr);
  196. return NULL;
  197. }
  198. void _xs_destroy(char **var)
  199. {
  200. /*
  201. if (_xs_debug)
  202. printf("_xs_destroy %p\n", var);
  203. */
  204. xs_free(*var);
  205. }
  206. int _xs_blk_size(int sz)
  207. /* calculates the block size */
  208. {
  209. int blk_size = 4096;
  210. if (sz < 256)
  211. blk_size = 32;
  212. else
  213. if (sz < 4096)
  214. blk_size = 256;
  215. return ((((sz) + blk_size) / blk_size) * blk_size);
  216. }
  217. xstype xs_type(const xs_val *data)
  218. /* return the type of data */
  219. {
  220. xstype t;
  221. if (data == NULL)
  222. t = XSTYPE_NULL;
  223. else
  224. switch (data[0]) {
  225. case XSTYPE_NULL:
  226. case XSTYPE_TRUE:
  227. case XSTYPE_FALSE:
  228. case XSTYPE_LIST:
  229. case XSTYPE_LITEM:
  230. case XSTYPE_DICT:
  231. case XSTYPE_KEYVAL:
  232. case XSTYPE_NUMBER:
  233. case XSTYPE_DATA:
  234. t = data[0];
  235. break;
  236. default:
  237. t = XSTYPE_STRING;
  238. break;
  239. }
  240. return t;
  241. }
  242. void _xs_put_size(xs_val *ptr, int i)
  243. /* must match _XS_TYPE_SIZE */
  244. {
  245. memcpy(ptr + 1, &i, sizeof(i));
  246. }
  247. int _xs_get_size(const xs_val *ptr)
  248. /* must match _XS_TYPE_SIZE */
  249. {
  250. int i;
  251. memcpy(&i, ptr + 1, sizeof(i));
  252. return i;
  253. }
  254. int xs_size(const xs_val *data)
  255. /* returns the size of data in bytes */
  256. {
  257. int len = 0;
  258. const char *p;
  259. if (data == NULL)
  260. return 0;
  261. switch (xs_type(data)) {
  262. case XSTYPE_STRING:
  263. len = strlen(data) + 1;
  264. break;
  265. case XSTYPE_LIST:
  266. case XSTYPE_DICT:
  267. case XSTYPE_DATA:
  268. len = _xs_get_size(data);
  269. break;
  270. case XSTYPE_KEYVAL:
  271. /* calculate the size of the key and the value */
  272. p = data + 1;
  273. p += xs_size(p);
  274. p += xs_size(p);
  275. len = p - data;
  276. break;
  277. case XSTYPE_LITEM:
  278. /* it's the size of the item + 1 */
  279. p = data + 1;
  280. p += xs_size(p);
  281. len = p - data;
  282. break;
  283. case XSTYPE_NUMBER:
  284. len = 1 + xs_size(data + 1);
  285. break;
  286. default:
  287. len = 1;
  288. }
  289. return len;
  290. }
  291. int xs_is_null(const xs_val *data)
  292. /* checks for null */
  293. {
  294. return (xs_type(data) == XSTYPE_NULL);
  295. }
  296. int xs_cmp(const xs_val *v1, const xs_val *v2)
  297. /* compares two values */
  298. {
  299. xstype t1 = xs_type(v1);
  300. xstype t2 = xs_type(v2);
  301. if (t1 == XSTYPE_NULL || t2 == XSTYPE_NULL)
  302. return 0;
  303. if (t1 == XSTYPE_STRING && t2 == XSTYPE_STRING)
  304. return strcmp(v1, v2);
  305. if (t1 == XSTYPE_NUMBER && t2 == XSTYPE_NUMBER) {
  306. double d1 = xs_number_get(v1);
  307. double d2 = xs_number_get(v2);
  308. return d1 == d2 ? 0 : d1 < d2 ? -1 : 1;
  309. }
  310. int s1 = xs_size(v1);
  311. int s2 = xs_size(v2);
  312. int d = s1 - s2;
  313. return d == 0 ? memcmp(v1, v2, s1) : d;
  314. }
  315. const xs_val *xs_or(const xs_val *v1, const xs_val *v2)
  316. /* returns v1 if it's not NULL, else v2 */
  317. {
  318. return v1 == NULL ? v2 : v1;
  319. }
  320. xs_val *xs_dup(const xs_val *data)
  321. /* creates a duplicate of data */
  322. {
  323. xs_val *s = NULL;
  324. if (data) {
  325. int sz = xs_size(data);
  326. s = xs_realloc(NULL, _xs_blk_size(sz));
  327. memcpy(s, data, sz);
  328. }
  329. return s;
  330. }
  331. xs_val *xs_expand(xs_val *data, int offset, int size)
  332. /* opens a hole in data */
  333. {
  334. xstype type = xs_type(data);
  335. int sz = xs_size(data);
  336. int n;
  337. sz += size;
  338. /* open room */
  339. data = xs_realloc(data, _xs_blk_size(sz));
  340. /* move up the rest of the data */
  341. for (n = sz - 1; n >= offset + size; n--)
  342. data[n] = data[n - size];
  343. if (type == XSTYPE_LIST ||
  344. type == XSTYPE_DICT ||
  345. type == XSTYPE_DATA)
  346. _xs_put_size(data, sz);
  347. return data;
  348. }
  349. xs_val *xs_collapse(xs_val *data, int offset, int size)
  350. /* shrinks data */
  351. {
  352. int sz = xs_size(data);
  353. int n;
  354. /* don't try to delete beyond the limit */
  355. if (offset + size > sz)
  356. size = sz - offset;
  357. /* shrink total size */
  358. sz -= size;
  359. for (n = offset; n < sz; n++)
  360. data[n] = data[n + size];
  361. if (xs_type(data) == XSTYPE_LIST ||
  362. xs_type(data) == XSTYPE_DICT ||
  363. xs_type(data) == XSTYPE_DATA)
  364. _xs_put_size(data, sz);
  365. return xs_realloc(data, _xs_blk_size(sz));
  366. }
  367. xs_val *xs_insert_m(xs_val *data, int offset, const char *mem, int size)
  368. /* inserts a memory block */
  369. {
  370. data = xs_expand(data, offset, size);
  371. memcpy(data + offset, mem, size);
  372. return data;
  373. }
  374. xs_val *xs_stock(int type)
  375. /* returns stock values */
  376. {
  377. static xs_val stock_null[] = { XSTYPE_NULL };
  378. static xs_val stock_true[] = { XSTYPE_TRUE };
  379. static xs_val stock_false[] = { XSTYPE_FALSE };
  380. static xs_val stock_0[] = { XSTYPE_NUMBER, '0', '\0' };
  381. static xs_val stock_1[] = { XSTYPE_NUMBER, '1', '\0' };
  382. static xs_list *stock_list = NULL;
  383. static xs_dict *stock_dict = NULL;
  384. switch (type) {
  385. case 0: return stock_0;
  386. case 1: return stock_1;
  387. case XSTYPE_NULL: return stock_null;
  388. case XSTYPE_TRUE: return stock_true;
  389. case XSTYPE_FALSE: return stock_false;
  390. case XSTYPE_LIST:
  391. if (stock_list == NULL)
  392. stock_list = xs_list_new();
  393. return stock_list;
  394. case XSTYPE_DICT:
  395. if (stock_dict == NULL)
  396. stock_dict = xs_dict_new();
  397. return stock_dict;
  398. }
  399. return NULL;
  400. }
  401. /** strings **/
  402. xs_str *xs_str_new(const char *str)
  403. /* creates a new string */
  404. {
  405. return xs_insert(NULL, 0, str ? str : "");
  406. }
  407. xs_str *xs_str_new_sz(const char *mem, int sz)
  408. /* creates a new string from a memory block, adding an asciiz */
  409. {
  410. xs_str *s = xs_realloc(NULL, _xs_blk_size(sz + 1));
  411. memcpy(s, mem, sz);
  412. s[sz] = '\0';
  413. return s;
  414. }
  415. xs_str *xs_str_wrap_i(const char *prefix, xs_str *str, const char *suffix)
  416. /* wraps str with prefix and suffix */
  417. {
  418. XS_ASSERT_TYPE(str, XSTYPE_STRING);
  419. if (prefix)
  420. str = xs_insert_m(str, 0, prefix, strlen(prefix));
  421. if (suffix)
  422. str = xs_insert_m(str, strlen(str), suffix, strlen(suffix));
  423. return str;
  424. }
  425. xs_str *_xs_str_cat(xs_str *str, const char *strs[])
  426. /* concatenates all strings after str */
  427. {
  428. int o = strlen(str);
  429. while (*strs) {
  430. int sz = strlen(*strs);
  431. str = xs_insert_m(str, o, *strs, sz);
  432. o += sz;
  433. strs++;
  434. }
  435. return str;
  436. }
  437. xs_str *xs_replace_in(xs_str *str, const char *sfrom, const char *sto, int times)
  438. /* replaces inline all sfrom with sto */
  439. {
  440. XS_ASSERT_TYPE(str, XSTYPE_STRING);
  441. int sfsz = strlen(sfrom);
  442. int stsz = strlen(sto);
  443. int diff = stsz - sfsz;
  444. char *ss;
  445. int offset = 0;
  446. while (times > 0 && (ss = strstr(str + offset, sfrom)) != NULL) {
  447. int n_offset = ss - str;
  448. if (diff < 0)
  449. str = xs_collapse(str, n_offset, -diff);
  450. else
  451. if (diff > 0)
  452. str = xs_expand(str, n_offset, diff);
  453. memcpy(str + n_offset, sto, stsz);
  454. offset = n_offset + stsz;
  455. times--;
  456. }
  457. return str;
  458. }
  459. xs_str *xs_fmt(const char *fmt, ...)
  460. /* formats a string with printf()-like marks */
  461. {
  462. int n;
  463. xs_str *s = NULL;
  464. va_list ap;
  465. va_start(ap, fmt);
  466. n = vsnprintf(s, 0, fmt, ap);
  467. va_end(ap);
  468. if (n > 0) {
  469. s = xs_realloc(NULL, _xs_blk_size(n + 1));
  470. va_start(ap, fmt);
  471. vsnprintf(s, n + 1, fmt, ap);
  472. va_end(ap);
  473. }
  474. return s;
  475. }
  476. int xs_str_in(const char *haystack, const char *needle)
  477. /* finds needle in haystack and returns the offset or -1 */
  478. {
  479. char *s;
  480. int r = -1;
  481. if ((s = strstr(haystack, needle)) != NULL)
  482. r = s - haystack;
  483. return r;
  484. }
  485. int xs_between(const char *prefix, const char *str, const char *suffix)
  486. /* returns true if str starts with prefix and ends with suffix */
  487. {
  488. int sz = strlen(str);
  489. int psz = prefix ? strlen(prefix) : 0;
  490. int ssz = suffix ? strlen(suffix) : 0;
  491. if (sz < psz || sz < ssz)
  492. return 0;
  493. if (prefix && memcmp(str, prefix, psz) != 0)
  494. return 0;
  495. if (suffix && memcmp(str + sz - ssz, suffix, ssz) != 0)
  496. return 0;
  497. return 1;
  498. }
  499. xs_str *xs_crop_i(xs_str *str, int start, int end)
  500. /* crops the string to be only from start to end */
  501. {
  502. int sz = strlen(str);
  503. if (end <= 0)
  504. end = sz + end;
  505. /* crop from the top */
  506. if (end >= 0 && end < sz)
  507. str[end] = '\0';
  508. /* crop from the bottom */
  509. str = xs_collapse(str, 0, start);
  510. return str;
  511. }
  512. xs_str *xs_lstrip_chars_i(xs_str *str, const char *chars)
  513. /* strips all chars from the start of str */
  514. {
  515. int n;
  516. for (n = 0; str[n] && strchr(chars, str[n]); n++);
  517. if (n)
  518. str = xs_collapse(str, 0, n);
  519. return str;
  520. }
  521. xs_str *xs_rstrip_chars_i(xs_str *str, const char *chars)
  522. /* strips all chars from the end of str */
  523. {
  524. int n;
  525. for (n = strlen(str); n > 0 && strchr(chars, str[n - 1]); n--);
  526. str[n] = '\0';
  527. return str;
  528. }
  529. xs_str *xs_strip_chars_i(xs_str *str, const char *chars)
  530. /* strips the string of chars from the start and the end */
  531. {
  532. return xs_lstrip_chars_i(xs_rstrip_chars_i(str, chars), chars);
  533. }
  534. xs_str *xs_tolower_i(xs_str *str)
  535. /* convert to lowercase */
  536. {
  537. XS_ASSERT_TYPE(str, XSTYPE_STRING);
  538. int n;
  539. for (n = 0; str[n]; n++)
  540. str[n] = tolower(str[n]);
  541. return str;
  542. }
  543. xs_str *xs_toupper_i(xs_str *str)
  544. /* convert to lowercase */
  545. {
  546. XS_ASSERT_TYPE(str, XSTYPE_STRING);
  547. int n;
  548. for (n = 0; str[n]; n++)
  549. str[n] = toupper(str[n]);
  550. return str;
  551. }
  552. /** lists **/
  553. xs_list *xs_list_new(void)
  554. /* creates a new list */
  555. {
  556. int sz = 1 + _XS_TYPE_SIZE + 1;
  557. xs_list *l = xs_realloc(NULL, sz);
  558. memset(l, '\0', sz);
  559. l[0] = XSTYPE_LIST;
  560. _xs_put_size(l, sz);
  561. return l;
  562. }
  563. xs_list *_xs_list_write_litem(xs_list *list, int offset, const char *mem, int dsz)
  564. /* writes a list item */
  565. {
  566. XS_ASSERT_TYPE(list, XSTYPE_LIST);
  567. if (mem == NULL) {
  568. mem = xs_stock(XSTYPE_NULL);
  569. dsz = xs_size(mem);
  570. }
  571. list = xs_expand(list, offset, dsz + 1);
  572. list[offset] = XSTYPE_LITEM;
  573. memcpy(list + offset + 1, mem, dsz);
  574. return list;
  575. }
  576. xs_list *xs_list_append_m(xs_list *list, const char *mem, int dsz)
  577. /* adds a memory block to the list */
  578. {
  579. XS_ASSERT_TYPE(list, XSTYPE_LIST);
  580. return _xs_list_write_litem(list, xs_size(list) - 1, mem, dsz);
  581. }
  582. xs_list *_xs_list_append(xs_list *list, const xs_val *vals[])
  583. /* adds several values to the list */
  584. {
  585. /* special case: if the first argument is NULL, just insert it */
  586. if (*vals == NULL)
  587. return xs_list_append_m(list, NULL, 0);
  588. while (*vals) {
  589. list = xs_list_append_m(list, *vals, xs_size(*vals));
  590. vals++;
  591. }
  592. return list;
  593. }
  594. int xs_list_iter(xs_list **list, const xs_val **value)
  595. /* iterates a list value */
  596. {
  597. int goon = 1;
  598. xs_val *p = *list;
  599. /* skip the start of the list */
  600. if (xs_type(p) == XSTYPE_LIST)
  601. p += 1 + _XS_TYPE_SIZE;
  602. /* an element? */
  603. if (xs_type(p) == XSTYPE_LITEM) {
  604. p++;
  605. *value = p;
  606. p += xs_size(*value);
  607. }
  608. else {
  609. /* end of list */
  610. goon = 0;
  611. }
  612. /* store back the pointer */
  613. *list = p;
  614. return goon;
  615. }
  616. int xs_list_next(const xs_list *list, const xs_val **value, int *ctxt)
  617. /* iterates a list, with context */
  618. {
  619. if (xs_type(list) != XSTYPE_LIST)
  620. return 0;
  621. int goon = 1;
  622. const char *p = list;
  623. /* skip the start of the list */
  624. if (*ctxt == 0)
  625. *ctxt = 1 + _XS_TYPE_SIZE;
  626. p += *ctxt;
  627. /* an element? */
  628. if (xs_type(p) == XSTYPE_LITEM) {
  629. p++;
  630. *value = p;
  631. p += xs_size(*value);
  632. }
  633. else {
  634. /* end of list */
  635. goon = 0;
  636. }
  637. /* update the context */
  638. *ctxt = p - list;
  639. return goon;
  640. }
  641. int xs_list_cap(xs_list *list, int max)
  642. /* caps the list to have a maximum of max items */
  643. {
  644. int n, ctxt = 0;
  645. const char *v;
  646. /* advance upto max elements */
  647. for (n = 0; n < max + 1 && xs_list_next(list, &v, &ctxt); n++);
  648. /* more than max items? */
  649. if (n == max + 1) {
  650. /* insert an end of list over the XSTYPE_LITEM of the old item */
  651. char *p = (char *)v;
  652. p[-1] = '\0';
  653. /* rewrite the list len */
  654. _xs_put_size(list, p - list);
  655. }
  656. return n == max + 1;
  657. }
  658. xs_list *xs_list_reverse(const xs_list *l)
  659. /* creates a new list as a reverse version of l */
  660. {
  661. xs_list *n = xs_dup(l);
  662. const xs_val *v;
  663. /* move to one byte before the EOM */
  664. char *p = n + xs_size(n) - 1;
  665. xs_list_foreach(l, v) {
  666. /* size of v, plus the LITEM */
  667. int z = xs_size(v) + 1;
  668. p -= z;
  669. /* copy v, including its LITEM */
  670. memcpy(p, v - 1, z);
  671. }
  672. return n;
  673. }
  674. int xs_list_len(const xs_list *list)
  675. /* returns the number of elements in the list */
  676. {
  677. XS_ASSERT_TYPE_NULL(list, XSTYPE_LIST);
  678. int c = 0;
  679. const xs_val *v;
  680. xs_list_foreach(list, v)
  681. c++;
  682. return c;
  683. }
  684. const xs_val *xs_list_get(const xs_list *list, int num)
  685. /* returns the element #num */
  686. {
  687. XS_ASSERT_TYPE(list, XSTYPE_LIST);
  688. if (num < 0)
  689. num = xs_list_len(list) + num;
  690. int c = 0;
  691. const xs_val *v;
  692. xs_list_foreach(list, v) {
  693. if (c == num)
  694. return v;
  695. c++;
  696. }
  697. return NULL;
  698. }
  699. xs_list *xs_list_del(xs_list *list, int num)
  700. /* deletes element #num */
  701. {
  702. XS_ASSERT_TYPE(list, XSTYPE_LIST);
  703. const xs_val *v;
  704. if ((v = xs_list_get(list, num)) != NULL)
  705. list = xs_collapse(list, v - 1 - list, xs_size(v - 1));
  706. return list;
  707. }
  708. xs_list *xs_list_insert(xs_list *list, int num, const xs_val *data)
  709. /* inserts an element at #num position */
  710. {
  711. XS_ASSERT_TYPE(list, XSTYPE_LIST);
  712. const xs_val *v;
  713. int offset;
  714. if ((v = xs_list_get(list, num)) != NULL)
  715. offset = v - list;
  716. else
  717. offset = xs_size(list);
  718. return _xs_list_write_litem(list, offset - 1, data, xs_size(data));
  719. }
  720. xs_list *xs_list_set(xs_list *list, int num, const xs_val *data)
  721. /* sets the element at #num position */
  722. {
  723. XS_ASSERT_TYPE(list, XSTYPE_LIST);
  724. list = xs_list_del(list, num);
  725. list = xs_list_insert(list, num, data);
  726. return list;
  727. }
  728. xs_list *xs_list_dequeue(xs_list *list, xs_val **data, int last)
  729. /* gets a copy of the first or last element of a list, shrinking it */
  730. {
  731. XS_ASSERT_TYPE(list, XSTYPE_LIST);
  732. int ct = 0;
  733. const xs_val *v = NULL;
  734. if (!last) {
  735. /* get the first */
  736. xs_list_next(list, &v, &ct);
  737. }
  738. else {
  739. /* iterate to the end */
  740. while (xs_list_next(list, &v, &ct));
  741. }
  742. if (v != NULL) {
  743. *data = xs_dup(v);
  744. /* collapse from the address of the element */
  745. list = xs_collapse(list, v - 1 - list, xs_size(v - 1));
  746. }
  747. return list;
  748. }
  749. int xs_list_in(const xs_list *list, const xs_val *val)
  750. /* returns the position of val in list or -1 */
  751. {
  752. XS_ASSERT_TYPE_NULL(list, XSTYPE_LIST);
  753. int n = 0;
  754. const xs_val *v;
  755. xs_list_foreach(list, v) {
  756. if (xs_cmp(val, v) == 0)
  757. return n;
  758. n++;
  759. }
  760. return -1;
  761. }
  762. xs_str *xs_join(const xs_list *list, const char *sep)
  763. /* joins a list into a string */
  764. {
  765. XS_ASSERT_TYPE(list, XSTYPE_LIST);
  766. xs_str *s = NULL;
  767. const xs_val *v;
  768. int c = 0;
  769. int offset = 0;
  770. int ssz = strlen(sep);
  771. xs_list_foreach(list, v) {
  772. /* refuse to join non-string values */
  773. if (xs_type(v) == XSTYPE_NUMBER)
  774. v = xs_number_str(v);
  775. if (xs_type(v) == XSTYPE_STRING) {
  776. int sz;
  777. /* add the separator */
  778. if (c != 0 && ssz) {
  779. s = xs_realloc(s, offset + ssz);
  780. memcpy(s + offset, sep, ssz);
  781. offset += ssz;
  782. }
  783. /* add the element */
  784. if ((sz = strlen(v)) > 0) {
  785. s = xs_realloc(s, offset + sz);
  786. memcpy(s + offset, v, sz);
  787. offset += sz;
  788. }
  789. c++;
  790. }
  791. }
  792. /* null-terminate */
  793. s = xs_realloc(s, _xs_blk_size(offset + 1));
  794. s[offset] = '\0';
  795. return s;
  796. }
  797. xs_list *xs_split_n(const char *str, const char *sep, int times)
  798. /* splits a string into a list upto n times */
  799. {
  800. xs_list *list = xs_list_new();
  801. if (!xs_is_string(str) || !xs_is_string(sep))
  802. return list;
  803. int sz = strlen(sep);
  804. char *ss;
  805. while (times > 0 && (ss = strstr(str, sep)) != NULL) {
  806. /* create a new string with this slice and add it to the list */
  807. xs *s = xs_str_new_sz(str, ss - str);
  808. if (xs_is_string(s))
  809. list = xs_list_append(list, s);
  810. /* skip past the separator */
  811. str = ss + sz;
  812. times--;
  813. }
  814. /* add the rest of the string */
  815. if (xs_is_string(str))
  816. list = xs_list_append(list, str);
  817. return list;
  818. }
  819. xs_list *xs_list_cat(xs_list *l1, const xs_list *l2)
  820. /* concatenates list l2 to l1 */
  821. {
  822. XS_ASSERT_TYPE(l1, XSTYPE_LIST);
  823. XS_ASSERT_TYPE(l2, XSTYPE_LIST);
  824. /* inserts at the end of l1 the content of l2 (skipping header and footer) */
  825. return xs_insert_m(l1, xs_size(l1) - 1,
  826. l2 + 1 + _XS_TYPE_SIZE, xs_size(l2) - (1 + _XS_TYPE_SIZE + 1));
  827. }
  828. /** keyvals **/
  829. int xs_keyval_size(const xs_str *key, const xs_val *value)
  830. /* returns the needed size for a keyval */
  831. {
  832. return 1 + xs_size(key) + xs_size(value);
  833. }
  834. xs_str *xs_keyval_key(const xs_keyval *keyval)
  835. /* returns a pointer to the key of the keyval */
  836. {
  837. return (xs_str *)&keyval[1];
  838. }
  839. xs_val *xs_keyval_value(const xs_keyval *keyval)
  840. /* returns a pointer to the value of the keyval */
  841. {
  842. return (xs_val *)&keyval[1 + xs_size(xs_keyval_key(keyval))];
  843. }
  844. xs_keyval *xs_keyval_make(xs_keyval *keyval, const xs_str *key, const xs_val *value)
  845. /* builds a keyval into mem (should have enough size) */
  846. {
  847. keyval[0] = XSTYPE_KEYVAL;
  848. memcpy(xs_keyval_key(keyval), key, xs_size(key));
  849. memcpy(xs_keyval_value(keyval), value, xs_size(value));
  850. return keyval;
  851. }
  852. /** dicts **/
  853. typedef struct {
  854. int value_offset; /* offset to value (from dict start) */
  855. int next; /* next node in sequential scanning */
  856. int child[4]; /* child nodes in hashed search */
  857. char key[]; /* C string key */
  858. } ditem_hdr;
  859. typedef struct {
  860. int size; /* size of full dict (_XS_TYPE_SIZE) */
  861. int first; /* first node for sequential scanning */
  862. int last; /* last node for sequential scanning */
  863. int root; /* root node for hashed search */
  864. /* a bunch of ditem_hdr and value follows */
  865. } dict_hdr;
  866. xs_dict *xs_dict_new(void)
  867. /* creates a new dict */
  868. {
  869. /* size of dict */
  870. int sz = 1 + sizeof(dict_hdr);
  871. xs_dict *d = xs_realloc(NULL, sz);
  872. memset(d, '\0', sz);
  873. d[0] = XSTYPE_DICT;
  874. _xs_put_size(d, sz);
  875. return d;
  876. }
  877. static int *_xs_dict_locate(const xs_dict *dict, const char *key)
  878. /* locates a ditem */
  879. {
  880. unsigned int h = xs_hash_func(key, strlen(key));
  881. /* start from the root */
  882. dict_hdr *dh = (dict_hdr *)(dict + 1);
  883. int *off = &dh->root;
  884. while (*off) {
  885. /* pointer to ditem */
  886. ditem_hdr *di = (ditem_hdr *)(dict + *off);
  887. /* pointer to the key */
  888. const char *d_key = di->key;
  889. if (strcmp(key, d_key) == 0)
  890. break;
  891. off = &di->child[h >> 30];
  892. h <<= 2;
  893. }
  894. return off;
  895. }
  896. xs_dict *xs_dict_set(xs_dict *dict, const xs_str *key, const xs_val *value)
  897. /* sets a key/value pair */
  898. {
  899. if (value == NULL)
  900. value = xs_stock(XSTYPE_NULL);
  901. if (xs_type(dict) == XSTYPE_DICT) {
  902. int *o = _xs_dict_locate(dict, key);
  903. int end = xs_size(dict);
  904. if (!*o) {
  905. /* ditem does not exist yet: append to the end */
  906. *o = end;
  907. int ksz = xs_size(key);
  908. int vsz = xs_size(value);
  909. int dsz = sizeof(ditem_hdr) + ksz + vsz;
  910. /* open room in the dict for the full ditem */
  911. dict = xs_expand(dict, end, dsz);
  912. dict_hdr *dh = (dict_hdr *)(dict + 1);
  913. /* build the ditem */
  914. ditem_hdr *di = (ditem_hdr *)(dict + end);
  915. memset(di, '\0', dsz);
  916. /* set the offset to the value */
  917. di->value_offset = end + sizeof(ditem_hdr) + ksz;
  918. /* copy the key */
  919. memcpy(di->key, key, ksz);
  920. /* copy the value */
  921. memcpy(dict + di->value_offset, value, vsz);
  922. /* chain to the sequential list */
  923. if (dh->first == 0)
  924. dh->first = end;
  925. else {
  926. /* chain this new element to the last one */
  927. ditem_hdr *dil = (ditem_hdr *)(dict + dh->last);
  928. dil->next = end;
  929. }
  930. dh->last = end;
  931. }
  932. else {
  933. /* ditem already exists */
  934. ditem_hdr *di = (ditem_hdr *)(dict + *o);
  935. /* get pointer to the value offset */
  936. int *i = &di->value_offset;
  937. /* deleted? recover offset */
  938. if (*i < 0)
  939. *i *= -1;
  940. /* get old value */
  941. xs_val *o_value = dict + *i;
  942. /* will new value fit over the old one? */
  943. if (xs_size(value) <= xs_size(o_value)) {
  944. /* just overwrite */
  945. /* (difference is leaked inside the dict) */
  946. memcpy(o_value, value, xs_size(value));
  947. }
  948. else {
  949. /* not enough room: new value will live at the end of the dict */
  950. /* (old value is leaked inside the dict) */
  951. *i = end;
  952. dict = xs_insert(dict, end, value);
  953. }
  954. }
  955. }
  956. return dict;
  957. }
  958. xs_dict *xs_dict_append(xs_dict *dict, const xs_str *key, const xs_val *value)
  959. /* just an alias (for this implementation it's the same) */
  960. {
  961. return xs_dict_set(dict, key, value);
  962. }
  963. xs_dict *xs_dict_prepend(xs_dict *dict, const xs_str *key, const xs_val *value)
  964. /* just an alias (for this implementation it's the same) */
  965. {
  966. return xs_dict_set(dict, key, value);
  967. }
  968. xs_dict *xs_dict_del(xs_dict *dict, const xs_str *key)
  969. /* deletes a key/value pair */
  970. {
  971. if (xs_type(dict) == XSTYPE_DICT) {
  972. int *o = _xs_dict_locate(dict, key);
  973. if (*o) {
  974. /* found ditem */
  975. ditem_hdr *di = (ditem_hdr *)(dict + *o);
  976. /* deleted ditems have a negative value offset */
  977. di->value_offset *= -1;
  978. }
  979. }
  980. return dict;
  981. }
  982. const xs_val *xs_dict_get(const xs_dict *dict, const xs_str *key)
  983. /* gets a value by key, or NULL */
  984. {
  985. if (xs_type(dict) == XSTYPE_DICT) {
  986. int *o = _xs_dict_locate(dict, key);
  987. if (*o) {
  988. /* found ditem */
  989. ditem_hdr *di = (ditem_hdr *)(dict + *o);
  990. if (di->value_offset > 0)
  991. return dict + di->value_offset;
  992. }
  993. }
  994. return NULL;
  995. }
  996. int xs_dict_next(const xs_dict *dict, const xs_str **key, const xs_val **value, int *ctxt)
  997. /* dict iterator, with context */
  998. {
  999. if (xs_type(dict) != XSTYPE_DICT)
  1000. return 0;
  1001. if (*ctxt == 0) {
  1002. /* at the beginning: get the first sequential item */
  1003. const dict_hdr *dh = (dict_hdr *)(dict + 1);
  1004. *ctxt = dh->first;
  1005. }
  1006. *value = NULL;
  1007. while (*value == NULL && *ctxt > 0) {
  1008. const ditem_hdr *di = (ditem_hdr *)(dict + *ctxt);
  1009. /* get value */
  1010. if (di->value_offset > 0) {
  1011. *value = (xs_val *)(dict + di->value_offset);
  1012. /* get key */
  1013. *key = (xs_str *)&di->key;
  1014. }
  1015. /* get offset to next ditem */
  1016. *ctxt = di->next ? di->next : -1;
  1017. }
  1018. return *value != NULL;
  1019. }
  1020. xs_dict *xs_dict_gc(const xs_dict *dict)
  1021. /* creates a copy of dict, but garbage-collected */
  1022. {
  1023. xs_dict *nd = xs_dict_new();
  1024. const xs_str *k;
  1025. const xs_val *v;
  1026. xs_dict_foreach(dict, k, v) {
  1027. if (xs_type(v) == XSTYPE_DICT) {
  1028. xs *sd = xs_dict_gc(v);
  1029. nd = xs_dict_set(nd, k, sd);
  1030. }
  1031. else
  1032. nd = xs_dict_set(nd, k, v);
  1033. }
  1034. return nd;
  1035. }
  1036. const xs_val *xs_dict_get_path_sep(const xs_dict *dict, const char *path, const char *sep)
  1037. /* gets a value from dict given a path separated by sep */
  1038. {
  1039. /* split by the separator */
  1040. xs *l = xs_split_n(path, sep, 1);
  1041. /* only one part? just get */
  1042. if (xs_list_len(l) == 1)
  1043. return xs_dict_get(dict, path);
  1044. const char *prefix = xs_list_get(l, 0);
  1045. const char *rest = xs_list_get(l, 1);
  1046. const xs_dict *sd = xs_dict_get(dict, prefix);
  1047. if (xs_type(sd) == XSTYPE_DICT)
  1048. return xs_dict_get_path_sep(sd, rest, sep);
  1049. return NULL;
  1050. }
  1051. xs_dict *xs_dict_set_path_sep(xs_dict *dict, const char *path, const xs_val *value, const char *sep)
  1052. /* sets a value into dict given a path separated by sep;
  1053. intermediate dicts are created if needed */
  1054. {
  1055. /* split by the separator */
  1056. xs *l = xs_split_n(path, sep, 1);
  1057. /* only one part? just set */
  1058. if (xs_list_len(l) == 1)
  1059. return xs_dict_set(dict, path, value);
  1060. const char *prefix = xs_list_get(l, 0);
  1061. const char *rest = xs_list_get(l, 1);
  1062. xs *nd = NULL;
  1063. /* does the first part of path exist? */
  1064. const xs_dict *cd = xs_dict_get(dict, prefix);
  1065. if (xs_type(cd) == XSTYPE_DICT)
  1066. nd = xs_dup(cd);
  1067. else
  1068. nd = xs_dict_new();
  1069. /* move down the path */
  1070. nd = xs_dict_set_path_sep(nd, rest, value, sep);
  1071. /* set */
  1072. return xs_dict_set(dict, prefix, nd);
  1073. }
  1074. /** other values **/
  1075. xs_val *xs_val_new(xstype t)
  1076. /* adds a new special value */
  1077. {
  1078. xs_val *v = xs_realloc(NULL, _xs_blk_size(1));
  1079. v[0] = t;
  1080. return v;
  1081. }
  1082. /** numbers */
  1083. xs_number *_xs_number_new(const char *str)
  1084. {
  1085. xs_number *v = xs_realloc(NULL, 1 + xs_size(str));
  1086. v[0] = XSTYPE_NUMBER;
  1087. memcpy(&v[1], str, xs_size(str));
  1088. return v;
  1089. }
  1090. xs_number *xs_number_new(double f)
  1091. /* creates a new number value from a double */
  1092. {
  1093. char tmp[64];
  1094. snprintf(tmp, sizeof(tmp), "%.15lf", f);
  1095. /* strip useless zeros */
  1096. if (strchr(tmp, '.') != NULL) {
  1097. char *ptr;
  1098. for (ptr = tmp + strlen(tmp) - 1; *ptr == '0'; ptr--);
  1099. if (*ptr != '.')
  1100. ptr++;
  1101. *ptr = '\0';
  1102. }
  1103. return _xs_number_new(tmp);
  1104. }
  1105. xs_number *xs_number_new_l(long l)
  1106. /* creates a new number value from a long */
  1107. {
  1108. char tmp[64];
  1109. snprintf(tmp, sizeof(tmp), "%ld", l);
  1110. return _xs_number_new(tmp);
  1111. }
  1112. double xs_number_get(const xs_number *v)
  1113. /* gets the number as a double */
  1114. {
  1115. if (xs_type(v) == XSTYPE_NUMBER)
  1116. v++;
  1117. if (xs_type(v) == XSTYPE_STRING)
  1118. return atof(v);
  1119. return 0.0;
  1120. }
  1121. long xs_number_get_l(const xs_number *v)
  1122. /* gets the number as a long */
  1123. {
  1124. if (xs_type(v) == XSTYPE_NUMBER)
  1125. v++;
  1126. if (xs_type(v) == XSTYPE_STRING)
  1127. return atol(v);
  1128. return 0;
  1129. }
  1130. const char *xs_number_str(const xs_number *v)
  1131. /* gets the number as a string */
  1132. {
  1133. if (xs_type(v) == XSTYPE_NUMBER)
  1134. return v + 1;
  1135. return NULL;
  1136. }
  1137. /** raw data blocks **/
  1138. xs_data *xs_data_new(const void *data, int size)
  1139. /* returns a new raw data value */
  1140. {
  1141. xs_data *v;
  1142. /* add the overhead (data type + size) */
  1143. int total_size = size + 1 + _XS_TYPE_SIZE;
  1144. v = xs_realloc(NULL, _xs_blk_size(total_size));
  1145. v[0] = XSTYPE_DATA;
  1146. _xs_put_size(v, total_size);
  1147. memcpy(&v[1 + _XS_TYPE_SIZE], data, size);
  1148. return v;
  1149. }
  1150. int xs_data_size(const xs_data *value)
  1151. /* returns the size of the data stored inside value */
  1152. {
  1153. return _xs_get_size(value) - (1 + _XS_TYPE_SIZE);
  1154. }
  1155. const void *xs_data_ptr(const xs_data *value)
  1156. /* return an internal pointer to the stored data */
  1157. {
  1158. return &value[1 + _XS_TYPE_SIZE];
  1159. }
  1160. void xs_data_get(void *data, const xs_data *value)
  1161. /* copies the raw data stored inside value into data */
  1162. {
  1163. memcpy(data, xs_data_ptr(value), xs_data_size(value));
  1164. }
  1165. void *xs_memmem(const char *haystack, int h_size, const char *needle, int n_size)
  1166. /* clone of memmem */
  1167. {
  1168. char *p, *r = NULL;
  1169. int offset = 0;
  1170. while (!r && h_size - offset > n_size &&
  1171. (p = memchr(haystack + offset, *needle, h_size - offset))) {
  1172. if (memcmp(p, needle, n_size) == 0)
  1173. r = p;
  1174. else
  1175. offset = p - haystack + 1;
  1176. }
  1177. return r;
  1178. }
  1179. unsigned int xs_hash_func(const char *data, int size)
  1180. /* a general purpose hashing function */
  1181. {
  1182. unsigned int hash = 0x666;
  1183. for (int n = 0; n < size; n++) {
  1184. hash ^= (unsigned char)data[n];
  1185. hash *= 111111111;
  1186. }
  1187. return hash ^ hash >> 16;
  1188. }
  1189. uint64_t xs_hash64_func(const char *data, int size)
  1190. /* a general purpose hashing function (64 bit) */
  1191. {
  1192. uint64_t hash = 0x100;
  1193. for (int n = 0; n < size; n++) {
  1194. hash ^= (unsigned char)data[n];
  1195. hash *= 1111111111111111111;
  1196. }
  1197. return hash;
  1198. }
  1199. #endif /* XS_IMPLEMENTATION */
  1200. #endif /* _XS_H */