Skip to content

Commit 0585f66

Browse files
Fix hardcoded buffer size additions with exact calculations
Co-Authored-By: Dan Lynch <[email protected]>
1 parent e67134d commit 0585f66

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/wasm_wrapper.c

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ char* wasm_parse_plpgsql(const char* input) {
8888
}
8989

9090
size_t funcs_len = strlen(result.plpgsql_funcs);
91-
size_t json_len = funcs_len + 32; // Extra space for wrapper JSON
91+
size_t json_len = strlen("{\"plpgsql_funcs\":}") + funcs_len + 1;
9292
char* wrapped_result = safe_malloc(json_len);
9393

9494
if (!wrapped_result) {
@@ -230,7 +230,7 @@ WasmDetailedResult* wasm_parse_query_detailed(const char* input) {
230230

231231
if (parse_result.error) {
232232
result->has_error = 1;
233-
size_t message_len = strlen(parse_result.error->message) + 100;
233+
size_t message_len = strlen("Parse error: at line , position ") + strlen(parse_result.error->message) + 20;
234234
char* prefixed_message = safe_malloc(message_len);
235235
if (!prefixed_message) {
236236
pg_query_free_parse_result(parse_result);

0 commit comments

Comments
 (0)