ramkomd.c 12.8 KB
Newer Older
Linus Tolke Y's avatar
Linus Tolke Y committed
1
/*
Per Cederqvist's avatar
Per Cederqvist committed
2
 * $Id: ramkomd.c,v 0.53 1994/11/14 23:37:13 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
 */

Per Cederqvist's avatar
Per Cederqvist committed
34
static char *rcsid = "$Id: ramkomd.c,v 0.53 1994/11/14 23:37:13 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

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

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


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

105 106 107 108 109 110
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
111

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

115
static void dump_exit_statistics(void);
116

Per Cederqvist's avatar
Per Cederqvist committed
117
static void
118
server_init (char * client_port, char * mux_port)
Per Cederqvist's avatar
Per Cederqvist committed
119
{
Per Cederqvist's avatar
Per Cederqvist committed
120
    IscConfig config;
121
    IscAddress *isc_adr = NULL;
Per Cederqvist's avatar
Per Cederqvist committed
122
    
Per Cederqvist's avatar
Per Cederqvist committed
123 124 125
    /*
    ** Setup some parameters here
    */
Per Cederqvist's avatar
Per Cederqvist committed
126 127 128 129 130
    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;
131
    config.master.abortfn = NULL; /* Use default abort function. */
Per Cederqvist's avatar
Per Cederqvist committed
132
    config.session.version = 1001;
133
    config.session.max.msgsize = -1; /* Use default sizes. */
Per Cederqvist's avatar
Per Cederqvist committed
134 135 136 137
    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
138

139
    kom_server_mcb  = isc_initialize(&config);
Per Cederqvist's avatar
Per Cederqvist committed
140
    if ( kom_server_mcb == NULL )
Per Cederqvist's avatar
Per Cederqvist committed
141 142 143 144 145
	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
146

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

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

    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
161
    
162
    param.num_ip_mux_port = isc_getportnum (isc_adr);
163 164 165
    isc_freeaddress (isc_adr);

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

    /*
     * 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
177
init_data_base(const char *dir_base)
Per Cederqvist's avatar
Per Cederqvist committed
178
{
179 180 181 182 183 184
    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
185 186 187 188 189

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

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

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

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

Per Cederqvist's avatar
Per Cederqvist committed
216
static void
Per Cederqvist's avatar
Per Cederqvist committed
217
sighandler_usr2 (int sig)
Per Cederqvist's avatar
Per Cederqvist committed
218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237
{
    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
238
static void
Per Cederqvist's avatar
Per Cederqvist committed
239 240 241 242 243 244 245
save_pid(void)
{
    FILE *fp;

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

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

Per Cederqvist's avatar
Per Cederqvist committed
250
static void
Per Cederqvist's avatar
Per Cederqvist committed
251 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
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? */
292
	restart_kom("open of log file failed: %d\n", errno);
Per Cederqvist's avatar
Per Cederqvist committed
293 294
    }
    sfree(logfile);
295 296 297
    log("*** Version %d (process %d) coming up.\n",
#include "version.incl"
	, 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 389 390
    log("*** Version %d (process %d) started.\n",
#include "version.incl"
	, getpid());

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

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

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

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

411

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

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

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

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

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

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

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

441 442
    dump_exit_statistics();

Per Cederqvist's avatar
Per Cederqvist committed
443
    log("%s terminated normally.\n", argv[0]);
444 445 446 447 448 449 450 451 452 453 454 455
    
    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
456 457 458 459

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

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

462 463 464 465
    dump_cache_stats (stat_file);

    free_all_tmp();
    free_all_cache();
466
    free_all_jubel();
467

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

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

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

Per Cederqvist's avatar
Per Cederqvist committed
511 512 513 514 515 516 517
#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
518 519 520 521
    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
522
#endif
523
    
Per Cederqvist's avatar
Per Cederqvist committed
524 525
    abort();
}