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