ramkomd.c 12.8 KB
Newer Older
Linus Tolke Y's avatar
Linus Tolke Y committed
1
/*
2
 * $Id: ramkomd.c,v 0.52 1994/10/24 00:34:39 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.52 1994/10/24 00:34:39 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 55 56
#ifdef HAVE_SYS_RESOURCE_H
#  include <sys/resource.h>
#endif
Per Cederqvist's avatar
Per Cederqvist committed
57
#include <unistd.h>
Per Cederqvist's avatar
Per Cederqvist committed
58 59 60 61
#include <setjmp.h>
#if defined(HAVE_SYS_PARAM_H) && !defined(HAVE_GETCWD)
# include <sys/param.h>
#endif
Per Cederqvist's avatar
Per Cederqvist committed
62
#include <errno.h>
Per Cederqvist's avatar
Per Cederqvist committed
63
#include <sys/types.h>
64
#include <sys/socket.h>
Per Cederqvist's avatar
Per Cederqvist committed
65
#include <fcntl.h>
Per Cederqvist's avatar
Per Cederqvist committed
66

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

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

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


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

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

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

114
static void dump_exit_statistics(void);
115

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

#if defined(HAVE_SETRLIMIT) && defined(RLIMIT_NOFILE)
    if (param.no_file != -1)
    {
	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
369
    go_daemon(param.dbase_dir, param.logfile_name);
370 371 372 373
    server_init(param.ip_client_port, param.ip_mux_port);
    init_data_base(param.dbase_dir);
}

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

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

386 387 388 389
    log("*** Version %d (process %d) started.\n",
#include "version.incl"
	, getpid());

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();
}