PostgreSQL Source Code  git master
rls.c File Reference
#include "postgres.h"
#include "access/htup.h"
#include "access/htup_details.h"
#include "access/transam.h"
#include "catalog/namespace.h"
#include "catalog/pg_class.h"
#include "miscadmin.h"
#include "utils/acl.h"
#include "utils/builtins.h"
#include "utils/lsyscache.h"
#include "utils/rls.h"
#include "utils/syscache.h"
#include "utils/varlena.h"
Include dependency graph for rls.c:

Go to the source code of this file.

Functions

int check_enable_rls (Oid relid, Oid checkAsUser, bool noError)
 
Datum row_security_active (PG_FUNCTION_ARGS)
 
Datum row_security_active_name (PG_FUNCTION_ARGS)
 

Function Documentation

◆ check_enable_rls()

int check_enable_rls ( Oid  relid,
Oid  checkAsUser,
bool  noError 
)

Definition at line 52 of file rls.c.

References ereport, errcode(), errhint(), errmsg(), ERROR, FirstNormalObjectId, get_rel_name(), GETSTRUCT, GetUserId(), has_bypassrls_privilege(), HeapTupleIsValid, InNoForceRLSOperation(), ObjectIdGetDatum, pg_class_ownercheck(), ReleaseSysCache(), relforcerowsecurity, RELOID, relrowsecurity, RLS_ENABLED, RLS_NONE, RLS_NONE_ENV, row_security, and SearchSysCache1().

Referenced by BuildIndexValueDescription(), DoCopy(), ExecBuildSlotPartitionKeyDescription(), ExecBuildSlotValueDescription(), get_row_security_policies(), intorel_startup(), ri_ReportViolation(), row_security_active(), and row_security_active_name().

53 {
54  Oid user_id = checkAsUser ? checkAsUser : GetUserId();
55  HeapTuple tuple;
56  Form_pg_class classform;
57  bool relrowsecurity;
59  bool amowner;
60 
61  /* Nothing to do for built-in relations */
62  if (relid < (Oid) FirstNormalObjectId)
63  return RLS_NONE;
64 
65  /* Fetch relation's relrowsecurity and relforcerowsecurity flags */
66  tuple = SearchSysCache1(RELOID, ObjectIdGetDatum(relid));
67  if (!HeapTupleIsValid(tuple))
68  return RLS_NONE;
69  classform = (Form_pg_class) GETSTRUCT(tuple);
70 
71  relrowsecurity = classform->relrowsecurity;
72  relforcerowsecurity = classform->relforcerowsecurity;
73 
74  ReleaseSysCache(tuple);
75 
76  /* Nothing to do if the relation does not have RLS */
77  if (!relrowsecurity)
78  return RLS_NONE;
79 
80  /*
81  * BYPASSRLS users always bypass RLS. Note that superusers are always
82  * considered to have BYPASSRLS.
83  *
84  * Return RLS_NONE_ENV to indicate that this decision depends on the
85  * environment (in this case, the user_id).
86  */
87  if (has_bypassrls_privilege(user_id))
88  return RLS_NONE_ENV;
89 
90  /*
91  * Table owners generally bypass RLS, except if the table has been set (by
92  * an owner) to FORCE ROW SECURITY, and this is not a referential
93  * integrity check.
94  *
95  * Return RLS_NONE_ENV to indicate that this decision depends on the
96  * environment (in this case, the user_id).
97  */
98  amowner = pg_class_ownercheck(relid, user_id);
99  if (amowner)
100  {
101  /*
102  * If FORCE ROW LEVEL SECURITY has been set on the relation then we
103  * should return RLS_ENABLED to indicate that RLS should be applied.
104  * If not, or if we are in an InNoForceRLSOperation context, we return
105  * RLS_NONE_ENV.
106  *
107  * InNoForceRLSOperation indicates that we should not apply RLS even
108  * if the table has FORCE RLS set - IF the current user is the owner.
109  * This is specifically to ensure that referential integrity checks
110  * are able to still run correctly.
111  *
112  * This is intentionally only done after we have checked that the user
113  * is the table owner, which should always be the case for referential
114  * integrity checks.
115  */
116  if (!relforcerowsecurity || InNoForceRLSOperation())
117  return RLS_NONE_ENV;
118  }
119 
120  /*
121  * We should apply RLS. However, the user may turn off the row_security
122  * GUC to get a forced error instead.
123  */
124  if (!row_security && !noError)
125  ereport(ERROR,
126  (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
127  errmsg("query would be affected by row-level security policy for table \"%s\"",
128  get_rel_name(relid)),
129  amowner ? errhint("To disable the policy for the table's owner, use ALTER TABLE NO FORCE ROW LEVEL SECURITY.") : 0));
130 
131  /* RLS should be fully enabled for this relation. */
132  return RLS_ENABLED;
133 }
int errhint(const char *fmt,...)
Definition: elog.c:987
#define GETSTRUCT(TUP)
Definition: htup_details.h:668
Oid GetUserId(void)
Definition: miscinit.c:379
int errcode(int sqlerrcode)
Definition: elog.c:575
bool InNoForceRLSOperation(void)
Definition: miscinit.c:520
unsigned int Oid
Definition: postgres_ext.h:31
#define FirstNormalObjectId
Definition: transam.h:94
bool has_bypassrls_privilege(Oid roleid)
Definition: aclchk.c:5395
#define ObjectIdGetDatum(X)
Definition: postgres.h:492
#define ERROR
Definition: elog.h:43
bool relforcerowsecurity
Definition: pg_class.h:65
bool relrowsecurity
Definition: pg_class.h:64
#define ereport(elevel, rest)
Definition: elog.h:122
HeapTuple SearchSysCache1(int cacheId, Datum key1)
Definition: syscache.c:1112
void ReleaseSysCache(HeapTuple tuple)
Definition: syscache.c:1160
#define HeapTupleIsValid(tuple)
Definition: htup.h:78
bool row_security
Definition: guc.c:448
bool pg_class_ownercheck(Oid class_oid, Oid roleid)
Definition: aclchk.c:4751
FormData_pg_class * Form_pg_class
Definition: pg_class.h:93
int errmsg(const char *fmt,...)
Definition: elog.c:797
Definition: rls.h:43
char * get_rel_name(Oid relid)
Definition: lsyscache.c:1730

