src/pazpar2.c

   1 /* This file is part of Pazpar2.
   2    Copyright (C) 2006-2012 Index Data
   3
   4 Pazpar2 is free software; you can redistribute it and/or modify it under
   5 the terms of the GNU General Public License as published by the Free
   6 Software Foundation; either version 2, or (at your option) any later
   7 version.
   8
   9 Pazpar2 is distributed in the hope that it will be useful, but WITHOUT ANY
  10 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12 for more details.
  13
  14 You should have received a copy of the GNU General Public License
  15 along with this program; if not, write to the Free Software
  16 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  17
  18 */
  19
  20 #if HAVE_CONFIG_H
  21 #include <config.h>
  22 #endif
  23 #ifdef WIN32
  24 #include <winsock.h>
  25 #include <direct.h>
  26 #endif
  27 #if HAVE_UNISTD_H
  28 #include <unistd.h>
  29 #endif
  30
  31 #include <signal.h>
  32 #include <assert.h>
  33
  34 #include "parameters.h"
  35 #include "session.h"
  36 #include "ppmutex.h"
  37 #include <yaz/daemon.h>
  38 #include <yaz/log.h>
  39 #include <yaz/options.h>
  40 #include <yaz/sc.h>
  41
  42 // #define MTRACE
  43 #ifdef MTRACE
  44 #include <mcheck.h>
  45 #endif
  46
  47 static struct conf_config *sc_stop_config = 0;
  48
  49 void child_handler(void *data)
  50 {
  51     struct conf_config *config = (struct conf_config *) data;
  52
  53     config_process_events(config);
  54
  55     config_destroy(config);
  56 }
  57
  58 static void show_version(void)
  59 {
  60     char yaz_version_str[80];
  61     printf("Pazpar2 " PACKAGE_VERSION
  62 #ifdef PAZPAR2_VERSION_SHA1
  63            " "
  64            PAZPAR2_VERSION_SHA1
  65 #endif
  66 "\n");
  67
  68     yaz_version(yaz_version_str, 0);
  69
  70     printf("Configuration:");
  71 #if YAZ_HAVE_ICU
  72     printf(" icu:enabled");
  73 #else
  74     printf(" icu:disabled");
  75 #endif
  76     printf(" yaz:%s", yaz_version_str);
  77     printf("\n");
  78     exit(0);
  79 }
  80
  81 #ifdef WIN32
  82 static int tcpip_init (void)
  83 {
  84     WORD requested;
  85     WSADATA wd;
  86
  87     requested = MAKEWORD(1, 1);
  88     if (WSAStartup(requested, &wd))
  89         return 0;
  90     return 1;
  91 }
  92 #endif
  93
  94
  95 static int sc_main(
  96     yaz_sc_t s,
  97     int argc, char **argv)
  98 {
  99     int daemon = 0;
 100     int ret;
 101     int log_file_in_use = 0;
 102     char *arg;
 103     const char *pidfile = 0;
 104     const char *uid = 0;
 105     const char *listener_override = 0;
 106     const char *config_fname = 0;
 107     const char *record_fname = 0;
 108     struct conf_config *config = 0;
 109     int test_mode = 0;
 110
 111 #ifndef WIN32
 112     if (signal(SIGPIPE, SIG_IGN) == SIG_ERR)
 113         yaz_log(YLOG_WARN|YLOG_ERRNO, "signal");
 114 #else
 115     tcpip_init();
 116 #endif
 117
 118     yaz_log_init_prefix("pazpar2");
 119     yaz_log_xml_errors(0, YLOG_WARN);
 120
 121     while ((ret = options("dDf:h:l:p:R:tu:v:VX", argv, argc, &arg)) != -2)
 122     {
 123         switch (ret)
 124         {
 125         case 'd':
 126             global_parameters.dump_records++;
 127             break;
 128         case 'D':
 129             daemon = 1;
 130             break;
 131         case 'f':
 132             config_fname = arg;
 133             break;
 134         case 'h':
 135             listener_override = arg;
 136             break;
 137         case 'l':
 138             yaz_log_init_file(arg);
 139             log_file_in_use = 1;
 140             break;
 141         case 'p':
 142             pidfile = arg;
 143             break;
 144         case 'R':
 145             record_fname = arg;
 146             global_parameters.predictable_sessions = 1;
 147             break;
 148         case 't':
 149             test_mode = 1;
 150             break;
 151         case 'u':
 152             uid = arg;
 153             break;
 154         case 'v':
 155             yaz_log_init_level(yaz_log_mask_str(arg));
 156             break;
 157         case 'V':
 158             show_version();
 159             break;
 160         case 'w':
 161             if (
 162 #ifdef WIN32
 163               _chdir
 164 #else
 165               chdir
 166 #endif
 167                 (arg))
 168             {
 169                 yaz_log(YLOG_FATAL|YLOG_ERRNO, "chdir %s", arg);
 170                 return 1;
 171             }
 172             break;
 173         case 'X':
 174             global_parameters.debug_mode++;
 175             global_parameters.predictable_sessions = 1;
 176             break;
 177         default:
 178             fprintf(stderr, "Usage: pazpar2\n"
 179                     "    -d                      Show internal records\n"
 180                     "    -D                      Daemon mode (background)\n"
 181                     "    -f configfile           Configuration\n"
 182                     "    -h [host:]port          Listener port\n"
 183                     "    -l file                 Log to file\n"
 184                     "    -p pidfile              PID file\n"
 185                     "    -R recfile              HTTP recording file\n"
 186                     "    -t                      Test configuration\n"
 187                     "    -u uid                  Change user to uid\n"
 188                     "    -V                      Show version\n"
 189                     "    -v level                Set log level\n"
 190                     "    -w dir                  Working directory\n"
 191                     "    -X                      Debug mode\n"
 192 #ifdef WIN32
 193                     "    -install                Install windows service\n"
 194                     "    -remove                 Remove windows service\n"
 195 #endif
 196                 );
 197             return 1;
 198         }
 199     }
 200     if (!config_fname)
 201     {
 202         yaz_log(YLOG_FATAL, "Configuration must be given with option -f");
 203         return 1;
 204     }
 205     pazpar2_mutex_init();
 206
 207     config = config_create(config_fname, global_parameters.dump_records);
 208     if (!config)
 209         return 1;
 210     sc_stop_config = config;
 211     if (test_mode)
 212     {
 213         yaz_log(YLOG_LOG, "Configuration OK");
 214         config_destroy(config);
 215     }
 216     else
 217     {
 218         yaz_log(YLOG_LOG, "Pazpar2 " VERSION  " "
 219 #ifdef PAZPAR2_VERSION_SHA1
 220                 PAZPAR2_VERSION_SHA1
 221 #else
 222                 "-"
 223 #endif
 224                 " started");
 225         if (daemon && !log_file_in_use)
 226         {
 227             yaz_log(YLOG_FATAL, "Logfile must be given (option -l) for daemon "
 228                     "mode");
 229             return 1;
 230         }
 231         ret = config_start_listeners(config, listener_override, record_fname);
 232         if (ret)
 233             return ret; /* error starting http listener */
 234
 235         yaz_sc_running(s);
 236
 237         yaz_daemon("pazpar2",
 238                    (global_parameters.debug_mode ? YAZ_DAEMON_DEBUG : 0) +
 239                    (daemon ? YAZ_DAEMON_FORK : 0) + YAZ_DAEMON_KEEPALIVE,
 240                    child_handler, config /* child_data */,
 241                    pidfile, uid);
 242     }
 243     return 0;
 244 }
 245
 246
 247 static void sc_stop(yaz_sc_t s)
 248 {
 249     config_stop_listeners(sc_stop_config);
 250 }
 251
 252 int main(int argc, char **argv)
 253 {
 254     int ret;
 255     yaz_sc_t s = yaz_sc_create("pazpar2", "Pazpar2");
 256
 257 #ifdef MTRACE
 258     mtrace();
 259 #endif
 260
 261     ret = yaz_sc_program(s, argc, argv, sc_main, sc_stop);
 262
 263     yaz_sc_destroy(&s);
 264
 265 #ifdef MTRACE
 266     muntrace();
 267 #endif
 268
 269
 270     exit(ret);
 271 }
 272
 273 /*
 274  * Local variables:
 275  * c-basic-offset: 4
 276  * c-file-style: "Stroustrup"
 277  * indent-tabs-mode: nil
 278  * End:
 279  * vim: shiftwidth=4 tabstop=8 expandtab
 280  */
 281