1 /* Inner loops of cache daemon.
2 Copyright (C) 1998-2007, 2008, 2009 Free Software Foundation, Inc.
3 This file is part of the GNU C Library.
4 Contributed by Ulrich Drepper <drepper@cygnus.com>, 1998.
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
8 by the Free Software Foundation; version 2 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, write to the Free Software Foundation,
18 Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */
34 #include <arpa/inet.h>
36 # include <sys/epoll.h>
39 # include <sys/inotify.h>
42 #include <sys/param.h>
45 # include <sys/sendfile.h>
47 #include <sys/socket.h>
54 #include <resolv/resolv.h>
56 # include <kernel-features.h>
60 /* Wrapper functions with error checking for standard functions. */
61 extern void *xmalloc (size_t n
);
62 extern void *xcalloc (size_t n
, size_t s
);
63 extern void *xrealloc (void *o
, size_t n
);
65 /* Support to run nscd as an unprivileged user */
66 const char *server_user
;
67 static uid_t server_uid
;
68 static gid_t server_gid
;
69 const char *stat_user
;
71 static gid_t
*server_groups
;
75 static int server_ngroups
;
77 static pthread_attr_t attr
;
79 static void begin_drop_privileges (void);
80 static void finish_drop_privileges (void);
82 /* Map request type to a string. */
83 const char *const serv2str
[LASTREQ
] =
85 [GETPWBYNAME
] = "GETPWBYNAME",
86 [GETPWBYUID
] = "GETPWBYUID",
87 [GETGRBYNAME
] = "GETGRBYNAME",
88 [GETGRBYGID
] = "GETGRBYGID",
89 [GETHOSTBYNAME
] = "GETHOSTBYNAME",
90 [GETHOSTBYNAMEv6
] = "GETHOSTBYNAMEv6",
91 [GETHOSTBYADDR
] = "GETHOSTBYADDR",
92 [GETHOSTBYADDRv6
] = "GETHOSTBYADDRv6",
93 [SHUTDOWN
] = "SHUTDOWN",
94 [GETSTAT
] = "GETSTAT",
95 [INVALIDATE
] = "INVALIDATE",
96 [GETFDPW
] = "GETFDPW",
97 [GETFDGR
] = "GETFDGR",
98 [GETFDHST
] = "GETFDHST",
100 [INITGROUPS
] = "INITGROUPS",
101 [GETSERVBYNAME
] = "GETSERVBYNAME",
102 [GETSERVBYPORT
] = "GETSERVBYPORT",
103 [GETFDSERV
] = "GETFDSERV"
106 /* The control data structures for the services. */
107 struct database_dyn dbs
[lastdb
] =
110 .lock
= PTHREAD_RWLOCK_WRITER_NONRECURSIVE_INITIALIZER_NP
,
111 .prune_lock
= PTHREAD_MUTEX_INITIALIZER
,
112 .prune_run_lock
= PTHREAD_MUTEX_INITIALIZER
,
118 .max_db_size
= DEFAULT_MAX_DB_SIZE
,
119 .suggested_module
= DEFAULT_SUGGESTED_MODULE
,
121 .filename
= "/etc/passwd",
122 .db_filename
= _PATH_NSCD_PASSWD_DB
,
123 .disabled_iov
= &pwd_iov_disabled
,
131 .lock
= PTHREAD_RWLOCK_WRITER_NONRECURSIVE_INITIALIZER_NP
,
132 .prune_lock
= PTHREAD_MUTEX_INITIALIZER
,
133 .prune_run_lock
= PTHREAD_MUTEX_INITIALIZER
,
139 .max_db_size
= DEFAULT_MAX_DB_SIZE
,
140 .suggested_module
= DEFAULT_SUGGESTED_MODULE
,
142 .filename
= "/etc/group",
143 .db_filename
= _PATH_NSCD_GROUP_DB
,
144 .disabled_iov
= &grp_iov_disabled
,
152 .lock
= PTHREAD_RWLOCK_WRITER_NONRECURSIVE_INITIALIZER_NP
,
153 .prune_lock
= PTHREAD_MUTEX_INITIALIZER
,
154 .prune_run_lock
= PTHREAD_MUTEX_INITIALIZER
,
158 .propagate
= 0, /* Not used. */
160 .max_db_size
= DEFAULT_MAX_DB_SIZE
,
161 .suggested_module
= DEFAULT_SUGGESTED_MODULE
,
163 .filename
= "/etc/hosts",
164 .db_filename
= _PATH_NSCD_HOSTS_DB
,
165 .disabled_iov
= &hst_iov_disabled
,
173 .lock
= PTHREAD_RWLOCK_WRITER_NONRECURSIVE_INITIALIZER_NP
,
174 .prune_lock
= PTHREAD_MUTEX_INITIALIZER
,
175 .prune_run_lock
= PTHREAD_MUTEX_INITIALIZER
,
179 .propagate
= 0, /* Not used. */
181 .max_db_size
= DEFAULT_MAX_DB_SIZE
,
182 .suggested_module
= DEFAULT_SUGGESTED_MODULE
,
184 .filename
= "/etc/services",
185 .db_filename
= _PATH_NSCD_SERVICES_DB
,
186 .disabled_iov
= &serv_iov_disabled
,
196 /* Mapping of request type to database. */
200 struct database_dyn
*db
;
201 } const reqinfo
[LASTREQ
] =
203 [GETPWBYNAME
] = { true, &dbs
[pwddb
] },
204 [GETPWBYUID
] = { true, &dbs
[pwddb
] },
205 [GETGRBYNAME
] = { true, &dbs
[grpdb
] },
206 [GETGRBYGID
] = { true, &dbs
[grpdb
] },
207 [GETHOSTBYNAME
] = { true, &dbs
[hstdb
] },
208 [GETHOSTBYNAMEv6
] = { true, &dbs
[hstdb
] },
209 [GETHOSTBYADDR
] = { true, &dbs
[hstdb
] },
210 [GETHOSTBYADDRv6
] = { true, &dbs
[hstdb
] },
211 [SHUTDOWN
] = { false, NULL
},
212 [GETSTAT
] = { false, NULL
},
213 [SHUTDOWN
] = { false, NULL
},
214 [GETFDPW
] = { false, &dbs
[pwddb
] },
215 [GETFDGR
] = { false, &dbs
[grpdb
] },
216 [GETFDHST
] = { false, &dbs
[hstdb
] },
217 [GETAI
] = { true, &dbs
[hstdb
] },
218 [INITGROUPS
] = { true, &dbs
[grpdb
] },
219 [GETSERVBYNAME
] = { true, &dbs
[servdb
] },
220 [GETSERVBYPORT
] = { true, &dbs
[servdb
] },
221 [GETFDSERV
] = { false, &dbs
[servdb
] }
225 /* Initial number of threads to use. */
227 /* Maximum number of threads to use. */
228 int max_nthreads
= 32;
230 /* Socket for incoming connections. */
234 /* Inotify descriptor. */
235 static int inotify_fd
= -1;
237 /* Watch descriptor for resolver configuration file. */
238 static int resolv_conf_descr
= -1;
241 #ifndef __ASSUME_SOCK_CLOEXEC
242 /* Negative if SOCK_CLOEXEC is not supported, positive if it is, zero
243 before be know the result. */
244 static int have_sock_cloexec
;
246 #ifndef __ASSUME_ACCEPT4
247 static int have_accept4
;
250 /* Number of times clients had to wait. */
251 unsigned long int client_queued
;
255 writeall (int fd
, const void *buf
, size_t len
)
261 ret
= TEMP_FAILURE_RETRY (send (fd
, buf
, n
, MSG_NOSIGNAL
));
264 buf
= (const char *) buf
+ ret
;
268 return ret
< 0 ? ret
: len
- n
;
274 sendfileall (int tofd
, int fromfd
, off_t off
, size_t len
)
281 ret
= TEMP_FAILURE_RETRY (sendfile (tofd
, fromfd
, &off
, n
));
287 return ret
< 0 ? ret
: len
- n
;
295 /* The following three are not really used, they are symbolic constants. */
301 use_he_begin
= use_he
| use_begin
,
302 use_he_end
= use_he
| use_end
,
305 use_key_begin
= use_key
| use_begin
,
306 use_key_end
= use_key
| use_end
,
307 use_key_first
= use_key_begin
| use_first
,
310 use_data_begin
= use_data
| use_begin
,
311 use_data_end
= use_data
| use_end
,
312 use_data_first
= use_data_begin
| use_first
317 check_use (const char *data
, nscd_ssize_t first_free
, uint8_t *usemap
,
318 enum usekey use
, ref_t start
, size_t len
)
322 if (start
> first_free
|| start
+ len
> first_free
323 || (start
& BLOCK_ALIGN_M1
))
326 if (usemap
[start
] == use_not
)
328 /* Add the start marker. */
329 usemap
[start
] = use
| use_begin
;
333 if (usemap
[++start
] != use_not
)
338 /* Add the end marker. */
339 usemap
[start
] = use
| use_end
;
341 else if ((usemap
[start
] & ~use_first
) == ((use
| use_begin
) & ~use_first
))
343 /* Hash entries can't be shared. */
347 usemap
[start
] |= (use
& use_first
);
351 if (usemap
[++start
] != use
)
354 if (usemap
[++start
] != (use
| use_end
))
358 /* Points to a wrong object or somewhere in the middle. */
365 /* Verify data in persistent database. */
367 verify_persistent_db (void *mem
, struct database_pers_head
*readhead
, int dbnr
)
369 assert (dbnr
== pwddb
|| dbnr
== grpdb
|| dbnr
== hstdb
|| dbnr
== servdb
);
371 time_t now
= time (NULL
);
373 struct database_pers_head
*head
= mem
;
374 struct database_pers_head head_copy
= *head
;
376 /* Check that the header that was read matches the head in the database. */
377 if (memcmp (head
, readhead
, sizeof (*head
)) != 0)
380 /* First some easy tests: make sure the database header is sane. */
381 if (head
->version
!= DB_VERSION
382 || head
->header_size
!= sizeof (*head
)
383 /* We allow a timestamp to be one hour ahead of the current time.
384 This should cover daylight saving time changes. */
385 || head
->timestamp
> now
+ 60 * 60 + 60
386 || (head
->gc_cycle
& 1)
388 || (size_t) head
->module
> INT32_MAX
/ sizeof (ref_t
)
389 || (size_t) head
->data_size
> INT32_MAX
- head
->module
* sizeof (ref_t
)
390 || head
->first_free
< 0
391 || head
->first_free
> head
->data_size
392 || (head
->first_free
& BLOCK_ALIGN_M1
) != 0
393 || head
->maxnentries
< 0
394 || head
->maxnsearched
< 0)
397 uint8_t *usemap
= calloc (head
->first_free
, 1);
401 const char *data
= (char *) &head
->array
[roundup (head
->module
,
402 ALIGN
/ sizeof (ref_t
))];
404 nscd_ssize_t he_cnt
= 0;
405 for (nscd_ssize_t cnt
= 0; cnt
< head
->module
; ++cnt
)
407 ref_t trail
= head
->array
[cnt
];
411 while (work
!= ENDREF
)
413 if (! check_use (data
, head
->first_free
, usemap
, use_he
, work
,
414 sizeof (struct hashentry
)))
417 /* Now we know we can dereference the record. */
418 struct hashentry
*here
= (struct hashentry
*) (data
+ work
);
422 /* Make sure the record is for this type of service. */
423 if (here
->type
>= LASTREQ
424 || reqinfo
[here
->type
].db
!= &dbs
[dbnr
])
427 /* Validate boolean field value. */
428 if (here
->first
!= false && here
->first
!= true)
436 || here
->packet
> head
->first_free
437 || here
->packet
+ sizeof (struct datahead
) > head
->first_free
)
440 struct datahead
*dh
= (struct datahead
*) (data
+ here
->packet
);
442 if (! check_use (data
, head
->first_free
, usemap
,
443 use_data
| (here
->first
? use_first
: 0),
444 here
->packet
, dh
->allocsize
))
447 if (dh
->allocsize
< sizeof (struct datahead
)
448 || dh
->recsize
> dh
->allocsize
449 || (dh
->notfound
!= false && dh
->notfound
!= true)
450 || (dh
->usable
!= false && dh
->usable
!= true))
453 if (here
->key
< here
->packet
+ sizeof (struct datahead
)
454 || here
->key
> here
->packet
+ dh
->allocsize
455 || here
->key
+ here
->len
> here
->packet
+ dh
->allocsize
)
458 /* If keys can appear outside of data, this should be done
459 instead. But gc doesn't mark the data in that case. */
460 if (! check_use (data
, head
->first_free
, usemap
,
461 use_key
| (here
->first
? use_first
: 0),
462 here
->key
, here
->len
))
470 /* A circular list, this must not happen. */
473 trail
= ((struct hashentry
*) (data
+ trail
))->next
;
478 if (he_cnt
!= head
->nentries
)
481 /* See if all data and keys had at least one reference from
482 he->first == true hashentry. */
483 for (ref_t idx
= 0; idx
< head
->first_free
; ++idx
)
486 if (usemap
[idx
] == use_key_begin
)
489 if (usemap
[idx
] == use_data_begin
)
493 /* Finally, make sure the database hasn't changed since the first test. */
494 if (memcmp (mem
, &head_copy
, sizeof (*head
)) != 0)
507 # define EXTRA_O_FLAGS O_CLOEXEC
509 # define EXTRA_O_FLAGS 0
513 /* Initialize database information structures. */
517 /* Look up unprivileged uid/gid/groups before we start listening on the
519 if (server_user
!= NULL
)
520 begin_drop_privileges ();
523 /* No configuration for this value, assume a default. */
527 /* Use inotify to recognize changed files. */
528 inotify_fd
= inotify_init1 (IN_NONBLOCK
);
529 # ifndef __ASSUME_IN_NONBLOCK
530 if (inotify_fd
== -1 && errno
== ENOSYS
)
532 inotify_fd
= inotify_init ();
533 if (inotify_fd
!= -1)
534 fcntl (inotify_fd
, F_SETFL
, O_RDONLY
| O_NONBLOCK
);
539 for (size_t cnt
= 0; cnt
< lastdb
; ++cnt
)
540 if (dbs
[cnt
].enabled
)
542 pthread_rwlock_init (&dbs
[cnt
].lock
, NULL
);
543 pthread_mutex_init (&dbs
[cnt
].memlock
, NULL
);
545 if (dbs
[cnt
].persistent
)
547 /* Try to open the appropriate file on disk. */
548 int fd
= open (dbs
[cnt
].db_filename
, O_RDWR
| EXTRA_O_FLAGS
);
555 struct database_pers_head head
;
556 ssize_t n
= TEMP_FAILURE_RETRY (read (fd
, &head
,
558 if (n
!= sizeof (head
) || fstat64 (fd
, &st
) != 0)
561 /* The code is single-threaded at this point so
562 using strerror is just fine. */
563 msg
= strerror (errno
);
565 dbg_log (_("invalid persistent database file \"%s\": %s"),
566 dbs
[cnt
].db_filename
, msg
);
567 unlink (dbs
[cnt
].db_filename
);
569 else if (head
.module
== 0 && head
.data_size
== 0)
571 /* The file has been created, but the head has not
572 been initialized yet. */
573 msg
= _("uninitialized header");
576 else if (head
.header_size
!= (int) sizeof (head
))
578 msg
= _("header size does not match");
581 else if ((total
= (sizeof (head
)
582 + roundup (head
.module
* sizeof (ref_t
),
586 || total
< sizeof (head
))
588 msg
= _("file size does not match");
591 /* Note we map with the maximum size allowed for the
592 database. This is likely much larger than the
593 actual file size. This is OK on most OSes since
594 extensions of the underlying file will
595 automatically translate more pages available for
597 else if ((mem
= mmap (NULL
, dbs
[cnt
].max_db_size
,
598 PROT_READ
| PROT_WRITE
,
602 else if (!verify_persistent_db (mem
, &head
, cnt
))
605 msg
= _("verification failed");
610 /* Success. We have the database. */
612 dbs
[cnt
].memsize
= total
;
613 dbs
[cnt
].data
= (char *)
614 &dbs
[cnt
].head
->array
[roundup (dbs
[cnt
].head
->module
,
615 ALIGN
/ sizeof (ref_t
))];
616 dbs
[cnt
].mmap_used
= true;
618 if (dbs
[cnt
].suggested_module
> head
.module
)
619 dbg_log (_("suggested size of table for database %s larger than the persistent database's table"),
624 /* We also need a read-only descriptor. */
627 dbs
[cnt
].ro_fd
= open (dbs
[cnt
].db_filename
,
628 O_RDONLY
| EXTRA_O_FLAGS
);
629 if (dbs
[cnt
].ro_fd
== -1)
631 cannot create read-only descriptor for \"%s\"; no mmap"),
632 dbs
[cnt
].db_filename
);
635 // XXX Shall we test whether the descriptors actually
636 // XXX point to the same file?
639 /* Close the file descriptors in case something went
640 wrong in which case the variable have not been
645 else if (errno
== EACCES
)
646 error (EXIT_FAILURE
, 0, _("cannot access '%s'"),
647 dbs
[cnt
].db_filename
);
650 if (dbs
[cnt
].head
== NULL
)
652 /* No database loaded. Allocate the data structure,
654 struct database_pers_head head
;
655 size_t total
= (sizeof (head
)
656 + roundup (dbs
[cnt
].suggested_module
657 * sizeof (ref_t
), ALIGN
)
658 + (dbs
[cnt
].suggested_module
659 * DEFAULT_DATASIZE_PER_BUCKET
));
661 /* Try to create the database. If we do not need a
662 persistent database create a temporary file. */
665 if (dbs
[cnt
].persistent
)
667 fd
= open (dbs
[cnt
].db_filename
,
668 O_RDWR
| O_CREAT
| O_EXCL
| O_TRUNC
| EXTRA_O_FLAGS
,
670 if (fd
!= -1 && dbs
[cnt
].shared
)
671 ro_fd
= open (dbs
[cnt
].db_filename
,
672 O_RDONLY
| EXTRA_O_FLAGS
);
676 char fname
[] = _PATH_NSCD_XYZ_DB_TMP
;
677 fd
= mkostemp (fname
, EXTRA_O_FLAGS
);
679 /* We do not need the file name anymore after we
680 opened another file descriptor in read-only mode. */
684 ro_fd
= open (fname
, O_RDONLY
| EXTRA_O_FLAGS
);
694 dbg_log (_("database for %s corrupted or simultaneously used; remove %s manually if necessary and restart"),
695 dbnames
[cnt
], dbs
[cnt
].db_filename
);
696 // XXX Correct way to terminate?
700 if (dbs
[cnt
].persistent
)
701 dbg_log (_("cannot create %s; no persistent database used"),
702 dbs
[cnt
].db_filename
);
704 dbg_log (_("cannot create %s; no sharing possible"),
705 dbs
[cnt
].db_filename
);
707 dbs
[cnt
].persistent
= 0;
708 // XXX remember: no mmap
712 /* Tell the user if we could not create the read-only
714 if (ro_fd
== -1 && dbs
[cnt
].shared
)
716 cannot create read-only descriptor for \"%s\"; no mmap"),
717 dbs
[cnt
].db_filename
);
719 /* Before we create the header, initialiye the hash
720 table. So that if we get interrupted if writing
721 the header we can recognize a partially initialized
723 size_t ps
= sysconf (_SC_PAGESIZE
);
725 assert (~ENDREF
== 0);
726 memset (tmpbuf
, '\xff', ps
);
728 size_t remaining
= dbs
[cnt
].suggested_module
* sizeof (ref_t
);
729 off_t offset
= sizeof (head
);
732 if (offset
% ps
!= 0)
734 towrite
= MIN (remaining
, ps
- (offset
% ps
));
735 if (pwrite (fd
, tmpbuf
, towrite
, offset
) != towrite
)
738 remaining
-= towrite
;
741 while (remaining
> ps
)
743 if (pwrite (fd
, tmpbuf
, ps
, offset
) == -1)
750 && pwrite (fd
, tmpbuf
, remaining
, offset
) != remaining
)
753 /* Create the header of the file. */
754 struct database_pers_head head
=
756 .version
= DB_VERSION
,
757 .header_size
= sizeof (head
),
758 .module
= dbs
[cnt
].suggested_module
,
759 .data_size
= (dbs
[cnt
].suggested_module
760 * DEFAULT_DATASIZE_PER_BUCKET
),
765 if ((TEMP_FAILURE_RETRY (write (fd
, &head
, sizeof (head
)))
767 || (TEMP_FAILURE_RETRY_VAL (posix_fallocate (fd
, 0, total
))
769 || (mem
= mmap (NULL
, dbs
[cnt
].max_db_size
,
770 PROT_READ
| PROT_WRITE
,
771 MAP_SHARED
, fd
, 0)) == MAP_FAILED
)
774 unlink (dbs
[cnt
].db_filename
);
775 dbg_log (_("cannot write to database file %s: %s"),
776 dbs
[cnt
].db_filename
, strerror (errno
));
777 dbs
[cnt
].persistent
= 0;
783 dbs
[cnt
].data
= (char *)
784 &dbs
[cnt
].head
->array
[roundup (dbs
[cnt
].head
->module
,
785 ALIGN
/ sizeof (ref_t
))];
786 dbs
[cnt
].memsize
= total
;
787 dbs
[cnt
].mmap_used
= true;
789 /* Remember the descriptors. */
791 dbs
[cnt
].ro_fd
= ro_fd
;
803 #if !defined O_CLOEXEC || !defined __ASSUME_O_CLOEXEC
804 /* We do not check here whether the O_CLOEXEC provided to the
805 open call was successful or not. The two fcntl calls are
806 only performed once each per process start-up and therefore
807 is not noticeable at all. */
809 && ((dbs
[cnt
].wr_fd
!= -1
810 && fcntl (dbs
[cnt
].wr_fd
, F_SETFD
, FD_CLOEXEC
) == -1)
811 || (dbs
[cnt
].ro_fd
!= -1
812 && fcntl (dbs
[cnt
].ro_fd
, F_SETFD
, FD_CLOEXEC
) == -1)))
815 cannot set socket to close on exec: %s; disabling paranoia mode"),
821 if (dbs
[cnt
].head
== NULL
)
823 /* We do not use the persistent database. Just
824 create an in-memory data structure. */
825 assert (! dbs
[cnt
].persistent
);
827 dbs
[cnt
].head
= xmalloc (sizeof (struct database_pers_head
)
828 + (dbs
[cnt
].suggested_module
830 memset (dbs
[cnt
].head
, '\0', sizeof (struct database_pers_head
));
831 assert (~ENDREF
== 0);
832 memset (dbs
[cnt
].head
->array
, '\xff',
833 dbs
[cnt
].suggested_module
* sizeof (ref_t
));
834 dbs
[cnt
].head
->module
= dbs
[cnt
].suggested_module
;
835 dbs
[cnt
].head
->data_size
= (DEFAULT_DATASIZE_PER_BUCKET
836 * dbs
[cnt
].head
->module
);
837 dbs
[cnt
].data
= xmalloc (dbs
[cnt
].head
->data_size
);
838 dbs
[cnt
].head
->first_free
= 0;
841 assert (dbs
[cnt
].ro_fd
== -1);
844 dbs
[cnt
].inotify_descr
= -1;
845 if (dbs
[cnt
].check_file
)
849 || (dbs
[cnt
].inotify_descr
850 = inotify_add_watch (inotify_fd
, dbs
[cnt
].filename
,
851 IN_DELETE_SELF
| IN_MODIFY
)) < 0)
852 /* We cannot notice changes in the main thread. */
855 /* We need the modification date of the file. */
858 if (stat64 (dbs
[cnt
].filename
, &st
) < 0)
860 /* We cannot stat() the file, disable file checking. */
861 dbg_log (_("cannot stat() file `%s': %s"),
862 dbs
[cnt
].filename
, strerror (errno
));
863 dbs
[cnt
].check_file
= 0;
866 dbs
[cnt
].file_mtime
= st
.st_mtime
;
871 if (cnt
== hstdb
&& inotify_fd
>= -1)
872 /* We also monitor the resolver configuration file. */
873 resolv_conf_descr
= inotify_add_watch (inotify_fd
,
875 IN_DELETE_SELF
| IN_MODIFY
);
879 /* Create the socket. */
880 #ifndef __ASSUME_SOCK_CLOEXEC
882 if (have_sock_cloexec
>= 0)
885 sock
= socket (AF_UNIX
, SOCK_STREAM
| SOCK_CLOEXEC
| SOCK_NONBLOCK
, 0);
886 #ifndef __ASSUME_SOCK_CLOEXEC
887 if (have_sock_cloexec
== 0)
888 have_sock_cloexec
= sock
!= -1 || errno
!= EINVAL
? 1 : -1;
891 #ifndef __ASSUME_SOCK_CLOEXEC
892 if (have_sock_cloexec
< 0)
893 sock
= socket (AF_UNIX
, SOCK_STREAM
, 0);
897 dbg_log (_("cannot open socket: %s"), strerror (errno
));
898 exit (errno
== EACCES
? 4 : 1);
900 /* Bind a name to the socket. */
901 struct sockaddr_un sock_addr
;
902 sock_addr
.sun_family
= AF_UNIX
;
903 strcpy (sock_addr
.sun_path
, _PATH_NSCDSOCKET
);
904 if (bind (sock
, (struct sockaddr
*) &sock_addr
, sizeof (sock_addr
)) < 0)
906 dbg_log ("%s: %s", _PATH_NSCDSOCKET
, strerror (errno
));
907 exit (errno
== EACCES
? 4 : 1);
910 #ifndef __ASSUME_SOCK_CLOEXEC
911 if (have_sock_cloexec
< 0)
913 /* We don't want to get stuck on accept. */
914 int fl
= fcntl (sock
, F_GETFL
);
915 if (fl
== -1 || fcntl (sock
, F_SETFL
, fl
| O_NONBLOCK
) == -1)
917 dbg_log (_("cannot change socket to nonblocking mode: %s"),
922 /* The descriptor needs to be closed on exec. */
923 if (paranoia
&& fcntl (sock
, F_SETFD
, FD_CLOEXEC
) == -1)
925 dbg_log (_("cannot set socket to close on exec: %s"),
932 /* Set permissions for the socket. */
933 chmod (_PATH_NSCDSOCKET
, DEFFILEMODE
);
935 /* Set the socket up to accept connections. */
936 if (listen (sock
, SOMAXCONN
) < 0)
938 dbg_log (_("cannot enable socket to accept connections: %s"),
943 /* Change to unprivileged uid/gid/groups if specifed in config file */
944 if (server_user
!= NULL
)
945 finish_drop_privileges ();
949 /* Close the connections. */
958 invalidate_cache (char *key
, int fd
)
963 for (number
= pwddb
; number
< lastdb
; ++number
)
964 if (strcmp (key
, dbnames
[number
]) == 0)
966 if (dbs
[number
].reset_res
)
972 if (number
== lastdb
)
975 writeall (fd
, &resp
, sizeof (resp
));
979 if (dbs
[number
].enabled
)
981 pthread_mutex_lock (&dbs
[number
].prune_run_lock
);
982 prune_cache (&dbs
[number
], LONG_MAX
, fd
);
983 pthread_mutex_unlock (&dbs
[number
].prune_run_lock
);
988 writeall (fd
, &resp
, sizeof (resp
));
995 send_ro_fd (struct database_dyn
*db
, char *key
, int fd
)
997 /* If we do not have an read-only file descriptor do nothing. */
1001 /* We need to send some data along with the descriptor. */
1002 uint64_t mapsize
= (db
->head
->data_size
1003 + roundup (db
->head
->module
* sizeof (ref_t
), ALIGN
)
1004 + sizeof (struct database_pers_head
));
1005 struct iovec iov
[2];
1006 iov
[0].iov_base
= key
;
1007 iov
[0].iov_len
= strlen (key
) + 1;
1008 iov
[1].iov_base
= &mapsize
;
1009 iov
[1].iov_len
= sizeof (mapsize
);
1011 /* Prepare the control message to transfer the descriptor. */
1015 char bytes
[CMSG_SPACE (sizeof (int))];
1017 struct msghdr msg
= { .msg_iov
= iov
, .msg_iovlen
= 2,
1018 .msg_control
= buf
.bytes
,
1019 .msg_controllen
= sizeof (buf
) };
1020 struct cmsghdr
*cmsg
= CMSG_FIRSTHDR (&msg
);
1022 cmsg
->cmsg_level
= SOL_SOCKET
;
1023 cmsg
->cmsg_type
= SCM_RIGHTS
;
1024 cmsg
->cmsg_len
= CMSG_LEN (sizeof (int));
1026 *(int *) CMSG_DATA (cmsg
) = db
->ro_fd
;
1028 msg
.msg_controllen
= cmsg
->cmsg_len
;
1030 /* Send the control message. We repeat when we are interrupted but
1031 everything else is ignored. */
1032 #ifndef MSG_NOSIGNAL
1033 # define MSG_NOSIGNAL 0
1035 (void) TEMP_FAILURE_RETRY (sendmsg (fd
, &msg
, MSG_NOSIGNAL
));
1037 if (__builtin_expect (debug_level
> 0, 0))
1038 dbg_log (_("provide access to FD %d, for %s"), db
->ro_fd
, key
);
1040 #endif /* SCM_RIGHTS */
1043 /* Handle new request. */
1045 handle_request (int fd
, request_header
*req
, void *key
, uid_t uid
, pid_t pid
)
1047 if (__builtin_expect (req
->version
, NSCD_VERSION
) != NSCD_VERSION
)
1049 if (debug_level
> 0)
1051 cannot handle old request version %d; current version is %d"),
1052 req
->version
, NSCD_VERSION
);
1056 /* Perform the SELinux check before we go on to the standard checks. */
1057 if (selinux_enabled
&& nscd_request_avc_has_perm (fd
, req
->type
) != 0)
1059 if (debug_level
> 0)
1068 snprintf (buf
, sizeof (buf
), "/proc/%ld/exe", (long int) pid
);
1069 ssize_t n
= readlink (buf
, buf
, sizeof (buf
) - 1);
1073 request from %ld not handled due to missing permission"), (long int) pid
);
1078 request from '%s' [%ld] not handled due to missing permission"),
1079 buf
, (long int) pid
);
1082 dbg_log (_("request not handled due to missing permission"));
1088 struct database_dyn
*db
= reqinfo
[req
->type
].db
;
1090 /* See whether we can service the request from the cache. */
1091 if (__builtin_expect (reqinfo
[req
->type
].data_request
, true))
1093 if (__builtin_expect (debug_level
, 0) > 0)
1095 if (req
->type
== GETHOSTBYADDR
|| req
->type
== GETHOSTBYADDRv6
)
1097 char buf
[INET6_ADDRSTRLEN
];
1099 dbg_log ("\t%s (%s)", serv2str
[req
->type
],
1100 inet_ntop (req
->type
== GETHOSTBYADDR
1101 ? AF_INET
: AF_INET6
,
1102 key
, buf
, sizeof (buf
)));
1105 dbg_log ("\t%s (%s)", serv2str
[req
->type
], (char *) key
);
1108 /* Is this service enabled? */
1109 if (__builtin_expect (!db
->enabled
, 0))
1111 /* No, sent the prepared record. */
1112 if (TEMP_FAILURE_RETRY (send (fd
, db
->disabled_iov
->iov_base
,
1113 db
->disabled_iov
->iov_len
,
1115 != (ssize_t
) db
->disabled_iov
->iov_len
1116 && __builtin_expect (debug_level
, 0) > 0)
1118 /* We have problems sending the result. */
1120 dbg_log (_("cannot write result: %s"),
1121 strerror_r (errno
, buf
, sizeof (buf
)));
1127 /* Be sure we can read the data. */
1128 if (__builtin_expect (pthread_rwlock_tryrdlock (&db
->lock
) != 0, 0))
1130 ++db
->head
->rdlockdelayed
;
1131 pthread_rwlock_rdlock (&db
->lock
);
1134 /* See whether we can handle it from the cache. */
1135 struct datahead
*cached
;
1136 cached
= (struct datahead
*) cache_search (req
->type
, key
, req
->key_len
,
1140 /* Hurray it's in the cache. */
1143 #ifdef HAVE_SENDFILE
1144 if (__builtin_expect (db
->mmap_used
, 1))
1146 assert (db
->wr_fd
!= -1);
1147 assert ((char *) cached
->data
> (char *) db
->data
);
1148 assert ((char *) cached
->data
- (char *) db
->head
1150 <= (sizeof (struct database_pers_head
)
1151 + db
->head
->module
* sizeof (ref_t
)
1152 + db
->head
->data_size
));
1153 nwritten
= sendfileall (fd
, db
->wr_fd
,
1154 (char *) cached
->data
1155 - (char *) db
->head
, cached
->recsize
);
1156 # ifndef __ASSUME_SENDFILE
1157 if (nwritten
== -1 && errno
== ENOSYS
)
1162 # ifndef __ASSUME_SENDFILE
1166 nwritten
= writeall (fd
, cached
->data
, cached
->recsize
);
1168 if (nwritten
!= cached
->recsize
1169 && __builtin_expect (debug_level
, 0) > 0)
1171 /* We have problems sending the result. */
1173 dbg_log (_("cannot write result: %s"),
1174 strerror_r (errno
, buf
, sizeof (buf
)));
1177 pthread_rwlock_unlock (&db
->lock
);
1182 pthread_rwlock_unlock (&db
->lock
);
1184 else if (__builtin_expect (debug_level
, 0) > 0)
1186 if (req
->type
== INVALIDATE
)
1187 dbg_log ("\t%s (%s)", serv2str
[req
->type
], (char *) key
);
1189 dbg_log ("\t%s", serv2str
[req
->type
]);
1192 /* Handle the request. */
1196 addpwbyname (db
, fd
, req
, key
, uid
);
1200 addpwbyuid (db
, fd
, req
, key
, uid
);
1204 addgrbyname (db
, fd
, req
, key
, uid
);
1208 addgrbygid (db
, fd
, req
, key
, uid
);
1212 addhstbyname (db
, fd
, req
, key
, uid
);
1215 case GETHOSTBYNAMEv6
:
1216 addhstbynamev6 (db
, fd
, req
, key
, uid
);
1220 addhstbyaddr (db
, fd
, req
, key
, uid
);
1223 case GETHOSTBYADDRv6
:
1224 addhstbyaddrv6 (db
, fd
, req
, key
, uid
);
1228 addhstai (db
, fd
, req
, key
, uid
);
1232 addinitgroups (db
, fd
, req
, key
, uid
);
1236 addservbyname (db
, fd
, req
, key
, uid
);
1240 addservbyport (db
, fd
, req
, key
, uid
);
1247 /* Get the callers credentials. */
1249 struct ucred caller
;
1250 socklen_t optlen
= sizeof (caller
);
1252 if (getsockopt (fd
, SOL_SOCKET
, SO_PEERCRED
, &caller
, &optlen
) < 0)
1256 dbg_log (_("error getting caller's id: %s"),
1257 strerror_r (errno
, buf
, sizeof (buf
)));
1263 /* Some systems have no SO_PEERCRED implementation. They don't
1264 care about security so we don't as well. */
1269 /* Accept shutdown, getstat and invalidate only from root. For
1270 the stat call also allow the user specified in the config file. */
1271 if (req
->type
== GETSTAT
)
1273 if (uid
== 0 || uid
== stat_uid
)
1274 send_stats (fd
, dbs
);
1278 if (req
->type
== INVALIDATE
)
1279 invalidate_cache (key
, fd
);
1281 termination_handler (0);
1290 send_ro_fd (reqinfo
[req
->type
].db
, key
, fd
);
1295 /* Ignore the command, it's nothing we know. */
1301 /* Restart the process. */
1305 /* First determine the parameters. We do not use the parameters
1306 passed to main() since in case nscd is started by running the
1307 dynamic linker this will not work. Yes, this is not the usual
1308 case but nscd is part of glibc and we occasionally do this. */
1309 size_t buflen
= 1024;
1310 char *buf
= alloca (buflen
);
1312 int fd
= open ("/proc/self/cmdline", O_RDONLY
);
1316 cannot open /proc/self/cmdline: %s; disabling paranoia mode"),
1325 ssize_t n
= TEMP_FAILURE_RETRY (read (fd
, buf
+ readlen
,
1330 cannot read /proc/self/cmdline: %s; disabling paranoia mode"),
1340 if (readlen
< buflen
)
1343 /* We might have to extend the buffer. */
1344 size_t old_buflen
= buflen
;
1345 char *newp
= extend_alloca (buf
, buflen
, 2 * buflen
);
1346 buf
= memmove (newp
, buf
, old_buflen
);
1351 /* Parse the command line. Worst case scenario: every two
1352 characters form one parameter (one character plus NUL). */
1353 char **argv
= alloca ((readlen
/ 2 + 1) * sizeof (argv
[0]));
1357 while (cp
< buf
+ readlen
)
1360 cp
= (char *) rawmemchr (cp
, '\0') + 1;
1364 /* Second, change back to the old user if we changed it. */
1365 if (server_user
!= NULL
)
1367 if (setresuid (old_uid
, old_uid
, old_uid
) != 0)
1370 cannot change to old UID: %s; disabling paranoia mode"),
1377 if (setresgid (old_gid
, old_gid
, old_gid
) != 0)
1380 cannot change to old GID: %s; disabling paranoia mode"),
1383 setuid (server_uid
);
1389 /* Next change back to the old working directory. */
1390 if (chdir (oldcwd
) == -1)
1393 cannot change to old working directory: %s; disabling paranoia mode"),
1396 if (server_user
!= NULL
)
1398 setuid (server_uid
);
1399 setgid (server_gid
);
1405 /* Synchronize memory. */
1406 int32_t certainly
[lastdb
];
1407 for (int cnt
= 0; cnt
< lastdb
; ++cnt
)
1408 if (dbs
[cnt
].enabled
)
1410 /* Make sure nobody keeps using the database. */
1411 dbs
[cnt
].head
->timestamp
= 0;
1412 certainly
[cnt
] = dbs
[cnt
].head
->nscd_certainly_running
;
1413 dbs
[cnt
].head
->nscd_certainly_running
= 0;
1415 if (dbs
[cnt
].persistent
)
1417 msync (dbs
[cnt
].head
, dbs
[cnt
].memsize
, MS_ASYNC
);
1420 /* The preparations are done. */
1422 char pathbuf
[PATH_MAX
];
1426 /* Try to exec the real nscd program so the process name (as reported
1427 in /proc/PID/status) will be 'nscd', but fall back to /proc/self/exe
1428 if readlink fails */
1429 ssize_t n
= readlink ("/proc/self/exe", pathbuf
, sizeof (pathbuf
) - 1);
1431 execv ("/proc/self/exe", argv
);
1435 execv (pathbuf
, argv
);
1438 /* If we come here, we will never be able to re-exec. */
1439 dbg_log (_("re-exec failed: %s; disabling paranoia mode"),
1442 if (server_user
!= NULL
)
1444 setuid (server_uid
);
1445 setgid (server_gid
);
1447 if (chdir ("/") != 0)
1448 dbg_log (_("cannot change current working directory to \"/\": %s"),
1452 /* Reenable the databases. */
1453 time_t now
= time (NULL
);
1454 for (int cnt
= 0; cnt
< lastdb
; ++cnt
)
1455 if (dbs
[cnt
].enabled
)
1457 dbs
[cnt
].head
->timestamp
= now
;
1458 dbs
[cnt
].head
->nscd_certainly_running
= certainly
[cnt
];
1463 /* List of file descriptors. */
1467 struct fdlist
*next
;
1469 /* Memory allocated for the list. */
1470 static struct fdlist
*fdlist
;
1471 /* List of currently ready-to-read file descriptors. */
1472 static struct fdlist
*readylist
;
1474 /* Conditional variable and mutex to signal availability of entries in
1475 READYLIST. The condvar is initialized dynamically since we might
1476 use a different clock depending on availability. */
1477 static pthread_cond_t readylist_cond
= PTHREAD_COND_INITIALIZER
;
1478 static pthread_mutex_t readylist_lock
= PTHREAD_MUTEX_INITIALIZER
;
1480 /* The clock to use with the condvar. */
1481 static clockid_t timeout_clock
= CLOCK_REALTIME
;
1483 /* Number of threads ready to handle the READYLIST. */
1484 static unsigned long int nready
;
1487 /* Function for the clean-up threads. */
1489 __attribute__ ((__noreturn__
))
1490 nscd_run_prune (void *p
)
1492 const long int my_number
= (long int) p
;
1493 assert (dbs
[my_number
].enabled
);
1495 int dont_need_update
= setup_thread (&dbs
[my_number
]);
1497 time_t now
= time (NULL
);
1499 /* We are running. */
1500 dbs
[my_number
].head
->timestamp
= now
;
1502 struct timespec prune_ts
;
1503 if (__builtin_expect (clock_gettime (timeout_clock
, &prune_ts
) == -1, 0))
1504 /* Should never happen. */
1507 /* Compute the initial timeout time. Prevent all the timers to go
1508 off at the same time by adding a db-based value. */
1509 prune_ts
.tv_sec
+= CACHE_PRUNE_INTERVAL
+ my_number
;
1510 dbs
[my_number
].wakeup_time
= now
+ CACHE_PRUNE_INTERVAL
+ my_number
;
1512 pthread_mutex_t
*prune_lock
= &dbs
[my_number
].prune_lock
;
1513 pthread_mutex_t
*prune_run_lock
= &dbs
[my_number
].prune_run_lock
;
1514 pthread_cond_t
*prune_cond
= &dbs
[my_number
].prune_cond
;
1516 pthread_mutex_lock (prune_lock
);
1519 /* Wait, but not forever. */
1521 if (! dbs
[my_number
].clear_cache
)
1522 e
= pthread_cond_timedwait (prune_cond
, prune_lock
, &prune_ts
);
1523 assert (__builtin_expect (e
== 0 || e
== ETIMEDOUT
, 1));
1527 if (e
== ETIMEDOUT
|| now
>= dbs
[my_number
].wakeup_time
1528 || dbs
[my_number
].clear_cache
)
1530 /* We will determine the new timout values based on the
1531 cache content. Should there be concurrent additions to
1532 the cache which are not accounted for in the cache
1533 pruning we want to know about it. Therefore set the
1534 timeout to the maximum. It will be descreased when adding
1535 new entries to the cache, if necessary. */
1536 if (sizeof (time_t) == sizeof (long int))
1537 dbs
[my_number
].wakeup_time
= LONG_MAX
;
1539 dbs
[my_number
].wakeup_time
= INT_MAX
;
1541 /* Unconditionally reset the flag. */
1542 time_t prune_now
= dbs
[my_number
].clear_cache
? LONG_MAX
: now
;
1543 dbs
[my_number
].clear_cache
= 0;
1545 pthread_mutex_unlock (prune_lock
);
1547 /* We use a separate lock for running the prune function (instead
1548 of keeping prune_lock locked) because this enables concurrent
1549 invocations of cache_add which might modify the timeout value. */
1550 pthread_mutex_lock (prune_run_lock
);
1551 next_wait
= prune_cache (&dbs
[my_number
], prune_now
, -1);
1552 pthread_mutex_unlock (prune_run_lock
);
1554 next_wait
= MAX (next_wait
, CACHE_PRUNE_INTERVAL
);
1555 /* If clients cannot determine for sure whether nscd is running
1556 we need to wake up occasionally to update the timestamp.
1557 Wait 90% of the update period. */
1558 #define UPDATE_MAPPING_TIMEOUT (MAPPING_TIMEOUT * 9 / 10)
1559 if (__builtin_expect (! dont_need_update
, 0))
1561 next_wait
= MIN (UPDATE_MAPPING_TIMEOUT
, next_wait
);
1562 dbs
[my_number
].head
->timestamp
= now
;
1565 pthread_mutex_lock (prune_lock
);
1567 /* Make it known when we will wake up again. */
1568 if (now
+ next_wait
< dbs
[my_number
].wakeup_time
)
1569 dbs
[my_number
].wakeup_time
= now
+ next_wait
;
1571 next_wait
= dbs
[my_number
].wakeup_time
- now
;
1574 /* The cache was just pruned. Do not do it again now. Just
1575 use the new timeout value. */
1576 next_wait
= dbs
[my_number
].wakeup_time
- now
;
1578 if (clock_gettime (timeout_clock
, &prune_ts
) == -1)
1579 /* Should never happen. */
1582 /* Compute next timeout time. */
1583 prune_ts
.tv_sec
+= next_wait
;
1588 /* This is the main loop. It is replicated in different threads but
1589 the the use of the ready list makes sure only one thread handles an
1590 incoming connection. */
1592 __attribute__ ((__noreturn__
))
1593 nscd_run_worker (void *p
)
1597 /* Initial locking. */
1598 pthread_mutex_lock (&readylist_lock
);
1600 /* One more thread available. */
1605 while (readylist
== NULL
)
1606 pthread_cond_wait (&readylist_cond
, &readylist_lock
);
1608 struct fdlist
*it
= readylist
->next
;
1609 if (readylist
->next
== readylist
)
1610 /* Just one entry on the list. */
1613 readylist
->next
= it
->next
;
1615 /* Extract the information and mark the record ready to be used
1620 /* One more thread available. */
1623 /* We are done with the list. */
1624 pthread_mutex_unlock (&readylist_lock
);
1626 #ifndef __ASSUME_ACCEPT4
1627 if (have_accept4
< 0)
1629 /* We do not want to block on a short read or so. */
1630 int fl
= fcntl (fd
, F_GETFL
);
1631 if (fl
== -1 || fcntl (fd
, F_SETFL
, fl
| O_NONBLOCK
) == -1)
1636 /* Now read the request. */
1638 if (__builtin_expect (TEMP_FAILURE_RETRY (read (fd
, &req
, sizeof (req
)))
1639 != sizeof (req
), 0))
1641 /* We failed to read data. Note that this also might mean we
1642 failed because we would have blocked. */
1643 if (debug_level
> 0)
1644 dbg_log (_("short read while reading request: %s"),
1645 strerror_r (errno
, buf
, sizeof (buf
)));
1649 /* Check whether this is a valid request type. */
1650 if (req
.type
< GETPWBYNAME
|| req
.type
>= LASTREQ
)
1653 /* Some systems have no SO_PEERCRED implementation. They don't
1654 care about security so we don't as well. */
1659 if (__builtin_expect (debug_level
> 0, 0))
1661 struct ucred caller
;
1662 socklen_t optlen
= sizeof (caller
);
1664 if (getsockopt (fd
, SOL_SOCKET
, SO_PEERCRED
, &caller
, &optlen
) == 0)
1668 const pid_t pid
= 0;
1671 /* It should not be possible to crash the nscd with a silly
1672 request (i.e., a terribly large key). We limit the size to 1kb. */
1673 if (__builtin_expect (req
.key_len
, 1) < 0
1674 || __builtin_expect (req
.key_len
, 1) > MAXKEYLEN
)
1676 if (debug_level
> 0)
1677 dbg_log (_("key length in request too long: %d"), req
.key_len
);
1682 char keybuf
[MAXKEYLEN
];
1684 if (__builtin_expect (TEMP_FAILURE_RETRY (read (fd
, keybuf
,
1688 /* Again, this can also mean we would have blocked. */
1689 if (debug_level
> 0)
1690 dbg_log (_("short read while reading request key: %s"),
1691 strerror_r (errno
, buf
, sizeof (buf
)));
1695 if (__builtin_expect (debug_level
, 0) > 0)
1700 handle_request: request received (Version = %d) from PID %ld"),
1701 req
.version
, (long int) pid
);
1705 handle_request: request received (Version = %d)"), req
.version
);
1708 /* Phew, we got all the data, now process it. */
1709 handle_request (fd
, &req
, keybuf
, uid
, pid
);
1717 pthread_mutex_lock (&readylist_lock
);
1719 /* One more thread available. */
1725 static unsigned int nconns
;
1730 pthread_mutex_lock (&readylist_lock
);
1732 /* Find an empty entry in FDLIST. */
1734 for (inner
= 0; inner
< nconns
; ++inner
)
1735 if (fdlist
[inner
].next
== NULL
)
1737 assert (inner
< nconns
);
1739 fdlist
[inner
].fd
= fd
;
1741 if (readylist
== NULL
)
1742 readylist
= fdlist
[inner
].next
= &fdlist
[inner
];
1745 fdlist
[inner
].next
= readylist
->next
;
1746 readylist
= readylist
->next
= &fdlist
[inner
];
1749 bool do_signal
= true;
1750 if (__builtin_expect (nready
== 0, 0))
1755 /* Try to start another thread to help out. */
1757 if (nthreads
< max_nthreads
1758 && pthread_create (&th
, &attr
, nscd_run_worker
,
1759 (void *) (long int) nthreads
) == 0)
1761 /* We got another thread. */
1763 /* The new thread might need a kick. */
1769 pthread_mutex_unlock (&readylist_lock
);
1771 /* Tell one of the worker threads there is work to do. */
1773 pthread_cond_signal (&readylist_cond
);
1777 /* Check whether restarting should happen. */
1779 restart_p (time_t now
)
1781 return (paranoia
&& readylist
== NULL
&& nready
== nthreads
1782 && now
>= restart_time
);
1786 /* Array for times a connection was accepted. */
1787 static time_t *starttime
;
1791 __attribute__ ((__noreturn__
))
1792 main_loop_poll (void)
1794 struct pollfd
*conns
= (struct pollfd
*) xmalloc (nconns
1795 * sizeof (conns
[0]));
1798 conns
[0].events
= POLLRDNORM
;
1800 size_t firstfree
= 1;
1803 if (inotify_fd
!= -1)
1805 conns
[1].fd
= inotify_fd
;
1806 conns
[1].events
= POLLRDNORM
;
1814 /* Wait for any event. We wait at most a couple of seconds so
1815 that we can check whether we should close any of the accepted
1816 connections since we have not received a request. */
1817 #define MAX_ACCEPT_TIMEOUT 30
1818 #define MIN_ACCEPT_TIMEOUT 5
1819 #define MAIN_THREAD_TIMEOUT \
1820 (MAX_ACCEPT_TIMEOUT * 1000 \
1821 - ((MAX_ACCEPT_TIMEOUT - MIN_ACCEPT_TIMEOUT) * 1000 * nused) / (2 * nconns))
1823 int n
= poll (conns
, nused
, MAIN_THREAD_TIMEOUT
);
1825 time_t now
= time (NULL
);
1827 /* If there is a descriptor ready for reading or there is a new
1828 connection, process this now. */
1831 if (conns
[0].revents
!= 0)
1833 /* We have a new incoming connection. Accept the connection. */
1836 #ifndef __ASSUME_ACCEPT4
1838 if (have_accept4
>= 0)
1841 fd
= TEMP_FAILURE_RETRY (accept4 (sock
, NULL
, NULL
,
1843 #ifndef __ASSUME_ACCEPT4
1844 if (have_accept4
== 0)
1845 have_accept4
= fd
!= -1 || errno
!= ENOSYS
? 1 : -1;
1848 #ifndef __ASSUME_ACCEPT4
1849 if (have_accept4
< 0)
1850 fd
= TEMP_FAILURE_RETRY (accept (sock
, NULL
, NULL
));
1853 /* Use the descriptor if we have not reached the limit. */
1856 if (firstfree
< nconns
)
1858 conns
[firstfree
].fd
= fd
;
1859 conns
[firstfree
].events
= POLLRDNORM
;
1860 starttime
[firstfree
] = now
;
1861 if (firstfree
>= nused
)
1862 nused
= firstfree
+ 1;
1866 while (firstfree
< nused
&& conns
[firstfree
].fd
!= -1);
1869 /* We cannot use the connection so close it. */
1878 if (inotify_fd
!= -1 && conns
[1].fd
== inotify_fd
)
1880 if (conns
[1].revents
!= 0)
1882 bool to_clear
[lastdb
] = { false, };
1886 # define PATH_MAX 1024
1888 struct inotify_event i
;
1889 char buf
[sizeof (struct inotify_event
) + PATH_MAX
];
1894 ssize_t nb
= TEMP_FAILURE_RETRY (read (inotify_fd
, &inev
,
1896 if (nb
< (ssize_t
) sizeof (struct inotify_event
))
1898 if (__builtin_expect (nb
== -1 && errno
!= EAGAIN
,
1901 /* Something went wrong when reading the inotify
1902 data. Better disable inotify. */
1904 disabled inotify after read error %d"),
1916 /* Check which of the files changed. */
1917 for (size_t dbcnt
= 0; dbcnt
< lastdb
; ++dbcnt
)
1918 if (inev
.i
.wd
== dbs
[dbcnt
].inotify_descr
)
1920 to_clear
[dbcnt
] = true;
1924 if (inev
.i
.wd
== resolv_conf_descr
)
1927 to_clear
[hstdb
] = true;
1932 /* Actually perform the cache clearing. */
1933 for (size_t dbcnt
= 0; dbcnt
< lastdb
; ++dbcnt
)
1934 if (to_clear
[dbcnt
])
1936 pthread_mutex_lock (&dbs
[dbcnt
].prune_lock
);
1937 dbs
[dbcnt
].clear_cache
= 1;
1938 pthread_mutex_unlock (&dbs
[dbcnt
].prune_lock
);
1939 pthread_cond_signal (&dbs
[dbcnt
].prune_cond
);
1949 for (size_t cnt
= first
; cnt
< nused
&& n
> 0; ++cnt
)
1950 if (conns
[cnt
].revents
!= 0)
1952 fd_ready (conns
[cnt
].fd
);
1954 /* Clean up the CONNS array. */
1956 if (cnt
< firstfree
)
1958 if (cnt
== nused
- 1)
1961 while (conns
[nused
- 1].fd
== -1);
1967 /* Now find entries which have timed out. */
1970 /* We make the timeout length depend on the number of file
1971 descriptors currently used. */
1972 #define ACCEPT_TIMEOUT \
1973 (MAX_ACCEPT_TIMEOUT \
1974 - ((MAX_ACCEPT_TIMEOUT - MIN_ACCEPT_TIMEOUT) * nused) / nconns)
1975 time_t laststart
= now
- ACCEPT_TIMEOUT
;
1977 for (size_t cnt
= nused
- 1; cnt
> 0; --cnt
)
1979 if (conns
[cnt
].fd
!= -1 && starttime
[cnt
] < laststart
)
1981 /* Remove the entry, it timed out. */
1982 (void) close (conns
[cnt
].fd
);
1985 if (cnt
< firstfree
)
1987 if (cnt
== nused
- 1)
1990 while (conns
[nused
- 1].fd
== -1);
1994 if (restart_p (now
))
2002 main_loop_epoll (int efd
)
2004 struct epoll_event ev
= { 0, };
2008 /* Add the socket. */
2009 ev
.events
= EPOLLRDNORM
;
2011 if (epoll_ctl (efd
, EPOLL_CTL_ADD
, sock
, &ev
) == -1)
2012 /* We cannot use epoll. */
2015 # ifdef HAVE_INOTIFY
2016 if (inotify_fd
!= -1)
2018 ev
.events
= EPOLLRDNORM
;
2019 ev
.data
.fd
= inotify_fd
;
2020 if (epoll_ctl (efd
, EPOLL_CTL_ADD
, inotify_fd
, &ev
) == -1)
2021 /* We cannot use epoll. */
2029 struct epoll_event revs
[100];
2030 # define nrevs (sizeof (revs) / sizeof (revs[0]))
2032 int n
= epoll_wait (efd
, revs
, nrevs
, MAIN_THREAD_TIMEOUT
);
2034 time_t now
= time (NULL
);
2036 for (int cnt
= 0; cnt
< n
; ++cnt
)
2037 if (revs
[cnt
].data
.fd
== sock
)
2039 /* A new connection. */
2042 # ifndef __ASSUME_ACCEPT4
2044 if (have_accept4
>= 0)
2047 fd
= TEMP_FAILURE_RETRY (accept4 (sock
, NULL
, NULL
,
2049 # ifndef __ASSUME_ACCEPT4
2050 if (have_accept4
== 0)
2051 have_accept4
= fd
!= -1 || errno
!= ENOSYS
? 1 : -1;
2054 # ifndef __ASSUME_ACCEPT4
2055 if (have_accept4
< 0)
2056 fd
= TEMP_FAILURE_RETRY (accept (sock
, NULL
, NULL
));
2059 /* Use the descriptor if we have not reached the limit. */
2062 /* Try to add the new descriptor. */
2065 || epoll_ctl (efd
, EPOLL_CTL_ADD
, fd
, &ev
) == -1)
2066 /* The descriptor is too large or something went
2067 wrong. Close the descriptor. */
2071 /* Remember when we accepted the connection. */
2072 starttime
[fd
] = now
;
2081 # ifdef HAVE_INOTIFY
2082 else if (revs
[cnt
].data
.fd
== inotify_fd
)
2084 bool to_clear
[lastdb
] = { false, };
2087 struct inotify_event i
;
2088 char buf
[sizeof (struct inotify_event
) + PATH_MAX
];
2093 ssize_t nb
= TEMP_FAILURE_RETRY (read (inotify_fd
, &inev
,
2095 if (nb
< (ssize_t
) sizeof (struct inotify_event
))
2097 if (__builtin_expect (nb
== -1 && errno
!= EAGAIN
, 0))
2099 /* Something went wrong when reading the inotify
2100 data. Better disable inotify. */
2101 dbg_log (_("disabled inotify after read error %d"),
2103 (void) epoll_ctl (efd
, EPOLL_CTL_DEL
, inotify_fd
,
2111 /* Check which of the files changed. */
2112 for (size_t dbcnt
= 0; dbcnt
< lastdb
; ++dbcnt
)
2113 if (inev
.i
.wd
== dbs
[dbcnt
].inotify_descr
)
2115 to_clear
[dbcnt
] = true;
2119 if (inev
.i
.wd
== resolv_conf_descr
)
2122 to_clear
[hstdb
] = true;
2127 /* Actually perform the cache clearing. */
2128 for (size_t dbcnt
= 0; dbcnt
< lastdb
; ++dbcnt
)
2129 if (to_clear
[dbcnt
])
2131 pthread_mutex_lock (&dbs
[dbcnt
].prune_lock
);
2132 dbs
[dbcnt
].clear_cache
= 1;
2133 pthread_mutex_unlock (&dbs
[dbcnt
].prune_lock
);
2134 pthread_cond_signal (&dbs
[dbcnt
].prune_cond
);
2140 /* Remove the descriptor from the epoll descriptor. */
2141 (void) epoll_ctl (efd
, EPOLL_CTL_DEL
, revs
[cnt
].data
.fd
, NULL
);
2143 /* Get a worker to handle the request. */
2144 fd_ready (revs
[cnt
].data
.fd
);
2146 /* Reset the time. */
2147 starttime
[revs
[cnt
].data
.fd
] = 0;
2148 if (revs
[cnt
].data
.fd
== highest
)
2151 while (highest
> 0 && starttime
[highest
] == 0);
2156 /* Now look for descriptors for accepted connections which have
2157 no reply in too long of a time. */
2158 time_t laststart
= now
- ACCEPT_TIMEOUT
;
2159 assert (starttime
[sock
] == 0);
2160 assert (inotify_fd
== -1 || starttime
[inotify_fd
] == 0);
2161 for (int cnt
= highest
; cnt
> STDERR_FILENO
; --cnt
)
2162 if (starttime
[cnt
] != 0 && starttime
[cnt
] < laststart
)
2164 /* We are waiting for this one for too long. Close it. */
2165 (void) epoll_ctl (efd
, EPOLL_CTL_DEL
, cnt
, NULL
);
2173 else if (cnt
!= sock
&& starttime
[cnt
] == 0 && cnt
== highest
)
2176 if (restart_p (now
))
2183 /* Start all the threads we want. The initial process is thread no. 1. */
2185 start_threads (void)
2187 /* Initialize the conditional variable we will use. The only
2188 non-standard attribute we might use is the clock selection. */
2189 pthread_condattr_t condattr
;
2190 pthread_condattr_init (&condattr
);
2192 #if defined _POSIX_CLOCK_SELECTION && _POSIX_CLOCK_SELECTION >= 0 \
2193 && defined _POSIX_MONOTONIC_CLOCK && _POSIX_MONOTONIC_CLOCK >= 0
2194 /* Determine whether the monotonous clock is available. */
2195 struct timespec dummy
;
2196 # if _POSIX_MONOTONIC_CLOCK == 0
2197 if (sysconf (_SC_MONOTONIC_CLOCK
) > 0)
2199 # if _POSIX_CLOCK_SELECTION == 0
2200 if (sysconf (_SC_CLOCK_SELECTION
) > 0)
2202 if (clock_getres (CLOCK_MONOTONIC
, &dummy
) == 0
2203 && pthread_condattr_setclock (&condattr
, CLOCK_MONOTONIC
) == 0)
2204 timeout_clock
= CLOCK_MONOTONIC
;
2207 /* Create the attribute for the threads. They are all created
2209 pthread_attr_init (&attr
);
2210 pthread_attr_setdetachstate (&attr
, PTHREAD_CREATE_DETACHED
);
2211 /* Use 1MB stacks, twice as much for 64-bit architectures. */
2212 pthread_attr_setstacksize (&attr
, NSCD_THREAD_STACKSIZE
);
2214 /* We allow less than LASTDB threads only for debugging. */
2215 if (debug_level
== 0)
2216 nthreads
= MAX (nthreads
, lastdb
);
2218 /* Create the threads which prune the databases. */
2219 // XXX Ideally this work would be done by some of the worker threads.
2220 // XXX But this is problematic since we would need to be able to wake
2221 // XXX them up explicitly as well as part of the group handling the
2222 // XXX ready-list. This requires an operation where we can wait on
2223 // XXX two conditional variables at the same time. This operation
2224 // XXX does not exist (yet).
2225 for (long int i
= 0; i
< lastdb
; ++i
)
2227 /* Initialize the conditional variable. */
2228 if (pthread_cond_init (&dbs
[i
].prune_cond
, &condattr
) != 0)
2230 dbg_log (_("could not initialize conditional variable"));
2236 && pthread_create (&th
, &attr
, nscd_run_prune
, (void *) i
) != 0)
2238 dbg_log (_("could not start clean-up thread; terminating"));
2243 pthread_condattr_destroy (&condattr
);
2245 for (long int i
= 0; i
< nthreads
; ++i
)
2248 if (pthread_create (&th
, &attr
, nscd_run_worker
, NULL
) != 0)
2252 dbg_log (_("could not start any worker thread; terminating"));
2260 /* Determine how much room for descriptors we should initially
2261 allocate. This might need to change later if we cap the number
2263 const long int nfds
= sysconf (_SC_OPEN_MAX
);
2265 #define MAXCONN 16384
2266 if (nfds
== -1 || nfds
> MAXCONN
)
2268 else if (nfds
< MINCONN
)
2273 /* We need memory to pass descriptors on to the worker threads. */
2274 fdlist
= (struct fdlist
*) xcalloc (nconns
, sizeof (fdlist
[0]));
2275 /* Array to keep track when connection was accepted. */
2276 starttime
= (time_t *) xcalloc (nconns
, sizeof (starttime
[0]));
2278 /* In the main thread we execute the loop which handles incoming
2281 int efd
= epoll_create (100);
2284 main_loop_epoll (efd
);
2293 /* Look up the uid, gid, and supplementary groups to run nscd as. When
2294 this function is called, we are not listening on the nscd socket yet so
2295 we can just use the ordinary lookup functions without causing a lockup */
2297 begin_drop_privileges (void)
2299 struct passwd
*pwd
= getpwnam (server_user
);
2303 dbg_log (_("Failed to run nscd as user '%s'"), server_user
);
2304 error (EXIT_FAILURE
, 0, _("Failed to run nscd as user '%s'"),
2308 server_uid
= pwd
->pw_uid
;
2309 server_gid
= pwd
->pw_gid
;
2311 /* Save the old UID/GID if we have to change back. */
2314 old_uid
= getuid ();
2315 old_gid
= getgid ();
2318 if (getgrouplist (server_user
, server_gid
, NULL
, &server_ngroups
) == 0)
2320 /* This really must never happen. */
2321 dbg_log (_("Failed to run nscd as user '%s'"), server_user
);
2322 error (EXIT_FAILURE
, errno
, _("initial getgrouplist failed"));
2325 server_groups
= (gid_t
*) xmalloc (server_ngroups
* sizeof (gid_t
));
2327 if (getgrouplist (server_user
, server_gid
, server_groups
, &server_ngroups
)
2330 dbg_log (_("Failed to run nscd as user '%s'"), server_user
);
2331 error (EXIT_FAILURE
, errno
, _("getgrouplist failed"));
2336 /* Call setgroups(), setgid(), and setuid() to drop root privileges and
2337 run nscd as the user specified in the configuration file. */
2339 finish_drop_privileges (void)
2341 #if defined HAVE_LIBAUDIT && defined HAVE_LIBCAP
2342 /* We need to preserve the capabilities to connect to the audit daemon. */
2343 cap_t new_caps
= preserve_capabilities ();
2346 if (setgroups (server_ngroups
, server_groups
) == -1)
2348 dbg_log (_("Failed to run nscd as user '%s'"), server_user
);
2349 error (EXIT_FAILURE
, errno
, _("setgroups failed"));
2354 res
= setresgid (server_gid
, server_gid
, old_gid
);
2356 res
= setgid (server_gid
);
2359 dbg_log (_("Failed to run nscd as user '%s'"), server_user
);
2365 res
= setresuid (server_uid
, server_uid
, old_uid
);
2367 res
= setuid (server_uid
);
2370 dbg_log (_("Failed to run nscd as user '%s'"), server_user
);
2375 #if defined HAVE_LIBAUDIT && defined HAVE_LIBCAP
2376 /* Remove the temporary capabilities. */
2377 install_real_capabilities (new_caps
);