Add mtrace start/stop. Requires MTRACE define
[pazpar2-moved-to-github.git] / src / pazpar2.c
1 /* This file is part of Pazpar2.
2    Copyright (C) 2006-2011 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 #endif
26
27 #include <signal.h>
28 #include <assert.h>
29
30 #include "parameters.h"
31 #include "session.h"
32 #include "ppmutex.h"
33 #include <yaz/daemon.h>
34 #include <yaz/log.h>
35 #include <yaz/options.h>
36 #include <yaz/sc.h>
37
38 // #define MTRACE
39 #ifdef MTRACE
40 #include <mcheck.h>
41 #endif
42
43 static struct conf_config *sc_stop_config = 0;
44
45 void child_handler(void *data)
46 {
47     struct conf_config *config = (struct conf_config *) data;
48
49     config_process_events(config);
50
51     config_destroy(config);
52 }
53
54 static void show_version(void)
55 {
56     char yaz_version_str[80];
57     printf("Pazpar2 " PACKAGE_VERSION 
58 #ifdef PAZPAR2_VERSION_SHA1
59            " "
60            PAZPAR2_VERSION_SHA1
61 #endif
62 "\n");
63
64     yaz_version(yaz_version_str, 0);
65
66     printf("Configuration:");
67 #if YAZ_HAVE_ICU
68     printf(" icu:enabled");
69 #else
70     printf(" icu:disabled");
71 #endif
72     printf(" yaz:%s", yaz_version_str);
73     printf("\n");
74     exit(0);
75 }            
76
77 #ifdef WIN32
78 static int tcpip_init (void)
79 {
80     WORD requested;
81     WSADATA wd;
82
83     requested = MAKEWORD(1, 1);
84     if (WSAStartup(requested, &wd))
85         return 0;
86     return 1;
87 }
88 #endif
89
90
91 static int sc_main(
92     yaz_sc_t s, 
93     int argc, char **argv)
94 {
95     int daemon = 0;
96     int ret;
97     int log_file_in_use = 0;
98     char *arg;
99     const char *pidfile = 0;
100     const char *uid = 0;
101     const char *listener_override = 0;
102     const char *config_fname = 0;
103     struct conf_config *config = 0;
104     int test_mode = 0;
105
106 #ifndef WIN32
107     if (signal(SIGPIPE, SIG_IGN) == SIG_ERR)
108         yaz_log(YLOG_WARN|YLOG_ERRNO, "signal");
109 #else
110     tcpip_init();
111 #endif
112
113     yaz_log_init_prefix("pazpar2");
114     yaz_log_xml_errors(0, YLOG_WARN);
115
116     while ((ret = options("dDf:h:l:p:tu:v:VX", argv, argc, &arg)) != -2)
117     {
118         switch (ret)
119         {
120         case 'd':
121             global_parameters.dump_records = 1;
122             break;
123         case 'D':
124             daemon = 1;
125             break;
126         case 'f':
127             config_fname = arg;
128             break;
129         case 'h':
130             listener_override = arg;
131             break;
132         case 'l':
133             yaz_log_init_file(arg);
134             log_file_in_use = 1;
135             break;
136         case 'p':
137             pidfile = arg;
138             break;
139         case 't':
140             test_mode = 1;
141             break;
142         case 'u':
143             uid = arg;
144             break;
145         case 'v':
146             yaz_log_init_level(yaz_log_mask_str(arg));
147             break;
148         case 'V':
149             show_version();
150         case 'X':
151             global_parameters.debug_mode++;
152             break;
153         default:
154             fprintf(stderr, "Usage: pazpar2\n"
155                     "    -d                      Show internal records\n"
156                     "    -D                      Daemon mode (background)\n"
157                     "    -f configfile           Configuration\n"
158                     "    -h [host:]port          Listener port\n"
159                     "    -l file                 Log to file\n"
160                     "    -p pidfile              PID file\n"
161                     "    -t                      Test configuration\n"
162                     "    -u uid                  Change user to uid\n"
163                     "    -V                      Show version\n"
164                     "    -v level                Set log level\n"
165                     "    -X                      Debug mode\n"
166 #ifdef WIN32
167                     "    -install                Install windows service\n"
168                     "    -remove                 Remove windows service\n"
169 #endif
170                 );
171             return 1;
172         }
173     }
174     if (!config_fname)
175     {
176         yaz_log(YLOG_FATAL, "Configuration must be given with option -f");
177         return 1;
178     }
179     pazpar2_mutex_init();
180     
181     config = config_create(config_fname, global_parameters.dump_records);
182     if (!config)
183         return 1;
184     sc_stop_config = config;
185     if (test_mode)
186     {
187         yaz_log(YLOG_LOG, "Configuration OK");
188         config_destroy(config);
189     }
190     else
191     {
192         yaz_log(YLOG_LOG, "Pazpar2 " VERSION  " "
193 #ifdef PAZPAR2_VERSION_SHA1
194                 PAZPAR2_VERSION_SHA1
195 #else
196                 "-"
197 #endif
198                 " started");
199         if (daemon && !log_file_in_use)
200         {
201             yaz_log(YLOG_FATAL, "Logfile must be given (option -l) for daemon "
202                     "mode");
203             return 1;
204         }
205         ret = config_start_listeners(config, listener_override);
206         if (ret)
207             return ret; /* error starting http listener */
208         
209         yaz_sc_running(s);
210         
211         yaz_daemon("pazpar2",
212                    (global_parameters.debug_mode ? YAZ_DAEMON_DEBUG : 0) +
213                    (daemon ? YAZ_DAEMON_FORK : 0) + YAZ_DAEMON_KEEPALIVE,
214                    child_handler, config /* child_data */,
215                    pidfile, uid);
216     }
217     return 0;
218 }
219
220
221 static void sc_stop(yaz_sc_t s)
222 {
223     config_stop_listeners(sc_stop_config);
224 }
225
226 int main(int argc, char **argv)
227 {
228     int ret;
229     yaz_sc_t s = yaz_sc_create("pazpar2", "Pazpar2");
230     
231 #ifdef MTRACE
232     mtrace();
233 #endif
234
235     ret = yaz_sc_program(s, argc, argv, sc_main, sc_stop);
236
237     yaz_sc_destroy(&s);
238
239 #ifdef MTRACE
240     muntrace();
241 #endif
242
243
244     exit(ret);
245 }
246
247 /*
248  * Local variables:
249  * c-basic-offset: 4
250  * c-file-style: "Stroustrup"
251  * indent-tabs-mode: nil
252  * End:
253  * vim: shiftwidth=4 tabstop=8 expandtab
254  */
255