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