s3: Use cli_writeall instead of cli_write
[Samba.git] / source3 / lib / privileges.c
bloba793d7755953da58c7cbbdc67fcc3e23f3803595
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 "lib/privileges.h"
26 #include "dbwrap.h"
27 #include "libcli/security/privileges_private.h"
28 #include "../libcli/security/security.h"
29 #include "passdb.h"
31 #define PRIVPREFIX "PRIV_"
33 typedef struct {
34 uint32_t count;
35 struct dom_sid *list;
36 } SID_LIST;
38 typedef struct {
39 TALLOC_CTX *mem_ctx;
40 uint64_t privilege;
41 SID_LIST sids;
42 } PRIV_SID_LIST;
45 interpret an old style SE_PRIV structure
47 static uint64_t map_old_SE_PRIV(unsigned char *dptr)
49 uint32_t *old_masks = (uint32_t *)dptr;
51 * the old privileges code only ever used up to 0x800, except
52 * for a special case of 'SE_ALL_PRIVS' which was 0xFFFFFFFF
54 if (old_masks[0] == 0xFFFFFFFF) {
55 /* they set all privileges */
56 return SE_ALL_PRIVS;
59 /* the old code used the machine byte order, but we don't know
60 * the byte order of the machine that wrote it. However we can
61 * tell what byte order it was by taking advantage of the fact
62 * that it only ever use up to 0x800
64 if (dptr[0] || dptr[1]) {
65 /* it was little endian */
66 return IVAL(dptr, 0);
69 /* it was either zero or big-endian */
70 return RIVAL(dptr, 0);
74 static bool get_privileges( const struct dom_sid *sid, uint64_t *mask )
76 struct db_context *db = get_account_pol_db();
77 fstring tmp, keystr;
78 TDB_DATA data;
80 /* Fail if the admin has not enable privileges */
82 if ( !lp_enable_privileges() ) {
83 return False;
86 if ( db == NULL )
87 return False;
89 /* PRIV_<SID> (NULL terminated) as the key */
91 fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
93 data = dbwrap_fetch_bystring( db, talloc_tos(), keystr );
95 if ( !data.dptr ) {
96 DEBUG(4, ("get_privileges: No privileges assigned to SID "
97 "[%s]\n", sid_string_dbg(sid)));
98 return False;
101 if (data.dsize == 4*4) {
102 /* it's an old style SE_PRIV structure. */
103 *mask = map_old_SE_PRIV(data.dptr);
104 } else {
105 if (data.dsize != sizeof( uint64_t ) ) {
106 DEBUG(3, ("get_privileges: Invalid privileges record assigned to SID "
107 "[%s]\n", sid_string_dbg(sid)));
108 return False;
111 *mask = BVAL(data.dptr, 0);
114 TALLOC_FREE(data.dptr);
116 return True;
119 /***************************************************************************
120 Store the privilege mask (set) for a given SID
121 ****************************************************************************/
123 static bool set_privileges( const struct dom_sid *sid, uint64_t mask )
125 struct db_context *db = get_account_pol_db();
126 uint8_t privbuf[8];
127 fstring tmp, keystr;
128 TDB_DATA data;
130 if ( !lp_enable_privileges() )
131 return False;
133 if ( db == NULL )
134 return False;
136 if ( !sid || (sid->num_auths == 0) ) {
137 DEBUG(0,("set_privileges: Refusing to store empty SID!\n"));
138 return False;
141 /* PRIV_<SID> (NULL terminated) as the key */
143 fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
145 /* This writes the 64 bit bitmask out in little endian format */
146 SBVAL(privbuf,0,mask);
148 data.dptr = privbuf;
149 data.dsize = sizeof(privbuf);
151 return NT_STATUS_IS_OK(dbwrap_store_bystring(db, keystr, data,
152 TDB_REPLACE));
155 /*********************************************************************
156 get a list of all privileges for all sids in the list
157 *********************************************************************/
159 bool get_privileges_for_sids(uint64_t *privileges, struct dom_sid *slist, int scount)
161 uint64_t mask;
162 int i;
163 bool found = False;
165 *privileges = 0;
167 for ( i=0; i<scount; i++ ) {
168 /* don't add unless we actually have a privilege assigned */
170 if ( !get_privileges( &slist[i], &mask ) )
171 continue;
173 DEBUG(5,("get_privileges_for_sids: sid = %s\nPrivilege "
174 "set: 0x%llx\n", sid_string_dbg(&slist[i]),
175 (unsigned long long)mask));
177 *privileges |= mask;
178 found = True;
181 return found;
184 NTSTATUS get_privileges_for_sid_as_set(TALLOC_CTX *mem_ctx, PRIVILEGE_SET **privileges, struct dom_sid *sid)
186 uint64_t mask;
187 if (!get_privileges(sid, &mask)) {
188 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
191 *privileges = talloc_zero(mem_ctx, PRIVILEGE_SET);
192 if (!*privileges) {
193 return NT_STATUS_NO_MEMORY;
196 if (!se_priv_to_privilege_set(*privileges, mask)) {
197 return NT_STATUS_NO_MEMORY;
199 return NT_STATUS_OK;
202 /*********************************************************************
203 traversal functions for privilege_enumerate_accounts
204 *********************************************************************/
206 static int priv_traverse_fn(struct db_record *rec, void *state)
208 PRIV_SID_LIST *priv = (PRIV_SID_LIST *)state;
209 int prefixlen = strlen(PRIVPREFIX);
210 struct dom_sid sid;
211 fstring sid_string;
213 /* check we have a PRIV_+SID entry */
215 if ( strncmp((char *)rec->key.dptr, PRIVPREFIX, prefixlen) != 0)
216 return 0;
218 /* check to see if we are looking for a particular privilege */
220 fstrcpy( sid_string, (char *)&(rec->key.dptr[strlen(PRIVPREFIX)]) );
222 if (priv->privilege != 0) {
223 uint64_t mask;
225 if (rec->value.dsize == 4*4) {
226 mask = map_old_SE_PRIV(rec->value.dptr);
227 } else {
228 if (rec->value.dsize != sizeof( uint64_t ) ) {
229 DEBUG(3, ("get_privileges: Invalid privileges record assigned to SID "
230 "[%s]\n", sid_string));
231 return 0;
233 mask = BVAL(rec->value.dptr, 0);
236 /* if the SID does not have the specified privilege
237 then just return */
239 if ((mask & priv->privilege) == 0) {
240 return 0;
244 /* this is a last ditch safety check to preventing returning
245 and invalid SID (i've somehow run into this on development branches) */
247 if ( strcmp( "S-0-0", sid_string ) == 0 )
248 return 0;
250 if ( !string_to_sid(&sid, sid_string) ) {
251 DEBUG(0,("travsersal_fn_enum__acct: Could not convert SID [%s]\n",
252 sid_string));
253 return 0;
256 if (!NT_STATUS_IS_OK(add_sid_to_array(priv->mem_ctx, &sid,
257 &priv->sids.list,
258 &priv->sids.count)))
260 return 0;
263 return 0;
266 /*********************************************************************
267 Retreive list of privileged SIDs (for _lsa_enumerate_accounts()
268 *********************************************************************/
270 NTSTATUS privilege_enumerate_accounts(struct dom_sid **sids, int *num_sids)
272 struct db_context *db = get_account_pol_db();
273 PRIV_SID_LIST priv;
275 if (db == NULL) {
276 return NT_STATUS_ACCESS_DENIED;
279 ZERO_STRUCT(priv);
281 db->traverse_read(db, priv_traverse_fn, &priv);
283 /* give the memory away; caller will free */
285 *sids = priv.sids.list;
286 *num_sids = priv.sids.count;
288 return NT_STATUS_OK;
291 /*********************************************************************
292 Retrieve list of SIDs granted a particular privilege
293 *********************************************************************/
295 NTSTATUS privilege_enum_sids(enum sec_privilege privilege, TALLOC_CTX *mem_ctx,
296 struct dom_sid **sids, int *num_sids)
298 struct db_context *db = get_account_pol_db();
299 PRIV_SID_LIST priv;
301 if (db == NULL) {
302 return NT_STATUS_ACCESS_DENIED;
305 ZERO_STRUCT(priv);
307 priv.privilege = sec_privilege_mask(privilege);
308 priv.mem_ctx = mem_ctx;
310 db->traverse_read(db, priv_traverse_fn, &priv);
312 /* give the memory away; caller will free */
314 *sids = priv.sids.list;
315 *num_sids = priv.sids.count;
317 return NT_STATUS_OK;
320 /***************************************************************************
321 Add privilege to sid
322 ****************************************************************************/
324 static bool grant_privilege_bitmap(const struct dom_sid *sid, const uint64_t priv_mask)
326 uint64_t old_mask, new_mask;
328 ZERO_STRUCT( old_mask );
329 ZERO_STRUCT( new_mask );
331 if ( get_privileges( sid, &old_mask ) )
332 new_mask = old_mask;
333 else
334 new_mask = 0;
336 new_mask |= priv_mask;
338 DEBUG(10,("grant_privilege: %s\n", sid_string_dbg(sid)));
340 DEBUGADD( 10, ("original privilege mask: 0x%llx\n", (unsigned long long)new_mask));
342 DEBUGADD( 10, ("new privilege mask: 0x%llx\n", (unsigned long long)new_mask));
344 return set_privileges( sid, new_mask );
347 /*********************************************************************
348 Add a privilege based on its name
349 *********************************************************************/
351 bool grant_privilege_by_name(const struct dom_sid *sid, const char *name)
353 uint64_t mask;
355 if (! se_priv_from_name(name, &mask)) {
356 DEBUG(3, ("grant_privilege_by_name: "
357 "No Such Privilege Found (%s)\n", name));
358 return False;
361 return grant_privilege_bitmap( sid, mask );
364 /***************************************************************************
365 Grant a privilege set (list of LUID values) from a sid
366 ****************************************************************************/
368 bool grant_privilege_set(const struct dom_sid *sid, struct lsa_PrivilegeSet *set)
370 uint64_t privilege_mask;
371 if (!privilege_set_to_se_priv(&privilege_mask, set)) {
372 return false;
374 return grant_privilege_bitmap(sid, privilege_mask);
377 /***************************************************************************
378 Remove privilege from sid
379 ****************************************************************************/
381 static bool revoke_privilege_bitmap(const struct dom_sid *sid, const uint64_t priv_mask)
383 uint64_t mask;
385 /* if the user has no privileges, then we can't revoke any */
387 if ( !get_privileges( sid, &mask ) )
388 return True;
390 DEBUG(10,("revoke_privilege: %s\n", sid_string_dbg(sid)));
392 DEBUGADD( 10, ("original privilege mask: 0x%llx\n", (unsigned long long)mask));
394 mask &= ~priv_mask;
396 DEBUGADD( 10, ("new privilege mask: 0x%llx\n", (unsigned long long)mask));
398 return set_privileges( sid, mask );
401 /***************************************************************************
402 Remove a privilege set (list of LUID values) from a sid
403 ****************************************************************************/
405 bool revoke_privilege_set(const struct dom_sid *sid, struct lsa_PrivilegeSet *set)
407 uint64_t privilege_mask;
408 if (!privilege_set_to_se_priv(&privilege_mask, set)) {
409 return false;
411 return revoke_privilege_bitmap(sid, privilege_mask);
414 /*********************************************************************
415 Revoke all privileges
416 *********************************************************************/
418 bool revoke_all_privileges( const struct dom_sid *sid )
420 return revoke_privilege_bitmap( sid, SE_ALL_PRIVS);
423 /*********************************************************************
424 Add a privilege based on its name
425 *********************************************************************/
427 bool revoke_privilege_by_name(const struct dom_sid *sid, const char *name)
429 uint64_t mask;
431 if (! se_priv_from_name(name, &mask)) {
432 DEBUG(3, ("revoke_privilege_by_name: "
433 "No Such Privilege Found (%s)\n", name));
434 return False;
437 return revoke_privilege_bitmap(sid, mask);
441 /***************************************************************************
442 Retrieve the SIDs assigned to a given privilege
443 ****************************************************************************/
445 NTSTATUS privilege_create_account(const struct dom_sid *sid )
447 return ( grant_privilege_bitmap(sid, 0) ? NT_STATUS_OK : NT_STATUS_UNSUCCESSFUL);
450 /***************************************************************************
451 Delete a privileged account
452 ****************************************************************************/
454 NTSTATUS privilege_delete_account(const struct dom_sid *sid)
456 struct db_context *db = get_account_pol_db();
457 fstring tmp, keystr;
459 if (!lp_enable_privileges()) {
460 return NT_STATUS_OK;
463 if (!db) {
464 return NT_STATUS_INVALID_HANDLE;
467 if (!sid || (sid->num_auths == 0)) {
468 return NT_STATUS_INVALID_SID;
471 /* PRIV_<SID> (NULL terminated) as the key */
473 fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
475 return dbwrap_delete_bystring(db, keystr);
478 /*******************************************************************
479 *******************************************************************/
481 bool is_privileged_sid( const struct dom_sid *sid )
483 uint64_t mask;
485 return get_privileges( sid, &mask );
488 /*******************************************************************
489 *******************************************************************/
491 bool grant_all_privileges( const struct dom_sid *sid )
493 uint64_t mask;
495 se_priv_put_all_privileges(&mask);
497 return grant_privilege_bitmap( sid, mask );