lib/stkstring.c

   1 #include "lib/lib.h"
   2 #include "lib/stkstring.h"
   3
   4 #include <stdio.h>
   5
   6 uns
   7 stk_array_len(char **s, uns cnt)
   8 {
   9   uns l = 1;
  10   while (cnt--)
  11     l += strlen(*s++);
  12   return l;
  13 }
  14
  15 void
  16 stk_array_join(char *x, char **s, uns cnt, uns sep)
  17 {
  18   while (cnt--)
  19     {
  20       uns l = strlen(*s);
  21       memcpy(x, *s, l);
  22       x += l;
  23       s++;
  24       if (sep && cnt)
  25         *x++ = sep;
  26     }
  27   *x = 0;
  28 }
  29
  30 uns
  31 stk_printf_internal(const char *fmt, ...)
  32 {
  33   uns len = 256;
  34   char *buf = alloca(len);
  35   va_list args, args2;
  36   va_start(args, fmt);
  37   for (;;)
  38     {
  39       va_copy(args2, args);
  40       int l = vsnprintf(buf, len, fmt, args2);
  41       va_end(args2);
  42       if (l < 0)
  43         len *= 2;
  44       else
  45         {
  46           va_end(args);
  47           return l+1;
  48         }
  49       buf = alloca(len);
  50     }
  51 }
  52
  53 uns
  54 stk_vprintf_internal(const char *fmt, va_list args)
  55 {
  56   uns len = 256;
  57   char *buf = alloca(len);
  58   va_list args2;
  59   for (;;)
  60     {
  61       va_copy(args2, args);
  62       int l = vsnprintf(buf, len, fmt, args2);
  63       va_end(args2);
  64       if (l < 0)
  65         len *= 2;
  66       else
  67         {
  68           va_end(args);
  69           return l+1;
  70         }
  71       buf = alloca(len);
  72     }
  73 }
  74
  75 void
  76 stk_hexdump_internal(char *dst, byte *src, uns n)
  77 {
  78   for (uns i=0; i<n; i++)
  79     {
  80       if (i)
  81         *dst++ = ' ';
  82       dst += sprintf(dst, "%02x", *src++);
  83     }
  84   *dst = 0;
  85 }
  86
  87 #ifdef TEST
  88
  89 int main(void)
  90 {
  91   char *a = stk_strndup("are!",3);
  92   a = stk_strcat(a, " the ");
  93   a = stk_strmulticat(a, stk_strdup("Jabberwock, "), "my", NULL);
  94   char *arr[] = { a, " son" };
  95   a = stk_strarraycat(arr, 2);
  96   a = stk_printf("Bew%s!", a);
  97   puts(a);
  98   puts(stk_hexdump(a, 3));
  99   char *ary[] = { "The", "jaws", "that", "bite" };
 100   puts(stk_strjoin(ary, 4, ' '));
 101   return 0;
 102 }
 103
 104 #endif