ramkomd.c 12.8 KB
Newer Older
Linus Tolke Y's avatar
Linus Tolke Y committed
1
/*
2
 * $Id: ramkomd.c,v 0.54 1994/12/31 15:16:19 ceder Exp $
Per Cederqvist's avatar
Per Cederqvist committed
3
 * Copyright (C) 1991, 1992, 1993, 1994  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 23 24
 *
 * 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.
 *
 * Please mail bug reports to bug-lyskom@lysator.liu.se. 
 */
Per Cederqvist's avatar
Per Cederqvist committed
25 26 27 28 29 30 31 32 33
/*
 * 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
 */

34
static char *rcsid = "$Id: ramkomd.c,v 0.54 1994/12/31 15:16:19 ceder Exp $";
35 36
#include "rcs.h"
USE(rcsid);
Per Cederqvist's avatar
Per Cederqvist committed
37

38 39 40
#ifdef HAVE_LOCALE_H
#  include <locale.h>
#endif
Per Cederqvist's avatar
Per Cederqvist committed
41
#include <stdio.h>
Per Cederqvist's avatar
Per Cederqvist committed
42
#include <signal.h>
Per Cederqvist's avatar
Per Cederqvist committed
43 44 45
#ifdef HAVE_STDARG_H
#  include <stdarg.h>
#endif
46 47 48 49 50 51
#ifdef HAVE_STDLIB_H
#  include <stdlib.h>
#endif
#ifdef HAVE_STRING_H
#  include <string.h>
#endif
Per Cederqvist's avatar
Per Cederqvist committed
52
#include <sys/wait.h>
Per Cederqvist's avatar
Per Cederqvist committed
53
#include <time.h>
54
#ifdef HAVE_SYS_RESOURCE_H
Per Cederqvist's avatar
Per Cederqvist committed
55
#  include <sys/time.h>
56 57
#  include <sys/resource.h>
#endif
Per Cederqvist's avatar
Per Cederqvist committed
58
#include <unistd.h>
Per Cederqvist's avatar
Per Cederqvist committed
59 60 61 62
#include <setjmp.h>
#if defined(HAVE_SYS_PARAM_H) && !defined(HAVE_GETCWD)
# include <sys/param.h>
#endif
Per Cederqvist's avatar
Per Cederqvist committed
63
#include <errno.h>
Per Cederqvist's avatar
Per Cederqvist committed
64
#include <sys/types.h>
65
#include <sys/socket.h>
Per Cederqvist's avatar
Per Cederqvist committed
66
#include <fcntl.h>
Per Cederqvist's avatar
Per Cederqvist committed
67

Per Cederqvist's avatar
Per Cederqvist committed
68
#include "exp.h"
69
#include "s-string.h"
Per Cederqvist's avatar
Per Cederqvist committed
70
#include "misc-types.h"
71
#include "kom-types.h"
Per Cederqvist's avatar
Per Cederqvist committed
72 73 74 75
#include "com.h"
#include "connections.h"
#include "internal-connections.h"
#include "kom-errno.h"
Per Cederqvist's avatar
Per Cederqvist committed
76
#include "isc-malloc.h"
77
#include "isc-interface.h"
Per Cederqvist's avatar
Per Cederqvist committed
78
#include "config.h"
Per Cederqvist's avatar
Per Cederqvist committed
79
#include "cache.h"
Per Cederqvist's avatar
Per Cederqvist committed
80 81
#include "string-malloc.h"
#include "lyskomd.h"
Per Cederqvist's avatar
Per Cederqvist committed
82
#include "log.h"
Per Cederqvist's avatar
Per Cederqvist committed
83 84 85
#include "internal-services.h"
#include "server/smalloc.h"
#include "kom-memory.h"
86 87
#include "conf-file.h"
#include "param.h"
88 89
#include "server-config.h"
#include "manipulate.h"
90
#include "version-info.h"
91

92 93 94 95
#if defined(HAVE_SETRLIMIT) && defined(RLIMIT_OFILE) && !defined(RLIMIT_NOFILE)
#  define RLIMIT_NOFILE RLIMIT_OFILE
#endif

96 97 98 99 100 101
struct kom_par param;
 
Kom_err		  kom_errno = KOM_NO_ERROR;
unsigned long	  err_stat = 0;


Per Cederqvist's avatar
Per Cederqvist committed
102
#ifndef NDEBUG
103
int	buglevel = 0;
104
#endif
Per Cederqvist's avatar
Per Cederqvist committed
105

