ramkomd.c 18.3 KB
Newer Older
Linus Tolke Y's avatar
Linus Tolke Y committed
1
/*
2 3
 * $Id: ramkomd.c,v 0.133 2005/12/18 22:18:14 ceder Exp $
 * Copyright (C) 1991-1999, 2001-2005  Lysator Academic Computer Association.
Linus Tolke Y's avatar
Linus Tolke Y committed
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
 *
 * This file is part of the LysKOM server.
 * 
 * LysKOM is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by 
 * the Free Software Foundation; either version 1, or (at your option) 
 * any later version.
 * 
 * LysKOM is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with LysKOM; see the file COPYING.  If not, write to
 * Lysator, c/o ISY, Linkoping University, S-581 83 Linkoping, SWEDEN,
 * or the Free Software Foundation, Inc., 675 Mass Ave, Cambridge, 
 * MA 02139, USA.
 *
Per Cederqvist's avatar
Per Cederqvist committed
23
 * Please report bugs at http://bugzilla.lysator.liu.se/. 
Linus Tolke Y's avatar
Linus Tolke Y committed
24
 */
25 26 27 28 29 30 31
/*
 * The next comment block is a historic comment, written in Swedish
 * using ISO 646 since nobody in Lysator knew about ISO 8859-1 by
 * then.  It translates, rougly, to "This is the main program of the
 * server.  It will hopefully be bigger than it is at the moment.
 * Created by Willfr 31-mar-1990."
 */
Per Cederqvist's avatar
Per Cederqvist committed
32 33 34 35 36 37 38 39 40
/*
 * Detta {r serverns huvudprogram. Det kommer f|rhoppningsvis bli st|rre
 * {n det {r just nu...
 *
 * Created by Willf|r 31/3-90
 *
 * It has grown! /ceder
 */

David Byers's avatar
David Byers committed
41 42 43 44 45

#ifdef HAVE_CONFIG_H
#  include <config.h>
#endif

46 47 48
#ifdef HAVE_LOCALE_H
#  include <locale.h>
#endif
Per Cederqvist's avatar
Per Cederqvist committed
49
#include <stdio.h>
Per Cederqvist's avatar
Per Cederqvist committed
50
#include <signal.h>
51 52 53 54 55 56
#ifdef HAVE_STDLIB_H
#  include <stdlib.h>
#endif
#ifdef HAVE_STRING_H
#  include <string.h>
#endif
Per Cederqvist's avatar
Per Cederqvist committed
57
#include <sys/wait.h>
58
#include "timewrap.h"
59 60 61
#ifdef HAVE_SYS_RESOURCE_H
#  include <sys/resource.h>
#endif
Per Cederqvist's avatar
Per Cederqvist committed
62
#include <unistd.h>
Per Cederqvist's avatar
Per Cederqvist committed
63 64 65 66
#include <setjmp.h>
#if defined(HAVE_SYS_PARAM_H) && !defined(HAVE_GETCWD)
# include <sys/param.h>
#endif
Per Cederqvist's avatar
Per Cederqvist committed
67
#include <errno.h>
Per Cederqvist's avatar
Per Cederqvist committed
68
#include <sys/types.h>
69
#include <sys/socket.h>
Per Cederqvist's avatar
Per Cederqvist committed
70
#include <fcntl.h>
71
#include <assert.h>
Per Cederqvist's avatar
Per Cederqvist committed
72

73 74
#include "oop.h"

