1 files changed, 10 insertions, 5 deletions

diff --git a/src/sp_config_utils.c b/src/sp_config_utils.c
index 130f07e..d72561b 100644
--- a/src/sp_config_utils.c
+++ b/src/sp_config_utils.c
@@ -32,18 +32,18 @@ int parse_keywords(sp_config_functions *funcs, char *line) {
   return 0;
 }
 
-char *get_param(size_t *consumed, char *restrict line, sp_type type,
-                const char *restrict keyword) {
+zend_string *get_param(size_t *consumed, char *restrict line, sp_type type,
+                       const char *restrict keyword) {
   enum { IN_ESCAPE, NONE } state = NONE;
   char *original_line = line;
   size_t j = 0;
 
-  char *ret = NULL;
+  zend_string *ret = NULL;
   if (NULL == line || '\0' == *line) {
     goto err;
   }
 
-  ret = pecalloc(sizeof(char), strlen(original_line) + 1, 1);
+  ret = zend_string_alloc(strlen(line) + 1, 1);
 
   /* The first char of a string is always '"', since they MUST be quoted. */
   if ('"' == *line) {
@@ -65,6 +65,11 @@ char *get_param(size_t *consumed, char *restrict line, sp_type type,
            2. the SP_TOKEN_END_PARAM
            */
           *consumed = i + 2;
+          // Make sure that the string we return is the right size,
+          // as it can be smaller than strlen(line)
+          ret = zend_string_truncate(ret, j, 1);
+          // truncate does not add a \0
+          ZSTR_VAL(ret)[ZSTR_LEN(ret)] = 0;
           return ret;
         } else if (state == IN_ESCAPE) {
           break;  // we're on an escped double quote
@@ -82,7 +87,7 @@ char *get_param(size_t *consumed, char *restrict line, sp_type type,
     if (state == IN_ESCAPE) {
       state = NONE;
     }
-    ret[j++] = line[i];
+    ZSTR_VAL(ret)[j++] = line[i];
   }
 err:
   if (0 == j) {