106 107 108 109 110 111
EXPORT char datafilename[BUFSIZ];      /* Full pathname to the database file */
EXPORT char backupfilename[BUFSIZ];    /* Full pathname to the backup file */
EXPORT char textfilename[BUFSIZ];      /* Full pathname to the text file. */
EXPORT char statisticfile[BUFSIZ];     /* Dito to statistics file. */
EXPORT char pidfile[BUFSIZ];	       /* Dito pid-file. */
INTERNAL char memusefile[BUFSIZ];      /* Memory usage file. */
Per Cederqvist's avatar
Per Cederqvist committed
112

113 114
static IscSession *listen_client = NULL;      /* ISC listen identifier */
static IscSession *listen_mux    = NULL;      /*          -"-          */
Per Cederqvist's avatar
Per Cederqvist committed
115

116
static void dump_exit_statistics(void);
117

Per Cederqvist's avatar
Per Cederqvist committed
118
static void
119
server_init (char * client_port, char * mux_port)
Per Cederqvist's avatar
Per Cederqvist committed
120
{
Per Cederqvist's avatar
Per Cederqvist committed
121
    IscConfig config;
122
    IscAddress *isc_adr = NULL;
Per Cederqvist's avatar
Per Cederqvist committed
123
    
Per Cederqvist's avatar
Per Cederqvist committed
124 125 126
    /*
    ** Setup some parameters here
    */
Per Cederqvist's avatar
Per Cederqvist committed
127 128 129 130 131
    config.version = 1005;
    config.master.version = 1001;
    config.master.memfn.alloc = &isc_malloc_wrapper;
    config.master.memfn.realloc = &isc_realloc_wrapper;
    config.master.memfn.free = &isc_free_wrapper;
132
    config.master.abortfn = NULL; /* Use default abort function. */
Per Cederqvist's avatar
Per Cederqvist committed
133
    config.session.version = 1001;
134
    config.session.max.msgsize = -1; /* Use default sizes. */
Per Cederqvist's avatar
Per Cederqvist committed
135 136 137 138
    config.session.max.queuedsize = -1;
    config.session.max.dequeuelen = -1;
    config.session.max.openretries = -1;
    config.session.max.backlog = -1;
Per Cederqvist's avatar
Per Cederqvist committed
139

140
    kom_server_mcb  = isc_initialize(&config);
Per Cederqvist's avatar
Per Cederqvist committed
141
    if ( kom_server_mcb == NULL )
Per Cederqvist's avatar
Per Cederqvist committed
142 143 144 145 146
	restart_kom("server_init: can't isc_initialize()\n");

    listen_client = isc_listentcp(kom_server_mcb, NULL, client_port);
    if (listen_client == NULL)
        restart_kom("server_init: can't isc_listentcp(CLIENT)\n");
Per Cederqvist's avatar
Per Cederqvist committed
147

148 149 150 151
    isc_adr = isc_getladdress (listen_client);
    if (isc_adr == NULL)
	restart_kom("server_init(): can't isc_getladdress (listen_client)\n");

152
    param.num_ip_client_port = isc_getportnum (isc_adr);
153 154 155
    isc_freeaddress (isc_adr);
    
    listen_mux = isc_listentcp (kom_server_mcb, NULL, mux_port);
Per Cederqvist's avatar
Per Cederqvist committed
156 157
    if (listen_mux == NULL)
        restart_kom("server_init: can't isc_listentcp(MUX)\n");
158 159 160 161

    isc_adr = isc_getladdress (listen_mux);
    if (isc_adr == NULL)
	restart_kom("server_init(): can't isc_getladdress (listen_mux)\n");
Per Cederqvist's avatar
Per Cederqvist committed
162
    
163
    param.num_ip_mux_port = isc_getportnum (isc_adr);
164 165 166
    isc_freeaddress (isc_adr);

    log("Listening for clients on %d and muxes on %d.\n",
167
	param.num_ip_client_port, param.num_ip_mux_port);
Per Cederqvist's avatar
Per Cederqvist committed
168 169 170 171 172 173 174 175 176 177

    /*
     * 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.
     */
    signal(SIGPIPE, SIG_IGN);
}

