]> git.pond.sub.org Git - empserver/blobdiff - src/server/main.c
Simplify checks whether player thread may sleep
[empserver] / src / server / main.c
index ad527c4fd85dc40a8017d52ca31f9248950357ac..336eb6a65d1e4c408b910b9271f4481bbdce08f8 100644 (file)
@@ -1,6 +1,6 @@
 /*
  *  Empire - A multi-player, client/server Internet based war game.
- *  Copyright (C) 1986-2007, Dave Pare, Jeff Bailey, Thomas Ruschak,
+ *  Copyright (C) 1986-2009, Dave Pare, Jeff Bailey, Thomas Ruschak,
  *                           Ken Stevens, Steve McClure
  *
  *  This program is free software; you can redistribute it and/or modify
  *  ---
  *
  *  main.c: Empire Server main, startup and shutdown
- * 
+ *
  *  Known contributors to this file:
  *     Dave Pare, 1994
  *     Steve McClure, 1996, 1998
  *     Doug Hay, 1998
- *     Ron Koenderink, 2004-2005
- *     Markus Armbruster, 2005-2006
+ *     Ron Koenderink, 2004-2009
+ *     Markus Armbruster, 2005-2008
  */
 
 #include <config.h>
 #include <errno.h>
 #include <signal.h>
 #include <stdio.h>
+#ifndef _WIN32
+#include <sys/wait.h>
+#endif
+#include <unistd.h>
 
 #if defined(_WIN32)
-#define WIN32
 #include <winsock2.h>
 #undef NS_ALL
 #include <process.h>
-#include <direct.h>
 #include "service.h"
-#include "../lib/gen/getopt.h"
-#else
-#include <sys/types.h>
-#include <unistd.h>
 #endif
 
 #include "empio.h"
@@ -59,6 +57,7 @@
 #include "file.h"
 #include "journal.h"
 #include "land.h"
+#include "match.h"
 #include "misc.h"
 #include "nat.h"
 #include "nuke.h"
@@ -72,6 +71,8 @@
 #include "ship.h"
 #include "version.h"
 
+static void ignore(void);
+static void crash_dump(void);
 static void create_pidfile(char *, pid_t);
 
 #if defined(_WIN32)
@@ -79,18 +80,33 @@ static void loc_NTInit(void);
 static void loc_NTTerm(void);
 #endif
 
+/*
+ * Lock to synchronize player threads with update and shutdown.
+ * Update and shutdown takes it exclusive, commands take it shared.
+ */
+empth_rwlock_t *play_lock;
+
 static char pidfname[] = "server.pid";
 
 /* Run as daemon?  If yes, detach from controlling terminal etc. */
 static int daemonize = 1;
 
