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