s3-privs Overhaul PRIVILEGE_SET handling, avoid dealing with the bitmap
[abartlet/samba.git/.git] / source3 / lib / privileges.c
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
8
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.
13
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.
18
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/>.
21 */
22
23
24 #include "includes.h"
25 #include "dbwrap.h"
26
27 #define PRIVPREFIX              "PRIV_"
28
29 typedef struct {
30         uint32_t count;
31         struct dom_sid *list;
32 } SID_LIST;
33
34 typedef struct {
35         TALLOC_CTX *mem_ctx;
36         uint64_t privilege;
37         SID_LIST sids;
38 } PRIV_SID_LIST;
39
40
41 static bool get_privileges( const struct dom_sid *sid, uint64_t *mask )
42 {
43         struct db_context *db = get_account_pol_db();
44         fstring tmp, keystr;
45         TDB_DATA data;
46
47         /* Fail if the admin has not enable privileges */
48
49         if ( !lp_enable_privileges() ) {
50                 return False;
51         }
52
53         if ( db == NULL )
54                 return False;
55
56         /* PRIV_<SID> (NULL terminated) as the key */
57
58         fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
59
60         data = dbwrap_fetch_bystring( db, talloc_tos(), keystr );
61
62         if ( !data.dptr ) {
63                 DEBUG(3, ("get_privileges: No privileges assigned to SID "
64                           "[%s]\n", sid_string_dbg(sid)));
65                 return False;
66         }
67
68         if (data.dsize == 4*4) {
69                 DEBUG(3, ("get_privileges: Should not have obtained old-style privileges record for SID "
70                           "[%s]\n", sid_string_dbg(sid)));
71                 return False;
72         }
73
74         if (data.dsize != sizeof( uint64_t ) ) {
75                 DEBUG(3, ("get_privileges: Invalid privileges record assigned to SID "
76                           "[%s]\n", sid_string_dbg(sid)));
77                 return False;
78         }
79
80         *mask = BVAL(data.dptr, 0);
81
82         TALLOC_FREE(data.dptr);
83
84         return True;
85 }
86
87 /***************************************************************************
88  Store the privilege mask (set) for a given SID
89 ****************************************************************************/
90
91 static bool set_privileges( const struct dom_sid *sid, uint64_t *mask )
92 {
93         struct db_context *db = get_account_pol_db();
94         uint8_t privbuf[8];
95         fstring tmp, keystr;
96         TDB_DATA data;
97
98         if ( !lp_enable_privileges() )
99                 return False;
100
101         if ( db == NULL )
102                 return False;
103
104         if ( !sid || (sid->num_auths == 0) ) {
105                 DEBUG(0,("set_privileges: Refusing to store empty SID!\n"));
106                 return False;
107         }
108
109         /* PRIV_<SID> (NULL terminated) as the key */
110
111         fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
112
113         /* This writes the 64 bit bitmask out in little endian format */
114         SBVAL(privbuf,0,*mask);
115
116         data.dptr  = (uint8 *)mask;
117         data.dsize = sizeof(uint64_t);
118
119         return NT_STATUS_IS_OK(dbwrap_store_bystring(db, keystr, data,
120                                                      TDB_REPLACE));
121 }
122
123 /*********************************************************************
124  get a list of all privileges for all sids in the list
125 *********************************************************************/
126
127 bool get_privileges_for_sids(uint64_t *privileges, struct dom_sid *slist, int scount)
128 {
129         uint64_t mask;
130         int i;
131         bool found = False;
132
133         *privileges = 0;
134
135         for ( i=0; i<scount; i++ ) {
136                 /* don't add unless we actually have a privilege assigned */
137
138                 if ( !get_privileges( &slist[i], &mask ) )
139                         continue;
140
141                 DEBUG(5,("get_privileges_for_sids: sid = %s\nPrivilege "
142                          "set: 0x%llx\n", sid_string_dbg(&slist[i]),
143                          (unsigned long long)mask));
144
145                 *privileges |= mask;
146                 found = True;
147         }
148
149         return found;
150 }
151
152 NTSTATUS get_privileges_for_sid_as_set(TALLOC_CTX *mem_ctx, PRIVILEGE_SET **privileges, struct dom_sid *sid)
153 {
154         uint64_t mask;
155         if (!get_privileges(sid, &mask)) {
156                 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
157         }
158
159         *privileges = talloc_zero(mem_ctx, PRIVILEGE_SET);
160         if (!*privileges) {
161                 return NT_STATUS_NO_MEMORY;
162         }
163
164         if (!se_priv_to_privilege_set(*privileges, mask)) {
165                 return NT_STATUS_NO_MEMORY;
166         }
167         return NT_STATUS_OK;
168 }
169
170 /*********************************************************************
171  traversal functions for privilege_enumerate_accounts
172 *********************************************************************/
173
174 static int priv_traverse_fn(struct db_record *rec, void *state)
175 {
176         PRIV_SID_LIST *priv = (PRIV_SID_LIST *)state;
177         int  prefixlen = strlen(PRIVPREFIX);
178         struct dom_sid sid;
179         fstring sid_string;
180
181         /* easy check first */
182
183         if (rec->value.dsize != sizeof(uint64_t) )
184                 return 0;
185
186         /* check we have a PRIV_+SID entry */
187
188         if ( strncmp((char *)rec->key.dptr, PRIVPREFIX, prefixlen) != 0)
189                 return 0;
190
191         /* check to see if we are looking for a particular privilege */
192
193         fstrcpy( sid_string, (char *)&(rec->key.dptr[strlen(PRIVPREFIX)]) );
194
195         if (priv->privilege != 0) {
196                 uint64_t mask;
197
198                 if (rec->value.dsize == 4*4) {
199                         DEBUG(3, ("get_privileges: Should not have obtained old-style privileges record for SID "
200                                   "[%s]\n", sid_string));
201                         return 0;
202                 }
203
204                 if (rec->value.dsize != sizeof( uint64_t ) ) {
205                         DEBUG(3, ("get_privileges: Invalid privileges record assigned to SID "
206                                   "[%s]\n", sid_string));
207                         return 0;
208                 }
209
210                 mask = BVAL(rec->value.dptr, 0);
211
212                 /* if the SID does not have the specified privilege
213                    then just return */
214
215                 if ((mask & priv->privilege) == 0) {
216                         return 0;
217                 }
218         }
219
220         /* this is a last ditch safety check to preventing returning
221            and invalid SID (i've somehow run into this on development branches) */
222
223         if ( strcmp( "S-0-0", sid_string ) == 0 )
224                 return 0;
225
226         if ( !string_to_sid(&sid, sid_string) ) {
227                 DEBUG(0,("travsersal_fn_enum__acct: Could not convert SID [%s]\n",
228                         sid_string));
229                 return 0;
230         }
231
232         if (!NT_STATUS_IS_OK(add_sid_to_array(priv->mem_ctx, &sid,
233                                               &priv->sids.list,
234                                               &priv->sids.count)))
235         {
236                 return 0;
237         }
238
239         return 0;
240 }
241
242 /*********************************************************************
243  Retreive list of privileged SIDs (for _lsa_enumerate_accounts()
244 *********************************************************************/
245
246 NTSTATUS privilege_enumerate_accounts(struct dom_sid **sids, int *num_sids)
247 {
248         struct db_context *db = get_account_pol_db();
249         PRIV_SID_LIST priv;
250
251         if (db == NULL) {
252                 return NT_STATUS_ACCESS_DENIED;
253         }
254
255         ZERO_STRUCT(priv);
256
257         db->traverse_read(db, priv_traverse_fn, &priv);
258
259         /* give the memory away; caller will free */
260
261         *sids      = priv.sids.list;
262         *num_sids  = priv.sids.count;
263
264         return NT_STATUS_OK;
265 }
266
267 /*********************************************************************
268  Retrieve list of SIDs granted a particular privilege
269 *********************************************************************/
270
271 NTSTATUS privilege_enum_sids(enum sec_privilege privilege, TALLOC_CTX *mem_ctx,
272                              struct dom_sid **sids, int *num_sids)
273 {
274         struct db_context *db = get_account_pol_db();
275         PRIV_SID_LIST priv;
276
277         if (db == NULL) {
278                 return NT_STATUS_ACCESS_DENIED;
279         }
280
281         ZERO_STRUCT(priv);
282
283         priv.privilege = sec_privilege_mask(privilege);
284         priv.mem_ctx = mem_ctx;
285
286         db->traverse_read(db, priv_traverse_fn, &priv);
287
288         /* give the memory away; caller will free */
289
290         *sids      = priv.sids.list;
291         *num_sids  = priv.sids.count;
292
293         return NT_STATUS_OK;
294 }
295
296 /***************************************************************************
297  Add privilege to sid
298 ****************************************************************************/
299
300 static bool grant_privilege_bitmap(const struct dom_sid *sid, const uint64_t priv_mask)
301 {
302         uint64_t old_mask, new_mask;
303
304         ZERO_STRUCT( old_mask );
305         ZERO_STRUCT( new_mask );
306
307         if ( get_privileges( sid, &old_mask ) )
308                 new_mask = old_mask;
309         else
310                 new_mask = 0;
311
312         new_mask |= priv_mask;
313
314         DEBUG(10,("grant_privilege: %s\n", sid_string_dbg(sid)));
315
316         DEBUGADD( 10, ("original privilege mask: 0x%llx\n", (unsigned long long)new_mask));
317
318         DEBUGADD( 10, ("new privilege mask:      0x%llx\n", (unsigned long long)new_mask));
319
320         return set_privileges( sid, &new_mask );
321 }
322
323 /*********************************************************************
324  Add a privilege based on its name
325 *********************************************************************/
326
327 bool grant_privilege_by_name(struct dom_sid *sid, const char *name)
328 {
329         uint64_t mask;
330
331         if (! se_priv_from_name(name, &mask)) {
332                 DEBUG(3, ("grant_privilege_by_name: "
333                           "No Such Privilege Found (%s)\n", name));
334                 return False;
335         }
336
337         return grant_privilege_bitmap( sid, mask );
338 }
339
340 /***************************************************************************
341  Grant a privilege set (list of LUID values) from a sid
342 ****************************************************************************/
343
344 bool grant_privilege_set(const struct dom_sid *sid, struct lsa_PrivilegeSet *set)
345 {
346         uint64_t privilege_mask;
347         if (!privilege_set_to_se_priv(&privilege_mask, set)) {
348                 return false;
349         }
350         return grant_privilege_bitmap(sid, privilege_mask);
351 }
352
353 /***************************************************************************
354  Remove privilege from sid
355 ****************************************************************************/
356
357 static bool revoke_privilege_bitmap(const struct dom_sid *sid, const uint64_t priv_mask)
358 {
359         uint64_t mask;
360
361         /* if the user has no privileges, then we can't revoke any */
362
363         if ( !get_privileges( sid, &mask ) )
364                 return True;
365
366         DEBUG(10,("revoke_privilege: %s\n", sid_string_dbg(sid)));
367
368         DEBUGADD( 10, ("original privilege mask: 0x%llx\n", (unsigned long long)mask));
369
370         mask &= ~priv_mask;
371
372         DEBUGADD( 10, ("new privilege mask:      0x%llx\n", (unsigned long long)mask));
373
374         return set_privileges( sid, &mask );
375 }
376
377 /***************************************************************************
378  Remove a privilege set (list of LUID values) from a sid
379 ****************************************************************************/
380
381 bool revoke_privilege_set(const struct dom_sid *sid, struct lsa_PrivilegeSet *set)
382 {
383         uint64_t privilege_mask;
384         if (!privilege_set_to_se_priv(&privilege_mask, set)) {
385                 return false;
386         }
387         return revoke_privilege_bitmap(sid, privilege_mask);
388 }
389
390 /*********************************************************************
391  Revoke all privileges
392 *********************************************************************/
393
394 bool revoke_all_privileges( struct dom_sid *sid )
395 {
396         return revoke_privilege_bitmap( sid, SE_ALL_PRIVS);
397 }
398
399 /*********************************************************************
400  Add a privilege based on its name
401 *********************************************************************/
402
403 bool revoke_privilege_by_name(struct dom_sid *sid, const char *name)
404 {
405         uint64_t mask;
406
407         if (! se_priv_from_name(name, &mask)) {
408                 DEBUG(3, ("revoke_privilege_by_name: "
409                           "No Such Privilege Found (%s)\n", name));
410                 return False;
411         }
412
413         return revoke_privilege_bitmap(sid, mask);
414
415 }
416
417 /***************************************************************************
418  Retrieve the SIDs assigned to a given privilege
419 ****************************************************************************/
420
421 NTSTATUS privilege_create_account(const struct dom_sid *sid )
422 {
423         return ( grant_privilege_bitmap(sid, 0) ? NT_STATUS_OK : NT_STATUS_UNSUCCESSFUL);
424 }
425
426 /***************************************************************************
427  Delete a privileged account
428 ****************************************************************************/
429
430 NTSTATUS privilege_delete_account(const struct dom_sid *sid)
431 {
432         struct db_context *db = get_account_pol_db();
433         fstring tmp, keystr;
434
435         if (!lp_enable_privileges()) {
436                 return NT_STATUS_OK;
437         }
438
439         if (!db) {
440                 return NT_STATUS_INVALID_HANDLE;
441         }
442
443         if (!sid || (sid->num_auths == 0)) {
444                 return NT_STATUS_INVALID_SID;
445         }
446
447         /* PRIV_<SID> (NULL terminated) as the key */
448
449         fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
450
451         return dbwrap_delete_bystring(db, keystr);
452 }
453
454 /****************************************************************************
455  duplicate alloc luid_attr
456  ****************************************************************************/
457
458 NTSTATUS dup_luid_attr(TALLOC_CTX *mem_ctx, struct lsa_LUIDAttribute **new_la, struct lsa_LUIDAttribute *old_la, int count)
459 {
460         int i;
461
462         if ( !old_la )
463                 return NT_STATUS_OK;
464
465         if (count) {
466                 *new_la = TALLOC_ARRAY(mem_ctx, struct lsa_LUIDAttribute, count);
467                 if ( !*new_la ) {
468                         DEBUG(0,("dup_luid_attr: failed to alloc new struct lsa_LUIDAttribute array [%d]\n", count));
469                         return NT_STATUS_NO_MEMORY;
470                 }
471         } else {
472                 *new_la = NULL;
473         }
474
475         for (i=0; i<count; i++) {
476                 (*new_la)[i].luid.high = old_la[i].luid.high;
477                 (*new_la)[i].luid.low = old_la[i].luid.low;
478                 (*new_la)[i].attribute = old_la[i].attribute;
479         }
480
481         return NT_STATUS_OK;
482 }
483
484 /*******************************************************************
485 *******************************************************************/
486
487 bool is_privileged_sid( const struct dom_sid *sid )
488 {
489         uint64_t mask;
490
491         return get_privileges( sid, &mask );
492 }
493
494 /*******************************************************************
495 *******************************************************************/
496
497 bool grant_all_privileges( const struct dom_sid *sid )
498 {
499         uint64_t mask;
500
501         if (!se_priv_put_all_privileges(&mask)) {
502                 return False;
503         }
504
505         return grant_privilege_bitmap( sid, mask );
506 }