s4/server: simplify and replace bad portable gettimeofday by time
[Samba/wip.git] / source4 / smbd / server.c
blob6acccd14ae4c9888d00e08647b7c14ba931ea36a
1 /*
2 Unix SMB/CIFS implementation.
4 Main SMB server routines
6 Copyright (C) Andrew Tridgell 1992-2005
7 Copyright (C) Martin Pool 2002
8 Copyright (C) Jelmer Vernooij 2002
9 Copyright (C) James J Myers 2003 <myersjj@samba.org>
11 This program is free software; you can redistribute it and/or modify
12 it under the terms of the GNU General Public License as published by
13 the Free Software Foundation; either version 3 of the License, or
14 (at your option) any later version.
16 This program is distributed in the hope that it will be useful,
17 but WITHOUT ANY WARRANTY; without even the implied warranty of
18 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 GNU General Public License for more details.
21 You should have received a copy of the GNU General Public License
22 along with this program. If not, see <http://www.gnu.org/licenses/>.
25 #include "includes.h"
26 #include "lib/events/events.h"
27 #include "version.h"
28 #include "lib/cmdline/popt_common.h"
29 #include "system/dir.h"
30 #include "system/filesys.h"
31 #include "ntvfs/ntvfs.h"
32 #include "ntptr/ntptr.h"
33 #include "auth/gensec/gensec.h"
34 #include "libcli/auth/schannel.h"
35 #include "smbd/process_model.h"
36 #include "param/secrets.h"
37 #include "smbd/pidfile.h"
38 #include "param/param.h"
39 #include "dsdb/samdb/samdb.h"
40 #include "auth/session.h"
41 #include "lib/messaging/irpc.h"
42 #include "librpc/gen_ndr/ndr_irpc.h"
43 #include "cluster/cluster.h"
44 #include "dynconfig/dynconfig.h"
47 recursively delete a directory tree
49 static void recursive_delete(const char *path)
51 DIR *dir;
52 struct dirent *de;
54 dir = opendir(path);
55 if (!dir) {
56 return;
59 for (de=readdir(dir);de;de=readdir(dir)) {
60 char *fname;
61 struct stat st;
63 if (ISDOT(de->d_name) || ISDOTDOT(de->d_name)) {
64 continue;
67 fname = talloc_asprintf(path, "%s/%s", path, de->d_name);
68 if (stat(fname, &st) != 0) {
69 continue;
71 if (S_ISDIR(st.st_mode)) {
72 recursive_delete(fname);
73 talloc_free(fname);
74 continue;
76 if (unlink(fname) != 0) {
77 DEBUG(0,("Unabled to delete '%s' - %s\n",
78 fname, strerror(errno)));
79 smb_panic("unable to cleanup tmp files");
81 talloc_free(fname);
83 closedir(dir);
87 cleanup temporary files. This is the new alternative to
88 TDB_CLEAR_IF_FIRST. Unfortunately TDB_CLEAR_IF_FIRST is not
89 efficient on unix systems due to the lack of scaling of the byte
90 range locking system. So instead of putting the burden on tdb to
91 cleanup tmp files, this function deletes them.
93 static void cleanup_tmp_files(struct loadparm_context *lp_ctx)
95 char *path;
96 TALLOC_CTX *mem_ctx = talloc_new(NULL);
98 path = smbd_tmp_path(mem_ctx, lp_ctx, NULL);
100 recursive_delete(path);
101 talloc_free(mem_ctx);
104 static void sig_hup(int sig)
106 debug_schedule_reopen_logs();
109 static void sig_term(int sig)
111 #if HAVE_GETPGRP
112 static int done_sigterm;
113 if (done_sigterm == 0 && getpgrp() == getpid()) {
114 DEBUG(0,("SIGTERM: killing children\n"));
115 done_sigterm = 1;
116 kill(-getpgrp(), SIGTERM);
118 #endif
119 DEBUG(0,("Exiting pid %d on SIGTERM\n", (int)getpid()));
120 exit(127);
124 setup signal masks
126 static void setup_signals(void)
128 /* we are never interested in SIGPIPE */
129 BlockSignals(true,SIGPIPE);
131 #if defined(SIGFPE)
132 /* we are never interested in SIGFPE */
133 BlockSignals(true,SIGFPE);
134 #endif
136 /* We are no longer interested in USR1 */
137 BlockSignals(true, SIGUSR1);
139 #if defined(SIGUSR2)
140 /* We are no longer interested in USR2 */
141 BlockSignals(true,SIGUSR2);
142 #endif
144 /* POSIX demands that signals are inherited. If the invoking process has
145 * these signals masked, we will have problems, as we won't receive them. */
146 BlockSignals(false, SIGHUP);
147 BlockSignals(false, SIGTERM);
149 CatchSignal(SIGHUP, sig_hup);
150 CatchSignal(SIGTERM, sig_term);
154 handle io on stdin
156 static void server_stdin_handler(struct tevent_context *event_ctx, struct tevent_fd *fde,
157 uint16_t flags, void *private_data)
159 const char *binary_name = (const char *)private_data;
160 uint8_t c;
161 if (read(0, &c, 1) == 0) {
162 DEBUG(0,("%s: EOF on stdin - terminating\n", binary_name));
163 #if HAVE_GETPGRP
164 if (getpgrp() == getpid()) {
165 DEBUG(0,("Sending SIGTERM from pid %d\n", (int)getpid()));
166 kill(-getpgrp(), SIGTERM);
168 #endif
169 exit(0);
174 die if the user selected maximum runtime is exceeded
176 _NORETURN_ static void max_runtime_handler(struct tevent_context *ev,
177 struct tevent_timer *te,
178 struct timeval t, void *private_data)
180 const char *binary_name = (const char *)private_data;
181 DEBUG(0,("%s: maximum runtime exceeded - terminating, current ts: %d\n", binary_name, time(NULL)));
182 exit(0);
186 pre-open the key databases. This saves a lot of time in child
187 processes
189 static void prime_ldb_databases(struct tevent_context *event_ctx)
191 TALLOC_CTX *db_context;
192 db_context = talloc_new(event_ctx);
194 samdb_connect(db_context, event_ctx, cmdline_lp_ctx, system_session(cmdline_lp_ctx), 0);
195 privilege_connect(db_context, cmdline_lp_ctx);
197 /* we deliberately leave these open, which allows them to be
198 * re-used in ldb_wrap_connect() */
203 called when a fatal condition occurs in a child task
205 static NTSTATUS samba_terminate(struct irpc_message *msg,
206 struct samba_terminate *r)
208 DEBUG(0,("samba_terminate: %s\n", r->in.reason));
209 exit(1);
213 setup messaging for the top level samba (parent) task
215 static NTSTATUS setup_parent_messaging(struct tevent_context *event_ctx,
216 struct loadparm_context *lp_ctx)
218 struct imessaging_context *msg;
219 NTSTATUS status;
221 msg = imessaging_init(talloc_autofree_context(),
222 lpcfg_imessaging_path(event_ctx, lp_ctx),
223 cluster_id(0, SAMBA_PARENT_TASKID), event_ctx);
224 NT_STATUS_HAVE_NO_MEMORY(msg);
226 irpc_add_name(msg, "samba");
228 status = IRPC_REGISTER(msg, irpc, SAMBA_TERMINATE,
229 samba_terminate, NULL);
231 return status;
236 show build info
238 static void show_build(void)
240 #define CONFIG_OPTION(n) { #n, dyn_ ## n }
241 struct {
242 const char *name;
243 const char *value;
244 } config_options[] = {
245 CONFIG_OPTION(BINDIR),
246 CONFIG_OPTION(SBINDIR),
247 CONFIG_OPTION(CONFIGFILE),
248 CONFIG_OPTION(NCALRPCDIR),
249 CONFIG_OPTION(LOGFILEBASE),
250 CONFIG_OPTION(LMHOSTSFILE),
251 CONFIG_OPTION(DATADIR),
252 CONFIG_OPTION(MODULESDIR),
253 CONFIG_OPTION(LOCKDIR),
254 CONFIG_OPTION(PIDDIR),
255 CONFIG_OPTION(PRIVATE_DIR),
256 CONFIG_OPTION(SWATDIR),
257 CONFIG_OPTION(CODEPAGEDIR),
258 CONFIG_OPTION(SETUPDIR),
259 CONFIG_OPTION(WINBINDD_SOCKET_DIR),
260 CONFIG_OPTION(WINBINDD_PRIVILEGED_SOCKET_DIR),
261 CONFIG_OPTION(NTP_SIGND_SOCKET_DIR),
262 { NULL, NULL}
264 int i;
266 printf("Samba version: %s\n", SAMBA_VERSION_STRING);
267 printf("Build environment:\n");
268 #ifdef BUILD_SYSTEM
269 printf(" Build host: %s\n", BUILD_SYSTEM);
270 #endif
272 printf("Paths:\n");
273 for (i=0; config_options[i].name; i++) {
274 printf(" %s: %s\n", config_options[i].name, config_options[i].value);
277 exit(0);
281 main server.
283 static int binary_smbd_main(const char *binary_name, int argc, const char *argv[])
285 bool opt_daemon = false;
286 bool opt_interactive = false;
287 int opt;
288 poptContext pc;
289 #define _MODULE_PROTO(init) extern NTSTATUS init(void);
290 STATIC_service_MODULES_PROTO;
291 init_module_fn static_init[] = { STATIC_service_MODULES };
292 init_module_fn *shared_init;
293 struct tevent_context *event_ctx;
294 uint16_t stdin_event_flags;
295 NTSTATUS status;
296 const char *model = "standard";
297 int max_runtime = 0;
298 enum {
299 OPT_DAEMON = 1000,
300 OPT_INTERACTIVE,
301 OPT_PROCESS_MODEL,
302 OPT_SHOW_BUILD
304 struct poptOption long_options[] = {
305 POPT_AUTOHELP
306 {"daemon", 'D', POPT_ARG_NONE, NULL, OPT_DAEMON,
307 "Become a daemon (default)", NULL },
308 {"interactive", 'i', POPT_ARG_NONE, NULL, OPT_INTERACTIVE,
309 "Run interactive (not a daemon)", NULL},
310 {"model", 'M', POPT_ARG_STRING, NULL, OPT_PROCESS_MODEL,
311 "Select process model", "MODEL"},
312 {"maximum-runtime",0, POPT_ARG_INT, &max_runtime, 0,
313 "set maximum runtime of the server process, till autotermination", "seconds"},
314 {"show-build", 'b', POPT_ARG_NONE, NULL, OPT_SHOW_BUILD, "show build info", NULL },
315 POPT_COMMON_SAMBA
316 POPT_COMMON_VERSION
317 { NULL }
320 pc = poptGetContext(binary_name, argc, argv, long_options, 0);
321 while((opt = poptGetNextOpt(pc)) != -1) {
322 switch(opt) {
323 case OPT_DAEMON:
324 opt_daemon = true;
325 break;
326 case OPT_INTERACTIVE:
327 opt_interactive = true;
328 break;
329 case OPT_PROCESS_MODEL:
330 model = poptGetOptArg(pc);
331 break;
332 case OPT_SHOW_BUILD:
333 show_build();
334 break;
335 default:
336 fprintf(stderr, "\nInvalid option %s: %s\n\n",
337 poptBadOption(pc, 0), poptStrerror(opt));
338 poptPrintUsage(pc, stderr, 0);
339 return 1;
343 if (opt_daemon && opt_interactive) {
344 fprintf(stderr,"\nERROR: "
345 "Option -i|--interactive is not allowed together with -D|--daemon\n\n");
346 poptPrintUsage(pc, stderr, 0);
347 return 1;
348 } else if (!opt_interactive) {
349 /* default is --daemon */
350 opt_daemon = true;
353 poptFreeContext(pc);
355 setup_logging(binary_name, opt_interactive?DEBUG_STDOUT:DEBUG_FILE);
356 setup_signals();
358 /* we want total control over the permissions on created files,
359 so set our umask to 0 */
360 umask(0);
362 DEBUG(0,("%s version %s started.\n", binary_name, SAMBA_VERSION_STRING));
363 DEBUGADD(0,("Copyright Andrew Tridgell and the Samba Team 1992-2011\n"));
365 if (sizeof(uint16_t) < 2 || sizeof(uint32_t) < 4 || sizeof(uint64_t) < 8) {
366 DEBUG(0,("ERROR: Samba is not configured correctly for the word size on your machine\n"));
367 DEBUGADD(0,("sizeof(uint16_t) = %u, sizeof(uint32_t) %u, sizeof(uint64_t) = %u\n",
368 (unsigned int)sizeof(uint16_t), (unsigned int)sizeof(uint32_t), (unsigned int)sizeof(uint64_t)));
369 return 1;
372 if (opt_daemon) {
373 DEBUG(3,("Becoming a daemon.\n"));
374 become_daemon(true, false, false);
377 cleanup_tmp_files(cmdline_lp_ctx);
379 if (!directory_exist(lpcfg_lockdir(cmdline_lp_ctx))) {
380 mkdir(lpcfg_lockdir(cmdline_lp_ctx), 0755);
383 pidfile_create(lpcfg_piddir(cmdline_lp_ctx), binary_name);
385 /* Set up a database to hold a random seed, in case we don't
386 * have /dev/urandom */
387 if (!randseed_init(talloc_autofree_context(), cmdline_lp_ctx)) {
388 return 1;
391 if (lpcfg_server_role(cmdline_lp_ctx) == ROLE_DOMAIN_CONTROLLER) {
392 if (!open_schannel_session_store(talloc_autofree_context(), lpcfg_private_dir(cmdline_lp_ctx))) {
393 DEBUG(0,("ERROR: Samba cannot open schannel store for secured NETLOGON operations.\n"));
394 exit(1);
398 gensec_init(); /* FIXME: */
400 ntptr_init(); /* FIXME: maybe run this in the initialization function
401 of the spoolss RPC server instead? */
403 ntvfs_init(cmdline_lp_ctx); /* FIXME: maybe run this in the initialization functions
404 of the SMB[,2] server instead? */
406 process_model_init(cmdline_lp_ctx);
408 shared_init = load_samba_modules(NULL, "service");
410 run_init_functions(static_init);
411 run_init_functions(shared_init);
413 talloc_free(shared_init);
415 /* the event context is the top level structure in smbd. Everything else
416 should hang off that */
417 event_ctx = s4_event_context_init(talloc_autofree_context());
419 if (event_ctx == NULL) {
420 DEBUG(0,("Initializing event context failed\n"));
421 return 1;
424 if (opt_interactive) {
425 /* terminate when stdin goes away */
426 stdin_event_flags = TEVENT_FD_READ;
427 } else {
428 /* stay alive forever */
429 stdin_event_flags = 0;
432 /* catch EOF on stdin */
433 #ifdef SIGTTIN
434 signal(SIGTTIN, SIG_IGN);
435 #endif
436 tevent_add_fd(event_ctx, event_ctx, 0, stdin_event_flags,
437 server_stdin_handler,
438 discard_const(binary_name));
440 if (max_runtime) {
441 DEBUG(0,("Called with maxruntime %d - current ts %d\n", max_runtime, time(NULL)));
442 tevent_add_timer(event_ctx, event_ctx,
443 timeval_current_ofs(max_runtime, 0),
444 max_runtime_handler,
445 discard_const(binary_name));
448 prime_ldb_databases(event_ctx);
450 status = setup_parent_messaging(event_ctx, cmdline_lp_ctx);
451 if (!NT_STATUS_IS_OK(status)) {
452 DEBUG(0,("Failed to setup parent messaging - %s\n", nt_errstr(status)));
453 return 1;
456 DEBUG(0,("%s: using '%s' process model\n", binary_name, model));
458 status = server_service_startup(event_ctx, cmdline_lp_ctx, model,
459 lpcfg_server_services(cmdline_lp_ctx));
460 if (!NT_STATUS_IS_OK(status)) {
461 DEBUG(0,("Starting Services failed - %s\n", nt_errstr(status)));
462 return 1;
465 /* wait for events - this is where smbd sits for most of its
466 life */
467 tevent_loop_wait(event_ctx);
469 /* as everything hangs off this event context, freeing it
470 should initiate a clean shutdown of all services */
471 talloc_free(event_ctx);
473 return 0;
476 int main(int argc, const char *argv[])
478 return binary_smbd_main("samba", argc, argv);