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