static void
178
init_data_base(const char *dir_base)
Per Cederqvist's avatar
Per Cederqvist committed
179
{
180 181 182 183 184 185
    sprintf(datafilename,   "%s/%s", dir_base, param.datafile_name);
    sprintf(backupfilename, "%s/%s", dir_base, param.backupfile_name);
    sprintf(textfilename, "%s/%s", dir_base, param.textfile_name);
    sprintf(statisticfile, "%s/%s", dir_base, param.statistic_name);
    sprintf(pidfile, "%s/%s", dir_base, param.pid_name);
    sprintf(memusefile, "%s/%s", dir_base, param.memuse_name);
Per Cederqvist's avatar
Per Cederqvist committed
186 187 188 189 190

    log("Database = %s\n",  datafilename);
    log("Backup = %s\n", backupfilename);
    
    if ( init_cache() == FAILURE )
191
	restart_kom ("Cannot find database.\n");
Per Cederqvist's avatar
Per Cederqvist committed
192 193
}

Per Cederqvist's avatar
Per Cederqvist committed
194
static void
Per Cederqvist's avatar
Per Cederqvist committed
195
sighandler_hup (int sig)
Per Cederqvist's avatar
Per Cederqvist committed
196 197 198 199 200
{
    log ("Signal HUP received. Shutting down server.\n");
    go_and_die = TRUE;
}

Per Cederqvist's avatar
Per Cederqvist committed
201
static void
Per Cederqvist's avatar
Per Cederqvist committed
202
sighandler_quit (int sig)
Per Cederqvist's avatar
Per Cederqvist committed
203 204 205 206 207 208 209
{
    log ("Signal QUIT received - syncing...\n");
    cache_sync_all();
    log ("Dumping core now.\n");
    abort();
}

Per Cederqvist's avatar
Per Cederqvist committed
210
static void
Per Cederqvist's avatar
Per Cederqvist committed
211
sighandler_usr1 (int sig)
Per Cederqvist's avatar
Per Cederqvist committed
212
{
213
    signal(SIGUSR1, sighandler_usr1);
Per Cederqvist's avatar
Per Cederqvist committed
214 215 216
    do_sync_db = TRUE;
}

Per Cederqvist's avatar
Per Cederqvist committed
217
static void
Per Cederqvist's avatar
Per Cederqvist committed
218
sighandler_usr2 (int sig)
Per Cederqvist's avatar
Per Cederqvist committed
219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238
{
    int		  child;

    log ("Signal USR2 received - will dump core now.  (Check that child dies.)\n");
    if ((child = fork()) == 0)
    {
	abort();
	log ("Abort() failed!!!");
	exit(1);
    }
    else if (child < 0)
    {
	log ("Couldn't fork.\n");
    }
    else
    {
	wait (NULL);
    }
}

Per Cederqvist's avatar
Per Cederqvist committed
239
static void
Per Cederqvist's avatar
Per Cederqvist committed
240 241 242 243 244 245 246
save_pid(void)
{
    FILE *fp;

    if ( (fp = fopen(pidfile, "w")) == NULL )
	return;

247
    fprintf(fp, "%ld\n", (long)getpid());
Per Cederqvist's avatar
Per Cederqvist committed
248 249 250
    fclose(fp);
}

