pkstr: clone functions if PK_IMPL_MEM defined

1 files changed, 26 insertions, 0 deletions

diff --git a/pkstr.h b/pkstr.h
index 3ba0630..f4814eb 100644
--- a/pkstr.h
+++ b/pkstr.h
@@ -1,6 +1,8 @@
 #ifndef PK_STR_H
 #define PK_STR_H
 
+#include "pkmem.h" /* deleteme */
+
 #include <stdint.h>
 
 struct pk_str {
@@ -18,6 +20,8 @@ struct pk_str cstring_to_pk_str(char *s);
 struct pk_cstr cstring_to_pk_cstr(const char *s);
 struct pk_str pk_cstr_to_pk_str(const struct pk_cstr *s);
 struct pk_cstr pk_str_to_pk_cstr(const struct pk_str *s);
+struct pk_str pk_str_clone(const struct pk_str *s, struct pk_membucket *bkt);
+struct pk_cstr pk_cstr_clone(const struct pk_cstr *s, struct pk_membucket *bkt);
 int pk_compare_str(const struct pk_str *lhs, const struct pk_str *rhs);
 int pk_compare_cstr(const struct pk_cstr *lhs, const struct pk_cstr *rhs);
 
@@ -68,6 +72,28 @@ pk_str_to_pk_cstr(const struct pk_str *s)
 	return ret;
 }
 
+struct pk_str
+pk_str_clone(const struct pk_str *s, struct pk_membucket *bkt) {
+	struct pk_str str;
+	str.length = s->length == 0 ? strlen(s->val) : s->length;
+	str.reserved = s->length + 1;
+	char *ss = (char*)pk_new(str.reserved * sizeof(char), alignof(char), bkt);
+	strncpy(ss, s->val, str.reserved);
+	str.val = ss;
+	return str;
+}
+
+struct pk_cstr
+pk_cstr_clone(const struct pk_cstr *s, struct pk_membucket *bkt) {
+	struct pk_cstr str;
+	str.length = s->length == 0 ? strlen(s->val) : s->length;
+	str.reserved = s->length + 1;
+	char *ss = (char*)pk_new(str.reserved * sizeof(char), alignof(char), bkt);
+	strncpy(ss, s->val, str.reserved);
+	str.val = ss;
+	return str;
+}
+
 int
 pk_compare_str(const struct pk_str *lhs, const struct pk_str *rhs)
 {