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