◆ row_security_active()

Datum row_security_active ( PG_FUNCTION_ARGS  )

Definition at line 142 of file rls.c.

References check_enable_rls(), InvalidOid, PG_GETARG_OID, PG_RETURN_BOOL, and RLS_ENABLED.

143 {
144  /* By OID */
145  Oid tableoid = PG_GETARG_OID(0);
146  int rls_status;
147 
148  rls_status = check_enable_rls(tableoid, InvalidOid, true);
149  PG_RETURN_BOOL(rls_status == RLS_ENABLED);
150 }
unsigned int Oid
Definition: postgres_ext.h:31
#define PG_GETARG_OID(n)
Definition: fmgr.h:245
#define PG_RETURN_BOOL(x)
Definition: fmgr.h:324
#define InvalidOid
Definition: postgres_ext.h:36
int check_enable_rls(Oid relid, Oid checkAsUser, bool noError)
Definition: rls.c:52

◆ row_security_active_name()

Datum row_security_active_name ( PG_FUNCTION_ARGS  )

Definition at line 153 of file rls.c.

References check_enable_rls(), InvalidOid, makeRangeVarFromNameList(), NoLock, PG_GETARG_TEXT_PP, PG_RETURN_BOOL, RangeVarGetRelid, RLS_ENABLED, and textToQualifiedNameList().

154 {
155  /* By qualified name */
156  text *tablename = PG_GETARG_TEXT_PP(0);
157  RangeVar *tablerel;
158  Oid tableoid;
159  int rls_status;
160 
161  /* Look up table name. Can't lock it - we might not have privileges. */
162  tablerel = makeRangeVarFromNameList(textToQualifiedNameList(tablename));
163  tableoid = RangeVarGetRelid(tablerel, NoLock, false);
164 
165  rls_status = check_enable_rls(tableoid, InvalidOid, true);
166  PG_RETURN_BOOL(rls_status == RLS_ENABLED);
167 }
#define RangeVarGetRelid(relation, lockmode, missing_ok)
Definition: namespace.h:63
RangeVar * makeRangeVarFromNameList(List *names)
Definition: namespace.c:3042
unsigned int Oid
Definition: postgres_ext.h:31
#define PG_GETARG_TEXT_PP(n)
Definition: fmgr.h:278
#define NoLock
Definition: lockdefs.h:34
List * textToQualifiedNameList(text *textval)
Definition: varlena.c:3232
#define PG_RETURN_BOOL(x)
Definition: fmgr.h:324
#define InvalidOid
Definition: postgres_ext.h:36
int check_enable_rls(Oid relid, Oid checkAsUser, bool noError)
Definition: rls.c:52
Definition: c.h:516