Per Cederqvist's avatar
Per Cederqvist committed
75
#include "exp.h"
76
#include "s-string.h"
Per Cederqvist's avatar
Per Cederqvist committed
77
#include "misc-types.h"
78
#include "kom-types.h"
Per Cederqvist's avatar
Per Cederqvist committed
79
#include "com.h"
80
#include "async.h"
Per Cederqvist's avatar
Per Cederqvist committed
81 82 83
#include "connections.h"
#include "internal-connections.h"
#include "kom-errno.h"
84
#include "oop-malloc.h"
Per Cederqvist's avatar
Per Cederqvist committed
85
#include "isc-malloc.h"
86
#include "isc-interface.h"
David Byers's avatar
David Byers committed
87
#include "kom-config.h"
Per Cederqvist's avatar
Per Cederqvist committed
88
#include "cache.h"
Per Cederqvist's avatar
Per Cederqvist committed
89 90
#include "string-malloc.h"
#include "lyskomd.h"
Per Cederqvist's avatar
Per Cederqvist committed
91
#include "log.h"
Per Cederqvist's avatar
Per Cederqvist committed
92 93
#include "server/smalloc.h"
#include "kom-memory.h"
94 95
#include "conf-file.h"
#include "param.h"
96 97
#include "server-config.h"
#include "manipulate.h"
98
#include "version-info.h"
99
#include "aux-items.h"
David Byers's avatar
David Byers committed
100
#include "admin.h"
101
#include "unused.h"
Per Cederqvist's avatar
Per Cederqvist committed
102
#include "sigflags.h"
Per Cederqvist's avatar
Per Cederqvist committed
103
#include "local-to-global.h"
104
#include "server-time.h"
105
#include "lockdb.h"
Per Cederqvist's avatar
Per Cederqvist committed
106
#include "linkansi.h"
107 108 109
#ifdef TRACED_ALLOCATIONS
#  include "trace-alloc.h"
#endif
110
#include "eintr.h"
111
#include "stats.h"
112

113 114 115 116
#if defined(HAVE_SETRLIMIT) && defined(RLIMIT_OFILE) && !defined(RLIMIT_NOFILE)
#  define RLIMIT_NOFILE RLIMIT_OFILE
#endif

117 118 119 120
#if !HAVE_RLIM_T
typedef int rlim_t;
#endif

121
struct timeval current_time = { 0, 0 };
122 123


Per Cederqvist's avatar
Per Cederqvist committed
124
#ifndef NDEBUG
125
int	buglevel = 0;
126
#endif
Per Cederqvist's avatar
Per Cederqvist committed
127

128 129 130 131 132
/* Don't place lyskomd in the background.  Write the log to stdout.
   Prompt for input on stdin before exiting.  This is primarily
   intended for debugging and the test suite. */
static int foreground = 0;

133 134
static oop_adapter_signal *kom_signal_adapter;

135
static void dump_exit_statistics(void);
136
static oop_call_signal sighandler_term;
137 138 139 140
static oop_call_signal sighandler_quit;
static oop_call_signal sighandler_usr1;
static oop_call_signal sighandler_usr2;
static oop_call_signal sighandler_winch;
141 142 143
/* handle_accept_event is declared in connections.h.  We redeclare it
   once more here, just to make sure that its signature matches the
   isc_accept_callback typedef. */
144
isc_accept_callback handle_accept_event;
145

146

147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163
#ifdef HAVE_STRUCT_SIGACTION

/* Assigning SIG_IGN to sa_handler results in

    ramkomd.c:310: warning: function declaration isn't a prototype

  on certain compilers.  By performing the assignment in a function
  the number of warnings are reduced. */

static inline void
set_handler_sig_ign(struct sigaction *ptr)
{
    ptr->sa_handler = SIG_IGN;
}

#endif

164 165 166 167 168 169 170 171 172
static isc_write_queue_change_cb write_queue_change_callback;

static void
write_queue_change_callback(int delta)
{
    update_stat(STAT_SEND_QUEUE, delta);
}


Per Cederqvist's avatar
Per Cederqvist committed
173
static void
174
start_listen_sockets(void)
Per Cederqvist's avatar
Per Cederqvist committed
175
{
176
    struct isc_scb *listen_client;
177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201
    int i;

    for (i = 0; i < param.listen.size; ++i)
    {
	struct ipport_entry *port = &param.listen.entries[i];

	listen_client = isc_listentcp(kom_server_mcb,
				      port->ipaddr,
				      port->port,
				      handle_accept_event);
	if (listen_client == NULL)
	    restart_kom("server_init: can't isc_listentcp(%s, %s)\n",
			port->ipaddr ? port->ipaddr : "*", port->port);
	assert(listen_client->laddr != NULL);
	kom_log("Listening for clients on %s:%d.\n",
		isc_getipnum(listen_client->laddr, NULL, 0),
		isc_getportnum(listen_client->laddr));
    }
}