+static void
+help(char *program_name, char *complaint)
+{
+    if (complaint)
+       fprintf(stderr, "%s: %s\n", program_name, complaint);
+    fprintf(stderr, "Try -h for help.\n");
+}
+
 static void
 print_usage(char *program_name)
 {
     printf("Usage: %s [OPTION]...\n"
-          "  -d              debug mode\n"
+          "  -d              debug mode, implies -E abort\n"
           "  -e CONFIG-FILE  configuration file\n"
           "                  (default %s)\n"
+          "  -E ACTION       what to do on oops: abort, crash-dump, nothing (default)\n"
           "  -h              display this help and exit\n"
 #ifdef _WIN32
           "  -i              install service `%s'\n"
@@ -98,10 +114,11 @@ print_usage(char *program_name)
 #endif
           "  -p              threading debug mode, implies -d\n"
 #ifdef _WIN32
-          "  -r              remove service `%s'\n"
-          "  -R NAME         remove service NAME\n"
+          "  -u              uninstall service `%s'\n"
+          "  -U NAME         uninstall service NAME\n"
 #endif
           "  -s              enable stack checking\n"
+          "  -R RANDOM-SEED  random seed\n"
           "  -v              display version information and exit\n",
           program_name, dflt_econfig
 #ifdef _WIN32
@@ -113,6 +130,8 @@ print_usage(char *program_name)
 int
 main(int argc, char **argv)
 {
+    static char *oops_key[] = { "abort", "crash-dump", "nothing", NULL };
+    static void (*oops_hndlr[])(void) = { abort, crash_dump, ignore };
     int flags = 0;
 #if defined(_WIN32)
     int install_service_set = 0;
@@ -121,25 +140,36 @@ main(int argc, char **argv)
     int remove_service_set = 0;
 #endif
     char *config_file = NULL;
-    int op, sig;
+    int op, idx, sig;
+    unsigned seed = time(NULL);
+
+    oops_handler = ignore;
 
 #ifdef _WIN32
-# define XOPTS "iI:rR:"
+# define XOPTS "iI:uU:"
 #else
 # define XOPTS
 #endif
-    while ((op = getopt(argc, argv, "de:hpsv" XOPTS)) != EOF) {
+    while ((op = getopt(argc, argv, "de:E:hpsR:v" XOPTS)) != EOF) {
        switch (op) {
        case 'p':
            flags |= EMPTH_PRINT;
            /* fall through */
        case 'd':
-           debug = 1;
+           oops_handler = abort;
            daemonize = 0;
            break;
        case 'e':
            config_file = optarg;
            break;
+       case 'E':
+           idx = stmtch(optarg, oops_key, 0, sizeof(*oops_key));
+           if (idx < 0) {
+               help(argv[0], "invalid argument for -E");
+               return EXIT_FAILURE;
+           }
+           oops_handler = oops_hndlr[idx];
+           break;
 #if defined(_WIN32)
        case 'I':
            service_name = optarg;
@@ -147,16 +177,19 @@ main(int argc, char **argv)
        case 'i':
            install_service_set++;
            break;
-       case 'R':
+       case 'U':
            service_name = optarg;
            /* fall through */
-       case 'r':
+       case 'u':
            remove_service_set++;
            break;
 #endif /* _WIN32 */
        case 's':
            flags |= EMPTH_STACKCHECK;
            break;
+       case 'R':
+           seed = strtoul(optarg, NULL, 10);
+           break;
        case 'v':
            printf("%s\n\n%s", version, legal);
            return EXIT_SUCCESS;
@@ -164,25 +197,25 @@ main(int argc, char **argv)
            print_usage(argv[0]);
            return EXIT_SUCCESS;
        default:
-           fprintf(stderr, "Try -h for help.\n");
+           help(argv[0], NULL);
            return EXIT_FAILURE;
        }
     }
 
 #if defined(_WIN32)
-    if ((debug || flags || config_file != NULL) &&
+    if ((!daemonize || flags || config_file != NULL) &&
        remove_service_set) {
        fprintf(stderr, "Can't use -p, -s, -d or -e with either "
-           "-r or -R options\n");
+           "-u or -U options\n");
        exit(EXIT_FAILURE);
     }
-    if ((debug || flags) && install_service_set) {
+    if ((!daemonize || flags) && install_service_set) {
        fprintf(stderr, "Can't use -d, -p or -s with either "
            "-i or -I options\n");
        exit(EXIT_FAILURE);
     }
     if (install_service_set && remove_service_set) {
-       fprintf(stderr, "Can't use both -r or -R and -i or -I "
+       fprintf(stderr, "Can't use both -u or -U and -i or -I "
            "options\n");
        exit(EXIT_FAILURE);
     }
@@ -196,26 +229,12 @@ main(int argc, char **argv)
     }
 #endif /* _WIN32 */
 
+    empfile_init();
     if (emp_config(config_file) < 0)
        exit(EXIT_FAILURE);
-    ef_init();
-    if (chdir(configdir)) {
-       fprintf(stderr, "Can't chdir to %s (%s)\n",
-               configdir, strerror(errno));
-       exit(EXIT_FAILURE);
-    }
-    if (chdir(builtindir)) {
-       fprintf(stderr, "Can't chdir to %s (%s)\n",
-               builtindir, strerror(errno));
-       exit(EXIT_FAILURE);
-    }
+    empfile_fixup();
     if (read_builtin_tables() < 0)
        exit(EXIT_FAILURE);
-    if (chdir(configdir)) {
-       fprintf(stderr, "Can't chdir to %s (%s)\n",
-               configdir, strerror(errno));
-       exit(EXIT_FAILURE);
-    }
     if (read_custom_tables() < 0)
        exit(EXIT_FAILURE);
     if (chdir(gamedir)) {
@@ -229,7 +248,7 @@ main(int argc, char **argv)
        return install_service(program_name, service_name, config_file);
 #endif /* _WIN32 */
 
-    init_server();
+    init_server(seed);
 
 #if defined(_WIN32)
     if (daemonize != 0) {
@@ -262,13 +281,16 @@ main(int argc, char **argv)
     }
 #endif /* !_WIN32 */
     start_server(flags);
+    journal_prng(seed);
 
     for (;;) {
        sig = empth_wait_for_signal();
 #ifdef SIGHUP
        if (sig == SIGHUP) {
-           journal_close();
-           journal_open();
+           /* if you make changes here, also update relo() */
+           journal_reopen();
+           update_reschedule();
+           logreopen();
            continue;
        }
 #endif
@@ -281,28 +303,57 @@ main(int argc, char **argv)
     return EXIT_SUCCESS;
 }
 
+static void
+ignore(void)
+{
+}
+
+static void
+crash_dump(void)
+{
+#ifdef _WIN32
+    logerror("Crash dump is not implemented");
+#else
+    pid_t pid;
+    int status;
+
+    fflush(NULL);
+    pid = fork();
+    if (pid < 0) {
+       logerror("Can't fork for crash dump (%s)", strerror(errno));
+       return;
+    }
+    if (pid == 0)
+       raise(SIGABRT);         /* child */
+
+    /* parent */
+    while (waitpid(pid, &status, 0) < 0) {
+       if (errno != EINTR) {
+           logerror("Can't get crash dumping child's status (%s)",
+                    strerror(errno));
+           return;
+       }
+    }
+    run_hook(post_crash_dump_hook, "post-crash-dump");
+    logerror("Crash dump complete");
+#endif
+}
 
 /*
  * Initialize for serving, acquire resources.
  */
 void
-init_server(void)
+init_server(unsigned seed)
 {
-    srandom(time(NULL));
+    srandom(seed);
 #if defined(_WIN32)
     loc_NTInit();
 #endif
-    update_policy_check();
     player_init();
     ef_init_srv();
     io_init();
     init_nreport();
 
-    if (opt_MOB_ACCESS) {
-       /* This fixes up mobility upon restart */
-       mobility_init();
-    }
-
     loginit("server");
 }
 
@@ -321,17 +372,10 @@ start_server(int flags)
 
     empth_init((void **)&player, flags);
 
-    empth_create(PP_ACCEPT, player_accept, (50 * 1024), flags,
-                "AcceptPlayers", "Accept network connections", 0);
-    empth_create(PP_KILLIDLE, player_kill_idle, (50 * 1024), flags,
-                "KillIdle", "Kills idle player connections", 0);
-    empth_create(PP_TIMESTAMP, delete_lostitems, (50 * 1024), flags,
-                "DeleteItems", "Deletes old lost items", 0);
-    if (opt_MOB_ACCESS) {
-       /* Start the mobility access check thread */
-       empth_create(PP_TIMESTAMP, mobility_check, (50 * 1024), flags,
-                    "MobilityCheck", "Writes the timestamp file", 0);
-    }
+    if (journal_startup() < 0)
+       exit(1);
+
+    empth_create(player_accept, 50 * 1024, flags, "AcceptPlayers", NULL);
 
     market_init();
     update_init();
@@ -347,6 +391,7 @@ finish_server(void)
 #if defined(_WIN32)
     loc_NTTerm();
 #endif
+    journal_shutdown();
     remove(pidfname);
 }
 
@@ -369,19 +414,19 @@ shutdwn(int sig)
 
     logerror("Shutdown commencing (cleaning up threads.)");
 
-    for (p = player_next(0); p != 0; p = player_next(p)) {
+    for (p = player_next(NULL); p; p = player_next(p)) {
        if (p->state != PS_PLAYING)
            continue;
        pr_flash(p, "Server shutting down...\n");
        p->state = PS_SHUTDOWN;
+       p->may_sleep = PLAYER_SLEEP_NEVER;
        p->aborted++;
        if (p->command) {
            pr_flash(p, "Shutdown aborting command\n");
        }
        empth_wakeup(p->proc);
     }
-    empth_rwlock_wrlock(update_lock);
-    /* rely on player_kill_idle() for killing hung player threads */
+    empth_rwlock_wrlock(play_lock);
     if (sig)
        logerror("Server shutting down on signal %d", sig);
     else