57c4d81fa4a6e177998d41d7f3fe56001cc53608
[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         se_priv_copy( privileges, &se_priv_none );
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                 se_priv_add( privileges, &mask );
146                 found = True;
147         }
148
149         return found;
150 }
151
152
153 /*********************************************************************
154  traversal functions for privilege_enumerate_accounts
155 *********************************************************************/
156
157 static int priv_traverse_fn(struct db_record *rec, void *state)
158 {
159         PRIV_SID_LIST *priv = (PRIV_SID_LIST *)state;
160         int  prefixlen = strlen(PRIVPREFIX);
161         struct dom_sid sid;
162         fstring sid_string;
163
164         /* easy check first */
165
166         if (rec->value.dsize != sizeof(uint64_t) )
167                 return 0;
168
169         /* check we have a PRIV_+SID entry */
170
171         if ( strncmp((char *)rec->key.dptr, PRIVPREFIX, prefixlen) != 0)
172                 return 0;
173
174         /* check to see if we are looking for a particular privilege */
175
176         if ( !se_priv_equal(&priv->privilege, &se_priv_none) ) {
177                 uint64_t mask;
178
179                 se_priv_copy( &mask, (uint64_t*)rec->value.dptr );
180
181                 /* if the SID does not have the specified privilege
182                    then just return */
183
184                 if ( !is_privilege_assigned( &mask, &priv->privilege) )
185                         return 0;
186         }
187
188         fstrcpy( sid_string, (char *)&(rec->key.dptr[strlen(PRIVPREFIX)]) );
189
190         /* this is a last ditch safety check to preventing returning
191            and invalid SID (i've somehow run into this on development branches) */
192
193         if ( strcmp( "S-0-0", sid_string ) == 0 )
194                 return 0;
195
196         if ( !string_to_sid(&sid, sid_string) ) {
197                 DEBUG(0,("travsersal_fn_enum__acct: Could not convert SID [%s]\n",
198                         sid_string));
199                 return 0;
200         }
201
202         if (!NT_STATUS_IS_OK(add_sid_to_array(priv->mem_ctx, &sid,
203                                               &priv->sids.list,
204                                               &priv->sids.count)))
205         {
206                 return 0;
207         }
208
209         return 0;
210 }
211
212 /*********************************************************************
213  Retreive list of privileged SIDs (for _lsa_enumerate_accounts()
214 *********************************************************************/
215
216 NTSTATUS privilege_enumerate_accounts(struct dom_sid **sids, int *num_sids)
217 {
218         struct db_context *db = get_account_pol_db();
219         PRIV_SID_LIST priv;
220
221         if (db == NULL) {
222                 return NT_STATUS_ACCESS_DENIED;
223         }
224
225         ZERO_STRUCT(priv);
226
227         se_priv_copy( &priv.privilege, &se_priv_none );
228
229         db->traverse_read(db, priv_traverse_fn, &priv);
230
231         /* give the memory away; caller will free */
232
233         *sids      = priv.sids.list;
234         *num_sids  = priv.sids.count;
235
236         return NT_STATUS_OK;
237 }
238
239 /*********************************************************************
240  Retrieve list of SIDs granted a particular privilege
241 *********************************************************************/
242
243 NTSTATUS privilege_enum_sids(const uint64_t *mask, TALLOC_CTX *mem_ctx,
244                              struct dom_sid **sids, int *num_sids)
245 {
246         struct db_context *db = get_account_pol_db();
247         PRIV_SID_LIST priv;
248
249         if (db == NULL) {
250                 return NT_STATUS_ACCESS_DENIED;
251         }
252
253         ZERO_STRUCT(priv);
254
255         se_priv_copy(&priv.privilege, mask);
256         priv.mem_ctx = mem_ctx;
257
258         db->traverse_read(db, priv_traverse_fn, &priv);
259
260         /* give the memory away; caller will free */
261
262         *sids      = priv.sids.list;
263         *num_sids  = priv.sids.count;
264
265         return NT_STATUS_OK;
266 }
267
268 /***************************************************************************
269  Add privilege to sid
270 ****************************************************************************/
271
272 bool grant_privilege(const struct dom_sid *sid, const uint64_t *priv_mask)
273 {
274         uint64_t old_mask, new_mask;
275
276         ZERO_STRUCT( old_mask );
277         ZERO_STRUCT( new_mask );
278
279         if ( get_privileges( sid, &old_mask ) )
280                 se_priv_copy( &new_mask, &old_mask );
281         else
282                 se_priv_copy( &new_mask, &se_priv_none );
283
284         se_priv_add( &new_mask, priv_mask );
285
286         DEBUG(10,("grant_privilege: %s\n", sid_string_dbg(sid)));
287
288         DEBUGADD( 10, ("original privilege mask: 0x%llx\n", (unsigned long long)new_mask));
289
290         DEBUGADD( 10, ("new privilege mask:      0x%llx\n", (unsigned long long)new_mask));
291
292         return set_privileges( sid, &new_mask );
293 }
294
295 /*********************************************************************
296  Add a privilege based on its name
297 *********************************************************************/
298
299 bool grant_privilege_by_name(struct dom_sid *sid, const char *name)
300 {
301         uint64_t mask;
302
303         if (! se_priv_from_name(name, &mask)) {
304                 DEBUG(3, ("grant_privilege_by_name: "
305                           "No Such Privilege Found (%s)\n", name));
306                 return False;
307         }
308
309         return grant_privilege( sid, &mask );
310 }
311
312 /***************************************************************************
313  Remove privilege from sid
314 ****************************************************************************/
315
316 bool revoke_privilege(const struct dom_sid *sid, const uint64_t priv_mask)
317 {
318         uint64_t mask;
319
320         /* if the user has no privileges, then we can't revoke any */
321
322         if ( !get_privileges( sid, &mask ) )
323                 return True;
324
325         DEBUG(10,("revoke_privilege: %s\n", sid_string_dbg(sid)));
326
327         DEBUGADD( 10, ("original privilege mask: 0x%llx\n", (unsigned long long)mask));
328
329         mask &= ~priv_mask;
330
331         DEBUGADD( 10, ("new privilege mask:      0x%llx\n", (unsigned long long)mask));
332
333         return set_privileges( sid, &mask );
334 }
335
336 /*********************************************************************
337  Revoke all privileges
338 *********************************************************************/
339
340 bool revoke_all_privileges( struct dom_sid *sid )
341 {
342         return revoke_privilege( sid, SE_ALL_PRIVS);
343 }
344
345 /*********************************************************************
346  Add a privilege based on its name
347 *********************************************************************/
348
349 bool revoke_privilege_by_name(struct dom_sid *sid, const char *name)
350 {
351         uint64_t mask;
352
353         if (! se_priv_from_name(name, &mask)) {
354                 DEBUG(3, ("revoke_privilege_by_name: "
355                           "No Such Privilege Found (%s)\n", name));
356                 return False;
357         }
358
359         return revoke_privilege(sid, mask);
360
361 }
362
363 /***************************************************************************
364  Retrieve the SIDs assigned to a given privilege
365 ****************************************************************************/
366
367 NTSTATUS privilege_create_account(const struct dom_sid *sid )
368 {
369         return ( grant_privilege(sid, &se_priv_none) ? NT_STATUS_OK : NT_STATUS_UNSUCCESSFUL);
370 }
371
372 /***************************************************************************
373  Delete a privileged account
374 ****************************************************************************/
375
376 NTSTATUS privilege_delete_account(const struct dom_sid *sid)
377 {
378         struct db_context *db = get_account_pol_db();
379         fstring tmp, keystr;
380
381         if (!lp_enable_privileges()) {
382                 return NT_STATUS_OK;
383         }
384
385         if (!db) {
386                 return NT_STATUS_INVALID_HANDLE;
387         }
388
389         if (!sid || (sid->num_auths == 0)) {
390                 return NT_STATUS_INVALID_SID;
391         }
392
393         /* PRIV_<SID> (NULL terminated) as the key */
394
395         fstr_sprintf(keystr, "%s%s", PRIVPREFIX, sid_to_fstring(tmp, sid));
396
397         return dbwrap_delete_bystring(db, keystr);
398 }
399
400 /****************************************************************************
401  initialise a privilege list and set the talloc context
402  ****************************************************************************/
403
404 NTSTATUS privilege_set_init(PRIVILEGE_SET *priv_set)
405 {
406         TALLOC_CTX *mem_ctx;
407
408         ZERO_STRUCTP( priv_set );
409
410         mem_ctx = talloc_init("privilege set");
411         if ( !mem_ctx ) {
412                 DEBUG(0,("privilege_set_init: failed to initialize talloc ctx!\n"));
413                 return NT_STATUS_NO_MEMORY;
414         }
415
416         priv_set->mem_ctx = mem_ctx;
417
418         return NT_STATUS_OK;
419 }
420
421 /****************************************************************************
422   initialise a privilege list and with someone else's talloc context
423 ****************************************************************************/
424
425 NTSTATUS privilege_set_init_by_ctx(TALLOC_CTX *mem_ctx, PRIVILEGE_SET *priv_set)
426 {
427         ZERO_STRUCTP( priv_set );
428
429         priv_set->mem_ctx = mem_ctx;
430         priv_set->ext_ctx = True;
431
432         return NT_STATUS_OK;
433 }
434
435 /****************************************************************************
436  Free all memory used by a PRIVILEGE_SET
437 ****************************************************************************/
438
439 void privilege_set_free(PRIVILEGE_SET *priv_set)
440 {
441         if ( !priv_set )
442                 return;
443
444         if ( !( priv_set->ext_ctx ) )
445                 talloc_destroy( priv_set->mem_ctx );
446
447         ZERO_STRUCTP( priv_set );
448 }
449
450 /****************************************************************************
451  duplicate alloc luid_attr
452  ****************************************************************************/
453
454 NTSTATUS dup_luid_attr(TALLOC_CTX *mem_ctx, struct lsa_LUIDAttribute **new_la, struct lsa_LUIDAttribute *old_la, int count)
455 {
456         int i;
457
458         if ( !old_la )
459                 return NT_STATUS_OK;
460
461         if (count) {
462                 *new_la = TALLOC_ARRAY(mem_ctx, struct lsa_LUIDAttribute, count);
463                 if ( !*new_la ) {
464                         DEBUG(0,("dup_luid_attr: failed to alloc new struct lsa_LUIDAttribute array [%d]\n", count));
465                         return NT_STATUS_NO_MEMORY;
466                 }
467         } else {
468                 *new_la = NULL;
469         }
470
471         for (i=0; i<count; i++) {
472                 (*new_la)[i].luid.high = old_la[i].luid.high;
473                 (*new_la)[i].luid.low = old_la[i].luid.low;
474                 (*new_la)[i].attribute = old_la[i].attribute;
475         }
476
477         return NT_STATUS_OK;
478 }
479
480 /*******************************************************************
481 *******************************************************************/
482
483 bool is_privileged_sid( const struct dom_sid *sid )
484 {
485         uint64_t mask;
486
487         return get_privileges( sid, &mask );
488 }
489
490 /*******************************************************************
491 *******************************************************************/
492
493 bool grant_all_privileges( const struct dom_sid *sid )
494 {
495         uint64_t mask;
496
497         if (!se_priv_put_all_privileges(&mask)) {
498                 return False;
499         }
500
501         return grant_privilege( sid, &mask );
502 }