static void
server_init(void)
{
202
    oop_source *src;
203 204 205 206
#ifdef HAVE_STRUCT_SIGACTION
    struct sigaction act;
#endif

207 208 209 210 211 212 213
    oop_malloc = &oop_malloc_wrapper;
    oop_realloc = &oop_realloc_wrapper;
    oop_free = &oop_free_wrapper;

    kom_server_oop_src = oop_sys_new();
    if (kom_server_oop_src == NULL)
	restart_kom("server_init: can't get system event source\n");
214 215 216 217 218 219 220 221 222

    kom_signal_adapter = oop_signal_new(oop_sys_source(kom_server_oop_src));
    if (kom_signal_adapter == NULL)
	restart_kom("server_init: can't create signal adapter\n");

    /* Ignore the signals before we register them with liboop.  That
       way, when liboop reinstalls the old signal handler during
       shutdown, we will ignore them rather than stop prematurely
       during the shutdown.  This is especially important for the
223
       SIGTERM signal, which is sent periodically by updateLysKOM. */
224 225 226 227 228
#ifdef HAVE_STRUCT_SIGACTION
    sigemptyset(&act.sa_mask);
    act.sa_flags = 0;
    set_handler_sig_ign(&act);
    sigaction(SIGHUP, &act, NULL);
229 230
    sigaction(SIGTERM, &act, NULL);
    sigaction(SIGINT, &act, NULL);
231 232 233 234 235 236
    sigaction(SIGQUIT, &act, NULL);
    sigaction(SIGUSR1, &act, NULL);
    sigaction(SIGUSR2, &act, NULL);
    sigaction(SIGWINCH, &act, NULL);
#else
    signal(SIGHUP, SIG_IGN);
237 238
    signal(SIGTERM, SIG_IGN);
    signal(SIGINT, SIG_IGN);
239 240 241 242 243 244 245 246
    signal(SIGQUIT, SIG_IGN);
    signal(SIGUSR1, SIG_IGN);
    signal(SIGUSR2, SIG_IGN);
    signal(SIGWINCH, SIG_IGN);
#endif

    src = oop_signal_source(kom_signal_adapter);
    oop_signal_use_sa_restart();
247 248 249
    src->on_signal(src, SIGHUP, sighandler_term, NULL);
    src->on_signal(src, SIGTERM, sighandler_term, NULL);
    src->on_signal(src, SIGINT, sighandler_term, NULL);
250 251 252 253 254
    src->on_signal(src, SIGQUIT, sighandler_quit, NULL);
    src->on_signal(src, SIGUSR1, sighandler_usr1, NULL);
    src->on_signal(src, SIGUSR2, sighandler_usr2, NULL);
    src->on_signal(src, SIGWINCH, sighandler_winch, NULL);

Per Cederqvist's avatar
Per Cederqvist committed
255 256 257
    /*
    ** Setup some parameters here
    */
258 259 260 261
    isc_setallocfn(&isc_malloc_wrapper,
		   &isc_realloc_wrapper,
		   &isc_free_wrapper);

262 263
    kom_server_mcb  = isc_initialize(oop_sys_source(kom_server_oop_src),
				     write_queue_change_callback);
Per Cederqvist's avatar
Per Cederqvist committed
264
    if ( kom_server_mcb == NULL )
Per Cederqvist's avatar
Per Cederqvist committed
265
	restart_kom("server_init: can't isc_initialize()\n");
266
    isc_cfg_fd_relocate(kom_server_mcb, PROTECTED_FDS);
267 268
    isc_cfg_stale_timeout(kom_server_mcb, param.stale_timeout,
			  param.connect_timeout);
269 270 271 272 273
    isc_cfg_queue_size(kom_server_mcb,
		       param.maxqueuedsize_bytes,
		       param.maxmsgsize,
		       param.maxqueuedsize,
		       param.maxdequeuelen);
Per Cederqvist's avatar
Per Cederqvist committed
274

Per Cederqvist's avatar
Per Cederqvist committed
275

276
    start_listen_sockets();
Per Cederqvist's avatar
Per Cederqvist committed
277 278 279 280 281
    /*
     * Ignore SIGPIPE, which the server gets if it tries to write to a
     * socket and the client has died. The server will anyhow handle
     * this situation correct.
     */
282 283 284
#ifdef HAVE_STRUCT_SIGACTION
    sigemptyset(&act.sa_mask);
    act.sa_flags = 0;
285
    set_handler_sig_ign(&act);
286
    sigaction(SIGPIPE, &act, NULL);
287
#else
Per Cederqvist's avatar
Per Cederqvist committed
288
    signal(SIGPIPE, SIG_IGN);
289
#endif
Per Cederqvist's avatar
Per Cederqvist committed
290 291
}

