Facility to change working directory for daemon
[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     const char *record_fname = 0;
104     struct conf_config *config = 0;
105     int test_mode = 0;
106
107 #ifndef WIN32
108     if (signal(SIGPIPE, SIG_IGN) == SIG_ERR)
109         yaz_log(YLOG_WARN|YLOG_ERRNO, "signal");
110 #else
111     tcpip_init();
112 #endif
113
114     yaz_log_init_prefix("pazpar2");
115     yaz_log_xml_errors(0, YLOG_WARN);
116
117     while ((ret = options("dDf:h:l:p:R:tu:v:VX", argv, argc, &arg)) != -2)
118     {
119         switch (ret)
120         {
121         case 'd':
122             global_parameters.dump_records = 1;
123             break;
124         case 'D':
125             daemon = 1;
126             break;
127         case 'f':
128             config_fname = arg;
129             break;
130         case 'h':
131             listener_override = arg;
132             break;
133         case 'l':
134             yaz_log_init_file(arg);
135             log_file_in_use = 1;
136             break;
137         case 'p':
138             pidfile = arg;
139             break;
140         case 'R':
141             record_fname = arg;
142             global_parameters.predictable_sessions = 1;
143             break;
144         case 't':
145             test_mode = 1;
146             break;
147         case 'u':
148             uid = arg;
149             break;
150         case 'v':
151             yaz_log_init_level(yaz_log_mask_str(arg));
152             break;
153         case 'V':
154             show_version();
155             break;
156         case 'w':
157             if (chdir(arg))
158             {
159                 yaz_log(YLOG_FATAL|YLOG_ERRNO, "chdir %s", arg);
160                 return 1;
161             }
162             break;
163         case 'X':
164             global_parameters.debug_mode++;
165             global_parameters.predictable_sessions = 1;
166             break;
167         default:
168             fprintf(stderr, "Usage: pazpar2\n"
169                     "    -d                      Show internal records\n"
170                     "    -D                      Daemon mode (background)\n"
171                     "    -f configfile           Configuration\n"
172                     "    -h [host:]port          Listener port\n"
173                     "    -l file                 Log to file\n"
174                     "    -p pidfile              PID file\n"
175                     "    -R recfile              HTTP recording file\n"
176                     "    -t                      Test configuration\n"
177                     "    -u uid                  Change user to uid\n"
178                     "    -V                      Show version\n"
179                     "    -v level                Set log level\n"
180                     "    -w dir                  Working directory\n"
181                     "    -X                      Debug mode\n"
182 #ifdef WIN32
183                     "    -install                Install windows service\n"
184                     "    -remove                 Remove windows service\n"
185 #endif
186                 );
187             return 1;
188         }
189     }
190     if (!config_fname)
191     {
192         yaz_log(YLOG_FATAL, "Configuration must be given with option -f");
193         return 1;
194     }
195     pazpar2_mutex_init();
196     
197     config = config_create(config_fname, global_parameters.dump_records);
198     if (!config)
199         return 1;
200     sc_stop_config = config;
201     if (test_mode)
202     {
203         yaz_log(YLOG_LOG, "Configuration OK");
204         config_destroy(config);
205     }
206     else
207     {
208         yaz_log(YLOG_LOG, "Pazpar2 " VERSION  " "
209 #ifdef PAZPAR2_VERSION_SHA1
210                 PAZPAR2_VERSION_SHA1
211 #else
212                 "-"
213 #endif
214                 " started");
215         if (daemon && !log_file_in_use)
216         {
217             yaz_log(YLOG_FATAL, "Logfile must be given (option -l) for daemon "
218                     "mode");
219             return 1;
220         }
221         ret = config_start_listeners(config, listener_override, record_fname);
222         if (ret)
223             return ret; /* error starting http listener */
224         
225         yaz_sc_running(s);
226         
227         yaz_daemon("pazpar2",
228                    (global_parameters.debug_mode ? YAZ_DAEMON_DEBUG : 0) +
229                    (daemon ? YAZ_DAEMON_FORK : 0) + YAZ_DAEMON_KEEPALIVE,
230                    child_handler, config /* child_data */,
231                    pidfile, uid);
232     }
233     return 0;
234 }
235
236
237 static void sc_stop(yaz_sc_t s)
238 {
239     config_stop_listeners(sc_stop_config);
240 }
241
242 int main(int argc, char **argv)
243 {
244     int ret;
245     yaz_sc_t s = yaz_sc_create("pazpar2", "Pazpar2");
246     
247 #ifdef MTRACE
248     mtrace();
249 #endif
250
251     ret = yaz_sc_program(s, argc, argv, sc_main, sc_stop);
252
253     yaz_sc_destroy(&s);
254
255 #ifdef MTRACE
256     muntrace();
257 #endif
258
259
260     exit(ret);
261 }
262
263 /*
264  * Local variables:
265  * c-basic-offset: 4
266  * c-file-style: "Stroustrup"
267  * indent-tabs-mode: nil
268  * End:
269  * vim: shiftwidth=4 tabstop=8 expandtab
270  */
271