selftest: Use TESTS env var to specify tests to run like Samba4 does.
[Samba.git] / source / lib / privileges.c
blobc1bb783fbc2174ed28d97cedfe5e1b89952765bd
1 /*
2 Unix SMB/CIFS implementation.
3 Privileges handling functions
4 Copyright (C) Jean François Micouleau 1998-2001
5 Copyright (C) Simo Sorce 2002-2003
6 Copyright (C) Gerald (Jerry) Carter 2005
7 Copyright (C) Michael Adam 2007
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 3 of the License, or
12 (at your option) any later version.
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
19 You should have received a copy of the GNU General Public License
20 along with this program. If not, see <http://www.gnu.org/licenses/>.
24 #include "includes.h"
26 #define PRIVPREFIX "PRIV_"
28 typedef struct {
29 size_t count;
30 DOM_SID *list;
31 } SID_LIST;
33 typedef struct {
34 TALLOC_CTX *mem_ctx;
35 SE_PRIV privilege;
36 SID_LIST sids;
37 } PRIV_SID_LIST;
40 static bool get_privileges( const DOM_SID *sid, SE_PRIV *mask )
42 struct db_context *db = get_account_pol_db();
43 fstring tmp, keystr;
44 TDB_DATA data;
46 /* Fail if the admin has not enable privileges */
48 if ( !lp_enable_privileges() ) {
49 return False;
52 if ( db == NULL )
53 return False;
55 /* PRIV_<SID> (NULL terminated) as the key */
57 fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
59 data = dbwrap_fetch_bystring( db, talloc_tos(), keystr );
61 if ( !data.dptr ) {
62 DEBUG(3, ("get_privileges: No privileges assigned to SID "
63 "[%s]\n", sid_string_dbg(sid)));
64 return False;
67 SMB_ASSERT( data.dsize == sizeof( SE_PRIV ) );
69 se_priv_copy( mask, (SE_PRIV*)data.dptr );
70 TALLOC_FREE(data.dptr);
72 return True;
75 /***************************************************************************
76 Store the privilege mask (set) for a given SID
77 ****************************************************************************/
79 static bool set_privileges( const DOM_SID *sid, SE_PRIV *mask )
81 struct db_context *db = get_account_pol_db();
82 fstring tmp, keystr;
83 TDB_DATA data;
85 if ( !lp_enable_privileges() )
86 return False;
88 if ( db == NULL )
89 return False;
91 if ( !sid || (sid->num_auths == 0) ) {
92 DEBUG(0,("set_privileges: Refusing to store empty SID!\n"));
93 return False;
96 /* PRIV_<SID> (NULL terminated) as the key */
98 fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
100 /* no packing. static size structure, just write it out */
102 data.dptr = (uint8 *)mask;
103 data.dsize = sizeof(SE_PRIV);
105 return NT_STATUS_IS_OK(dbwrap_store_bystring(db, keystr, data,
106 TDB_REPLACE));
109 /*********************************************************************
110 get a list of all privileges for all sids in the list
111 *********************************************************************/
113 bool get_privileges_for_sids(SE_PRIV *privileges, DOM_SID *slist, int scount)
115 SE_PRIV mask;
116 int i;
117 bool found = False;
119 se_priv_copy( privileges, &se_priv_none );
121 for ( i=0; i<scount; i++ ) {
122 /* don't add unless we actually have a privilege assigned */
124 if ( !get_privileges( &slist[i], &mask ) )
125 continue;
127 DEBUG(5,("get_privileges_for_sids: sid = %s\nPrivilege "
128 "set:\n", sid_string_dbg(&slist[i])));
129 dump_se_priv( DBGC_ALL, 5, &mask );
131 se_priv_add( privileges, &mask );
132 found = True;
135 return found;
139 /*********************************************************************
140 traversal functions for privilege_enumerate_accounts
141 *********************************************************************/
143 static int priv_traverse_fn(struct db_record *rec, void *state)
145 PRIV_SID_LIST *priv = (PRIV_SID_LIST *)state;
146 int prefixlen = strlen(PRIVPREFIX);
147 DOM_SID sid;
148 fstring sid_string;
150 /* easy check first */
152 if (rec->value.dsize != sizeof(SE_PRIV) )
153 return 0;
155 /* check we have a PRIV_+SID entry */
157 if ( strncmp((char *)rec->key.dptr, PRIVPREFIX, prefixlen) != 0)
158 return 0;
160 /* check to see if we are looking for a particular privilege */
162 if ( !se_priv_equal(&priv->privilege, &se_priv_none) ) {
163 SE_PRIV mask;
165 se_priv_copy( &mask, (SE_PRIV*)rec->value.dptr );
167 /* if the SID does not have the specified privilege
168 then just return */
170 if ( !is_privilege_assigned( &mask, &priv->privilege) )
171 return 0;
174 fstrcpy( sid_string, (char *)&(rec->key.dptr[strlen(PRIVPREFIX)]) );
176 /* this is a last ditch safety check to preventing returning
177 and invalid SID (i've somehow run into this on development branches) */
179 if ( strcmp( "S-0-0", sid_string ) == 0 )
180 return 0;
182 if ( !string_to_sid(&sid, sid_string) ) {
183 DEBUG(0,("travsersal_fn_enum__acct: Could not convert SID [%s]\n",
184 sid_string));
185 return 0;
188 if (!NT_STATUS_IS_OK(add_sid_to_array(priv->mem_ctx, &sid,
189 &priv->sids.list,
190 &priv->sids.count)))
192 return 0;
195 return 0;
198 /*********************************************************************
199 Retreive list of privileged SIDs (for _lsa_enumerate_accounts()
200 *********************************************************************/
202 NTSTATUS privilege_enumerate_accounts(DOM_SID **sids, int *num_sids)
204 struct db_context *db = get_account_pol_db();
205 PRIV_SID_LIST priv;
207 if (db == NULL) {
208 return NT_STATUS_ACCESS_DENIED;
211 ZERO_STRUCT(priv);
213 se_priv_copy( &priv.privilege, &se_priv_none );
215 db->traverse_read(db, priv_traverse_fn, &priv);
217 /* give the memory away; caller will free */
219 *sids = priv.sids.list;
220 *num_sids = priv.sids.count;
222 return NT_STATUS_OK;
225 /*********************************************************************
226 Retrieve list of SIDs granted a particular privilege
227 *********************************************************************/
229 NTSTATUS privilege_enum_sids(const SE_PRIV *mask, TALLOC_CTX *mem_ctx,
230 DOM_SID **sids, int *num_sids)
232 struct db_context *db = get_account_pol_db();
233 PRIV_SID_LIST priv;
235 if (db == NULL) {
236 return NT_STATUS_ACCESS_DENIED;
239 ZERO_STRUCT(priv);
241 se_priv_copy(&priv.privilege, mask);
242 priv.mem_ctx = mem_ctx;
244 db->traverse_read(db, priv_traverse_fn, &priv);
246 /* give the memory away; caller will free */
248 *sids = priv.sids.list;
249 *num_sids = priv.sids.count;
251 return NT_STATUS_OK;
254 /***************************************************************************
255 Add privilege to sid
256 ****************************************************************************/
258 bool grant_privilege(const DOM_SID *sid, const SE_PRIV *priv_mask)
260 SE_PRIV old_mask, new_mask;
262 ZERO_STRUCT( old_mask );
263 ZERO_STRUCT( new_mask );
265 if ( get_privileges( sid, &old_mask ) )
266 se_priv_copy( &new_mask, &old_mask );
267 else
268 se_priv_copy( &new_mask, &se_priv_none );
270 se_priv_add( &new_mask, priv_mask );
272 DEBUG(10,("grant_privilege: %s\n", sid_string_dbg(sid)));
274 DEBUGADD( 10, ("original privilege mask:\n"));
275 dump_se_priv( DBGC_ALL, 10, &old_mask );
277 DEBUGADD( 10, ("new privilege mask:\n"));
278 dump_se_priv( DBGC_ALL, 10, &new_mask );
280 return set_privileges( sid, &new_mask );
283 /*********************************************************************
284 Add a privilege based on its name
285 *********************************************************************/
287 bool grant_privilege_by_name(DOM_SID *sid, const char *name)
289 SE_PRIV mask;
291 if (! se_priv_from_name(name, &mask)) {
292 DEBUG(3, ("grant_privilege_by_name: "
293 "No Such Privilege Found (%s)\n", name));
294 return False;
297 return grant_privilege( sid, &mask );
300 /***************************************************************************
301 Remove privilege from sid
302 ****************************************************************************/
304 bool revoke_privilege(const DOM_SID *sid, const SE_PRIV *priv_mask)
306 SE_PRIV mask;
308 /* if the user has no privileges, then we can't revoke any */
310 if ( !get_privileges( sid, &mask ) )
311 return True;
313 DEBUG(10,("revoke_privilege: %s\n", sid_string_dbg(sid)));
315 DEBUGADD( 10, ("original privilege mask:\n"));
316 dump_se_priv( DBGC_ALL, 10, &mask );
318 se_priv_remove( &mask, priv_mask );
320 DEBUGADD( 10, ("new privilege mask:\n"));
321 dump_se_priv( DBGC_ALL, 10, &mask );
323 return set_privileges( sid, &mask );
326 /*********************************************************************
327 Revoke all privileges
328 *********************************************************************/
330 bool revoke_all_privileges( DOM_SID *sid )
332 return revoke_privilege( sid, &se_priv_all );
335 /*********************************************************************
336 Add a privilege based on its name
337 *********************************************************************/
339 bool revoke_privilege_by_name(DOM_SID *sid, const char *name)
341 SE_PRIV mask;
343 if (! se_priv_from_name(name, &mask)) {
344 DEBUG(3, ("revoke_privilege_by_name: "
345 "No Such Privilege Found (%s)\n", name));
346 return False;
349 return revoke_privilege(sid, &mask);
353 /***************************************************************************
354 Retrieve the SIDs assigned to a given privilege
355 ****************************************************************************/
357 NTSTATUS privilege_create_account(const DOM_SID *sid )
359 return ( grant_privilege(sid, &se_priv_none) ? NT_STATUS_OK : NT_STATUS_UNSUCCESSFUL);
362 /****************************************************************************
363 initialise a privilege list and set the talloc context
364 ****************************************************************************/
366 NTSTATUS privilege_set_init(PRIVILEGE_SET *priv_set)
368 TALLOC_CTX *mem_ctx;
370 ZERO_STRUCTP( priv_set );
372 mem_ctx = talloc_init("privilege set");
373 if ( !mem_ctx ) {
374 DEBUG(0,("privilege_set_init: failed to initialize talloc ctx!\n"));
375 return NT_STATUS_NO_MEMORY;
378 priv_set->mem_ctx = mem_ctx;
380 return NT_STATUS_OK;
383 /****************************************************************************
384 initialise a privilege list and with someone else's talloc context
385 ****************************************************************************/
387 NTSTATUS privilege_set_init_by_ctx(TALLOC_CTX *mem_ctx, PRIVILEGE_SET *priv_set)
389 ZERO_STRUCTP( priv_set );
391 priv_set->mem_ctx = mem_ctx;
392 priv_set->ext_ctx = True;
394 return NT_STATUS_OK;
397 /****************************************************************************
398 Free all memory used by a PRIVILEGE_SET
399 ****************************************************************************/
401 void privilege_set_free(PRIVILEGE_SET *priv_set)
403 if ( !priv_set )
404 return;
406 if ( !( priv_set->ext_ctx ) )
407 talloc_destroy( priv_set->mem_ctx );
409 ZERO_STRUCTP( priv_set );
412 /****************************************************************************
413 duplicate alloc luid_attr
414 ****************************************************************************/
416 NTSTATUS dup_luid_attr(TALLOC_CTX *mem_ctx, LUID_ATTR **new_la, LUID_ATTR *old_la, int count)
418 int i;
420 if ( !old_la )
421 return NT_STATUS_OK;
423 if (count) {
424 *new_la = TALLOC_ARRAY(mem_ctx, LUID_ATTR, count);
425 if ( !*new_la ) {
426 DEBUG(0,("dup_luid_attr: failed to alloc new LUID_ATTR array [%d]\n", count));
427 return NT_STATUS_NO_MEMORY;
429 } else {
430 *new_la = NULL;
433 for (i=0; i<count; i++) {
434 (*new_la)[i].luid.high = old_la[i].luid.high;
435 (*new_la)[i].luid.low = old_la[i].luid.low;
436 (*new_la)[i].attr = old_la[i].attr;
439 return NT_STATUS_OK;
442 /*******************************************************************
443 *******************************************************************/
445 bool is_privileged_sid( const DOM_SID *sid )
447 SE_PRIV mask;
449 return get_privileges( sid, &mask );
452 /*******************************************************************
453 *******************************************************************/
455 bool grant_all_privileges( const DOM_SID *sid )
457 SE_PRIV mask;
459 if (!se_priv_put_all_privileges(&mask)) {
460 return False;
463 return grant_privilege( sid, &mask );