292

Per Cederqvist's avatar
Per Cederqvist committed
293
static void
294
init_data_base(void)
Per Cederqvist's avatar
Per Cederqvist committed
295
{
David Byers's avatar
David Byers committed
296 297 298
    kom_log("Database = %s\n",  param.datafile_name);
    kom_log("Backup = %s\n", param.backupfile_name);
    kom_log("2nd Backup = %s\n", param.backupfile_name_2);
299
    kom_log("Lock File = %s\n", param.lockfile_name);
Per Cederqvist's avatar
Per Cederqvist committed
300 301
    
    if ( init_cache() == FAILURE )
302
	restart_kom ("Cannot find database.\n");
Per Cederqvist's avatar
Per Cederqvist committed
303 304
}

305
static void *
306 307
sighandler_term(oop_source *UNUSED(source),
	       int sig,
308
	       void *UNUSED(user))
Per Cederqvist's avatar
Per Cederqvist committed
309
{
310 311 312 313 314 315 316 317 318 319 320 321
    if (sig == SIGTERM)
	kom_log("Signal TERM received. Shutting down server.\n");
    else if (sig == SIGHUP)
	kom_log("Signal HUP received. Shutting down server."
		" Please use SIGTERM instead.\n");
    else if (sig == SIGINT)
	kom_log("Signal INT received. Shutting down server."
		" Please use SIGTERM instead.\n");
    else
	kom_log("Some signal received. Shutting down server."
		" Please use SIGTERM instead.\n");

322
    go_and_die = TRUE;
323
    return OOP_HALT;
Per Cederqvist's avatar
Per Cederqvist committed
324 325
}

326 327 328 329
static void *
sighandler_quit(oop_source *UNUSED(source),
		int UNUSED(sig),
		void *UNUSED(user))
Per Cederqvist's avatar
Per Cederqvist committed
330
{
David Byers's avatar
David Byers committed
331
    kom_log ("Signal QUIT received - syncing...\n");
Per Cederqvist's avatar
Per Cederqvist committed
332
    cache_sync_all();
David Byers's avatar
David Byers committed
333
    kom_log ("Dumping core now.\n");
Per Cederqvist's avatar
Per Cederqvist committed
334 335 336
    abort();
}

337 338 339 340
static void *
sighandler_usr1(oop_source *UNUSED(source),
		int UNUSED(sig),
		void *UNUSED(user))
Per Cederqvist's avatar
Per Cederqvist committed
341
{
342 343
    dump_statistics();
    return OOP_CONTINUE;
Per Cederqvist's avatar
Per Cederqvist committed
344 345
}

346 347 348 349
static void *
sighandler_usr2(oop_source *UNUSED(source),
		int UNUSED(sig),
		void *UNUSED(user))
