]>
Commit | Line | Data |
---|---|---|
1 | /* $Id$ | |
2 | * | |
3 | * Moira server process. | |
4 | * | |
5 | * Copyright (C) 1987-1998 by the Massachusetts Institute of Technology | |
6 | * For copying and distribution information, please see the file | |
7 | * <mit-copyright.h>. | |
8 | * | |
9 | */ | |
10 | ||
11 | #include <mit-copyright.h> | |
12 | #include "mr_server.h" | |
13 | ||
14 | #include <sys/socket.h> | |
15 | #include <sys/stat.h> | |
16 | #include <sys/types.h> | |
17 | #include <sys/utsname.h> | |
18 | #include <sys/wait.h> | |
19 | ||
20 | #include <netinet/in.h> | |
21 | #include <arpa/inet.h> | |
22 | ||
23 | #include <ctype.h> | |
24 | #include <errno.h> | |
25 | #include <signal.h> | |
26 | #include <stdio.h> | |
27 | #include <stdlib.h> | |
28 | #include <string.h> | |
29 | #include <unistd.h> | |
30 | ||
31 | #ifdef HAVE_KRB4 | |
32 | #include <krb.h> | |
33 | #endif | |
34 | #include <krb5.h> | |
35 | ||
36 | RCSID("$Header$"); | |
37 | ||
38 | client *cur_client; | |
39 | ||
40 | char *whoami; | |
41 | char *takedown; | |
42 | FILE *journal; | |
43 | ||
44 | time_t now; | |
45 | ||
46 | char *host; | |
47 | krb5_context context = NULL; | |
48 | char *krb_realm = NULL; | |
49 | ||
50 | /* Client array and associated data. This needs to be global for _list_users */ | |
51 | client **clients; | |
52 | int nclients, clientssize; | |
53 | ||
54 | int dormant; | |
55 | int child_exited_abnormally = 0; | |
56 | int child_pid, child_signal, child_status; | |
57 | ||
58 | void reapchild(int x); | |
59 | void godormant(int x); | |
60 | void gowakeup(int x); | |
61 | void clist_append(client *cp); | |
62 | void mr_setup_signals(void); | |
63 | ||
64 | /* | |
65 | * Main Moira server loop. | |
66 | * | |
67 | * Initialize the world, then start accepting connections and | |
68 | * making progress on current connections. | |
69 | */ | |
70 | ||
71 | int main(int argc, char **argv) | |
72 | { | |
73 | int status, i, listener; | |
74 | time_t tardy; | |
75 | char *port, *p; | |
76 | extern char *database; | |
77 | struct stat stbuf; | |
78 | struct utsname uts; | |
79 | fd_set readfds, writefds, xreadfds, xwritefds; | |
80 | int nfds, counter = 0; | |
81 | ||
82 | whoami = argv[0]; | |
83 | /* | |
84 | * Error handler init. | |
85 | */ | |
86 | mr_init(); | |
87 | set_com_err_hook(mr_com_err); | |
88 | setvbuf(stderr, NULL, _IOLBF, BUFSIZ); | |
89 | ||
90 | port = strchr(MOIRA_SERVER, ':') + 1; | |
91 | ||
92 | for (i = 1; i < argc; i++) | |
93 | { | |
94 | if (!strcmp(argv[i], "-db") && i + 1 < argc) | |
95 | { | |
96 | database = argv[i + 1]; | |
97 | i++; | |
98 | } | |
99 | else if (!strcmp(argv[i], "-p") && i + 1 < argc) | |
100 | { | |
101 | port = argv[i + 1]; | |
102 | i++; | |
103 | } | |
104 | else | |
105 | { | |
106 | com_err(whoami, 0, "Usage: moirad [-db database][-p port]"); | |
107 | exit(1); | |
108 | } | |
109 | } | |
110 | ||
111 | status = krb5_init_context(&context); | |
112 | if (status) | |
113 | { | |
114 | com_err(whoami, status, "Initializing krb5 context."); | |
115 | exit(1); | |
116 | } | |
117 | ||
118 | status = krb5_get_default_realm(context, &krb_realm); | |
119 | if (status) | |
120 | { | |
121 | com_err(whoami, status, "Getting default Kerberos realm."); | |
122 | exit(1); | |
123 | } | |
124 | ||
125 | /* | |
126 | * Database initialization. Only init if database should be open. | |
127 | */ | |
128 | ||
129 | if (stat(MOIRA_MOTD_FILE, &stbuf) != 0) | |
130 | { | |
131 | if ((status = mr_open_database())) | |
132 | { | |
133 | com_err(whoami, status, "trying to open database."); | |
134 | exit(1); | |
135 | } | |
136 | sanity_check_database(); | |
137 | } | |
138 | else | |
139 | { | |
140 | dormant = ASLEEP; | |
141 | com_err(whoami, 0, "sleeping, not opening database"); | |
142 | } | |
143 | ||
144 | sanity_check_queries(); | |
145 | ||
146 | /* | |
147 | * Get moira server hostname for authentication | |
148 | */ | |
149 | if (uname(&uts) < 0) | |
150 | { | |
151 | com_err(whoami, errno, "Unable to get local hostname"); | |
152 | exit(1); | |
153 | } | |
154 | host = canonicalize_hostname(xstrdup(uts.nodename)); | |
155 | for (p = host; *p && *p != '.'; p++) | |
156 | { | |
157 | if (isupper(*p)) | |
158 | *p = tolower(*p); | |
159 | } | |
160 | *p = '\0'; | |
161 | ||
162 | /* | |
163 | * Set up client array handler. | |
164 | */ | |
165 | nclients = 0; | |
166 | clientssize = 10; | |
167 | clients = xmalloc(clientssize * sizeof(client *)); | |
168 | ||
169 | mr_setup_signals(); | |
170 | ||
171 | journal = fopen(JOURNAL, "a"); | |
172 | if (!journal) | |
173 | { | |
174 | com_err(whoami, errno, "opening journal file"); | |
175 | exit(1); | |
176 | } | |
177 | ||
178 | /* | |
179 | * Establish template connection. | |
180 | */ | |
181 | listener = mr_listen(port); | |
182 | if (listener == -1) | |
183 | { | |
184 | com_err(whoami, MR_ABORTED, "trying to create listening connection"); | |
185 | exit(1); | |
186 | } | |
187 | FD_ZERO(&xreadfds); | |
188 | FD_ZERO(&xwritefds); | |
189 | FD_SET(listener, &xreadfds); | |
190 | nfds = listener + 1; | |
191 | ||
192 | com_err(whoami, 0, "started (pid %d)", getpid()); | |
193 | com_err(whoami, 0, rcsid); | |
194 | if (dormant != ASLEEP) | |
195 | send_zgram("MOIRA", "server started"); | |
196 | else | |
197 | send_zgram("MOIRA", "server started, but database closed"); | |
198 | ||
199 | /* | |
200 | * Run until shut down. | |
201 | */ | |
202 | while (!takedown) | |
203 | { | |
204 | int i; | |
205 | struct timeval timeout = {60, 0}; /* 1 minute */ | |
206 | ||
207 | /* If we're supposed to go down and we can, do it */ | |
208 | if (((dormant == AWAKE) && (nclients == 0) && | |
209 | (stat(MOIRA_MOTD_FILE, &stbuf) == 0)) || | |
210 | (dormant == SLEEPY)) | |
211 | { | |
212 | mr_close_database(); | |
213 | com_err(whoami, 0, "database closed"); | |
214 | mr_setup_signals(); | |
215 | send_zgram("MOIRA", "database closed"); | |
216 | dormant = ASLEEP; | |
217 | } | |
218 | ||
219 | /* Block until something happens. */ | |
220 | memcpy(&readfds, &xreadfds, sizeof(readfds)); | |
221 | memcpy(&writefds, &xwritefds, sizeof(writefds)); | |
222 | if (select(nfds, &readfds, &writefds, NULL, &timeout) == -1) | |
223 | { | |
224 | if (errno != EINTR) | |
225 | com_err(whoami, errno, "in select"); | |
226 | if (!inc_running || now - inc_started > INC_TIMEOUT) | |
227 | next_incremental(); | |
228 | continue; | |
229 | } | |
230 | ||
231 | if (takedown) | |
232 | break; | |
233 | ||
234 | if (child_exited_abnormally) | |
235 | { | |
236 | critical_alert(whoami, "moirad", "%d: child exits with signal %d status %d", | |
237 | child_pid, child_signal, child_status); | |
238 | child_exited_abnormally = 0; | |
239 | } | |
240 | ||
241 | time(&now); | |
242 | if (!inc_running || now - inc_started > INC_TIMEOUT) | |
243 | next_incremental(); | |
244 | tardy = now - 30 * 60; | |
245 | ||
246 | /* If we're asleep and we should wake up, do it */ | |
247 | if ((dormant == ASLEEP) && (stat(MOIRA_MOTD_FILE, &stbuf) == -1) && | |
248 | (errno == ENOENT)) | |
249 | { | |
250 | mr_open_database(); | |
251 | com_err(whoami, 0, "database open"); | |
252 | mr_setup_signals(); | |
253 | send_zgram("MOIRA", "database open again"); | |
254 | dormant = AWAKE; | |
255 | } | |
256 | ||
257 | /* Handle any new connections */ | |
258 | if (FD_ISSET(listener, &readfds)) | |
259 | { | |
260 | int newconn, addrlen = sizeof(struct sockaddr_in); | |
261 | struct sockaddr_in addr; | |
262 | client *cp; | |
263 | ||
264 | newconn = accept(listener, (struct sockaddr *)&addr, &addrlen); | |
265 | if (newconn == -1) | |
266 | com_err(whoami, errno, "accepting new connection"); | |
267 | else if (newconn > 0) | |
268 | { | |
269 | if (newconn + 1 > nfds) | |
270 | nfds = newconn + 1; | |
271 | FD_SET(newconn, &xreadfds); | |
272 | ||
273 | /* Add a new client to the array */ | |
274 | nclients++; | |
275 | if (nclients > clientssize) | |
276 | { | |
277 | clientssize = 2 * clientssize; | |
278 | clients = xrealloc(clients, clientssize * sizeof(client *)); | |
279 | } | |
280 | ||
281 | clients[nclients - 1] = cp = xmalloc(sizeof(client)); | |
282 | memset(cp, 0, sizeof(client)); | |
283 | cp->con = newconn; | |
284 | cp->id = counter++; | |
285 | cp->last_time_used = now; | |
286 | cp->haddr = addr; | |
287 | cp->tuplessize = 1; | |
288 | cp->tuples = xmalloc(sizeof(mr_params)); | |
289 | memset(cp->tuples, 0, sizeof(mr_params)); | |
290 | cp->state = CL_ACCEPTING; | |
291 | cp->version = 2; | |
292 | ||
293 | cur_client = cp; | |
294 | com_err(whoami, 0, | |
295 | "New connection from %s port %d (now %d client%s)", | |
296 | inet_ntoa(cp->haddr.sin_addr), | |
297 | (int)ntohs(cp->haddr.sin_port), | |
298 | nclients, nclients != 1 ? "s" : ""); | |
299 | } | |
300 | } | |
301 | ||
302 | /* Handle any existing connections. */ | |
303 | for (i = 0; i < nclients; i++) | |
304 | { | |
305 | cur_client = clients[i]; | |
306 | ||
307 | if (FD_ISSET(clients[i]->con, &writefds)) | |
308 | { | |
309 | client_write(clients[i]); | |
310 | if (!clients[i]->ntuples) | |
311 | { | |
312 | FD_CLR(clients[i]->con, &xwritefds); | |
313 | FD_SET(clients[i]->con, &xreadfds); | |
314 | } | |
315 | clients[i]->last_time_used = now; | |
316 | } | |
317 | ||
318 | if (FD_ISSET(clients[i]->con, &readfds)) | |
319 | { | |
320 | if (clients[i]->state == CL_ACCEPTING) | |
321 | { | |
322 | switch(mr_cont_accept(clients[i]->con, | |
323 | &clients[i]->hsbuf, | |
324 | &clients[i]->hslen)) | |
325 | { | |
326 | case -1: | |
327 | break; | |
328 | ||
329 | case 0: | |
330 | clients[i]->state = CL_CLOSING; | |
331 | break; | |
332 | ||
333 | default: | |
334 | clients[i]->state = CL_ACTIVE; | |
335 | clients[i]->hsbuf = NULL; | |
336 | break; | |
337 | } | |
338 | } | |
339 | else | |
340 | { | |
341 | client_read(clients[i]); | |
342 | if (clients[i]->ntuples) | |
343 | { | |
344 | FD_CLR(clients[i]->con, &xreadfds); | |
345 | FD_SET(clients[i]->con, &xwritefds); | |
346 | } | |
347 | clients[i]->last_time_used = now; | |
348 | } | |
349 | } | |
350 | ||
351 | if (clients[i]->last_time_used < tardy) | |
352 | { | |
353 | com_err(whoami, 0, "Shutting down connection due to inactivity"); | |
354 | clients[i]->state = CL_CLOSING; | |
355 | } | |
356 | ||
357 | if (clients[i]->state == CL_CLOSING) | |
358 | { | |
359 | client *old; | |
360 | ||
361 | com_err(whoami, 0, "Closed connection (now %d client%s, " | |
362 | "%d queries)", nclients - 1, nclients != 2 ? "s" : "", | |
363 | newqueries); | |
364 | ||
365 | shutdown(clients[i]->con, 2); | |
366 | close(clients[i]->con); | |
367 | FD_CLR(clients[i]->con, &xreadfds); | |
368 | FD_CLR(clients[i]->con, &xwritefds); | |
369 | free_rtn_tuples(clients[i]); | |
370 | free(clients[i]->tuples); | |
371 | free(clients[i]->hsbuf); | |
372 | old = clients[i]; | |
373 | clients[i] = clients[--nclients]; | |
374 | free(old); | |
375 | } | |
376 | ||
377 | cur_client = NULL; | |
378 | if (takedown) | |
379 | break; | |
380 | } | |
381 | } | |
382 | ||
383 | com_err(whoami, 0, "%s", takedown); | |
384 | if (dormant != ASLEEP) | |
385 | mr_close_database(); | |
386 | send_zgram("MOIRA", takedown); | |
387 | return 0; | |
388 | } | |
389 | ||
390 | void reapchild(int x) | |
391 | { | |
392 | int status, pid; | |
393 | ||
394 | while ((pid = waitpid(-1, &status, WNOHANG)) > 0) | |
395 | { | |
396 | if (pid == inc_pid) | |
397 | inc_running = 0; | |
398 | if (!takedown && (WTERMSIG(status) != 0 || WEXITSTATUS(status) != 0)) | |
399 | { | |
400 | child_exited_abnormally = 1; | |
401 | child_pid = pid; | |
402 | child_signal = WTERMSIG(status); | |
403 | child_status = WEXITSTATUS(status); | |
404 | } | |
405 | } | |
406 | } | |
407 | ||
408 | ||
409 | void godormant(int x) | |
410 | { | |
411 | switch (dormant) | |
412 | { | |
413 | case AWAKE: | |
414 | case GROGGY: | |
415 | com_err(whoami, 0, "requested to go dormant"); | |
416 | break; | |
417 | case ASLEEP: | |
418 | com_err(whoami, 0, "already asleep"); | |
419 | break; | |
420 | case SLEEPY: | |
421 | break; | |
422 | } | |
423 | dormant = SLEEPY; | |
424 | } | |
425 | ||
426 | ||
427 | void gowakeup(int x) | |
428 | { | |
429 | switch (dormant) | |
430 | { | |
431 | case ASLEEP: | |
432 | case SLEEPY: | |
433 | com_err(whoami, 0, "Good morning"); | |
434 | break; | |
435 | case AWAKE: | |
436 | com_err(whoami, 0, "already awake"); | |
437 | break; | |
438 | case GROGGY: | |
439 | break; | |
440 | } | |
441 | dormant = GROGGY; | |
442 | } | |
443 | ||
444 | ||
445 | void mr_setup_signals(void) | |
446 | { | |
447 | struct sigaction action; | |
448 | ||
449 | action.sa_flags = 0; | |
450 | sigemptyset(&action.sa_mask); | |
451 | ||
452 | /* There should probably be a few more of these. */ | |
453 | ||
454 | action.sa_handler = sigshut; | |
455 | if ((sigaction(SIGTERM, &action, NULL) < 0) || | |
456 | (sigaction(SIGINT, &action, NULL) < 0) || | |
457 | (sigaction(SIGHUP, &action, NULL) < 0)) | |
458 | { | |
459 | com_err(whoami, errno, "Unable to establish signal handlers."); | |
460 | exit(1); | |
461 | } | |
462 | ||
463 | action.sa_handler = godormant; | |
464 | if (sigaction(SIGUSR1, &action, NULL) < 0) | |
465 | { | |
466 | com_err(whoami, errno, "Unable to establish signal handlers."); | |
467 | exit(1); | |
468 | } | |
469 | ||
470 | action.sa_handler = gowakeup; | |
471 | if (sigaction(SIGUSR2, &action, NULL) < 0) | |
472 | { | |
473 | com_err(whoami, errno, "Unable to establish signal handlers."); | |
474 | exit(1); | |
475 | } | |
476 | ||
477 | action.sa_handler = SIG_IGN; | |
478 | if (sigaction(SIGPIPE, &action, NULL) < 0) | |
479 | { | |
480 | com_err(whoami, errno, "Unable to establish signal handlers."); | |
481 | exit(1); | |
482 | } | |
483 | ||
484 | action.sa_handler = reapchild; | |
485 | sigaddset(&action.sa_mask, SIGCHLD); | |
486 | if (sigaction(SIGCHLD, &action, NULL) < 0) | |
487 | { | |
488 | com_err(whoami, errno, "Unable to establish signal handlers."); | |
489 | exit(1); | |
490 | } | |
491 | } |