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