Per Cederqvist's avatar
Per Cederqvist committed
251
static void
Per Cederqvist's avatar
Per Cederqvist committed
252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292
go_daemon(const char *prefix, const char *logf)
{
    pid_t child;
    int fd;
    char *logfile;

    if (getppid() != 1)
    {
	/* We were not invoked from /etc/inittab, so
	   disassociate from controlling terminal. */

#ifdef SIGTTOU
	signal(SIGTTOU, SIG_IGN);
#endif
#ifdef SIGTTIN
	signal(SIGTTIN, SIG_IGN);
#endif
#ifdef SIGTSTP
	signal(SIGTSTP, SIG_IGN);
#endif

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

    logfile = smalloc(strlen(prefix) + 1 + strlen(logf) + 1);
    sprintf(logfile, "%s/%s", prefix, logf);
    /* Close all file descriptors */
    for (fd = 0; fd < MAX_NO_OF_CONNECTIONS + PROTECTED_FDS; fd++)
	close(fd);
    if (open("/dev/null", O_RDONLY) != 0
	||open("/dev/null", O_WRONLY) != 1
	||open(logfile, O_WRONLY|O_CREAT|O_APPEND, 0644) != 2)
    {
	/* Kinda stupid to try to log an error message now, but
	   who knows? */
293
	restart_kom("open of log file failed: %d\n", errno);
Per Cederqvist's avatar
Per Cederqvist committed
294 295
    }
    sfree(logfile);
296 297
    log("*** Version %s (process %d) coming up.\n",
	kom_version_info.server_version, getpid());
Per Cederqvist's avatar
Per Cederqvist committed
298 299
}

300 301 302
void
initialize(const char *config_file)
{
303 304 305 306
#if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE)
    struct rlimit rlim;
#endif

307
    read_configuration(config_file);
308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323

#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
324 325

#if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE)
Per Cederqvist's avatar
Per Cederqvist committed
326
    if (param.no_files != -1)
327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369
    {
	if (getrlimit(RLIMIT_NOFILE, &rlim) < 0)
	{
	    perror("getrlimit(RLIMIT_NOFILE) failed");
	    exit(1);
	}
	if (param.no_files > rlim.rlim_max)
	{
	    fprintf(stderr, "attempt to raise open files from %d to %d, "
		    "but only %d is allowed\n", rlim.rlim_cur, param.no_files,
		    rlim.rlim_max);
	    rlim.rlim_cur = rlim.rlim_max;
	}
	else
	    rlim.rlim_cur = param.no_files;
	
	if (setrlimit(RLIMIT_NOFILE, &rlim) < 0)
	{
	    perror("setrlimit failed");
	    exit(1);
	}
    }
#else
    if (param.no_files != -1)
    {
	fprintf(stderr,
		"can't alter number of open files in your environment.\n");
	exit(1);
    }
#endif

    /* Find out how many connections we can handle. */
#if defined (HAVE_SYSCONF)
    MAX_NO_OF_CONNECTIONS = sysconf(_SC_OPEN_MAX) - PROTECTED_FDS;
#else
# ifdef HAVE_GETDTABLESIZE
    MAX_NO_OF_CONNECTIONS = getdtablesize() - PROTECTED_FDS;
# else
    /* If we don't have getdtablesize or sysconf, MAX_NO_OF_CONNECTIONS is
       set at compile time. */
# endif
#endif

Per Cederqvist's avatar
Per Cederqvist committed
370
    go_daemon(param.dbase_dir, param.logfile_name);
371 372 373 374
    server_init(param.ip_client_port, param.ip_mux_port);
    init_data_base(param.dbase_dir);
}

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

378
	
Per Cederqvist's avatar
Per Cederqvist committed
379 380 381 382 383
int
main (int    argc,
      char **argv)
{
    int i;
384 385
    char *default_config_file;
    char *config_file;
Per Cederqvist's avatar
Per Cederqvist committed
386

387 388
    log("*** Version %s (process %d) started.\n",
	kom_version_info.server_version, getpid());
389

Per Cederqvist's avatar
Per Cederqvist committed
390
#ifdef ENCRYPT_PASSWORDS
391
    /* Seed the random number generator. */
Per Cederqvist's avatar
Per Cederqvist committed
392 393
    srand(time(NULL) + getpid());
#endif
Per Cederqvist's avatar
Per Cederqvist committed
394

395 396
    /* Initialize the string handling package. */
    s_set_storage_management(string_malloc, string_realloc, string_free);
397

398
    /* Parse command line arguments. */
Per Cederqvist's avatar
Per Cederqvist committed
399 400 401 402 403 404 405
    for (i = 1; i < argc && argv[i][0] == '-'; i++)
	switch (argv[i][1])
	{
	case 'd':
	    buglevel++;
	    break;

406 407 408
	default:
	    restart_kom("usage: %s [-d ...] [config-file]\n", argv[0]);
	}
Per Cederqvist's avatar
Per Cederqvist committed
409

410

411
    /* Read in the configuration file. */
412

413 414
    default_config_file = smalloc(strlen(DEFAULT_DBASE_DIR) +
				  strlen(CONFIG_FILE) + 2);
415
    sprintf(default_config_file, "%s/%s", DEFAULT_DBASE_DIR, CONFIG_FILE);
Per Cederqvist's avatar
Per Cederqvist committed
416
    if (i < argc)
417 418 419
	config_file = argv[i++];
    else
	config_file = default_config_file;
Per Cederqvist's avatar
Per Cederqvist committed
420

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

Per Cederqvist's avatar
Per Cederqvist committed
424 425 426 427 428
    signal(SIGHUP, sighandler_hup);
    signal(SIGQUIT, sighandler_quit);
    signal(SIGUSR1, sighandler_usr1);
    signal(SIGUSR2, sighandler_usr2);
  
429 430 431
    initialize(config_file);	/* Read config, listen, and start db */
    sfree(default_config_file);

Per Cederqvist's avatar
Per Cederqvist committed
432
    save_pid();
Per Cederqvist's avatar
Per Cederqvist committed
433

Per Cederqvist's avatar
Per Cederqvist committed
434
    toploop();
Per Cederqvist's avatar
Per Cederqvist committed
435 436

    logout_all_clients();
437
    isc_shutdown(kom_server_mcb);
Per Cederqvist's avatar
Per Cederqvist committed
438
    cache_sync_all();
Per Cederqvist's avatar
Per Cederqvist committed
439

440 441
    dump_exit_statistics();

Per Cederqvist's avatar
Per Cederqvist committed
442
    log("%s terminated normally.\n", argv[0]);
443 444 445 446 447 448 449 450 451 452 453 454
    
    return 0;
}

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

    time(&now);
    stat_file = fopen(memusefile, "a");
Per Cederqvist's avatar
Per Cederqvist committed
455 456 457 458

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

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

461 462 463 464
    dump_cache_stats (stat_file);

    free_all_tmp();
    free_all_cache();
465
    free_all_jubel();
466

Per Cederqvist's avatar
Per Cederqvist committed
467 468
    dump_smalloc_counts(stat_file);
    dump_alloc_counts(stat_file);
469
    dump_cache_mem_usage(stat_file);
Per Cederqvist's avatar
Per Cederqvist committed
470
    dump_string_alloc_counts(stat_file);
471
    dump_allocated_connections(stat_file);
472
    dump_isc_alloc_counts(stat_file);
Per Cederqvist's avatar
Per Cederqvist committed
473
    fclose (stat_file);
Per Cederqvist's avatar
Per Cederqvist committed
474
}
475
    