Per Cederqvist's avatar
Per Cederqvist committed
350 351 352
{
    int		  child;

David Byers's avatar
David Byers committed
353
    kom_log ("Signal USR2 received - will dump core now.  (Check that child dies.)\n");
Per Cederqvist's avatar
Per Cederqvist committed
354 355 356
    if ((child = fork()) == 0)
    {
	abort();
357
	kom_log ("Abort() failed!!!\n");
Per Cederqvist's avatar
Per Cederqvist committed
358 359 360 361
	exit(1);
    }
    else if (child < 0)
    {
David Byers's avatar
David Byers committed
362
	kom_log ("Couldn't fork.\n");
Per Cederqvist's avatar
Per Cederqvist committed
363 364 365 366 367
    }
    else
    {
	wait (NULL);
    }
368
    return OOP_CONTINUE;
Per Cederqvist's avatar
Per Cederqvist committed
369 370
}

371 372 373 374
static void *
sighandler_winch(oop_source *UNUSED(source),
		 int UNUSED(sig),
		 void *UNUSED(user))
375
{
376 377 378 379
    free_aux_item_definitions();
    initialize_aux_items(param.aux_def_file);
    kom_log("Signal WINCH received. aux definitions reloaded.\n");
    return OOP_CONTINUE;
380 381
}

Per Cederqvist's avatar
Per Cederqvist committed
382
static void
Per Cederqvist's avatar
Per Cederqvist committed
383 384 385 386
save_pid(void)
{
    FILE *fp;

387
    if ((fp = i_fopen(param.pid_name, "w")) == NULL)
Per Cederqvist's avatar
Per Cederqvist committed
388 389
	return;

390
    fprintf(fp, "%ld\n", (long)getpid());
391
    i_fclose(fp);
Per Cederqvist's avatar
Per Cederqvist committed
392 393
}

Per Cederqvist's avatar
Per Cederqvist committed
394
static void
395
go_daemon(void)
Per Cederqvist's avatar
Per Cederqvist committed
396 397 398
{
    pid_t child;
    int fd;
399 400 401 402
#ifdef HAVE_STRUCT_SIGACTION
    struct sigaction act;
#endif
    
403
    if (foreground != 0)
404 405 406 407
    {
	return;
    }

Per Cederqvist's avatar
Per Cederqvist committed
408 409 410 411 412
    if (getppid() != 1)
    {
	/* We were not invoked from /etc/inittab, so
	   disassociate from controlling terminal. */

413 414 415
#ifdef HAVE_STRUCT_SIGACTION
	sigemptyset(&act.sa_mask);
	act.sa_flags = 0;
416
	set_handler_sig_ign(&act);
417 418 419 420 421 422 423 424 425 426 427 428 429 430

#  ifdef SIGTTOU
	sigaction(SIGTTOU, &act, NULL);
#  endif
#  ifdef SIGTTIN
	sigaction(SIGTTIN, &act, NULL);
#  endif
#  ifdef SIGTSTP
	sigaction(SIGTSTP, &act, NULL);
#  endif

#else  /* !HAVE_STRUCT_SIGACTION */

#  ifdef SIGTTOU
Per Cederqvist's avatar
Per Cederqvist committed
431
	signal(SIGTTOU, SIG_IGN);
432 433
#  endif
#  ifdef SIGTTIN
Per Cederqvist's avatar
Per Cederqvist committed
434
	signal(SIGTTIN, SIG_IGN);
435 436
#  endif
#  ifdef SIGTSTP
Per Cederqvist's avatar
Per Cederqvist committed
437
	signal(SIGTSTP, SIG_IGN);
438 439
#  endif

Per Cederqvist's avatar
Per Cederqvist committed
440 441 442 443 444 445 446 447 448 449 450 451
#endif

	child = fork();
	if (child < 0)
	    restart_kom("fork failed: %d\n", errno);
	else if (child > 0)
	    exit (0);		/* parent */
	
	setsid();
    }

    /* Close all file descriptors */
452
    for (fd = 0; fd < fd_ceiling; fd++)
Per Cederqvist's avatar
Per Cederqvist committed
453 454 455
	close(fd);
    if (open("/dev/null", O_RDONLY) != 0
	||open("/dev/null", O_WRONLY) != 1
456
	||open(param.logfile_name, O_WRONLY|O_CREAT|O_APPEND, 0644) != 2)
Per Cederqvist's avatar
Per Cederqvist committed
457 458 459
    {
	/* Kinda stupid to try to log an error message now, but
	   who knows? */
460
	restart_kom("open of log file failed: %d\n", errno);
Per Cederqvist's avatar
Per Cederqvist committed
461
    }
David Byers's avatar
David Byers committed
462 463
    kom_log("*** Version %s (process %lu) coming up.\n",
            kom_version_info.server_version, (unsigned long)getpid());
Per Cederqvist's avatar
Per Cederqvist committed
464 465
}

