Fixed stk_strndup(): it works on non-terminated input strings now

and without an off-by-one bug :)

diff --git a/lib/stkstring.c b/lib/stkstring.c
index c7ac9ef5b75f528eef06862510a304c94a299a91..ccc689da64ba04e10f656631a1895f457520ba32 100644 (file)
--- a/lib/stkstring.c
+++ b/lib/stkstring.c
@@ -72,9 +72,9 @@ stk_hexdump_internal(char *dst, byte *src, uns n)
 
 int main(void)
 {
-  char *a = stk_strdup("are");
+  char *a = stk_strndup("are!",3);
   a = stk_strcat(a, " the ");
-  a = stk_strmulticat(a, "Jabberwock, ", "my", NULL);
+  a = stk_strmulticat(a, stk_strdup("Jabberwock, "), "my", NULL);
   char *arr[] = { a, " son" };
   a = stk_strarraycat(arr, 2);
   a = stk_printf("Bew%s!", a);

diff --git a/lib/stkstring.h b/lib/stkstring.h
index 7314d9832ed987e1b83ca72cc1a6f5202969a237..8f58aec736b87012c1990fde567e118de8e6c348 100644 (file)
--- a/lib/stkstring.h
+++ b/lib/stkstring.h
@@ -1,7 +1,7 @@
 /*
  *     UCW Library -- Strings Allocated on the Stack
  *
- *     (c) 2005 Martin Mares <mj@ucw.cz>
+ *     (c) 2005--2006 Martin Mares <mj@ucw.cz>
  *     (c) 2005 Tomas Valla <tom@ucw.cz>
  *
  *     This software may be freely distributed and used according to the terms
@@ -12,7 +12,7 @@
 #include <string.h>
 
 #define stk_strdup(s) ({ char *_s=(s); uns _l=strlen(_s)+1; char *_x=alloca(_l); memcpy(_x, _s, _l); _x; })
-#define stk_strndup(s,n) ({ char *_s=(s); uns _n=(n); uns _k = strlen(_s); uns _l=MIN(_n,_k)+1; char *_x=alloca(_l); memcpy(_x, _s, _l); _x[_l]=0; _x; })
+#define stk_strndup(s,n) ({ char *_s=(s); uns _l=strnlen(_s,(n)); char *_x=alloca(_l+1); memcpy(_x, _s, _l); _x[_l]=0; _x; })
 #define stk_strcat(s1,s2) ({ char *_s1=(s1); char *_s2=(s2); uns _l1=strlen(_s1); uns _l2=strlen(_s2); char *_x=alloca(_l1+_l2+1); memcpy(_x,_s1,_l1); memcpy(_x+_l1,_s2,_l2+1); _x; })
 #define stk_strmulticat(s...) ({ char *_s[]={s}; char *_x=alloca(stk_array_len(_s, ARRAY_SIZE(_s)-1)); stk_array_copy(_x, _s, ARRAY_SIZE(_s)-1); _x; })
 #define stk_strarraycat(s,n) ({ char **_s=(s); int _n=(n); char *_x=alloca(stk_array_len(_s,_n)); stk_array_copy(_x, _s, _n); _x; })