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