From 4289d4a981aa992f108860a7977e454dac498a25 Mon Sep 17 00:00:00 2001
From: Halil Ozan Akgul <hozanakgul@gmail.com>
Date: Fri, 10 Mar 2023 12:20:15 +0300
Subject: [PATCH] safe functions

---
 src/backend/distributed/utils/attribute.c | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/backend/distributed/utils/attribute.c b/src/backend/distributed/utils/attribute.c
index 34791a81a..54b2d5b22 100644
--- a/src/backend/distributed/utils/attribute.c
+++ b/src/backend/distributed/utils/attribute.c
@@ -11,6 +11,7 @@
 #include "postgres.h"
 #include "unistd.h"
 
+#include "distributed/citus_safe_lib.h"
 #include "distributed/log_utils.h"
 #include "distributed/listutils.h"
 #include "distributed/tuplestore.h"
@@ -204,7 +205,7 @@ AttributeQueryIfAnnotated(const char *query_string, CmdType commandType)
 		}
 
 		/*	attributeToTenant = (char *) malloc(strlen(tenantId)); */
-		strcpy(attributeToTenant, tenantId);
+		strcpy_s(attributeToTenant, tenantId);
 	}
 	else
 	{
@@ -549,7 +550,7 @@ CreateTenantStats(MultiTenantMonitor *monitor)
 {
 	int tenantIndex = monitor->tenantCount;
 
-	strcpy(monitor->tenants[tenantIndex].tenantAttribute, attributeToTenant);
+	strcpy_s(monitor->tenants[tenantIndex].tenantAttribute, attributeToTenant);
 	monitor->tenants[tenantIndex].colocationGroupId = colocationGroupId;
 
 	monitor->tenants[tenantIndex].namedLockTranche.trancheId = LWLockNewTrancheId();