466
static void
467
initialize(const char *config_file)
468
{
469
#ifdef USING_RLIMIT_NOFILE
470 471 472
    struct rlimit rlim;
#endif

473
    read_configuration(config_file);
474
    initialize_aux_items(param.aux_def_file);
475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490

#ifdef HAVE_LOCALE_H
    if (param.use_locale != NULL)
	if (setlocale(LC_CTYPE, param.use_locale) == NULL)
	{
	    fprintf(stderr, "setlocale: ");
	    perror(param.use_locale);
	    exit(1);
	}
#else
    if (param.use_locale != NULL)
    {
	fprintf(stderr, "locale not supported in your environment.\n");
	exit(1);
    }
#endif
491

492 493 494 495 496 497 498 499
    if (param.no_files != -1 && param.no_files > FD_SETSIZE)
    {
	fprintf(stderr,
		"The \"Open files: %ld\" is too large: FD_SETSIZE is %ld\n",
		(long)param.no_files, (long)FD_SETSIZE);
	exit(1);
    }

500
#ifdef USING_RLIMIT_NOFILE
501 502 503 504 505 506
    if (getrlimit(RLIMIT_NOFILE, &rlim) < 0)
    {
	perror("getrlimit(RLIMIT_NOFILE) failed");
	exit(1);
    }

Per Cederqvist's avatar
Per Cederqvist committed
507
    if (param.no_files != -1)
508
    {
509
	if ((rlim_t)param.no_files > rlim.rlim_max)
510
	{
511 512 513 514
	    fprintf(stderr, "attempt to raise open files from %ld to %ld, "
		    "but only %ld is allowed\n",
		    (long)rlim.rlim_cur, (long)param.no_files,
		    (long)rlim.rlim_max);
515 516 517 518 519 520 521 522 523 524
	    rlim.rlim_cur = rlim.rlim_max;
	}
	else
	    rlim.rlim_cur = param.no_files;
	
	if (setrlimit(RLIMIT_NOFILE, &rlim) < 0)
	{
	    perror("setrlimit failed");
	    exit(1);
	}
525 526 527 528 529 530 531 532
	if (getrlimit(RLIMIT_NOFILE, &rlim) < 0)
	{
	    perror("getrlimit(RLIMIT_NOFILE) failed");
	    exit(1);
	}
	if (rlim.rlim_cur != (rlim_t)param.no_files)
	    restart_kom("getrlimit after setrlimit returns %ld, not %ld.\n",
			(long)rlim.rlim_cur, (long)param.no_files);
533
    }
534 535 536 537 538
    fd_ceiling = rlim.rlim_cur;
#elif HAVE_GETDTABLESIZE
    if (param.no_files == -1)
	fd_ceiling = getdtablesize();
    else if (param.no_files >= getdtablesize())
539
    {
540 541
	fprintf(stderr, "getdtablesize() indicates that at most %ld files"
		" can be used.\n", (long)getdtablesize());
542 543
	exit(1);
    }
544 545 546 547 548 549 550 551 552 553 554 555 556
    else
	fd_ceiling = param.no_files;
#elif defined(OPEN_MAX)
    if (param.no_files == -1)
	fd_ceiling = OPEN_MAX;
    else if (param.no_files >= OPEN_MAX)
    {
	fprintf(stderr, "OPEN_MAX indicates that at most %ld files"
		" can be used.\n", (long)OPEN_MAX);
	exit(1);
    }
    else
	fd_ceiling = param.no_files;
557
#else
558
#  error Do not know how to find maximum number of open files.
559 560
#endif

561 562 563
    if (fd_ceiling > FD_SETSIZE)
	fd_ceiling = FD_SETSIZE;

564
    go_daemon();
565
    if (lock_db() < 0)
566 567 568 569 570
    {
	/* Don't actually die until something is entered on stdin in debug
	   mode.  This is mainly here for the benefit of the test suite,
	   but is could also be useful to be able to attach a debugger and
	   do pre-mortem debugging of the process at this point.  */
571
	kom_log("Cannot obtain database lock.  Exiting.\n");
572
	if (foreground)
573 574 575 576 577
	{
	    kom_log("Press enter to terminate lyskomd\n");
	    getchar();
	}

578
	exit(1);
579 580
    }

581
    server_init();
582
    init_data_base();
583 584
}

