external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Review changes for pg16 update GRANT and REVOKE

pg16_grant_inherit_set
Jodi-Ann Francis 2023-09-07 12:12:16 -04:00 committed by francisjodi
parent 587718aeae
commit c6d5240af7
2 changed files with 56 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
src/backend/distributed/deparser/deparse_role_stmts.c
View File

@ -408,20 +408,31 @@ AppendRevokeAdminOptionFor(StringInfo buf, GrantRoleStmt *stmt)
{ {
switch (opt->defname) switch (opt->defname)
{ {
case "admin":
appendStringInfo(buf, "ADMIN OPTION FOR "); appendStringInfo(buf, "ADMIN OPTION FOR ");
} opt_count++;
else if (strcmp(opt->defname, "inherit") == 0); break;
case "inherit":
if (opt_count > 0)
{ {
appendStringInfo(buf, "INHERIT TRUE"); appendStringInfo(buf, ", ");
appendStringInfo(buf, "GRANT x TO y WITH INHERIT TRUE, SET TRUE;");
} }
else if (strcmp(opt->defname, "set") == 0) appendStringInfo(buf, "INHERIT OPTION FOR ");
opt_count++;
break;
case "set":
if (opt_count > 0)
{ {
appendStringInfo(buf, "SET TRUE"); appendStringInfo(buf, ", ");
}
appendStringInfo(buf, "SET OPTION FOR ");
opt_count++;
break;
} }
} }
} }
}
#else #else
if (!stmt->is_grant && stmt->admin_opt) if (!stmt->is_grant && stmt->admin_opt)
{ {
@ -438,16 +449,34 @@ AppendGrantWithAdminOption(StringInfo buf, GrantRoleStmt *stmt)
{ {
#if PG_VERSION_NUM >= PG_VERSION_16 #if PG_VERSION_NUM >= PG_VERSION_16
DefElem *opt = NULL; DefElem *opt = NULL;
int opt_count = 0;
foreach_ptr(opt, stmt->opt) foreach_ptr(opt, stmt->opt)
{ {
bool admin_option = false; switch (opt->defname)
char *optval = defGetString(opt);
if (strcmp(opt->defname, "admin") == 0 &&
parse_bool(optval, &admin_option) && admin_option)
{ {
case "admin":
appendStringInfo(buf, " WITH ADMIN OPTION"); appendStringInfo(buf, " WITH ADMIN OPTION");
opt_count++;
break; break;
case "inherit":
if (opt_count > 0)
{
appendStringInfo(buf, ", ");
}
appendStringInfo(buf, "INHERIT OPTION ");
opt_count++;
break;
case "set":
if (opt_count > 0)
{
appendStringInfo(buf, ", ");
}
appendStringInfo(buf, "SET OPTION ");
opt_count++;
break;
}
} }
} }
} }

14
src/test/regress/sql/pg16.sql
View File

@ -899,10 +899,14 @@ GRANT create_group TO create_role_4 WITH SET;
-- ADMIN role can perfom administrative tasks -- ADMIN role can perfom administrative tasks
-- role can now access the data and permissions of the table (owner of table) -- role can now access the data and permissions of the table (owner of table)
-- role can change current user to any other user/role that has access -- role can change current user to any other user/role that has access
GRANT ADMIN ON DATABASE db_name TO role_name; GRANT ADMIN TO joe;
GRANT INHERIT ON TABLE table_name TO role_name; GRANT INHERIT ON ROLE joe TO james;
GRANT SET SESSION AUTHORIZATION TO role_name;
SELECT * FROM table_name WHERE column_name = 'value'; GRANT SELECT ON companies TO joe WITH GRANT OPTION;
GRANT SET (SELECT) ON companies TO james;
\set VERBOSITY terse
SET client_min_messages TO ERROR;
DROP SCHEMA pg16 CASCADE;
SELECT COUNT(*) FROM table_name WHERE column_name = 'value';