From ff03b35f945d68e57acebbe433f8c000617f22cd Mon Sep 17 00:00:00 2001
From: Sait Talha Nisanci <s.talhanisanci@gmail.com>
Date: Wed, 30 Dec 2020 14:18:40 +0300
Subject: [PATCH] Add security utils

---
 src/include/distributed/security_utils.h | 26 ++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 src/include/distributed/security_utils.h

diff --git a/src/include/distributed/security_utils.h b/src/include/distributed/security_utils.h
new file mode 100644
index 000000000..549ae641f
--- /dev/null
+++ b/src/include/distributed/security_utils.h
@@ -0,0 +1,26 @@
+/*-------------------------------------------------------------------------
+ *
+ * security_utils.h
+ *   security related utility functions.
+ *
+ * Copyright (c) Citus Data, Inc.
+ *
+ *-------------------------------------------------------------------------
+ */
+
+#ifndef SECURITY_UTILS_H_
+#define SECURITY_UTILS_H_
+
+#include "postgres.h"
+#include "miscadmin.h"
+
+#define PushCitusSecurityContext() \
+    Oid savedUserId_DONTUSE = InvalidOid; \
+	int savedSecurityContext_DONTUSE = 0; \
+    GetUserIdAndSecContext(&savedUserId_DONTUSE, &savedSecurityContext_DONTUSE); \
+	SetUserIdAndSecContext(CitusExtensionOwner(), SECURITY_LOCAL_USERID_CHANGE);
+
+#define PopCitusSecurityContext() \
+    SetUserIdAndSecContext(savedUserId_DONTUSE, savedSecurityContext_DONTUSE);
+
+#endif