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