s3: Lift the smbd_messaging_context from start_background_queue
[amitay/samba.git] / source3 / smbd / server.c
1 /*
2    Unix SMB/CIFS implementation.
3    Main SMB server routines
4    Copyright (C) Andrew Tridgell                1992-1998
5    Copyright (C) Martin Pool                    2002
6    Copyright (C) Jelmer Vernooij                2002-2003
7    Copyright (C) Volker Lendecke                1993-2007
8    Copyright (C) Jeremy Allison                 1993-2007
9
10    This program is free software; you can redistribute it and/or modify
11    it under the terms of the GNU General Public License as published by
12    the Free Software Foundation; either version 3 of the License, or
13    (at your option) any later version.
14
15    This program is distributed in the hope that it will be useful,
16    but WITHOUT ANY WARRANTY; without even the implied warranty of
17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18    GNU General Public License for more details.
19
20    You should have received a copy of the GNU General Public License
21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
22 */
23
24 #include "includes.h"
25 #include "popt_common.h"
26 #include "smbd/globals.h"
27 #include "librpc/gen_ndr/messaging.h"
28 #include "registry.h"
29 #include "libcli/auth/schannel.h"
30 #include "secrets.h"
31
32 static_decl_rpc;
33
34 #ifdef WITH_DFS
35 extern int dcelogin_atmost_once;
36 #endif /* WITH_DFS */
37
38 int smbd_server_fd(void)
39 {
40         return server_fd;
41 }
42
43 static void smbd_set_server_fd(int fd)
44 {
45         server_fd = fd;
46 }
47
48 int get_client_fd(void)
49 {
50         return server_fd;
51 }
52
53 struct event_context *smbd_event_context(void)
54 {
55         return server_event_context();
56 }
57
58 /*******************************************************************
59  What to do when smb.conf is updated.
60  ********************************************************************/
61
62 static void smb_conf_updated(struct messaging_context *msg,
63                              void *private_data,
64                              uint32_t msg_type,
65                              struct server_id server_id,
66                              DATA_BLOB *data)
67 {
68         DEBUG(10,("smb_conf_updated: Got message saying smb.conf was "
69                   "updated. Reloading.\n"));
70         change_to_root_user();
71         reload_services(smbd_messaging_context(), False);
72 }
73
74
75 /*******************************************************************
76  Delete a statcache entry.
77  ********************************************************************/
78
79 static void smb_stat_cache_delete(struct messaging_context *msg,
80                                   void *private_data,
81                                   uint32_t msg_tnype,
82                                   struct server_id server_id,
83                                   DATA_BLOB *data)
84 {
85         const char *name = (const char *)data->data;
86         DEBUG(10,("smb_stat_cache_delete: delete name %s\n", name));
87         stat_cache_delete(name);
88 }
89
90 /****************************************************************************
91   Send a SIGTERM to our process group.
92 *****************************************************************************/
93
94 static void  killkids(void)
95 {
96         if(am_parent) kill(0,SIGTERM);
97 }
98
99 /****************************************************************************
100  Process a sam sync message - not sure whether to do this here or
101  somewhere else.
102 ****************************************************************************/
103
104 static void msg_sam_sync(struct messaging_context *msg,
105                          void *private_data,
106                          uint32_t msg_type,
107                          struct server_id server_id,
108                          DATA_BLOB *data)
109 {
110         DEBUG(10, ("** sam sync message received, ignoring\n"));
111 }
112
113 static void msg_exit_server(struct messaging_context *msg,
114                             void *private_data,
115                             uint32_t msg_type,
116                             struct server_id server_id,
117                             DATA_BLOB *data)
118 {
119         DEBUG(3, ("got a SHUTDOWN message\n"));
120         exit_server_cleanly(NULL);
121 }
122
123 #ifdef DEVELOPER
124 static void msg_inject_fault(struct messaging_context *msg,
125                              void *private_data,
126                              uint32_t msg_type,
127                              struct server_id src,
128                              DATA_BLOB *data)
129 {
130         int sig;
131
132         if (data->length != sizeof(sig)) {
133                 DEBUG(0, ("Process %s sent bogus signal injection request\n",
134                           procid_str_static(&src)));
135                 return;
136         }
137
138         sig = *(int *)data->data;
139         if (sig == -1) {
140                 exit_server("internal error injected");
141                 return;
142         }
143
144 #if HAVE_STRSIGNAL
145         DEBUG(0, ("Process %s requested injection of signal %d (%s)\n",
146                   procid_str_static(&src), sig, strsignal(sig)));
147 #else
148         DEBUG(0, ("Process %s requested injection of signal %d\n",
149                   procid_str_static(&src), sig));
150 #endif
151
152         kill(sys_getpid(), sig);
153 }
154 #endif /* DEVELOPER */
155
156 /*
157  * Parent smbd process sets its own debug level first and then
158  * sends a message to all the smbd children to adjust their debug
159  * level to that of the parent.
160  */
161
162 static void smbd_msg_debug(struct messaging_context *msg_ctx,
163                            void *private_data,
164                            uint32_t msg_type,
165                            struct server_id server_id,
166                            DATA_BLOB *data)
167 {
168         struct child_pid *child;
169
170         debug_message(msg_ctx, private_data, MSG_DEBUG, server_id, data);
171
172         for (child = children; child != NULL; child = child->next) {
173                 messaging_send_buf(msg_ctx, pid_to_procid(child->pid),
174                                    MSG_DEBUG,
175                                    data->data,
176                                    strlen((char *) data->data) + 1);
177         }
178 }
179
180 static void add_child_pid(pid_t pid)
181 {
182         struct child_pid *child;
183
184         child = SMB_MALLOC_P(struct child_pid);
185         if (child == NULL) {
186                 DEBUG(0, ("Could not add child struct -- malloc failed\n"));
187                 return;
188         }
189         child->pid = pid;
190         DLIST_ADD(children, child);
191         num_children += 1;
192 }
193
194 /*
195   at most every smbd:cleanuptime seconds (default 20), we scan the BRL
196   and locking database for entries to cleanup. As a side effect this
197   also cleans up dead entries in the connections database (due to the
198   traversal in message_send_all()
199
200   Using a timer for this prevents a flood of traversals when a large
201   number of clients disconnect at the same time (perhaps due to a
202   network outage).  
203 */
204
205 static void cleanup_timeout_fn(struct event_context *event_ctx,
206                                 struct timed_event *te,
207                                 struct timeval now,
208                                 void *private_data)
209 {
210         struct timed_event **cleanup_te = (struct timed_event **)private_data;
211
212         DEBUG(1,("Cleaning up brl and lock database after unclean shutdown\n"));
213         message_send_all(smbd_messaging_context(), MSG_SMB_UNLOCK, NULL, 0, NULL);
214         messaging_send_buf(smbd_messaging_context(), procid_self(),
215                                 MSG_SMB_BRL_VALIDATE, NULL, 0);
216         /* mark the cleanup as having been done */
217         (*cleanup_te) = NULL;
218 }
219
220 static void remove_child_pid(pid_t pid, bool unclean_shutdown)
221 {
222         struct child_pid *child;
223         static struct timed_event *cleanup_te;
224         struct server_id child_id;
225
226         if (unclean_shutdown) {
227                 /* a child terminated uncleanly so tickle all
228                    processes to see if they can grab any of the
229                    pending locks
230                 */
231                 DEBUG(3,(__location__ " Unclean shutdown of pid %u\n",
232                         (unsigned int)pid));
233                 if (!cleanup_te) {
234                         /* call the cleanup timer, but not too often */
235                         int cleanup_time = lp_parm_int(-1, "smbd", "cleanuptime", 20);
236                         cleanup_te = event_add_timed(smbd_event_context(), NULL,
237                                                 timeval_current_ofs(cleanup_time, 0),
238                                                 cleanup_timeout_fn,
239                                                 &cleanup_te);
240                         DEBUG(1,("Scheduled cleanup of brl and lock database after unclean shutdown\n"));
241                 }
242         }
243
244         child_id = procid_self(); /* Just initialize pid and potentially vnn */
245         child_id.pid = pid;
246
247         if (!serverid_deregister(child_id)) {
248                 DEBUG(1, ("Could not remove pid %d from serverid.tdb\n",
249                           (int)pid));
250         }
251
252         for (child = children; child != NULL; child = child->next) {
253                 if (child->pid == pid) {
254                         struct child_pid *tmp = child;
255                         DLIST_REMOVE(children, child);
256                         SAFE_FREE(tmp);
257                         num_children -= 1;
258                         return;
259                 }
260         }
261
262         DEBUG(0, ("Could not find child %d -- ignoring\n", (int)pid));
263 }
264
265 /****************************************************************************
266  Have we reached the process limit ?
267 ****************************************************************************/
268
269 static bool allowable_number_of_smbd_processes(void)
270 {
271         int max_processes = lp_max_smbd_processes();
272
273         if (!max_processes)
274                 return True;
275
276         return num_children < max_processes;
277 }
278
279 static void smbd_sig_chld_handler(struct tevent_context *ev,
280                                   struct tevent_signal *se,
281                                   int signum,
282                                   int count,
283                                   void *siginfo,
284                                   void *private_data)
285 {
286         pid_t pid;
287         int status;
288
289         while ((pid = sys_waitpid(-1, &status, WNOHANG)) > 0) {
290                 bool unclean_shutdown = False;
291
292                 /* If the child terminated normally, assume
293                    it was an unclean shutdown unless the
294                    status is 0
295                 */
296                 if (WIFEXITED(status)) {
297                         unclean_shutdown = WEXITSTATUS(status);
298                 }
299                 /* If the child terminated due to a signal
300                    we always assume it was unclean.
301                 */
302                 if (WIFSIGNALED(status)) {
303                         unclean_shutdown = True;
304                 }
305                 remove_child_pid(pid, unclean_shutdown);
306         }
307 }
308
309 static void smbd_setup_sig_chld_handler(void)
310 {
311         struct tevent_signal *se;
312
313         se = tevent_add_signal(smbd_event_context(),
314                                smbd_event_context(),
315                                SIGCHLD, 0,
316                                smbd_sig_chld_handler,
317                                NULL);
318         if (!se) {
319                 exit_server("failed to setup SIGCHLD handler");
320         }
321 }
322
323 struct smbd_open_socket;
324
325 struct smbd_parent_context {
326         bool interactive;
327
328         /* the list of listening sockets */
329         struct smbd_open_socket *sockets;
330 };
331
332 struct smbd_open_socket {
333         struct smbd_open_socket *prev, *next;
334         struct smbd_parent_context *parent;
335         int fd;
336         struct tevent_fd *fde;
337 };
338
339 static void smbd_open_socket_close_fn(struct tevent_context *ev,
340                                       struct tevent_fd *fde,
341                                       int fd,
342                                       void *private_data)
343 {
344         /* this might be the socket_wrapper swrap_close() */
345         close(fd);
346 }
347
348 static void smbd_accept_connection(struct tevent_context *ev,
349                                    struct tevent_fd *fde,
350                                    uint16_t flags,
351                                    void *private_data)
352 {
353         struct smbd_open_socket *s = talloc_get_type_abort(private_data,
354                                      struct smbd_open_socket);
355         struct sockaddr_storage addr;
356         socklen_t in_addrlen = sizeof(addr);
357         pid_t pid = 0;
358         uint64_t unique_id;
359
360         smbd_set_server_fd(accept(s->fd, (struct sockaddr *)(void *)&addr,&in_addrlen));
361
362         if (smbd_server_fd() == -1 && errno == EINTR)
363                 return;
364
365         if (smbd_server_fd() == -1) {
366                 DEBUG(0,("open_sockets_smbd: accept: %s\n",
367                          strerror(errno)));
368                 return;
369         }
370
371         if (s->parent->interactive) {
372                 smbd_process();
373                 exit_server_cleanly("end of interactive mode");
374                 return;
375         }
376
377         if (!allowable_number_of_smbd_processes()) {
378                 close(smbd_server_fd());
379                 smbd_set_server_fd(-1);
380                 return;
381         }
382
383         /*
384          * Generate a unique id in the parent process so that we use
385          * the global random state in the parent.
386          */
387         generate_random_buffer((uint8_t *)&unique_id, sizeof(unique_id));
388
389         pid = sys_fork();
390         if (pid == 0) {
391                 NTSTATUS status = NT_STATUS_OK;
392
393                 /* Child code ... */
394                 am_parent = 0;
395
396                 set_my_unique_id(unique_id);
397
398                 /* Stop zombies, the parent explicitly handles
399                  * them, counting worker smbds. */
400                 CatchChild();
401
402                 /* close our standard file
403                    descriptors */
404                 close_low_fds(False);
405
406                 /*
407                  * Can't use TALLOC_FREE here. Nulling out the argument to it
408                  * would overwrite memory we've just freed.
409                  */
410                 talloc_free(s->parent);
411                 s = NULL;
412
413                 status = reinit_after_fork(smbd_messaging_context(),
414                                            smbd_event_context(), procid_self(),
415                                            true);
416                 if (!NT_STATUS_IS_OK(status)) {
417                         if (NT_STATUS_EQUAL(status,
418                                             NT_STATUS_TOO_MANY_OPENED_FILES)) {
419                                 DEBUG(0,("child process cannot initialize "
420                                          "because too many files are open\n"));
421                                 goto exit;
422                         }
423                         DEBUG(0,("reinit_after_fork() failed\n"));
424                         smb_panic("reinit_after_fork() failed");
425                 }
426
427                 smbd_setup_sig_term_handler();
428                 smbd_setup_sig_hup_handler(server_event_context(),
429                                            server_messaging_context());
430
431                 if (!serverid_register(procid_self(),
432                                        FLAG_MSG_GENERAL|FLAG_MSG_SMBD
433                                        |FLAG_MSG_DBWRAP
434                                        |FLAG_MSG_PRINT_GENERAL)) {
435                         exit_server_cleanly("Could not register myself in "
436                                             "serverid.tdb");
437                 }
438
439                 smbd_process();
440          exit:
441                 exit_server_cleanly("end of child");
442                 return;
443         }
444
445         if (pid < 0) {
446                 DEBUG(0,("smbd_accept_connection: sys_fork() failed: %s\n",
447                          strerror(errno)));
448         }
449
450         /* The parent doesn't need this socket */
451         close(smbd_server_fd());
452
453         /* Sun May 6 18:56:14 2001 ackley@cs.unm.edu:
454                 Clear the closed fd info out of server_fd --
455                 and more importantly, out of client_fd in
456                 util_sock.c, to avoid a possible
457                 getpeername failure if we reopen the logs
458                 and use %I in the filename.
459         */
460
461         smbd_set_server_fd(-1);
462
463         if (pid != 0) {
464                 add_child_pid(pid);
465         }
466
467         /* Force parent to check log size after
468          * spawning child.  Fix from
469          * klausr@ITAP.Physik.Uni-Stuttgart.De.  The
470          * parent smbd will log to logserver.smb.  It
471          * writes only two messages for each child
472          * started/finished. But each child writes,
473          * say, 50 messages also in logserver.smb,
474          * begining with the debug_count of the
475          * parent, before the child opens its own log
476          * file logserver.client. In a worst case
477          * scenario the size of logserver.smb would be
478          * checked after about 50*50=2500 messages
479          * (ca. 100kb).
480          * */
481         force_check_log_size();
482 }
483
484 static bool smbd_open_one_socket(struct smbd_parent_context *parent,
485                                  const struct sockaddr_storage *ifss,
486                                  uint16_t port)
487 {
488         struct smbd_open_socket *s;
489
490         s = talloc(parent, struct smbd_open_socket);
491         if (!s) {
492                 return false;
493         }
494
495         s->parent = parent;
496         s->fd = open_socket_in(SOCK_STREAM,
497                                port,
498                                parent->sockets == NULL ? 0 : 2,
499                                ifss,
500                                true);
501         if (s->fd == -1) {
502                 DEBUG(0,("smbd_open_once_socket: open_socket_in: "
503                         "%s\n", strerror(errno)));
504                 TALLOC_FREE(s);
505                 /*
506                  * We ignore an error here, as we've done before
507                  */
508                 return true;
509         }
510
511         /* ready to listen */
512         set_socket_options(s->fd, "SO_KEEPALIVE");
513         set_socket_options(s->fd, lp_socket_options());
514
515         /* Set server socket to
516          * non-blocking for the accept. */
517         set_blocking(s->fd, False);
518
519         if (listen(s->fd, SMBD_LISTEN_BACKLOG) == -1) {
520                 DEBUG(0,("open_sockets_smbd: listen: "
521                         "%s\n", strerror(errno)));
522                         close(s->fd);
523                 TALLOC_FREE(s);
524                 return false;
525         }
526
527         s->fde = tevent_add_fd(smbd_event_context(),
528                                s,
529                                s->fd, TEVENT_FD_READ,
530                                smbd_accept_connection,
531                                s);
532         if (!s->fde) {
533                 DEBUG(0,("open_sockets_smbd: "
534                          "tevent_add_fd: %s\n",
535                          strerror(errno)));
536                 close(s->fd);
537                 TALLOC_FREE(s);
538                 return false;
539         }
540         tevent_fd_set_close_fn(s->fde, smbd_open_socket_close_fn);
541
542         DLIST_ADD_END(parent->sockets, s, struct smbd_open_socket *);
543
544         return true;
545 }
546
547 /****************************************************************************
548  Open the socket communication.
549 ****************************************************************************/
550
551 static bool open_sockets_smbd(struct smbd_parent_context *parent,
552                               const char *smb_ports)
553 {
554         int num_interfaces = iface_count();
555         int i;
556         char *ports;
557         unsigned dns_port = 0;
558
559 #ifdef HAVE_ATEXIT
560         atexit(killkids);
561 #endif
562
563         /* Stop zombies */
564         smbd_setup_sig_chld_handler();
565
566         /* use a reasonable default set of ports - listing on 445 and 139 */
567         if (!smb_ports) {
568                 ports = lp_smb_ports();
569                 if (!ports || !*ports) {
570                         ports = talloc_strdup(talloc_tos(), SMB_PORTS);
571                 } else {
572                         ports = talloc_strdup(talloc_tos(), ports);
573                 }
574         } else {
575                 ports = talloc_strdup(talloc_tos(), smb_ports);
576         }
577
578         if (lp_interfaces() && lp_bind_interfaces_only()) {
579                 /* We have been given an interfaces line, and been
580                    told to only bind to those interfaces. Create a
581                    socket per interface and bind to only these.
582                 */
583
584                 /* Now open a listen socket for each of the
585                    interfaces. */
586                 for(i = 0; i < num_interfaces; i++) {
587                         const struct sockaddr_storage *ifss =
588                                         iface_n_sockaddr_storage(i);
589                         char *tok;
590                         const char *ptr;
591
592                         if (ifss == NULL) {
593                                 DEBUG(0,("open_sockets_smbd: "
594                                         "interface %d has NULL IP address !\n",
595                                         i));
596                                 continue;
597                         }
598
599                         for (ptr=ports;
600                              next_token_talloc(talloc_tos(),&ptr, &tok, " \t,");) {
601                                 unsigned port = atoi(tok);
602                                 if (port == 0 || port > 0xffff) {
603                                         continue;
604                                 }
605
606                                 if (!smbd_open_one_socket(parent, ifss, port)) {
607                                         return false;
608                                 }
609                         }
610                 }
611         } else {
612                 /* Just bind to 0.0.0.0 - accept connections
613                    from anywhere. */
614
615                 char *tok;
616                 const char *ptr;
617                 const char *sock_addr = lp_socket_address();
618                 char *sock_tok;
619                 const char *sock_ptr;
620
621                 if (strequal(sock_addr, "0.0.0.0") ||
622                     strequal(sock_addr, "::")) {
623 #if HAVE_IPV6
624                         sock_addr = "::,0.0.0.0";
625 #else
626                         sock_addr = "0.0.0.0";
627 #endif
628                 }
629
630                 for (sock_ptr=sock_addr;
631                      next_token_talloc(talloc_tos(), &sock_ptr, &sock_tok, " \t,"); ) {
632                         for (ptr=ports; next_token_talloc(talloc_tos(), &ptr, &tok, " \t,"); ) {
633                                 struct sockaddr_storage ss;
634
635                                 unsigned port = atoi(tok);
636                                 if (port == 0 || port > 0xffff) {
637                                         continue;
638                                 }
639
640                                 /* Keep the first port for mDNS service
641                                  * registration.
642                                  */
643                                 if (dns_port == 0) {
644                                         dns_port = port;
645                                 }
646
647                                 /* open an incoming socket */
648                                 if (!interpret_string_addr(&ss, sock_tok,
649                                                 AI_NUMERICHOST|AI_PASSIVE)) {
650                                         continue;
651                                 }
652
653                                 if (!smbd_open_one_socket(parent, &ss, port)) {
654                                         return false;
655                                 }
656                         }
657                 }
658         }
659
660         if (parent->sockets == NULL) {
661                 DEBUG(0,("open_sockets_smbd: No "
662                         "sockets available to bind to.\n"));
663                 return false;
664         }
665
666         /* Setup the main smbd so that we can get messages. Note that
667            do this after starting listening. This is needed as when in
668            clustered mode, ctdb won't allow us to start doing database
669            operations until it has gone thru a full startup, which
670            includes checking to see that smbd is listening. */
671
672         if (!serverid_register(procid_self(),
673                                FLAG_MSG_GENERAL|FLAG_MSG_SMBD
674                                |FLAG_MSG_DBWRAP)) {
675                 DEBUG(0, ("open_sockets_smbd: Failed to register "
676                           "myself in serverid.tdb\n"));
677                 return false;
678         }
679
680         /* Listen to messages */
681
682         messaging_register(smbd_messaging_context(), NULL,
683                            MSG_SMB_SAM_SYNC, msg_sam_sync);
684         messaging_register(smbd_messaging_context(), NULL,
685                            MSG_SHUTDOWN, msg_exit_server);
686         messaging_register(smbd_messaging_context(), NULL,
687                            MSG_SMB_FILE_RENAME, msg_file_was_renamed);
688         messaging_register(smbd_messaging_context(), NULL,
689                            MSG_SMB_CONF_UPDATED, smb_conf_updated);
690         messaging_register(smbd_messaging_context(), NULL,
691                            MSG_SMB_STAT_CACHE_DELETE, smb_stat_cache_delete);
692         messaging_register(smbd_messaging_context(), NULL,
693                            MSG_DEBUG, smbd_msg_debug);
694         brl_register_msgs(smbd_messaging_context());
695
696 #ifdef CLUSTER_SUPPORT
697         if (lp_clustering()) {
698                 ctdbd_register_reconfigure(
699                         messaging_ctdbd_connection(procid_self()));
700         }
701 #endif
702
703 #ifdef DEVELOPER
704         messaging_register(smbd_messaging_context(), NULL,
705                            MSG_SMB_INJECT_FAULT, msg_inject_fault);
706 #endif
707
708         if (dns_port != 0) {
709 #ifdef WITH_DNSSD_SUPPORT
710                 smbd_setup_mdns_registration(smbd_event_context(),
711                                              parent, dns_port);
712 #endif
713 #ifdef WITH_AVAHI_SUPPORT
714                 void *avahi_conn;
715
716                 avahi_conn = avahi_start_register(
717                         smbd_event_context(), smbd_event_context(), dns_port);
718                 if (avahi_conn == NULL) {
719                         DEBUG(10, ("avahi_start_register failed\n"));
720                 }
721 #endif
722         }
723
724         return true;
725 }
726
727 static void smbd_parent_loop(struct smbd_parent_context *parent)
728 {
729         /* now accept incoming connections - forking a new process
730            for each incoming connection */
731         DEBUG(2,("waiting for connections\n"));
732         while (1) {
733                 int ret;
734                 TALLOC_CTX *frame = talloc_stackframe();
735
736                 ret = tevent_loop_once(smbd_event_context());
737                 if (ret != 0) {
738                         exit_server_cleanly("tevent_loop_once() error");
739                 }
740
741                 TALLOC_FREE(frame);
742         } /* end while 1 */
743
744 /* NOTREACHED   return True; */
745 }
746
747
748 /****************************************************************************
749  Initialise connect, service and file structs.
750 ****************************************************************************/
751
752 static bool init_structs(void )
753 {
754         /*
755          * Set the machine NETBIOS name if not already
756          * set from the config file.
757          */
758
759         if (!init_names())
760                 return False;
761
762         file_init();
763
764         if (!secrets_init())
765                 return False;
766
767         return True;
768 }
769
770 /****************************************************************************
771  main program.
772 ****************************************************************************/
773
774 /* Declare prototype for build_options() to avoid having to run it through
775    mkproto.h.  Mixing $(builddir) and $(srcdir) source files in the current
776    prototype generation system is too complicated. */
777
778 extern void build_options(bool screen);
779
780  int main(int argc,const char *argv[])
781 {
782         /* shall I run as a daemon */
783         bool is_daemon = false;
784         bool interactive = false;
785         bool Fork = true;
786         bool no_process_group = false;
787         bool log_stdout = false;
788         char *ports = NULL;
789         char *profile_level = NULL;
790         int opt;
791         poptContext pc;
792         bool print_build_options = False;
793         enum {
794                 OPT_DAEMON = 1000,
795                 OPT_INTERACTIVE,
796                 OPT_FORK,
797                 OPT_NO_PROCESS_GROUP,
798                 OPT_LOG_STDOUT
799         };
800         struct poptOption long_options[] = {
801         POPT_AUTOHELP
802         {"daemon", 'D', POPT_ARG_NONE, NULL, OPT_DAEMON, "Become a daemon (default)" },
803         {"interactive", 'i', POPT_ARG_NONE, NULL, OPT_INTERACTIVE, "Run interactive (not a daemon)"},
804         {"foreground", 'F', POPT_ARG_NONE, NULL, OPT_FORK, "Run daemon in foreground (for daemontools, etc.)" },
805         {"no-process-group", '\0', POPT_ARG_NONE, NULL, OPT_NO_PROCESS_GROUP, "Don't create a new process group" },
806         {"log-stdout", 'S', POPT_ARG_NONE, NULL, OPT_LOG_STDOUT, "Log to stdout" },
807         {"build-options", 'b', POPT_ARG_NONE, NULL, 'b', "Print build options" },
808         {"port", 'p', POPT_ARG_STRING, &ports, 0, "Listen on the specified ports"},
809         {"profiling-level", 'P', POPT_ARG_STRING, &profile_level, 0, "Set profiling level","PROFILE_LEVEL"},
810         POPT_COMMON_SAMBA
811         POPT_COMMON_DYNCONFIG
812         POPT_TABLEEND
813         };
814         struct smbd_parent_context *parent = NULL;
815         TALLOC_CTX *frame = talloc_stackframe(); /* Setup tos. */
816         NTSTATUS status;
817         uint64_t unique_id;
818
819         smbd_init_globals();
820
821         TimeInit();
822
823 #ifdef HAVE_SET_AUTH_PARAMETERS
824         set_auth_parameters(argc,argv);
825 #endif
826
827         pc = poptGetContext("smbd", argc, argv, long_options, 0);
828         while((opt = poptGetNextOpt(pc)) != -1) {
829                 switch (opt)  {
830                 case OPT_DAEMON:
831                         is_daemon = true;
832                         break;
833                 case OPT_INTERACTIVE:
834                         interactive = true;
835                         break;
836                 case OPT_FORK:
837                         Fork = false;
838                         break;
839                 case OPT_NO_PROCESS_GROUP:
840                         no_process_group = true;
841                         break;
842                 case OPT_LOG_STDOUT:
843                         log_stdout = true;
844                         break;
845                 case 'b':
846                         print_build_options = True;
847                         break;
848                 default:
849                         d_fprintf(stderr, "\nInvalid option %s: %s\n\n",
850                                   poptBadOption(pc, 0), poptStrerror(opt));
851                         poptPrintUsage(pc, stderr, 0);
852                         exit(1);
853                 }
854         }
855         poptFreeContext(pc);
856
857         if (interactive) {
858                 Fork = False;
859                 log_stdout = True;
860         }
861
862         setup_logging(argv[0],log_stdout);
863
864         if (print_build_options) {
865                 build_options(True); /* Display output to screen as well as debug */
866                 exit(0);
867         }
868
869         load_case_tables();
870
871 #ifdef HAVE_SETLUID
872         /* needed for SecureWare on SCO */
873         setluid(0);
874 #endif
875
876         sec_init();
877
878         set_remote_machine_name("smbd", False);
879
880         if (interactive && (DEBUGLEVEL >= 9)) {
881                 talloc_enable_leak_report();
882         }
883
884         if (log_stdout && Fork) {
885                 DEBUG(0,("ERROR: Can't log to stdout (-S) unless daemon is in foreground (-F) or interactive (-i)\n"));
886                 exit(1);
887         }
888
889         /* we want to re-seed early to prevent time delays causing
890            client problems at a later date. (tridge) */
891         generate_random_buffer(NULL, 0);
892
893         /* make absolutely sure we run as root - to handle cases where people
894            are crazy enough to have it setuid */
895
896         gain_root_privilege();
897         gain_root_group_privilege();
898
899         fault_setup((void (*)(void *))exit_server_fault);
900         dump_core_setup("smbd");
901
902         /* we are never interested in SIGPIPE */
903         BlockSignals(True,SIGPIPE);
904
905 #if defined(SIGFPE)
906         /* we are never interested in SIGFPE */
907         BlockSignals(True,SIGFPE);
908 #endif
909
910 #if defined(SIGUSR2)
911         /* We are no longer interested in USR2 */
912         BlockSignals(True,SIGUSR2);
913 #endif
914
915         /* POSIX demands that signals are inherited. If the invoking process has
916          * these signals masked, we will have problems, as we won't recieve them. */
917         BlockSignals(False, SIGHUP);
918         BlockSignals(False, SIGUSR1);
919         BlockSignals(False, SIGTERM);
920
921         /* Ensure we leave no zombies until we
922          * correctly set up child handling below. */
923
924         CatchChild();
925
926         /* we want total control over the permissions on created files,
927            so set our umask to 0 */
928         umask(0);
929
930         init_sec_ctx();
931
932         reopen_logs();
933
934         DEBUG(0,("smbd version %s started.\n", samba_version_string()));
935         DEBUGADD(0,("%s\n", COPYRIGHT_STARTUP_MESSAGE));
936
937         DEBUG(2,("uid=%d gid=%d euid=%d egid=%d\n",
938                  (int)getuid(),(int)getgid(),(int)geteuid(),(int)getegid()));
939
940         /* Output the build options to the debug log */ 
941         build_options(False);
942
943         if (sizeof(uint16) < 2 || sizeof(uint32) < 4) {
944                 DEBUG(0,("ERROR: Samba is not configured correctly for the word size on your machine\n"));
945                 exit(1);
946         }
947
948         if (!lp_load_initial_only(get_dyn_CONFIGFILE())) {
949                 DEBUG(0, ("error opening config file\n"));
950                 exit(1);
951         }
952
953         if (smbd_messaging_context() == NULL)
954                 exit(1);
955
956         if (!reload_services(smbd_messaging_context(), False))
957                 return(-1);     
958
959         init_structs();
960
961 #ifdef WITH_PROFILE
962         if (!profile_setup(smbd_messaging_context(), False)) {
963                 DEBUG(0,("ERROR: failed to setup profiling\n"));
964                 return -1;
965         }
966         if (profile_level != NULL) {
967                 int pl = atoi(profile_level);
968                 struct server_id src;
969
970                 DEBUG(1, ("setting profiling level: %s\n",profile_level));
971                 src.pid = getpid();
972                 set_profile_level(pl, src);
973         }
974 #endif
975
976         DEBUG(3,( "loaded services\n"));
977
978         if (!is_daemon && !is_a_socket(0)) {
979                 if (!interactive)
980                         DEBUG(0,("standard input is not a socket, assuming -D option\n"));
981
982                 /*
983                  * Setting is_daemon here prevents us from eventually calling
984                  * the open_sockets_inetd()
985                  */
986
987                 is_daemon = True;
988         }
989
990         if (is_daemon && !interactive) {
991                 DEBUG( 3, ( "Becoming a daemon.\n" ) );
992                 become_daemon(Fork, no_process_group, log_stdout);
993         }
994
995         generate_random_buffer((uint8_t *)&unique_id, sizeof(unique_id));
996         set_my_unique_id(unique_id);
997
998 #if HAVE_SETPGID
999         /*
1000          * If we're interactive we want to set our own process group for
1001          * signal management.
1002          */
1003         if (interactive && !no_process_group)
1004                 setpgid( (pid_t)0, (pid_t)0);
1005 #endif
1006
1007         if (!directory_exist(lp_lockdir()))
1008                 mkdir(lp_lockdir(), 0755);
1009
1010         if (is_daemon)
1011                 pidfile_create("smbd");
1012
1013         status = reinit_after_fork(smbd_messaging_context(),
1014                                    smbd_event_context(),
1015                                    procid_self(), false);
1016         if (!NT_STATUS_IS_OK(status)) {
1017                 DEBUG(0,("reinit_after_fork() failed\n"));
1018                 exit(1);
1019         }
1020
1021         smbd_server_conn->msg_ctx = smbd_messaging_context();
1022
1023         smbd_setup_sig_term_handler();
1024         smbd_setup_sig_hup_handler(smbd_event_context(),
1025                                    smbd_server_conn->msg_ctx);
1026
1027         /* Setup all the TDB's - including CLEAR_IF_FIRST tdb's. */
1028
1029         if (smbd_memcache() == NULL) {
1030                 exit(1);
1031         }
1032
1033         memcache_set_global(smbd_memcache());
1034
1035         /* Initialise the password backed before the global_sam_sid
1036            to ensure that we fetch from ldap before we make a domain sid up */
1037
1038         if(!initialize_password_db(False, smbd_event_context()))
1039                 exit(1);
1040
1041         if (!secrets_init()) {
1042                 DEBUG(0, ("ERROR: smbd can not open secrets.tdb\n"));
1043                 exit(1);
1044         }
1045
1046         if (lp_server_role() == ROLE_DOMAIN_BDC || lp_server_role() == ROLE_DOMAIN_PDC) {
1047                 if (!open_schannel_session_store(NULL, lp_private_dir())) {
1048                         DEBUG(0,("ERROR: Samba cannot open schannel store for secured NETLOGON operations.\n"));
1049                         exit(1);
1050                 }
1051         }
1052
1053         if(!get_global_sam_sid()) {
1054                 DEBUG(0,("ERROR: Samba cannot create a SAM SID.\n"));
1055                 exit(1);
1056         }
1057
1058         if (!sessionid_init()) {
1059                 exit(1);
1060         }
1061
1062         if (!connections_init(True))
1063                 exit(1);
1064
1065         if (!locking_init())
1066                 exit(1);
1067
1068         if (!messaging_tdb_parent_init()) {
1069                 exit(1);
1070         }
1071
1072         if (!notify_internal_parent_init()) {
1073                 exit(1);
1074         }
1075
1076         if (!serverid_parent_init()) {
1077                 exit(1);
1078         }
1079
1080         if (!W_ERROR_IS_OK(registry_init_full()))
1081                 exit(1);
1082
1083 #if 0
1084         if (!init_svcctl_db())
1085                 exit(1);
1086 #endif
1087
1088         if (!init_system_info()) {
1089                 DEBUG(0,("ERROR: failed to setup system user info.\n"));
1090                 return -1;
1091         }
1092
1093         if (!print_backend_init(smbd_messaging_context()))
1094                 exit(1);
1095
1096         if (!init_guest_info()) {
1097                 DEBUG(0,("ERROR: failed to setup guest info.\n"));
1098                 return -1;
1099         }
1100
1101         /* Open the share_info.tdb here, so we don't have to open
1102            after the fork on every single connection.  This is a small
1103            performance improvment and reduces the total number of system
1104            fds used. */
1105         if (!share_info_db_init()) {
1106                 DEBUG(0,("ERROR: failed to load share info db.\n"));
1107                 exit(1);
1108         }
1109
1110         /* only start the background queue daemon if we are 
1111            running as a daemon -- bad things will happen if
1112            smbd is launched via inetd and we fork a copy of 
1113            ourselves here */
1114
1115         if (is_daemon && !interactive
1116             && lp_parm_bool(-1, "smbd", "backgroundqueue", true)) {
1117                 start_background_queue(smbd_event_context(),
1118                                        smbd_messaging_context());
1119         }
1120
1121         if (!is_daemon) {
1122                 /* inetd mode */
1123                 TALLOC_FREE(frame);
1124
1125                 /* Started from inetd. fd 0 is the socket. */
1126                 /* We will abort gracefully when the client or remote system
1127                    goes away */
1128                 smbd_set_server_fd(dup(0));
1129
1130                 /* close our standard file descriptors */
1131                 close_low_fds(False); /* Don't close stderr */
1132
1133 #ifdef HAVE_ATEXIT
1134                 atexit(killkids);
1135 #endif
1136
1137                 /* Stop zombies */
1138                 smbd_setup_sig_chld_handler();
1139
1140                 smbd_process();
1141
1142                 exit_server_cleanly(NULL);
1143                 return(0);
1144         }
1145
1146         parent = talloc_zero(smbd_event_context(), struct smbd_parent_context);
1147         if (!parent) {
1148                 exit_server("talloc(struct smbd_parent_context) failed");
1149         }
1150         parent->interactive = interactive;
1151
1152         if (!open_sockets_smbd(parent, ports))
1153                 exit_server("open_sockets_smbd() failed");
1154
1155         TALLOC_FREE(frame);
1156         /* make sure we always have a valid stackframe */
1157         frame = talloc_stackframe();
1158
1159         smbd_parent_loop(parent);
1160
1161         exit_server_cleanly(NULL);
1162         TALLOC_FREE(frame);
1163         return(0);
1164 }