xs.h 36 KB

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