s3-privs Remove extra pointer on privilege mask
[Samba.git] / source3 / lib / privileges.c
blob5a54a140333065d23fb29be02284325bca141efb
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"
25 #include "dbwrap.h"
27 #define PRIVPREFIX "PRIV_"
29 typedef struct {
30 uint32_t count;
31 struct dom_sid *list;
32 } SID_LIST;
34 typedef struct {
35 TALLOC_CTX *mem_ctx;
36 uint64_t privilege;
37 SID_LIST sids;
38 } PRIV_SID_LIST;
41 interpret an old style SE_PRIV structure
43 static uint64_t map_old_SE_PRIV(unsigned char *dptr)
45 uint32_t *old_masks = (uint32_t *)dptr;
47 * the old privileges code only ever used up to 0x800, except
48 * for a special case of 'SE_ALL_PRIVS' which was 0xFFFFFFFF
50 if (old_masks[0] == 0xFFFFFFFF) {
51 /* they set all privileges */
52 return SE_ALL_PRIVS;
55 /* the old code used the machine byte order, but we don't know
56 * the byte order of the machine that wrote it. However we can
57 * tell what byte order it was by taking advantage of the fact
58 * that it only ever use up to 0x800
60 if (dptr[0] || dptr[1]) {
61 /* it was little endian */
62 return IVAL(dptr, 0);
65 /* it was either zero or big-endian */
66 return RIVAL(dptr, 0);
70 static bool get_privileges( const struct dom_sid *sid, uint64_t *mask )
72 struct db_context *db = get_account_pol_db();
73 fstring tmp, keystr;
74 TDB_DATA data;
76 /* Fail if the admin has not enable privileges */
78 if ( !lp_enable_privileges() ) {
79 return False;
82 if ( db == NULL )
83 return False;
85 /* PRIV_<SID> (NULL terminated) as the key */
87 fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
89 data = dbwrap_fetch_bystring( db, talloc_tos(), keystr );
91 if ( !data.dptr ) {
92 DEBUG(3, ("get_privileges: No privileges assigned to SID "
93 "[%s]\n", sid_string_dbg(sid)));
94 return False;
97 if (data.dsize == 4*4) {
98 /* it's an old style SE_PRIV structure. */
99 *mask = map_old_SE_PRIV(data.dptr);
100 } else {
101 if (data.dsize != sizeof( uint64_t ) ) {
102 DEBUG(3, ("get_privileges: Invalid privileges record assigned to SID "
103 "[%s]\n", sid_string_dbg(sid)));
104 return False;
107 *mask = BVAL(data.dptr, 0);
110 TALLOC_FREE(data.dptr);
112 return True;
115 /***************************************************************************
116 Store the privilege mask (set) for a given SID
117 ****************************************************************************/
119 static bool set_privileges( const struct dom_sid *sid, uint64_t mask )
121 struct db_context *db = get_account_pol_db();
122 uint8_t privbuf[8];
123 fstring tmp, keystr;
124 TDB_DATA data;
126 if ( !lp_enable_privileges() )
127 return False;
129 if ( db == NULL )
130 return False;
132 if ( !sid || (sid->num_auths == 0) ) {
133 DEBUG(0,("set_privileges: Refusing to store empty SID!\n"));
134 return False;
137 /* PRIV_<SID> (NULL terminated) as the key */
139 fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
141 /* This writes the 64 bit bitmask out in little endian format */
142 SBVAL(privbuf,0,mask);
144 data.dptr = privbuf;
145 data.dsize = sizeof(privbuf);
147 return NT_STATUS_IS_OK(dbwrap_store_bystring(db, keystr, data,
148 TDB_REPLACE));
151 /*********************************************************************
152 get a list of all privileges for all sids in the list
153 *********************************************************************/
155 bool get_privileges_for_sids(uint64_t *privileges, struct dom_sid *slist, int scount)
157 uint64_t mask;
158 int i;
159 bool found = False;
161 *privileges = 0;
163 for ( i=0; i<scount; i++ ) {
164 /* don't add unless we actually have a privilege assigned */
166 if ( !get_privileges( &slist[i], &mask ) )
167 continue;
169 DEBUG(5,("get_privileges_for_sids: sid = %s\nPrivilege "
170 "set: 0x%llx\n", sid_string_dbg(&slist[i]),
171 (unsigned long long)mask));
173 *privileges |= mask;
174 found = True;
177 return found;
180 NTSTATUS get_privileges_for_sid_as_set(TALLOC_CTX *mem_ctx, PRIVILEGE_SET **privileges, struct dom_sid *sid)
182 uint64_t mask;
183 if (!get_privileges(sid, &mask)) {
184 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
187 *privileges = talloc_zero(mem_ctx, PRIVILEGE_SET);
188 if (!*privileges) {
189 return NT_STATUS_NO_MEMORY;
192 if (!se_priv_to_privilege_set(*privileges, mask)) {
193 return NT_STATUS_NO_MEMORY;
195 return NT_STATUS_OK;
198 /*********************************************************************
199 traversal functions for privilege_enumerate_accounts
200 *********************************************************************/
202 static int priv_traverse_fn(struct db_record *rec, void *state)
204 PRIV_SID_LIST *priv = (PRIV_SID_LIST *)state;
205 int prefixlen = strlen(PRIVPREFIX);
206 struct dom_sid sid;
207 fstring sid_string;
209 /* check we have a PRIV_+SID entry */
211 if ( strncmp((char *)rec->key.dptr, PRIVPREFIX, prefixlen) != 0)
212 return 0;
214 /* check to see if we are looking for a particular privilege */
216 fstrcpy( sid_string, (char *)&(rec->key.dptr[strlen(PRIVPREFIX)]) );
218 if (priv->privilege != 0) {
219 uint64_t mask;
221 if (rec->value.dsize == 4*4) {
222 mask = map_old_SE_PRIV(rec->value.dptr);
223 } else {
224 if (rec->value.dsize != sizeof( uint64_t ) ) {
225 DEBUG(3, ("get_privileges: Invalid privileges record assigned to SID "
226 "[%s]\n", sid_string));
227 return 0;
229 mask = BVAL(rec->value.dptr, 0);
232 /* if the SID does not have the specified privilege
233 then just return */
235 if ((mask & priv->privilege) == 0) {
236 return 0;
240 /* this is a last ditch safety check to preventing returning
241 and invalid SID (i've somehow run into this on development branches) */
243 if ( strcmp( "S-0-0", sid_string ) == 0 )
244 return 0;
246 if ( !string_to_sid(&sid, sid_string) ) {
247 DEBUG(0,("travsersal_fn_enum__acct: Could not convert SID [%s]\n",
248 sid_string));
249 return 0;
252 if (!NT_STATUS_IS_OK(add_sid_to_array(priv->mem_ctx, &sid,
253 &priv->sids.list,
254 &priv->sids.count)))
256 return 0;
259 return 0;
262 /*********************************************************************
263 Retreive list of privileged SIDs (for _lsa_enumerate_accounts()
264 *********************************************************************/
266 NTSTATUS privilege_enumerate_accounts(struct dom_sid **sids, int *num_sids)
268 struct db_context *db = get_account_pol_db();
269 PRIV_SID_LIST priv;
271 if (db == NULL) {
272 return NT_STATUS_ACCESS_DENIED;
275 ZERO_STRUCT(priv);
277 db->traverse_read(db, priv_traverse_fn, &priv);
279 /* give the memory away; caller will free */
281 *sids = priv.sids.list;
282 *num_sids = priv.sids.count;
284 return NT_STATUS_OK;
287 /*********************************************************************
288 Retrieve list of SIDs granted a particular privilege
289 *********************************************************************/
291 NTSTATUS privilege_enum_sids(enum sec_privilege privilege, TALLOC_CTX *mem_ctx,
292 struct dom_sid **sids, int *num_sids)
294 struct db_context *db = get_account_pol_db();
295 PRIV_SID_LIST priv;
297 if (db == NULL) {
298 return NT_STATUS_ACCESS_DENIED;
301 ZERO_STRUCT(priv);
303 priv.privilege = sec_privilege_mask(privilege);
304 priv.mem_ctx = mem_ctx;
306 db->traverse_read(db, priv_traverse_fn, &priv);
308 /* give the memory away; caller will free */
310 *sids = priv.sids.list;
311 *num_sids = priv.sids.count;
313 return NT_STATUS_OK;
316 /***************************************************************************
317 Add privilege to sid
318 ****************************************************************************/
320 static bool grant_privilege_bitmap(const struct dom_sid *sid, const uint64_t priv_mask)
322 uint64_t old_mask, new_mask;
324 ZERO_STRUCT( old_mask );
325 ZERO_STRUCT( new_mask );
327 if ( get_privileges( sid, &old_mask ) )
328 new_mask = old_mask;
329 else
330 new_mask = 0;
332 new_mask |= priv_mask;
334 DEBUG(10,("grant_privilege: %s\n", sid_string_dbg(sid)));
336 DEBUGADD( 10, ("original privilege mask: 0x%llx\n", (unsigned long long)new_mask));
338 DEBUGADD( 10, ("new privilege mask: 0x%llx\n", (unsigned long long)new_mask));
340 return set_privileges( sid, new_mask );
343 /*********************************************************************
344 Add a privilege based on its name
345 *********************************************************************/
347 bool grant_privilege_by_name(struct dom_sid *sid, const char *name)
349 uint64_t mask;
351 if (! se_priv_from_name(name, &mask)) {
352 DEBUG(3, ("grant_privilege_by_name: "
353 "No Such Privilege Found (%s)\n", name));
354 return False;
357 return grant_privilege_bitmap( sid, mask );
360 /***************************************************************************
361 Grant a privilege set (list of LUID values) from a sid
362 ****************************************************************************/
364 bool grant_privilege_set(const struct dom_sid *sid, struct lsa_PrivilegeSet *set)
366 uint64_t privilege_mask;
367 if (!privilege_set_to_se_priv(&privilege_mask, set)) {
368 return false;
370 return grant_privilege_bitmap(sid, privilege_mask);
373 /***************************************************************************
374 Remove privilege from sid
375 ****************************************************************************/
377 static bool revoke_privilege_bitmap(const struct dom_sid *sid, const uint64_t priv_mask)
379 uint64_t mask;
381 /* if the user has no privileges, then we can't revoke any */
383 if ( !get_privileges( sid, &mask ) )
384 return True;
386 DEBUG(10,("revoke_privilege: %s\n", sid_string_dbg(sid)));
388 DEBUGADD( 10, ("original privilege mask: 0x%llx\n", (unsigned long long)mask));
390 mask &= ~priv_mask;
392 DEBUGADD( 10, ("new privilege mask: 0x%llx\n", (unsigned long long)mask));
394 return set_privileges( sid, mask );
397 /***************************************************************************
398 Remove a privilege set (list of LUID values) from a sid
399 ****************************************************************************/
401 bool revoke_privilege_set(const struct dom_sid *sid, struct lsa_PrivilegeSet *set)
403 uint64_t privilege_mask;
404 if (!privilege_set_to_se_priv(&privilege_mask, set)) {
405 return false;
407 return revoke_privilege_bitmap(sid, privilege_mask);
410 /*********************************************************************
411 Revoke all privileges
412 *********************************************************************/
414 bool revoke_all_privileges( struct dom_sid *sid )
416 return revoke_privilege_bitmap( sid, SE_ALL_PRIVS);
419 /*********************************************************************
420 Add a privilege based on its name
421 *********************************************************************/
423 bool revoke_privilege_by_name(struct dom_sid *sid, const char *name)
425 uint64_t mask;
427 if (! se_priv_from_name(name, &mask)) {
428 DEBUG(3, ("revoke_privilege_by_name: "
429 "No Such Privilege Found (%s)\n", name));
430 return False;
433 return revoke_privilege_bitmap(sid, mask);
437 /***************************************************************************
438 Retrieve the SIDs assigned to a given privilege
439 ****************************************************************************/
441 NTSTATUS privilege_create_account(const struct dom_sid *sid )
443 return ( grant_privilege_bitmap(sid, 0) ? NT_STATUS_OK : NT_STATUS_UNSUCCESSFUL);
446 /***************************************************************************
447 Delete a privileged account
448 ****************************************************************************/
450 NTSTATUS privilege_delete_account(const struct dom_sid *sid)
452 struct db_context *db = get_account_pol_db();
453 fstring tmp, keystr;
455 if (!lp_enable_privileges()) {
456 return NT_STATUS_OK;
459 if (!db) {
460 return NT_STATUS_INVALID_HANDLE;
463 if (!sid || (sid->num_auths == 0)) {
464 return NT_STATUS_INVALID_SID;
467 /* PRIV_<SID> (NULL terminated) as the key */
469 fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
471 return dbwrap_delete_bystring(db, keystr);
474 /*******************************************************************
475 *******************************************************************/
477 bool is_privileged_sid( const struct dom_sid *sid )
479 uint64_t mask;
481 return get_privileges( sid, &mask );
484 /*******************************************************************
485 *******************************************************************/
487 bool grant_all_privileges( const struct dom_sid *sid )
489 uint64_t mask;
491 if (!se_priv_put_all_privileges(&mask)) {
492 return False;
495 return grant_privilege_bitmap( sid, mask );