Compile nscd if have-thread-library. Rename nscd-routines to
[kopensolaris-gnu/glibc.git] / nscd / nscd.c
1 /* Copyright (c) 1998 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3    Contributed by Thorsten Kukuk <kukuk@vt.uni-paderborn.de>, 1998.
4
5    The GNU C Library is free software; you can redistribute it and/or
6    modify it under the terms of the GNU Library General Public License as
7    published by the Free Software Foundation; either version 2 of the
8    License, or (at your option) any later version.
9
10    The GNU C Library is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13    Library General Public License for more details.
14
15    You should have received a copy of the GNU Library General Public
16    License along with the GNU C Library; see the file COPYING.LIB.  If not,
17    write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18    Boston, MA 02111-1307, USA. */
19
20 /* nscd - Name Service Cache Daemon. Caches passwd and group.  */
21
22 #include <argp.h>
23 #include <errno.h>
24 #include <error.h>
25 #include <libintl.h>
26 #include <locale.h>
27 #include <pthread.h>
28 #include <pwd.h>
29 #include <signal.h>
30 #include <stdio.h>
31 #include <stdlib.h>
32 #include <string.h>
33 #include <syslog.h>
34 #include <sys/socket.h>
35 #include <sys/un.h>
36
37 #include "dbg_log.h"
38 #include "nscd.h"
39
40 /* Get libc version number.  */
41 #include <version.h>
42
43 #define PACKAGE _libc_intl_domainname
44
45 /* Structure used by main() thread to keep track of the number of
46    active threads.  Used to limit how many threads it will create
47    and under a shutdown condition to wait till all in-progress
48    requests have finished before "turning off the lights".  */
49
50 typedef struct
51 {
52   int             num_active;
53   pthread_cond_t  thread_exit_cv;
54   pthread_mutex_t mutex;
55 } thread_info_t;
56
57 thread_info_t thread_info;
58
59 int do_shutdown = 0;
60 int disabled_passwd = 0;
61 int disabled_group = 0;
62 int go_background = 1;
63 const char *conffile = _PATH_NSCDCONF;
64
65 static void termination_handler (int signum);
66 static int check_pid (const char *file);
67 static int write_pid (const char *file);
68 static void handle_requests (void);
69
70 /* Name and version of program.  */
71 static void print_version (FILE *stream, struct argp_state *state);
72 void (*argp_program_version_hook) (FILE *, struct argp_state *) = print_version;
73
74 /* Definitions of arguments for argp functions.  */
75 static const struct argp_option options[] =
76 {
77   { "config-file", 'f', N_("NAME"), 0,
78     N_("Read configuration data from NAME") },
79   { "debug", 'd', NULL, 0,
80     N_("Do not fork and display messages on the current tty") },
81   { "shutdown", 'K', NULL, 0, N_("Shut the server down") },
82   { NULL, 0, NULL, 0, NULL }
83 };
84
85 /* Short description of program.  */
86 static const char doc[] = N_("Name Switch Cache Daemon.");
87
88 /* Prototype for option handler.  */
89 static error_t parse_opt __P ((int key, char *arg, struct argp_state *state));
90
91 /* Data structure to communicate with argp functions.  */
92 static struct argp argp =
93 {
94   options, parse_opt, NULL, doc,
95 };
96
97 int
98 main (int argc, char **argv)
99 {
100   int remaining;
101
102   /* Set locale via LC_ALL.  */
103   setlocale (LC_ALL, "");
104   /* Set the text message domain.  */
105   textdomain (PACKAGE);
106
107   /* Parse and process arguments.  */
108   argp_parse (&argp, argc, argv, 0, &remaining, NULL);
109
110   if (remaining != argc)
111     {
112       error (0, 0, gettext ("wrong number of arguments"));
113       argp_help (&argp, stdout, ARGP_HELP_SEE, program_invocation_short_name);
114       exit (EXIT_FAILURE);
115     }
116
117   signal (SIGINT, termination_handler);
118   signal (SIGQUIT, termination_handler);
119   signal (SIGTERM, termination_handler);
120   signal (SIGPIPE, SIG_IGN);
121
122   /* Check if we are already running. */
123   if (check_pid (_PATH_NSCDPID))
124     {
125       fputs (_("already running"), stderr);
126       exit (EXIT_FAILURE);
127     }
128
129   /* Behave like a daemon.  */
130   if (go_background)
131     {
132       openlog ("nscd", LOG_CONS | LOG_ODELAY, LOG_DAEMON);
133
134       if (daemon (0, 0) < 0)
135         {
136           fprintf (stderr, _("connot auto-background: %s\n"),
137                    strerror (errno));
138           exit (EXIT_FAILURE);
139         }
140       if (write_pid (_PATH_NSCDPID) < 0)
141         dbg_log ("%s: %s", _PATH_NSCDPID, strerror (errno));
142
143       /* Ignore job control signals */
144       signal (SIGTTOU, SIG_IGN);
145       signal (SIGTTIN, SIG_IGN);
146       signal (SIGTSTP, SIG_IGN);
147     }
148   /* Cleanup files created by a previous `bind' */
149   unlink (_PATH_NSCDSOCKET);
150
151   nscd_parse_file (conffile);
152
153   /* Create first sockets */
154   init_sockets ();
155   /* Init databases */
156   if ((cache_pwdinit () < 0) || (cache_grpinit () < 0))
157     {
158       fputs (_("Not enough memory\n"), stderr);
159       return 1;
160     }
161   /* Handle incoming requests */
162   handle_requests ();
163
164   return 0;
165 }
166
167
168 /* Handle program arguments.  */
169 static error_t
170 parse_opt (int key, char *arg, struct argp_state *state)
171 {
172   switch (key)
173     {
174     case 'd':
175       debug_flag = 1;
176       go_background = 0;
177       break;
178     case 'f':
179       conffile = arg;
180       break;
181     case 'K':
182       if (getuid () != 0)
183         {
184           printf (_("Only root is allowed to use this option!\n\n"));
185           exit (EXIT_FAILURE);
186         }
187       {
188         int sock = __nscd_open_socket ();
189         request_header req;
190         ssize_t nbytes;
191
192         if (sock == -1)
193           exit (EXIT_FAILURE);
194
195         req.version = NSCD_VERSION;
196         req.type = SHUTDOWN;
197         req.key_len = 0;
198         nbytes = write (sock, &req, sizeof (request_header));
199         close (sock);
200         if (nbytes != req.key_len)
201           exit (EXIT_FAILURE);
202         else
203           exit (EXIT_SUCCESS);
204       }
205     case 'g':
206       print_stat ();
207       exit (EXIT_SUCCESS);
208     default:
209       return ARGP_ERR_UNKNOWN;
210     }
211   return 0;
212 }
213
214 /* Print the version information.  */
215 static void
216 print_version (FILE *stream, struct argp_state *state)
217 {
218   fprintf (stream, "nscd (GNU %s) %s\n", PACKAGE, VERSION);
219   fprintf (stream, gettext ("\
220 Copyright (C) %s Free Software Foundation, Inc.\n\
221 This is free software; see the source for copying conditions.  There is NO\n\
222 warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n\
223 "), "1998");
224   fprintf (stream, gettext ("Written by %s.\n"), "Thorsten Kukuk");
225 }
226
227
228 /* Create a socket connected to a name. */
229 int
230 __nscd_open_socket (void)
231 {
232   struct sockaddr_un addr;
233   int sock;
234
235   sock = socket (PF_UNIX, SOCK_STREAM, 0);
236   if (sock < 0)
237     return -1;
238
239   addr.sun_family = AF_UNIX;
240   strcpy (addr.sun_path, _PATH_NSCDSOCKET);
241   if (connect (sock, (struct sockaddr *) &addr, sizeof (addr)) < 0)
242     {
243       close (sock);
244       return -1;
245     }
246
247   return sock;
248 }
249
250 /* Cleanup.  */
251 static void
252 termination_handler (int signum)
253 {
254   close_sockets ();
255
256   /* Clean up the files created by `bind'.  */
257   unlink (_PATH_NSCDSOCKET);
258
259   /* Clean up pid file.  */
260   unlink (_PATH_NSCDPID);
261
262   exit (EXIT_SUCCESS);
263 }
264
265 /* Returns 1 if the process in pid file FILE is running, 0 if not.  */
266 static int
267 check_pid (const char *file)
268 {
269   FILE *fp;
270
271   fp = fopen (file, "r");
272   if (fp)
273     {
274       pid_t pid;
275
276       fscanf (fp, "%d", &pid);
277       fclose (fp);
278
279       if (kill (pid, 0) == 0)
280         return 1;
281     }
282
283   return 0;
284 }
285
286 /* Write the current process id to the file FILE.
287    Returns 0 if successful, -1 if not.  */
288 static int
289 write_pid (const char *file)
290 {
291   FILE *fp;
292
293   fp = fopen (file, "w");
294   if (fp == NULL)
295     return -1;
296
297   fprintf (fp, "%d\n", getpid ());
298   if (ferror (fp))
299     return -1;
300
301   fclose (fp);
302
303   return 0;
304 }
305
306 /* Type of the lookup function for netname2user.  */
307 typedef int (*pwbyname_function) (const char *name, struct passwd *pw,
308                                    char *buffer, size_t buflen);
309
310 /* Hanlde incoming requests.  */
311 static
312 void handle_requests (void)
313 {
314   request_header req;
315   int conn; /* Handle on which connection (client) the request came from.  */
316   int done = 0;
317   char *key;
318
319   while (!done)
320     {
321       key = NULL;
322       get_request (&conn, &req, &key);
323       if (debug_flag)
324         dbg_log (_("handle_requests: request received (Version = %d)"),
325                  req.version);
326       switch (req.type)
327         {
328         case GETPWBYNAME:
329           {
330             param_t *param = malloc (sizeof (param_t));
331             pthread_t thread;
332
333             if (debug_flag)
334               dbg_log ("\tGETPWBYNAME (%s)", key);
335             param->key = key;
336             param->conn = conn;
337             if (disabled_passwd)
338               pthread_create (&thread, NULL, cache_pw_disabled, (void *)param);
339             else
340               pthread_create (&thread, NULL, cache_getpwnam, (void *)param);
341             pthread_detach (thread);
342           }
343           break;
344         case GETPWBYUID:
345           {
346             param_t *param = malloc (sizeof (param_t));
347             pthread_t thread;
348
349             if (debug_flag)
350               dbg_log ("\tGETPWBYUID (%s)", key);
351             param->key = key;
352             param->conn = conn;
353             if (disabled_passwd)
354               pthread_create (&thread, NULL, cache_pw_disabled, (void *)param);
355             else
356               pthread_create (&thread, NULL, cache_getpwuid, (void *)param);
357             pthread_detach (thread);
358           }
359           break;
360         case GETGRBYNAME:
361           {
362             param_t *param = malloc (sizeof (param_t));
363             pthread_t thread;
364
365             if (debug_flag)
366               dbg_log ("\tGETGRBYNAME (%s)", key);
367             param->key = key;
368             param->conn = conn;
369             if (disabled_group)
370               pthread_create (&thread, NULL, cache_gr_disabled, (void *)param);
371             else
372               pthread_create (&thread, NULL, cache_getgrnam, (void *)param);
373             pthread_detach (thread);
374           }
375           break;
376         case GETGRBYGID:
377           {
378             param_t *param = malloc (sizeof (param_t));
379             pthread_t thread;
380
381             if (debug_flag)
382               dbg_log ("\tGETGRBYGID (%s)", key);
383             param->key = key;
384             param->conn = conn;
385             if (disabled_group)
386               pthread_create (&thread, NULL, cache_gr_disabled, (void *)param);
387             else
388               pthread_create (&thread, NULL, cache_getgrgid, (void *)param);
389             pthread_detach (thread);
390           }
391           break;
392         case GETHOSTBYNAME:
393           /* Not yetimplemented.  */
394           close_socket (conn);
395           break;
396         case GETHOSTBYADDR:
397           /* Not yet implemented. */
398           close_socket (conn);
399           break;
400         case SHUTDOWN:
401           do_shutdown = 1;
402           close_socket (0);
403           close_socket (conn);
404           /* Clean up the files created by `bind'.  */
405           unlink (_PATH_NSCDSOCKET);
406           /* Clean up pid file.  */
407           unlink (_PATH_NSCDPID);
408           done = 1;
409           break;
410         case GETSTAT:
411           {
412             stat_response_header resp;
413
414             if (debug_flag)
415               dbg_log ("\tGETSTAT");
416
417             get_pw_stat (&resp);
418             get_gr_stat (&resp);
419             resp.debug_level = debug_flag;
420             resp.pw_enabled = !disabled_passwd;
421             resp.gr_enabled = !disabled_group;
422
423             stat_send (conn, &resp);
424
425             close_socket (conn);
426           }
427           break;
428         default:
429           dbg_log (_("Unknown request (%d)"), req.type);
430           break;
431         }
432     }
433 }