Age Owner Branch data TLA Line data Source code
1 : : /*
2 : : * pg_db_role_setting.c
3 : : * Routines to support manipulation of the pg_db_role_setting relation
4 : : *
5 : : * Portions Copyright (c) 1996-2024, PostgreSQL Global Development Group
6 : : * Portions Copyright (c) 1994, Regents of the University of California
7 : : *
8 : : * IDENTIFICATION
9 : : * src/backend/catalog/pg_db_role_setting.c
10 : : */
11 : : #include "postgres.h"
12 : :
13 : : #include "access/genam.h"
14 : : #include "access/heapam.h"
15 : : #include "access/htup_details.h"
16 : : #include "access/tableam.h"
17 : : #include "catalog/indexing.h"
18 : : #include "catalog/objectaccess.h"
19 : : #include "catalog/pg_db_role_setting.h"
20 : : #include "utils/fmgroids.h"
21 : : #include "utils/rel.h"
22 : :
23 : : void
5303 alvherre@alvh.no-ip. 24 :CBC 570 : AlterSetting(Oid databaseid, Oid roleid, VariableSetStmt *setstmt)
25 : : {
26 : : char *valuestr;
27 : : HeapTuple tuple;
28 : : Relation rel;
29 : : ScanKeyData scankey[2];
30 : : SysScanDesc scan;
31 : :
32 : 570 : valuestr = ExtractSetVariableArgs(setstmt);
33 : :
34 : : /* Get the old tuple, if any. */
35 : :
1910 andres@anarazel.de 36 : 570 : rel = table_open(DbRoleSettingRelationId, RowExclusiveLock);
5303 alvherre@alvh.no-ip. 37 : 570 : ScanKeyInit(&scankey[0],
38 : : Anum_pg_db_role_setting_setdatabase,
39 : : BTEqualStrategyNumber, F_OIDEQ,
40 : : ObjectIdGetDatum(databaseid));
41 : 570 : ScanKeyInit(&scankey[1],
42 : : Anum_pg_db_role_setting_setrole,
43 : : BTEqualStrategyNumber, F_OIDEQ,
44 : : ObjectIdGetDatum(roleid));
45 : 570 : scan = systable_beginscan(rel, DbRoleSettingDatidRolidIndexId, true,
46 : : NULL, 2, scankey);
47 : 570 : tuple = systable_getnext(scan);
48 : :
49 : : /*
50 : : * There are three cases:
51 : : *
52 : : * - in RESET ALL, request GUC to reset the settings array and update the
53 : : * catalog if there's anything left, delete it otherwise
54 : : *
55 : : * - in other commands, if there's a tuple in pg_db_role_setting, update
56 : : * it; if it ends up empty, delete it
57 : : *
58 : : * - otherwise, insert a new pg_db_role_setting tuple, but only if the
59 : : * command is not RESET
60 : : */
61 [ + + ]: 570 : if (setstmt->kind == VAR_RESET_ALL)
62 : : {
63 [ + - ]: 1 : if (HeapTupleIsValid(tuple))
64 : : {
5134 65 : 1 : ArrayType *new = NULL;
66 : : Datum datum;
67 : : bool isnull;
68 : :
69 : 1 : datum = heap_getattr(tuple, Anum_pg_db_role_setting_setconfig,
70 : : RelationGetDescr(rel), &isnull);
71 : :
72 [ + - ]: 1 : if (!isnull)
333 akorotkov@postgresql 73 : 1 : new = GUCArrayReset(DatumGetArrayTypeP(datum));
74 : :
5134 alvherre@alvh.no-ip. 75 [ + - ]: 1 : if (new)
76 : : {
77 : : Datum repl_val[Natts_pg_db_role_setting];
78 : : bool repl_null[Natts_pg_db_role_setting];
79 : : bool repl_repl[Natts_pg_db_role_setting];
80 : : HeapTuple newtuple;
81 : :
82 : 1 : memset(repl_repl, false, sizeof(repl_repl));
83 : :
84 : 1 : repl_val[Anum_pg_db_role_setting_setconfig - 1] =
85 : 1 : PointerGetDatum(new);
86 : 1 : repl_repl[Anum_pg_db_role_setting_setconfig - 1] = true;
87 : 1 : repl_null[Anum_pg_db_role_setting_setconfig - 1] = false;
88 : :
89 : 1 : newtuple = heap_modify_tuple(tuple, RelationGetDescr(rel),
90 : : repl_val, repl_null, repl_repl);
2630 91 : 1 : CatalogTupleUpdate(rel, &tuple->t_self, newtuple);
92 : : }
93 : : else
2629 tgl@sss.pgh.pa.us 94 :UBC 0 : CatalogTupleDelete(rel, &tuple->t_self);
95 : : }
96 : : }
5303 alvherre@alvh.no-ip. 97 [ + + ]:CBC 569 : else if (HeapTupleIsValid(tuple))
98 : : {
99 : : Datum repl_val[Natts_pg_db_role_setting];
100 : : bool repl_null[Natts_pg_db_role_setting];
101 : : bool repl_repl[Natts_pg_db_role_setting];
102 : : HeapTuple newtuple;
103 : : Datum datum;
104 : : bool isnull;
105 : : ArrayType *a;
106 : :
107 : 463 : memset(repl_repl, false, sizeof(repl_repl));
108 : 463 : repl_repl[Anum_pg_db_role_setting_setconfig - 1] = true;
109 : 463 : repl_null[Anum_pg_db_role_setting_setconfig - 1] = false;
110 : :
111 : : /* Extract old value of setconfig */
112 : 463 : datum = heap_getattr(tuple, Anum_pg_db_role_setting_setconfig,
113 : : RelationGetDescr(rel), &isnull);
114 [ + - ]: 463 : a = isnull ? NULL : DatumGetArrayTypeP(datum);
115 : :
116 : : /* Update (valuestr is NULL in RESET cases) */
117 [ + + ]: 463 : if (valuestr)
333 akorotkov@postgresql 118 : 452 : a = GUCArrayAdd(a, setstmt->name, valuestr);
119 : : else
120 : 11 : a = GUCArrayDelete(a, setstmt->name);
121 : :
5303 alvherre@alvh.no-ip. 122 [ + + ]: 462 : if (a)
123 : : {
124 : 452 : repl_val[Anum_pg_db_role_setting_setconfig - 1] =
125 : 452 : PointerGetDatum(a);
126 : :
127 : 452 : newtuple = heap_modify_tuple(tuple, RelationGetDescr(rel),
128 : : repl_val, repl_null, repl_repl);
2630 129 : 452 : CatalogTupleUpdate(rel, &tuple->t_self, newtuple);
130 : : }
131 : : else
2629 tgl@sss.pgh.pa.us 132 : 10 : CatalogTupleDelete(rel, &tuple->t_self);
133 : : }
5303 alvherre@alvh.no-ip. 134 [ + + ]: 106 : else if (valuestr)
135 : : {
136 : : /* non-null valuestr means it's not RESET, so insert a new tuple */
137 : : HeapTuple newtuple;
138 : : Datum values[Natts_pg_db_role_setting];
139 : : bool nulls[Natts_pg_db_role_setting];
140 : : ArrayType *a;
141 : :
142 : 104 : memset(nulls, false, sizeof(nulls));
143 : :
333 akorotkov@postgresql 144 : 104 : a = GUCArrayAdd(NULL, setstmt->name, valuestr);
145 : :
5303 alvherre@alvh.no-ip. 146 : 104 : values[Anum_pg_db_role_setting_setdatabase - 1] =
147 : 104 : ObjectIdGetDatum(databaseid);
148 : 104 : values[Anum_pg_db_role_setting_setrole - 1] = ObjectIdGetDatum(roleid);
149 : 104 : values[Anum_pg_db_role_setting_setconfig - 1] = PointerGetDatum(a);
150 : 104 : newtuple = heap_form_tuple(RelationGetDescr(rel), values, nulls);
151 : :
2630 152 : 104 : CatalogTupleInsert(rel, newtuple);
153 : : }
154 : :
4046 rhaas@postgresql.org 155 [ - + ]: 569 : InvokeObjectPostAlterHookArg(DbRoleSettingRelationId,
156 : : databaseid, 0, roleid, false);
157 : :
5303 alvherre@alvh.no-ip. 158 : 569 : systable_endscan(scan);
159 : :
160 : : /* Close pg_db_role_setting, but keep lock till commit */
1910 andres@anarazel.de 161 : 569 : table_close(rel, NoLock);
5303 alvherre@alvh.no-ip. 162 : 569 : }
163 : :
164 : : /*
165 : : * Drop some settings from the catalog. These can be for a particular
166 : : * database, or for a particular role. (It is of course possible to do both
167 : : * too, but it doesn't make sense for current uses.)
168 : : */
169 : : void
170 : 677 : DropSetting(Oid databaseid, Oid roleid)
171 : : {
172 : : Relation relsetting;
173 : : TableScanDesc scan;
174 : : ScanKeyData keys[2];
175 : : HeapTuple tup;
5161 bruce@momjian.us 176 : 677 : int numkeys = 0;
177 : :
1910 andres@anarazel.de 178 : 677 : relsetting = table_open(DbRoleSettingRelationId, RowExclusiveLock);
179 : :
5303 alvherre@alvh.no-ip. 180 [ + + ]: 677 : if (OidIsValid(databaseid))
181 : : {
182 : 30 : ScanKeyInit(&keys[numkeys],
183 : : Anum_pg_db_role_setting_setdatabase,
184 : : BTEqualStrategyNumber,
185 : : F_OIDEQ,
186 : : ObjectIdGetDatum(databaseid));
187 : 30 : numkeys++;
188 : : }
189 [ + + ]: 677 : if (OidIsValid(roleid))
190 : : {
191 : 647 : ScanKeyInit(&keys[numkeys],
192 : : Anum_pg_db_role_setting_setrole,
193 : : BTEqualStrategyNumber,
194 : : F_OIDEQ,
195 : : ObjectIdGetDatum(roleid));
196 : 647 : numkeys++;
197 : : }
198 : :
1861 andres@anarazel.de 199 : 677 : scan = table_beginscan_catalog(relsetting, numkeys, keys);
5303 alvherre@alvh.no-ip. 200 [ + + ]: 682 : while (HeapTupleIsValid(tup = heap_getnext(scan, ForwardScanDirection)))
201 : : {
2629 tgl@sss.pgh.pa.us 202 : 5 : CatalogTupleDelete(relsetting, &tup->t_self);
203 : : }
1861 andres@anarazel.de 204 : 677 : table_endscan(scan);
205 : :
1910 206 : 677 : table_close(relsetting, RowExclusiveLock);
5303 alvherre@alvh.no-ip. 207 : 677 : }
208 : :
209 : : /*
210 : : * Scan pg_db_role_setting looking for applicable settings, and load them on
211 : : * the current process.
212 : : *
213 : : * relsetting is pg_db_role_setting, already opened and locked.
214 : : *
215 : : * Note: we only consider setting for the exact databaseid/roleid combination.
216 : : * This probably needs to be called more than once, with InvalidOid passed as
217 : : * databaseid/roleid.
218 : : */
219 : : void
3939 rhaas@postgresql.org 220 : 52244 : ApplySetting(Snapshot snapshot, Oid databaseid, Oid roleid,
221 : : Relation relsetting, GucSource source)
222 : : {
223 : : SysScanDesc scan;
224 : : ScanKeyData keys[2];
225 : : HeapTuple tup;
226 : :
5303 alvherre@alvh.no-ip. 227 : 52244 : ScanKeyInit(&keys[0],
228 : : Anum_pg_db_role_setting_setdatabase,
229 : : BTEqualStrategyNumber,
230 : : F_OIDEQ,
231 : : ObjectIdGetDatum(databaseid));
232 : 52244 : ScanKeyInit(&keys[1],
233 : : Anum_pg_db_role_setting_setrole,
234 : : BTEqualStrategyNumber,
235 : : F_OIDEQ,
236 : : ObjectIdGetDatum(roleid));
237 : :
238 : 52244 : scan = systable_beginscan(relsetting, DbRoleSettingDatidRolidIndexId, true,
239 : : snapshot, 2, keys);
240 [ + + ]: 55346 : while (HeapTupleIsValid(tup = systable_getnext(scan)))
241 : : {
242 : : bool isnull;
243 : : Datum datum;
244 : :
245 : 3102 : datum = heap_getattr(tup, Anum_pg_db_role_setting_setconfig,
246 : : RelationGetDescr(relsetting), &isnull);
247 [ + - ]: 3102 : if (!isnull)
248 : : {
249 : 3102 : ArrayType *a = DatumGetArrayTypeP(datum);
250 : :
251 : : /*
252 : : * We process all the options at SUSET level. We assume that the
253 : : * right to insert an option into pg_db_role_setting was checked
254 : : * when it was inserted.
255 : : */
333 akorotkov@postgresql 256 : 3102 : ProcessGUCArray(a, PGC_SUSET, source, GUC_ACTION_SET);
257 : : }
258 : : }
259 : :
5303 alvherre@alvh.no-ip. 260 : 52244 : systable_endscan(scan);
261 : 52244 : }
|
