apr_buffer: Advance buffer during apr_buffer_pstrncat().

git-svn-id: https://svn.apache.org/repos/asf/apr/apr/trunk@1920293 13f79535-47bb-0310-9956-ffa450edef68
apache · Aug 30, 2024 · f1886c4 · f1886c4
1 parent 0b10ac3
commit f1886c4
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 7 deletions.
diff --git a/buffer/apr_buffer.c b/buffer/apr_buffer.c
@@ -394,6 +394,7 @@ APR_DECLARE(char *) apr_buffer_pstrncat(apr_pool_t *p, const apr_buffer_t *buf,
         else {
             if (APR_BUFFER_PLAIN == flags) {
                 memcpy(dst, src->d.mem, src->size);
+                dst += src->size;
             }
             else if (APR_BUFFER_BASE64 == flags) {
                 apr_size_t b64len;

diff --git a/test/abts_tests.h b/test/abts_tests.h
@@ -74,6 +74,7 @@ const struct testlist {
     {testuri},
     {testuuid},
     {testbuckets},
+    {testbuffer},
     {testpass},
     {testbase64},
     {testmd4},
@@ -94,7 +95,6 @@ const struct testlist {
     {testsiphash},
     {testjson},
     {testjose},
-    {testbuffer},
     {testldap}
 };
 

diff --git a/test/testbuffer.c b/test/testbuffer.c
@@ -167,7 +167,7 @@ static void test_buffers(abts_case *tc, void *data)
     apr_pool_t *pool;
     apr_array_header_t *vals;
 
-    apr_buffer_t src[2];
+    apr_buffer_t src[4];
     apr_buffer_t *dst;
 
     char *str;
@@ -180,11 +180,13 @@ static void test_buffers(abts_case *tc, void *data)
     /* populate our source buffers */
     apr_buffer_mem_set(&src[0], test_memory, sizeof(test_memory));
     apr_buffer_str_set(&src[1], test_string, strlen(test_string));
+    apr_buffer_mem_set(&src[2], test_memory, sizeof(test_memory));
+    apr_buffer_str_set(&src[3], test_string, strlen(test_string));
 
     /* duplicate the source buffers, allocating memory from a pool */
-    vals = apr_array_make(pool, 2, sizeof(apr_buffer_t));
-    apr_buffer_arraydup((apr_buffer_t **)(&vals->elts), src, test_buffers_palloc, pool, 2);
-    vals->nelts = 2;
+    vals = apr_array_make(pool, 4, sizeof(apr_buffer_t));
+    apr_buffer_arraydup((apr_buffer_t **)(&vals->elts), src, test_buffers_palloc, pool, 4);
+    vals->nelts = 4;
 
     dst = apr_array_pop(vals);
 
@@ -201,10 +203,10 @@ static void test_buffers(abts_case *tc, void *data)
     ABTS_ASSERT(tc, "buffer copy fail",
                     !apr_buffer_cmp(dst, &src[1]));
 
-    str = apr_buffer_pstrncat(pool, &src[0], 2, "; ", APR_BUFFER_BASE64, &len);
+    str = apr_buffer_pstrncat(pool, &src[0], 4, "; ", APR_BUFFER_BASE64, &len);
 
     ABTS_ASSERT(tc, "buffer strcat fail",
-                    !strcmp(str, "AQIDBA==; Hello"));
+                    !strcmp(str, "AQIDBA==; Hello; AQIDBA==; Hello"));
 
     apr_pool_destroy(pool);
 }