s3-build: only include memcache.h where needed.
[kai/samba.git] / source3 / smbd / server.c
1 /*
2    Unix SMB/CIFS implementation.
3    Main SMB server routines
4    Copyright (C) Andrew Tridgell                1992-1998
5    Copyright (C) Martin Pool                    2002
6    Copyright (C) Jelmer Vernooij                2002-2003
7    Copyright (C) Volker Lendecke                1993-2007
8    Copyright (C) Jeremy Allison                 1993-2007
9
10    This program is free software; you can redistribute it and/or modify
11    it under the terms of the GNU General Public License as published by
12    the Free Software Foundation; either version 3 of the License, or
13    (at your option) any later version.
14
15    This program is distributed in the hope that it will be useful,
16    but WITHOUT ANY WARRANTY; without even the implied warranty of
17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18    GNU General Public License for more details.
19
20    You should have received a copy of the GNU General Public License
21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
22 */
23
24 #include "includes.h"
25 #include "popt_common.h"
26 #include "smbd/globals.h"
27 #include "librpc/gen_ndr/messaging.h"
28 #include "registry.h"
29 #include "libcli/auth/schannel.h"
30 #include "secrets.h"
31 #include "memcache.h"
32
33 #include "../librpc/gen_ndr/srv_dfs.h"
34 #include "../librpc/gen_ndr/srv_dssetup.h"
35 #include "../librpc/gen_ndr/srv_echo.h"
36 #include "../librpc/gen_ndr/srv_eventlog.h"
37 #include "../librpc/gen_ndr/srv_initshutdown.h"
38 #include "../librpc/gen_ndr/srv_lsa.h"
39 #include "../librpc/gen_ndr/srv_netlogon.h"
40 #include "../librpc/gen_ndr/srv_ntsvcs.h"
41 #include "../librpc/gen_ndr/srv_samr.h"
42 #include "../librpc/gen_ndr/srv_spoolss.h"
43 #include "../librpc/gen_ndr/srv_srvsvc.h"
44 #include "../librpc/gen_ndr/srv_svcctl.h"
45 #include "../librpc/gen_ndr/srv_winreg.h"
46 #include "../librpc/gen_ndr/srv_wkssvc.h"
47
48 #include "printing/nt_printing_migrate.h"
49
50 static_decl_rpc;
51
52 #ifdef WITH_DFS
53 extern int dcelogin_atmost_once;
54 #endif /* WITH_DFS */
55
56 int smbd_server_fd(void)
57 {
58         return server_fd;
59 }
60
61 static void smbd_set_server_fd(int fd)
62 {
63         struct smbd_server_connection *sconn = smbd_server_conn;
64         const char *name;
65
66         server_fd = fd;
67
68         sconn->sock = fd;
69
70         /*
71          * Initialize sconn->client_id: If we can't find the client's
72          * name, default to its address.
73          */
74
75         client_addr(fd, sconn->client_id.addr, sizeof(sconn->client_id.addr));
76
77         name = client_name(sconn->sock);
78         if (strcmp(name, "UNKNOWN") != 0) {
79                 name = talloc_strdup(sconn, name);
80         } else {
81                 name = NULL;
82         }
83         sconn->client_id.name =
84                 (name != NULL) ? name : sconn->client_id.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_fd(), 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         pid_t pid = 0;
392         uint64_t unique_id;
393
394         smbd_set_server_fd(accept(s->fd, (struct sockaddr *)(void *)&addr,&in_addrlen));
395
396         if (smbd_server_fd() == -1 && errno == EINTR)
397                 return;
398
399         if (smbd_server_fd() == -1) {
400                 DEBUG(0,("open_sockets_smbd: accept: %s\n",
401                          strerror(errno)));
402                 return;
403         }
404
405         if (s->parent->interactive) {
406                 smbd_process(smbd_server_conn);
407                 exit_server_cleanly("end of interactive mode");
408                 return;
409         }
410
411         if (!allowable_number_of_smbd_processes()) {
412                 close(smbd_server_fd());
413                 smbd_set_server_fd(-1);
414                 return;
415         }
416
417         /*
418          * Generate a unique id in the parent process so that we use
419          * the global random state in the parent.
420          */
421         generate_random_buffer((uint8_t *)&unique_id, sizeof(unique_id));
422
423         pid = sys_fork();
424         if (pid == 0) {
425                 NTSTATUS status = NT_STATUS_OK;
426
427                 /* Child code ... */
428                 am_parent = 0;
429
430                 set_my_unique_id(unique_id);
431
432                 /* Stop zombies, the parent explicitly handles
433                  * them, counting worker smbds. */
434                 CatchChild();
435
436                 /* close our standard file
437                    descriptors */
438                 close_low_fds(False);
439
440                 /*
441                  * Can't use TALLOC_FREE here. Nulling out the argument to it
442                  * would overwrite memory we've just freed.
443                  */
444                 talloc_free(s->parent);
445                 s = NULL;
446
447                 status = reinit_after_fork(smbd_messaging_context(),
448                                            smbd_event_context(), procid_self(),
449                                            true);
450                 if (!NT_STATUS_IS_OK(status)) {
451                         if (NT_STATUS_EQUAL(status,
452                                             NT_STATUS_TOO_MANY_OPENED_FILES)) {
453                                 DEBUG(0,("child process cannot initialize "
454                                          "because too many files are open\n"));
455                                 goto exit;
456                         }
457                         DEBUG(0,("reinit_after_fork() failed\n"));
458                         smb_panic("reinit_after_fork() failed");
459                 }
460
461                 smbd_setup_sig_term_handler();
462                 smbd_setup_sig_hup_handler(server_event_context(),
463                                            server_messaging_context());
464
465                 if (!serverid_register(procid_self(),
466                                        FLAG_MSG_GENERAL|FLAG_MSG_SMBD
467                                        |FLAG_MSG_DBWRAP
468                                        |FLAG_MSG_PRINT_GENERAL)) {
469                         exit_server_cleanly("Could not register myself in "
470                                             "serverid.tdb");
471                 }
472
473                 smbd_process(smbd_server_conn);
474          exit:
475                 exit_server_cleanly("end of child");
476                 return;
477         }
478
479         if (pid < 0) {
480                 DEBUG(0,("smbd_accept_connection: sys_fork() failed: %s\n",
481                          strerror(errno)));
482         }
483
484         /* The parent doesn't need this socket */
485         close(smbd_server_fd());
486
487         /* Sun May 6 18:56:14 2001 ackley@cs.unm.edu:
488                 Clear the closed fd info out of server_fd --
489                 and more importantly, out of client_fd in
490                 util_sock.c, to avoid a possible
491                 getpeername failure if we reopen the logs
492                 and use %I in the filename.
493         */
494
495         smbd_set_server_fd(-1);
496
497         if (pid != 0) {
498                 add_child_pid(pid);
499         }
500
501         /* Force parent to check log size after
502          * spawning child.  Fix from
503          * klausr@ITAP.Physik.Uni-Stuttgart.De.  The
504          * parent smbd will log to logserver.smb.  It
505          * writes only two messages for each child
506          * started/finished. But each child writes,
507          * say, 50 messages also in logserver.smb,
508          * begining with the debug_count of the
509          * parent, before the child opens its own log
510          * file logserver.client. In a worst case
511          * scenario the size of logserver.smb would be
512          * checked after about 50*50=2500 messages
513          * (ca. 100kb).
514          * */
515         force_check_log_size();
516 }
517
518 static bool smbd_open_one_socket(struct smbd_parent_context *parent,
519                                  const struct sockaddr_storage *ifss,
520                                  uint16_t port)
521 {
522         struct smbd_open_socket *s;
523
524         s = talloc(parent, struct smbd_open_socket);
525         if (!s) {
526                 return false;
527         }
528
529         s->parent = parent;
530         s->fd = open_socket_in(SOCK_STREAM,
531                                port,
532                                parent->sockets == NULL ? 0 : 2,
533                                ifss,
534                                true);
535         if (s->fd == -1) {
536                 DEBUG(0,("smbd_open_once_socket: open_socket_in: "
537                         "%s\n", strerror(errno)));
538                 TALLOC_FREE(s);
539                 /*
540                  * We ignore an error here, as we've done before
541                  */
542                 return true;
543         }
544
545         /* ready to listen */
546         set_socket_options(s->fd, "SO_KEEPALIVE");
547         set_socket_options(s->fd, lp_socket_options());
548
549         /* Set server socket to
550          * non-blocking for the accept. */
551         set_blocking(s->fd, False);
552
553         if (listen(s->fd, SMBD_LISTEN_BACKLOG) == -1) {
554                 DEBUG(0,("open_sockets_smbd: listen: "
555                         "%s\n", strerror(errno)));
556                         close(s->fd);
557                 TALLOC_FREE(s);
558                 return false;
559         }
560
561         s->fde = tevent_add_fd(smbd_event_context(),
562                                s,
563                                s->fd, TEVENT_FD_READ,
564                                smbd_accept_connection,
565                                s);
566         if (!s->fde) {
567                 DEBUG(0,("open_sockets_smbd: "
568                          "tevent_add_fd: %s\n",
569                          strerror(errno)));
570                 close(s->fd);
571                 TALLOC_FREE(s);
572                 return false;
573         }
574         tevent_fd_set_close_fn(s->fde, smbd_open_socket_close_fn);
575
576         DLIST_ADD_END(parent->sockets, s, struct smbd_open_socket *);
577
578         return true;
579 }
580
581 /****************************************************************************
582  Open the socket communication.
583 ****************************************************************************/
584
585 static bool open_sockets_smbd(struct smbd_parent_context *parent,
586                               const char *smb_ports)
587 {
588         int num_interfaces = iface_count();
589         int i;
590         char *ports;
591         unsigned dns_port = 0;
592
593 #ifdef HAVE_ATEXIT
594         atexit(killkids);
595 #endif
596
597         /* Stop zombies */
598         smbd_setup_sig_chld_handler();
599
600         /* use a reasonable default set of ports - listing on 445 and 139 */
601         if (!smb_ports) {
602                 ports = lp_smb_ports();
603                 if (!ports || !*ports) {
604                         ports = talloc_strdup(talloc_tos(), SMB_PORTS);
605                 } else {
606                         ports = talloc_strdup(talloc_tos(), ports);
607                 }
608         } else {
609                 ports = talloc_strdup(talloc_tos(), smb_ports);
610         }
611
612         if (lp_interfaces() && lp_bind_interfaces_only()) {
613                 /* We have been given an interfaces line, and been
614                    told to only bind to those interfaces. Create a
615                    socket per interface and bind to only these.
616                 */
617
618                 /* Now open a listen socket for each of the
619                    interfaces. */
620                 for(i = 0; i < num_interfaces; i++) {
621                         const struct sockaddr_storage *ifss =
622                                         iface_n_sockaddr_storage(i);
623                         char *tok;
624                         const char *ptr;
625
626                         if (ifss == NULL) {
627                                 DEBUG(0,("open_sockets_smbd: "
628                                         "interface %d has NULL IP address !\n",
629                                         i));
630                                 continue;
631                         }
632
633                         for (ptr=ports;
634                              next_token_talloc(talloc_tos(),&ptr, &tok, " \t,");) {
635                                 unsigned port = atoi(tok);
636                                 if (port == 0 || port > 0xffff) {
637                                         continue;
638                                 }
639
640                                 if (!smbd_open_one_socket(parent, ifss, port)) {
641                                         return false;
642                                 }
643                         }
644                 }
645         } else {
646                 /* Just bind to 0.0.0.0 - accept connections
647                    from anywhere. */
648
649                 char *tok;
650                 const char *ptr;
651                 const char *sock_addr = lp_socket_address();
652                 char *sock_tok;
653                 const char *sock_ptr;
654
655                 if (strequal(sock_addr, "0.0.0.0") ||
656                     strequal(sock_addr, "::")) {
657 #if HAVE_IPV6
658                         sock_addr = "::,0.0.0.0";
659 #else
660                         sock_addr = "0.0.0.0";
661 #endif
662                 }
663
664                 for (sock_ptr=sock_addr;
665                      next_token_talloc(talloc_tos(), &sock_ptr, &sock_tok, " \t,"); ) {
666                         for (ptr=ports; next_token_talloc(talloc_tos(), &ptr, &tok, " \t,"); ) {
667                                 struct sockaddr_storage ss;
668
669                                 unsigned port = atoi(tok);
670                                 if (port == 0 || port > 0xffff) {
671                                         continue;
672                                 }
673
674                                 /* Keep the first port for mDNS service
675                                  * registration.
676                                  */
677                                 if (dns_port == 0) {
678                                         dns_port = port;
679                                 }
680
681                                 /* open an incoming socket */
682                                 if (!interpret_string_addr(&ss, sock_tok,
683                                                 AI_NUMERICHOST|AI_PASSIVE)) {
684                                         continue;
685                                 }
686
687                                 if (!smbd_open_one_socket(parent, &ss, port)) {
688                                         return false;
689                                 }
690                         }
691                 }
692         }
693
694         if (parent->sockets == NULL) {
695                 DEBUG(0,("open_sockets_smbd: No "
696                         "sockets available to bind to.\n"));
697                 return false;
698         }
699
700         /* Setup the main smbd so that we can get messages. Note that
701            do this after starting listening. This is needed as when in
702            clustered mode, ctdb won't allow us to start doing database
703            operations until it has gone thru a full startup, which
704            includes checking to see that smbd is listening. */
705
706         if (!serverid_register(procid_self(),
707                                FLAG_MSG_GENERAL|FLAG_MSG_SMBD
708                                |FLAG_MSG_DBWRAP)) {
709                 DEBUG(0, ("open_sockets_smbd: Failed to register "
710                           "myself in serverid.tdb\n"));
711                 return false;
712         }
713
714         /* Listen to messages */
715
716         messaging_register(smbd_messaging_context(), NULL,
717                            MSG_SMB_SAM_SYNC, msg_sam_sync);
718         messaging_register(smbd_messaging_context(), NULL,
719                            MSG_SHUTDOWN, msg_exit_server);
720         messaging_register(smbd_messaging_context(), NULL,
721                            MSG_SMB_FILE_RENAME, msg_file_was_renamed);
722         messaging_register(smbd_messaging_context(), NULL,
723                            MSG_SMB_CONF_UPDATED, smb_conf_updated);
724         messaging_register(smbd_messaging_context(), NULL,
725                            MSG_SMB_STAT_CACHE_DELETE, smb_stat_cache_delete);
726         messaging_register(smbd_messaging_context(), NULL,
727                            MSG_DEBUG, smbd_msg_debug);
728         brl_register_msgs(smbd_messaging_context());
729
730 #ifdef CLUSTER_SUPPORT
731         if (lp_clustering()) {
732                 ctdbd_register_reconfigure(
733                         messaging_ctdbd_connection(procid_self()));
734         }
735 #endif
736
737 #ifdef DEVELOPER
738         messaging_register(smbd_messaging_context(), NULL,
739                            MSG_SMB_INJECT_FAULT, msg_inject_fault);
740 #endif
741
742         if (dns_port != 0) {
743 #ifdef WITH_DNSSD_SUPPORT
744                 smbd_setup_mdns_registration(smbd_event_context(),
745                                              parent, dns_port);
746 #endif
747 #ifdef WITH_AVAHI_SUPPORT
748                 void *avahi_conn;
749
750                 avahi_conn = avahi_start_register(
751                         smbd_event_context(), smbd_event_context(), dns_port);
752                 if (avahi_conn == NULL) {
753                         DEBUG(10, ("avahi_start_register failed\n"));
754                 }
755 #endif
756         }
757
758         return true;
759 }
760
761 static void smbd_parent_loop(struct smbd_parent_context *parent)
762 {
763         /* now accept incoming connections - forking a new process
764            for each incoming connection */
765         DEBUG(2,("waiting for connections\n"));
766         while (1) {
767                 int ret;
768                 TALLOC_CTX *frame = talloc_stackframe();
769
770                 ret = tevent_loop_once(smbd_event_context());
771                 if (ret != 0) {
772                         exit_server_cleanly("tevent_loop_once() error");
773                 }
774
775                 TALLOC_FREE(frame);
776         } /* end while 1 */
777
778 /* NOTREACHED   return True; */
779 }
780
781
782 /****************************************************************************
783  Initialise connect, service and file structs.
784 ****************************************************************************/
785
786 static bool init_structs(void )
787 {
788         /*
789          * Set the machine NETBIOS name if not already
790          * set from the config file.
791          */
792
793         if (!init_names())
794                 return False;
795
796         file_init();
797
798         if (!secrets_init())
799                 return False;
800
801         return True;
802 }
803
804 static bool spoolss_init_cb(void *ptr)
805 {
806         struct messaging_context *msg_ctx = talloc_get_type_abort(
807                 ptr, struct messaging_context);
808         return nt_printing_tdb_migrate(msg_ctx);
809 }
810
811 /****************************************************************************
812  main program.
813 ****************************************************************************/
814
815 /* Declare prototype for build_options() to avoid having to run it through
816    mkproto.h.  Mixing $(builddir) and $(srcdir) source files in the current
817    prototype generation system is too complicated. */
818
819 extern void build_options(bool screen);
820
821  int main(int argc,const char *argv[])
822 {
823         /* shall I run as a daemon */
824         bool is_daemon = false;
825         bool interactive = false;
826         bool Fork = true;
827         bool no_process_group = false;
828         bool log_stdout = false;
829         char *ports = NULL;
830         char *profile_level = NULL;
831         int opt;
832         poptContext pc;
833         bool print_build_options = False;
834         enum {
835                 OPT_DAEMON = 1000,
836                 OPT_INTERACTIVE,
837                 OPT_FORK,
838                 OPT_NO_PROCESS_GROUP,
839                 OPT_LOG_STDOUT
840         };
841         struct poptOption long_options[] = {
842         POPT_AUTOHELP
843         {"daemon", 'D', POPT_ARG_NONE, NULL, OPT_DAEMON, "Become a daemon (default)" },
844         {"interactive", 'i', POPT_ARG_NONE, NULL, OPT_INTERACTIVE, "Run interactive (not a daemon)"},
845         {"foreground", 'F', POPT_ARG_NONE, NULL, OPT_FORK, "Run daemon in foreground (for daemontools, etc.)" },
846         {"no-process-group", '\0', POPT_ARG_NONE, NULL, OPT_NO_PROCESS_GROUP, "Don't create a new process group" },
847         {"log-stdout", 'S', POPT_ARG_NONE, NULL, OPT_LOG_STDOUT, "Log to stdout" },
848         {"build-options", 'b', POPT_ARG_NONE, NULL, 'b', "Print build options" },
849         {"port", 'p', POPT_ARG_STRING, &ports, 0, "Listen on the specified ports"},
850         {"profiling-level", 'P', POPT_ARG_STRING, &profile_level, 0, "Set profiling level","PROFILE_LEVEL"},
851         POPT_COMMON_SAMBA
852         POPT_COMMON_DYNCONFIG
853         POPT_TABLEEND
854         };
855         struct smbd_parent_context *parent = NULL;
856         TALLOC_CTX *frame = talloc_stackframe(); /* Setup tos. */
857         NTSTATUS status;
858         uint64_t unique_id;
859         struct rpc_srv_callbacks spoolss_cb;
860
861         /* Initialize the event context, it will panic on error */
862         smbd_event_context();
863
864         smbd_init_globals();
865
866         TimeInit();
867
868 #ifdef HAVE_SET_AUTH_PARAMETERS
869         set_auth_parameters(argc,argv);
870 #endif
871
872         pc = poptGetContext("smbd", argc, argv, long_options, 0);
873         while((opt = poptGetNextOpt(pc)) != -1) {
874                 switch (opt)  {
875                 case OPT_DAEMON:
876                         is_daemon = true;
877                         break;
878                 case OPT_INTERACTIVE:
879                         interactive = true;
880                         break;
881                 case OPT_FORK:
882                         Fork = false;
883                         break;
884                 case OPT_NO_PROCESS_GROUP:
885                         no_process_group = true;
886                         break;
887                 case OPT_LOG_STDOUT:
888                         log_stdout = true;
889                         break;
890                 case 'b':
891                         print_build_options = True;
892                         break;
893                 default:
894                         d_fprintf(stderr, "\nInvalid option %s: %s\n\n",
895                                   poptBadOption(pc, 0), poptStrerror(opt));
896                         poptPrintUsage(pc, stderr, 0);
897                         exit(1);
898                 }
899         }
900         poptFreeContext(pc);
901
902         if (interactive) {
903                 Fork = False;
904                 log_stdout = True;
905         }
906
907         setup_logging(argv[0],log_stdout);
908
909         if (print_build_options) {
910                 build_options(True); /* Display output to screen as well as debug */
911                 exit(0);
912         }
913
914         load_case_tables();
915
916 #ifdef HAVE_SETLUID
917         /* needed for SecureWare on SCO */
918         setluid(0);
919 #endif
920
921         set_remote_machine_name("smbd", False);
922
923         if (interactive && (DEBUGLEVEL >= 9)) {
924                 talloc_enable_leak_report();
925         }
926
927         if (log_stdout && Fork) {
928                 DEBUG(0,("ERROR: Can't log to stdout (-S) unless daemon is in foreground (-F) or interactive (-i)\n"));
929                 exit(1);
930         }
931
932         /* we want to re-seed early to prevent time delays causing
933            client problems at a later date. (tridge) */
934         generate_random_buffer(NULL, 0);
935
936         /* get initial effective uid and gid */
937         sec_init();
938
939         /* make absolutely sure we run as root - to handle cases where people
940            are crazy enough to have it setuid */
941         gain_root_privilege();
942         gain_root_group_privilege();
943
944         fault_setup((void (*)(void *))exit_server_fault);
945         dump_core_setup("smbd");
946
947         /* we are never interested in SIGPIPE */
948         BlockSignals(True,SIGPIPE);
949
950 #if defined(SIGFPE)
951         /* we are never interested in SIGFPE */
952         BlockSignals(True,SIGFPE);
953 #endif
954
955 #if defined(SIGUSR2)
956         /* We are no longer interested in USR2 */
957         BlockSignals(True,SIGUSR2);
958 #endif
959
960         /* POSIX demands that signals are inherited. If the invoking process has
961          * these signals masked, we will have problems, as we won't recieve them. */
962         BlockSignals(False, SIGHUP);
963         BlockSignals(False, SIGUSR1);
964         BlockSignals(False, SIGTERM);
965
966         /* Ensure we leave no zombies until we
967          * correctly set up child handling below. */
968
969         CatchChild();
970
971         /* we want total control over the permissions on created files,
972            so set our umask to 0 */
973         umask(0);
974
975         reopen_logs();
976
977         DEBUG(0,("smbd version %s started.\n", samba_version_string()));
978         DEBUGADD(0,("%s\n", COPYRIGHT_STARTUP_MESSAGE));
979
980         DEBUG(2,("uid=%d gid=%d euid=%d egid=%d\n",
981                  (int)getuid(),(int)getgid(),(int)geteuid(),(int)getegid()));
982
983         /* Output the build options to the debug log */ 
984         build_options(False);
985
986         if (sizeof(uint16) < 2 || sizeof(uint32) < 4) {
987                 DEBUG(0,("ERROR: Samba is not configured correctly for the word size on your machine\n"));
988                 exit(1);
989         }
990
991         if (!lp_load_initial_only(get_dyn_CONFIGFILE())) {
992                 DEBUG(0, ("error opening config file\n"));
993                 exit(1);
994         }
995
996         /* Init the security context and global current_user */
997         init_sec_ctx();
998
999         if (smbd_messaging_context() == NULL)
1000                 exit(1);
1001
1002         /*
1003          * Reloading of the printers will not work here as we don't have a
1004          * server info and rpc services set up. It will be called later.
1005          */
1006         if (!reload_services(smbd_messaging_context(), smbd_server_fd(),
1007                              False)) {
1008                 exit(1);
1009         }
1010
1011         /* ...NOTE... Log files are working from this point! */
1012
1013         DEBUG(3,("loaded services\n"));
1014
1015         init_structs();
1016
1017 #ifdef WITH_PROFILE
1018         if (!profile_setup(smbd_messaging_context(), False)) {
1019                 DEBUG(0,("ERROR: failed to setup profiling\n"));
1020                 return -1;
1021         }
1022         if (profile_level != NULL) {
1023                 int pl = atoi(profile_level);
1024                 struct server_id src;
1025
1026                 DEBUG(1, ("setting profiling level: %s\n",profile_level));
1027                 src.pid = getpid();
1028                 set_profile_level(pl, src);
1029         }
1030 #endif
1031
1032         if (!is_daemon && !is_a_socket(0)) {
1033                 if (!interactive)
1034                         DEBUG(0,("standard input is not a socket, assuming -D option\n"));
1035
1036                 /*
1037                  * Setting is_daemon here prevents us from eventually calling
1038                  * the open_sockets_inetd()
1039                  */
1040
1041                 is_daemon = True;
1042         }
1043
1044         if (is_daemon && !interactive) {
1045                 DEBUG( 3, ( "Becoming a daemon.\n" ) );
1046                 become_daemon(Fork, no_process_group, log_stdout);
1047         }
1048
1049         generate_random_buffer((uint8_t *)&unique_id, sizeof(unique_id));
1050         set_my_unique_id(unique_id);
1051
1052 #if HAVE_SETPGID
1053         /*
1054          * If we're interactive we want to set our own process group for
1055          * signal management.
1056          */
1057         if (interactive && !no_process_group)
1058                 setpgid( (pid_t)0, (pid_t)0);
1059 #endif
1060
1061         if (!directory_exist(lp_lockdir()))
1062                 mkdir(lp_lockdir(), 0755);
1063
1064         if (is_daemon)
1065                 pidfile_create("smbd");
1066
1067         status = reinit_after_fork(smbd_messaging_context(),
1068                                    smbd_event_context(),
1069                                    procid_self(), false);
1070         if (!NT_STATUS_IS_OK(status)) {
1071                 DEBUG(0,("reinit_after_fork() failed\n"));
1072                 exit(1);
1073         }
1074
1075         smbd_server_conn->msg_ctx = smbd_messaging_context();
1076
1077         smbd_setup_sig_term_handler();
1078         smbd_setup_sig_hup_handler(smbd_event_context(),
1079                                    smbd_server_conn->msg_ctx);
1080
1081         /* Setup all the TDB's - including CLEAR_IF_FIRST tdb's. */
1082
1083         if (smbd_memcache() == NULL) {
1084                 exit(1);
1085         }
1086
1087         memcache_set_global(smbd_memcache());
1088
1089         /* Initialise the password backed before the global_sam_sid
1090            to ensure that we fetch from ldap before we make a domain sid up */
1091
1092         if(!initialize_password_db(False, smbd_event_context()))
1093                 exit(1);
1094
1095         if (!secrets_init()) {
1096                 DEBUG(0, ("ERROR: smbd can not open secrets.tdb\n"));
1097                 exit(1);
1098         }
1099
1100         if (lp_server_role() == ROLE_DOMAIN_BDC || lp_server_role() == ROLE_DOMAIN_PDC) {
1101                 if (!open_schannel_session_store(NULL, lp_private_dir())) {
1102                         DEBUG(0,("ERROR: Samba cannot open schannel store for secured NETLOGON operations.\n"));
1103                         exit(1);
1104                 }
1105         }
1106
1107         if(!get_global_sam_sid()) {
1108                 DEBUG(0,("ERROR: Samba cannot create a SAM SID.\n"));
1109                 exit(1);
1110         }
1111
1112         if (!sessionid_init()) {
1113                 exit(1);
1114         }
1115
1116         if (!connections_init(True))
1117                 exit(1);
1118
1119         if (!locking_init())
1120                 exit(1);
1121
1122         if (!messaging_tdb_parent_init()) {
1123                 exit(1);
1124         }
1125
1126         if (!notify_internal_parent_init()) {
1127                 exit(1);
1128         }
1129
1130         if (!serverid_parent_init()) {
1131                 exit(1);
1132         }
1133
1134         if (!W_ERROR_IS_OK(registry_init_full()))
1135                 exit(1);
1136
1137         if (!print_backend_init(smbd_messaging_context()))
1138                 exit(1);
1139
1140 #if 0
1141         if (!init_svcctl_db())
1142                 exit(1);
1143 #endif
1144
1145         /* Open the share_info.tdb here, so we don't have to open
1146            after the fork on every single connection.  This is a small
1147            performance improvment and reduces the total number of system
1148            fds used. */
1149         if (!share_info_db_init()) {
1150                 DEBUG(0,("ERROR: failed to load share info db.\n"));
1151                 exit(1);
1152         }
1153
1154         if (!init_system_info()) {
1155                 DEBUG(0,("ERROR: failed to setup system user info.\n"));
1156                 return -1;
1157         }
1158
1159         if (!init_guest_info()) {
1160                 DEBUG(0,("ERROR: failed to setup guest info.\n"));
1161                 return -1;
1162         }
1163
1164         /*
1165          * Initialize spoolss with an init function to convert printers first.
1166          * static_init_rpc will try to initialize the spoolss server too but you
1167          * can't register it twice.
1168          */
1169         spoolss_cb.init = spoolss_init_cb;
1170         spoolss_cb.shutdown = NULL;
1171         spoolss_cb.private_data = smbd_server_conn->msg_ctx;
1172
1173         /* Spoolss depends on a winreg pipe, so start it first. */
1174         if (!NT_STATUS_IS_OK(rpc_winreg_init(NULL))) {
1175                 exit(1);
1176         }
1177
1178         if (!NT_STATUS_IS_OK(rpc_spoolss_init(&spoolss_cb))) {
1179                 exit(1);
1180         }
1181
1182         static_init_rpc;
1183
1184         /* Publish nt printers, this requires a working winreg pipe */
1185         reload_printers(smbd_messaging_context());
1186
1187         /* only start the background queue daemon if we are 
1188            running as a daemon -- bad things will happen if
1189            smbd is launched via inetd and we fork a copy of 
1190            ourselves here */
1191
1192         if (is_daemon && !interactive
1193             && lp_parm_bool(-1, "smbd", "backgroundqueue", true)) {
1194                 start_background_queue(smbd_event_context(),
1195                                        smbd_messaging_context());
1196         }
1197
1198         if (!is_daemon) {
1199                 /* inetd mode */
1200                 TALLOC_FREE(frame);
1201
1202                 /* Started from inetd. fd 0 is the socket. */
1203                 /* We will abort gracefully when the client or remote system
1204                    goes away */
1205                 smbd_set_server_fd(dup(0));
1206
1207                 /* close our standard file descriptors */
1208                 close_low_fds(False); /* Don't close stderr */
1209
1210 #ifdef HAVE_ATEXIT
1211                 atexit(killkids);
1212 #endif
1213
1214                 /* Stop zombies */
1215                 smbd_setup_sig_chld_handler();
1216
1217                 smbd_process(smbd_server_conn);
1218
1219                 exit_server_cleanly(NULL);
1220                 return(0);
1221         }
1222
1223         parent = talloc_zero(smbd_event_context(), struct smbd_parent_context);
1224         if (!parent) {
1225                 exit_server("talloc(struct smbd_parent_context) failed");
1226         }
1227         parent->interactive = interactive;
1228
1229         if (!open_sockets_smbd(parent, ports))
1230                 exit_server("open_sockets_smbd() failed");
1231
1232         TALLOC_FREE(frame);
1233         /* make sure we always have a valid stackframe */
1234         frame = talloc_stackframe();
1235
1236         smbd_parent_loop(parent);
1237
1238         exit_server_cleanly(NULL);
1239         TALLOC_FREE(frame);
1240         return(0);
1241 }