fmt/pl: add pl_alloc_str

include/re_fmt.h

@@ -29,6 +29,7 @@ struct pl {
 
 extern const struct pl pl_null;
 
+struct pl *pl_alloc_str(const char *str);
 void     pl_set_str(struct pl *pl, const char *str);
 void     pl_set_mbuf(struct pl *pl, const struct mbuf *mb);
 int32_t  pl_i32(const struct pl *pl);

src/fmt/pl.c

@@ -21,6 +21,49 @@
 const struct pl pl_null = {NULL, 0};
 
 
+static void pl_alloc_destruct(void *arg)
+{
+	struct pl *pl = arg;
+
+	mem_deref((void *)pl->p);
+}
+
+
+/**
+ * Allocate a pointer-length object from a NULL-terminated string
+ *
+ * @param pl  Pointer-length object to be initialised
+ * @param str NULL-terminated string
+ *
+ * @return Allocated Pointer-length object or NULL
+ */
+struct pl *pl_alloc_str(const char *str)
+{
+	struct pl *pl;
+
+	if (!str)
+		return NULL;
+
+	size_t sz = strlen(str);
+
+	pl = mem_zalloc(sizeof(struct pl), pl_alloc_destruct);
+	if (!pl)
+		return NULL;
+
+	pl->p = mem_alloc(sz, NULL);
+	if (!pl->p) {
+		mem_deref(pl);
+		return NULL;
+	}
+
+	memcpy((void *)pl->p, str, sz);
+
+	pl->l = sz;
+
+	return pl;
+}
+
+
 /**
  * Initialise a pointer-length object from a NULL-terminated string
  *

test/fmt.c

@@ -138,6 +138,24 @@ int test_fmt_pl(void)
 }
 
 
+int test_fmt_pl_alloc_str(void)
+{
+	int err		= 0;
+	char test_str[] = "Test String";
+
+	struct pl *pl = pl_alloc_str(test_str);
+	if (!pl)
+		return ENOMEM;
+
+	TEST_MEMCMP(test_str, str_len(test_str), pl->p, pl->l);
+
+out:
+	mem_deref(pl);
+
+	return err;
+}
+
+
 int test_fmt_pl_i32(void)
 {
 	const struct {

test/test.c

@@ -79,6 +79,7 @@ static const struct test tests[] = {
 	TEST(test_fmt_human_time),
 	TEST(test_fmt_param),
 	TEST(test_fmt_pl),
+	TEST(test_fmt_pl_alloc_str),
 	TEST(test_fmt_pl_float),
 	TEST(test_fmt_pl_i32),
 	TEST(test_fmt_pl_i64),

test/test.h

@@ -181,6 +181,7 @@ int test_fmt_hexdump(void);
 int test_fmt_human_time(void);
 int test_fmt_param(void);
 int test_fmt_pl(void);
+int test_fmt_pl_alloc_str(void);
 int test_fmt_pl_float(void);
 int test_fmt_pl_i32(void);
 int test_fmt_pl_i64(void);