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