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