s3:util_sec: add a cache to set_thread_credentials()
[Samba.git] / source3 / lib / util_sec.c
blob703c522d77b88be84e43c160c42d31bd5ab64ee8
1 /*
2 Unix SMB/CIFS implementation.
3 Copyright (C) Jeremy Allison 1998.
4 rewritten for version 2.0.6 by Tridge
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 3 of the License, or
9 (at your option) any later version.
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.
16 You should have received a copy of the GNU General Public License
17 along with this program. If not, see <http://www.gnu.org/licenses/>.
20 #ifndef AUTOCONF_TEST
21 #include "includes.h"
22 #include "system/passwd.h" /* uid_wrapper */
23 #include "../lib/util/setid.h"
25 #else
26 /* we are running this code in autoconf test mode to see which type of setuid
27 function works */
28 #if defined(HAVE_UNISTD_H)
29 #include <unistd.h>
30 #endif
31 #include <stdlib.h>
32 #include <stdio.h>
33 #include <sys/types.h>
34 #include <errno.h>
36 #ifdef HAVE_SYS_PRIV_H
37 #include <sys/priv.h>
38 #endif
39 #ifdef HAVE_SYS_ID_H
40 #include <sys/id.h>
41 #endif
43 #define DEBUG(x, y) printf y
44 #define smb_panic(x) exit(1)
45 #define bool int
46 #endif
48 /* are we running as non-root? This is used by the regresison test code,
49 and potentially also for sites that want non-root smbd */
50 static uid_t initial_uid;
51 static gid_t initial_gid;
53 /****************************************************************************
54 remember what uid we got started as - this allows us to run correctly
55 as non-root while catching trapdoor systems
56 ****************************************************************************/
58 void sec_init(void)
60 static int initialized;
62 if (!initialized) {
64 #ifndef AUTOCONF_TEST
65 if (uid_wrapper_enabled()) {
66 setenv("UID_WRAPPER_MYUID", "1", 1);
68 #endif
70 initial_uid = geteuid();
71 initial_gid = getegid();
73 #ifndef AUTOCONF_TEST
74 if (uid_wrapper_enabled()) {
75 unsetenv("UID_WRAPPER_MYUID");
77 #endif
79 initialized = 1;
83 /****************************************************************************
84 some code (eg. winbindd) needs to know what uid we started as
85 ****************************************************************************/
86 uid_t sec_initial_uid(void)
88 return initial_uid;
91 /****************************************************************************
92 some code (eg. winbindd, profiling shm) needs to know what gid we started as
93 ****************************************************************************/
94 gid_t sec_initial_gid(void)
96 return initial_gid;
99 /**
100 * @brief Check if we are running in root mode.
102 * @return If we samba root privileges it returns true, false otehrwise.
104 bool root_mode(void)
106 uid_t euid;
108 euid = geteuid();
110 #ifndef AUTOCONF_TEST
111 if (uid_wrapper_enabled()) {
112 return (euid == initial_uid || euid == (uid_t)0);
114 #endif
116 return (initial_uid == euid);
119 /****************************************************************************
120 are we running in non-root mode?
121 ****************************************************************************/
122 bool non_root_mode(void)
124 return (initial_uid != (uid_t)0);
127 /****************************************************************************
128 abort if we haven't set the uid correctly
129 ****************************************************************************/
130 static void assert_uid(uid_t ruid, uid_t euid)
132 if ((euid != (uid_t)-1 && geteuid() != euid) ||
133 (ruid != (uid_t)-1 && getuid() != ruid)) {
134 if (!non_root_mode()) {
135 DEBUG(0,("Failed to set uid privileges to (%d,%d) now set to (%d,%d)\n",
136 (int)ruid, (int)euid,
137 (int)getuid(), (int)geteuid()));
138 smb_panic("failed to set uid\n");
139 exit(1);
144 /****************************************************************************
145 abort if we haven't set the gid correctly
146 ****************************************************************************/
147 static void assert_gid(gid_t rgid, gid_t egid)
149 if ((egid != (gid_t)-1 && getegid() != egid) ||
150 (rgid != (gid_t)-1 && getgid() != rgid)) {
151 if (!non_root_mode()) {
152 DEBUG(0,("Failed to set gid privileges to (%d,%d) now set to (%d,%d) uid=(%d,%d)\n",
153 (int)rgid, (int)egid,
154 (int)getgid(), (int)getegid(),
155 (int)getuid(), (int)geteuid()));
156 smb_panic("failed to set gid\n");
157 exit(1);
162 /****************************************************************************
163 Gain root privilege before doing something.
164 We want to end up with ruid==euid==0
165 ****************************************************************************/
166 void gain_root_privilege(void)
168 #if defined(USE_SETRESUID) || defined(HAVE_LINUX_THREAD_CREDENTIALS)
169 samba_setresuid(0,0,0);
170 #endif
172 #if USE_SETEUID
173 samba_seteuid(0);
174 #endif
176 #if USE_SETREUID
177 samba_setreuid(0, 0);
178 #endif
180 #if USE_SETUIDX
181 samba_setuidx(ID_EFFECTIVE, 0);
182 samba_setuidx(ID_REAL, 0);
183 #endif
185 /* this is needed on some systems */
186 samba_setuid(0);
188 assert_uid(0, 0);
192 /****************************************************************************
193 Ensure our real and effective groups are zero.
194 we want to end up with rgid==egid==0
195 ****************************************************************************/
196 void gain_root_group_privilege(void)
198 #if defined(USE_SETRESUID) || defined(HAVE_LINUX_THREAD_CREDENTIALS)
199 samba_setresgid(0,0,0);
200 #endif
202 #if USE_SETREUID
203 samba_setregid(0,0);
204 #endif
206 #if USE_SETEUID
207 samba_setegid(0);
208 #endif
210 #if USE_SETUIDX
211 samba_setgidx(ID_EFFECTIVE, 0);
212 samba_setgidx(ID_REAL, 0);
213 #endif
215 samba_setgid(0);
217 assert_gid(0, 0);
221 /****************************************************************************
222 Set effective uid, and possibly the real uid too.
223 We want to end up with either:
225 ruid==uid and euid==uid
229 ruid==0 and euid==uid
231 depending on what the local OS will allow us to regain root from.
232 ****************************************************************************/
233 void set_effective_uid(uid_t uid)
235 #if defined(USE_SETRESUID) || defined(HAVE_LINUX_THREAD_CREDENTIALS)
236 /* Set the effective as well as the real uid. */
237 if (samba_setresuid(uid,uid,-1) == -1) {
238 if (errno == EAGAIN) {
239 DEBUG(0, ("samba_setresuid failed with EAGAIN. uid(%d) "
240 "might be over its NPROC limit\n",
241 (int)uid));
244 #endif
246 #if USE_SETREUID
247 samba_setreuid(-1,uid);
248 #endif
250 #if USE_SETEUID
251 samba_seteuid(uid);
252 #endif
254 #if USE_SETUIDX
255 samba_setuidx(ID_EFFECTIVE, uid);
256 #endif
258 assert_uid(-1, uid);
261 /****************************************************************************
262 Set *only* the effective gid.
263 we want to end up with rgid==0 and egid==gid
264 ****************************************************************************/
265 void set_effective_gid(gid_t gid)
267 #if defined(USE_SETRESUID) || defined(HAVE_LINUX_THREAD_CREDENTIALS)
268 samba_setresgid(-1,gid,-1);
269 #endif
271 #if USE_SETREUID
272 samba_setregid(-1,gid);
273 #endif
275 #if USE_SETEUID
276 samba_setegid(gid);
277 #endif
279 #if USE_SETUIDX
280 samba_setgidx(ID_EFFECTIVE, gid);
281 #endif
283 assert_gid(-1, gid);
286 static uid_t saved_euid, saved_ruid;
287 static gid_t saved_egid, saved_rgid;
289 /****************************************************************************
290 save the real and effective uid for later restoration. Used by the quotas
291 code
292 ****************************************************************************/
293 void save_re_uid(void)
295 saved_ruid = getuid();
296 saved_euid = geteuid();
300 /****************************************************************************
301 and restore them!
302 ****************************************************************************/
304 void restore_re_uid_fromroot(void)
306 #if defined(USE_SETRESUID) || defined(HAVE_LINUX_THREAD_CREDENTIALS)
307 samba_setresuid(saved_ruid, saved_euid, -1);
308 #elif USE_SETREUID
309 samba_setreuid(saved_ruid, -1);
310 samba_setreuid(-1,saved_euid);
311 #elif USE_SETUIDX
312 samba_setuidx(ID_REAL, saved_ruid);
313 samba_setuidx(ID_EFFECTIVE, saved_euid);
314 #else
315 set_effective_uid(saved_euid);
316 if (getuid() != saved_ruid)
317 samba_setuid(saved_ruid);
318 set_effective_uid(saved_euid);
319 #endif
321 assert_uid(saved_ruid, saved_euid);
324 void restore_re_uid(void)
326 set_effective_uid(0);
327 restore_re_uid_fromroot();
330 /****************************************************************************
331 save the real and effective gid for later restoration. Used by the
332 getgroups code
333 ****************************************************************************/
334 void save_re_gid(void)
336 saved_rgid = getgid();
337 saved_egid = getegid();
340 /****************************************************************************
341 and restore them!
342 ****************************************************************************/
343 void restore_re_gid(void)
345 #if defined(USE_SETRESUID) || defined(HAVE_LINUX_THREAD_CREDENTIALS)
346 samba_setresgid(saved_rgid, saved_egid, -1);
347 #elif USE_SETREUID
348 samba_setregid(saved_rgid, -1);
349 samba_setregid(-1,saved_egid);
350 #elif USE_SETUIDX
351 samba_setgidx(ID_REAL, saved_rgid);
352 samba_setgidx(ID_EFFECTIVE, saved_egid);
353 #else
354 set_effective_gid(saved_egid);
355 if (getgid() != saved_rgid)
356 samba_setgid(saved_rgid);
357 set_effective_gid(saved_egid);
358 #endif
360 assert_gid(saved_rgid, saved_egid);
364 /****************************************************************************
365 set the real AND effective uid to the current effective uid in a way that
366 allows root to be regained.
367 This is only possible on some platforms.
368 ****************************************************************************/
369 int set_re_uid(void)
371 uid_t uid = geteuid();
373 #if defined(USE_SETRESUID) || defined(HAVE_LINUX_THREAD_CREDENTIALS)
374 samba_setresuid(uid, uid, -1);
375 #endif
377 #if USE_SETREUID
378 samba_setreuid(0, 0);
379 samba_setreuid(uid, -1);
380 samba_setreuid(-1, uid);
381 #endif
383 #if USE_SETEUID
384 /* can't be done */
385 return -1;
386 #endif
388 #if USE_SETUIDX
389 /* can't be done */
390 return -1;
391 #endif
393 assert_uid(uid, uid);
394 return 0;
398 /****************************************************************************
399 Become the specified uid and gid - permanently !
400 there should be no way back if possible
401 ****************************************************************************/
402 void become_user_permanently(uid_t uid, gid_t gid)
405 * First - gain root privilege. We do this to ensure
406 * we can lose it again.
409 gain_root_privilege();
410 gain_root_group_privilege();
412 #if defined(USE_SETRESUID) || defined(HAVE_LINUX_THREAD_CREDENTIALS)
413 samba_setresgid(gid,gid,gid);
414 samba_setgid(gid);
415 samba_setresuid(uid,uid,uid);
416 samba_setuid(uid);
417 #endif
419 #if USE_SETREUID
420 samba_setregid(gid,gid);
421 samba_setgid(gid);
422 samba_setreuid(uid,uid);
423 samba_setuid(uid);
424 #endif
426 #if USE_SETEUID
427 samba_setegid(gid);
428 samba_setgid(gid);
429 samba_setuid(uid);
430 samba_seteuid(uid);
431 samba_setuid(uid);
432 #endif
434 #if USE_SETUIDX
435 samba_setgidx(ID_REAL, gid);
436 samba_setgidx(ID_EFFECTIVE, gid);
437 samba_setgid(gid);
438 samba_setuidx(ID_REAL, uid);
439 samba_setuidx(ID_EFFECTIVE, uid);
440 samba_setuid(uid);
441 #endif
443 assert_uid(uid, uid);
444 assert_gid(gid, gid);
447 /**********************************************************
448 Function to set thread specific credentials. Leave
449 saved-set uid/gid alone.Must be thread-safe code.
450 **********************************************************/
452 int set_thread_credentials(uid_t uid,
453 gid_t gid,
454 size_t setlen,
455 const gid_t *gidset)
457 #if defined(HAVE_LINUX_THREAD_CREDENTIALS)
459 * With Linux thread-specific credentials
460 * we know we have setresuid/setresgid
461 * available.
463 #ifdef HAVE___THREAD
464 static struct {
465 bool active;
466 uid_t uid;
467 gid_t gid;
468 size_t setlen;
469 uintptr_t gidset;
470 } __thread cache;
472 if (cache.active &&
473 cache.uid == uid &&
474 cache.gid == gid &&
475 cache.setlen == setlen &&
476 (const gid_t *)cache.gidset == gidset)
478 return 0;
480 #endif /* HAVE___THREAD */
482 /* Become root. */
483 /* Set ru=0, eu=0 */
484 if (samba_setresuid(0, 0, -1) != 0) {
485 return -1;
487 /* Set our primary gid. */
488 /* Set rg=gid, eg=gid */
489 if (samba_setresgid(gid, gid, -1) != 0) {
490 return -1;
492 /* Set extra groups list. */
493 if (samba_setgroups(setlen, gidset) != 0) {
494 return -1;
496 /* Become the requested user. */
497 /* Set ru=uid, eu=uid */
498 if (samba_setresuid(uid, uid, -1) != 0) {
499 return -1;
501 if (geteuid() != uid || getuid() != uid ||
502 getegid() != gid || getgid() != gid) {
503 smb_panic("set_thread_credentials failed\n");
504 return -1;
507 #ifdef HAVE___THREAD
508 cache.active = true;
509 cache.uid = uid;
510 cache.gid = gid;
511 cache.setlen = setlen;
512 cache.gidset = (uintptr_t)gidset;
513 #endif /* HAVE___THREAD */
515 return 0;
516 #else
517 errno = ENOSYS;
518 return -1;
519 #endif
522 #ifdef AUTOCONF_TEST
524 /****************************************************************************
525 this function just checks that we don't get ENOSYS back
526 ****************************************************************************/
527 static int have_syscall(void)
529 errno = 0;
531 #if defined(USE_SETRESUID) || defined(HAVE_LINUX_THREAD_CREDENTIALS)
532 samba_setresuid(-1,-1,-1);
533 #endif
535 #if USE_SETREUID
536 samba_setreuid(-1,-1);
537 #endif
539 #if USE_SETEUID
540 samba_seteuid(-1);
541 #endif
543 #if USE_SETUIDX
544 samba_setuidx(ID_EFFECTIVE, -1);
545 #endif
547 if (errno == ENOSYS) return -1;
549 return 0;
552 main()
554 if (getuid() != 0) {
555 #if (defined(AIX) && defined(USE_SETREUID))
556 /* setreuid is badly broken on AIX 4.1, we avoid it completely */
557 fprintf(stderr,"avoiding possibly broken setreuid\n");
558 exit(1);
559 #endif
561 /* if not running as root then at least check to see if we get ENOSYS - this
562 handles Linux 2.0.x with glibc 2.1 */
563 fprintf(stderr,"not running as root: checking for ENOSYS\n");
564 exit(have_syscall());
567 gain_root_privilege();
568 gain_root_group_privilege();
569 set_effective_gid(1);
570 set_effective_uid(1);
571 save_re_uid();
572 restore_re_uid();
573 gain_root_privilege();
574 gain_root_group_privilege();
575 become_user_permanently(1, 1);
576 samba_setuid(0);
577 if (getuid() == 0) {
578 fprintf(stderr,"uid not set permanently\n");
579 exit(1);
582 printf("OK\n");
584 exit(0);
586 #endif
588 /****************************************************************************
589 Check if we are setuid root. Used in libsmb and smbpasswd paranoia checks.
590 ****************************************************************************/
591 bool is_setuid_root(void)
593 return (geteuid() == (uid_t)0) && (getuid() != (uid_t)0);