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