Per Cederqvist's avatar
Per Cederqvist committed
476
void
Per Cederqvist's avatar
Per Cederqvist committed
477
#if defined(HAVE_VFPRINTF) && defined(HAVE_STDARG_H)
Per Cederqvist's avatar
Per Cederqvist committed
478
restart_kom(const char * format, ...)
479 480 481 482 483 484
#else
restart_kom(format,
	    a, b, c, d, e, f, g)
     const char *format;
     int a, b, c, d, e, f, g;
#endif
Per Cederqvist's avatar
Per Cederqvist committed
485
{
Per Cederqvist's avatar
Per Cederqvist committed
486
#if defined(HAVE_VFPRINTF) && defined(HAVE_STDARG_H)
Per Cederqvist's avatar
Per Cederqvist committed
487
    va_list AP;
488
#endif
Per Cederqvist's avatar
Per Cederqvist committed
489 490 491
#if defined(HAVE_GETCWD)
    char pathname[1026];
#else /* !defined(HAVE_GETCWD) */
492
    char *getwd(char *); /* +++ should be in sys/param.h */
Per Cederqvist's avatar
Per Cederqvist committed
493 494 495 496 497 498
# if defined(MAXPATHLEN)
    char pathname[MAXPATHLEN];
# else
    char pathname[1024];
# endif /* !defined(MAXPATHLEN) */
#endif /* !defined(HAVE_GETCWD) */
Per Cederqvist's avatar
Per Cederqvist committed
499

Per Cederqvist's avatar
Per Cederqvist committed
500
#if defined(HAVE_VFPRINTF) && defined(HAVE_STDARG_H)
Per Cederqvist's avatar
Per Cederqvist committed
501 502 503
    va_start(AP, format);
    logv(format, AP);
    va_end(AP);
504 505 506
#else
    log(format, a, b, c, d, e, f, g);
#endif
Per Cederqvist's avatar
Per Cederqvist committed
507 508

    log("Previous message is fatal. Will dump core now.\n");
509

Per Cederqvist's avatar
Per Cederqvist committed
510 511 512 513 514 515 516
#ifdef HAVE_GETCWD
    /* getcwd is POSIX, so try that first. */
    if (getcwd(pathname, 1026) == NULL)
	log("getcwd failed: errno %d\n", errno);
    else
	log("Search for the core in %s\n", pathname);
#else
517 518 519 520
    if ( getwd(pathname) == NULL )
	log("getwd failed: %s\n", pathname);
    else
	log("Search for the core in %s\n", pathname);
Per Cederqvist's avatar
Per Cederqvist committed
521
#endif
522
    
Per Cederqvist's avatar
Per Cederqvist committed
523 524
    abort();
}