changes from feedback

php · May 14, 2024 · 87c67bc · 87c67bc
1 parent 5564458
commit 87c67bc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ext/pgsql/pgsql.c b/ext/pgsql/pgsql.c
@@ -3289,11 +3289,12 @@ PHP_FUNCTION(pg_copy_from)
 					if (UNEXPECTED(!tmp)) {
 						return;
 					}
-					zend_string *zquery = zend_string_alloc(ZSTR_LEN(tmp) + 2, false);
+					zend_string *zquery = zend_string_alloc(ZSTR_LEN(tmp) + 1, false);
 					memcpy(ZSTR_VAL(zquery), ZSTR_VAL(tmp), ZSTR_LEN(tmp) + 1);
-					ZSTR_VAL(zquery)[ZSTR_LEN(tmp) + 1] = 0;
+					ZSTR_LEN(zquery) = ZSTR_LEN(tmp);
 					if (ZSTR_LEN(tmp) > 0 && ZSTR_VAL(zquery)[ZSTR_LEN(tmp)]  != '\n') {
 						ZSTR_VAL(zquery)[ZSTR_LEN(tmp)] = '\n';
+						ZSTR_LEN(zquery) ++;
 					}
 					if (PQputCopyData(pgsql, ZSTR_VAL(zquery), ZSTR_LEN(tmp) > 0 ? ZSTR_LEN(tmp) + 1 : 0) != 1) {
 						zend_string_release_ex(zquery, false);