Per Cederqvist's avatar
Per Cederqvist committed
585 586 587
/* Stop complaint from gcc 2.0 about "no previous prototype for `main'". */
int main(int argc, char **argv);

588
	
Per Cederqvist's avatar
Per Cederqvist committed
589 590 591 592 593
int
main (int    argc,
      char **argv)
{
    int i;
594
    const char *config_file;
595
    oop_source *src;
Per Cederqvist's avatar
Per Cederqvist committed
596 597 598

    link_ansi();

599 600
    if (gettimeofday(&current_time, NULL) < 0)
	restart_kom("gettimeofday failed: %s\n", strerror(errno));
Per Cederqvist's avatar
Per Cederqvist committed
601

602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617
#ifdef TRACED_ALLOCATIONS
    /* We must do this before we allocate any memory... */
    {
      char buf[1024];
      char *nl;

      fputs("Where does the trace want to go today? [stderr]\n", stdout);
      fflush(stdout);
      if (fgets(buf, sizeof(buf), stdin) != buf)
	  restart_kom("main(): unable to read trace location\n");
      if ((nl = strchr(buf, '\n')) != NULL)
	  *nl = '\0';
      trace_alloc_file(buf);
    }
#endif

David Byers's avatar
David Byers committed
618
    kom_log("*** Version %s (process %lu) started.\n",
619
	kom_version_info.server_version, (unsigned long)getpid());
620
#ifdef DEBUG_CALLS
David Byers's avatar
David Byers committed
621 622
    kom_log("WARNING: This server was compiled with --with-debug-calls.\n");
    kom_log("It isn't safe to use in a production environment.\n");
623 624
#else
    kom_log("Debug calls are disabled, as they should be.\n");
625
#endif
626

Per Cederqvist's avatar
Per Cederqvist committed
627
#ifdef ENCRYPT_PASSWORDS
628
    /* Seed the random number generator. */
629
    /* FIXME (bug 1068): This is not a good way to seed it... */
Per Cederqvist's avatar
Per Cederqvist committed
630 631
    srand(time(NULL) + getpid());
#endif
Per Cederqvist's avatar
Per Cederqvist committed
632

633 634
    /* Initialize the string handling package. */
    s_set_storage_management(string_malloc, string_realloc, string_free);
635

636
    /* Parse command line arguments. */
Per Cederqvist's avatar
Per Cederqvist committed
637 638 639 640 641 642 643
    for (i = 1; i < argc && argv[i][0] == '-'; i++)
	switch (argv[i][1])
	{
	case 'd':
	    buglevel++;
	    break;

644 645 646 647
	case 'f':
	    foreground = 1;
	    break;

648
	default:
649
	    restart_kom("usage: %s [-f] [-d ...] [config-file]\n", argv[0]);
650
	}
Per Cederqvist's avatar
Per Cederqvist committed
651

652

653
    /* Read in the configuration file. */
654

Per Cederqvist's avatar
Per Cederqvist committed
655
    if (i < argc)
656 657
	config_file = argv[i++];
    else
658
	config_file = get_default_config_file_name();
Per Cederqvist's avatar
Per Cederqvist committed
659

660 661
    if (i < argc)
	    restart_kom("usage: %s [-d ...] [config-file]\n", argv[0]);
662

663
    init_stats();
664
    initialize(config_file);	/* Read config, listen, and start db */
665
    chdir(param.core_dir);
666

Per Cederqvist's avatar
Per Cederqvist committed
667
    save_pid();
Per Cederqvist's avatar
Per Cederqvist committed
668

Per Cederqvist's avatar
Per Cederqvist committed
669
    toploop();
Per Cederqvist's avatar
Per Cederqvist committed
670

671 672 673
    /* There is no use sending logout messages to all sessions. */
    param.send_async_messages = FALSE;
    
Per Cederqvist's avatar
Per Cederqvist committed
674
    logout_all_clients();
675
    isc_shutdown(kom_server_mcb);
Per Cederqvist's avatar
Per Cederqvist committed
676
    cache_sync_all();
677
    unlock_db();
678
    src = oop_signal_source(kom_signal_adapter);
679 680 681 682
    src->cancel_signal(src, SIGWINCH, sighandler_winch, NULL);
    src->cancel_signal(src, SIGUSR2, sighandler_usr2, NULL);
    src->cancel_signal(src, SIGUSR1, sighandler_usr1, NULL);
    src->cancel_signal(src, SIGQUIT, sighandler_quit, NULL);
683 684 685
    src->cancel_signal(src, SIGHUP, sighandler_term, NULL);
    src->cancel_signal(src, SIGTERM, sighandler_term, NULL);
    src->cancel_signal(src, SIGINT, sighandler_term, NULL);
686
    oop_signal_delete(kom_signal_adapter);
687
    oop_sys_delete(kom_server_oop_src);
Per Cederqvist's avatar
Per Cederqvist committed
688

David Byers's avatar
David Byers committed
689 690
    /* Finish */

691 692
    dump_exit_statistics();

David Byers's avatar
David Byers committed
693
    kom_log("%s terminated normally.\n", argv[0]);
694 695 696 697 698

    /* Don't actually die until something is entered on stdin in debug
       mode.  This is mainly here for the benefit of the test suite,
       but is could also be useful to be able to attach a debugger and
       do pre-mortem debugging of the process at this point.  */
699
    if (foreground)
700
    {
David Byers's avatar
David Byers committed
701
	kom_log("Press enter to terminate lyskomd\n");
702 703
	getchar();
    }
David Byers's avatar
David Byers committed
704

705
    exit(0);
706 707 708 709 710 711 712 713
}

