1 /* Inner loops of cache daemon.
2 Copyright (C) 1998-2013 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, see <http://www.gnu.org/licenses/>. */
35 #include <arpa/inet.h>
37 # include <linux/netlink.h>
38 # include <linux/rtnetlink.h>
41 # include <sys/epoll.h>
44 # include <sys/inotify.h>
47 #include <sys/param.h>
50 # include <sys/sendfile.h>
52 #include <sys/socket.h>
59 #include <resolv/resolv.h>
61 #include <kernel-features.h>
64 /* Support to run nscd as an unprivileged user */
65 const char *server_user
;
66 static uid_t server_uid
;
67 static gid_t server_gid
;
68 const char *stat_user
;
70 static gid_t
*server_groups
;
74 static int server_ngroups
;
76 static pthread_attr_t attr
;
78 static void begin_drop_privileges (void);
79 static void finish_drop_privileges (void);
81 /* Map request type to a string. */
82 const char *const serv2str
[LASTREQ
] =
84 [GETPWBYNAME
] = "GETPWBYNAME",
85 [GETPWBYUID
] = "GETPWBYUID",
86 [GETGRBYNAME
] = "GETGRBYNAME",
87 [GETGRBYGID
] = "GETGRBYGID",
88 [GETHOSTBYNAME
] = "GETHOSTBYNAME",
89 [GETHOSTBYNAMEv6
] = "GETHOSTBYNAMEv6",
90 [GETHOSTBYADDR
] = "GETHOSTBYADDR",
91 [GETHOSTBYADDRv6
] = "GETHOSTBYADDRv6",
92 [SHUTDOWN
] = "SHUTDOWN",
93 [GETSTAT
] = "GETSTAT",
94 [INVALIDATE
] = "INVALIDATE",
95 [GETFDPW
] = "GETFDPW",
96 [GETFDGR
] = "GETFDGR",
97 [GETFDHST
] = "GETFDHST",
99 [INITGROUPS
] = "INITGROUPS",
100 [GETSERVBYNAME
] = "GETSERVBYNAME",
101 [GETSERVBYPORT
] = "GETSERVBYPORT",
102 [GETFDSERV
] = "GETFDSERV",
103 [GETNETGRENT
] = "GETNETGRENT",
104 [INNETGR
] = "INNETGR",
105 [GETFDNETGR
] = "GETFDNETGR"
108 /* The control data structures for the services. */
109 struct database_dyn dbs
[lastdb
] =
112 .lock
= PTHREAD_RWLOCK_WRITER_NONRECURSIVE_INITIALIZER_NP
,
113 .prune_lock
= PTHREAD_MUTEX_INITIALIZER
,
114 .prune_run_lock
= PTHREAD_MUTEX_INITIALIZER
,
120 .max_db_size
= DEFAULT_MAX_DB_SIZE
,
121 .suggested_module
= DEFAULT_SUGGESTED_MODULE
,
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
,
141 .db_filename
= _PATH_NSCD_GROUP_DB
,
142 .disabled_iov
= &grp_iov_disabled
,
150 .lock
= PTHREAD_RWLOCK_WRITER_NONRECURSIVE_INITIALIZER_NP
,
151 .prune_lock
= PTHREAD_MUTEX_INITIALIZER
,
152 .prune_run_lock
= PTHREAD_MUTEX_INITIALIZER
,
156 .propagate
= 0, /* Not used. */
158 .max_db_size
= DEFAULT_MAX_DB_SIZE
,
159 .suggested_module
= DEFAULT_SUGGESTED_MODULE
,
160 .db_filename
= _PATH_NSCD_HOSTS_DB
,
161 .disabled_iov
= &hst_iov_disabled
,
169 .lock
= PTHREAD_RWLOCK_WRITER_NONRECURSIVE_INITIALIZER_NP
,
170 .prune_lock
= PTHREAD_MUTEX_INITIALIZER
,
171 .prune_run_lock
= PTHREAD_MUTEX_INITIALIZER
,
175 .propagate
= 0, /* Not used. */
177 .max_db_size
= DEFAULT_MAX_DB_SIZE
,
178 .suggested_module
= DEFAULT_SUGGESTED_MODULE
,
179 .db_filename
= _PATH_NSCD_SERVICES_DB
,
180 .disabled_iov
= &serv_iov_disabled
,
188 .lock
= PTHREAD_RWLOCK_WRITER_NONRECURSIVE_INITIALIZER_NP
,
189 .prune_lock
= PTHREAD_MUTEX_INITIALIZER
,
190 .prune_run_lock
= PTHREAD_MUTEX_INITIALIZER
,
194 .propagate
= 0, /* Not used. */
196 .max_db_size
= DEFAULT_MAX_DB_SIZE
,
197 .suggested_module
= DEFAULT_SUGGESTED_MODULE
,
198 .db_filename
= _PATH_NSCD_NETGROUP_DB
,
199 .disabled_iov
= &netgroup_iov_disabled
,
209 /* Mapping of request type to database. */
213 struct database_dyn
*db
;
214 } const reqinfo
[LASTREQ
] =
216 [GETPWBYNAME
] = { true, &dbs
[pwddb
] },
217 [GETPWBYUID
] = { true, &dbs
[pwddb
] },
218 [GETGRBYNAME
] = { true, &dbs
[grpdb
] },
219 [GETGRBYGID
] = { true, &dbs
[grpdb
] },
220 [GETHOSTBYNAME
] = { true, &dbs
[hstdb
] },
221 [GETHOSTBYNAMEv6
] = { true, &dbs
[hstdb
] },
222 [GETHOSTBYADDR
] = { true, &dbs
[hstdb
] },
223 [GETHOSTBYADDRv6
] = { true, &dbs
[hstdb
] },
224 [SHUTDOWN
] = { false, NULL
},
225 [GETSTAT
] = { false, NULL
},
226 [SHUTDOWN
] = { false, NULL
},
227 [GETFDPW
] = { false, &dbs
[pwddb
] },
228 [GETFDGR
] = { false, &dbs
[grpdb
] },
229 [GETFDHST
] = { false, &dbs
[hstdb
] },
230 [GETAI
] = { true, &dbs
[hstdb
] },
231 [INITGROUPS
] = { true, &dbs
[grpdb
] },
232 [GETSERVBYNAME
] = { true, &dbs
[servdb
] },
233 [GETSERVBYPORT
] = { true, &dbs
[servdb
] },
234 [GETFDSERV
] = { false, &dbs
[servdb
] },
235 [GETNETGRENT
] = { true, &dbs
[netgrdb
] },
236 [INNETGR
] = { true, &dbs
[netgrdb
] },
237 [GETFDNETGR
] = { false, &dbs
[netgrdb
] }
241 /* Initial number of threads to use. */
243 /* Maximum number of threads to use. */
244 int max_nthreads
= 32;
246 /* Socket for incoming connections. */
250 /* Inotify descriptor. */
255 /* Descriptor for netlink status updates. */
256 static int nl_status_fd
= -1;
259 #ifndef __ASSUME_SOCK_CLOEXEC
260 /* Negative if SOCK_CLOEXEC is not supported, positive if it is, zero
261 before be know the result. */
262 static int have_sock_cloexec
;
264 #ifndef __ASSUME_ACCEPT4
265 static int have_accept4
;
268 /* Number of times clients had to wait. */
269 unsigned long int client_queued
;
273 writeall (int fd
, const void *buf
, size_t len
)
279 ret
= TEMP_FAILURE_RETRY (send (fd
, buf
, n
, MSG_NOSIGNAL
));
282 buf
= (const char *) buf
+ ret
;
286 return ret
< 0 ? ret
: len
- n
;
292 sendfileall (int tofd
, int fromfd
, off_t off
, size_t len
)
299 ret
= TEMP_FAILURE_RETRY (sendfile (tofd
, fromfd
, &off
, n
));
305 return ret
< 0 ? ret
: len
- n
;
313 /* The following three are not really used, they are symbolic constants. */
319 use_he_begin
= use_he
| use_begin
,
320 use_he_end
= use_he
| use_end
,
323 use_key_begin
= use_key
| use_begin
,
324 use_key_end
= use_key
| use_end
,
325 use_key_first
= use_key_begin
| use_first
,
328 use_data_begin
= use_data
| use_begin
,
329 use_data_end
= use_data
| use_end
,
330 use_data_first
= use_data_begin
| use_first
335 check_use (const char *data
, nscd_ssize_t first_free
, uint8_t *usemap
,
336 enum usekey use
, ref_t start
, size_t len
)
340 if (start
> first_free
|| start
+ len
> first_free
341 || (start
& BLOCK_ALIGN_M1
))
344 if (usemap
[start
] == use_not
)
346 /* Add the start marker. */
347 usemap
[start
] = use
| use_begin
;
351 if (usemap
[++start
] != use_not
)
356 /* Add the end marker. */
357 usemap
[start
] = use
| use_end
;
359 else if ((usemap
[start
] & ~use_first
) == ((use
| use_begin
) & ~use_first
))
361 /* Hash entries can't be shared. */
365 usemap
[start
] |= (use
& use_first
);
369 if (usemap
[++start
] != use
)
372 if (usemap
[++start
] != (use
| use_end
))
376 /* Points to a wrong object or somewhere in the middle. */
383 /* Verify data in persistent database. */
385 verify_persistent_db (void *mem
, struct database_pers_head
*readhead
, int dbnr
)
387 assert (dbnr
== pwddb
|| dbnr
== grpdb
|| dbnr
== hstdb
|| dbnr
== servdb
390 time_t now
= time (NULL
);
392 struct database_pers_head
*head
= mem
;
393 struct database_pers_head head_copy
= *head
;
395 /* Check that the header that was read matches the head in the database. */
396 if (memcmp (head
, readhead
, sizeof (*head
)) != 0)
399 /* First some easy tests: make sure the database header is sane. */
400 if (head
->version
!= DB_VERSION
401 || head
->header_size
!= sizeof (*head
)
402 /* We allow a timestamp to be one hour ahead of the current time.
403 This should cover daylight saving time changes. */
404 || head
->timestamp
> now
+ 60 * 60 + 60
405 || (head
->gc_cycle
& 1)
407 || (size_t) head
->module
> INT32_MAX
/ sizeof (ref_t
)
408 || (size_t) head
->data_size
> INT32_MAX
- head
->module
* sizeof (ref_t
)
409 || head
->first_free
< 0
410 || head
->first_free
> head
->data_size
411 || (head
->first_free
& BLOCK_ALIGN_M1
) != 0
412 || head
->maxnentries
< 0
413 || head
->maxnsearched
< 0)
416 uint8_t *usemap
= calloc (head
->first_free
, 1);
420 const char *data
= (char *) &head
->array
[roundup (head
->module
,
421 ALIGN
/ sizeof (ref_t
))];
423 nscd_ssize_t he_cnt
= 0;
424 for (nscd_ssize_t cnt
= 0; cnt
< head
->module
; ++cnt
)
426 ref_t trail
= head
->array
[cnt
];
430 while (work
!= ENDREF
)
432 if (! check_use (data
, head
->first_free
, usemap
, use_he
, work
,
433 sizeof (struct hashentry
)))
436 /* Now we know we can dereference the record. */
437 struct hashentry
*here
= (struct hashentry
*) (data
+ work
);
441 /* Make sure the record is for this type of service. */
442 if (here
->type
>= LASTREQ
443 || reqinfo
[here
->type
].db
!= &dbs
[dbnr
])
446 /* Validate boolean field value. */
447 if (here
->first
!= false && here
->first
!= true)
455 || here
->packet
> head
->first_free
456 || here
->packet
+ sizeof (struct datahead
) > head
->first_free
)
459 struct datahead
*dh
= (struct datahead
*) (data
+ here
->packet
);
461 if (! check_use (data
, head
->first_free
, usemap
,
462 use_data
| (here
->first
? use_first
: 0),
463 here
->packet
, dh
->allocsize
))
466 if (dh
->allocsize
< sizeof (struct datahead
)
467 || dh
->recsize
> dh
->allocsize
468 || (dh
->notfound
!= false && dh
->notfound
!= true)
469 || (dh
->usable
!= false && dh
->usable
!= true))
472 if (here
->key
< here
->packet
+ sizeof (struct datahead
)
473 || here
->key
> here
->packet
+ dh
->allocsize
474 || here
->key
+ here
->len
> here
->packet
+ dh
->allocsize
)
477 /* If keys can appear outside of data, this should be done
478 instead. But gc doesn't mark the data in that case. */
479 if (! check_use (data
, head
->first_free
, usemap
,
480 use_key
| (here
->first
? use_first
: 0),
481 here
->key
, here
->len
))
489 /* A circular list, this must not happen. */
492 trail
= ((struct hashentry
*) (data
+ trail
))->next
;
497 if (he_cnt
!= head
->nentries
)
500 /* See if all data and keys had at least one reference from
501 he->first == true hashentry. */
502 for (ref_t idx
= 0; idx
< head
->first_free
; ++idx
)
505 if (usemap
[idx
] == use_key_begin
)
508 if (usemap
[idx
] == use_data_begin
)
512 /* Finally, make sure the database hasn't changed since the first test. */
513 if (memcmp (mem
, &head_copy
, sizeof (*head
)) != 0)
526 # define EXTRA_O_FLAGS O_CLOEXEC
528 # define EXTRA_O_FLAGS 0
532 /* Initialize database information structures. */
536 /* Look up unprivileged uid/gid/groups before we start listening on the
538 if (server_user
!= NULL
)
539 begin_drop_privileges ();
542 /* No configuration for this value, assume a default. */
545 for (size_t cnt
= 0; cnt
< lastdb
; ++cnt
)
546 if (dbs
[cnt
].enabled
)
548 pthread_rwlock_init (&dbs
[cnt
].lock
, NULL
);
549 pthread_mutex_init (&dbs
[cnt
].memlock
, NULL
);
551 if (dbs
[cnt
].persistent
)
553 /* Try to open the appropriate file on disk. */
554 int fd
= open (dbs
[cnt
].db_filename
, O_RDWR
| EXTRA_O_FLAGS
);
561 struct database_pers_head head
;
562 ssize_t n
= TEMP_FAILURE_RETRY (read (fd
, &head
,
564 if (n
!= sizeof (head
) || fstat64 (fd
, &st
) != 0)
567 /* The code is single-threaded at this point so
568 using strerror is just fine. */
569 msg
= strerror (errno
);
571 dbg_log (_("invalid persistent database file \"%s\": %s"),
572 dbs
[cnt
].db_filename
, msg
);
573 unlink (dbs
[cnt
].db_filename
);
575 else if (head
.module
== 0 && head
.data_size
== 0)
577 /* The file has been created, but the head has not
578 been initialized yet. */
579 msg
= _("uninitialized header");
582 else if (head
.header_size
!= (int) sizeof (head
))
584 msg
= _("header size does not match");
587 else if ((total
= (sizeof (head
)
588 + roundup (head
.module
* sizeof (ref_t
),
592 || total
< sizeof (head
))
594 msg
= _("file size does not match");
597 /* Note we map with the maximum size allowed for the
598 database. This is likely much larger than the
599 actual file size. This is OK on most OSes since
600 extensions of the underlying file will
601 automatically translate more pages available for
603 else if ((mem
= mmap (NULL
, dbs
[cnt
].max_db_size
,
604 PROT_READ
| PROT_WRITE
,
608 else if (!verify_persistent_db (mem
, &head
, cnt
))
611 msg
= _("verification failed");
616 /* Success. We have the database. */
618 dbs
[cnt
].memsize
= total
;
619 dbs
[cnt
].data
= (char *)
620 &dbs
[cnt
].head
->array
[roundup (dbs
[cnt
].head
->module
,
621 ALIGN
/ sizeof (ref_t
))];
622 dbs
[cnt
].mmap_used
= true;
624 if (dbs
[cnt
].suggested_module
> head
.module
)
625 dbg_log (_("suggested size of table for database %s larger than the persistent database's table"),
630 /* We also need a read-only descriptor. */
633 dbs
[cnt
].ro_fd
= open (dbs
[cnt
].db_filename
,
634 O_RDONLY
| EXTRA_O_FLAGS
);
635 if (dbs
[cnt
].ro_fd
== -1)
637 cannot create read-only descriptor for \"%s\"; no mmap"),
638 dbs
[cnt
].db_filename
);
641 // XXX Shall we test whether the descriptors actually
642 // XXX point to the same file?
645 /* Close the file descriptors in case something went
646 wrong in which case the variable have not been
651 else if (errno
== EACCES
)
652 error (EXIT_FAILURE
, 0, _("cannot access '%s'"),
653 dbs
[cnt
].db_filename
);
656 if (dbs
[cnt
].head
== NULL
)
658 /* No database loaded. Allocate the data structure,
660 struct database_pers_head head
;
661 size_t total
= (sizeof (head
)
662 + roundup (dbs
[cnt
].suggested_module
663 * sizeof (ref_t
), ALIGN
)
664 + (dbs
[cnt
].suggested_module
665 * DEFAULT_DATASIZE_PER_BUCKET
));
667 /* Try to create the database. If we do not need a
668 persistent database create a temporary file. */
671 if (dbs
[cnt
].persistent
)
673 fd
= open (dbs
[cnt
].db_filename
,
674 O_RDWR
| O_CREAT
| O_EXCL
| O_TRUNC
| EXTRA_O_FLAGS
,
676 if (fd
!= -1 && dbs
[cnt
].shared
)
677 ro_fd
= open (dbs
[cnt
].db_filename
,
678 O_RDONLY
| EXTRA_O_FLAGS
);
682 char fname
[] = _PATH_NSCD_XYZ_DB_TMP
;
683 fd
= mkostemp (fname
, EXTRA_O_FLAGS
);
685 /* We do not need the file name anymore after we
686 opened another file descriptor in read-only mode. */
690 ro_fd
= open (fname
, O_RDONLY
| EXTRA_O_FLAGS
);
700 dbg_log (_("database for %s corrupted or simultaneously used; remove %s manually if necessary and restart"),
701 dbnames
[cnt
], dbs
[cnt
].db_filename
);
702 // XXX Correct way to terminate?
706 if (dbs
[cnt
].persistent
)
707 dbg_log (_("cannot create %s; no persistent database used"),
708 dbs
[cnt
].db_filename
);
710 dbg_log (_("cannot create %s; no sharing possible"),
711 dbs
[cnt
].db_filename
);
713 dbs
[cnt
].persistent
= 0;
714 // XXX remember: no mmap
718 /* Tell the user if we could not create the read-only
720 if (ro_fd
== -1 && dbs
[cnt
].shared
)
722 cannot create read-only descriptor for \"%s\"; no mmap"),
723 dbs
[cnt
].db_filename
);
725 /* Before we create the header, initialize the hash
726 table. That way if we get interrupted while writing
727 the header we can recognize a partially initialized
729 size_t ps
= sysconf (_SC_PAGESIZE
);
731 assert (~ENDREF
== 0);
732 memset (tmpbuf
, '\xff', ps
);
734 size_t remaining
= dbs
[cnt
].suggested_module
* sizeof (ref_t
);
735 off_t offset
= sizeof (head
);
738 if (offset
% ps
!= 0)
740 towrite
= MIN (remaining
, ps
- (offset
% ps
));
741 if (pwrite (fd
, tmpbuf
, towrite
, offset
) != towrite
)
744 remaining
-= towrite
;
747 while (remaining
> ps
)
749 if (pwrite (fd
, tmpbuf
, ps
, offset
) == -1)
756 && pwrite (fd
, tmpbuf
, remaining
, offset
) != remaining
)
759 /* Create the header of the file. */
760 struct database_pers_head head
=
762 .version
= DB_VERSION
,
763 .header_size
= sizeof (head
),
764 .module
= dbs
[cnt
].suggested_module
,
765 .data_size
= (dbs
[cnt
].suggested_module
766 * DEFAULT_DATASIZE_PER_BUCKET
),
771 if ((TEMP_FAILURE_RETRY (write (fd
, &head
, sizeof (head
)))
773 || (TEMP_FAILURE_RETRY_VAL (posix_fallocate (fd
, 0, total
))
775 || (mem
= mmap (NULL
, dbs
[cnt
].max_db_size
,
776 PROT_READ
| PROT_WRITE
,
777 MAP_SHARED
, fd
, 0)) == MAP_FAILED
)
780 unlink (dbs
[cnt
].db_filename
);
781 dbg_log (_("cannot write to database file %s: %s"),
782 dbs
[cnt
].db_filename
, strerror (errno
));
783 dbs
[cnt
].persistent
= 0;
789 dbs
[cnt
].data
= (char *)
790 &dbs
[cnt
].head
->array
[roundup (dbs
[cnt
].head
->module
,
791 ALIGN
/ sizeof (ref_t
))];
792 dbs
[cnt
].memsize
= total
;
793 dbs
[cnt
].mmap_used
= true;
795 /* Remember the descriptors. */
797 dbs
[cnt
].ro_fd
= ro_fd
;
809 #if !defined O_CLOEXEC || !defined __ASSUME_O_CLOEXEC
810 /* We do not check here whether the O_CLOEXEC provided to the
811 open call was successful or not. The two fcntl calls are
812 only performed once each per process start-up and therefore
813 is not noticeable at all. */
815 && ((dbs
[cnt
].wr_fd
!= -1
816 && fcntl (dbs
[cnt
].wr_fd
, F_SETFD
, FD_CLOEXEC
) == -1)
817 || (dbs
[cnt
].ro_fd
!= -1
818 && fcntl (dbs
[cnt
].ro_fd
, F_SETFD
, FD_CLOEXEC
) == -1)))
821 cannot set socket to close on exec: %s; disabling paranoia mode"),
827 if (dbs
[cnt
].head
== NULL
)
829 /* We do not use the persistent database. Just
830 create an in-memory data structure. */
831 assert (! dbs
[cnt
].persistent
);
833 dbs
[cnt
].head
= xmalloc (sizeof (struct database_pers_head
)
834 + (dbs
[cnt
].suggested_module
836 memset (dbs
[cnt
].head
, '\0', sizeof (struct database_pers_head
));
837 assert (~ENDREF
== 0);
838 memset (dbs
[cnt
].head
->array
, '\xff',
839 dbs
[cnt
].suggested_module
* sizeof (ref_t
));
840 dbs
[cnt
].head
->module
= dbs
[cnt
].suggested_module
;
841 dbs
[cnt
].head
->data_size
= (DEFAULT_DATASIZE_PER_BUCKET
842 * dbs
[cnt
].head
->module
);
843 dbs
[cnt
].data
= xmalloc (dbs
[cnt
].head
->data_size
);
844 dbs
[cnt
].head
->first_free
= 0;
847 assert (dbs
[cnt
].ro_fd
== -1);
851 /* Create the socket. */
852 #ifndef __ASSUME_SOCK_CLOEXEC
854 if (have_sock_cloexec
>= 0)
857 sock
= socket (AF_UNIX
, SOCK_STREAM
| SOCK_CLOEXEC
| SOCK_NONBLOCK
, 0);
858 #ifndef __ASSUME_SOCK_CLOEXEC
859 if (have_sock_cloexec
== 0)
860 have_sock_cloexec
= sock
!= -1 || errno
!= EINVAL
? 1 : -1;
863 #ifndef __ASSUME_SOCK_CLOEXEC
864 if (have_sock_cloexec
< 0)
865 sock
= socket (AF_UNIX
, SOCK_STREAM
, 0);
869 dbg_log (_("cannot open socket: %s"), strerror (errno
));
870 exit (errno
== EACCES
? 4 : 1);
872 /* Bind a name to the socket. */
873 struct sockaddr_un sock_addr
;
874 sock_addr
.sun_family
= AF_UNIX
;
875 strcpy (sock_addr
.sun_path
, _PATH_NSCDSOCKET
);
876 if (bind (sock
, (struct sockaddr
*) &sock_addr
, sizeof (sock_addr
)) < 0)
878 dbg_log ("%s: %s", _PATH_NSCDSOCKET
, strerror (errno
));
879 exit (errno
== EACCES
? 4 : 1);
882 #ifndef __ASSUME_SOCK_CLOEXEC
883 if (have_sock_cloexec
< 0)
885 /* We don't want to get stuck on accept. */
886 int fl
= fcntl (sock
, F_GETFL
);
887 if (fl
== -1 || fcntl (sock
, F_SETFL
, fl
| O_NONBLOCK
) == -1)
889 dbg_log (_("cannot change socket to nonblocking mode: %s"),
894 /* The descriptor needs to be closed on exec. */
895 if (paranoia
&& fcntl (sock
, F_SETFD
, FD_CLOEXEC
) == -1)
897 dbg_log (_("cannot set socket to close on exec: %s"),
904 /* Set permissions for the socket. */
905 chmod (_PATH_NSCDSOCKET
, DEFFILEMODE
);
907 /* Set the socket up to accept connections. */
908 if (listen (sock
, SOMAXCONN
) < 0)
910 dbg_log (_("cannot enable socket to accept connections: %s"),
916 if (dbs
[hstdb
].enabled
)
918 /* Try to open netlink socket to monitor network setting changes. */
919 nl_status_fd
= socket (AF_NETLINK
,
920 SOCK_RAW
| SOCK_CLOEXEC
| SOCK_NONBLOCK
,
922 if (nl_status_fd
!= -1)
924 struct sockaddr_nl snl
;
925 memset (&snl
, '\0', sizeof (snl
));
926 snl
.nl_family
= AF_NETLINK
;
927 /* XXX Is this the best set to use? */
928 snl
.nl_groups
= (RTMGRP_IPV4_IFADDR
| RTMGRP_TC
| RTMGRP_IPV4_MROUTE
929 | RTMGRP_IPV4_ROUTE
| RTMGRP_IPV4_RULE
930 | RTMGRP_IPV6_IFADDR
| RTMGRP_IPV6_MROUTE
931 | RTMGRP_IPV6_ROUTE
| RTMGRP_IPV6_IFINFO
932 | RTMGRP_IPV6_PREFIX
);
934 if (bind (nl_status_fd
, (struct sockaddr
*) &snl
, sizeof (snl
)) != 0)
936 close (nl_status_fd
);
941 /* Start the timestamp process. */
942 dbs
[hstdb
].head
->extra_data
[NSCD_HST_IDX_CONF_TIMESTAMP
]
943 = __bump_nl_timestamp ();
945 # ifndef __ASSUME_SOCK_CLOEXEC
946 if (have_sock_cloexec
< 0)
948 /* We don't want to get stuck on accept. */
949 int fl
= fcntl (nl_status_fd
, F_GETFL
);
951 || fcntl (nl_status_fd
, F_SETFL
, fl
| O_NONBLOCK
) == -1)
954 cannot change socket to nonblocking mode: %s"),
959 /* The descriptor needs to be closed on exec. */
961 && fcntl (nl_status_fd
, F_SETFD
, FD_CLOEXEC
) == -1)
963 dbg_log (_("cannot set socket to close on exec: %s"),
974 /* Change to unprivileged uid/gid/groups if specified in config file */
975 if (server_user
!= NULL
)
976 finish_drop_privileges ();
980 /* Register the file in FINFO as a traced file for the database DBS[DBIX].
982 We support registering multiple files per database. Each call to
983 register_traced_file adds to the list of registered files.
985 When we prune the database, either through timeout or a request to
986 invalidate, we will check to see if any of the registered files has changed.
987 When we accept new connections to handle a cache request we will also
988 check to see if any of the registered files has changed.
990 If we have inotify support then we install an inotify fd to notify us of
991 file deletion or modification, both of which will require we invalidate
992 the cache for the database. Without inotify support we stat the file and
993 store st_mtime to determine if the file has been modified. */
995 register_traced_file (size_t dbidx
, struct traced_file
*finfo
)
997 /* If the database is disabled or file checking is disabled
998 then ignore the registration. */
999 if (! dbs
[dbidx
].enabled
|| ! dbs
[dbidx
].check_file
)
1002 if (__builtin_expect (debug_level
> 0, 0))
1003 dbg_log (_("register trace file %s for database %s"),
1004 finfo
->fname
, dbnames
[dbidx
]);
1008 || (finfo
->inotify_descr
= inotify_add_watch (inotify_fd
, finfo
->fname
,
1013 /* We need the modification date of the file. */
1016 if (stat64 (finfo
->fname
, &st
) < 0)
1018 /* We cannot stat() the file, disable file checking. */
1019 dbg_log (_("cannot stat() file `%s': %s"),
1020 finfo
->fname
, strerror (errno
));
1024 finfo
->inotify_descr
= -1;
1025 finfo
->mtime
= st
.st_mtime
;
1028 /* Queue up the file name. */
1029 finfo
->next
= dbs
[dbidx
].traced_files
;
1030 dbs
[dbidx
].traced_files
= finfo
;
1034 /* Close the connections. */
1036 close_sockets (void)
1043 invalidate_cache (char *key
, int fd
)
1048 for (number
= pwddb
; number
< lastdb
; ++number
)
1049 if (strcmp (key
, dbnames
[number
]) == 0)
1051 if (number
== hstdb
)
1053 struct traced_file
*runp
= dbs
[hstdb
].traced_files
;
1054 while (runp
!= NULL
)
1055 if (runp
->call_res_init
)
1066 if (number
== lastdb
)
1069 writeall (fd
, &resp
, sizeof (resp
));
1073 if (dbs
[number
].enabled
)
1075 pthread_mutex_lock (&dbs
[number
].prune_run_lock
);
1076 prune_cache (&dbs
[number
], LONG_MAX
, fd
);
1077 pthread_mutex_unlock (&dbs
[number
].prune_run_lock
);
1082 writeall (fd
, &resp
, sizeof (resp
));
1089 send_ro_fd (struct database_dyn
*db
, char *key
, int fd
)
1091 /* If we do not have an read-only file descriptor do nothing. */
1092 if (db
->ro_fd
== -1)
1095 /* We need to send some data along with the descriptor. */
1096 uint64_t mapsize
= (db
->head
->data_size
1097 + roundup (db
->head
->module
* sizeof (ref_t
), ALIGN
)
1098 + sizeof (struct database_pers_head
));
1099 struct iovec iov
[2];
1100 iov
[0].iov_base
= key
;
1101 iov
[0].iov_len
= strlen (key
) + 1;
1102 iov
[1].iov_base
= &mapsize
;
1103 iov
[1].iov_len
= sizeof (mapsize
);
1105 /* Prepare the control message to transfer the descriptor. */
1109 char bytes
[CMSG_SPACE (sizeof (int))];
1111 struct msghdr msg
= { .msg_iov
= iov
, .msg_iovlen
= 2,
1112 .msg_control
= buf
.bytes
,
1113 .msg_controllen
= sizeof (buf
) };
1114 struct cmsghdr
*cmsg
= CMSG_FIRSTHDR (&msg
);
1116 cmsg
->cmsg_level
= SOL_SOCKET
;
1117 cmsg
->cmsg_type
= SCM_RIGHTS
;
1118 cmsg
->cmsg_len
= CMSG_LEN (sizeof (int));
1120 int *ip
= (int *) CMSG_DATA (cmsg
);
1123 msg
.msg_controllen
= cmsg
->cmsg_len
;
1125 /* Send the control message. We repeat when we are interrupted but
1126 everything else is ignored. */
1127 #ifndef MSG_NOSIGNAL
1128 # define MSG_NOSIGNAL 0
1130 (void) TEMP_FAILURE_RETRY (sendmsg (fd
, &msg
, MSG_NOSIGNAL
));
1132 if (__builtin_expect (debug_level
> 0, 0))
1133 dbg_log (_("provide access to FD %d, for %s"), db
->ro_fd
, key
);
1135 #endif /* SCM_RIGHTS */
1138 /* Handle new request. */
1140 handle_request (int fd
, request_header
*req
, void *key
, uid_t uid
, pid_t pid
)
1142 if (__builtin_expect (req
->version
, NSCD_VERSION
) != NSCD_VERSION
)
1144 if (debug_level
> 0)
1146 cannot handle old request version %d; current version is %d"),
1147 req
->version
, NSCD_VERSION
);
1151 /* Perform the SELinux check before we go on to the standard checks. */
1152 if (selinux_enabled
&& nscd_request_avc_has_perm (fd
, req
->type
) != 0)
1154 if (debug_level
> 0)
1163 snprintf (buf
, sizeof (buf
), "/proc/%ld/exe", (long int) pid
);
1164 ssize_t n
= readlink (buf
, buf
, sizeof (buf
) - 1);
1168 request from %ld not handled due to missing permission"), (long int) pid
);
1173 request from '%s' [%ld] not handled due to missing permission"),
1174 buf
, (long int) pid
);
1177 dbg_log (_("request not handled due to missing permission"));
1183 struct database_dyn
*db
= reqinfo
[req
->type
].db
;
1185 /* See whether we can service the request from the cache. */
1186 if (__builtin_expect (reqinfo
[req
->type
].data_request
, true))
1188 if (__builtin_expect (debug_level
, 0) > 0)
1190 if (req
->type
== GETHOSTBYADDR
|| req
->type
== GETHOSTBYADDRv6
)
1192 char buf
[INET6_ADDRSTRLEN
];
1194 dbg_log ("\t%s (%s)", serv2str
[req
->type
],
1195 inet_ntop (req
->type
== GETHOSTBYADDR
1196 ? AF_INET
: AF_INET6
,
1197 key
, buf
, sizeof (buf
)));
1200 dbg_log ("\t%s (%s)", serv2str
[req
->type
], (char *) key
);
1203 /* Is this service enabled? */
1204 if (__builtin_expect (!db
->enabled
, 0))
1206 /* No, sent the prepared record. */
1207 if (TEMP_FAILURE_RETRY (send (fd
, db
->disabled_iov
->iov_base
,
1208 db
->disabled_iov
->iov_len
,
1210 != (ssize_t
) db
->disabled_iov
->iov_len
1211 && __builtin_expect (debug_level
, 0) > 0)
1213 /* We have problems sending the result. */
1215 dbg_log (_("cannot write result: %s"),
1216 strerror_r (errno
, buf
, sizeof (buf
)));
1222 /* Be sure we can read the data. */
1223 if (__builtin_expect (pthread_rwlock_tryrdlock (&db
->lock
) != 0, 0))
1225 ++db
->head
->rdlockdelayed
;
1226 pthread_rwlock_rdlock (&db
->lock
);
1229 /* See whether we can handle it from the cache. */
1230 struct datahead
*cached
;
1231 cached
= (struct datahead
*) cache_search (req
->type
, key
, req
->key_len
,
1235 /* Hurray it's in the cache. */
1238 #ifdef HAVE_SENDFILE
1239 if (__builtin_expect (db
->mmap_used
, 1))
1241 assert (db
->wr_fd
!= -1);
1242 assert ((char *) cached
->data
> (char *) db
->data
);
1243 assert ((char *) cached
->data
- (char *) db
->head
1245 <= (sizeof (struct database_pers_head
)
1246 + db
->head
->module
* sizeof (ref_t
)
1247 + db
->head
->data_size
));
1248 nwritten
= sendfileall (fd
, db
->wr_fd
,
1249 (char *) cached
->data
1250 - (char *) db
->head
, cached
->recsize
);
1251 # ifndef __ASSUME_SENDFILE
1252 if (nwritten
== -1 && errno
== ENOSYS
)
1257 # ifndef __ASSUME_SENDFILE
1261 nwritten
= writeall (fd
, cached
->data
, cached
->recsize
);
1263 if (nwritten
!= cached
->recsize
1264 && __builtin_expect (debug_level
, 0) > 0)
1266 /* We have problems sending the result. */
1268 dbg_log (_("cannot write result: %s"),
1269 strerror_r (errno
, buf
, sizeof (buf
)));
1272 pthread_rwlock_unlock (&db
->lock
);
1277 pthread_rwlock_unlock (&db
->lock
);
1279 else if (__builtin_expect (debug_level
, 0) > 0)
1281 if (req
->type
== INVALIDATE
)
1282 dbg_log ("\t%s (%s)", serv2str
[req
->type
], (char *) key
);
1284 dbg_log ("\t%s", serv2str
[req
->type
]);
1287 /* Handle the request. */
1291 addpwbyname (db
, fd
, req
, key
, uid
);
1295 addpwbyuid (db
, fd
, req
, key
, uid
);
1299 addgrbyname (db
, fd
, req
, key
, uid
);
1303 addgrbygid (db
, fd
, req
, key
, uid
);
1307 addhstbyname (db
, fd
, req
, key
, uid
);
1310 case GETHOSTBYNAMEv6
:
1311 addhstbynamev6 (db
, fd
, req
, key
, uid
);
1315 addhstbyaddr (db
, fd
, req
, key
, uid
);
1318 case GETHOSTBYADDRv6
:
1319 addhstbyaddrv6 (db
, fd
, req
, key
, uid
);
1323 addhstai (db
, fd
, req
, key
, uid
);
1327 addinitgroups (db
, fd
, req
, key
, uid
);
1331 addservbyname (db
, fd
, req
, key
, uid
);
1335 addservbyport (db
, fd
, req
, key
, uid
);
1339 addgetnetgrent (db
, fd
, req
, key
, uid
);
1343 addinnetgr (db
, fd
, req
, key
, uid
);
1350 /* Get the callers credentials. */
1352 struct ucred caller
;
1353 socklen_t optlen
= sizeof (caller
);
1355 if (getsockopt (fd
, SOL_SOCKET
, SO_PEERCRED
, &caller
, &optlen
) < 0)
1359 dbg_log (_("error getting caller's id: %s"),
1360 strerror_r (errno
, buf
, sizeof (buf
)));
1366 /* Some systems have no SO_PEERCRED implementation. They don't
1367 care about security so we don't as well. */
1372 /* Accept shutdown, getstat and invalidate only from root. For
1373 the stat call also allow the user specified in the config file. */
1374 if (req
->type
== GETSTAT
)
1376 if (uid
== 0 || uid
== stat_uid
)
1377 send_stats (fd
, dbs
);
1381 if (req
->type
== INVALIDATE
)
1382 invalidate_cache (key
, fd
);
1384 termination_handler (0);
1394 send_ro_fd (reqinfo
[req
->type
].db
, key
, fd
);
1399 /* Ignore the command, it's nothing we know. */
1405 /* Restart the process. */
1409 /* First determine the parameters. We do not use the parameters
1410 passed to main() since in case nscd is started by running the
1411 dynamic linker this will not work. Yes, this is not the usual
1412 case but nscd is part of glibc and we occasionally do this. */
1413 size_t buflen
= 1024;
1414 char *buf
= alloca (buflen
);
1416 int fd
= open ("/proc/self/cmdline", O_RDONLY
);
1420 cannot open /proc/self/cmdline: %s; disabling paranoia mode"),
1429 ssize_t n
= TEMP_FAILURE_RETRY (read (fd
, buf
+ readlen
,
1434 cannot read /proc/self/cmdline: %s; disabling paranoia mode"),
1444 if (readlen
< buflen
)
1447 /* We might have to extend the buffer. */
1448 size_t old_buflen
= buflen
;
1449 char *newp
= extend_alloca (buf
, buflen
, 2 * buflen
);
1450 buf
= memmove (newp
, buf
, old_buflen
);
1455 /* Parse the command line. Worst case scenario: every two
1456 characters form one parameter (one character plus NUL). */
1457 char **argv
= alloca ((readlen
/ 2 + 1) * sizeof (argv
[0]));
1461 while (cp
< buf
+ readlen
)
1464 cp
= (char *) rawmemchr (cp
, '\0') + 1;
1468 /* Second, change back to the old user if we changed it. */
1469 if (server_user
!= NULL
)
1471 if (setresuid (old_uid
, old_uid
, old_uid
) != 0)
1474 cannot change to old UID: %s; disabling paranoia mode"),
1481 if (setresgid (old_gid
, old_gid
, old_gid
) != 0)
1484 cannot change to old GID: %s; disabling paranoia mode"),
1487 setuid (server_uid
);
1493 /* Next change back to the old working directory. */
1494 if (chdir (oldcwd
) == -1)
1497 cannot change to old working directory: %s; disabling paranoia mode"),
1500 if (server_user
!= NULL
)
1502 setuid (server_uid
);
1503 setgid (server_gid
);
1509 /* Synchronize memory. */
1510 int32_t certainly
[lastdb
];
1511 for (int cnt
= 0; cnt
< lastdb
; ++cnt
)
1512 if (dbs
[cnt
].enabled
)
1514 /* Make sure nobody keeps using the database. */
1515 dbs
[cnt
].head
->timestamp
= 0;
1516 certainly
[cnt
] = dbs
[cnt
].head
->nscd_certainly_running
;
1517 dbs
[cnt
].head
->nscd_certainly_running
= 0;
1519 if (dbs
[cnt
].persistent
)
1521 msync (dbs
[cnt
].head
, dbs
[cnt
].memsize
, MS_ASYNC
);
1524 /* The preparations are done. */
1526 char pathbuf
[PATH_MAX
];
1530 /* Try to exec the real nscd program so the process name (as reported
1531 in /proc/PID/status) will be 'nscd', but fall back to /proc/self/exe
1532 if readlink or the exec with the result of the readlink call fails. */
1533 ssize_t n
= readlink ("/proc/self/exe", pathbuf
, sizeof (pathbuf
) - 1);
1537 execv (pathbuf
, argv
);
1539 execv ("/proc/self/exe", argv
);
1541 /* If we come here, we will never be able to re-exec. */
1542 dbg_log (_("re-exec failed: %s; disabling paranoia mode"),
1545 if (server_user
!= NULL
)
1547 setuid (server_uid
);
1548 setgid (server_gid
);
1550 if (chdir ("/") != 0)
1551 dbg_log (_("cannot change current working directory to \"/\": %s"),
1555 /* Reenable the databases. */
1556 time_t now
= time (NULL
);
1557 for (int cnt
= 0; cnt
< lastdb
; ++cnt
)
1558 if (dbs
[cnt
].enabled
)
1560 dbs
[cnt
].head
->timestamp
= now
;
1561 dbs
[cnt
].head
->nscd_certainly_running
= certainly
[cnt
];
1566 /* List of file descriptors. */
1570 struct fdlist
*next
;
1572 /* Memory allocated for the list. */
1573 static struct fdlist
*fdlist
;
1574 /* List of currently ready-to-read file descriptors. */
1575 static struct fdlist
*readylist
;
1577 /* Conditional variable and mutex to signal availability of entries in
1578 READYLIST. The condvar is initialized dynamically since we might
1579 use a different clock depending on availability. */
1580 static pthread_cond_t readylist_cond
= PTHREAD_COND_INITIALIZER
;
1581 static pthread_mutex_t readylist_lock
= PTHREAD_MUTEX_INITIALIZER
;
1583 /* The clock to use with the condvar. */
1584 static clockid_t timeout_clock
= CLOCK_REALTIME
;
1586 /* Number of threads ready to handle the READYLIST. */
1587 static unsigned long int nready
;
1590 /* Function for the clean-up threads. */
1592 __attribute__ ((__noreturn__
))
1593 nscd_run_prune (void *p
)
1595 const long int my_number
= (long int) p
;
1596 assert (dbs
[my_number
].enabled
);
1598 int dont_need_update
= setup_thread (&dbs
[my_number
]);
1600 time_t now
= time (NULL
);
1602 /* We are running. */
1603 dbs
[my_number
].head
->timestamp
= now
;
1605 struct timespec prune_ts
;
1606 if (__builtin_expect (clock_gettime (timeout_clock
, &prune_ts
) == -1, 0))
1607 /* Should never happen. */
1610 /* Compute the initial timeout time. Prevent all the timers to go
1611 off at the same time by adding a db-based value. */
1612 prune_ts
.tv_sec
+= CACHE_PRUNE_INTERVAL
+ my_number
;
1613 dbs
[my_number
].wakeup_time
= now
+ CACHE_PRUNE_INTERVAL
+ my_number
;
1615 pthread_mutex_t
*prune_lock
= &dbs
[my_number
].prune_lock
;
1616 pthread_mutex_t
*prune_run_lock
= &dbs
[my_number
].prune_run_lock
;
1617 pthread_cond_t
*prune_cond
= &dbs
[my_number
].prune_cond
;
1619 pthread_mutex_lock (prune_lock
);
1622 /* Wait, but not forever. */
1624 if (! dbs
[my_number
].clear_cache
)
1625 e
= pthread_cond_timedwait (prune_cond
, prune_lock
, &prune_ts
);
1626 assert (__builtin_expect (e
== 0 || e
== ETIMEDOUT
, 1));
1630 if (e
== ETIMEDOUT
|| now
>= dbs
[my_number
].wakeup_time
1631 || dbs
[my_number
].clear_cache
)
1633 /* We will determine the new timout values based on the
1634 cache content. Should there be concurrent additions to
1635 the cache which are not accounted for in the cache
1636 pruning we want to know about it. Therefore set the
1637 timeout to the maximum. It will be descreased when adding
1638 new entries to the cache, if necessary. */
1639 dbs
[my_number
].wakeup_time
= MAX_TIMEOUT_VALUE
;
1641 /* Unconditionally reset the flag. */
1642 time_t prune_now
= dbs
[my_number
].clear_cache
? LONG_MAX
: now
;
1643 dbs
[my_number
].clear_cache
= 0;
1645 pthread_mutex_unlock (prune_lock
);
1647 /* We use a separate lock for running the prune function (instead
1648 of keeping prune_lock locked) because this enables concurrent
1649 invocations of cache_add which might modify the timeout value. */
1650 pthread_mutex_lock (prune_run_lock
);
1651 next_wait
= prune_cache (&dbs
[my_number
], prune_now
, -1);
1652 pthread_mutex_unlock (prune_run_lock
);
1654 next_wait
= MAX (next_wait
, CACHE_PRUNE_INTERVAL
);
1655 /* If clients cannot determine for sure whether nscd is running
1656 we need to wake up occasionally to update the timestamp.
1657 Wait 90% of the update period. */
1658 #define UPDATE_MAPPING_TIMEOUT (MAPPING_TIMEOUT * 9 / 10)
1659 if (__builtin_expect (! dont_need_update
, 0))
1661 next_wait
= MIN (UPDATE_MAPPING_TIMEOUT
, next_wait
);
1662 dbs
[my_number
].head
->timestamp
= now
;
1665 pthread_mutex_lock (prune_lock
);
1667 /* Make it known when we will wake up again. */
1668 if (now
+ next_wait
< dbs
[my_number
].wakeup_time
)
1669 dbs
[my_number
].wakeup_time
= now
+ next_wait
;
1671 next_wait
= dbs
[my_number
].wakeup_time
- now
;
1674 /* The cache was just pruned. Do not do it again now. Just
1675 use the new timeout value. */
1676 next_wait
= dbs
[my_number
].wakeup_time
- now
;
1678 if (clock_gettime (timeout_clock
, &prune_ts
) == -1)
1679 /* Should never happen. */
1682 /* Compute next timeout time. */
1683 prune_ts
.tv_sec
+= next_wait
;
1688 /* This is the main loop. It is replicated in different threads but
1689 the use of the ready list makes sure only one thread handles an
1690 incoming connection. */
1692 __attribute__ ((__noreturn__
))
1693 nscd_run_worker (void *p
)
1697 /* Initial locking. */
1698 pthread_mutex_lock (&readylist_lock
);
1700 /* One more thread available. */
1705 while (readylist
== NULL
)
1706 pthread_cond_wait (&readylist_cond
, &readylist_lock
);
1708 struct fdlist
*it
= readylist
->next
;
1709 if (readylist
->next
== readylist
)
1710 /* Just one entry on the list. */
1713 readylist
->next
= it
->next
;
1715 /* Extract the information and mark the record ready to be used
1720 /* One more thread available. */
1723 /* We are done with the list. */
1724 pthread_mutex_unlock (&readylist_lock
);
1726 #ifndef __ASSUME_ACCEPT4
1727 if (have_accept4
< 0)
1729 /* We do not want to block on a short read or so. */
1730 int fl
= fcntl (fd
, F_GETFL
);
1731 if (fl
== -1 || fcntl (fd
, F_SETFL
, fl
| O_NONBLOCK
) == -1)
1736 /* Now read the request. */
1738 if (__builtin_expect (TEMP_FAILURE_RETRY (read (fd
, &req
, sizeof (req
)))
1739 != sizeof (req
), 0))
1741 /* We failed to read data. Note that this also might mean we
1742 failed because we would have blocked. */
1743 if (debug_level
> 0)
1744 dbg_log (_("short read while reading request: %s"),
1745 strerror_r (errno
, buf
, sizeof (buf
)));
1749 /* Check whether this is a valid request type. */
1750 if (req
.type
< GETPWBYNAME
|| req
.type
>= LASTREQ
)
1753 /* Some systems have no SO_PEERCRED implementation. They don't
1754 care about security so we don't as well. */
1759 if (__builtin_expect (debug_level
> 0, 0))
1761 struct ucred caller
;
1762 socklen_t optlen
= sizeof (caller
);
1764 if (getsockopt (fd
, SOL_SOCKET
, SO_PEERCRED
, &caller
, &optlen
) == 0)
1768 const pid_t pid
= 0;
1771 /* It should not be possible to crash the nscd with a silly
1772 request (i.e., a terribly large key). We limit the size to 1kb. */
1773 if (__builtin_expect (req
.key_len
, 1) < 0
1774 || __builtin_expect (req
.key_len
, 1) > MAXKEYLEN
)
1776 if (debug_level
> 0)
1777 dbg_log (_("key length in request too long: %d"), req
.key_len
);
1782 char keybuf
[MAXKEYLEN
+ 1];
1784 if (__builtin_expect (TEMP_FAILURE_RETRY (read (fd
, keybuf
,
1788 /* Again, this can also mean we would have blocked. */
1789 if (debug_level
> 0)
1790 dbg_log (_("short read while reading request key: %s"),
1791 strerror_r (errno
, buf
, sizeof (buf
)));
1794 keybuf
[req
.key_len
] = '\0';
1796 if (__builtin_expect (debug_level
, 0) > 0)
1801 handle_request: request received (Version = %d) from PID %ld"),
1802 req
.version
, (long int) pid
);
1806 handle_request: request received (Version = %d)"), req
.version
);
1809 /* Phew, we got all the data, now process it. */
1810 handle_request (fd
, &req
, keybuf
, uid
, pid
);
1818 pthread_mutex_lock (&readylist_lock
);
1820 /* One more thread available. */
1827 static unsigned int nconns
;
1832 pthread_mutex_lock (&readylist_lock
);
1834 /* Find an empty entry in FDLIST. */
1836 for (inner
= 0; inner
< nconns
; ++inner
)
1837 if (fdlist
[inner
].next
== NULL
)
1839 assert (inner
< nconns
);
1841 fdlist
[inner
].fd
= fd
;
1843 if (readylist
== NULL
)
1844 readylist
= fdlist
[inner
].next
= &fdlist
[inner
];
1847 fdlist
[inner
].next
= readylist
->next
;
1848 readylist
= readylist
->next
= &fdlist
[inner
];
1851 bool do_signal
= true;
1852 if (__builtin_expect (nready
== 0, 0))
1857 /* Try to start another thread to help out. */
1859 if (nthreads
< max_nthreads
1860 && pthread_create (&th
, &attr
, nscd_run_worker
,
1861 (void *) (long int) nthreads
) == 0)
1863 /* We got another thread. */
1865 /* The new thread might need a kick. */
1871 pthread_mutex_unlock (&readylist_lock
);
1873 /* Tell one of the worker threads there is work to do. */
1875 pthread_cond_signal (&readylist_cond
);
1879 /* Check whether restarting should happen. */
1881 restart_p (time_t now
)
1883 return (paranoia
&& readylist
== NULL
&& nready
== nthreads
1884 && now
>= restart_time
);
1888 /* Array for times a connection was accepted. */
1889 static time_t *starttime
;
1892 /* Inotify event for changed file. */
1895 struct inotify_event i
;
1897 # define PATH_MAX 1024
1899 char buf
[sizeof (struct inotify_event
) + PATH_MAX
];
1902 /* Process the inotify event in INEV. If the event matches any of the files
1903 registered with a database then mark that database as requiring its cache
1904 to be cleared. We indicate the cache needs clearing by setting
1905 TO_CLEAR[DBCNT] to true for the matching database. */
1907 inotify_check_files (bool *to_clear
, union __inev
*inev
)
1909 /* Check which of the files changed. */
1910 for (size_t dbcnt
= 0; dbcnt
< lastdb
; ++dbcnt
)
1912 struct traced_file
*finfo
= dbs
[dbcnt
].traced_files
;
1914 while (finfo
!= NULL
)
1916 /* Inotify event watch descriptor matches. */
1917 if (finfo
->inotify_descr
== inev
->i
.wd
)
1919 /* Mark cache as needing to be cleared and reinitialize. */
1920 to_clear
[dbcnt
] = true;
1921 if (finfo
->call_res_init
)
1926 finfo
= finfo
->next
;
1931 /* If an entry in the array of booleans TO_CLEAR is TRUE then clear the cache
1932 for the associated database, otherwise do nothing. The TO_CLEAR array must
1933 have LASTDB entries. */
1935 clear_db_cache (bool *to_clear
)
1937 for (size_t dbcnt
= 0; dbcnt
< lastdb
; ++dbcnt
)
1938 if (to_clear
[dbcnt
])
1940 pthread_mutex_lock (&dbs
[dbcnt
].prune_lock
);
1941 dbs
[dbcnt
].clear_cache
= 1;
1942 pthread_mutex_unlock (&dbs
[dbcnt
].prune_lock
);
1943 pthread_cond_signal (&dbs
[dbcnt
].prune_cond
);
1950 __attribute__ ((__noreturn__
))
1951 main_loop_poll (void)
1953 struct pollfd
*conns
= (struct pollfd
*) xmalloc (nconns
1954 * sizeof (conns
[0]));
1957 conns
[0].events
= POLLRDNORM
;
1959 size_t firstfree
= 1;
1962 if (inotify_fd
!= -1)
1964 conns
[1].fd
= inotify_fd
;
1965 conns
[1].events
= POLLRDNORM
;
1972 size_t idx_nl_status_fd
= 0;
1973 if (nl_status_fd
!= -1)
1975 idx_nl_status_fd
= nused
;
1976 conns
[nused
].fd
= nl_status_fd
;
1977 conns
[nused
].events
= POLLRDNORM
;
1985 /* Wait for any event. We wait at most a couple of seconds so
1986 that we can check whether we should close any of the accepted
1987 connections since we have not received a request. */
1988 #define MAX_ACCEPT_TIMEOUT 30
1989 #define MIN_ACCEPT_TIMEOUT 5
1990 #define MAIN_THREAD_TIMEOUT \
1991 (MAX_ACCEPT_TIMEOUT * 1000 \
1992 - ((MAX_ACCEPT_TIMEOUT - MIN_ACCEPT_TIMEOUT) * 1000 * nused) / (2 * nconns))
1994 int n
= poll (conns
, nused
, MAIN_THREAD_TIMEOUT
);
1996 time_t now
= time (NULL
);
1998 /* If there is a descriptor ready for reading or there is a new
1999 connection, process this now. */
2002 if (conns
[0].revents
!= 0)
2004 /* We have a new incoming connection. Accept the connection. */
2007 #ifndef __ASSUME_ACCEPT4
2009 if (have_accept4
>= 0)
2012 fd
= TEMP_FAILURE_RETRY (accept4 (sock
, NULL
, NULL
,
2014 #ifndef __ASSUME_ACCEPT4
2015 if (have_accept4
== 0)
2016 have_accept4
= fd
!= -1 || errno
!= ENOSYS
? 1 : -1;
2019 #ifndef __ASSUME_ACCEPT4
2020 if (have_accept4
< 0)
2021 fd
= TEMP_FAILURE_RETRY (accept (sock
, NULL
, NULL
));
2024 /* Use the descriptor if we have not reached the limit. */
2027 if (firstfree
< nconns
)
2029 conns
[firstfree
].fd
= fd
;
2030 conns
[firstfree
].events
= POLLRDNORM
;
2031 starttime
[firstfree
] = now
;
2032 if (firstfree
>= nused
)
2033 nused
= firstfree
+ 1;
2037 while (firstfree
< nused
&& conns
[firstfree
].fd
!= -1);
2040 /* We cannot use the connection so close it. */
2049 if (inotify_fd
!= -1 && conns
[1].fd
== inotify_fd
)
2051 if (conns
[1].revents
!= 0)
2053 bool to_clear
[lastdb
] = { false, };
2056 /* Read all inotify events for files registered via
2057 register_traced_file(). */
2060 ssize_t nb
= TEMP_FAILURE_RETRY (read (inotify_fd
, &inev
,
2062 if (nb
< (ssize_t
) sizeof (struct inotify_event
))
2064 if (__builtin_expect (nb
== -1 && errno
!= EAGAIN
,
2067 /* Something went wrong when reading the inotify
2068 data. Better disable inotify. */
2070 disabled inotify after read error %d"),
2082 /* Check which of the files changed. */
2083 inotify_check_files (to_clear
, &inev
);
2086 /* Actually perform the cache clearing. */
2087 clear_db_cache (to_clear
);
2097 if (idx_nl_status_fd
!= 0 && conns
[idx_nl_status_fd
].revents
!= 0)
2100 /* Read all the data. We do not interpret it here. */
2101 while (TEMP_FAILURE_RETRY (read (nl_status_fd
, buf
,
2102 sizeof (buf
))) != -1)
2105 dbs
[hstdb
].head
->extra_data
[NSCD_HST_IDX_CONF_TIMESTAMP
]
2106 = __bump_nl_timestamp ();
2110 for (size_t cnt
= first
; cnt
< nused
&& n
> 0; ++cnt
)
2111 if (conns
[cnt
].revents
!= 0)
2113 fd_ready (conns
[cnt
].fd
);
2115 /* Clean up the CONNS array. */
2117 if (cnt
< firstfree
)
2119 if (cnt
== nused
- 1)
2122 while (conns
[nused
- 1].fd
== -1);
2128 /* Now find entries which have timed out. */
2131 /* We make the timeout length depend on the number of file
2132 descriptors currently used. */
2133 #define ACCEPT_TIMEOUT \
2134 (MAX_ACCEPT_TIMEOUT \
2135 - ((MAX_ACCEPT_TIMEOUT - MIN_ACCEPT_TIMEOUT) * nused) / nconns)
2136 time_t laststart
= now
- ACCEPT_TIMEOUT
;
2138 for (size_t cnt
= nused
- 1; cnt
> 0; --cnt
)
2140 if (conns
[cnt
].fd
!= -1 && starttime
[cnt
] < laststart
)
2142 /* Remove the entry, it timed out. */
2143 (void) close (conns
[cnt
].fd
);
2146 if (cnt
< firstfree
)
2148 if (cnt
== nused
- 1)
2151 while (conns
[nused
- 1].fd
== -1);
2155 if (restart_p (now
))
2163 main_loop_epoll (int efd
)
2165 struct epoll_event ev
= { 0, };
2169 /* Add the socket. */
2170 ev
.events
= EPOLLRDNORM
;
2172 if (epoll_ctl (efd
, EPOLL_CTL_ADD
, sock
, &ev
) == -1)
2173 /* We cannot use epoll. */
2176 # ifdef HAVE_INOTIFY
2177 if (inotify_fd
!= -1)
2179 ev
.events
= EPOLLRDNORM
;
2180 ev
.data
.fd
= inotify_fd
;
2181 if (epoll_ctl (efd
, EPOLL_CTL_ADD
, inotify_fd
, &ev
) == -1)
2182 /* We cannot use epoll. */
2188 # ifdef HAVE_NETLINK
2189 if (nl_status_fd
!= -1)
2191 ev
.events
= EPOLLRDNORM
;
2192 ev
.data
.fd
= nl_status_fd
;
2193 if (epoll_ctl (efd
, EPOLL_CTL_ADD
, nl_status_fd
, &ev
) == -1)
2194 /* We cannot use epoll. */
2201 struct epoll_event revs
[100];
2202 # define nrevs (sizeof (revs) / sizeof (revs[0]))
2204 int n
= epoll_wait (efd
, revs
, nrevs
, MAIN_THREAD_TIMEOUT
);
2206 time_t now
= time (NULL
);
2208 for (int cnt
= 0; cnt
< n
; ++cnt
)
2209 if (revs
[cnt
].data
.fd
== sock
)
2211 /* A new connection. */
2214 # ifndef __ASSUME_ACCEPT4
2216 if (have_accept4
>= 0)
2219 fd
= TEMP_FAILURE_RETRY (accept4 (sock
, NULL
, NULL
,
2221 # ifndef __ASSUME_ACCEPT4
2222 if (have_accept4
== 0)
2223 have_accept4
= fd
!= -1 || errno
!= ENOSYS
? 1 : -1;
2226 # ifndef __ASSUME_ACCEPT4
2227 if (have_accept4
< 0)
2228 fd
= TEMP_FAILURE_RETRY (accept (sock
, NULL
, NULL
));
2231 /* Use the descriptor if we have not reached the limit. */
2234 /* Try to add the new descriptor. */
2237 || epoll_ctl (efd
, EPOLL_CTL_ADD
, fd
, &ev
) == -1)
2238 /* The descriptor is too large or something went
2239 wrong. Close the descriptor. */
2243 /* Remember when we accepted the connection. */
2244 starttime
[fd
] = now
;
2253 # ifdef HAVE_INOTIFY
2254 else if (revs
[cnt
].data
.fd
== inotify_fd
)
2256 bool to_clear
[lastdb
] = { false, };
2259 /* Read all inotify events for files registered via
2260 register_traced_file(). */
2263 ssize_t nb
= TEMP_FAILURE_RETRY (read (inotify_fd
, &inev
,
2265 if (nb
< (ssize_t
) sizeof (struct inotify_event
))
2267 if (__builtin_expect (nb
== -1 && errno
!= EAGAIN
, 0))
2269 /* Something went wrong when reading the inotify
2270 data. Better disable inotify. */
2271 dbg_log (_("disabled inotify after read error %d"),
2273 (void) epoll_ctl (efd
, EPOLL_CTL_DEL
, inotify_fd
,
2281 /* Check which of the files changed. */
2282 inotify_check_files(to_clear
, &inev
);
2285 /* Actually perform the cache clearing. */
2286 clear_db_cache (to_clear
);
2289 # ifdef HAVE_NETLINK
2290 else if (revs
[cnt
].data
.fd
== nl_status_fd
)
2293 /* Read all the data. We do not interpret it here. */
2294 while (TEMP_FAILURE_RETRY (read (nl_status_fd
, buf
,
2295 sizeof (buf
))) != -1)
2298 __bump_nl_timestamp ();
2303 /* Remove the descriptor from the epoll descriptor. */
2304 (void) epoll_ctl (efd
, EPOLL_CTL_DEL
, revs
[cnt
].data
.fd
, NULL
);
2306 /* Get a worker to handle the request. */
2307 fd_ready (revs
[cnt
].data
.fd
);
2309 /* Reset the time. */
2310 starttime
[revs
[cnt
].data
.fd
] = 0;
2311 if (revs
[cnt
].data
.fd
== highest
)
2314 while (highest
> 0 && starttime
[highest
] == 0);
2319 /* Now look for descriptors for accepted connections which have
2320 no reply in too long of a time. */
2321 time_t laststart
= now
- ACCEPT_TIMEOUT
;
2322 assert (starttime
[sock
] == 0);
2323 assert (inotify_fd
== -1 || starttime
[inotify_fd
] == 0);
2324 assert (nl_status_fd
== -1 || starttime
[nl_status_fd
] == 0);
2325 for (int cnt
= highest
; cnt
> STDERR_FILENO
; --cnt
)
2326 if (starttime
[cnt
] != 0 && starttime
[cnt
] < laststart
)
2328 /* We are waiting for this one for too long. Close it. */
2329 (void) epoll_ctl (efd
, EPOLL_CTL_DEL
, cnt
, NULL
);
2337 else if (cnt
!= sock
&& starttime
[cnt
] == 0 && cnt
== highest
)
2340 if (restart_p (now
))
2347 /* Start all the threads we want. The initial process is thread no. 1. */
2349 start_threads (void)
2351 /* Initialize the conditional variable we will use. The only
2352 non-standard attribute we might use is the clock selection. */
2353 pthread_condattr_t condattr
;
2354 pthread_condattr_init (&condattr
);
2356 #if defined _POSIX_CLOCK_SELECTION && _POSIX_CLOCK_SELECTION >= 0 \
2357 && defined _POSIX_MONOTONIC_CLOCK && _POSIX_MONOTONIC_CLOCK >= 0
2358 /* Determine whether the monotonous clock is available. */
2359 struct timespec dummy
;
2360 # if _POSIX_MONOTONIC_CLOCK == 0
2361 if (sysconf (_SC_MONOTONIC_CLOCK
) > 0)
2363 # if _POSIX_CLOCK_SELECTION == 0
2364 if (sysconf (_SC_CLOCK_SELECTION
) > 0)
2366 if (clock_getres (CLOCK_MONOTONIC
, &dummy
) == 0
2367 && pthread_condattr_setclock (&condattr
, CLOCK_MONOTONIC
) == 0)
2368 timeout_clock
= CLOCK_MONOTONIC
;
2371 /* Create the attribute for the threads. They are all created
2373 pthread_attr_init (&attr
);
2374 pthread_attr_setdetachstate (&attr
, PTHREAD_CREATE_DETACHED
);
2375 /* Use 1MB stacks, twice as much for 64-bit architectures. */
2376 pthread_attr_setstacksize (&attr
, NSCD_THREAD_STACKSIZE
);
2378 /* We allow less than LASTDB threads only for debugging. */
2379 if (debug_level
== 0)
2380 nthreads
= MAX (nthreads
, lastdb
);
2382 /* Create the threads which prune the databases. */
2383 // XXX Ideally this work would be done by some of the worker threads.
2384 // XXX But this is problematic since we would need to be able to wake
2385 // XXX them up explicitly as well as part of the group handling the
2386 // XXX ready-list. This requires an operation where we can wait on
2387 // XXX two conditional variables at the same time. This operation
2388 // XXX does not exist (yet).
2389 for (long int i
= 0; i
< lastdb
; ++i
)
2391 /* Initialize the conditional variable. */
2392 if (pthread_cond_init (&dbs
[i
].prune_cond
, &condattr
) != 0)
2394 dbg_log (_("could not initialize conditional variable"));
2400 && pthread_create (&th
, &attr
, nscd_run_prune
, (void *) i
) != 0)
2402 dbg_log (_("could not start clean-up thread; terminating"));
2407 pthread_condattr_destroy (&condattr
);
2409 for (long int i
= 0; i
< nthreads
; ++i
)
2412 if (pthread_create (&th
, &attr
, nscd_run_worker
, NULL
) != 0)
2416 dbg_log (_("could not start any worker thread; terminating"));
2424 /* Determine how much room for descriptors we should initially
2425 allocate. This might need to change later if we cap the number
2427 const long int nfds
= sysconf (_SC_OPEN_MAX
);
2429 #define MAXCONN 16384
2430 if (nfds
== -1 || nfds
> MAXCONN
)
2432 else if (nfds
< MINCONN
)
2437 /* We need memory to pass descriptors on to the worker threads. */
2438 fdlist
= (struct fdlist
*) xcalloc (nconns
, sizeof (fdlist
[0]));
2439 /* Array to keep track when connection was accepted. */
2440 starttime
= (time_t *) xcalloc (nconns
, sizeof (starttime
[0]));
2442 /* In the main thread we execute the loop which handles incoming
2445 int efd
= epoll_create (100);
2448 main_loop_epoll (efd
);
2457 /* Look up the uid, gid, and supplementary groups to run nscd as. When
2458 this function is called, we are not listening on the nscd socket yet so
2459 we can just use the ordinary lookup functions without causing a lockup */
2461 begin_drop_privileges (void)
2463 struct passwd
*pwd
= getpwnam (server_user
);
2467 dbg_log (_("Failed to run nscd as user '%s'"), server_user
);
2468 error (EXIT_FAILURE
, 0, _("Failed to run nscd as user '%s'"),
2472 server_uid
= pwd
->pw_uid
;
2473 server_gid
= pwd
->pw_gid
;
2475 /* Save the old UID/GID if we have to change back. */
2478 old_uid
= getuid ();
2479 old_gid
= getgid ();
2482 if (getgrouplist (server_user
, server_gid
, NULL
, &server_ngroups
) == 0)
2484 /* This really must never happen. */
2485 dbg_log (_("Failed to run nscd as user '%s'"), server_user
);
2486 error (EXIT_FAILURE
, errno
, _("initial getgrouplist failed"));
2489 server_groups
= (gid_t
*) xmalloc (server_ngroups
* sizeof (gid_t
));
2491 if (getgrouplist (server_user
, server_gid
, server_groups
, &server_ngroups
)
2494 dbg_log (_("Failed to run nscd as user '%s'"), server_user
);
2495 error (EXIT_FAILURE
, errno
, _("getgrouplist failed"));
2500 /* Call setgroups(), setgid(), and setuid() to drop root privileges and
2501 run nscd as the user specified in the configuration file. */
2503 finish_drop_privileges (void)
2505 #if defined HAVE_LIBAUDIT && defined HAVE_LIBCAP
2506 /* We need to preserve the capabilities to connect to the audit daemon. */
2507 cap_t new_caps
= preserve_capabilities ();
2510 if (setgroups (server_ngroups
, server_groups
) == -1)
2512 dbg_log (_("Failed to run nscd as user '%s'"), server_user
);
2513 error (EXIT_FAILURE
, errno
, _("setgroups failed"));
2518 res
= setresgid (server_gid
, server_gid
, old_gid
);
2520 res
= setgid (server_gid
);
2523 dbg_log (_("Failed to run nscd as user '%s'"), server_user
);
2529 res
= setresuid (server_uid
, server_uid
, old_uid
);
2531 res
= setuid (server_uid
);
2534 dbg_log (_("Failed to run nscd as user '%s'"), server_user
);
2539 #if defined HAVE_LIBAUDIT && defined HAVE_LIBCAP
2540 /* Remove the temporary capabilities. */
2541 install_real_capabilities (new_caps
);