kstring.h

   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
  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
  58 static inline int kputs(const char *p, kstring_t *s)
  59 {
  60         return kputsn(p, strlen(p), s);
  61 }
  62
  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
  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
  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
 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