From fdb4d415d105ec65538bd27d2e3645b1b792445c Mon Sep 17 00:00:00 2001
From: Yura Sokolov <funny.falcon@gmail.com>
Date: Wed, 20 Oct 2021 11:31:01 +0300
Subject: [PATCH] ee: Use memory context instead of malloc in ATX handling

PGPRO-5692 / PGPRO-5336
---
 src/multimaster.c | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/multimaster.c b/src/multimaster.c
index f595e90d6c..9f4d3c6479 100644
--- a/src/multimaster.c
+++ b/src/multimaster.c
@@ -366,7 +366,7 @@ MtmDeserializeTransactionState(void *ctx)
 static void *
 MtmSuspendTransaction(void)
 {
-	MtmCurrentTrans *ctx = malloc(sizeof(MtmCurrentTrans));
+	MtmCurrentTrans *ctx = MemoryContextAlloc(CurTransactionContext, sizeof(MtmCurrentTrans));
 
 	*ctx = MtmTx;
 	return ctx;
@@ -376,7 +376,7 @@ static void
 MtmResumeTransaction(void *ctx)
 {
 	MtmTx = *(MtmCurrentTrans *) ctx;
-	free(ctx);
+	pfree(ctx);
 }
 #endif