Fix dsa_allocate for aqo_qtext_store to avoid segfault when out of memory (#166)

queenofpigeons · web-flow · commit c80987852866 · 2023-04-28T15:11:42.000+03:00
diff --git a/storage.c b/storage.c
@@ -1111,7 +1111,7 @@ aqo_qtext_store(uint64 queryid, const char *query_string)
 
 		entry->queryid = queryid;
 		size = size > querytext_max_size ? querytext_max_size : size;
-		entry->qtext_dp = dsa_allocate(qtext_dsa, size);
+		entry->qtext_dp = dsa_allocate0(qtext_dsa, size);
 
 		if (!_check_dsa_validity(entry->qtext_dp))
 		{

Original file line number	Diff line number	Diff line change
`@@ -1111,7 +1111,7 @@ aqo_qtext_store(uint64 queryid, const char *query_string)`
`1111`	`1111`
`1112`	`1112`	`entry->queryid = queryid;`
`1113`	`1113`	`size = size > querytext_max_size ? querytext_max_size : size;`
`1114`		`- entry->qtext_dp = dsa_allocate(qtext_dsa, size);`
	`1114`	`+ entry->qtext_dp = dsa_allocate0(qtext_dsa, size);`
`1115`	`1115`
`1116`	`1116`	`if (!_check_dsa_validity(entry->qtext_dp))`
`1117`	`1117`	`{`