830e759d0939bf1d455586251b87f1850c21b47c
[nivanova/samba-autobuild/.git] / source3 / smbd / sec_ctx.c
1 /* 
2    Unix SMB/Netbios implementation.
3    Version 1.9.
4    uid/user handling
5    Copyright (C) Tim Potter 2000
6    
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 */
21
22 #include "includes.h"
23
24 extern int DEBUGLEVEL;
25 extern struct current_user current_user;
26
27 struct sec_ctx {
28         uid_t uid;
29         uid_t gid;
30         int ngroups;
31         gid_t *groups;
32         NT_USER_TOKEN *token;
33 };
34
35 /* A stack of security contexts.  We include the current context as being
36    the first one, so there is room for another MAX_SEC_CTX_DEPTH more. */
37
38 static struct sec_ctx sec_ctx_stack[MAX_SEC_CTX_DEPTH + 1];
39 static int sec_ctx_stack_ndx;
40
41 /****************************************************************************
42  Become the specified uid.
43 ****************************************************************************/
44
45 static BOOL become_uid(uid_t uid)
46 {
47         /* Check for dodgy uid values */
48
49         if (uid == (uid_t)-1 || 
50             ((sizeof(uid_t) == 2) && (uid == (uid_t)65535))) {
51                 static int done;
52  
53                 if (!done) {
54                         DEBUG(1,("WARNING: using uid %d is a security risk\n",
55                                  (int)uid));
56                         done = 1;
57                 }
58         }
59
60         /* Set effective user id */
61
62         set_effective_uid(uid);
63         current_user.uid = uid;
64
65         DO_PROFILE_INC(uid_changes);
66
67         return True;
68 }
69
70 /****************************************************************************
71  Become the specified gid.
72 ****************************************************************************/
73
74 static BOOL become_gid(gid_t gid)
75 {
76         /* Check for dodgy gid values */
77
78         if (gid == (gid_t)-1 || ((sizeof(gid_t) == 2) && 
79                                  (gid == (gid_t)65535))) {
80                 static int done;
81                 
82                 if (!done) {
83                         DEBUG(1,("WARNING: using gid %d is a security risk\n",
84                                  (int)gid));  
85                         done = 1;
86                 }
87         }
88   
89         /* Set effective group id */
90
91         set_effective_gid(gid);
92         current_user.gid = gid;
93         
94         return True;
95 }
96
97 /****************************************************************************
98  Become the specified uid and gid.
99 ****************************************************************************/
100
101 static BOOL become_id(uid_t uid, gid_t gid)
102 {
103         return become_gid(gid) && become_uid(uid);
104 }
105
106 /****************************************************************************
107  Drop back to root privileges in order to change to another user.
108 ****************************************************************************/
109
110 static void gain_root(void)
111 {
112         if (non_root_mode()) {
113                 return;
114         }
115
116         if (geteuid() != 0) {
117                 set_effective_uid(0);
118
119                 if (geteuid() != 0) {
120                         DEBUG(0,
121                               ("Warning: You appear to have a trapdoor "
122                                "uid system\n"));
123                 }
124         }
125
126         if (getegid() != 0) {
127                 set_effective_gid(0);
128
129                 if (getegid() != 0) {
130                         DEBUG(0,
131                               ("Warning: You appear to have a trapdoor "
132                                "gid system\n"));
133                 }
134         }
135 }
136
137 /****************************************************************************
138  Get the list of current groups.
139 ****************************************************************************/
140
141 int get_current_groups(int *p_ngroups, gid_t **p_groups)
142 {
143         int i;
144         gid_t grp;
145         int ngroups = sys_getgroups(0,&grp);
146         gid_t *groups;
147
148         (*p_ngroups) = 0;
149         (*p_groups) = NULL;
150
151         if (ngroups <= 0)
152                 return -1;
153
154         if((groups = (gid_t *)malloc(sizeof(gid_t)*ngroups)) == NULL) {
155                 DEBUG(0,("setup_groups malloc fail !\n"));
156                 return -1;
157         }
158
159         if ((ngroups = sys_getgroups(ngroups,groups)) == -1) {
160                 SAFE_FREE(groups);
161                 return -1;
162         }
163
164         (*p_ngroups) = ngroups;
165         (*p_groups) = groups;
166
167         DEBUG( 3, ( "get_current_groups: uid %u is in %u groups: ", (unsigned int)getuid() , ngroups ) );
168         for (i = 0; i < ngroups; i++ ) {
169                 DEBUG( 3, ( "%s%d", (i ? ", " : ""), (int)groups[i] ) );
170         }
171         DEBUG( 3, ( "\n" ) );
172
173     return ngroups;
174 }
175
176 /****************************************************************************
177  Delete a SID token.
178 ****************************************************************************/
179
180 void delete_nt_token(NT_USER_TOKEN **pptoken)
181 {
182     if (*pptoken) {
183                 NT_USER_TOKEN *ptoken = *pptoken;
184         SAFE_FREE( ptoken->user_sids );
185         ZERO_STRUCTP(ptoken);
186     }
187     SAFE_FREE(*pptoken);
188 }
189
190 /****************************************************************************
191  Duplicate a SID token.
192 ****************************************************************************/
193
194 NT_USER_TOKEN *dup_nt_token(NT_USER_TOKEN *ptoken)
195 {
196         NT_USER_TOKEN *token;
197
198         if (!ptoken)
199                 return NULL;
200
201     if ((token = (NT_USER_TOKEN *)malloc( sizeof(NT_USER_TOKEN) ) ) == NULL)
202         return NULL;
203
204     ZERO_STRUCTP(token);
205
206     if ((token->user_sids = (DOM_SID *)memdup( ptoken->user_sids, sizeof(DOM_SID) * ptoken->num_sids )) == NULL) {
207         SAFE_FREE(token);
208         return NULL;
209     }
210
211     token->num_sids = ptoken->num_sids;
212
213         return token;
214 }
215
216 /****************************************************************************
217  Initialize the groups a user belongs to.
218 ****************************************************************************/
219
220 BOOL initialise_groups(char *user, uid_t uid, gid_t gid)
221 {
222         struct sec_ctx *prev_ctx_p;
223         BOOL result = True;
224
225         if (non_root_mode()) {
226                 return True;
227         }
228
229         become_root();
230
231         /* Call initgroups() to get user groups */
232
233         if (winbind_initgroups(user,gid) == -1) {
234                 DEBUG(0,("Unable to initgroups. Error was %s\n", strerror(errno) ));
235                 if (getuid() == 0) {
236                         if (gid < 0 || gid > 32767 || uid < 0 || uid > 32767) {
237                                 DEBUG(0,("This is probably a problem with the account %s\n", user));
238                         }
239                 }
240                 result = False;
241                 goto done;
242         }
243
244         /* Store groups in previous user's security context.  This will
245            always work as the become_root() call increments the stack
246            pointer. */
247
248         prev_ctx_p = &sec_ctx_stack[sec_ctx_stack_ndx - 1];
249
250         SAFE_FREE(prev_ctx_p->groups);
251         prev_ctx_p->ngroups = 0;
252
253         get_current_groups(&prev_ctx_p->ngroups, &prev_ctx_p->groups);
254
255  done:
256         unbecome_root();
257
258         return result;
259 }
260
261 /****************************************************************************
262  Create a new security context on the stack.  It is the same as the old
263  one.  User changes are done using the set_sec_ctx() function.
264 ****************************************************************************/
265
266 BOOL push_sec_ctx(void)
267 {
268         struct sec_ctx *ctx_p;
269
270         /* Check we don't overflow our stack */
271
272         if (sec_ctx_stack_ndx == MAX_SEC_CTX_DEPTH) {
273                 DEBUG(0, ("Security context stack overflow!\n"));
274                 smb_panic("Security context stack overflow!\n");
275         }
276
277         /* Store previous user context */
278
279         sec_ctx_stack_ndx++;
280
281         ctx_p = &sec_ctx_stack[sec_ctx_stack_ndx];
282
283         ctx_p->uid = geteuid();
284         ctx_p->gid = getegid();
285
286         DEBUG(3, ("push_sec_ctx(%u, %u) : sec_ctx_stack_ndx = %d\n", 
287                   (unsigned int)ctx_p->uid, (unsigned int)ctx_p->gid, sec_ctx_stack_ndx ));
288
289         ctx_p->token = dup_nt_token(sec_ctx_stack[sec_ctx_stack_ndx-1].token);
290
291         ctx_p->ngroups = sys_getgroups(0, NULL);
292
293         if (ctx_p->ngroups != 0) {
294                 if (!(ctx_p->groups = malloc(ctx_p->ngroups * sizeof(gid_t)))) {
295                         DEBUG(0, ("Out of memory in push_sec_ctx()\n"));
296                         delete_nt_token(&ctx_p->token);
297                         return False;
298                 }
299
300                 sys_getgroups(ctx_p->ngroups, ctx_p->groups);
301         } else {
302                 ctx_p->groups = NULL;
303         }
304
305         return True;
306 }
307
308 /****************************************************************************
309  Set the current security context to a given user.
310 ****************************************************************************/
311
312 void set_sec_ctx(uid_t uid, gid_t gid, int ngroups, gid_t *groups, NT_USER_TOKEN *token)
313 {
314         struct sec_ctx *ctx_p = &sec_ctx_stack[sec_ctx_stack_ndx];
315         
316         /* Set the security context */
317
318         DEBUG(3, ("setting sec ctx (%u, %u) - sec_ctx_stack_ndx = %d\n", 
319                 (unsigned int)uid, (unsigned int)gid, sec_ctx_stack_ndx));
320
321         if (ngroups) {
322                 int i;
323
324                 DEBUG(3, ("%d user groups: \n", ngroups));
325                 for (i = 0; i < ngroups; i++) {
326                         DEBUGADD(3, ("%u ", (unsigned int)groups[i]));
327                 }
328
329                 DEBUG(3, ("\n"));
330         }
331         
332
333         gain_root();
334
335 #ifdef HAVE_SETGROUPS
336         sys_setgroups(ngroups, groups);
337 #endif
338
339         ctx_p->ngroups = ngroups;
340
341         SAFE_FREE(ctx_p->groups);
342         if (token && (token == ctx_p->token))
343                 smb_panic("DUPLICATE_TOKEN");
344
345         delete_nt_token(&ctx_p->token);
346         
347         ctx_p->groups = memdup(groups, sizeof(gid_t) * ngroups);
348         ctx_p->token = dup_nt_token(token);
349
350         become_id(uid, gid);
351
352         ctx_p->uid = uid;
353         ctx_p->gid = gid;
354
355         /* Update current_user stuff */
356
357         current_user.uid = uid;
358         current_user.gid = gid;
359         current_user.ngroups = ngroups;
360         current_user.groups = groups;
361         current_user.nt_user_token = ctx_p->token;
362 }
363
364 /****************************************************************************
365  Become root context.
366 ****************************************************************************/
367
368 void set_root_sec_ctx(void)
369 {
370         /* May need to worry about supplementary groups at some stage */
371
372         set_sec_ctx(0, 0, 0, NULL, NULL);
373 }
374
375 /****************************************************************************
376  Pop a security context from the stack.
377 ****************************************************************************/
378
379 BOOL pop_sec_ctx(void)
380 {
381         struct sec_ctx *ctx_p;
382         struct sec_ctx *prev_ctx_p;
383
384         /* Check for stack underflow */
385
386         if (sec_ctx_stack_ndx == 0) {
387                 DEBUG(0, ("Security context stack underflow!\n"));
388                 smb_panic("Security context stack underflow!\n");
389         }
390
391         ctx_p = &sec_ctx_stack[sec_ctx_stack_ndx];
392
393         /* Clear previous user info */
394
395         ctx_p->uid = (uid_t)-1;
396         ctx_p->gid = (gid_t)-1;
397
398         SAFE_FREE(ctx_p->groups);
399         ctx_p->ngroups = 0;
400
401         delete_nt_token(&ctx_p->token);
402
403         /* Pop back previous user */
404
405         sec_ctx_stack_ndx--;
406
407         gain_root();
408
409         prev_ctx_p = &sec_ctx_stack[sec_ctx_stack_ndx];
410
411 #ifdef HAVE_SETGROUPS
412         sys_setgroups(prev_ctx_p->ngroups, prev_ctx_p->groups);
413 #endif
414
415         become_id(prev_ctx_p->uid, prev_ctx_p->gid);
416
417         /* Update current_user stuff */
418
419         current_user.uid = prev_ctx_p->uid;
420         current_user.gid = prev_ctx_p->gid;
421         current_user.ngroups = prev_ctx_p->ngroups;
422         current_user.groups = prev_ctx_p->groups;
423         current_user.nt_user_token = prev_ctx_p->token;
424
425         DEBUG(3, ("pop_sec_ctx (%u, %u) - sec_ctx_stack_ndx = %d\n", 
426                 (unsigned int)geteuid(), (unsigned int)getegid(), sec_ctx_stack_ndx));
427
428         return True;
429 }
430
431 /* Initialise the security context system */
432
433 void init_sec_ctx(void)
434 {
435         int i;
436         struct sec_ctx *ctx_p;
437
438         /* Initialise security context stack */
439
440         memset(sec_ctx_stack, 0, sizeof(struct sec_ctx) * MAX_SEC_CTX_DEPTH);
441
442         for (i = 0; i < MAX_SEC_CTX_DEPTH; i++) {
443                 sec_ctx_stack[i].uid = (uid_t)-1;
444                 sec_ctx_stack[i].gid = (gid_t)-1;
445         }
446
447         /* Initialise first level of stack.  It is the current context */
448         ctx_p = &sec_ctx_stack[0];
449
450         ctx_p->uid = geteuid();
451         ctx_p->gid = getegid();
452
453         get_current_groups(&ctx_p->ngroups, &ctx_p->groups);
454
455         ctx_p->token = NULL; /* Maps to guest user. */
456
457         /* Initialise current_user global */
458
459         current_user.uid = ctx_p->uid;
460         current_user.gid = ctx_p->gid;
461         current_user.ngroups = ctx_p->ngroups;
462         current_user.groups = ctx_p->groups;
463
464         /* The conn and vuid are usually taken care of by other modules.
465            We initialise them here. */
466
467         current_user.conn = NULL;
468         current_user.vuid = UID_FIELD_INVALID;
469         current_user.nt_user_token = NULL;
470 }