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