static void 
dump_exit_statistics(void)
{
    FILE *stat_file;
    time_t now;

714 715
    time(&now);
    stat_file = i_fopen(param.memuse_name, "a");
Per Cederqvist's avatar
Per Cederqvist committed
716 717 718 719

    if ( stat_file == NULL )
	restart_kom("Can't open file to save memory usage to.\n");

720 721
    fprintf(stat_file, "\nLysKOM Server going down at %s\n", ctime(&now));

722 723 724 725
    dump_cache_stats (stat_file);

    free_all_tmp();
    free_all_cache();
726
    free_all_jubel();
727
    clear_info(&kom_info);
David Byers's avatar
David Byers committed
728 729
    free_aux_item_definitions();
    free_configuration();
730
    free_default_config_file_name();
731

Per Cederqvist's avatar
Per Cederqvist committed
732 733
    dump_smalloc_counts(stat_file);
    dump_alloc_counts(stat_file);
734
    dump_cache_mem_usage(stat_file);
Per Cederqvist's avatar
Per Cederqvist committed
735
    dump_string_alloc_counts(stat_file);
736
    dump_allocated_connections(stat_file);
737
    dump_isc_alloc_counts(stat_file);
738
    dump_oop_alloc_counts(stat_file);
739
    dump_l2g_stats(stat_file);
740
    i_fclose (stat_file);
Per Cederqvist's avatar
Per Cederqvist committed
741
}