1 #ifndef KSTRING_H
2 #define KSTRING_H
3
4 #include <stdlib.h>
5 #include <string.h>
6 #include <stdint.h>
7
8 #ifndef kroundup32
9 #define kroundup32(x) (--(x), (x)|=(x)>>1, (x)|=(x)>>2, (x)|=(x)>>4, (x)|=(x)>>8, (x)|=(x)>>16, ++(x))
10 #endif
11
12 #ifndef KSTRING_T
13 #define KSTRING_T kstring_t
14 typedef struct __kstring_t {
15 size_t l, m;
16 char *s;
17 } kstring_t;
18 #endif
19
20 typedef struct {
21 uint64_t tab[4];
22 const char *p; // end of the current token
23 } ks_tokaux_t;
24
25 #ifdef __cplusplus
26 extern "C" {
27 #endif
28
29 int ksprintf(kstring_t *s, const char *fmt, ...);
30 int ksplit_core(char *s, int delimiter, int *_max, int **_offsets);
31 char *kstrstr(const char *str, const char *pat, int **_prep);
32 char *kstrnstr(const char *str, const char *pat, int n, int **_prep);
33 void *kmemmem(const void *_str, int n, const void *_pat, int m, int **_prep);
34
35 /* kstrtok() is similar to strtok_r() except that str is not
36 * modified and both str and sep can be NULL. For efficiency, it is
37 * actually recommended to set both to NULL in the subsequent calls
38 * if sep is not changed. */
39 char *kstrtok(const char *str, const char *sep, ks_tokaux_t *aux);
40
41 #ifdef __cplusplus
42 }
43 #endif
44
kputsn(const char * p,int l,kstring_t * s)45 static inline int kputsn(const char *p, int l, kstring_t *s)
46 {
47 if (s->l + l + 1 >= s->m) {
48 s->m = s->l + l + 2;
49 kroundup32(s->m);
50 s->s = (char*)realloc(s->s, s->m);
51 }
52 memcpy(s->s + s->l, p, l);
53 s->l += l;
54 s->s[s->l] = 0;
55 return l;
56 }
57
kputs(const char * p,kstring_t * s)58 static inline int kputs(const char *p, kstring_t *s)
59 {
60 return kputsn(p, strlen(p), s);
61 }
62
kputc(int c,kstring_t * s)63 static inline int kputc(int c, kstring_t *s)
64 {
65 if (s->l + 1 >= s->m) {
66 s->m = s->l + 2;
67 kroundup32(s->m);
68 s->s = (char*)realloc(s->s, s->m);
69 }
70 s->s[s->l++] = c;
71 s->s[s->l] = 0;
72 return c;
73 }
74
kputw(int c,kstring_t * s)75 static inline int kputw(int c, kstring_t *s)
76 {
77 char buf[16];
78 int l, x;
79 if (c == 0) return kputc('0', s);
80 for (l = 0, x = c < 0? -c : c; x > 0; x /= 10) buf[l++] = x%10 + '0';
81 if (c < 0) buf[l++] = '-';
82 if (s->l + l + 1 >= s->m) {
83 s->m = s->l + l + 2;
84 kroundup32(s->m);
85 s->s = (char*)realloc(s->s, s->m);
86 }
87 for (x = l - 1; x >= 0; --x) s->s[s->l++] = buf[x];
88 s->s[s->l] = 0;
89 return 0;
90 }
91
kputuw(unsigned c,kstring_t * s)92 static inline int kputuw(unsigned c, kstring_t *s)
93 {
94 char buf[16];
95 int l, i;
96 unsigned x;
97 if (c == 0) return kputc('0', s);
98 for (l = 0, x = c; x > 0; x /= 10) buf[l++] = x%10 + '0';
99 if (s->l + l + 1 >= s->m) {
100 s->m = s->l + l + 2;
101 kroundup32(s->m);
102 s->s = (char*)realloc(s->s, s->m);
103 }
104 for (i = l - 1; i >= 0; --i) s->s[s->l++] = buf[i];
105 s->s[s->l] = 0;
106 return 0;
107 }
108
ksplit(kstring_t * s,int delimiter,int * n)109 static inline int *ksplit(kstring_t *s, int delimiter, int *n)
110 {
111 int max = 0, *offsets = 0;
112 *n = ksplit_core(s->s, delimiter, &max, &offsets);
113 return offsets;
114 }
115
116 #endif
117