s3:smbd: remove brl_register_msgs()
[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         messaging_register(msg_ctx, NULL, MSG_SMB_BRL_VALIDATE,
769                            brl_revalidate);
770
771         msg_idmap_register_msg(msg_ctx);
772
773 #ifdef CLUSTER_SUPPORT
774         if (lp_clustering()) {
775                 ctdbd_register_reconfigure(messaging_ctdbd_connection());
776         }
777 #endif
778
779 #ifdef DEVELOPER
780         messaging_register(msg_ctx, NULL, MSG_SMB_INJECT_FAULT,
781                            msg_inject_fault);
782 #endif
783
784         if (lp_multicast_dns_register() && (dns_port != 0)) {
785 #ifdef WITH_DNSSD_SUPPORT
786                 smbd_setup_mdns_registration(ev_ctx,
787                                              parent, dns_port);
788 #endif
789 #ifdef WITH_AVAHI_SUPPORT
790                 void *avahi_conn;
791
792                 avahi_conn = avahi_start_register(ev_ctx,
793                                                   ev_ctx,
794                                                   dns_port);
795                 if (avahi_conn == NULL) {
796                         DEBUG(10, ("avahi_start_register failed\n"));
797                 }
798 #endif
799         }
800
801         return true;
802 }
803
804 static void smbd_parent_loop(struct tevent_context *ev_ctx,
805                              struct smbd_parent_context *parent)
806 {
807         /* now accept incoming connections - forking a new process
808            for each incoming connection */
809         DEBUG(2,("waiting for connections\n"));
810         while (1) {
811                 int ret;
812                 TALLOC_CTX *frame = talloc_stackframe();
813
814                 ret = tevent_loop_once(ev_ctx);
815                 if (ret != 0) {
816                         exit_server_cleanly("tevent_loop_once() error");
817                 }
818
819                 TALLOC_FREE(frame);
820         } /* end while 1 */
821
822 /* NOTREACHED   return True; */
823 }
824
825
826 /****************************************************************************
827  Initialise connect, service and file structs.
828 ****************************************************************************/
829
830 static bool init_structs(void )
831 {
832         /*
833          * Set the machine NETBIOS name if not already
834          * set from the config file.
835          */
836
837         if (!init_names())
838                 return False;
839
840         if (!secrets_init())
841                 return False;
842
843         return True;
844 }
845
846 static void smbd_parent_sig_term_handler(struct tevent_context *ev,
847                                          struct tevent_signal *se,
848                                          int signum,
849                                          int count,
850                                          void *siginfo,
851                                          void *private_data)
852 {
853         exit_server_cleanly("termination signal");
854 }
855
856 static void smbd_parent_sig_hup_handler(struct tevent_context *ev,
857                                         struct tevent_signal *se,
858                                         int signum,
859                                         int count,
860                                         void *siginfo,
861                                         void *private_data)
862 {
863         struct smbd_parent_context *parent =
864                 talloc_get_type_abort(private_data,
865                 struct smbd_parent_context);
866
867         change_to_root_user();
868         DEBUG(1,("parent: Reloading services after SIGHUP\n"));
869         reload_services(parent->msg_ctx, -1, false);
870
871         printing_subsystem_update(parent->ev_ctx, parent->msg_ctx, true);
872 }
873
874 /****************************************************************************
875  main program.
876 ****************************************************************************/
877
878 /* Declare prototype for build_options() to avoid having to run it through
879    mkproto.h.  Mixing $(builddir) and $(srcdir) source files in the current
880    prototype generation system is too complicated. */
881
882 extern void build_options(bool screen);
883
884  int main(int argc,const char *argv[])
885 {
886         /* shall I run as a daemon */
887         bool is_daemon = false;
888         bool interactive = false;
889         bool Fork = true;
890         bool no_process_group = false;
891         bool log_stdout = false;
892         char *ports = NULL;
893         char *profile_level = NULL;
894         int opt;
895         poptContext pc;
896         bool print_build_options = False;
897         enum {
898                 OPT_DAEMON = 1000,
899                 OPT_INTERACTIVE,
900                 OPT_FORK,
901                 OPT_NO_PROCESS_GROUP,
902                 OPT_LOG_STDOUT
903         };
904         struct poptOption long_options[] = {
905         POPT_AUTOHELP
906         {"daemon", 'D', POPT_ARG_NONE, NULL, OPT_DAEMON, "Become a daemon (default)" },
907         {"interactive", 'i', POPT_ARG_NONE, NULL, OPT_INTERACTIVE, "Run interactive (not a daemon)"},
908         {"foreground", 'F', POPT_ARG_NONE, NULL, OPT_FORK, "Run daemon in foreground (for daemontools, etc.)" },
909         {"no-process-group", '\0', POPT_ARG_NONE, NULL, OPT_NO_PROCESS_GROUP, "Don't create a new process group" },
910         {"log-stdout", 'S', POPT_ARG_NONE, NULL, OPT_LOG_STDOUT, "Log to stdout" },
911         {"build-options", 'b', POPT_ARG_NONE, NULL, 'b', "Print build options" },
912         {"port", 'p', POPT_ARG_STRING, &ports, 0, "Listen on the specified ports"},
913         {"profiling-level", 'P', POPT_ARG_STRING, &profile_level, 0, "Set profiling level","PROFILE_LEVEL"},
914         POPT_COMMON_SAMBA
915         POPT_COMMON_DYNCONFIG
916         POPT_TABLEEND
917         };
918         struct smbd_parent_context *parent = NULL;
919         TALLOC_CTX *frame;
920         NTSTATUS status;
921         uint64_t unique_id;
922         struct tevent_context *ev_ctx;
923         struct messaging_context *msg_ctx;
924         struct tevent_signal *se;
925
926         /*
927          * Do this before any other talloc operation
928          */
929         talloc_enable_null_tracking();
930         frame = talloc_stackframe();
931
932         setup_logging(argv[0], DEBUG_DEFAULT_STDOUT);
933
934         load_case_tables();
935
936         smbd_init_globals();
937
938         TimeInit();
939
940 #ifdef HAVE_SET_AUTH_PARAMETERS
941         set_auth_parameters(argc,argv);
942 #endif
943
944         pc = poptGetContext("smbd", argc, argv, long_options, 0);
945         while((opt = poptGetNextOpt(pc)) != -1) {
946                 switch (opt)  {
947                 case OPT_DAEMON:
948                         is_daemon = true;
949                         break;
950                 case OPT_INTERACTIVE:
951                         interactive = true;
952                         break;
953                 case OPT_FORK:
954                         Fork = false;
955                         break;
956                 case OPT_NO_PROCESS_GROUP:
957                         no_process_group = true;
958                         break;
959                 case OPT_LOG_STDOUT:
960                         log_stdout = true;
961                         break;
962                 case 'b':
963                         print_build_options = True;
964                         break;
965                 default:
966                         d_fprintf(stderr, "\nInvalid option %s: %s\n\n",
967                                   poptBadOption(pc, 0), poptStrerror(opt));
968                         poptPrintUsage(pc, stderr, 0);
969                         exit(1);
970                 }
971         }
972         poptFreeContext(pc);
973
974         if (interactive) {
975                 Fork = False;
976                 log_stdout = True;
977         }
978
979         if (log_stdout) {
980                 setup_logging(argv[0], DEBUG_STDOUT);
981         } else {
982                 setup_logging(argv[0], DEBUG_FILE);
983         }
984
985         if (print_build_options) {
986                 build_options(True); /* Display output to screen as well as debug */
987                 exit(0);
988         }
989
990 #ifdef HAVE_SETLUID
991         /* needed for SecureWare on SCO */
992         setluid(0);
993 #endif
994
995         set_remote_machine_name("smbd", False);
996
997         if (interactive && (DEBUGLEVEL >= 9)) {
998                 talloc_enable_leak_report();
999         }
1000
1001         if (log_stdout && Fork) {
1002                 DEBUG(0,("ERROR: Can't log to stdout (-S) unless daemon is in foreground (-F) or interactive (-i)\n"));
1003                 exit(1);
1004         }
1005
1006         /* we want to re-seed early to prevent time delays causing
1007            client problems at a later date. (tridge) */
1008         generate_random_buffer(NULL, 0);
1009
1010         /* get initial effective uid and gid */
1011         sec_init();
1012
1013         /* make absolutely sure we run as root - to handle cases where people
1014            are crazy enough to have it setuid */
1015         gain_root_privilege();
1016         gain_root_group_privilege();
1017
1018         fault_setup();
1019         dump_core_setup("smbd", lp_logfile());
1020
1021         /* we are never interested in SIGPIPE */
1022         BlockSignals(True,SIGPIPE);
1023
1024 #if defined(SIGFPE)
1025         /* we are never interested in SIGFPE */
1026         BlockSignals(True,SIGFPE);
1027 #endif
1028
1029 #if defined(SIGUSR2)
1030         /* We are no longer interested in USR2 */
1031         BlockSignals(True,SIGUSR2);
1032 #endif
1033
1034         /* POSIX demands that signals are inherited. If the invoking process has
1035          * these signals masked, we will have problems, as we won't recieve them. */
1036         BlockSignals(False, SIGHUP);
1037         BlockSignals(False, SIGUSR1);
1038         BlockSignals(False, SIGTERM);
1039
1040         /* Ensure we leave no zombies until we
1041          * correctly set up child handling below. */
1042
1043         CatchChild();
1044
1045         /* we want total control over the permissions on created files,
1046            so set our umask to 0 */
1047         umask(0);
1048
1049         reopen_logs();
1050
1051         DEBUG(0,("smbd version %s started.\n", samba_version_string()));
1052         DEBUGADD(0,("%s\n", COPYRIGHT_STARTUP_MESSAGE));
1053
1054         DEBUG(2,("uid=%d gid=%d euid=%d egid=%d\n",
1055                  (int)getuid(),(int)getgid(),(int)geteuid(),(int)getegid()));
1056
1057         /* Output the build options to the debug log */ 
1058         build_options(False);
1059
1060         if (sizeof(uint16) < 2 || sizeof(uint32) < 4) {
1061                 DEBUG(0,("ERROR: Samba is not configured correctly for the word size on your machine\n"));
1062                 exit(1);
1063         }
1064
1065         if (!lp_load_initial_only(get_dyn_CONFIGFILE())) {
1066                 DEBUG(0, ("error opening config file '%s'\n", get_dyn_CONFIGFILE()));
1067                 exit(1);
1068         }
1069
1070         /* Init the security context and global current_user */
1071         init_sec_ctx();
1072
1073         /*
1074          * Initialize the event context. The event context needs to be
1075          * initialized before the messaging context, cause the messaging
1076          * context holds an event context.
1077          * FIXME: This should be s3_tevent_context_init()
1078          */
1079         ev_ctx = server_event_context();
1080         if (ev_ctx == NULL) {
1081                 exit(1);
1082         }
1083
1084         /*
1085          * Init the messaging context
1086          * FIXME: This should only call messaging_init()
1087          */
1088         msg_ctx = server_messaging_context();
1089         if (msg_ctx == NULL) {
1090                 exit(1);
1091         }
1092
1093         /*
1094          * Reloading of the printers will not work here as we don't have a
1095          * server info and rpc services set up. It will be called later.
1096          */
1097         if (!reload_services(NULL, -1, False)) {
1098                 exit(1);
1099         }
1100
1101         /* ...NOTE... Log files are working from this point! */
1102
1103         DEBUG(3,("loaded services\n"));
1104
1105         init_structs();
1106
1107 #ifdef WITH_PROFILE
1108         if (!profile_setup(msg_ctx, False)) {
1109                 DEBUG(0,("ERROR: failed to setup profiling\n"));
1110                 return -1;
1111         }
1112         if (profile_level != NULL) {
1113                 int pl = atoi(profile_level);
1114                 struct server_id src;
1115
1116                 DEBUG(1, ("setting profiling level: %s\n",profile_level));
1117                 src.pid = getpid();
1118                 set_profile_level(pl, src);
1119         }
1120 #endif
1121
1122         if (!is_daemon && !is_a_socket(0)) {
1123                 if (!interactive)
1124                         DEBUG(0,("standard input is not a socket, assuming -D option\n"));
1125
1126                 /*
1127                  * Setting is_daemon here prevents us from eventually calling
1128                  * the open_sockets_inetd()
1129                  */
1130
1131                 is_daemon = True;
1132         }
1133
1134         if (is_daemon && !interactive) {
1135                 DEBUG( 3, ( "Becoming a daemon.\n" ) );
1136                 become_daemon(Fork, no_process_group, log_stdout);
1137         }
1138
1139         generate_random_buffer((uint8_t *)&unique_id, sizeof(unique_id));
1140         set_my_unique_id(unique_id);
1141
1142 #if HAVE_SETPGID
1143         /*
1144          * If we're interactive we want to set our own process group for
1145          * signal management.
1146          */
1147         if (interactive && !no_process_group)
1148                 setpgid( (pid_t)0, (pid_t)0);
1149 #endif
1150
1151         if (!directory_exist(lp_lockdir()))
1152                 mkdir(lp_lockdir(), 0755);
1153
1154         if (is_daemon)
1155                 pidfile_create("smbd");
1156
1157         status = reinit_after_fork(msg_ctx,
1158                                    ev_ctx,
1159                                    false);
1160         if (!NT_STATUS_IS_OK(status)) {
1161                 DEBUG(0,("reinit_after_fork() failed\n"));
1162                 exit(1);
1163         }
1164
1165         smbd_server_conn->msg_ctx = msg_ctx;
1166
1167         parent = talloc_zero(ev_ctx, struct smbd_parent_context);
1168         if (!parent) {
1169                 exit_server("talloc(struct smbd_parent_context) failed");
1170         }
1171         parent->interactive = interactive;
1172         parent->ev_ctx = ev_ctx;
1173         parent->msg_ctx = msg_ctx;
1174         am_parent = parent;
1175
1176         se = tevent_add_signal(parent->ev_ctx,
1177                                parent,
1178                                SIGTERM, 0,
1179                                smbd_parent_sig_term_handler,
1180                                parent);
1181         if (!se) {
1182                 exit_server("failed to setup SIGTERM handler");
1183         }
1184         se = tevent_add_signal(parent->ev_ctx,
1185                                parent,
1186                                SIGHUP, 0,
1187                                smbd_parent_sig_hup_handler,
1188                                parent);
1189         if (!se) {
1190                 exit_server("failed to setup SIGHUP handler");
1191         }
1192
1193         /* Setup all the TDB's - including CLEAR_IF_FIRST tdb's. */
1194
1195         if (smbd_memcache() == NULL) {
1196                 exit(1);
1197         }
1198
1199         memcache_set_global(smbd_memcache());
1200
1201         /* Initialise the password backed before the global_sam_sid
1202            to ensure that we fetch from ldap before we make a domain sid up */
1203
1204         if(!initialize_password_db(false, ev_ctx))
1205                 exit(1);
1206
1207         if (!secrets_init()) {
1208                 DEBUG(0, ("ERROR: smbd can not open secrets.tdb\n"));
1209                 exit(1);
1210         }
1211
1212         if (lp_server_role() == ROLE_DOMAIN_BDC || lp_server_role() == ROLE_DOMAIN_PDC) {
1213                 struct loadparm_context *lp_ctx = loadparm_init_s3(NULL, loadparm_s3_context());
1214                 if (!open_schannel_session_store(NULL, lp_ctx)) {
1215                         DEBUG(0,("ERROR: Samba cannot open schannel store for secured NETLOGON operations.\n"));
1216                         exit(1);
1217                 }
1218                 TALLOC_FREE(lp_ctx);
1219         }
1220
1221         if(!get_global_sam_sid()) {
1222                 DEBUG(0,("ERROR: Samba cannot create a SAM SID.\n"));
1223                 exit(1);
1224         }
1225
1226         if (!sessionid_init()) {
1227                 exit(1);
1228         }
1229
1230         if (!connections_init(True))
1231                 exit(1);
1232
1233         if (!locking_init())
1234                 exit(1);
1235
1236         if (!messaging_tdb_parent_init(ev_ctx)) {
1237                 exit(1);
1238         }
1239
1240         if (!notify_internal_parent_init(ev_ctx)) {
1241                 exit(1);
1242         }
1243
1244         if (!serverid_parent_init(ev_ctx)) {
1245                 exit(1);
1246         }
1247
1248         if (!W_ERROR_IS_OK(registry_init_full()))
1249                 exit(1);
1250
1251         /* Open the share_info.tdb here, so we don't have to open
1252            after the fork on every single connection.  This is a small
1253            performance improvment and reduces the total number of system
1254            fds used. */
1255         if (!share_info_db_init()) {
1256                 DEBUG(0,("ERROR: failed to load share info db.\n"));
1257                 exit(1);
1258         }
1259
1260         status = init_system_info();
1261         if (!NT_STATUS_IS_OK(status)) {
1262                 DEBUG(1, ("ERROR: failed to setup system user info: %s.\n",
1263                           nt_errstr(status)));
1264                 return -1;
1265         }
1266
1267         if (!init_guest_info()) {
1268                 DEBUG(0,("ERROR: failed to setup guest info.\n"));
1269                 return -1;
1270         }
1271
1272         if (!file_init(smbd_server_conn)) {
1273                 DEBUG(0, ("ERROR: file_init failed\n"));
1274                 return -1;
1275         }
1276
1277         /* This MUST be done before start_epmd() because otherwise
1278          * start_epmd() forks and races against dcesrv_ep_setup() to
1279          * call directory_create_or_exist() */
1280         if (!directory_create_or_exist(lp_ncalrpc_dir(), geteuid(), 0755)) {
1281                 DEBUG(0, ("Failed to create pipe directory %s - %s\n",
1282                           lp_ncalrpc_dir(), strerror(errno)));
1283                 return -1;
1284         }
1285
1286         if (is_daemon && !interactive) {
1287                 if (rpc_epmapper_daemon() == RPC_DAEMON_FORK) {
1288                         start_epmd(ev_ctx, msg_ctx);
1289                 }
1290         }
1291
1292         if (!dcesrv_ep_setup(ev_ctx, msg_ctx)) {
1293                 exit(1);
1294         }
1295
1296         /* only start other daemons if we are running as a daemon
1297          * -- bad things will happen if smbd is launched via inetd
1298          *  and we fork a copy of ourselves here */
1299         if (is_daemon && !interactive) {
1300
1301                 if (rpc_lsasd_daemon() == RPC_DAEMON_FORK) {
1302                         start_lsasd(ev_ctx, msg_ctx);
1303                 }
1304
1305                 if (!_lp_disable_spoolss() &&
1306                     (rpc_spoolss_daemon() != RPC_DAEMON_DISABLED)) {
1307                         bool bgq = lp_parm_bool(-1, "smbd", "backgroundqueue", true);
1308
1309                         if (!printing_subsystem_init(ev_ctx, msg_ctx, true, bgq)) {
1310                                 exit(1);
1311                         }
1312                 }
1313         } else if (!_lp_disable_spoolss() &&
1314                    (rpc_spoolss_daemon() != RPC_DAEMON_DISABLED)) {
1315                 if (!printing_subsystem_init(ev_ctx, msg_ctx, false, false)) {
1316                         exit(1);
1317                 }
1318         }
1319
1320         if (!is_daemon) {
1321                 /* inetd mode */
1322                 TALLOC_FREE(frame);
1323
1324                 /* Started from inetd. fd 0 is the socket. */
1325                 /* We will abort gracefully when the client or remote system
1326                    goes away */
1327                 smbd_server_conn->sock = dup(0);
1328
1329                 /* close our standard file descriptors */
1330                 if (!debug_get_output_is_stdout()) {
1331                         close_low_fds(False); /* Don't close stderr */
1332                 }
1333
1334 #ifdef HAVE_ATEXIT
1335                 atexit(killkids);
1336 #endif
1337
1338                 /* Stop zombies */
1339                 smbd_setup_sig_chld_handler(parent);
1340
1341                 smbd_process(ev_ctx, smbd_server_conn);
1342
1343                 exit_server_cleanly(NULL);
1344                 return(0);
1345         }
1346
1347         if (!open_sockets_smbd(parent, ev_ctx, msg_ctx, ports))
1348                 exit_server("open_sockets_smbd() failed");
1349
1350         /* do a printer update now that all messaging has been set up,
1351          * before we allow clients to start connecting */
1352         printing_subsystem_update(ev_ctx, msg_ctx, false);
1353
1354         TALLOC_FREE(frame);
1355         /* make sure we always have a valid stackframe */
1356         frame = talloc_stackframe();
1357
1358         smbd_parent_loop(ev_ctx, parent);
1359
1360         exit_server_cleanly(NULL);
1361         TALLOC_FREE(frame);
1362         return(0);
1363 }